01: /*
02: * Copyright 2006 Sun Microsystems, Inc. All Rights Reserved.
03: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
04: *
05: * This code is free software; you can redistribute it and/or modify it
06: * under the terms of the GNU General Public License version 2 only, as
07: * published by the Free Software Foundation. Sun designates this
08: * particular file as subject to the "Classpath" exception as provided
09: * by Sun in the LICENSE file that accompanied this code.
10: *
11: * This code is distributed in the hope that it will be useful, but WITHOUT
12: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13: * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14: * version 2 for more details (a copy is included in the LICENSE file that
15: * accompanied this code).
16: *
17: * You should have received a copy of the GNU General Public License version
18: * 2 along with this work; if not, write to the Free Software Foundation,
19: * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20: *
21: * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
22: * CA 95054 USA or visit www.sun.com if you need additional information or
23: * have any questions.
24: */
25:
26: package com.sun.codemodel.internal.writer;
27:
28: import java.io.FilterOutputStream;
29: import java.io.IOException;
30: import java.io.OutputStream;
31: import java.io.PrintStream;
32:
33: import com.sun.codemodel.internal.CodeWriter;
34: import com.sun.codemodel.internal.JPackage;
35:
36: /**
37: * Output all source files into a single stream with a little
38: * formatting header in front of each file.
39: *
40: * This is primarily for human consumption of the generated source
41: * code, such as to debug/test CodeModel or to quickly inspect the result.
42: *
43: * @author
44: * Kohsuke Kawaguchi (kohsuke.kawaguchi@sun.com)
45: */
46: public class SingleStreamCodeWriter extends CodeWriter {
47:
48: private final PrintStream out;
49:
50: /**
51: * @param os
52: * This stream will be closed at the end of the code generation.
53: */
54: public SingleStreamCodeWriter(OutputStream os) {
55: out = new PrintStream(os);
56: }
57:
58: public OutputStream openBinary(JPackage pkg, String fileName)
59: throws IOException {
60: String pkgName = pkg.name();
61: if (pkgName.length() != 0)
62: pkgName += '.';
63:
64: out.println("-----------------------------------" + pkgName
65: + fileName + "-----------------------------------");
66:
67: return new FilterOutputStream(out) {
68: public void close() {
69: // don't let this stream close
70: }
71: };
72: }
73:
74: public void close() throws IOException {
75: out.close();
76: }
77:
78: }
|