com.lowagie.text.rtf.document.RtfDocument Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of rtf-gen Show documentation
Show all versions of rtf-gen Show documentation
RTF generation library derived from iText-2.1.7. All PDF features and RTF parsing are removed. Code is modernized for Java 8.
/*
* $Id: RtfDocument.java 3580 2008-08-06 15:52:00Z howard_s $
*
* 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:
* https://github.com/osobolev/rtf-gen
*/
package com.lowagie.text.rtf.document;
import com.lowagie.text.DocWriter;
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;
import java.io.IOException;
import java.io.OutputStream;
import java.util.HashSet;
import java.util.Set;
/**
* The RtfDocument stores all document related data and also the main data stream.
* INTERNAL CLASS - NOT TO BE USED DIRECTLY
*
* @author Mark Hall ([email protected])
* @author Todd Bush [Tab support]
* @author Thomas Bickel ([email protected])
* @version $Id: RtfDocument.java 3580 2008-08-06 15:52:00Z howard_s $
* @since 1.x
*/
public class RtfDocument extends RtfElement {
/**
* Stores the actual document data
*/
private RtfDataCache data;
/**
* The RtfMapper to use in this RtfDocument
*/
private final RtfMapper mapper;
/**
* The RtfDocumentHeader that handles all document header methods
*/
private final RtfDocumentHeader documentHeader;
/**
* Stores integers that have been generated as unique random numbers
*/
private final Set previousRandomInts = new HashSet<>();
/**
* Whether to automatically generate TOC entries for Chapters and Sections. Defaults to false
*/
private boolean autogenerateTOCEntries = false;
/**
* The RtfDocumentSettings for this RtfDocument.
*/
private final RtfDocumentSettings documentSettings;
/**
* 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 = DocWriter.getISOBytes("\\rtf1");
private static final byte[] FSC_LINE = DocWriter.getISOBytes("\\line ");
private static final byte[] FSC_PAR = DocWriter.getISOBytes("\\par ");
private static final byte[] FSC_TAB = DocWriter.getISOBytes("\\tab ");
private static final byte[] FSC_PAGE_PAR = DocWriter.getISOBytes("\\page\\par ");
private static final byte[] FSC_NEWPAGE = DocWriter.getISOBytes("$newpage$");
private static final byte[] FSC_BACKSLASH = DocWriter.getISOBytes("\\");
private static final byte[] FSC_HEX_PREFIX = DocWriter.getISOBytes("\\'");
private static final byte[] FSC_UNI_PREFIX = DocWriter.getISOBytes("\\u");
/**
* 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.documentSettings = new RtfDocumentSettings(this);
}
/**
* unused
*/
@Override
public void writeContent(OutputStream out) {
}
/**
* Writes the document
*
* @param out The OutputStream to write the RTF document to.
*/
public void writeDocument(OutputStream out) throws IOException {
out.write(OPEN_GROUP);
out.write(RTF_DOCUMENT);
this.documentHeader.writeContent(out);
this.data.writeTo(out);
out.write(CLOSE_GROUP);
}
/**
* Opens the RtfDocument and initializes the data cache. If the data cache is
* set to CACHE_DISK, but the cache cannot be initialized 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) {
log("Could not initialize disk cache. Using memory cache.", ioe);
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) {
log(ioe);
}
}
/**
* 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.
* Will not return a number between -1 and -5 because some values in that range are invalid.
*
* @return A random int
*/
public int getRandomInt() {
while (true){
int newInt = (int) (Math.random() * Integer.MAX_VALUE);
if (this.previousRandomInts.add(newInt))
return newInt;
}
}
/**
* Gets the RtfDocumentHeader of this RtfDocument
*
* @return The RtfDocumentHeader of this RtfDocument
*/
public RtfDocumentHeader getDocumentHeader() {
return this.documentHeader;
}
/**
* 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 true hex encoding characters is preferred to unicode encoding if possible
* @param softLineBreaks if true return characters are written as soft line breaks
* @throws IOException
*/
public void filterSpecialChar(OutputStream out, String str, boolean useHex, boolean softLineBreaks) throws IOException {
if (out == null) {
throw new NullPointerException("null OutpuStream");
}
boolean alwaysUseUniCode = this.documentSettings.isAlwaysUseUnicode();
if (str == null) {
return;
}
int len = str.length();
if (len == 0) {
return;
}
for (int k = 0; k < len; k++) {
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)) {
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 true if m.length characters in str, starting at offset soff
* match the bytes in the given array m.
*
* @param str the string to search for a match
* @param soff the starting offset in str
* @param m the array to match
* @return true if there is match
*/
private static boolean subMatch(String str, int soff, byte[] m) {
for (byte b : m) {
if (str.charAt(soff++) != b) {
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 automatically generate table of contents entries
*
* @return Whether 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;
}
/**
* Helper method outputs linebreak in document if debugging is turned on.
*
* @param result the OutputStream to write the linebreak to.
* @throws IOException
* @since 2.1.3
*/
public final void outputDebugLinebreak(OutputStream result) throws IOException {
if (this.getDocumentSettings().isOutputDebugLineBreaks()) {
result.write('\n');
}
}
public final void log(String message, Throwable error) {
RtfLogger logger = getDocumentSettings().getLogger();
logger.warn(message);
logger.error(error);
}
public final void log(Throwable error) {
RtfLogger logger = getDocumentSettings().getLogger();
logger.error(error);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy