Skip to content
Snippets Groups Projects
Commit 3a85c7ef authored by Mattia Mancini's avatar Mattia Mancini
Browse files

SSB-44: reordering of the indexes

parent 6362e3bb
No related branches found
No related tags found
1 merge request!44Merge back holography to master
...@@ -89,7 +89,7 @@ class HolographyDataTable(LazyH5Table): ...@@ -89,7 +89,7 @@ class HolographyDataTable(LazyH5Table):
try: try:
self.__get_index_arrays() self.__get_index_arrays()
except KeyError: except KeyError:
logger.debug('uri missing creating datable at: ' + self.uri) logger.debug('uri missing, creating datable at: ' + self.uri)
def __get_index_arrays(self): def __get_index_arrays(self):
""" """
...@@ -109,12 +109,12 @@ class HolographyDataTable(LazyH5Table): ...@@ -109,12 +109,12 @@ class HolographyDataTable(LazyH5Table):
# I select the first reference station just for convenience to iterate on the third level # I select the first reference station just for convenience to iterate on the third level
first_frequency = frequencies_string[0] first_frequency = frequencies_string[0]
beam_numbers = {beam_number for beam_number in self.h5_file[self.uri] beam_numbers = {beam_number for beam_number in self.h5_file[self.uri]
[first_frequency] [first_reference_station]
[first_reference_station]} [first_frequency]}
self._frequencies = set(sorted(map(float, frequencies_string))) self._frequencies = tuple(sorted(map(float, frequencies_string)))
self._reference_stations = reference_stations self._reference_stations = tuple(reference_stations)
self._beam_numbers = set(sorted(map(int, beam_numbers))) self._beam_numbers = tuple(sorted(map(int, beam_numbers)))
@property @property
def frequencies(self): def frequencies(self):
...@@ -173,7 +173,7 @@ class HolographyDataTable(LazyH5Table): ...@@ -173,7 +173,7 @@ class HolographyDataTable(LazyH5Table):
reference_station, frequency, beam_number = key reference_station, frequency, beam_number = key
self._frequencies.add(float(frequency)) self._frequencies.add(float(frequency))
self._reference_stations.add(frequency) self._reference_stations.add(reference_station)
self._beam_numbers.add(int(beam_number)) self._beam_numbers.add(int(beam_number))
super().__setitem__((reference_station, super().__setitem__((reference_station,
...@@ -184,13 +184,14 @@ class HolographyDataTable(LazyH5Table): ...@@ -184,13 +184,14 @@ class HolographyDataTable(LazyH5Table):
for frequency in self._frequencies: for frequency in self._frequencies:
for station in self._reference_stations: for station in self._reference_stations:
for beam in self._beam_numbers: for beam in self._beam_numbers:
yield (float(frequency), station, int(beam)), \ yield (station, float(frequency), int(beam)), \
self.__getitem__((station, frequency, beam)) self.__getitem__((station, frequency, beam))
def keys(self): def keys(self):
key_set = set() key_set = set()
for frequency in self._frequencies:
for station in self._reference_stations: for station in self._reference_stations:
for frequency in self._frequencies:
for beam in self._beam_numbers: for beam in self._beam_numbers:
key_set.add((frequency, station, beam)) key_set.add((station, frequency, beam))
return key_set return key_set
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment