bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/pdf/PDFPainter.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.modeling.ui.pdf;
  13
  14 import java.util.concurrent.Semaphore;
  15 import java.util.concurrent.atomic.AtomicReference;
  16
  17 import org.simantics.db.ReadGraph;
  18 import org.simantics.db.Resource;
  19 import org.simantics.db.common.request.PossibleIndexRoot;
  20 import org.simantics.db.common.request.UniqueRead;
  21 import org.simantics.db.common.utils.NameUtils;
  22 import org.simantics.db.exception.DatabaseException;
  23 import org.simantics.db.exception.ValidationException;
  24 import org.simantics.db.layer0.variable.Variable;
  25 import org.simantics.db.layer0.variable.Variables;
  26 import org.simantics.db.management.ISessionContext;
  27 import org.simantics.diagram.elements.DiagramNodeUtil;
  28 import org.simantics.diagram.stubs.DiagramResource;
  29 import org.simantics.g2d.canvas.Hints;
  30 import org.simantics.g2d.canvas.impl.CanvasContext;
  31 import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;
  32 import org.simantics.modeling.requests.Node;
  33 import org.simantics.structural.stubs.StructuralResource2;
  34 import org.simantics.utils.DataContainer;
  35 import org.simantics.utils.datastructures.Pair;
  36 import org.simantics.utils.page.PageDesc;
  37 import org.simantics.utils.threads.IThreadWorkQueue;
  38 import org.simantics.utils.threads.ThreadUtils;
  39
  40 import com.lowagie.text.Rectangle;
  41 import com.lowagie.text.pdf.FontMapper;
  42 import com.lowagie.text.pdf.PdfWriter;
  43
  44 /**
  45  * Rasterizes diagram into a PDF document using the com.lowagie.text plug-in.
  46  *
  47  * @author Tuukka Lehtonen
  48  */
  49 public class PDFPainter {
  50
  51     public static boolean render(
  52             final IThreadWorkQueue thread,
  53             final ISessionContext sessionContext,
  54             PDFExportPlan exportModel,
  55             final Node node,
  56             final PdfWriter writer,
  57             final FontMapper mapper,
  58             final Rectangle pageSize,
  59             final PageDesc pageDesc,
  60             final boolean fitDiagramContentsToPageMargins,
  61             long timeout)
  62     throws InterruptedException, DatabaseException
  63     {
  64         final DataContainer<Boolean> result = new DataContainer<Boolean>(false);
  65         final DataContainer<DatabaseException> exception = new DataContainer<DatabaseException>();
  66
  67         final CanvasContext ctx = new CanvasContext(thread);
  68         final AtomicReference<ICanvasSceneGraphProvider> sgProvider = new AtomicReference<ICanvasSceneGraphProvider>();
  69
  70         try {
  71             final Semaphore done = new Semaphore(0);
  72             // IMPORTANT: Load diagram in a different thread than the canvas context thread!
  73             ThreadUtils.getBlockingWorkExecutor().execute(new Runnable() {
  74                 @Override
  75                 public void run() {
  76                     try {
  77                         Pair<Resource, String> modelAndRVI = sessionContext.getSession().syncRequest(new UniqueRead<Pair<Resource, String>>() {
  78                             @Override
  79                             public Pair<Resource, String> perform(ReadGraph graph) throws DatabaseException {
  80                                 return new Pair<Resource, String>( resolveModel(graph, node), resolveRVI(graph, node) );
  81                             }
  82                         });
  83
  84                         final Boolean isSymbol = sessionContext.getSession().syncRequest(new UniqueRead<Boolean>() {
  85                             @Override
  86                             public Boolean perform(ReadGraph graph) throws DatabaseException {
  87                                 StructuralResource2 STR = StructuralResource2.getInstance(graph);
  88                                 DiagramResource DIA = DiagramResource.getInstance(graph);
  89                                 Resource possibleSymbol = graph.getPossibleObject(node.getDiagramResource(), STR.Defines);
  90                                 return possibleSymbol != null && graph.isInstanceOf(possibleSymbol, DIA.ElementClass);
  91                             }
  92                         });
  93
  94                         ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(ctx, modelAndRVI.first, node.getDiagramResource(), modelAndRVI.second, 5000);
  95                         sgProvider.set( provider );
  96                         ctx.getDefaultHintContext().setHint(Hints.KEY_PAGE_DESC, pageDesc);
  97
  98 //                        StringBuilder b = new StringBuilder();
  99 //                        NodeUtil.printTreeNodes(ctx.getCanvasNode(), b);
 100 //                        System.err.println(b.toString());
 101
 102                         ThreadUtils.asyncExec(thread, new Runnable() {
 103                             @Override
 104                             public void run() {
 105                                 try {
 106                                     PDFBuilder chassis = new PDFBuilder(writer, mapper, pageSize, pageDesc, fitDiagramContentsToPageMargins || isSymbol);
 107
 108                                     chassis.paint(ctx, true);
 109                                 } finally {
 110                                     done.release();
 111                                 }
 112                             }
 113                         });
 114                     } catch (DatabaseException e) {
 115                         done.release();
 116                         exception.set(e);
 117                     } catch (Throwable e) {
 118                         done.release();
 119                         exception.set(new DatabaseException(e));
 120                     } finally {
 121                         done.release();
 122                     }
 123                 }
 124             });
 125
 126             done.acquire(2);
 127             if (exception.get() != null)
 128                 throw exception.get();
 129             return result.get();
 130         } finally {
 131             if (sgProvider.get() != null)
 132                 sgProvider.get().dispose();
 133             ctx.dispose();
 134         }
 135     }
 136
 137     private static Resource resolveModel(ReadGraph graph, Node node) throws DatabaseException {
 138         Resource composite = node.getDefiningResources().head();
 139         Resource model = graph.syncRequest(new PossibleIndexRoot(composite));
 140 //        Resource model = StructuralVariables.getModel(graph, composite);
 141         if (model == null)
 142             throw new ValidationException("no model found for composite " + NameUtils.getSafeName(graph, composite));
 143         return model;
 144     }
 145
 146 //    private static String resolveModelURI(ReadGraph graph, final Node node) throws DatabaseException {
 147 //        return graph.getURI(resolveModel(graph, node));
 148 //    }
 149
 150     private static String resolveRVI(ReadGraph graph, final Node node) throws DatabaseException {
 151         String RVI = node.getRVI();
 152         if(RVI != null) return RVI;
 153         Resource composite = node.getDefiningResources().head();
 154         Variable var = Variables.getVariable(graph, composite);
 155         org.simantics.db.layer0.variable.RVI rvi = var.getPossibleRVI(graph);
 156         if(rvi == null) return null;
 157         return rvi.toString();
 158 //        final ResourceArray compositePath = StructuralVariables.getCompositeArray(graph, composite);
 159 //        final ResourceArray variablePath = compositePath.removeFromBeginning(1);
 160 //        return StructuralVariables.getRVI(graph, variablePath);
 161     }
 162
 163 }