Add splitData() to data interval sets

paranoiq · Feb 14, 2020 · 2e2e76b · 2e2e76b
1 parent 8971f61
commit 2e2e76b
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/src/Time/IntervalData/DateIntervalDataSet.php b/src/Time/IntervalData/DateIntervalDataSet.php
@@ -322,4 +322,26 @@ public function modifyData(self $other, callable $reducer): self
         return new static($results);
     }
 
+    /**
+     * Split interval set to more interval sets with different subsets of original data.
+     * @param callable $splitter Maps original data set to a group of data sets. Should return array with keys indicating the data set group.
+     * @return self[] $this
+     */
+    public function splitData(callable $splitter): array
+    {
+        $intervalGroups = [];
+        foreach ($this->intervals as $interval) {
+            foreach ($splitter($interval->getData()) as $key => $values) {
+                $intervalGroups[$key][] = new DateIntervalData($interval->getStart(), $interval->getEnd(), $values);
+            }
+        }
+
+        $intervalSets = [];
+        foreach ($intervalGroups as $intervals) {
+            $intervalSets[] = (new DateIntervalDataSet($intervals))->normalize();
+        }
+
+        return $intervalSets;
+    }
+
 }
diff --git a/src/Time/IntervalData/NightIntervalDataSet.php b/src/Time/IntervalData/NightIntervalDataSet.php
@@ -322,4 +322,26 @@ public function modifyData(self $other, callable $reducer): self
         return new static($results);
     }
 
+    /**
+     * Split interval set to more interval sets with different subsets of original data.
+     * @param callable $splitter Maps original data set to a group of data sets. Should return array with keys indicating the data set group.
+     * @return self[] $this
+     */
+    public function splitData(callable $splitter): array
+    {
+        $intervalGroups = [];
+        foreach ($this->intervals as $interval) {
+            foreach ($splitter($interval->getData()) as $key => $values) {
+                $intervalGroups[$key][] = new NightIntervalData($interval->getStart(), $interval->getEnd(), $values);
+            }
+        }
+
+        $intervalSets = [];
+        foreach ($intervalGroups as $intervals) {
+            $intervalSets[] = (new NightIntervalDataSet($intervals))->normalize();
+        }
+
+        return $intervalSets;
+    }
+
 }