Added velocity to sws_ola

olaflow/processing/olaflow.py

@@ -25,6 +25,21 @@ class OFModel:
                 )
             )
 
+    def write_vector_field(self, field, values):
+        with open(self._root.joinpath("0", field), "r") as aw_file:
+            aw_raw = aw_file.read()
+
+        with open(self._root.joinpath("0", field), "w") as aw_file:
+            aw_file.write(
+                re.sub(
+                    r"(?<=\(\n).*?(?=\n\))",
+                    "\n".join(map(lambda x: f"({' '.join(x.astype('str'))})", values)),
+                    aw_raw,
+                    count=1,
+                    flags=re.S,
+                )
+            )
+
     @property
     def x(self):
         return self._x