bundles/org.simantics.databoard/src/org/simantics/databoard/parser/unparsing/DataTypePrinter2.java

   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.databoard.parser.unparsing;
  13 \r
  14 import gnu.trove.map.hash.THashMap;\r
  15 import gnu.trove.procedure.TObjectObjectProcedure;\r
  16 import gnu.trove.set.hash.THashSet;\r
  17 \r
  18 import org.simantics.databoard.type.ArrayType;\r
  19 import org.simantics.databoard.type.BooleanType;\r
  20 import org.simantics.databoard.type.ByteType;\r
  21 import org.simantics.databoard.type.Component;\r
  22 import org.simantics.databoard.type.Datatype;\r
  23 import org.simantics.databoard.type.Datatype.Visitor;\r
  24 import org.simantics.databoard.type.DoubleType;\r
  25 import org.simantics.databoard.type.FloatType;\r
  26 import org.simantics.databoard.type.IntegerType;\r
  27 import org.simantics.databoard.type.LongType;\r
  28 import org.simantics.databoard.type.MapType;\r
  29 import org.simantics.databoard.type.OptionalType;\r
  30 import org.simantics.databoard.type.RecordType;\r
  31 import org.simantics.databoard.type.StringType;\r
  32 import org.simantics.databoard.type.UnionType;\r
  33 import org.simantics.databoard.type.VariantType;\r
  34
  35
  36 /**
  37  * @author Hannu Niemist&ouml;
  38  */
  39 public class DataTypePrinter2 {
  40
  41     StringBuilder sb;
  42     int indentation = 0;\r
  43     THashMap<Datatype, String> refs = new THashMap<Datatype, String>();
  44 \r
  45     private DataTypePrinter2(StringBuilder sb) {\r
  46         this.sb = sb;\r
  47     }\r
  48 \r
  49     void newLine() {\r
  50         sb.append('\n');\r
  51         for(int i=0;i<indentation;++i)\r
  52             sb.append("    ");\r
  53     }\r
  54     \r
  55     boolean useRef = true;\r
  56     Visitor<Object> printVisitor = new Visitor<Object>() {\r
  57 \r
  58         @Override\r
  59         public Object visit(ArrayType b) {\r
  60             useRef = true;\r
  61             sb.append("Array(");\r
  62             b.componentType.accept(printVisitor);\r
  63             sb.append(")");\r
  64             return null;\r
  65         }\r
  66 \r
  67         @Override\r
  68         public Object visit(BooleanType b) {\r
  69             useRef = true;\r
  70             sb.append("Boolean");\r
  71             return null;\r
  72         }\r
  73 \r
  74         @Override\r
  75         public Object visit(DoubleType b) {\r
  76             useRef = true;\r
  77             sb.append("Double");\r
  78             return null;\r
  79         }\r
  80 \r
  81         @Override\r
  82         public Object visit(FloatType b) {\r
  83             useRef = true;\r
  84             sb.append("Float");\r
  85             return null;\r
  86         }\r
  87 \r
  88         @Override\r
  89         public Object visit(IntegerType b) {\r
  90             useRef = true;\r
  91             sb.append("Integer");\r
  92             return null;\r
  93         }\r
  94 \r
  95         @Override\r
  96         public Object visit(ByteType b) {\r
  97             useRef = true;\r
  98             sb.append("Byte");\r
  99             return null;\r
 100         }\r
 101 \r
 102         @Override\r
 103         public Object visit(LongType b) {\r
 104             useRef = true;\r
 105             sb.append("Long");\r
 106             return null;\r
 107         }\r
 108 \r
 109         @Override\r
 110         public Object visit(OptionalType b) {\r
 111             useRef = true;\r
 112             sb.append("Optional(");\r
 113             b.componentType.accept(printVisitor);\r
 114             sb.append(")");\r
 115             return null;\r
 116         }\r
 117 \r
 118         @Override\r
 119         public Object visit(RecordType b) {\r
 120             if(refs.containsKey(b) && useRef)\r
 121                 sb.append(refs.get(b));\r
 122             else {\r
 123                 useRef = true;\r
 124                 sb.append("{");\r
 125                 ++indentation;\r
 126                 for(int i=0;i<b.getComponentCount();++i) {\r
 127                     if(i > 0)\r
 128                         sb.append(", ");\r
 129                     Component c = b.getComponent(i);\r
 130                     newLine();\r
 131                     sb.append(c.name);\r
 132                     sb.append(" : ");\r
 133                     c.type.accept(printVisitor);\r
 134                 }\r
 135                 --indentation;\r
 136                 newLine();\r
 137                 sb.append("}");\r
 138             }\r
 139             return null;\r
 140         }\r
 141 \r
 142         @Override\r
 143         public Object visit(StringType b) {\r
 144             useRef = true;\r
 145             sb.append("String");\r
 146             return null;\r
 147         }\r
 148 \r
 149         @Override\r
 150         public Object visit(UnionType b) {\r
 151             if(refs.containsKey(b) && useRef)\r
 152                 sb.append(refs.get(b));\r
 153             else {\r
 154                 useRef = true;\r
 155                 ++indentation;\r
 156                 for(int i=0;i<b.getComponentCount();++i) {\r
 157                     newLine();\r
 158                     sb.append("| ");\r
 159                     Component c = b.getComponent(i);\r
 160                     sb.append(c.name);\r
 161                     sb.append(" ");\r
 162                     c.type.accept(printVisitor);\r
 163                 }\r
 164                 --indentation;\r
 165             }\r
 166             return null;\r
 167         }\r
 168 \r
 169         @Override\r
 170         public Object visit(VariantType b) {\r
 171             useRef = true;\r
 172             sb.append("Variant");\r
 173             return null;\r
 174         }\r
 175 \r
 176         @Override\r
 177         public Object visit(MapType b) {\r
 178             useRef = true;\r
 179             sb.append("Map(");\r
 180             b.keyType.accept(printVisitor);\r
 181             sb.append(", ");\r
 182             b.valueType.accept(printVisitor);\r
 183             sb.append(")");\r
 184             return null;\r
 185         }\r
 186 \r
 187     };\r
 188 \r
 189     THashSet<Datatype> seen = new THashSet<Datatype>();\r
 190 \r
 191     private void collectRefs(Datatype dt) {\r
 192         if(dt instanceof RecordType || dt instanceof UnionType) {\r
 193             if(!seen.add(dt)) {\r
 194                 if(!refs.containsKey(dt))\r
 195                     refs.put(dt, "T" + refs.size());\r
 196                 return;\r
 197             }\r
 198         }\r
 199         dt.accept(refCollectVisitor);\r
 200     }\r
 201 \r
 202     Visitor<Object> refCollectVisitor = new Visitor<Object>() {\r
 203 \r
 204         @Override\r
 205         public Object visit(ArrayType b) {\r
 206             collectRefs(b.componentType);\r
 207             return null;\r
 208         }\r
 209 \r
 210         @Override\r
 211         public Object visit(BooleanType b) {\r
 212             return null;\r
 213         }\r
 214 \r
 215         @Override\r
 216         public Object visit(DoubleType b) {\r
 217             return null;\r
 218         }\r
 219 \r
 220         @Override\r
 221         public Object visit(FloatType b) {\r
 222             return null;\r
 223         }\r
 224 \r
 225         @Override\r
 226         public Object visit(IntegerType b) {\r
 227             return null;\r
 228         }\r
 229 \r
 230         @Override\r
 231         public Object visit(ByteType b) {\r
 232             return null;\r
 233         }\r
 234 \r
 235         @Override\r
 236         public Object visit(LongType b) {\r
 237             return null;\r
 238         }\r
 239 \r
 240         @Override\r
 241         public Object visit(OptionalType b) {\r
 242             collectRefs(b.componentType);\r
 243             return null;\r
 244         }\r
 245 \r
 246         @Override\r
 247         public Object visit(RecordType b) {\r
 248             for(Component c : b.getComponents())\r
 249                 collectRefs(c.type);\r
 250             return null;\r
 251         }\r
 252 \r
 253         @Override\r
 254         public Object visit(StringType b) {\r
 255             return null;\r
 256         }\r
 257 \r
 258         @Override\r
 259         public Object visit(UnionType b) {\r
 260             for(Component c : b.getComponents())\r
 261                 collectRefs(c.type);\r
 262             return null;\r
 263         }\r
 264 \r
 265         @Override\r
 266         public Object visit(VariantType b) {\r
 267             return null;\r
 268         }\r
 269 \r
 270         @Override\r
 271         public Object visit(MapType b) {\r
 272             collectRefs(b.keyType);\r
 273             collectRefs(b.valueType);\r
 274             return null;\r
 275         }\r
 276 \r
 277     };\r
 278 \r
 279     void printDt(Datatype dt) {\r
 280         collectRefs(dt);\r
 281         dt.accept(printVisitor);\r
 282         if(!refs.isEmpty()) {\r
 283             newLine();\r
 284             sb.append("where");\r
 285             ++indentation;\r
 286             refs.forEachEntry(new TObjectObjectProcedure<Datatype, String>() {\r
 287                 @Override\r
 288                 public boolean execute(Datatype a, String b) {\r
 289                     newLine();\r
 290                     sb.append(b).append(" = ");\r
 291                     useRef = false;\r
 292                     a.accept(printVisitor);\r
 293                     return true;\r
 294                 }\r
 295             });\r
 296             --indentation;\r
 297             refs.clear();\r
 298         }\r
 299     }\r
 300     \r
 301     public static String print(Datatype dt) {\r
 302         StringBuilder sb = new StringBuilder();\r
 303         new DataTypePrinter2(sb).printDt(dt);\r
 304         return sb.toString();\r
 305     }
 306
 307 }