/* * $Id: RtfDocument.java 2996 2007-11-20 22:40:36Z hallm $ * $Name: $ * * Copyright 2003, 2004, 2005 by Mark Hall * * The contents of this file are subject to the Mozilla Public License Version 1.1 * (the "License"); you may not use this file except in compliance with the License. * You may obtain a copy of the License at http://www.mozilla.org/MPL/ * * Software distributed under the License is distributed on an "AS IS" basis, * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License * for the specific language governing rights and limitations under the License. * * The Original Code is 'iText, a free JAVA-PDF library'. * * The Initial Developer of the Original Code is Bruno Lowagie. Portions created by * the Initial Developer are Copyright (C) 1999, 2000, 2001, 2002 by Bruno Lowagie. * All Rights Reserved. * Co-Developer of the code is Paulo Soares. Portions created by the Co-Developer * are Copyright (C) 2000, 2001, 2002 by Paulo Soares. All Rights Reserved. * * Contributor(s): all the names of the contributors are added in the source code * where applicable. * * Alternatively, the contents of this file may be used under the terms of the * LGPL license (the ?GNU LIBRARY GENERAL PUBLIC LICENSE?), in which case the * provisions of LGPL are applicable instead of those above. If you wish to * allow use of your version of this file only under the terms of the LGPL * License and not to allow others to use your version of this file under * the MPL, indicate your decision by deleting the provisions above and * replace them with the notice and other provisions required by the LGPL. * If you do not delete the provisions above, a recipient may use your version * of this file under either the MPL or the GNU LIBRARY GENERAL PUBLIC LICENSE. * * This library is free software; you can redistribute it and/or modify it * under the terms of the MPL as stated above or under the terms of the GNU * Library General Public License as published by the Free Software Foundation; * either version 2 of the License, or any later version. * * This library 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 Library general Public License for more * details. * * If you didn't download this code from the following link, you should check if * you aren't using an obsolete version: * http://www.lowagie.com/iText/ */ package com.lowagie.text.rtf.document; import java.io.IOException; import java.io.OutputStream; import java.util.ArrayList; import com.lowagie.text.rtf.RtfBasicElement; import com.lowagie.text.rtf.RtfElement; import com.lowagie.text.rtf.RtfMapper; import com.lowagie.text.rtf.document.output.*; import com.lowagie.text.rtf.graphic.RtfImage; /** * The RtfDocument stores all document related data and also the main data stream. * INTERNAL CLASS - NOT TO BE USED DIRECTLY * * @version $Id: RtfDocument.java 2996 2007-11-20 22:40:36Z hallm $ * @author Mark Hall (mhall@edu.uni-klu.ac.at) * @author Todd Bush [Tab support] * @author Thomas Bickel (tmb99@inode.at) */ public class RtfDocument extends RtfElement { /** * Stores the actual document data */ private RtfDataCache data = null; /** * The RtfMapper to use in this RtfDocument */ private RtfMapper mapper = null; /** * The RtfDocumentHeader that handles all document header methods */ private RtfDocumentHeader documentHeader = null; /** * Stores integers that have been generated as unique random numbers */ private ArrayList previousRandomInts = null; /** * Whether to automatically generate TOC entries for Chapters and Sections. Defaults to false */ private boolean autogenerateTOCEntries = false; /** * The RtfDocumentSettings for this RtfDocument. */ private RtfDocumentSettings documentSettings = null; /** * The last RtfBasicElement that was added directly to the RtfDocument. */ private RtfBasicElement lastElementWritten = null; /** * Constant for the Rtf document start */ private static final byte[] RTF_DOCUMENT = "\\rtf1".getBytes(); private final static byte[] FSC_LINE = "\\line ".getBytes(); private final static byte[] FSC_PAR = "\\par ".getBytes(); private final static byte[] FSC_TAB = "\\tab ".getBytes(); private final static byte[] FSC_PAGE_PAR = "\\page\\par ".getBytes(); private final static byte[] FSC_NEWPAGE = "$newpage$".getBytes(); private final static byte[] FSC_BACKSLASH = "\\".getBytes(); private final static byte[] FSC_HEX_PREFIX = "\\\'".getBytes(); private final static byte[] FSC_UNI_PREFIX = "\\u".getBytes(); /** * The default constructor for a RtfDocument */ public RtfDocument() { super(null); this.data = new RtfMemoryCache(); this.mapper = new RtfMapper(this); this.documentHeader = new RtfDocumentHeader(this); this.documentHeader.init(); this.previousRandomInts = new ArrayList(); this.documentSettings = new RtfDocumentSettings(this); } /** * unused * @deprecated As of iText 2.0.6 or earlier, replaced by * {@link #writeContent(OutputStream)}, scheduled for removal at or after 2.1.0 */ public byte[] write() { return(new byte[0]); } /** * unused */ public void writeContent(final OutputStream out) throws IOException { } /** * Writes the document * * @param out The <code>OutputStream</code> to write the RTF document to. */ public void writeDocument(OutputStream out) { try { out.write(OPEN_GROUP); out.write(RtfDocument.RTF_DOCUMENT); //out.write(documentHeader.write()); this.documentHeader.writeContent(out); this.data.writeTo(out); out.write(CLOSE_GROUP); } catch(IOException ioe) { ioe.printStackTrace(); } } /** * Opens the RtfDocument and initialises the data cache. If the data cache is * set to CACHE_DISK, but the cache cannot be initialised then the memory cache * is used. */ public void open() { try { switch(this.documentSettings.getDataCacheStyle()) { case RtfDataCache.CACHE_MEMORY_EFFICIENT: this.data = new RtfEfficientMemoryCache(); break; case RtfDataCache.CACHE_MEMORY: this.data = new RtfMemoryCache(); break; case RtfDataCache.CACHE_DISK: this.data = new RtfDiskCache(); break; default: throw(new RuntimeException("unknown")); } } catch(IOException ioe) { System.err.println("Could not initialise disk cache. Using memory cache."); ioe.printStackTrace(); this.data = new RtfMemoryCache(); } } /** * Adds an element to the rtf document * * @param element The element to add */ public void add(RtfBasicElement element) { try { if(element instanceof RtfInfoElement) { this.documentHeader.addInfoElement((RtfInfoElement) element); } else { if(element instanceof RtfImage) { ((RtfImage) element).setTopLevelElement(true); } element.writeContent( this.data.getOutputStream() ); this.lastElementWritten = element; } } catch(IOException ioe) { ioe.printStackTrace(); } } /** * Gets the RtfMapper object of this RtfDocument * * @return The RtfMapper */ public RtfMapper getMapper() { return this.mapper; } /** * Generates a random integer that is unique with respect to the document. * * @return A random int */ public int getRandomInt() { Integer newInt = null; do { newInt = new Integer((int) (Math.random() * Integer.MAX_VALUE)); } while(this.previousRandomInts.contains(newInt)); this.previousRandomInts.add(newInt); return newInt.intValue(); } /** * Gets the RtfDocumentHeader of this RtfDocument * * @return The RtfDocumentHeader of this RtfDocument */ public RtfDocumentHeader getDocumentHeader() { return this.documentHeader; } /** * Replaces special characters with their unicode values * @param str The original <code>String</code> * @param useHex indicated if the hexadecimal value has to be used * @param softLineBreaks whether to use soft line breaks instead of default hard ones. * * @deprecated As of iText 2.0.6 or earlier, replaced by * {@link #filterSpecialChar(OutputStream, String, boolean, boolean)}, scheduled for removal at or after 2.1.0 * @return The converted String */ public String filterSpecialChar(String str, boolean useHex, boolean softLineBreaks) { if(str == null) { return ""; } int length = str.length(); int z = 'z'; StringBuffer ret = new StringBuffer(length); for (int i = 0; i < length; i++) { char ch = str.charAt(i); if (ch == '\\') { ret.append("\\\\"); } else if (ch == '\n') { if(softLineBreaks) { ret.append("\\line "); } else { ret.append("\\par "); } } else if (ch == '\t') { ret.append("\\tab "); } else if ((ch) > z && this.documentSettings.isAlwaysUseUnicode()) { if(useHex) { ret.append("\\\'").append(Long.toHexString(ch)); } else { ret.append("\\u").append((long) ch).append('?'); } } else { ret.append(ch); } } String s = ret.toString(); if(s.indexOf("$newpage$") >= 0) { String before = s.substring(0, s.indexOf("$newpage$")); String after = s.substring(s.indexOf("$newpage$") + 9); ret = new StringBuffer(before); ret.append("\\page\\par "); ret.append(after); return ret.toString(); } return s; } /** * Writes the given string to the given {@link OutputStream} encoding the string characters. * * @param out destination OutputStream * @param str string to write * @param useHex if <code>true</code> hex encoding characters is preferred to unicode encoding if possible * @param softLineBreaks if <code>true</code> return characters are written as soft line breaks * * @throws IOException */ public void filterSpecialChar(final OutputStream out, final String str, final boolean useHex, final boolean softLineBreaks) throws IOException { if(out == null) { throw(new NullPointerException("null OutpuStream")); } final boolean alwaysUseUniCode = this.documentSettings.isAlwaysUseUnicode(); if(str == null) { return; } final int len = str.length(); if(len == 0) { return; } for(int k = 0; k < len; k++) { final char c = str.charAt(k); if(c < 0x20) { //allow return and tab only if(c == '\n') { out.write(softLineBreaks ? FSC_LINE : FSC_PAR); } else if(c == '\t') { out.write(FSC_TAB); } else { out.write('?'); } } else if((c == '\\') || (c == '{') || (c == '}')) { //escape out.write(FSC_BACKSLASH); out.write(c); } else if((c == '$') && (len-k >= FSC_NEWPAGE.length) && subMatch(str, k, FSC_NEWPAGE)) { //"$newpage$" -> "\\page\\par " out.write(FSC_PAGE_PAR); k += FSC_NEWPAGE.length-1; } else { if((c > 0xff) || ((c > 'z') && alwaysUseUniCode)) { if(useHex && (c <= 0xff)) { //encode as 2 char hex string out.write(FSC_HEX_PREFIX); out.write(RtfImage.byte2charLUT, c*2, 2); } else { //encode as decimal, signed short value out.write(FSC_UNI_PREFIX); String s = Short.toString((short)c); for(int x = 0; x < s.length(); x++) { out.write(s.charAt(x)); } out.write('?'); } } else { out.write(c); } } } } /** * Returns <code>true</code> if <tt>m.length</tt> characters in <tt>str</tt>, starting at offset <tt>soff</tt> * match the bytes in the given array <tt>m</tt>. * * @param str the string to search for a match * @param soff the starting offset in str * @param m the array to match * @return <code>true</code> if there is match */ private static boolean subMatch(final String str, int soff, final byte[] m) { for(int k = 0; k < m.length; k++) { if(str.charAt(soff++) != m[k]) { return(false); } } return(true); } /** * Whether to automagically generate table of contents entries when * adding Chapters or Sections. * * @param autogenerate Whether to automatically generate TOC entries */ public void setAutogenerateTOCEntries(boolean autogenerate) { this.autogenerateTOCEntries = autogenerate; } /** * Get whether to autmatically generate table of contents entries * * @return Wheter to automatically generate TOC entries */ public boolean getAutogenerateTOCEntries() { return this.autogenerateTOCEntries; } /** * Gets the RtfDocumentSettings that specify how the rtf document is generated. * * @return The current RtfDocumentSettings. */ public RtfDocumentSettings getDocumentSettings() { return this.documentSettings; } /** * Gets the last RtfBasicElement that was directly added to the RtfDocument. * * @return The last RtfBasicElement that was directly added to the RtfDocument. */ public RtfBasicElement getLastElementWritten() { return this.lastElementWritten; } }