add array of rotation

Author: AntSimi

setup.py

@@ -24,7 +24,7 @@
     ],
     zip_safe=False,
     entry_points=dict(console_scripts=[
-        'EddyUpdate = py_eddy_tracker.update.__init__:eddy_update',
+        'MergeEddies = py_eddy_tracker.appli:merge_eddies',
     ]),
     package_data={
         'py_eddy_tracker.featured_tracking': ['*.nc'],

src/py_eddy_tracker/appli.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+from py_eddy_tracker import EddyParser
+from py_eddy_tracker.observations.tracking import TrackEddiesObservations
+
+
+def merge_eddies():
+    parser = EddyParser('Merge eddies')
+    parser.add_argument('filename', nargs='+', help='all file to merge')
+    parser.add_argument('out', help='output file')
+    parser.add_argument('--add_rotation_variable', help='add rotation variables', action='store_true')
+    parser.add_argument('--include_var', nargs='+', type=str, help='use only listed variable, remove_var will be ignored')
+    args = parser.parse_args()
+
+    obs = TrackEddiesObservations.load_file(args.filename[0], raw_data=True, include_vars=args.include_var)
+    obs = obs.add_rotation_type()
+    for filename in args.filename[1:]:
+        other = TrackEddiesObservations.load_file(filename, raw_data=True, include_vars=args.include_var)
+        other = other.add_rotation_type()
+        obs = obs.merge(other)
+    obs.write_file(filename=args.out)
+

src/py_eddy_tracker/observations/observation.py

@@ -238,6 +238,7 @@ def add_fields(self, fields):
             track_extra_variables=list(concatenate((self.track_extra_variables, fields))),
             track_array_variables=self.track_array_variables,
             array_variables=self.array_variables,
+            only_variables=list(concatenate((self.obs.dtype.names, fields))),
             raw_data=self.raw_data
         )
         new.sign_type = self.sign_type
@@ -247,6 +248,11 @@ def add_fields(self, fields):
             new.obs[var] = self.obs[var]
         return new
 
+    def add_rotation_type(self):
+        new = self.add_fields(('type_cyc',))
+        new.observations['type_cyc'] = self.sign_type
+        return new
+
     @property
     def dtype(self):
         """Return dtype to build numpy array
@@ -378,6 +384,7 @@ def new_like(eddies, new_size):
             track_extra_variables=eddies.track_extra_variables,
             track_array_variables=eddies.track_array_variables,
             array_variables=eddies.array_variables,
+            only_variables=eddies.only_variables,
         )
 
     def index(self, index):