Skip to content
Snippets Groups Projects
Commit 2c669096 authored by Simon Ding's avatar Simon Ding
Browse files

Support loading observation data for a given frequency range

parent 40bd9444
No related branches found
No related tags found
1 merge request!12Mpi likelihood
Pipeline #94091 passed
...@@ -93,7 +93,7 @@ class Observation: ...@@ -93,7 +93,7 @@ class Observation:
f(file_name, **dct) f(file_name, **dct)
@staticmethod @staticmethod
def load(file_name): def load(file_name, lo_hi_index=None):
dct = dict(np.load(file_name)) dct = dict(np.load(file_name))
antpos = [] antpos = []
for ii in range(4): for ii in range(4):
...@@ -103,6 +103,17 @@ class Observation: ...@@ -103,6 +103,17 @@ class Observation:
antpos.append(val) antpos.append(val)
pol = Polarization.from_list(dct["polarization"]) pol = Polarization.from_list(dct["polarization"])
direction = Direction.from_list(dct["direction"]) direction = Direction.from_list(dct["direction"])
if lo_hi_index is not None:
lo, hi = lo_hi_index
# FIXME: implement with less code duplication
return Observation(
AntennaPositions.from_list(antpos),
dct["vis"][:, :, lo:hi],
dct["weight"][:, :, lo:hi],
pol,
dct["freq"][lo:hi],
direction,
)
return Observation( return Observation(
AntennaPositions.from_list(antpos), AntennaPositions.from_list(antpos),
dct["vis"], dct["vis"],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment