--- /dev/null
+/*******************************************************************************\r
+ * Copyright (c) 2007 VTT Technical Research Centre of Finland and others.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ * VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.trend;\r
+\r
+import java.io.File;\r
+import java.util.UUID;\r
+\r
+import org.simantics.databoard.util.Bean;\r
+import org.simantics.history.Collector;\r
+import org.simantics.history.History;\r
+import org.simantics.history.HistoryException;\r
+import org.simantics.history.util.subscription.SubscriptionItem;\r
+import org.simantics.history.HistoryManager;\r
+import org.simantics.history.impl.CollectorImpl;\r
+import org.simantics.simulation.data.DatasourceAdapter;\r
+import org.simantics.simulation.data.PseudoSolver;\r
+import org.simantics.trend.configuration.TrendSamplingFormats;\r
+import org.simantics.utils.FileUtils;\r
+\r
+/**\r
+ * Test Data initializes PseudoSolver, HistoryCollector and DatasourceRepository\r
+ *\r
+ * @author Toni Kalajainen <toni.kalajainen@vtt.fi>\r
+ */\r
+public class TestData {\r
+\r
+ public PseudoSolver solver;\r
+ public File workarea;\r
+ public HistoryManager historyManager;\r
+ public Collector collector;\r
+ public DatasourceAdapter adapter;\r
+ public String subscriptionId = UUID.randomUUID().toString();\r
+\r
+ public TestData() throws HistoryException {\r
+ historyManager = History.createMemoryHistory();\r
+ collector = new CollectorImpl( historyManager );\r
+ init();\r
+ }\r
+ \r
+ public TestData(File workarea) throws HistoryException {\r
+ this.workarea = workarea;\r
+ historyManager = History.openFileHistory(workarea);\r
+ collector = new CollectorImpl( historyManager );\r
+ init();\r
+ }\r
+ \r
+ void init() throws HistoryException {\r
+ solver = new PseudoSolver();\r
+ SubscriptionItem[] hi = null;\r
+ hi = SubscriptionItem.createItems( PseudoSolver.SINE, subscriptionId, TrendSamplingFormats.createAnalogSamplingFormats(0, 0, "m") );\r
+ historyManager.create(hi);\r
+ collector.addItems(hi);\r
+ hi = SubscriptionItem.createItems( PseudoSolver.RAMP, subscriptionId, TrendSamplingFormats.createBinarySamplingFormats(0) );\r
+ historyManager.create(hi);\r
+ collector.addItems(hi);\r
+ hi = SubscriptionItem.createItems( PseudoSolver.RANDOM, subscriptionId, TrendSamplingFormats.createAnalogSamplingFormats(0, 0, "j") );\r
+ historyManager.create(hi);\r
+ collector.addItems(hi);\r
+ \r
+ adapter = new DatasourceAdapter( collector );\r
+ solver.addListener(adapter);\r
+ }\r
+ \r
+ public void dispose() {\r
+ solver.removeListener(adapter);\r
+ adapter.reset();\r
+ solver.stop();\r
+ collector.close();\r
+ try {\r
+ for (Bean bean : collector.getItems()) {\r
+ String id = (String) bean.getFieldUnchecked("id");\r
+ historyManager.delete(id);\r
+ }\r
+ } catch (HistoryException e) {\r
+ e.printStackTrace();\r
+ }\r
+ historyManager.close();\r
+ if (workarea != null) {\r
+ FileUtils.deleteDir(workarea);\r
+ }\r
+ }\r
+ \r
+}\r
+\r