X-Git-Url: https://gerrit.simantics.org/r/gitweb?p=simantics%2Fplatform.git;a=blobdiff_plain;f=bundles%2Forg.simantics.simulation%2Fsrc%2Forg%2Fsimantics%2Fsimulation%2Fexport%2FExportSubscriptionCSV.java;fp=bundles%2Forg.simantics.simulation%2Fsrc%2Forg%2Fsimantics%2Fsimulation%2Fexport%2FExportSubscriptionCSV.java;h=7c69c1a2eb67f3b7a9aa2433cc4011abc2a18dfa;hp=0000000000000000000000000000000000000000;hb=969bd23cab98a79ca9101af33334000879fb60c5;hpb=866dba5cd5a3929bbeae85991796acb212338a08

diff --git a/bundles/org.simantics.simulation/src/org/simantics/simulation/export/ExportSubscriptionCSV.java b/bundles/org.simantics.simulation/src/org/simantics/simulation/export/ExportSubscriptionCSV.java
new file mode 100644
index 000000000..7c69c1a2e
--- /dev/null
+++ b/bundles/org.simantics.simulation/src/org/simantics/simulation/export/ExportSubscriptionCSV.java
@@ -0,0 +1,131 @@
+package org.simantics.simulation.export;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.simantics.Simantics;
+import org.simantics.databoard.binding.error.BindingException;
+import org.simantics.databoard.binding.mutable.Variant;
+import org.simantics.databoard.util.Bean;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.export.core.ExportContext;
+import org.simantics.export.core.error.ExportException;
+import org.simantics.export.core.intf.ExportClass;
+import org.simantics.export.core.manager.Content;
+import org.simantics.history.History;
+import org.simantics.history.HistoryException;
+import org.simantics.history.HistoryManager;
+import org.simantics.history.ItemManager;
+import org.simantics.history.util.subscription.SamplingFormat;
+import org.simantics.simulation.experiment.IExperiment;
+import org.simantics.simulation.experiment.IHistoryExperiment;
+import org.simantics.simulation.export.CSVItemsQuery.CSVItem;
+import org.simantics.simulation.project.ExperimentManager;
+import org.simantics.simulation.project.IExperimentManager;
+import org.simantics.utils.datastructures.MapList;
+
+public class ExportSubscriptionCSV extends ExperimentExportClass implements ExportClass {
+	
+	@Override
+	public void export(List<Content> contents, 
+			Object writer_,
+			ExportContext context, 
+			Variant options,
+			IProgressMonitor monitor, 
+			MapList<Content, Content> attachmentMap
+			) throws ExportException {
+		
+		// Flush all experiments, just in case.
+		IExperimentManager em = Simantics.getProject().getHint( ExperimentManager.KEY_EXPERIMENT_MANAGER );
+		if ( em != null ) {
+			for (IExperiment exp : em.getExperiments()) {
+				if ( exp instanceof IHistoryExperiment ) {
+					IHistoryExperiment he = (IHistoryExperiment) exp;
+					try {
+						he.flushHistory();
+					} catch (HistoryException e) {
+					}
+				}
+			}
+		}
+		
+		CSVWriter writer = (CSVWriter) writer_;		
+		List<HistoryManager> historiesToClose = new ArrayList<HistoryManager>(); 
+		try {
+			List<ModelRef> models = ExperimentExportClass.getResult(context, options, true);
+			
+			// Label format is the following
+			// "MODEL EXPERIMENT/RUN ITEM" 
+			// MODEL is omited, if itemCount <= 1
+			// RUN is omited, if runCount <= 1 
+			int runCount = 0;
+			int modelCount = models.size();
+			for (ModelRef model : models) runCount += model.enabledRunCount();
+
+			List<CSVItem> items = new ArrayList<CSVItem>();
+			for (Content content : contents) {
+				items.addAll( context.session.syncRequest( new CSVItemsQuery( content.url ) ) );
+			}
+
+			// 
+			for (ModelRef model : models) {
+				for (ExperimentRef experiment : model.experiments) {
+					if ( experiment.runs.isEmpty() ) continue;					
+					for (RunRef run : experiment.runs) {						
+						if ( run.historyFolder == null || !run.historyFolder.exists() ) continue;
+			
+						HistoryManager history = History.openFileHistory( run.historyFolder );
+						historiesToClose.add( history );
+		        		ItemManager im = new ItemManager( history.getItems() );
+						
+						for ( CSVItem item : items ) {
+							if ( !item.modelUri.equals( model.uri ) ) continue;
+							StringBuilder labelBuilder = new StringBuilder();
+							labelBuilder.append( item.label );
+							if ( modelCount>1 ) {
+								labelBuilder.append(" (");
+								labelBuilder.append(model.label);
+								labelBuilder.append(")");
+							}
+							if ( runCount>1 ) {
+								labelBuilder.append(" (");
+								labelBuilder.append( experiment.label );
+								labelBuilder.append( "\\" );
+								labelBuilder.append( run.label );
+								labelBuilder.append(")");
+							}
+							
+			            	List<Bean> historyItems = im.search("variableId", item.variableReference);
+			            	Collections.sort(historyItems, SamplingFormat.INTERVAL_COMPARATOR);
+			            	if (items.isEmpty()) continue;
+			            	Bean config = historyItems.get(0);
+			            	String historyId = (String) config.getFieldUnchecked("id");
+														
+							writer.addItem(history, historyId, labelBuilder.toString(), item.variableReference, item.unit);
+						}
+						
+					}
+				}
+			}
+					
+		} catch (DatabaseException e) {
+			throw new ExportException( e );
+		} catch (HistoryException e) {
+			throw new ExportException( e );
+		} catch (BindingException e) {
+			throw new ExportException( e );
+		} finally {
+			for (HistoryManager history : historiesToClose) history.close();
+		}
+		
+	}
+
+	@Override
+	public List<String> validate(String contentUri, ExportContext context, Variant options) {
+		return Collections.emptyList();
+	}
+	
+}
+