vidavis 0.0.7__py3-none-any.whl → 0.0.8__py3-none-any.whl

vidavis/__version__.py

@@ -1 +1 @@
-__version__ = '0.0.7'
+__version__ = '0.0.8'

vidavis/apps/_ms_raster.py

@@ -841,11 +841,14 @@ class MsRaster(MsPlot):
                 xds = set_index_coordinates(self._plot_data, tuple(cursor_position.keys()))
                 sel_xds = xds.sel(indexers=None, method='nearest', tolerance=None, drop=False, **cursor_position)
                 for coord in sel_xds.coords:
-                    if coord != 'uvw_label':
-                        val, unit = self._get_xda_val_unit(sel_xds[coord])
-                        values[coord] = val
-                        units[coord] = unit
+                    if coord == 'uvw_label' or ('baseline_antenna' in coord and 'baseline_name' in sel_xds.coords):
+                        continue
+                    val, unit = self._get_xda_val_unit(sel_xds[coord])
+                    values[coord] = val
+                    units[coord] = unit
                 for data_var in sel_xds.data_vars:
+                    if 'TIME_CENTROID' in data_var:
+                        continue
                     val, unit = self._get_xda_val_unit(sel_xds[data_var])
                     if data_var == 'UVW':
                         names = ['U', 'V', 'W']
@@ -860,7 +863,7 @@ class MsRaster(MsPlot):
 
         # Set complex component name for visibilities
         if 'VISIBILITY' in values:
-            values[self._plot_inputs['vis_axis'].capitalize()] = values.pop('VISIBILITY')
+            values[self._plot_inputs['vis_axis'].upper()] = values.pop('VISIBILITY')
         return values, units
 
     def _update_cursor_box(self, location_values, units):

vidavis/data/measurement_set/processing_set/_ps_data.py

@@ -29,8 +29,7 @@ class PsData:
         if not ms:
             raise RuntimeError("MS path not available for reading MeasurementSet")
 
-        # Open processing set from zarr
-        # Converts msv2 if ms path is not zarr
+        # Open processing set from zarr. Converts msv2 if ms path is not zarr
         self._ps_xdt, self._zarr_path = get_processing_set(ms, logger)
 
         self._logger = logger
@@ -120,6 +119,7 @@ class PsData:
     def get_dimension_values(self, dimension):
         ''' Return sorted list of unique values for input dimension in selected ProcessingSet.
             For 'time', returns datetime strings.
+            For spectrum datasets, 'antenna2' returns empty list.
         '''
         ps_xdt = self._get_ps_xdt()
         dim_values = []
@@ -134,11 +134,18 @@ class PsData:
             elif dimension == 'antenna2':
                 dimension = 'baseline_antenna2_name'
             for ms_xdt in ps_xdt.values():
+                if dimension not in ms_xdt.coords:
+                    if 'antenna1' in dimension and 'antenna_name' in ms_xdt.coords:
+                        dimension = 'antenna_name' # spectrum dataset
+                    else:
+                        continue
                 try:
                     dim_values.extend([value.item() for value in ms_xdt[dimension].values])
                 except TypeError:
                     dim_values.append(ms_xdt[dimension].values.item())
 
+        if not dim_values:
+            return dim_values
         return sorted(set(dim_values))
 
     def _get_time_strings(self, ps):
@@ -152,13 +159,17 @@ class PsData:
         return times
 
     def _get_baselines(self, ps):
-        ''' Return baseline strings as ant1_name & ant2_name '''
+        ''' Return baseline strings as ant1_name & ant2_name.
+            For spectrum datasets, return list of antenna_name. '''
         baselines = []
         for ms_xdt in ps.values():
-            ant1_names = ms_xdt.baseline_antenna1_name.values
-            ant2_names = ms_xdt.baseline_antenna2_name.values
-            for ant1, ant2 in zip(ant1_names, ant2_names):
-                baselines.append(f"{ant1} & {ant2}")
+            if 'antenna_name' in ms_xdt.coords:
+                baselines.extend(ms_xdt.antenna_name.values)
+            else:
+                ant1_names = ms_xdt.baseline_antenna1_name.values
+                ant2_names = ms_xdt.baseline_antenna2_name.values
+                for ant1, ant2 in zip(ant1_names, ant2_names):
+                    baselines.append(f"{ant1} & {ant2}")
         return baselines
 
     def get_dimension_attrs(self, dim):

{vidavis-0.0.7.dist-info → vidavis-0.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vidavis
-Version: 0.0.7
+Version: 0.0.8
 Summary: Radio astronomy visibility data visualization
 License: LGPL
 Author-email: Darrell Schiebel <darrell@schiebel.us>,Pam Harris <pharris@nrao.edu>

{vidavis-0.0.7.dist-info → vidavis-0.0.8.dist-info}/RECORD

@@ -1,7 +1,7 @@
 vidavis/LICENSE.rst,sha256=qzGpkvhDzf_MgF1PIn6rCmYPrcEhkfrBUchosLJj-U4,26371
 vidavis/__init__.py,sha256=nhvlqlYhl4NzFO2WPTYfhyZkNdpfkBUTUOgJWKCK4Tc,1681
 vidavis/apps/__init__.py,sha256=ZQ5v1VFtjn3ztmuOHLOk5WbC1uLNIgL9rbHQ4v0zJwY,87
-vidavis/apps/_ms_raster.py,sha256=z6abcuznCSudJVcBnp0KpF5mwuRzbON8IFZtKhWnWKk,46831
+vidavis/apps/_ms_raster.py,sha256=dsvqK2BmFCjigUbv0DXrFkWzelzXJvOxqxUpjXPMAjs,47003
 vidavis/bokeh/__init__.py,sha256=gdPPxBCe0enCSjvPFxkgMlhpVnAMFXKcw9GN28tVdXM,95
 vidavis/bokeh/_palette.py,sha256=gzfJHuUgqxd8hJpZe-gQPFTCPq9f5I8uLEkHAK5FNDM,2480
 vidavis/data/__init__.py,sha256=-RDRe0PYK6vPlhdRV2Dy1vGbnDGoXWDATmfxaR-gXcE,48
@@ -10,7 +10,7 @@ vidavis/data/measurement_set/_ms_data.py,sha256=7ATsPbGFAv8WtWk_PRxlfoLdqapd6eMv
 vidavis/data/measurement_set/processing_set/__init__.py,sha256=TVQe5Nl4APCB2E1T1giAkOvSxg7OBNcOm9-BeZelY2Q,95
 vidavis/data/measurement_set/processing_set/_ps_concat.py,sha256=bCRsYFJQTp65QEAk_wGZQ2rZNkn6E3Gg2gKKFQfV5mw,3499
 vidavis/data/measurement_set/processing_set/_ps_coords.py,sha256=pdmojL-_55x3CjFcMIMuIqsI4ADG-TlLxc-RTgBQZC8,3640
-vidavis/data/measurement_set/processing_set/_ps_data.py,sha256=4-LXSk9ROnmWG8vZmSuBD7ObRIyNSTTfyjmYlULikpE,10767
+vidavis/data/measurement_set/processing_set/_ps_data.py,sha256=2Ss3iUdyIgtMuMTpl9mAGt4K89zgDmRsoTjbjurpDmc,11353
 vidavis/data/measurement_set/processing_set/_ps_io.py,sha256=VeNi-s1hozgCAGAGHs4NUXtlVFwUh-mkqrY9iYWOfW4,1717
 vidavis/data/measurement_set/processing_set/_ps_raster_data.py,sha256=6mU_V68lvJvIktEuyE5tWlyj7GaRmOnkCWSjHXc5lWs,7249
 vidavis/data/measurement_set/processing_set/_ps_select.py,sha256=GhAPxZ5mwvmUhCqKETVkhecjbwfrtDAG7-ibM19GDg0,10831
@@ -30,8 +30,8 @@ vidavis/toolbox/__init__.py,sha256=jqFa-eziVz_frNnXxwjJFK36qNpz1H38s-VlpBcq-R8,1
 vidavis/toolbox/_app_context.py,sha256=H7gtF8RrAH46FqDcMobv3KM1Osbnapgu6aTG-m3VCWA,3049
 vidavis/toolbox/_logging.py,sha256=OEisrd8FM8VTNBMc7neLh9ekelf29ZILYB5pScebly0,2739
 vidavis/toolbox/_static.py,sha256=HJLMtClppgOJXWAtV6Umn5EqN80u0oZiIouQ1JsB9PM,2346
-vidavis/__version__.py,sha256=7sNbee72r3qCAVloFm2RRWCTa9gVjMS7nCJcJ-URzl4,21
-vidavis-0.0.7.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
-vidavis-0.0.7.dist-info/METADATA,sha256=OxbpEuebhkcUmr7Ka5Tx4rK2GMfmw15Ls8yegy3XwRA,2242
-vidavis-0.0.7.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-vidavis-0.0.7.dist-info/RECORD,,
+vidavis/__version__.py,sha256=QaKrrCeLmUWRaUA8pjWo9GWzxTk9cCTRKQXEuQX5viQ,21
+vidavis-0.0.8.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
+vidavis-0.0.8.dist-info/METADATA,sha256=WRk7xKWALHxvBv3Zh3qdhqPoPxx9r6pU3IDw4d74k9s,2242
+vidavis-0.0.8.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+vidavis-0.0.8.dist-info/RECORD,,