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

   1 /*******************************************************************************\r
   2  * Copyright (c) 2010 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.representation.expressions;\r
  13 \r
  14 import java.util.ArrayList;\r
  15 import java.util.Iterator;\r
  16 \r
  17 import org.simantics.objmap.annotations.GraphType;\r
  18 import org.simantics.objmap.annotations.RelatedValue;\r
  19 import org.simantics.sysdyn.SysdynResource;\r
  20 import org.simantics.sysdyn.representation.ArrayIndexes;\r
  21 import org.simantics.sysdyn.representation.Enumeration;\r
  22 import org.simantics.sysdyn.representation.IndependentVariable;\r
  23 import org.simantics.sysdyn.representation.utils.FormatUtils;\r
  24 import org.simantics.sysdyn.representation.utils.IndexUtils;\r
  25 \r
  26 \r
  27 @GraphType(SysdynResource.URIs.WithLookupExpression)\r
  28 public class WithLookupExpression extends Expression {\r
  29 \r
  30     @RelatedValue(SysdynResource.URIs.WithLookupExpression_lookup)\r
  31     private String lookupTable;\r
  32     @RelatedValue(SysdynResource.URIs.WithLookupExpression_expression)\r
  33     private String equation;\r
  34 \r
  35     @Override\r
  36     public String getDeclaration(IndependentVariable variable) {\r
  37         ArrayIndexes ai = variable.getArrayIndexes();\r
  38         ArrayList<Enumeration> enumerations = null;\r
  39         if(ai != null) \r
  40                 enumerations = ai.getEnumerations();\r
  41         \r
  42         String range = "";\r
  43         if(enumerations != null && enumerations.size() > 0) {\r
  44                 StringBuilder sb = new StringBuilder();\r
  45                 sb.append("[");\r
  46                 Iterator<Enumeration> iterator = enumerations.iterator();\r
  47                 while(iterator.hasNext()) {\r
  48                         sb.append(iterator.next().getName() + ".size");\r
  49                         if(iterator.hasNext()) {\r
  50                                 sb.append(", ");\r
  51                         }\r
  52                 }\r
  53                 sb.append("]");\r
  54                 range = sb.toString();\r
  55         }\r
  56         \r
  57         return "    " + variable.getType() + " " + variable.getName() + range + ";\n";\r
  58 \r
  59     }\r
  60 \r
  61     @Override\r
  62     public String getEquation(IndependentVariable variable) {\r
  63         String equation = FormatUtils.formatExpressionForModelica(variable, this.equation);\r
  64         String range = IndexUtils.rangeToIndexes(variable, this.getArrayRange());\r
  65 \r
  66         return \r
  67         "    " + variable.getName() + (range.equals("[:]") ? "" : range) + " = interpolate(" + equation + ", " + lookupTable + ");\n";\r
  68     }\r
  69 \r
  70 }\r