added dtf aware constructor in CsvResultEntitySource

ie3-institute · Jul 9, 2021 · 3ca7269 · 3ca7269
1 parent b8b4447
commit 3ca7269
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/main/java/edu/ie3/datamodel/io/source/csv/CsvResultEntitySource.java b/src/main/java/edu/ie3/datamodel/io/source/csv/CsvResultEntitySource.java
@@ -54,6 +54,18 @@ public CsvResultEntitySource(
     this.connectorResultFactory = new ConnectorResultFactory();
   }
 
+  public CsvResultEntitySource(
+      String csvSep, String folderPath, FileNamingStrategy fileNamingStrategy, String dtfPattern) {
+    super(csvSep, folderPath, fileNamingStrategy);
+
+    // init factories
+    this.systemParticipantResultFactory = new SystemParticipantResultFactory(dtfPattern);
+    this.thermalResultFactory = new ThermalResultFactory(dtfPattern);
+    this.switchResultFactory = new SwitchResultFactory(dtfPattern);
+    this.nodeResultFactory = new NodeResultFactory(dtfPattern);
+    this.connectorResultFactory = new ConnectorResultFactory(dtfPattern);
+  }
+
   // Grid
   @Override
   public Set<NodeResult> getNodeResults() {