bundles/org.simantics.trend/example/org/simantics/trend/TestData.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007 VTT Technical Research Centre of Finland and others.\r
   3  * All rights reserved. This program and the accompanying materials\r
   4  * are made available under the terms of the Eclipse Public License v1.0\r
   5  * which accompanies this distribution, and is available at\r
   6  * http://www.eclipse.org/legal/epl-v10.html\r
   7  *\r
   8  * Contributors:\r
   9  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  10  *******************************************************************************/\r
  11 package org.simantics.trend;\r
  12 \r
  13 import java.io.File;\r
  14 import java.util.UUID;\r
  15 \r
  16 import org.simantics.databoard.util.Bean;\r
  17 import org.simantics.history.Collector;\r
  18 import org.simantics.history.History;\r
  19 import org.simantics.history.HistoryException;\r
  20 import org.simantics.history.util.subscription.SubscriptionItem;\r
  21 import org.simantics.history.HistoryManager;\r
  22 import org.simantics.history.impl.CollectorImpl;\r
  23 import org.simantics.simulation.data.DatasourceAdapter;\r
  24 import org.simantics.simulation.data.PseudoSolver;\r
  25 import org.simantics.trend.configuration.TrendSamplingFormats;\r
  26 import org.simantics.utils.FileUtils;\r
  27 \r
  28 /**\r
  29  * Test Data initializes PseudoSolver, HistoryCollector and DatasourceRepository\r
  30  *\r
  31  * @author Toni Kalajainen <toni.kalajainen@vtt.fi>\r
  32  */\r
  33 public class TestData {\r
  34 \r
  35         public PseudoSolver solver;\r
  36         public File workarea;\r
  37         public HistoryManager historyManager;\r
  38         public Collector collector;\r
  39         public DatasourceAdapter adapter;\r
  40         public String subscriptionId = UUID.randomUUID().toString();\r
  41 \r
  42         public TestData() throws HistoryException {\r
  43                 historyManager = History.createMemoryHistory();\r
  44                 collector = new CollectorImpl( historyManager );\r
  45                 init();\r
  46         }\r
  47         \r
  48         public TestData(File workarea) throws HistoryException {\r
  49                 this.workarea = workarea;\r
  50                 historyManager = History.openFileHistory(workarea);\r
  51                 collector = new CollectorImpl( historyManager );\r
  52                 init();\r
  53         }\r
  54         \r
  55         void init() throws HistoryException {\r
  56                 solver = new PseudoSolver();\r
  57                 SubscriptionItem[] hi = null;\r
  58                 hi = SubscriptionItem.createItems( PseudoSolver.SINE, subscriptionId, TrendSamplingFormats.createAnalogSamplingFormats(0, 0, "m") );\r
  59                 historyManager.create(hi);\r
  60                 collector.addItems(hi);\r
  61                 hi = SubscriptionItem.createItems( PseudoSolver.RAMP, subscriptionId, TrendSamplingFormats.createBinarySamplingFormats(0) );\r
  62                 historyManager.create(hi);\r
  63                 collector.addItems(hi);\r
  64                 hi = SubscriptionItem.createItems( PseudoSolver.RANDOM, subscriptionId, TrendSamplingFormats.createAnalogSamplingFormats(0, 0, "j") );\r
  65                 historyManager.create(hi);\r
  66                 collector.addItems(hi);\r
  67                 \r
  68                 adapter = new DatasourceAdapter( collector );\r
  69                 solver.addListener(adapter);\r
  70         }\r
  71         \r
  72         public void dispose() {\r
  73                 solver.removeListener(adapter);\r
  74                 adapter.reset();\r
  75                 solver.stop();\r
  76                 collector.close();\r
  77                 try {\r
  78                         for (Bean bean : collector.getItems()) {\r
  79                                 String id = (String) bean.getFieldUnchecked("id");\r
  80                                 historyManager.delete(id);\r
  81                         }\r
  82                 } catch (HistoryException e) {\r
  83                         e.printStackTrace();\r
  84                 }\r
  85                 historyManager.close();\r
  86                 if (workarea != null) {\r
  87                         FileUtils.deleteDir(workarea);\r
  88                 }\r
  89         }\r
  90         \r
  91 }\r
  92 \r