bundles/org.simantics.modeling/src/org/simantics/modeling/scl/GraphModuleSourceRepository.java

   1 package org.simantics.modeling.scl;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collection;
   5
   6 import org.simantics.Simantics;
   7 import org.simantics.db.ReadGraph;
   8 import org.simantics.db.RequestProcessorSpecific;
   9 import org.simantics.db.Resource;
  10 import org.simantics.db.WriteGraph;
  11 import org.simantics.db.common.request.UnaryRead;
  12 import org.simantics.db.common.request.WriteRequest;
  13 import org.simantics.db.exception.DatabaseException;
  14 import org.simantics.db.procedure.SyncListener;
  15 import org.simantics.db.request.Read;
  16 import org.simantics.layer0.Layer0;
  17 import org.simantics.modeling.ModelingUtils;
  18 import org.simantics.scl.compiler.module.repository.UpdateListener;
  19 import org.simantics.scl.compiler.module.repository.UpdateListener.Observable;
  20 import org.simantics.scl.compiler.source.ModuleSource;
  21 import org.simantics.scl.compiler.source.StringModuleSource;
  22 import org.simantics.scl.compiler.source.repository.ModuleSourceRepository;
  23 import org.simantics.scl.runtime.SCLContext;
  24 import org.simantics.scl.runtime.tuple.Tuple0;
  25 import org.simantics.structural2.utils.StructuralUtils;
  26
  27 import gnu.trove.procedure.TObjectProcedure;
  28 import gnu.trove.set.hash.THashSet;
  29
  30 public enum GraphModuleSourceRepository implements ModuleSourceRepository {
  31     INSTANCE;
  32
  33     @Override
  34     public ModuleSource getModuleSource(final String moduleName, UpdateListener listener) {
  35         if(!moduleName.startsWith("http://"))
  36             return null;
  37
  38         Object graph = SCLContext.getCurrent().get("graph");
  39         RequestProcessorSpecific requestProcessor;
  40         if(graph instanceof ReadGraph)
  41             requestProcessor = (ReadGraph)graph;
  42         else
  43             requestProcessor = Simantics.getSession();
  44
  45         Read<ModuleSource> request = new ReadModuleSource(moduleName);
  46
  47         try {
  48             if(listener != null)
  49                 return requestProcessor.syncRequest(request, new ModuleListener(listener, moduleName));
  50             else
  51                 return requestProcessor.syncRequest(request);
  52         } catch (DatabaseException e) {
  53             e.printStackTrace();
  54             return null;
  55         }
  56     }
  57
  58     static class ModuleListener implements SyncListener<ModuleSource>, Observable {
  59         UpdateListener listener;
  60         boolean alreadyExecutedOnce;
  61         final String moduleName;
  62         public ModuleListener(UpdateListener listener, String moduleName) {
  63             this.listener = listener;
  64             this.listener.addObservable(this);
  65             this.moduleName = moduleName;
  66         }
  67         @Override
  68         public void removeListener(UpdateListener listener) {
  69             listener = null;
  70         }
  71         @Override
  72         public boolean isDisposed() {
  73             return listener == null;
  74         }
  75         private void fireUpdate(ReadGraph graph) {
  76             if(listener != null) {
  77                 SCLContext context = SCLContext.getCurrent();
  78                 Object oldGraph = context.put("graph", graph);
  79                 try {
  80                     listener.notifyAboutUpdate();
  81                 } finally {
  82                     listener = null;
  83                     context.put("graph", oldGraph);
  84                 }
  85             }
  86         }
  87         @Override
  88         public void execute(ReadGraph graph, ModuleSource result)
  89                 throws DatabaseException {
  90             if(alreadyExecutedOnce)
  91                 fireUpdate(graph);
  92             else
  93                 alreadyExecutedOnce = true;
  94         }
  95         @Override
  96         public void exception(ReadGraph graph, Throwable t)
  97                 throws DatabaseException {
  98             t.printStackTrace();
  99             if(alreadyExecutedOnce && listener != null)
 100                 fireUpdate(graph);
 101         }
 102     };
 103
 104     public static class GraphModuleSource extends StringModuleSource {
 105
 106         private final boolean immutable;
 107
 108         public GraphModuleSource(String moduleName, ClassLoader classLoader, String moduleText, boolean immutable) {
 109             super(moduleName, classLoader, moduleText);
 110             this.immutable = immutable;
 111         }
 112
 113         @Override
 114         public boolean isUpdateable() {
 115             return !immutable;
 116         }
 117
 118         @Override
 119         public void update(String newSourceText) {
 120             try {
 121                 Simantics.getSession().syncRequest(new WriteModuleSource(getModuleName(), newSourceText));
 122             } catch (DatabaseException e) {
 123                 e.printStackTrace();
 124             }
 125         }
 126     }
 127
 128     static class ReadModuleSource extends UnaryRead<String, ModuleSource> {
 129         public ReadModuleSource(String moduleName) {
 130             super(moduleName);
 131         }
 132
 133         @Override
 134         public ModuleSource perform(ReadGraph graph) throws DatabaseException {
 135             Resource moduleResource = graph.getPossibleResource(parameter);
 136             if(moduleResource == null)
 137                 return null;
 138             Layer0 L0 = Layer0.getInstance(graph);
 139             if(!graph.isInstanceOf(moduleResource, L0.SCLModule))
 140                 return null;
 141             String text = graph.getRelatedValue(moduleResource, L0.SCLModule_definition);
 142             boolean immutable = StructuralUtils.isImmutable(graph, moduleResource);
 143             return new GraphModuleSource(parameter, getClass().getClassLoader(), text, immutable);
 144         }
 145     }
 146
 147     static class WriteModuleSource extends WriteRequest {
 148         private final String moduleURI;
 149         private final String sourceText;
 150
 151         public WriteModuleSource(String moduleURI, String sourceText) {
 152             this.moduleURI = moduleURI;
 153             this.sourceText = sourceText;
 154         }
 155
 156         @Override
 157         public void perform(WriteGraph graph) throws DatabaseException {
 158             Resource moduleResource = graph.getPossibleResource(moduleURI);
 159             if(moduleResource == null)
 160                 return;
 161             Layer0 L0 = Layer0.getInstance(graph);
 162             if(!graph.isInstanceOf(moduleResource, L0.SCLModule))
 163                 return;
 164             graph.claimLiteral(moduleResource, L0.SCLModule_definition, sourceText);
 165         }
 166     }
 167
 168     @Override
 169     public void forAllModules(TObjectProcedure<String> procedure) {
 170         THashSet<String> moduleURIs;
 171         try {
 172             moduleURIs = Simantics.getSession().syncRequest(new Read<THashSet<String>>() {
 173                 @Override
 174                 public THashSet<String> perform(ReadGraph graph)
 175                         throws DatabaseException {
 176                     THashSet<String> result = new THashSet<String>();
 177                     Resource projectResource = Simantics.getProjectResource();
 178                     Layer0 L0 = Layer0.getInstance(graph);
 179                     for(Resource model : graph.getObjects(projectResource, L0.ConsistsOf)) {
 180                         if(graph.isInstanceOf(model, L0.IndexRoot)) {
 181                             for(Resource module : ModelingUtils.searchByType(graph, model, L0.SCLModule))
 182                                 result.add(graph.getURI(module));
 183                         }
 184                     }
 185
 186                     Collection<Resource> ontologies = Simantics.applySCL("Simantics/SharedOntologies", "getSharedOntologies", graph, Tuple0.INSTANCE);
 187                     for (Resource ontology : ontologies) {
 188                         for(Resource module : ModelingUtils.searchByType(graph, ontology, L0.SCLModule))
 189                             result.add(graph.getURI(module));
 190                     }
 191
 192                     return result;
 193                 }
 194             });
 195             moduleURIs.forEach(procedure);
 196         } catch (DatabaseException e) {
 197             e.printStackTrace();
 198         }
 199     }
 200
 201     @Override
 202     public Collection<String> getModuleNames() {
 203         ArrayList<String> result = new ArrayList<>();
 204         forAllModules((String name) -> {
 205             result.add(name);
 206             return true;
 207         });
 208         return result;
 209     }
 210 }