KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > columba > ristretto > coder > CharsetEncoderInputStream


1 /* ***** BEGIN LICENSE BLOCK *****
2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
3  *
4  * The contents of this file are subject to the Mozilla Public License Version
5  * 1.1 (the "License"); you may not use this file except in compliance with
6  * the License. You may obtain a copy of the License at
7  * http://www.mozilla.org/MPL/
8  *
9  * Software distributed under the License is distributed on an "AS IS" basis,
10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
11  * for the specific language governing rights and limitations under the
12  * License.
13  *
14  * The Original Code is Ristretto Mail API.
15  *
16  * The Initial Developers of the Original Code are
17  * Timo Stich and Frederik Dietz.
18  * Portions created by the Initial Developers are Copyright (C) 2004
19  * All Rights Reserved.
20  *
21  * Contributor(s):
22  *
23  * Alternatively, the contents of this file may be used under the terms of
24  * either the GNU General Public License Version 2 or later (the "GPL"), or
25  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
26  * in which case the provisions of the GPL or the LGPL are applicable instead
27  * of those above. If you wish to allow use of your version of this file only
28  * under the terms of either the GPL or the LGPL, and not to allow others to
29  * use your version of this file under the terms of the MPL, indicate your
30  * decision by deleting the provisions above and replace them with the notice
31  * and other provisions required by the GPL or the LGPL. If you do not delete
32  * the provisions above, a recipient may use your version of this file under
33  * the terms of any one of the MPL, the GPL or the LGPL.
34  *
35  * ***** END LICENSE BLOCK ***** */

36 package org.columba.ristretto.coder;
37
38 import java.io.FilterInputStream JavaDoc;
39 import java.io.IOException JavaDoc;
40 import java.io.InputStream JavaDoc;
41 import java.nio.ByteBuffer JavaDoc;
42 import java.nio.CharBuffer JavaDoc;
43 import java.nio.charset.Charset JavaDoc;
44 import java.nio.charset.CharsetEncoder JavaDoc;
45 import java.nio.charset.CoderResult JavaDoc;
46
47 /**
48  * FilterInputStream that encodes a characterstream into a bytestream.
49  *
50  * @author Timo Stich <tstich@users.sourceforge.net>
51  */

52 public class CharsetEncoderInputStream extends FilterInputStream JavaDoc {
53
54     private Charset JavaDoc charset;
55     private CharsetEncoder JavaDoc encoder;
56     
57     private CharBuffer JavaDoc inBytes;
58     private ByteBuffer JavaDoc outBytes;
59
60     private int pos;
61     private int available;
62
63     /**
64      * Constructs a CharsetEncoderInputStream.
65      *
66      * @param arg0 the characterstream
67      * @param charset the charset the characterstream is encoded in
68      */

69     public CharsetEncoderInputStream(InputStream JavaDoc arg0, Charset JavaDoc charset) {
70         super(arg0);
71         this.charset = charset;
72         encoder = charset.newEncoder();
73         
74         inBytes = CharBuffer.allocate(1);
75         outBytes = ByteBuffer.allocate(5);
76     }
77     
78     /**
79      * Read one character and encode it to the according charset.
80      *
81      * @return the encoded char
82      * @throws IOException
83      */

84     private int encodeNextCharacter() throws IOException JavaDoc {
85         int read = in.read();
86         
87         if( read == -1) return -1;
88         
89         inBytes.clear();
90         inBytes.put(0,(char) read);
91         
92         outBytes.clear();
93         CoderResult JavaDoc result = encoder.encode( inBytes, outBytes, in.available() == 0 );
94         
95         // If the character cannot be encoded add a dummy character
96
if( result.isMalformed() || result.isUnmappable() ) {
97             outBytes.put((byte)0);
98             return outBytes.limit();
99         }
100         
101         return outBytes.position();
102     }
103     
104     /**
105      * @see java.io.InputStream#read()
106      */

107     public int read() throws IOException JavaDoc {
108         if( pos == available ) {
109             available = encodeNextCharacter();
110             pos = 0;
111         }
112         
113         if( available == -1) return -1;
114         
115         return outBytes.get(pos++);
116     }
117     
118     /**
119      * @see java.io.InputStream#read(byte[], int, int)
120      */

121     public int read(byte[] arg0, int arg1, int arg2) throws IOException JavaDoc {
122         int next;
123         for( int i=0; i<arg2; i++) {
124             next = read();
125             if( next == -1 ) {
126                 if( i == 0 ) {
127                     return -1;
128                 } else {
129                     return i;
130                 }
131             }
132             arg0[arg1+i] = (byte) next;
133         }
134         return arg2;
135     }
136     
137
138 }
139
Popular Tags