KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > com > quadcap > io > TeeWriter


1 package com.quadcap.io;
2
3 /* Copyright 2002 - 2003 Quadcap Software. All rights reserved.
4  *
5  * This software is distributed under the Quadcap Free Software License.
6  * This software may be used or modified for any purpose, personal or
7  * commercial. Open Source redistributions are permitted. Commercial
8  * redistribution of larger works derived from, or works which bundle
9  * this software requires a "Commercial Redistribution License"; see
10  * http://www.quadcap.com/purchase.
11  *
12  * Redistributions qualify as "Open Source" under one of the following terms:
13  *
14  * Redistributions are made at no charge beyond the reasonable cost of
15  * materials and delivery.
16  *
17  * Redistributions are accompanied by a copy of the Source Code or by an
18  * irrevocable offer to provide a copy of the Source Code for up to three
19  * years at the cost of materials and delivery. Such redistributions
20  * must allow further use, modification, and redistribution of the Source
21  * Code under substantially the same terms as this license.
22  *
23  * Redistributions of source code must retain the copyright notices as they
24  * appear in each source code file, these license terms, and the
25  * disclaimer/limitation of liability set forth as paragraph 6 below.
26  *
27  * Redistributions in binary form must reproduce this Copyright Notice,
28  * these license terms, and the disclaimer/limitation of liability set
29  * forth as paragraph 6 below, in the documentation and/or other materials
30  * provided with the distribution.
31  *
32  * The Software is provided on an "AS IS" basis. No warranty is
33  * provided that the Software is free of defects, or fit for a
34  * particular purpose.
35  *
36  * Limitation of Liability. Quadcap Software shall not be liable
37  * for any damages suffered by the Licensee or any third party resulting
38  * from use of the Software.
39  */
40
41 import java.io.IOException JavaDoc;
42 import java.io.Writer JavaDoc;
43
44 import java.util.HashMap JavaDoc;
45 import java.util.Iterator JavaDoc;
46
47 /**
48  *
49  * @author Stan Bailes
50  */
51 public class TeeWriter extends Writer JavaDoc {
52     class WriterBinding {
53         Writer JavaDoc w;
54         boolean passFlush = true;
55         boolean passClose = true;
56         boolean autoFlush = false;
57         boolean enabled = true;
58         WriterBinding(Writer JavaDoc w) { this.w = w; }
59     }
60     
61     HashMap JavaDoc writers = new HashMap JavaDoc();
62
63     /**
64      * Public constructor
65      */
66     public TeeWriter() {}
67
68     public TeeWriter(Writer JavaDoc w) { addWriter("", w); }
69
70     /**
71      * Write some characters to the buffer and the downstream writer
72      */
73     public void write(char[] buf, int off, int len) throws IOException JavaDoc {
74         Iterator JavaDoc iter = writers.values().iterator();
75         while (iter.hasNext()) {
76             WriterBinding wb = (WriterBinding)iter.next();
77             if (wb.enabled) {
78                 wb.w.write(buf, off, len);
79                 if (wb.autoFlush) {
80                     wb.w.flush();
81                 }
82             }
83         }
84     }
85
86     /**
87      * Flush (the downstream writer)
88      */
89     public void flush() throws IOException JavaDoc {
90         Iterator JavaDoc iter = writers.values().iterator();
91         while (iter.hasNext()) {
92             WriterBinding wb = (WriterBinding)iter.next();
93             if (wb.enabled && wb.passFlush) wb.w.flush();
94         }
95     }
96
97     /**
98      * Close (the downstream writer)
99      */
100     public void close() throws IOException JavaDoc {
101         Iterator JavaDoc iter = writers.values().iterator();
102         while (iter.hasNext()) {
103             WriterBinding wb = (WriterBinding)iter.next();
104             if (wb.enabled && wb.passClose) wb.w.close();
105         }
106     }
107
108     /**
109      * Add a new writer
110      */
111     public void addWriter(String JavaDoc name, Writer JavaDoc w) {
112         WriterBinding wb = new WriterBinding(w);
113         writers.put(name, wb);
114     }
115
116     /**
117      * Remove a writer
118      */
119     public void removeWriter(String JavaDoc name) {
120         writers.remove(name);
121     }
122
123     public void setPassClose(String JavaDoc w, boolean b) throws IOException JavaDoc {
124         getBinding(w).passClose = b;
125     }
126
127     public void setPassFlush(String JavaDoc w, boolean b) throws IOException JavaDoc {
128         getBinding(w).passFlush = b;
129     }
130         
131     public void setAutoFlush(String JavaDoc w, boolean b) throws IOException JavaDoc {
132         getBinding(w).autoFlush = b;
133     }
134
135     public void setEnabled(String JavaDoc w, boolean b) throws IOException JavaDoc {
136         getBinding(w).enabled = b;
137     }
138
139     final private WriterBinding getBinding(String JavaDoc w) throws IOException JavaDoc {
140         WriterBinding wb = (WriterBinding)writers.get(w);
141         if (wb == null) {
142             throw new IOException JavaDoc("No writer: " + w);
143         }
144         return wb;
145     }
146             
147 }
148
Popular Tags