| /* |
| * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| package com.sun.tools.internal.ws.processor.util; |
| |
| import com.sun.tools.internal.ws.processor.generator.GeneratorException; |
| |
| import java.io.BufferedWriter; |
| import java.io.IOException; |
| import java.io.Writer; |
| import java.nio.charset.Charset; |
| import java.nio.charset.CharsetEncoder; |
| import java.text.MessageFormat; |
| |
| /** |
| * |
| * @author WS Development Team |
| */ |
| public class IndentingWriter extends BufferedWriter { |
| |
| private boolean beginningOfLine = true; |
| private int currentIndent = 0; |
| private int indentStep = 4; |
| |
| public IndentingWriter(Writer out) { |
| super(out); |
| } |
| |
| public IndentingWriter(Writer out,int step) { |
| this(out); |
| |
| if (indentStep < 0) { |
| throw new IllegalArgumentException("negative indent step"); |
| } |
| indentStep = step; |
| } |
| |
| public void write(int c) throws IOException { |
| checkWrite(); |
| super.write(c); |
| } |
| |
| public void write(char[] cbuf, int off, int len) throws IOException { |
| if (len > 0) { |
| checkWrite(); |
| } |
| super.write(cbuf, off, len); |
| } |
| |
| public void write(String s, int off, int len) throws IOException { |
| if (len > 0) { |
| checkWrite(); |
| } |
| super.write(s, off, len); |
| } |
| |
| public void newLine() throws IOException { |
| super.newLine(); |
| beginningOfLine = true; |
| } |
| |
| protected void checkWrite() throws IOException { |
| if (beginningOfLine) { |
| beginningOfLine = false; |
| int i = currentIndent; |
| while (i > 0) { |
| super.write(' '); |
| -- i; |
| } |
| } |
| } |
| |
| protected void indentIn() { |
| currentIndent += indentStep; |
| } |
| |
| protected void indentOut() { |
| currentIndent -= indentStep; |
| if (currentIndent < 0) { |
| currentIndent = 0; |
| } |
| } |
| |
| public void pI() { |
| indentIn(); |
| } |
| |
| public void pO() { |
| indentOut(); |
| } |
| |
| public void pI(int levels) { |
| for (int i = 0; i < levels; ++i) { |
| indentIn(); |
| } |
| } |
| |
| public void pO(int levels) { |
| for (int i = 0; i < levels; ++i) { |
| indentOut(); |
| } |
| } |
| |
| public void p(String s) throws IOException { |
| /* |
| int tabCount = 0; |
| for (int i = 0; i < s.length(); ++i) { |
| if (s.charAt(i) == '\t') { |
| ++tabCount; |
| indentIn(); |
| } |
| } |
| |
| String printStr = s.substring(tabCount); |
| */ |
| boolean canEncode = true; |
| |
| //bug fix: 4839636 |
| try{ |
| if(!canEncode(s)) { |
| canEncode = false; |
| } |
| } catch (Throwable t) { |
| |
| // there was some exception, what should we do? |
| // lets ignore it for now and proceed with the code generation! |
| } |
| |
| if(!canEncode) { |
| throw new GeneratorException( |
| "generator.indentingwriter.charset.cantencode", s); |
| } |
| write(s); |
| /* |
| while (tabCount-- > 0) { |
| indentOut(); |
| } |
| */ |
| } |
| |
| /** |
| * Check if encode can handle the chars in this string. |
| * |
| */ |
| protected boolean canEncode(String s) { |
| final CharsetEncoder encoder = |
| Charset.forName(System.getProperty("file.encoding")).newEncoder(); |
| char[] chars = s.toCharArray(); |
| for (int i=0; i<chars.length; i++) { |
| if(!encoder.canEncode(chars[i])) { |
| return false; |
| } |
| } |
| return true; |
| } |
| |
| public void p(String s1, String s2) throws IOException { |
| p(s1); |
| p(s2); |
| } |
| |
| public void p(String s1, String s2, String s3) throws IOException { |
| p(s1); |
| p(s2); |
| p(s3); |
| } |
| |
| public void p(String s1, String s2, String s3, String s4) throws IOException { |
| p(s1); |
| p(s2); |
| p(s3); |
| p(s4); |
| } |
| |
| public void p(String s1, String s2, String s3, String s4, String s5) throws IOException { |
| p(s1); |
| p(s2); |
| p(s3); |
| p(s4); |
| p(s5); |
| } |
| |
| public void pln() throws IOException { |
| newLine(); |
| } |
| |
| public void pln(String s) throws IOException { |
| p(s); |
| pln(); |
| } |
| |
| public void pln(String s1, String s2) throws IOException { |
| p(s1, s2); |
| pln(); |
| } |
| |
| public void pln(String s1, String s2, String s3) throws IOException { |
| p(s1, s2, s3); |
| pln(); |
| } |
| |
| public void pln(String s1, String s2, String s3, String s4) throws IOException { |
| p(s1, s2, s3, s4); |
| pln(); |
| } |
| |
| public void pln(String s1, String s2, String s3, String s4, String s5) throws IOException { |
| p(s1, s2, s3, s4, s5); |
| pln(); |
| } |
| |
| public void plnI(String s) throws IOException { |
| p(s); |
| pln(); |
| pI(); |
| } |
| |
| public void pO(String s) throws IOException { |
| pO(); |
| p(s); |
| } |
| |
| public void pOln(String s) throws IOException { |
| pO(s); |
| pln(); |
| } |
| |
| public void pOlnI(String s) throws IOException { |
| pO(s); |
| pln(); |
| pI(); |
| } |
| |
| public void p(Object o) throws IOException { |
| write(o.toString()); |
| } |
| |
| public void pln(Object o) throws IOException { |
| p(o.toString()); |
| pln(); |
| } |
| |
| public void plnI(Object o) throws IOException { |
| p(o.toString()); |
| pln(); |
| pI(); |
| } |
| |
| public void pO(Object o) throws IOException { |
| pO(); |
| p(o.toString()); |
| } |
| |
| public void pOln(Object o) throws IOException { |
| pO(o.toString()); |
| pln(); |
| } |
| |
| public void pOlnI(Object o) throws IOException { |
| pO(o.toString()); |
| pln(); |
| pI(); |
| } |
| |
| public void pM(String s) throws IOException { |
| int i = 0; |
| while (i < s.length()) { |
| int j = s.indexOf('\n', i); |
| if (j == -1) { |
| p(s.substring(i)); |
| break; |
| } else { |
| pln(s.substring(i, j)); |
| i = j + 1; |
| } |
| } |
| } |
| |
| public void pMln(String s) throws IOException { |
| pM(s); |
| pln(); |
| } |
| |
| public void pMlnI(String s) throws IOException { |
| pM(s); |
| pln(); |
| pI(); |
| } |
| |
| public void pMO(String s) throws IOException { |
| pO(); |
| pM(s); |
| } |
| |
| public void pMOln(String s) throws IOException { |
| pMO(s); |
| pln(); |
| } |
| |
| public void pF(String pattern, Object[] arguments) throws IOException { |
| pM(MessageFormat.format(pattern, arguments)); |
| } |
| |
| public void pFln(String pattern, Object[] arguments) throws IOException { |
| pF(pattern, arguments); |
| pln(); |
| } |
| } |