gerrit.simantics Code Review - simantics/sysdyn.git/blob

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2011 Association for Decentralized Information Management in\r
   3  * Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.sysdyn.ui.browser.contributions;\r
  13 \r
  14 import java.util.ArrayList;\r
  15 import java.util.Collection;\r
  16 import java.util.TreeMap;\r
  17 \r
  18 import org.simantics.browsing.ui.common.node.AbstractNode;\r
  19 import org.simantics.browsing.ui.graph.contributor.viewpoint.ViewpointContributor;\r
  20 import org.simantics.db.ReadGraph;\r
  21 import org.simantics.db.Resource;\r
  22 import org.simantics.db.common.request.ObjectsWithType;\r
  23 import org.simantics.db.common.utils.NameUtils;\r
  24 import org.simantics.db.exception.DatabaseException;\r
  25 import org.simantics.layer0.Layer0;\r
  26 import org.simantics.sysdyn.SysdynResource;\r
  27 import org.simantics.sysdyn.ui.browser.nodes.FunctionLibraryNode;\r
  28 import org.simantics.sysdyn.ui.browser.nodes.FunctionNode;\r
  29 import org.simantics.sysdyn.ui.browser.nodes.FunctionsFolder;\r
  30 import org.simantics.sysdyn.ui.browser.nodes.SharedFunctionsFolder;\r
  31 import org.simantics.utils.strings.AlphanumComparator;\r
  32 \r
  33 public class FunctionLibraries  extends ViewpointContributor<FunctionsFolder> {\r
  34 \r
  35         @Override\r
  36     public Collection<?> getContribution(ReadGraph graph, FunctionsFolder functionsFolder)\r
  37     throws DatabaseException {\r
  38 \r
  39         Layer0 l0 = Layer0.getInstance(graph);\r
  40         SysdynResource sr = SysdynResource.getInstance(graph);\r
  41         \r
  42         ArrayList<AbstractNode<Resource>> result = new ArrayList<AbstractNode<Resource>>();\r
  43         \r
  44                 TreeMap<String, Resource> sortResult = new TreeMap<String, Resource>(AlphanumComparator.CASE_INSENSITIVE_COMPARATOR);\r
  45         // Find and sort model functions\r
  46         for(Resource function : graph.syncRequest(new ObjectsWithType(functionsFolder.data, l0.ConsistsOf, sr.SysdynModelicaFunction))) {\r
  47                         sortResult.put(NameUtils.getSafeName(graph, function), function);\r
  48         }\r
  49                 for(Resource function : sortResult.values())\r
  50                         result.add(new FunctionNode(function));\r
  51         \r
  52         // Find and sort model function libraries\r
  53                 sortResult.clear();\r
  54         for(Resource functionLibrary : graph.syncRequest(new ObjectsWithType(functionsFolder.data, l0.ConsistsOf, sr.SysdynModelicaFunctionLibrary))) {\r
  55                         sortResult.put(NameUtils.getSafeName(graph, functionLibrary), functionLibrary);\r
  56                 }\r
  57                 for(Resource functionLibrary : sortResult.values())\r
  58                         result.add(new FunctionLibraryNode<Resource>(functionLibrary));\r
  59                 \r
  60                 // Find built-in functions\r
  61         Resource sysdyn = graph.getPossibleResource("http://www.simantics.org/Sysdyn-1.1");\r
  62         if(sysdyn != null) {\r
  63                 for(Resource library : graph.syncRequest(new ObjectsWithType(sysdyn, l0.ConsistsOf, sr.SysdynModelicaFunctionLibrary))) {\r
  64                          result.add(new FunctionLibraryNode<Resource>(library));\r
  65                 }\r
  66         }\r
  67         \r
  68         \r
  69         result.add(new SharedFunctionsFolder(functionsFolder.data));\r
  70         \r
  71         return result;\r
  72 \r
  73     }\r
  74 \r
  75     @Override\r
  76     public String getViewpointId() {\r
  77         return "Standard";\r
  78     }\r
  79     \r
  80 \r
  81 }\r