X-Git-Url: https://gerrit.simantics.org/r/gitweb?p=simantics%2Fplatform.git;a=blobdiff_plain;f=bundles%2Forg.simantics.document.linking.ui%2Fsrc%2Forg%2Fsimantics%2Fdocument%2Flinking%2Freport%2FExportToPDF.java;fp=bundles%2Forg.simantics.document.linking.ui%2Fsrc%2Forg%2Fsimantics%2Fdocument%2Flinking%2Freport%2FExportToPDF.java;h=4b3873e0a3965998442ae6606b51ad5c2180fec7;hp=0000000000000000000000000000000000000000;hb=969bd23cab98a79ca9101af33334000879fb60c5;hpb=866dba5cd5a3929bbeae85991796acb212338a08

diff --git a/bundles/org.simantics.document.linking.ui/src/org/simantics/document/linking/report/ExportToPDF.java b/bundles/org.simantics.document.linking.ui/src/org/simantics/document/linking/report/ExportToPDF.java
new file mode 100644
index 000000000..4b3873e0a
--- /dev/null
+++ b/bundles/org.simantics.document.linking.ui/src/org/simantics/document/linking/report/ExportToPDF.java
@@ -0,0 +1,168 @@
+package org.simantics.document.linking.report;
+
+import java.awt.Font;
+import java.io.File;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.eclipse.core.runtime.IStatus;
+import org.eclipse.core.runtime.NullProgressMonitor;
+import org.eclipse.core.runtime.Status;
+import org.eclipse.core.runtime.jobs.Job;
+import org.simantics.Simantics;
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.Session;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.request.Read;
+import org.simantics.document.linking.Activator;
+import org.simantics.document.linking.report.html.HTMLDocument;
+import org.simantics.document.linking.report.pdf.PDFDocument;
+import org.simantics.document.linking.report.templates.CompleteStructureWriter;
+import org.simantics.document.linking.report.templates.DocumentStructureWriter;
+import org.simantics.document.linking.report.templates.ReportWriter;
+
+
+public class ExportToPDF {
+	public static final int DOCUMENT_BROWSE_STRCTURE = 0;
+	public static final int DOCUMENT_STRUCTURE = 1;
+	public static final int DIAGRAM_STRUCTURE = 2;
+	public static final int COMPLETE_STRUCTURE = 3;
+	
+	private Session session;
+	private Resource model;
+	
+	public ExportToPDF(Session session,Resource model) {
+		this.model = model;
+		this.session = session;
+	}
+	
+	public  void exportPDF(String filename, int report) {
+		ReportWriter<?> reportWriter = null;
+		switch (report) {
+		case DOCUMENT_BROWSE_STRCTURE:
+			
+			break;
+		case DOCUMENT_STRUCTURE:
+			reportWriter = new DocumentStructureWriter();
+			break;
+		case DIAGRAM_STRUCTURE:
+			
+			break;
+		case COMPLETE_STRUCTURE:
+			reportWriter = new CompleteStructureWriter();
+			break;
+		default:
+			break;
+		}
+		ExportJob job = new ExportJob(filename, reportWriter);
+		job.setUser(true);
+		job.schedule();
+		
+	}
+	
+	private class ExportJob extends Job {
+		
+		String filename;
+		ReportWriter<?> reportWriter;
+		
+		public ExportJob(String filename, ReportWriter<?> reportWriter) {
+			super(reportWriter.getName());
+			this.filename = filename;
+			this.reportWriter = reportWriter;
+		}
+		
+		@Override
+		protected IStatus run(IProgressMonitor monitor) {
+			return export(filename, reportWriter, monitor);
+		}
+	}
+	
+	public <T> IStatus export(String filename, ReportWriter<T> reportWriter, IProgressMonitor monitor) {
+		
+		Document lineWriter = null;
+		try {
+			if (filename.toLowerCase().endsWith(".pdf"))
+				lineWriter = new PDFDocument(filename, new Font("Arial", Font.PLAIN, 10),new Font("Arial", Font.BOLD, 10));
+			else if (filename.toLowerCase().endsWith(".html") ||
+					filename.toLowerCase().endsWith(".htm"))
+				lineWriter = new HTMLDocument(new File(filename));
+			else {
+				throw new Exception("File has unknow extension " + filename);
+			}
+			export(lineWriter, reportWriter, monitor);
+			
+		} catch (Exception e) {
+			return new Status(IStatus.ERROR, Activator.PLUGIN_ID, "Could not generate report", e);
+		} finally {
+			if (lineWriter != null) {
+				try {
+					lineWriter.close();
+				} catch (Exception e) {
+					e.printStackTrace();
+				}
+			}
+			
+		}
+		String status  = "Report generated.";
+//		if (lineWriter instanceof PDFTableWriter) {
+//			status += " Report size " + ((PDFTableWriter)lineWriter).getCurrentPageIndex() + " pages.";
+//		}
+		
+		return new Status(IStatus.OK, Activator.PLUGIN_ID, status);
+	}
+	
+	
+
+	
+	private <T> void export(final Document document, final ReportWriter<T> reportWriter, final IProgressMonitor monitor) throws DatabaseException{
+		session.syncRequest(new Read<Object>() {
+			@Override
+			public Object perform(ReadGraph graph) throws DatabaseException {
+				
+				try {
+					Map<Object, Object> context = new HashMap<Object, Object>();
+					context.put("model", model);
+					context.put(Document.class, document);
+					context.put(ReportWriter.class, reportWriter);
+					context.put("DocumentName", reportWriter.getName());
+					
+					reportWriter.start(graph, model, document, context);
+					
+					List<T> list = reportWriter.getReportItems(graph);
+					
+			        monitor.beginTask("Write Report", list.size());
+	
+			        T previous = null;
+			        T current = null;
+			        T next = null;
+			        for (int i = 0; i < list.size(); i++) {
+			        	current = list.get(i);
+			        	if (i > 0)
+			        		previous = list.get(i-1);
+			        	else
+			        		previous = null;
+			        	if (i < list.size()-1)
+			        		next = list.get(i+1);
+			        	else
+			        		next = null;
+			        	reportWriter.write(previous, current, next, document);
+			        	monitor.worked(1);
+			        }
+				} catch (Exception e) {
+					throw new DatabaseException(e);
+				}
+		        monitor.done();
+		        return null;
+ 			}
+		});
+		
+	}
+	
+	public static void exportDocumentToPDF(Resource model, String filename) {
+	    ExportToPDF exportPdf = new ExportToPDF(Simantics.getSession(), model);
+	    exportPdf.export(filename, new DocumentStructureWriter(), new NullProgressMonitor());
+	}
+}