2e882525a03df470905c87ad42a189df8f79ac0e
[mir.git] / source / mir / misc / PDFUtil.java
1 /*
2  * Copyright (C) 2001, 2002 The Mir-coders group
3  *
4  * This file is part of Mir.
5  *
6  * Mir is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * Mir is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with Mir; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  * In addition, as a special exception, The Mir-coders gives permission to link
21  * the code of this program with  any library licensed under the Apache Software License, 
22  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library 
23  * (or with modified versions of the above that use the same license as the above), 
24  * and distribute linked combinations including the two.  You must obey the 
25  * GNU General Public License in all respects for all of the code used other than 
26  * the above mentioned libraries.  If you modify this file, you may extend this 
27  * exception to your version of the file, but you are not obligated to do so.  
28  * If you do not wish to do so, delete this exception statement from your version.
29  */
30 package mir.misc;
31
32 import java.io.BufferedOutputStream;
33 import java.io.ByteArrayOutputStream;
34 import java.io.File;
35 import java.io.FileOutputStream;
36
37 import javax.servlet.http.HttpServletResponse;
38
39 import org.apache.fop.apps.Driver;
40 import org.apache.fop.apps.InputHandler;
41 import org.apache.fop.apps.XSLTInputHandler;
42 import org.apache.log.Hierarchy;
43 import org.apache.log.Logger;
44 import org.apache.log.Priority;
45 import org.xml.sax.XMLReader;
46
47 public class PDFUtil {
48
49   public static void makePDF(String foFilePath,Object pdfDestination,String stylesheetPath) throws Exception
50   {
51     try{
52       Driver driver = new Driver();
53
54       //stupid logging that fop wants to use, needs to be changed
55       Hierarchy hierarchy = Hierarchy.getDefaultHierarchy();
56       Logger fopLog=null;
57       fopLog = hierarchy.getLoggerFor("fop");
58       fopLog.setPriority(Priority.WARN);
59       driver.setLogger(fopLog);
60
61       driver.setRenderer(Driver.RENDER_PDF);
62
63       File foFile=new File(foFilePath);
64
65       String html2foStyleSheetPath;
66       html2foStyleSheetPath=stylesheetPath;
67
68       File html2foStyleSheet=new File(html2foStyleSheetPath);
69       InputHandler inputHandler =
70         new XSLTInputHandler(foFile, html2foStyleSheet);
71       XMLReader parser = inputHandler.getParser();
72
73       if (pdfDestination instanceof String) {
74         String filePath = (String) pdfDestination;
75         driver.setOutputStream(new BufferedOutputStream(new FileOutputStream(filePath)));
76         driver.render(parser, inputHandler.getInputSource());
77       }
78       else if (pdfDestination instanceof HttpServletResponse){
79         HttpServletResponse res = (HttpServletResponse) pdfDestination;
80         ByteArrayOutputStream out = new ByteArrayOutputStream();
81         driver.setOutputStream(out);
82         res.setContentType("application/pdf");
83
84         driver.render(parser, inputHandler.getInputSource());
85
86         byte[] content = out.toByteArray();
87         res.setContentLength(content.length);
88         res.getOutputStream().write(content);
89         res.getOutputStream().flush();
90       }
91       else {
92         throw new Exception("I'm sorry but I don't know how to output a pdf to an object of type" + pdfDestination.getClass().getName());
93       }
94     }
95
96     catch (Exception ex){
97         throw(ex);
98     }
99   }
100 }
101
102
103