facturas.PDF.InputHandler.java Source code

Java tutorial

Introduction

Here is the source code for facturas.PDF.InputHandler.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (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.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/* $Id: InputHandler.java 426576 2006-07-28 15:44:37Z jeremias $ */

package facturas.PDF;

// Imported java.io classes
import java.io.File;
import java.io.OutputStream;
import java.io.StringReader;
import java.util.Vector;

// Imported TraX classes
import javax.xml.transform.ErrorListener;
import javax.xml.transform.Result;
import javax.xml.transform.Source;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.sax.SAXResult;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.fop.apps.FOPException;
import org.apache.fop.apps.FOUserAgent;
import org.apache.fop.apps.Fop;
import org.apache.fop.apps.FopFactory;
import org.apache.fop.render.awt.viewer.Renderable;

/**
 * Class for handling files input from command line either with XML and XSLT
 * files (and optionally xsl parameters) or FO File input alone
 */
public class InputHandler implements ErrorListener, Renderable {

    private File sourcefile = null; // either FO or XML/XSLT usage

    private File stylesheet = null; // for XML/XSLT usage

    private Vector xsltParams = null; // for XML/XSLT usage

    private PDFGeneador _generador = null;

    /** the logger */
    protected Log log = LogFactory.getLog(InputHandler.class);

    /**
     * Constructor for XML->XSLT->FO input
     *
     * @param xmlfile
     *            XML file
     * @param xsltfile
     *            XSLT file
     * @param params
     *            Vector of command-line parameters (name, value, name, value,
     *            ...) for XSL stylesheet, null if none
     */
    public InputHandler(File xmlfile, File xsltfile, Vector params, PDFGeneador generador) {
        sourcefile = xmlfile;
        stylesheet = xsltfile;
        xsltParams = params;
        _generador = generador;
    }

    /**
     * Constructor for FO input
     *
     * @param fofile
     *            the file to read the FO document.
     */
    public InputHandler(File fofile, PDFGeneador generador) {
        sourcefile = fofile;
        _generador = generador;
    }

    /**
     * Generate a document, given an initialized Fop object
     *
     * @param userAgent
     *            the user agent
     * @param outputFormat
     *            the output format to generate (MIME type, see MimeConstants)
     * @param out
     *            the output stream to write the generated output to (may be
     *            null if not applicable)
     * @throws FOPException
     *             in case of an error during processing
     */
    public void renderTo(FOUserAgent userAgent, String outputFormat, OutputStream out) throws FOPException {

        FopFactory factory = userAgent.getFactory();
        Fop fop;
        if (out != null) {
            fop = factory.newFop(outputFormat, userAgent, out);
        } else {
            fop = factory.newFop(outputFormat, userAgent);
        }

        // if base URL was not explicitly set in FOUserAgent, obtain here
        if (fop.getUserAgent().getBaseURL() == null) {
            String baseURL = null;

            try {
                baseURL = new File(sourcefile.getAbsolutePath()).getParentFile().toURL().toExternalForm();
            } catch (Exception e) {
                baseURL = "";
            }
            fop.getUserAgent().setBaseURL(baseURL);
        }

        // Resulting SAX events (the generated FO) must be piped through to FOP
        Result res = new SAXResult(fop.getDefaultHandler());

        transformTo(res);
    }

    /** @see org.apache.fop.render.awt.viewer.Renderable */
    public void renderTo(FOUserAgent userAgent, String outputFormat) throws FOPException {
        renderTo(userAgent, outputFormat, null);
    }

    /**
     * In contrast to render(Fop) this method only performs the XSLT stage and
     * saves the intermediate XSL-FO file to the output file.
     *
     * @param out
     *            OutputStream to write the transformation result to.
     * @throws FOPException
     *             in case of an error during processing
     */
    public void transformTo(OutputStream out) throws FOPException {
        Result res = new StreamResult(out);
        transformTo(res);
    }

    /**
     * Transforms the input document to the input format expected by FOP using
     * XSLT.
     *
     * @param result
     *            the Result object where the result of the XSL transformation
     *            is sent to
     * @throws FOPException
     *             in case of an error during processing
     */
    protected void transformTo(Result result) throws FOPException {
        try {
            // Setup XSLT
            TransformerFactory factory = TransformerFactory.newInstance();
            Transformer transformer;

            if (stylesheet == null) { // FO Input
                transformer = factory.newTransformer();
            } else { // XML/XSLT input
                transformer = factory.newTransformer(new StreamSource(stylesheet));

                // Set the value of parameters, if any, defined for stylesheet
                if (xsltParams != null) {
                    for (int i = 0; i < xsltParams.size(); i += 2) {
                        transformer.setParameter((String) xsltParams.elementAt(i),
                                (String) xsltParams.elementAt(i + 1));
                    }
                }
            }
            transformer.setErrorListener(this);

            // Create a SAXSource from the input Source file
            // Source src = new StreamSource(sourcefile);

            Source src = new StreamSource(new StringReader(_generador.createString()));

            // Start XSLT transformation and FOP processing
            transformer.transform(src, result);

        } catch (Exception e) {
            throw new FOPException(e);
        }
    }

    // --- Implementation of the ErrorListener interface ---

    /**
     * @see javax.xml.transform.ErrorListener#warning(javax.xml.transform.TransformerException)
     */
    public void warning(TransformerException exc) {
        log.warn(exc.toString());
    }

    /**
     * @see javax.xml.transform.ErrorListener#error(javax.xml.transform.TransformerException)
     */
    public void error(TransformerException exc) {
        log.error(exc.toString());
    }

    /**
     * @see javax.xml.transform.ErrorListener#fatalError(javax.xml.transform.TransformerException)
     */
    public void fatalError(TransformerException exc) throws TransformerException {
        throw exc;
    }

}