Revision 8b22682704c00bd278c44dae1686f726d261b718 authored by Steven Murray on 09 January 2023, 21:13:32 UTC, committed by Steven Murray on 09 January 2023, 21:13:32 UTC
1 parent e377413
uvdata.py
# -*- mode: python; coding: utf-8 -*-
# Copyright (c) 2018 Radio Astronomy Software Group
# Licensed under the 2-clause BSD License
"""Primary container for radio interferometer datasets."""
import copy
import os
import threading
import warnings
from collections.abc import Iterable
import astropy.units as units
import numpy as np
from astropy import constants as const
from astropy import coordinates as coord
from astropy.coordinates import FK5, Angle, EarthLocation, SkyCoord
from astropy.time import Time
from scipy import ndimage as nd
from .. import parameter as uvp
from .. import telescopes as uvtel
from .. import utils as uvutils
from ..uvbase import UVBase
__all__ = ["UVData"]
import logging
logger = logging.getLogger(__name__)
# standard angle tolerance: 1 mas in radians.
radian_tol = 1 * 2 * np.pi * 1e-3 / (60.0 * 60.0 * 360.0)
reporting_request = (
" Please report this in our issue log, we have not been able to find a file with "
"this feature, we would like to investigate this more."
)
class UVData(UVBase):
"""
A class for defining a radio interferometer dataset.
Attributes
----------
UVParameter objects :
For full list see the documentation on ReadTheDocs:
http://pyuvdata.readthedocs.io/en/latest/.
Some are always required, some are required only for flex_spw data
and others are always optional.
"""
def __init__(self):
"""Create a new UVData object."""
# add the UVParameters to the class
self._Ntimes = uvp.UVParameter(
"Ntimes", description="Number of times.", expected_type=int
)
self._Nbls = uvp.UVParameter(
"Nbls", description="Number of baselines.", expected_type=int
)
desc = (
"Number of baseline-times (i.e. number of spectra). Not necessarily "
"equal to Nbls * Ntimes."
)
self._Nblts = uvp.UVParameter("Nblts", description=desc, expected_type=int)
self._Nfreqs = uvp.UVParameter(
"Nfreqs", description="Number of frequency channels.", expected_type=int
)
self._Npols = uvp.UVParameter(
"Npols", description="Number of polarizations.", expected_type=int
)
desc = (
"Array of the visibility data, shape: (Nblts, 1, Nfreqs, "
"Npols) or (Nblts, Nfreqs, Npols) if future_array_shapes=True, "
"type = complex float, in units of self.vis_units."
)
# TODO: Spw axis to be collapsed in future release
self._data_array = uvp.UVParameter(
"data_array",
description=desc,
form=("Nblts", 1, "Nfreqs", "Npols"),
expected_type=complex,
)
desc = 'Visibility units, options are: "uncalib", "Jy" or "K str".'
self._vis_units = uvp.UVParameter(
"vis_units",
description=desc,
form="str",
expected_type=str,
acceptable_vals=["uncalib", "Jy", "K str"],
)
desc = (
"Number of data points averaged into each data element, "
"NOT required to be an integer, type = float, same shape as data_array."
"The product of the integration_time and the nsample_array "
"value for a visibility reflects the total amount of time "
"that went into the visibility. Best practice is for the "
"nsample_array to be used to track flagging within an integration_time "
"(leading to a decrease of the nsample array value below 1) and "
"LST averaging (leading to an increase in the nsample array "
"value). So datasets that have not been LST averaged should "
"have nsample array values less than or equal to 1."
"Note that many files do not follow this convention, but it is "
"safe to assume that the product of the integration_time and "
"the nsample_array is the total amount of time included in a visibility."
)
self._nsample_array = uvp.UVParameter(
"nsample_array",
description=desc,
form=("Nblts", 1, "Nfreqs", "Npols"),
expected_type=float,
)
desc = "Boolean flag, True is flagged, same shape as data_array."
self._flag_array = uvp.UVParameter(
"flag_array",
description=desc,
form=("Nblts", 1, "Nfreqs", "Npols"),
expected_type=bool,
)
self._Nspws = uvp.UVParameter(
"Nspws",
description="Number of spectral windows "
"(ie non-contiguous spectral chunks). ",
expected_type=int,
)
self._spw_array = uvp.UVParameter(
"spw_array",
description="Array of spectral window numbers, shape (Nspws).",
form=("Nspws",),
expected_type=int,
)
desc = (
"Projected baseline vectors relative to phase center, "
"shape (Nblts, 3), units meters. "
"Convention is: uvw = xyz(ant2) - xyz(ant1)."
"Note that this is the Miriad convention but it is different "
"from the AIPS/FITS convention (where uvw = xyz(ant1) - xyz(ant2))."
)
self._uvw_array = uvp.UVParameter(
"uvw_array",
description=desc,
form=("Nblts", 3),
expected_type=np.float64,
strict_type_check=True,
acceptable_range=(0, 1e8),
tols=1e-3,
)
desc = (
"Array of times, center of integration, shape (Nblts), units Julian Date."
)
self._time_array = uvp.UVParameter(
"time_array",
description=desc,
form=("Nblts",),
expected_type=np.float64,
strict_type_check=True,
tols=1e-3 / (60.0 * 60.0 * 24.0),
) # 1 ms in days
desc = (
"Array of local apparent sidereal times (LAST) at the center of "
"integration, shape (Nblts), units radians."
)
self._lst_array = uvp.UVParameter(
"lst_array",
description=desc,
form=("Nblts",),
expected_type=np.float64,
strict_type_check=True,
tols=radian_tol,
)
desc = (
"Array of numbers for the first antenna, which is matched to that in "
"the antenna_numbers attribute. Shape (Nblts), type = int."
)
self._ant_1_array = uvp.UVParameter(
"ant_1_array",
description=desc,
expected_type=int,
form=("Nblts",),
acceptable_range=(0, 2147483647),
)
desc = (
"Array of numbers for the second antenna, which is matched to that in "
"the antenna_numbers attribute. Shape (Nblts), type = int."
)
self._ant_2_array = uvp.UVParameter(
"ant_2_array",
description=desc,
expected_type=int,
form=("Nblts",),
acceptable_range=(0, 2147483647),
)
desc = (
"Array of baseline numbers, shape (Nblts), type = int; "
"by default baseline = 2048 * ant1 + ant2 + 2^16, "
"though other conventions are available."
)
self._baseline_array = uvp.UVParameter(
"baseline_array",
description=desc,
expected_type=int,
form=("Nblts",),
acceptable_range=(0, 4611686018498691072),
)
# this dimensionality of freq_array does not allow for different spws
# to have different dimensions
desc = (
"Array of frequencies, center of the channel, "
"shape (1, Nfreqs) or (Nfreqs,) if future_array_shapes=True, units Hz."
)
# TODO: Spw axis to be collapsed in future release
self._freq_array = uvp.UVParameter(
"freq_array",
description=desc,
form=(1, "Nfreqs"),
expected_type=float,
tols=1e-3,
) # mHz
desc = (
"Array of polarization integers, shape (Npols). "
"AIPS Memo 117 says: pseudo-stokes 1:4 (pI, pQ, pU, pV); "
"circular -1:-4 (RR, LL, RL, LR); linear -5:-8 (XX, YY, XY, YX). "
"NOTE: AIPS Memo 117 actually calls the pseudo-Stokes polarizations "
'"Stokes", but this is inaccurate as visibilities cannot be in '
"true Stokes polarizations for physical antennas. We adopt the "
"term pseudo-Stokes to refer to linear combinations of instrumental "
"visibility polarizations (e.g. pI = xx + yy)."
)
self._polarization_array = uvp.UVParameter(
"polarization_array",
description=desc,
expected_type=int,
acceptable_vals=list(np.arange(-8, 0)) + list(np.arange(1, 5)),
form=("Npols",),
)
desc = (
"Length of the integration in seconds, shape (Nblts). "
"The product of the integration_time and the nsample_array "
"value for a visibility reflects the total amount of time "
"that went into the visibility. Best practice is for the "
"integration_time to reflect the length of time a visibility "
"was integrated over (so it should vary in the case of "
"baseline-dependent averaging and be a way to do selections "
"for differently integrated baselines)."
"Note that many files do not follow this convention, but it is "
"safe to assume that the product of the integration_time and "
"the nsample_array is the total amount of time included in a visibility."
)
self._integration_time = uvp.UVParameter(
"integration_time",
description=desc,
form=("Nblts",),
expected_type=float,
tols=1e-3,
) # 1 ms
desc = (
"Width of frequency channels (Hz). If flex_spw = False and "
"future_array_shapes=False, then it is a "
"single value of type = float, otherwise it is an array of shape "
"(Nfreqs), type = float."
)
self._channel_width = uvp.UVParameter(
"channel_width", description=desc, expected_type=float, tols=1e-3
) # 1 mHz
self._telescope_name = uvp.UVParameter(
"telescope_name",
description="Name of telescope or array (string).",
form="str",
expected_type=str,
)
self._instrument = uvp.UVParameter(
"instrument",
description="Receiver or backend. Sometimes identical to telescope_name.",
form="str",
expected_type=str,
)
desc = (
"Telescope location: xyz position in specified frame (default ITRS). "
"Can also be accessed using telescope_location_lat_lon_alt or "
"telescope_location_lat_lon_alt_degrees properties."
)
self._telescope_location = uvp.LocationParameter(
"telescope_location",
description=desc,
acceptable_range=(6.35e6, 6.39e6),
frame="itrs",
tols=1e-3,
)
self._history = uvp.UVParameter(
"history",
description="String of history, units English.",
form="str",
expected_type=str,
)
# --- flexible spectral window information ---
desc = (
'Option to construct a "flexible spectral window", which stores'
"all spectral channels across the frequency axis of data_array. "
"Allows for spectral windows of variable sizes, and channels of "
"varying widths."
)
self._flex_spw = uvp.UVParameter(
"flex_spw", description=desc, expected_type=bool, value=False
)
desc = (
"Required if flex_spw = True. Maps individual channels along the "
"frequency axis to individual spectral windows, as listed in the "
"spw_array. Shape (Nfreqs), type = int."
)
self._flex_spw_id_array = uvp.UVParameter(
"flex_spw_id_array",
description=desc,
form=("Nfreqs",),
expected_type=int,
required=False,
)
desc = (
"Optional, only used if flex_spw = True. Allows for labeling individual "
"spectral windows with different polarizations. If set, Npols must be set "
"to 1 (i.e., only one polarization per spectral window allowed). Shape "
"(Nspws), type = int."
)
self._flex_spw_polarization_array = uvp.UVParameter(
"flex_spw_polarization_array",
description=desc,
form=("Nspws",),
expected_type=int,
acceptable_vals=list(np.arange(-8, 0)) + list(np.arange(1, 5)),
required=False,
)
desc = "Flag indicating that this object is using the future array shapes."
self._future_array_shapes = uvp.UVParameter(
"future_array_shapes", description=desc, expected_type=bool, value=False
)
# --- phasing information ---
desc = "Specifies the number of phase centers contained within the data set."
self._Nphase = uvp.UVParameter("Nphase", description=desc, expected_type=int)
desc = (
"Dictionary that acts as a catalog, containing information on individual "
"phase centers. Keys are the catalog IDs of the different phase centers in "
"the object (matched to the parameter `phase_center_id_array`). At a "
"minimum, each dictionary must contain the keys: "
"'cat_name' giving the phase center name (this does not have to be unique, "
"non-unique values can be used to indicate sets of phase centers that make "
"up a mosaic observation), "
"'cat_type', which can be 'sidereal' (fixed position in RA/Dec), 'ephem' "
"(position in RA/Dec which moves with time), 'driftscan' (fixed postion in "
"Az/El, NOT the same as the old `phase_type`='drift') or 'unprojected' "
"(baseline coordinates in ENU, but data are not phased, similar to "
"the old `phase_type`='drift')"
"'cat_lon' (longitude coord, e.g. RA, either a single value or a one "
"dimensional array of length Npts --the number of ephemeris data points-- "
"for ephem type phase centers), "
"'cat_lat' (latitude coord, e.g. Dec., either a single value or a one "
"dimensional array of length Npts --the number of ephemeris data points-- "
"for ephem type phase centers), "
"'cat_frame' (coordinate frame, e.g. icrs, must be a frame supported by "
"astropy). "
"Other optional keys include "
"'cat_epoch' (epoch and equinox of the coordinate frame, not needed for "
"frames without an epoch (e.g. ICRS) unless the there is proper motion), "
"'cat_times' (times for the coordinates, only used for 'ephem' types), "
"'cat_pm_ra' (proper motion in RA), "
"'cat_pm_dec' (proper motion in Dec), "
"'cat_dist' (physical distance to the source in parsec, useful if parallax "
"is important, either a single value or a one dimensional array of length "
"Npts --the number of ephemeris data points-- for ephem type phase "
"centers.), "
"'cat_vrad' (rest frame velocity in km/s, either a single value or a one "
"dimensional array of length Npts --the number of ephemeris data points-- "
"for ephem type phase centers.), and "
"'info_source' (describes where catalog info came from). "
"See the documentation of the `phase` method for more details."
)
self._phase_center_catalog = uvp.UVParameter(
"phase_center_catalog", description=desc, expected_type=dict
)
desc = (
"Apparent right ascension of phase center in the topocentric frame of the "
"observatory, units radians. Shape (Nblts,), type = float."
)
self._phase_center_app_ra = uvp.AngleParameter(
"phase_center_app_ra",
form=("Nblts",),
expected_type=float,
description=desc,
tols=radian_tol,
)
desc = (
"Apparent Declination of phase center in the topocentric frame of the "
"observatory, units radians. Shape (Nblts,), type = float."
)
self._phase_center_app_dec = uvp.AngleParameter(
"phase_center_app_dec",
form=("Nblts",),
expected_type=float,
description=desc,
tols=radian_tol,
)
desc = (
"Position angle between the hour circle (which is a great circle that goes "
"through the target postion and both poles) in the apparent/topocentric "
"frame, and the frame given in the phase_center_frame attribute."
"Shape (Nblts,), type = float."
)
# The tolerance here is set by the fact that is is calculated using an arctan,
# the limiting precision of which happens around values of 1.
self._phase_center_frame_pa = uvp.AngleParameter(
"phase_center_frame_pa",
form=("Nblts",),
expected_type=float,
description=desc,
tols=2e-8,
)
desc = (
"Maps individual indices along the Nblt axis to a key in "
"`phase_center_catalog`, which maps to a dict "
"containing the other metadata for each phase center."
"Shape (Nblts), type = int."
)
self._phase_center_id_array = uvp.UVParameter(
"phase_center_id_array",
description=desc,
form=("Nblts",),
expected_type=int,
)
desc = (
"Optional when reading a MS. Retains the scan number when reading a MS."
" Shape (Nblts), type = int."
)
self._scan_number_array = uvp.UVParameter(
"scan_number_array",
description=desc,
form=("Nblts",),
expected_type=int,
required=False,
)
# --- antenna information ----
desc = (
"Number of antennas with data present (i.e. number of unique "
"entries in ant_1_array and ant_2_array). May be smaller "
"than the number of antennas in the array."
)
self._Nants_data = uvp.UVParameter(
"Nants_data", description=desc, expected_type=int
)
desc = (
"Number of antennas in the array. May be larger "
"than the number of antennas with data."
)
self._Nants_telescope = uvp.UVParameter(
"Nants_telescope", description=desc, expected_type=int
)
desc = (
"List of antenna names, shape (Nants_telescope), "
"with numbers given by antenna_numbers (which can be matched "
"to ant_1_array and ant_2_array). There must be one entry "
"here for each unique entry in ant_1_array and "
"ant_2_array, but there may be extras as well. "
)
self._antenna_names = uvp.UVParameter(
"antenna_names",
description=desc,
form=("Nants_telescope",),
expected_type=str,
)
desc = (
"List of integer antenna numbers corresponding to antenna_names, "
"shape (Nants_telescope). There must be one "
"entry here for each unique entry in ant_1_array and "
"ant_2_array, but there may be extras as well."
"Note that these are not indices -- they do not need to start "
"at zero or be continuous."
)
self._antenna_numbers = uvp.UVParameter(
"antenna_numbers",
description=desc,
form=("Nants_telescope",),
expected_type=int,
)
desc = (
"Array giving coordinates of antennas relative to "
"telescope_location (ITRF frame), shape (Nants_telescope, 3), "
"units meters. See the tutorial page in the documentation "
"for an example of how to convert this to topocentric frame."
)
self._antenna_positions = uvp.UVParameter(
"antenna_positions",
description=desc,
form=("Nants_telescope", 3),
expected_type=float,
tols=1e-3, # 1 mm
)
# -------- extra, non-required parameters ----------
desc = (
"Orientation of the physical dipole corresponding to what is "
"labelled as the x polarization. Options are 'east' "
"(indicating east/west orientation) and 'north (indicating "
"north/south orientation)."
)
self._x_orientation = uvp.UVParameter(
"x_orientation",
description=desc,
required=False,
expected_type=str,
acceptable_vals=["east", "north"],
)
blt_order_options = ["time", "baseline", "ant1", "ant2", "bda"]
desc = (
"Ordering of the data array along the blt axis. A tuple with "
'the major and minor order (minor order is omitted if order is "bda"). '
"The allowed values are: "
+ " ,".join([str(val) for val in blt_order_options])
+ "."
)
self._blt_order = uvp.UVParameter(
"blt_order",
description=desc,
form=(2,),
required=False,
expected_type=str,
acceptable_vals=blt_order_options,
)
desc = (
"Any user supplied extra keywords, type=dict. Keys should be "
"8 character or less strings if writing to uvfits or miriad files. "
'Use the special key "comment" for long multi-line string comments.'
)
self._extra_keywords = uvp.UVParameter(
"extra_keywords",
required=False,
description=desc,
value={},
spoof_val={},
expected_type=dict,
)
desc = (
"Array of antenna diameters in meters. Used by CASA to "
"construct a default beam if no beam is supplied."
)
self._antenna_diameters = uvp.UVParameter(
"antenna_diameters",
required=False,
description=desc,
form=("Nants_telescope",),
expected_type=float,
tols=1e-3, # 1 mm
)
# --- other stuff ---
# the below are copied from AIPS memo 117, but could be revised to
# merge with other sources of data.
self._gst0 = uvp.UVParameter(
"gst0",
required=False,
description="Greenwich sidereal time at midnight on reference date.",
spoof_val=0.0,
expected_type=float,
)
self._rdate = uvp.UVParameter(
"rdate",
required=False,
description="Date for which the GST0 applies.",
spoof_val="",
form="str",
)
self._earth_omega = uvp.UVParameter(
"earth_omega",
required=False,
description="Earth's rotation rate in degrees per day.",
spoof_val=360.985,
expected_type=float,
)
self._dut1 = uvp.UVParameter(
"dut1",
required=False,
description="DUT1 (google it) AIPS 117 calls it UT1UTC.",
spoof_val=0.0,
expected_type=float,
)
self._timesys = uvp.UVParameter(
"timesys",
required=False,
description="We only support UTC.",
spoof_val="UTC",
form="str",
)
desc = (
"FHD thing we do not understand, something about the time "
"at which the phase center is normal to the chosen UV plane "
"for phasing."
)
self._uvplane_reference_time = uvp.UVParameter(
"uvplane_reference_time", required=False, description=desc, spoof_val=0
)
desc = "Per-antenna and per-frequency equalization coefficients."
self._eq_coeffs = uvp.UVParameter(
"eq_coeffs",
required=False,
description=desc,
form=("Nants_telescope", "Nfreqs"),
expected_type=float,
spoof_val=1.0,
)
desc = "Convention for how to remove eq_coeffs from data."
self._eq_coeffs_convention = uvp.UVParameter(
"eq_coeffs_convention",
required=False,
description=desc,
form="str",
spoof_val="divide",
)
desc = (
"List of strings containing the unique basenames (not the full path) of "
"input files."
)
self._filename = uvp.UVParameter(
"filename", required=False, description=desc, expected_type=str
)
super(UVData, self).__init__()
def _set_flex_spw(self):
"""
Set flex_spw to True, and adjust required parameters.
This method should not be called directly by users; instead it is called
by the file-reading methods to indicate that an object has multiple spectral
windows concatenated together across the frequency axis.
"""
# Mark once-optional arrays as now required
self.flex_spw = True
self._flex_spw_id_array.required = True
# If setting flex_spw, then we can set values in polarization_array to 0
# provided that we set flex_spw_polarization_array
self._polarization_array.acceptable_vals = list(np.arange(-8, 5))
# Now make sure that chan_width is set to be an array
self._channel_width.form = ("Nfreqs",)
def _set_scan_numbers(self, override=False):
"""
Set scan numbers by grouping consecutive integrations on the same phase center.
This approach mimics the definition of scan number in measurement sets and is
especially helpful for distinguishing between repeated visits to multiple
phase centers.
Parameters
----------
override : bool
When True, will redefine existing scan numbers. Default is False.
"""
if self.scan_number_array is None or override:
slice_list = []
# This loops over phase centers, finds contiguous integrations with
# ndimage.label, and then finds the slices to return those contiguous
# integrations with nd.find_objects.
for cat_id in self.phase_center_catalog:
slice_list.extend(
nd.find_objects(nd.label(self.phase_center_id_array == cat_id)[0])
)
# Sort by start integration number, which we can extract from
# the start of each slice in the list.
slice_list_ord = sorted(slice_list, key=lambda x: x[0].start)
# Incrementally increase the scan number with each group in
# slice_list_ord
scan_array = np.zeros_like(self.phase_center_id_array)
for ii, slice_scan in enumerate(slice_list_ord):
scan_array[slice_scan] = ii + 1
self.scan_number_array = scan_array
def _look_for_name(self, cat_name):
"""
Look up catalog IDs which match a given name.
Parameters
----------
cat_name : str
Name to match against entries in phase_center_catalog.
Returns
-------
cat_id_list : list
List of all catalog IDs which match the given name.
"""
return [
pc_id
for pc_id, pc_dict in self.phase_center_catalog.items()
if pc_dict["cat_name"] == cat_name
]
def _look_in_catalog(
self,
cat_name=None,
cat_type=None,
cat_lon=None,
cat_lat=None,
cat_frame=None,
cat_epoch=None,
cat_times=None,
cat_pm_ra=None,
cat_pm_dec=None,
cat_dist=None,
cat_vrad=None,
ignore_name=False,
target_cat_id=None,
phase_dict=None,
):
"""
Check the catalog to see if an existing entry matches provided data.
This is a helper function for verifying if an entry already exists within
the catalog, contained within the attribute `phase_center_catalog`.
Parameters
----------
cat_name : str
Name of the phase center, which should match a the value of "cat_name"
inside an entry of `phase_center_catalog`.
cat_type : str
Type of phase center of the entry. Must be one of:
"sidereal" (fixed RA/Dec),
"ephem" (RA/Dec that moves with time),
"driftscan" (fixed az/el position),
"unprojected" (no w-projection, equivalent to the old
`phase_type` == "drift").
cat_lon : float or ndarray
Value of the longitudinal coordinate (e.g., RA, Az, l) in radians of the
phase center. No default unless `cat_type="unprojected"`, in which case the
default is zero. Expected to be a float for sidereal and driftscan phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
cat_lat : float or ndarray
Value of the latitudinal coordinate (e.g., Dec, El, b) in radians of the
phase center. No default unless `cat_type="unprojected"`, in which case the
default is pi/2. Expected to be a float for sidereal and driftscan phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
cat_frame : str
Coordinate frame that cat_lon and cat_lat are given in. Only used for
sidereal and ephem phase centers. Can be any of the several supported frames
in astropy (a limited list: fk4, fk5, icrs, gcrs, cirs, galactic).
cat_epoch : str or float
Epoch of the coordinates, only used when cat_frame = fk4 or fk5. Given
in units of fractional years, either as a float or as a string with the
epoch abbreviation (e.g, Julian epoch 2000.0 would be J2000.0).
cat_times : ndarray of floats
Only used when `cat_type="ephem"`. Describes the time for which the values
of `cat_lon` and `cat_lat` are caclulated, in units of JD. Shape is (Npts,).
cat_pm_ra : float
Proper motion in RA, in units of mas/year. Only used for sidereal phase
centers.
cat_pm_dec : float
Proper motion in Dec, in units of mas/year. Only used for sidereal phase
centers.
cat_dist : float or ndarray of float
Distance of the source, in units of pc. Only used for sidereal and ephem
phase centers. Expected to be a float for sidereal and driftscan phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
cat_vrad : float or ndarray of float
Radial velocity of the source, in units of km/s. Only used for sidereal and
ephem phase centers. Expected to be a float for sidereal and driftscan phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
ignore_name : bool
Nominally, `_look_in_catalog` will only look at entries where `cat_name`
matches the name of an entry in the catalog. However, by setting this to
True, the method will search all entries in the catalog and see if any
match all of the provided data (excluding `cat_name`).
target_cat_id : int
Optional argument to specify a particular cat_id to check against.
phase_dict : dict
Instead of providing individual parameters, one may provide a dict which
matches that format used within `phase_center_catalog` for checking for
existing entries. If used, all other parameters (save for `ignore_name`
and `cat_name`) are disregarded.
Returns
-------
cat_id : int or None
The unique ID number for the phase center added to the internal catalog.
This value is used in the `phase_center_id_array` attribute to denote which
source a given baseline-time corresponds to. If no catalog entry matches,
then None is returned.
cat_diffs : int
The number of differences between the information provided and the catalog
entry contained within `phase_center_catalog`. If everything matches, then
`cat_diffs=0`.
"""
# 1 marcsec tols
radian_tols = (0, radian_tol)
default_tols = (1e-5, 1e-8)
match_id = None
match_diffs = 99999
if (cat_name is None) and (not ignore_name):
if phase_dict is None:
raise ValueError(
"Must specify either phase_dict or cat_name if ignore_name=False."
)
cat_name = phase_dict["cat_name"]
if cat_type == "unphased" or cat_name == "unphased":
warnings.warn(
"The 'unphased' catalog type has been renamed to 'unprojected'. Using "
"unprojected for now, this warning will become an error in version 2.4",
DeprecationWarning,
)
cat_type = "unprojected"
if cat_name == "unphased":
cat_name = "unprojected"
# Emulate the defaults that are set if None is detected for
# unprojected and driftscan types.
if cat_type in ["unprojected", "driftscan"]:
if cat_lon is None:
cat_lon = 0.0
if cat_lat is None:
cat_lat = np.pi / 2
if cat_frame is None:
cat_frame = "altaz"
if phase_dict is None:
phase_dict = {
"cat_type": cat_type,
"cat_lon": cat_lon,
"cat_lat": cat_lat,
"cat_frame": cat_frame,
"cat_epoch": cat_epoch,
"cat_times": cat_times,
"cat_pm_ra": cat_pm_ra,
"cat_pm_dec": cat_pm_dec,
"cat_dist": cat_dist,
"cat_vrad": cat_vrad,
}
this_dict = self.phase_center_catalog
tol_dict = {
"cat_type": None,
"cat_lon": radian_tols,
"cat_lat": radian_tols,
"cat_frame": None,
"cat_epoch": None,
"cat_times": default_tols,
"cat_pm_ra": default_tols,
"cat_pm_dec": default_tols,
"cat_dist": default_tols,
"cat_vrad": default_tols,
}
if target_cat_id is not None:
if target_cat_id not in self.phase_center_catalog:
raise ValueError(f"No phase center with ID number {target_cat_id}.")
name_dict = {
target_cat_id: self.phase_center_catalog[target_cat_id]["cat_name"]
}
else:
name_dict = {
key: cat_dict["cat_name"]
for key, cat_dict in self.phase_center_catalog.items()
}
for cat_id, name in name_dict.items():
cat_diffs = 0
if (cat_name != name) and (not ignore_name):
continue
check_dict = this_dict[cat_id]
for key in tol_dict.keys():
if phase_dict.get(key) is not None:
if check_dict.get(key) is None:
cat_diffs += 1
elif tol_dict[key] is None:
# If no tolerance specified, expect attributes to be identical
cat_diffs += phase_dict.get(key) != check_dict.get(key)
else:
# allclose will throw a Value error if you have two arrays
# of different shape, which we can catch to flag that
# the two arrays are actually not within tolerance.
if np.shape(phase_dict[key]) != np.shape(check_dict[key]):
cat_diffs += 1
else:
cat_diffs += not np.allclose(
phase_dict[key],
check_dict[key],
tol_dict[key][0],
tol_dict[key][1],
)
else:
cat_diffs += check_dict[key] is not None
if (cat_diffs == 0) or (cat_name == name):
if cat_diffs < match_diffs:
# If our current match is an improvement on any previous matches,
# then record it as the best match.
match_id = cat_id
match_diffs = cat_diffs
if match_diffs == 0:
# If we have a total match, we can bail at this point
break
return match_id, match_diffs
def _add_phase_center(
self,
cat_name,
cat_type=None,
cat_lon=None,
cat_lat=None,
cat_frame=None,
cat_epoch=None,
cat_times=None,
cat_pm_ra=None,
cat_pm_dec=None,
cat_dist=None,
cat_vrad=None,
info_source="user",
force_update=False,
cat_id=None,
):
"""
Add an entry to the internal object/source catalog or find a matching one.
This is a helper function for identifying a adding a phase center to the
internal catalog, contained within the attribute `phase_center_catalog`, unless
a phase center already exists that matches the passed parameters. If a matching
phase center is found, the catalog ID associated with that phase center is
returned.
Parameters
----------
cat_name : str
Name of the phase center to be added.
cat_type : str
Type of phase center to be added. Must be one of:
"sidereal" (fixed RA/Dec),
"ephem" (RA/Dec that moves with time),
"driftscan" (fixed az/el position),
"unprojected" (no w-projection, equivalent to the old
`phase_type` == "drift").
cat_lon : float or ndarray
Value of the longitudinal coordinate (e.g., RA, Az, l) in radians of the
phase center. No default unless `cat_type="unprojected"`, in which case the
default is zero. Expected to be a float for sidereal and driftscan phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
cat_lat : float or ndarray
Value of the latitudinal coordinate (e.g., Dec, El, b) in radians of the
phase center. No default unless `cat_type="unprojected"`, in which case the
default is pi/2. Expected to be a float for sidereal and driftscan phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
cat_frame : str
Coordinate frame that cat_lon and cat_lat are given in. Only used
for sidereal and ephem targets. Can be any of the several supported frames
in astropy (a limited list: fk4, fk5, icrs, gcrs, cirs, galactic).
cat_epoch : str or float
Epoch of the coordinates, only used when cat_frame = fk4 or fk5. Given
in units of fractional years, either as a float or as a string with the
epoch abbreviation (e.g, Julian epoch 2000.0 would be J2000.0).
cat_times : ndarray of floats
Only used when `cat_type="ephem"`. Describes the time for which the values
of `cat_lon` and `cat_lat` are caclulated, in units of JD. Shape is (Npts,).
cat_pm_ra : float
Proper motion in RA, in units of mas/year. Only used for sidereal phase
centers.
cat_pm_dec : float
Proper motion in Dec, in units of mas/year. Only used for sidereal phase
centers.
cat_dist : float or ndarray of float
Distance of the source, in units of pc. Only used for sidereal and ephem
phase centers. Expected to be a float for sidereal and driftscan phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
cat_vrad : float or ndarray of float
Radial velocity of the source, in units of km/s. Only used for sidereal and
ephem phase centers. Expected to be a float for sidereal and driftscan phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
info_source : str
Optional string describing the source of the information provided. Used
primarily in UVData to denote when an ephemeris has been supplied by the
JPL-Horizons system, user-supplied, or read in by one of the various file
interpreters. Default is 'user'.
force_update : bool
Normally, `_add_phase_center` will throw an error if there already exists a
phase_center with the given cat_id. However, if one sets
`force_update=True`, the method will overwrite the existing entry in
`phase_center_catalog` with the parameters supplied. Note that doing this
will _not_ update other atributes of the `UVData` object. Default is False.
cat_id : int
An integer signifying the ID number for the phase center, used in the
`phase_center_id_array` attribute. If a matching phase center entry exists
already, that phase center ID will be returned, which may be different than
the value specified to this parameter. The default is for the method to
assign this value automatically.
Returns
-------
cat_id : int
The unique ID number for the phase center that either matches the specified
parameters or was added to the internal catalog. If a matching entry was
found, this may not be the value passed to the `cat_id` parameter. This
value is used in the `phase_center_id_array` attribute to denote which
source a given baseline-time corresponds to.
Raises
------
ValueError
If attempting to add a non-unique source name or if adding a sidereal
source without coordinates.
"""
if not isinstance(cat_name, str):
raise ValueError("cat_name must be a string.")
if cat_type == "unphased":
warnings.warn(
"The 'unphased' catalog type has been renamed to 'unprojected'. Using "
"unprojected for now, this warning will become an error in version 2.4",
DeprecationWarning,
)
cat_type = "unprojected"
if cat_name == "unphased":
cat_name = "unprojected"
# We currently only have 4 supported types -- make sure the user supplied
# one of those
if cat_type not in ["sidereal", "ephem", "driftscan", "unprojected"]:
raise ValueError(
"Only sidereal, ephem, driftscan or unprojected may be used "
"for cat_type."
)
# Both proper motion parameters need to be set together
if (cat_pm_ra is None) != (cat_pm_dec is None):
raise ValueError(
"Must supply values for either both or neither of "
"cat_pm_ra and cat_pm_dec."
)
# If left unset, unprojected and driftscan defaulted to Az, El = (0 deg, 90 deg)
if cat_type in ["unprojected", "driftscan"]:
if cat_lon is None:
cat_lon = 0.0
if cat_lat is None:
cat_lat = np.pi / 2
if cat_frame is None:
cat_frame = "altaz"
# check some case-specific things and make sure all the entries are acceptable
if (cat_times is None) and (cat_type == "ephem"):
raise ValueError("cat_times cannot be None for ephem object.")
elif (cat_times is not None) and (cat_type != "ephem"):
raise ValueError("cat_times cannot be used for non-ephem phase centers.")
if (cat_lon is None) and (cat_type in ["sidereal", "ephem"]):
raise ValueError(
"cat_lon cannot be None for sidereal or ephem phase centers."
)
if (cat_lat is None) and (cat_type in ["sidereal", "ephem"]):
raise ValueError(
"cat_lat cannot be None for sidereal or ephem phase centers."
)
if (cat_frame is None) and (cat_type in ["sidereal", "ephem"]):
raise ValueError(
"cat_frame cannot be None for sidereal or ephem phase centers."
)
elif (cat_frame != "altaz") and (cat_type in ["driftscan", "unprojected"]):
raise ValueError(
"cat_frame must be either None or 'altaz' when the cat type "
"is either driftscan or unprojected."
)
if (cat_type == "unprojected") and (cat_lon != 0.0):
raise ValueError(
"Catalog entries that are unprojected must have cat_lon set to either "
"0 or None."
)
if (cat_type == "unprojected") and (cat_lat != (np.pi / 2)):
raise ValueError(
"Catalog entries that are unprojected must have cat_lat set to either "
"pi/2 or None."
)
if (cat_type != "sidereal") and (
(cat_pm_ra is not None) or (cat_pm_dec is not None)
):
raise ValueError(
"Non-zero proper motion values (cat_pm_ra, cat_pm_dec) "
"for cat types other than sidereal are not supported."
)
if isinstance(cat_epoch, Time) or isinstance(cat_epoch, str):
if cat_frame in ["fk4", "fk4noeterms"]:
cat_epoch = Time(cat_epoch).byear
else:
cat_epoch = Time(cat_epoch).jyear
elif cat_epoch is not None:
cat_epoch = float(cat_epoch)
if cat_type == "ephem":
cat_times = np.array(cat_times, dtype=float).reshape(-1)
cshape = cat_times.shape
try:
cat_lon = np.array(cat_lon, dtype=float).reshape(cshape)
cat_lat = np.array(cat_lat, dtype=float).reshape(cshape)
if cat_dist is not None:
cat_dist = np.array(cat_dist, dtype=float).reshape(cshape)
if cat_vrad is not None:
cat_vrad = np.array(cat_vrad, dtype=float).reshape(cshape)
except ValueError as err:
raise ValueError(
"Object properties -- lon, lat, pm_ra, pm_dec, dist, vrad -- must "
"be of the same size as cat_times for ephem phase centers."
) from err
else:
if cat_lon is not None:
cat_lon = float(cat_lon)
cat_lon = None if cat_lon is None else float(cat_lon)
cat_lat = None if cat_lat is None else float(cat_lat)
cat_pm_ra = None if cat_pm_ra is None else float(cat_pm_ra)
cat_pm_dec = None if cat_pm_dec is None else float(cat_pm_dec)
cat_dist = None if cat_dist is None else float(cat_dist)
cat_vrad = None if cat_vrad is None else float(cat_vrad)
if self.phase_center_catalog is not None:
# We want to create a unique ID for each source, for use in indexing arrays.
# The logic below ensures that we pick the lowest positive integer that is
# not currently being used by another source
if cat_id is None:
cat_id = (
set(range(self.Nphase + 1))
.difference(self.phase_center_catalog)
.pop()
)
elif cat_id in self.phase_center_catalog and not force_update:
raise ValueError(
"Provided cat_id belongs to another source (%s)."
% self.phase_center_catalog[cat_id]["cat_name"]
)
# Let's warn if this entry has the same name as an existing one
temp_id, cat_diffs = self._look_in_catalog(
cat_name=cat_name,
cat_type=cat_type,
cat_lon=cat_lon,
cat_lat=cat_lat,
cat_frame=cat_frame,
cat_epoch=cat_epoch,
cat_times=cat_times,
cat_pm_ra=cat_pm_ra,
cat_pm_dec=cat_pm_dec,
cat_dist=cat_dist,
cat_vrad=cat_vrad,
)
# If the source does have the same name, check to see if all the
# atributes match. If so, no problem, go about your business
if temp_id is not None:
if cat_diffs == 0:
# Everything matches, return the catalog ID of the matching entry
return temp_id
else:
warnings.warn(
f"The provided name {cat_name} is already used but has "
"different parameters. Adding another entry with the same name "
"but a different ID and parameters."
)
else:
if cat_id is None:
cat_id = 0
# If source is unique, begin creating a dictionary for it
phase_dict = {
"cat_name": cat_name,
"cat_type": cat_type,
"cat_lon": cat_lon,
"cat_lat": cat_lat,
"cat_frame": cat_frame,
"cat_epoch": cat_epoch,
"cat_times": cat_times,
"cat_pm_ra": cat_pm_ra,
"cat_pm_dec": cat_pm_dec,
"cat_vrad": cat_vrad,
"cat_dist": cat_dist,
"info_source": info_source,
}
if self.phase_center_catalog is not None:
self.phase_center_catalog[cat_id] = phase_dict
else:
self.phase_center_catalog = {cat_id: phase_dict}
self.Nphase = len(self.phase_center_catalog)
return cat_id
def _remove_phase_center(self, defunct_id):
"""
Remove an entry from the internal object/source catalog.
Removes an entry from the attribute `phase_center_catalog`.
Parameters
----------
defunct_id : int
Catalog ID of the source to be removed
Raises
------
IndexError
If the name provided is not found as a key in `phase_center_catalog`
"""
if defunct_id not in self.phase_center_catalog:
raise IndexError("No source by that ID contained in the catalog.")
del self.phase_center_catalog[defunct_id]
self.Nphase = len(self.phase_center_catalog)
def _clear_unused_phase_centers(self):
"""
Remove objects dictionaries and names that are no longer in use.
Goes through the `phase_center_catalog` attribute in of a UVData object and
clears out entries that are no longer being used, and appropriately updates
`phase_center_id_array` accordingly. This function is not typically called
by users, but instead is used by other methods.
"""
unique_cat_ids = np.unique(self.phase_center_id_array)
defunct_list = []
Nphase = 0
for cat_id in self.phase_center_catalog:
if cat_id in unique_cat_ids:
Nphase += 1
else:
defunct_list.append(cat_id)
# Check the number of "good" sources we have -- if we haven't dropped any,
# then we are free to bail, otherwise update the Nphase attribute
if Nphase == self.Nphase:
return
# Time to kill the entries that are no longer in the source stack
for defunct_id in defunct_list:
self._remove_phase_center(defunct_id)
def _check_for_cat_type(self, cat_type):
"""
Check which Nblts have a cat_type in `cat_type`.
This convenience method returns back a boolean mask to identify which data
along the Blt axis contains data in cat_type.
Parameters
----------
cat_type : str or list of str
Phase types to check for.
Returns
-------
blt_mask : ndarray of bool
A boolean mask for identifying which elements contain unprojected objects.
True where not projected, False where projected.
"""
# Check and see if we have any data with cat_type.
if not isinstance(cat_type, (list, tuple, np.ndarray)):
cat_type = [cat_type]
cat_type_list = [
cat_id
for cat_id, cat_dict in self.phase_center_catalog.items()
if cat_dict["cat_type"] in cat_type
]
# Construct a bool mask
blt_mask = np.isin(self.phase_center_id_array, cat_type_list)
return blt_mask
def rename_phase_center(self, catalog_identifier, new_name):
"""
Rename a phase center/catalog entry within a multi phase center data set.
Parameters
----------
catalog_identifier : str or int or list of int
Unique identifier of a phase center to be renamed. If supplied as a str,
will be matched against the phase center name. Otherwise if an int or list
of int, assumed to be the catalog ID number(s).
new_name : str
New name for the phase center.
Raises
------
ValueError
If attempting to run the method on a non multi phase center data set, or if
`catalog_identifier` is not found in `phase_center_catalog`.
TypeError
If `new_name` is not actually a string or if `catalog_identifier` is not a
string or an integer.
"""
if isinstance(catalog_identifier, (str, int)):
pass
elif isinstance(catalog_identifier, list) and all(
isinstance(cat, int) for cat in catalog_identifier
):
pass
else:
raise TypeError(
"catalog_identifier must be a string, an integer or a list of integers."
)
if isinstance(catalog_identifier, str):
cat_id = []
for key, ps_dict in self.phase_center_catalog.items():
if ps_dict["cat_name"] == catalog_identifier:
cat_id.append(key)
if len(cat_id) == 0:
raise ValueError(
f"No entry by the name {catalog_identifier} in the catalog."
)
else:
# Force cat_id to be a list to make downstream code simpler. If cat_id is
# an int, it will throw a TypeError on casting to list, which we can catch.
try:
cat_id = list(catalog_identifier)
except TypeError:
cat_id = [catalog_identifier]
for key in cat_id:
if key not in self.phase_center_catalog:
raise ValueError(f"No entry with the ID {key} in the catalog.")
if not isinstance(new_name, str):
raise TypeError("Value provided to new_name must be a string.")
if (new_name == catalog_identifier) or (len(cat_id) == 0):
# This is basically just a no-op, so return to user
return
for key in cat_id:
self.phase_center_catalog[key]["cat_name"] = new_name
def split_phase_center(
self,
catalog_identifier,
new_name=None,
select_mask=None,
new_id=None,
downselect=False,
):
"""
Rename the phase center (but preserve other properties) of a subset of data.
Allows you to rename a subset of the data phased to a particular phase center,
marked by a different name than the original. Useful when you want to phase to
one position, but want to differentiate different groups of data (e.g., marking
every other integration to make jackknifing easier).
Parameters
----------
catalog_identifier : str or int
Unique identifier of a phase center to be renamed. If supplied as a str,
will be matched against the phase center name. Otherwise if an int, assumed
to be the catalog ID number.
new_name : str
Name for the "split" portion of the phase center. Optional argument, default
is to use the same name as the existing phase center.
select_mask : array_like
Selection mask for which data should be identified as belonging to the phase
center labeled by `new_name`. Any array-like able to be used as an index
is suitable -- the most typical is an array of bool with length `Nblts`,
or an array of ints within the range (-Nblts, Nblts).
new_id : int
Catalog ID to assign to the new phase center. Optional argument, a unique
value will be automatically assigned if not provided.
downselect : bool
If selecting data that is not marked as belonging to `cat_name`,
normally an error is thrown. By setting this to True, `select_mask` will
be modified to exclude data not marked as belonging to `cat_name`.
Raises
------
ValueError
If catalog_identifier is not an int or string or if it is not found as a
cat_name in `phase_center_catalog` or if it is a string and is found
multiple times. If new_id is not an int or already exists in the catalog.
If new_name is not a string. If `select_mask` contains data that doesn't
belong to catalog_identifier, unless `downselect` is True.
IndexError
If select_mask is not a valid indexing array.
UserWarning
If all data for `cat_name` was selected (in which case `rename_phase_center`
is called instead), or if no valid data was selected.
"""
# Check to make sure that everything lines up with
if not isinstance(catalog_identifier, (str, int)):
raise TypeError("catalog_identifier must be a string or an integer.")
if isinstance(catalog_identifier, str):
cat_id = None
for pc_id, pc_dict in self.phase_center_catalog.items():
if pc_dict["cat_name"] == catalog_identifier:
if cat_id is None:
cat_id = pc_id
else:
raise ValueError(
f"The cat_name {catalog_identifier} has multiple entries "
"in the catalog. Please specify a cat_id in order to "
"eliminate ambiguity (which you can see using the "
"`print_phase_center_info` method)."
)
if cat_id is None:
raise ValueError(
f"No catalog entries matching the name {catalog_identifier}."
)
else:
cat_id = catalog_identifier
if cat_id not in self.phase_center_catalog:
raise ValueError(f"No entry with the ID {cat_id} found in the catalog.")
if new_id is None:
new_id = (
set(range(self.Nphase + 1)).difference(self.phase_center_catalog).pop()
)
elif not isinstance(new_id, int):
raise TypeError("Value provided to new_id must be an int.")
if new_id in self.phase_center_catalog:
raise ValueError(
f"The ID {new_id} is already in the catalog, choose another value for "
"new_id."
)
if not (isinstance(new_name, str) or new_name is None):
raise TypeError("Value provided to new_name must be a string.")
try:
inv_mask = np.ones(self.Nblts, dtype=bool)
inv_mask[select_mask] = False
except IndexError as err:
raise IndexError(
"select_mask must be an array-like, either of ints with shape (Nblts), "
"or of ints within the range (-Nblts, Nblts)."
) from err
# If we have selected any entries that don't correspond to the cat_id
# in question, either downselect or raise an error.
if np.any(
np.isin(self.phase_center_id_array[select_mask], cat_id, invert=True)
):
if downselect:
inv_mask |= np.isin(self.phase_center_id_array, cat_id, invert=True)
select_mask = ~inv_mask
else:
raise ValueError(
"Data selected with select_mask includes data which is not part of "
"the selected phase_center ({cat_id}). You can fix this by either "
"revising select_mask or setting downselect=True."
)
# Now check for no(-ish) ops
if np.all(inv_mask):
# You didn't actually select anything we could change
warnings.warn("No relevant data selected - check select_mask.")
elif not np.any(np.isin(self.phase_center_id_array[inv_mask], cat_id)):
# No matching catalog IDs found outside the range, so this is really a
# replace more than a split.
warnings.warn(
"All data for the source selected - updating the cat_id instead."
)
self._update_phase_center_id(cat_id, new_id)
if new_name is not None:
self.rename_phase_center(new_id, new_name)
else:
temp_dict = self.phase_center_catalog[cat_id]
cat_id = self._add_phase_center(
temp_dict["cat_name"] if new_name is None else new_name,
cat_type=temp_dict["cat_type"],
cat_lon=temp_dict.get("cat_lon"),
cat_lat=temp_dict.get("cat_lat"),
cat_frame=temp_dict.get("cat_frame"),
cat_epoch=temp_dict.get("cat_epoch"),
cat_times=temp_dict.get("cat_times"),
cat_pm_ra=temp_dict.get("cat_pm_ra"),
cat_pm_dec=temp_dict.get("cat_pm_dec"),
cat_dist=temp_dict.get("cat_dist"),
cat_vrad=temp_dict.get("cat_vrad"),
cat_id=new_id,
force_update=True,
)
self.phase_center_id_array[select_mask] = cat_id
def merge_phase_centers(
self, catalog_identifier, force_merge=False, ignore_name=False
):
"""
Merge two differently named objects into one within a multi-phase-ctr data set.
Recombines two different objects into a single catalog entry -- useful if
having previously used `split_phase_center` or when multiple objects with
different names share the same source parameters.
Parameters
----------
catalog_identifier : str or int or list of str or int
Unique identifier(s) of a phase center to be merged. If supplied as strings,
will be matched against the phase center name. Otherwise if supplied as
integers, assumed to be the catalog ID number(s).
force_merge : bool
Normally, the method will throw an error if the phase center properties
differ for the catalogs listed in catalog_identifier. This can be overriden
by setting this to True. Default is False.
ignore_name : bool
When comparing phase centers, all attributes are normally checked. However,
if set to True, the catalog name ("cat_name") will be ignored when
performing the comparison. Default is False.
Raises
------
ValueError
If catname1 or catname2 are not found in the UVData object, of if their
properties differ (and `force_merge` is not set to True).
UserWarning
If forcing the merge of two objects with different properties.
"""
if isinstance(catalog_identifier, (str, int)):
pass
elif isinstance(catalog_identifier, list) and all(
isinstance(cat, (str, int)) for cat in catalog_identifier
):
pass
else:
raise TypeError(
"catalog_identifier must be a string, an integer or a list of strings "
"or integers."
)
if not isinstance(catalog_identifier, list):
catalog_identifier = [catalog_identifier]
cat_id_list = []
for cat in catalog_identifier:
if isinstance(cat, str):
this_list = []
for key, ps_dict in self.phase_center_catalog.items():
if ps_dict["cat_name"] == cat:
this_list.append(key)
if len(this_list) == 0:
raise ValueError(f"No entry by the name {cat} in the catalog.")
cat_id_list.extend(this_list)
else:
# Force cat_id to be a list to make downstream code simpler. If cat_id
# is an int, it will throw a TypeError on casting to list, which we can
# catch.
if cat not in self.phase_center_catalog:
raise ValueError(f"No entry with the ID {cat} in the catalog.")
cat_id_list.append(cat)
# Check for the no-op
if len(cat_id_list) < 2:
warnings.warn(
"Selection matches less than two phase centers, no need to merge."
)
return
# First, let's check and see if the dict entries are identical
for cat_id in cat_id_list[1:]:
pc_id, pc_diffs = self._look_in_catalog(
phase_dict=self.phase_center_catalog[cat_id],
ignore_name=ignore_name,
target_cat_id=cat_id_list[0],
)
if (pc_diffs != 0) or (pc_id is None):
if force_merge:
warnings.warn(
"Forcing fields together, even though their attributes differ."
)
else:
raise ValueError(
"Attributes of phase centers differ in phase_center_catalog. "
"You can ignore this error and force merge_phase_centers to "
"complete by setting force_merge=True, but this should be done "
"with substantial caution."
)
# Set everything to the first cat ID in the list
self.phase_center_id_array[
np.isin(self.phase_center_id_array, cat_id_list)
] = cat_id_list[0]
# Finally, remove the defunct cat IDs
for cat_id in cat_id_list[1:]:
self._remove_phase_center(cat_id)
def print_phase_center_info(
self,
catalog_identifier=None,
hms_format=None,
return_str=False,
print_table=True,
):
"""
Print out the details of the phase centers.
Prints out an ASCII table that contains the details of the
`phase_center_catalog` attribute, which acts as the internal source catalog
for UVData objects.
Parameters
----------
catalog_identifier : str or int or list of str or int
Optional parameter which, if provided, will cause the method to only return
information on the phase center(s) with the matching name(s) or catalog ID
number(s). Default is to print out information on all catalog entries.
hms_format : bool
Optional parameter, which if selected, can be used to force coordinates to
be printed out in Hours-Min-Sec (if set to True) or Deg-Min-Sec (if set to
False) format. Default is to print out in HMS if all the objects have
coordinate frames of icrs, gcrs, fk5, fk4, and top; otherwise, DMS format
is used.
return_str: bool
If set to True, the method returns an ASCII string which contains all the
table infrmation. Default is False.
print_table : bool
If set to True, prints the table to the terminal window. Default is True.
Returns
-------
table_str : bool
If return_str=True, an ASCII string containing the entire table text
Raises
------
ValueError
If `cat_name` matches no keys in `phase_center_catalog`.
"""
r2d = 180.0 / np.pi
r2m = 60.0 * 180.0 / np.pi
r2s = 3600.0 * 180.0 / np.pi
ra_frames = ["icrs", "gcrs", "fk5", "fk4", "topo"]
if catalog_identifier is not None:
if isinstance(catalog_identifier, (str, int)):
pass
elif isinstance(catalog_identifier, list) and all(
isinstance(cat, (str, int)) for cat in catalog_identifier
):
pass
else:
raise TypeError(
"catalog_identifier must be a string, an integer or a list of "
"strings or integers."
)
if not isinstance(catalog_identifier, list):
catalog_identifier = [catalog_identifier]
cat_id_list = []
for cat in catalog_identifier:
if isinstance(cat, str):
this_list = []
for key, ps_dict in self.phase_center_catalog.items():
if ps_dict["cat_name"] == cat:
this_list.append(key)
if len(this_list) == 0:
raise ValueError(f"No entry by the name {cat} in the catalog.")
cat_id_list.extend(this_list)
else:
# Force cat_id to be a list to make downstream code simpler.
# If cat_id is an int, it will throw a TypeError on casting to
# list, which we can catch.
if cat not in self.phase_center_catalog:
raise ValueError(f"No entry with the ID {cat} in the catalog.")
cat_id_list.append(cat)
else:
cat_id_list = list(self.phase_center_catalog)
dict_list = [self.phase_center_catalog[cat_id] for cat_id in cat_id_list]
# We want to check and actually see which fields we need to
# print
any_lon = any_lat = any_frame = any_epoch = any_times = False
any_pm_ra = any_pm_dec = any_dist = any_vrad = False
for indv_dict in dict_list:
any_lon = any_lon or indv_dict.get("cat_lon") is not None
any_lat = any_lat or indv_dict.get("cat_lat") is not None
any_frame = any_frame or indv_dict.get("cat_frame") is not None
any_epoch = any_epoch or indv_dict.get("cat_epoch") is not None
any_times = any_times or indv_dict.get("cat_times") is not None
any_pm_ra = any_pm_ra or indv_dict.get("cat_pm_ra") is not None
any_pm_dec = any_pm_dec or indv_dict.get("cat_pm_dec") is not None
any_dist = any_dist or indv_dict.get("cat_dist") is not None
any_vrad = any_vrad or indv_dict.get("cat_vrad") is not None
if any_lon and (hms_format is None):
cat_frame = indv_dict.get("cat_frame")
cat_type = indv_dict["cat_type"]
if (cat_frame not in ra_frames) or (cat_type == "driftscan"):
hms_format = False
if hms_format is None:
hms_format = True
col_list = []
col_list.append(
{"hdr": ("ID", "#"), "fmt": "% 4i", "field": " %4s ", "name": "cat_id"}
)
col_list.append(
{
"hdr": ("Cat Entry", "Name"),
"fmt": "%12s",
"field": " %12s ",
"name": "cat_name",
}
)
col_list.append(
{"hdr": ("Type", ""), "fmt": "%12s", "field": " %12s ", "name": "cat_type"}
)
if any_lon:
col_list.append(
{
"hdr": ("Az/Lon/RA", "hours" if hms_format else "deg"),
"fmt": "% 3i:%02i:%05.2f",
"field": (" %12s " if hms_format else " %13s "),
"name": "cat_lon",
}
)
if any_lat:
col_list.append(
{
"hdr": ("El/Lat/Dec", "deg"),
"fmt": "%1s%2i:%02i:%05.2f",
"field": " %12s ",
"name": "cat_lat",
}
)
if any_frame:
col_list.append(
{
"hdr": ("Frame", ""),
"fmt": "%5s",
"field": " %5s ",
"name": "cat_frame",
}
)
if any_epoch:
col_list.append(
{
"hdr": ("Epoch", ""),
"fmt": "%7s",
"field": " %7s ",
"name": "cat_epoch",
}
)
if any_times:
col_list.append(
{
"hdr": (" Ephem Range ", "Start-MJD End-MJD"),
"fmt": " %8.2f % 8.2f",
"field": " %20s ",
"name": "cat_times",
}
)
if any_pm_ra:
col_list.append(
{
"hdr": ("PM-Ra", "mas/yr"),
"fmt": "%.4g",
"field": " %6s ",
"name": "cat_pm_ra",
}
)
if any_pm_dec:
col_list.append(
{
"hdr": ("PM-Dec", "mas/yr"),
"fmt": "%.4g",
"field": " %6s ",
"name": "cat_pm_dec",
}
)
if any_dist:
col_list.append(
{
"hdr": ("Dist", "pc"),
"fmt": "%.1e",
"field": " %7s ",
"name": "cat_dist",
}
)
if any_vrad:
col_list.append(
{
"hdr": ("V_rad", "km/s"),
"fmt": "%.4g",
"field": " %6s ",
"name": "cat_vrad",
}
)
top_str = ""
bot_str = ""
for col in col_list:
top_str += col["field"] % col["hdr"][0]
bot_str += col["field"] % col["hdr"][1]
info_str = ""
info_str += top_str + "\n"
info_str += bot_str + "\n"
info_str += ("-" * len(bot_str)) + "\n"
# We want to print in the order of cat_id
for idx in np.argsort(cat_id_list):
tbl_str = ""
for col in col_list:
# If we have a "special" field that needs extra handling,
# take care of that up front
if col["name"] == "cat_id":
temp_val = cat_id_list[idx]
else:
temp_val = dict_list[idx][col["name"]]
if temp_val is None:
temp_str = ""
elif col["name"] == "cat_lon":
# Force the longitude component to be a positive value
temp_val = np.mod(np.median(temp_val), 2 * np.pi)
temp_val /= 15.0 if hms_format else 1.0
coord_tuple = (
np.mod(temp_val * r2d, 360.0),
np.mod(temp_val * r2m, 60.0),
np.mod(temp_val * r2s, 60.0),
)
temp_str = col["fmt"] % coord_tuple
elif col["name"] == "cat_lat":
temp_val = np.median(temp_val)
coord_tuple = (
"-" if temp_val < 0.0 else "+",
np.mod(np.abs(temp_val) * r2d, 360.0),
np.mod(np.abs(temp_val) * r2m, 60.0),
np.mod(np.abs(temp_val) * r2s, 60.0),
)
temp_str = col["fmt"] % coord_tuple
elif col["name"] == "cat_epoch":
use_byrs = dict_list[idx]["cat_frame"] in ["fk4", "fk4noeterms"]
temp_val = ("B%6.1f" if use_byrs else "J%6.1f") % temp_val
temp_str = col["fmt"] % temp_val
elif col["name"] == "cat_times":
time_tuple = (
np.min(temp_val) - 2400000.5,
np.max(temp_val) - 2400000.5,
)
temp_str = col["fmt"] % time_tuple
elif (col["name"] == "cat_dist") or (col["name"] == "cat_vrad"):
temp_val = np.median(temp_val)
temp_str = col["fmt"] % temp_val
else:
temp_str = col["fmt"] % temp_val
tbl_str += col["field"] % temp_str
info_str += tbl_str + "\n"
if print_table:
# We need this extra bit of code to handle trailing whitespace, since
# otherwise some checks (e.g., doc check on tutorials) will balk
print(
"\n".join([line.rstrip() for line in info_str.split("\n")]), end=""
) # pragma: nocover
if return_str:
return info_str
def _update_phase_center_id(self, cat_id, new_id=None, reserved_ids=None):
"""
Update a phase center with a new catalog ID number.
Parameters
----------
cat_id : int
Current catalog ID of the phase center, which corresponds to a key in the
attribute `phase_center_catalog`.
new_id : int
Optional argument. If supplied, then the method will attempt to use the
provided value as the new catalog ID, provided that an existing catalog
entry is not already using the same value. If not supplied, then the
method will automatically assign a value.
reserved_ids : array-like in int
Optional argument. An array-like of ints that denotes which ID numbers
are already reserved. Useful for when combining two separate catalogs.
Raises
------
ValueError
If not using the method on a multi-phase-ctr data set, if there's no entry
that matches `cat_name`, or of the value `new_id` is already taken.
"""
if cat_id not in self.phase_center_catalog:
raise ValueError(
f"Cannot run _update_phase_center_id: no entry with id {cat_id}."
)
used_cat_ids = set(self.phase_center_catalog)
used_cat_ids.remove(cat_id)
if reserved_ids is not None:
used_cat_ids = used_cat_ids.union(reserved_ids)
if new_id is None:
# If the old ID is in the reserved list, then we'll need to update it
if cat_id not in used_cat_ids:
# Don't need to actually update anything
return
new_id = set(range(len(used_cat_ids) + 1)).difference(used_cat_ids).pop()
else:
if new_id in used_cat_ids:
raise ValueError("Catalog ID supplied already taken by another source.")
self.phase_center_id_array[self.phase_center_id_array == cat_id] = new_id
self.phase_center_catalog[new_id] = self.phase_center_catalog.pop(cat_id)
def _consolidate_phase_center_catalogs(
self, reference_catalog=None, other=None, ignore_name=False
):
"""
Consolidate phase center catalogs with a reference or another object.
This is a helper method which updates the phase_center_catalog and related
parameters to make this object consistent with a reference catalog or with
another object so the second object can be added or concatenated to this object.
If both `reference_catalog` and `other` are provided, both this object and the
one passed to `other` will have their catalogs updated.
Parameters
----------
reference_calalog : dict
A reference catalog to make this object consistent with.
other : UVData object
A UVData object which self needs to be consistent with because it will be
added to self. The phase_center_catalog from other is used as the reference
catalog if the reference_catalog is None. If `reference_catalog` is also
set, the phase_center_catalog on other will also be modified to be
consistent with the `reference_catalog`.
ignore_name : bool
Option to ignore the name of the phase center (`cat_name` in
`phase_center_catalog`) when identifying matching phase centers. If set to
True, phase centers that are the same up to their name will be combined with
the name set to the reference catalog name or the name found in the first
UVData object. If set to False, phase centers that are the same up to the
name will be kept as separate phase centers. Default is False.
"""
if reference_catalog is None and other is None:
raise ValueError(
"Either the reference_catalog or the other parameter must be set."
)
if reference_catalog is None:
reference_catalog = other.phase_center_catalog
elif other is not None:
# first update other to be consistent with the reference
other._consolidate_phase_center_catalogs(
reference_catalog=reference_catalog
)
# then use the updated other as the reference
reference_catalog = other.phase_center_catalog
reserved_ids = list(reference_catalog)
# First loop, we want to update all the catalog IDs so that we know there
# are no conflicts with the reference
for cat_id in list(self.phase_center_catalog):
self._update_phase_center_id(cat_id, reserved_ids=reserved_ids)
# Next loop, we want to update the IDs of sources that are in the reference.
for cat_id in list(reference_catalog):
match_id, match_diffs = self._look_in_catalog(
phase_dict=reference_catalog[cat_id], ignore_name=ignore_name
)
if match_id is not None and match_diffs == 0:
# If match_diffs is 0 then all the keys in the phase center catalog
# match, so this is functionally the same source
self._update_phase_center_id(match_id, new_id=cat_id)
if ignore_name:
# Make the names match if names were ignored in matching
self.phase_center_catalog[cat_id]["cat_name"] = reference_catalog[
cat_id
]["cat_name"]
# _look_in_catalog ignores the "info_source" field. Update it so
# it matches the reference.
self.phase_center_catalog[cat_id]["info_source"] = reference_catalog[
cat_id
]["info_source"]
# Finally, add those other objects not found in self
for cat_id in reference_catalog:
if cat_id not in self.phase_center_catalog:
self._add_phase_center(
reference_catalog[cat_id]["cat_name"],
cat_type=reference_catalog[cat_id]["cat_type"],
cat_lon=reference_catalog[cat_id]["cat_lon"],
cat_lat=reference_catalog[cat_id]["cat_lat"],
cat_frame=reference_catalog[cat_id]["cat_frame"],
cat_epoch=reference_catalog[cat_id]["cat_epoch"],
cat_times=reference_catalog[cat_id]["cat_times"],
cat_pm_ra=reference_catalog[cat_id]["cat_pm_ra"],
cat_pm_dec=reference_catalog[cat_id]["cat_pm_dec"],
cat_dist=reference_catalog[cat_id]["cat_dist"],
cat_vrad=reference_catalog[cat_id]["cat_vrad"],
info_source=reference_catalog[cat_id]["info_source"],
cat_id=cat_id,
)
@property
def _data_params(self):
"""List of strings giving the data-like parameters."""
return ["data_array", "nsample_array", "flag_array"]
@property
def data_like_parameters(self):
"""Iterate defined parameters which are data-like (not metadata-like)."""
for key in self._data_params:
if hasattr(self, key):
yield getattr(self, key)
@property
def metadata_only(self):
"""
Property that determines whether this is a metadata only object.
An object is metadata only if data_array, nsample_array and flag_array
are all None.
"""
metadata_only = all(d is None for d in self.data_like_parameters)
for param_name in self._data_params:
getattr(self, "_" + param_name).required = not metadata_only
return metadata_only
def _old_phase_attributes_compatible(self):
"""
Check if this object is compatible with the old phase attributes.
Returns
-------
compatible : bool
True if this object is compatible with the old phase attributes, False
otherwise
reason : str
Reason it is not compatible. None if compatible is True.
"""
if self.Nphase > 1:
return False, "multiple phase centers"
phase_dict = list(self.phase_center_catalog.values())[0]
if phase_dict["cat_type"] not in ["sidereal", "unprojected"]:
return False, f"{phase_dict['cat_type']} phase centers"
if phase_dict["cat_type"] == "sidereal" and phase_dict["cat_frame"] not in [
"icrs",
"gcrs",
]:
return False, f"{phase_dict['cat_frame']} phase frames"
else:
return True, None
def __getattribute__(self, __name):
"""Issue deprecation warnings for old phasing attributes."""
old_phase_attrs = [
"phase_type",
"phase_center_ra",
"phase_center_dec",
"phase_center_frame",
"phase_center_epoch",
"object_name",
]
if __name in old_phase_attrs:
compatible, reason = self._old_phase_attributes_compatible()
if not compatible:
raise ValueError(
"The older phase attributes, including "
+ ", ".join(old_phase_attrs)
+ " cannot be used with data phased with the new method. This was "
+ "phased with the new method because it has "
+ reason
+ "."
)
warn_str = (
"The older phase attributes, including "
+ ", ".join(old_phase_attrs)
+ ", are deprecated in favor of representing phasing using the "
"phase_center_catalog."
)
phase_dict = list(self.phase_center_catalog.values())[0]
if __name == "phase_type":
warn_str += (
" The phase_type is now represented as the 'cat_type' in the "
"phase_center_catalog (the old 'drift' type corresponds to the "
"new 'unprojected' type, the old 'phased' type corresponds to the "
"new 'sidereal' type, and there is also now support for an 'ephem' "
"type to support moving objects and a new 'driftscan' type which "
"can point at any alt/az (not just zenith) and which always has "
"w-projection applied)."
)
if phase_dict["cat_type"] == "unprojected":
ret_val = "drift"
else:
ret_val = "phased"
elif __name == "phase_center_ra":
warn_str += (
" The phase_center_ra is now represented as the 'cat_lon' in the "
"phase_center_catalog."
)
ret_val = phase_dict["cat_lon"]
elif __name == "phase_center_dec":
warn_str += (
" The phase_center_dec is now represented as the 'cat_lat' in the "
"phase_center_catalog."
)
ret_val = phase_dict["cat_lat"]
elif __name == "phase_center_frame":
warn_str += (
" The phase_center_frame is now represented as the 'cat_frame' in "
"the phase_center_catalog."
)
ret_val = phase_dict["cat_frame"]
elif __name == "phase_center_epoch":
warn_str += (
" The phase_center_epoch is now represented as the 'cat_epoch' in "
"the phase_center_catalog."
)
ret_val = phase_dict["cat_epoch"]
elif __name == "object_name":
warn_str += (
" The object_name is now represented as the 'cat_name' in "
"the phase_center_catalog."
)
ret_val = phase_dict["cat_name"]
warn_str += (
" To see a nice representation of the phase_center_catalog, use the "
"`print_phase_center_info` method. The older attributes will be "
"removed in version 3.0"
)
warnings.warn(warn_str, DeprecationWarning)
return ret_val
return super().__getattribute__(__name)
def _set_future_array_shapes(self):
"""
Set future_array_shapes to True and adjust required parameters.
This method should not be called directly by users; instead it is called
by file-reading methods and `use_future_array_shapes` to indicate the
`future_array_shapes` is True and define expected parameter shapes.
"""
self.future_array_shapes = True
self._freq_array.form = ("Nfreqs",)
self._channel_width.form = ("Nfreqs",)
for param_name in self._data_params:
getattr(self, "_" + param_name).form = ("Nblts", "Nfreqs", "Npols")
def use_future_array_shapes(self):
"""
Change the array shapes of this object to match the planned future shapes.
This method sets allows users to convert to the planned array shapes changes
before the changes go into effect. This method sets the `future_array_shapes`
parameter on this object to True.
"""
self._set_future_array_shapes()
if not self.metadata_only:
# remove the length-1 spw axis for all data-like parameters
for param_name in self._data_params:
setattr(self, param_name, (getattr(self, param_name))[:, 0, :, :])
# remove the length-1 spw axis for the freq_array
self.freq_array = self.freq_array[0, :]
if not self.flex_spw:
# make channel_width be an array of length Nfreqs rather than a single value
# (not needed with flexible spws because this is already done in that case)
self.channel_width = (
np.zeros(self.Nfreqs, dtype=np.float64) + self.channel_width
)
def use_current_array_shapes(self):
"""
Change the array shapes of this object to match the current future shapes.
This method sets allows users to convert back to the current array shapes.
This method sets the `future_array_shapes` parameter on this object to False.
"""
if not self.flex_spw:
unique_channel_widths = np.unique(self.channel_width)
if unique_channel_widths.size > 1:
raise ValueError(
"channel_width parameter contains multiple unique values, but "
"only one spectral window is present. Cannot collapse "
"channel_width to a single value."
)
self._channel_width.form = ()
self.channel_width = unique_channel_widths[0]
self.future_array_shapes = False
for param_name in self._data_params:
getattr(self, "_" + param_name).form = ("Nblts", 1, "Nfreqs", "Npols")
if not self.metadata_only:
for param_name in self._data_params:
setattr(
self, param_name, (getattr(self, param_name))[:, np.newaxis, :, :]
)
self._freq_array.form = (1, "Nfreqs")
self.freq_array = self.freq_array[np.newaxis, :]
def known_telescopes(self):
"""
Get a list of telescopes known to pyuvdata.
This is just a shortcut to uvdata.telescopes.known_telescopes()
Returns
-------
list of str
List of names of known telescopes
"""
return uvtel.known_telescopes()
def set_telescope_params(self, overwrite=False, warn=True):
"""
Set telescope related parameters.
If the telescope_name is in the known_telescopes, set any missing
telescope-associated parameters (e.g. telescope location) to the value
for the known telescope.
Parameters
----------
overwrite : bool
Option to overwrite existing telescope-associated parameters with
the values from the known telescope.
Raises
------
ValueError
if the telescope_name is not in known telescopes
"""
telescope_obj = uvtel.get_telescope(self.telescope_name)
if telescope_obj is not False:
params_set = []
for p in telescope_obj:
telescope_param = getattr(telescope_obj, p)
self_param = getattr(self, p)
if telescope_param.value is not None and (
overwrite is True or self_param.value is None
):
telescope_shape = telescope_param.expected_shape(telescope_obj)
self_shape = self_param.expected_shape(self)
if telescope_shape == self_shape:
params_set.append(self_param.name)
prop_name = self_param.name
setattr(self, prop_name, getattr(telescope_obj, prop_name))
else:
# expected shapes aren't equal. This can happen
# e.g. with diameters,
# which is a single value on the telescope object but is
# an array of length Nants_telescope on the UVData object
# use an assert here because we want an error if this condition
# isn't true, but it's really an internal consistency check.
# This will error if there are changes to the Telescope
# object definition, but nothing that a normal user
# does will cause an error
assert telescope_shape == () and self_shape != "str"
# this parameter is as of this comment most likely a float
# since only diameters and antenna positions will probably
# trigger this else statement
# assign float64 as the type of the array
array_val = (
np.zeros(self_shape, dtype=np.float64)
+ telescope_param.value
)
params_set.append(self_param.name)
prop_name = self_param.name
setattr(self, prop_name, array_val)
if len(params_set) > 0:
params_set_str = ", ".join(params_set)
if warn:
warnings.warn(
"{params} are not set or being overwritten. Using known values "
"for {telescope_name}.".format(
params=params_set_str,
telescope_name=telescope_obj.telescope_name,
)
)
else:
raise ValueError(
f"Telescope {self.telescope_name} is not in known_telescopes."
)
def _calc_single_integration_time(self):
"""
Calculate a single integration time in seconds when not otherwise specified.
This function computes the shortest time difference present in the
time_array, and returns it to be used as the integration time for all
samples.
Returns
-------
int_time : int
integration time in seconds to be assigned to all samples in the data.
"""
# The time_array is in units of days, and integration_time has units of
# seconds, so we need to convert.
return np.diff(np.sort(list(set(self.time_array))))[0] * 86400
def _set_lsts_helper(self):
latitude, longitude, altitude = self.telescope_location_lat_lon_alt_degrees
unique_times, inverse_inds = np.unique(self.time_array, return_inverse=True)
unique_lst_array = uvutils.get_lst_for_time(
unique_times,
latitude,
longitude,
altitude,
frame=self._telescope_location.frame,
)
self.lst_array = unique_lst_array[inverse_inds]
return
def _set_app_coords_helper(self, pa_only=False):
"""
Set values for the apparent coordinate arrays.
This is an internal helper function, which is not designed to be called by
users, but rather individual read/write functions for the UVData object.
Users should use the phase() method for updating/adjusting coordinate values.
Parameters
----------
pa_only : bool, False
Skip the calculation of the apparent RA/Dec, and only calculate the
position angle between `cat_frame` and the apparent coordinate
system. Useful for reading in data formats that do not calculate a PA.
"""
if pa_only:
app_ra = self.phase_center_app_ra
app_dec = self.phase_center_app_dec
else:
app_ra = np.zeros(self.Nblts, dtype=float)
app_dec = np.zeros(self.Nblts, dtype=float)
for cat_id in self.phase_center_catalog.keys():
temp_dict = self.phase_center_catalog[cat_id]
select_mask = self.phase_center_id_array == cat_id
cat_type = temp_dict["cat_type"]
lon_val = temp_dict.get("cat_lon")
lat_val = temp_dict.get("cat_lat")
time_val = temp_dict.get("cat_times")
epoch = temp_dict.get("cat_epoch")
frame = temp_dict.get("cat_frame")
pm_ra = temp_dict.get("cat_pm_ra")
pm_dec = temp_dict.get("cat_pm_dec")
vrad = temp_dict.get("vrad")
dist = temp_dict.get("cat_dist")
app_ra[select_mask], app_dec[select_mask] = uvutils.calc_app_coords(
lon_val,
lat_val,
frame,
coord_epoch=epoch,
coord_times=time_val,
pm_ra=pm_ra,
pm_dec=pm_dec,
vrad=vrad,
dist=dist,
time_array=self.time_array[select_mask],
lst_array=self.lst_array[select_mask],
telescope_loc=self.telescope_location_lat_lon_alt,
coord_type=cat_type,
)
# Now that we have the apparent coordinates sorted out, we can figure
# out what it is we want to do with the position angle
frame_pa = np.zeros(self.Nblts, dtype=float)
for cat_id in self.phase_center_catalog.keys():
temp_dict = self.phase_center_catalog[cat_id]
select_mask = self.phase_center_id_array == cat_id
if not np.any(select_mask):
continue
frame = temp_dict.get("cat_frame")
epoch = temp_dict.get("cat_epoch")
if not frame == "altaz":
frame_pa[select_mask] = uvutils.calc_frame_pos_angle(
self.time_array[select_mask],
app_ra[select_mask],
app_dec[select_mask],
self.telescope_location_lat_lon_alt,
frame,
ref_epoch=epoch,
)
self.phase_center_app_ra = app_ra
self.phase_center_app_dec = app_dec
self.phase_center_frame_pa = frame_pa
def set_lsts_from_time_array(self, background=False):
"""Set the lst_array based from the time_array.
Parameters
----------
background : bool, False
When set to True, start the calculation on a threading.Thread in the
background and return the thread to the user.
Returns
-------
proc : None or threading.Thread instance
When background is set to True, a thread is returned which must be
joined before the lst_array exists on the UVData object.
"""
if not background:
self._set_lsts_helper()
return
else:
proc = threading.Thread(target=self._set_lsts_helper)
proc.start()
return proc
def _check_flex_spw_contiguous(self):
"""
Check if the spectral windows are contiguous for flex_spw datasets.
This checks the flex_spw_id_array to make sure that all channels for each
spectral window are together in one block, versus being interspersed (e.g.,
channel #1 and #3 is in spw #1, channels #2 and #4 are in spw #2). In theory,
UVH5 and UVData objects can handle this, but MIRIAD, MIR, UVFITS, and MS file
formats cannot, so we just consider it forbidden.
"""
if self.flex_spw:
uvutils._check_flex_spw_contiguous(self.spw_array, self.flex_spw_id_array)
def _check_freq_spacing(self, raise_errors=True):
"""
Check if frequencies are evenly spaced and separated by their channel width.
This is a requirement for writing uvfits & miriad files.
Parameters
----------
raise_errors : bool
Option to raise errors if the various checks do not pass.
Returns
-------
spacing_error : bool
Flag that channel spacings or channel widths are not equal.
chanwidth_error : bool
Flag that channel spacing does not match channel width.
"""
return uvutils._check_freq_spacing(
self.freq_array,
self._freq_array.tols,
self.channel_width,
self._channel_width.tols,
self.flex_spw,
self.future_array_shapes,
self.spw_array,
self.flex_spw_id_array,
raise_errors=raise_errors,
)
def remove_flex_pol(self, combine_spws=True):
"""
Convert a flex-pol UVData object into one with a standard polarization axis.
This will convert a flexible-polarization dataset into one with standard
polarization handling, which is required for some operations or writing in
certain filetypes. Note that depending on how it is used, this can inflate
the size of data-like parameters by up to a factor of Nspws (the true value
depends on the number of unique entries in `flex_spw_polarization_array`).
"""
if self.flex_spw_polarization_array is None:
# There isn't anything to do, so just move along
return
unique_pols = np.unique(self.flex_spw_polarization_array)
n_pols = len(unique_pols)
if self.Nspws == 1 or n_pols == 1:
# Just remove the flex_spw_polarization_array and fix the polarization array
self.polarization_array = unique_pols
self.flex_spw_polarization_array = None
return
if combine_spws:
# check to see if there are spectral windows that have matching freq_array
# and channel_width (up to sorting). If so, they need to be combined.
if self.future_array_shapes:
freq_array_use = self.freq_array
else:
freq_array_use = self.freq_array[0, :]
# Now find matching sets of spws
# order spws by order of appearance in flex_spw_id_array
# this tends to get back to the original order in a convert/remove loop
spws_remaining = self.flex_spw_id_array[
np.sort(np.unique(self.flex_spw_id_array, return_index=True)[1])
].tolist()
# key is first spw in a set, value is another dict with keys:
# - "freqs" sorted array of frequencies
# - "widths" channel widths sorted to match frequencies (using argsort)
# - "spws" list of spws in this set
# - "pols" list of pols for the spws in this set
spw_dict = {}
# key is spw, value is first spw that matches (key into spw_dict)
first_spw_dict = {}
while len(spws_remaining) > 0:
this_spw = spws_remaining[0]
this_pol = self.flex_spw_polarization_array[self.spw_array == this_spw][
0
]
spw_inds = np.nonzero(self.flex_spw_id_array == this_spw)[0]
spw_inds = spw_inds[np.argsort(freq_array_use[spw_inds])]
this_freqs = freq_array_use[spw_inds]
this_widths = self.channel_width[spw_inds]
match = False
for spw1, fset1 in spw_dict.items():
if np.array_equal(fset1["freqs"], this_freqs) and np.array_equal(
fset1["widths"], this_widths
):
if this_pol in spw_dict[spw1]["pols"]:
raise ValueError(
"Some spectral windows have identical frequencies, "
"channel widths and polarizations, so spws cannot be "
"combined. Set combine_spws=False to avoid this error."
)
# this spw matches an existing set with no overlapping pols
spw_dict[spw1]["spws"].append(this_spw)
spw_dict[spw1]["pols"].append(this_pol)
first_spw_dict[this_spw] = spw1
spws_remaining.remove(this_spw)
match = True
continue
if not match:
# this spw does not match an existing set
spw_dict[this_spw] = {
"freqs": this_freqs,
"widths": this_widths,
"spws": [this_spw],
"pols": [this_pol],
}
first_spw_dict[this_spw] = this_spw
spws_remaining.remove(this_spw)
n_sets = len(spw_dict)
n_spws_per_set = []
n_freqs = 0
reorder_channels = False
for spw1, spw_set in spw_dict.items():
n_spws_per_set.append(len(spw_set["spws"]))
n_freqs += spw_set["freqs"].size
spw1_mask = self.flex_spw_id_array == spw1
for spw2 in spw_set["spws"][1:]:
spw2_mask = self.flex_spw_id_array == spw2
if not (
np.array_equal(
freq_array_use[spw1_mask], freq_array_use[spw2_mask]
)
and np.array_equal(
self.channel_width[spw1_mask], self.channel_width[spw2_mask]
)
):
reorder_channels = True
n_spws_per_set = np.array(n_spws_per_set)
if not np.all(n_spws_per_set == n_pols):
# If all the pols are not present in all sets, we cannot combine spws
warnings.warn(
"combine_spws is True but there are not matched spws for all "
"polarizations, so spws will not be combined."
)
combine_spws = False
if combine_spws:
# figure out whether reordering is required to use an inplace reshape
# Criteria:
# 1) polarization is the slowest changing axis
# 2) freqs and spw sets are in the same order for each pol
reorder_spws = False
spw_inds = np.zeros_like(self.flex_spw_id_array)
first_spw_array = np.zeros_like(self.flex_spw_id_array)
for spw_ind, spw in enumerate(self.spw_array):
these_freq_inds = np.nonzero(self.flex_spw_id_array == spw)[0]
spw_inds[these_freq_inds] = spw_ind
first_spw_array[these_freq_inds] = first_spw_dict[spw]
pol_array_check = self.flex_spw_polarization_array[spw_inds]
if not np.nonzero(np.diff(pol_array_check))[0].size == n_pols - 1:
reorder_spws = True
pol0_spw_order = first_spw_array[pol_array_check == unique_pols[0]]
for pol in unique_pols[1:]:
this_spw_order = first_spw_array[pol_array_check == pol]
if not np.array_equal(this_spw_order, pol0_spw_order):
reorder_spws = True
if reorder_channels or reorder_spws:
if reorder_channels:
channel_order = "freq"
else:
channel_order = None
if reorder_spws:
# note: spw_order is an index array into spw_array
spw_order = np.zeros(self.Nspws, dtype=int)
if np.all(unique_pols < 0):
# use more standard ordering for polarizations
unique_pols = unique_pols[np.argsort(np.abs(unique_pols))]
for pol_ind, pol in enumerate(unique_pols):
for spw_ind, (_, spw_set) in enumerate(spw_dict.items()):
this_ind = pol_ind * n_sets + spw_ind
this_spw = np.array(spw_set["spws"])[spw_set["pols"] == pol]
spw_order[this_ind] = np.nonzero(
self.spw_array == this_spw
)[0]
else:
spw_order = None
self.reorder_freqs(channel_order=channel_order, spw_order=spw_order)
# recalculate arrays used below
if self.future_array_shapes:
freq_array_use = self.freq_array
else:
freq_array_use = self.freq_array[0, :]
first_spw_array = np.zeros_like(self.flex_spw_id_array)
for spw_ind, spw in enumerate(self.spw_array):
these_freq_inds = np.nonzero(self.flex_spw_id_array == spw)[0]
spw_inds[these_freq_inds] = spw_ind
first_spw_array[these_freq_inds] = first_spw_dict[spw]
pol_array_check = self.flex_spw_polarization_array[spw_inds]
self.Npols = n_pols
self.Nspws = n_sets
self.Nfreqs = n_freqs
# now things are in the correct order to do a simple reshape
if self.future_array_shapes:
self.freq_array = freq_array_use[: self.Nfreqs]
else:
self.freq_array = freq_array_use[np.newaxis, : self.Nfreqs]
self.channel_width = self.channel_width[: self.Nfreqs]
self.polarization_array = pol_array_check[
np.sort(np.unique(pol_array_check, return_index=True)[1])
]
self.flex_spw_polarization_array = None
self.spw_array = first_spw_array[
np.sort(np.unique(first_spw_array, return_index=True)[1])
]
self.flex_spw_id_array = first_spw_array[: self.Nfreqs]
if not self.metadata_only:
# we use the order="F" parameter here to undo the reshape done in
# `convert_to_flexp_pol` (which uses it to ensure that polarization is
# the slowest changing axis)
if self.future_array_shapes:
self.data_array = self.data_array.reshape(
self.Nblts, self.Nfreqs, self.Npols, order="F"
)
self.flag_array = self.flag_array.reshape(
self.Nblts, self.Nfreqs, self.Npols, order="F"
)
self.nsample_array = self.nsample_array.reshape(
self.Nblts, self.Nfreqs, self.Npols, order="F"
)
else:
self.data_array = self.data_array.reshape(
self.Nblts, 1, self.Nfreqs, self.Npols, order="F"
)
self.flag_array = self.flag_array.reshape(
self.Nblts, 1, self.Nfreqs, self.Npols, order="F"
)
self.nsample_array = self.nsample_array.reshape(
self.Nblts, 1, self.Nfreqs, self.Npols, order="F"
)
return
self.Npols = n_pols
# If we have metadata only, or there was only one pol we were working with,
# then we do not need to do anything further aside from removing the array
# associated with flex_spw_polarization_array
if self.metadata_only:
self.polarization_array = unique_pols
self.flex_spw_polarization_array = None
return
# Otherwise, move through all of the data params
self.polarization_array = unique_pols
for name, param in zip(self._data_params, self.data_like_parameters):
# We need to construct arrays with the appropriate shape
new_shape = [self.Nblts, 1, self.Nfreqs, self.Npols]
# Pop the spw-axis if using future array shapes
if self.future_array_shapes:
del new_shape[1]
# Use full here, since we want zeros if we are working with nsample_array
# or data_array, otherwise we want True if working w/ flag_array.
new_param = np.full(new_shape, name == "flag_array", dtype=param.dtype)
# Now we have to iterate through each spectral window
for spw, pol in zip(self.spw_array, self.flex_spw_polarization_array):
pol_idx = np.intersect1d(
pol, self.polarization_array, return_indices=True
)[2][0]
spw_screen = self.flex_spw_id_array == spw
# Note that this works because pol_idx is an integer, ergo a simple
# slice (wherease spw_screen is a complex slice, which we can only have
# one of for an array).
if self.future_array_shapes:
new_param[:, spw_screen, pol_idx] = param[:, spw_screen, 0]
else:
new_param[:, :, spw_screen, pol_idx] = param[:, :, spw_screen, 0]
# With the new array defined and filled, set the attribute equal to it
setattr(self, name, new_param)
# Finally, remove the flex-pol attribute
self.flex_spw_polarization_array = None
def _make_flex_pol(self, raise_error=False, raise_warning=True):
"""
Convert a regular UVData object into one with flex-polarization enabled.
This is an internal helper function, which is not designed to be called by
users, but rather individual read/write functions for the UVData object.
This will convert a regular UVData object into one that uses flexible
polarization, which allows for each spectral window to have its own unique
polarization code, useful for storing data more compactly when different
windows have different polarizations recorded. Note that at this time,
only one polarization code per-spw is allowed -- if more than one polarization
is found to have unflagged data in a given spectral window, then the object
will not be converted.
Parameters
----------
raise_error : bool
If an object cannot be converted to flex-pol, then raise a ValueError.
Default is False.
raise_warning : bool
If an object cannot be converted to flex-pol, and `raise_error=False`, then
raise a warning that the conversion failed. Default is True.
"""
if not self.flex_spw:
msg = "Cannot make a flex-pol UVData object if flex_spw=False."
if raise_error:
raise ValueError(msg)
if raise_warning:
warnings.warn(msg)
return
if self.metadata_only:
msg = (
"Cannot make a metadata_only UVData object flex-pol because flagging "
"info is required. Consider using `convert_to_flex_pol` instead, but "
"be aware that the behavior is somewhat different"
)
if raise_error:
raise ValueError(msg)
if raise_warning:
warnings.warn(msg)
return
if self.Npols == 1:
# This is basically a no-op, fix the to pol-array attributes and exit
if self.flex_spw_polarization_array is None:
self.flex_spw_polarization_array = (
np.zeros_like(self.spw_array) + self.polarization_array[0]
)
self.polarization_array = np.array([0])
return
flex_pol_idx = np.zeros_like(self.spw_array)
for idx, spw in enumerate(self.spw_array):
spw_screen = self.flex_spw_id_array == spw
# For each window, we want to check that there is only one polarization with
# any unflagged data, which we can do by seeing if not all of the flags
# are set across the non-polarization axes (hence the ~np.all()).
if self.future_array_shapes:
pol_check = ~np.all(self.flag_array[:, spw_screen], axis=(0, 1))
else:
pol_check = ~np.all(self.flag_array[:, 0, spw_screen], axis=(0, 1))
if sum(pol_check) > 1:
msg = (
"Cannot make a flex-pol UVData object, as some windows have "
"unflagged data in mutiple polarizations."
)
if raise_error:
raise ValueError(msg)
if raise_warning:
warnings.warn(msg)
return
elif not np.any(pol_check):
flex_pol_idx[idx] = -1
else:
flex_pol_idx[idx] = np.where(pol_check)[0][0]
# If one window was all flagged out, but the others all belong to the same pol,
# assume we just want that polarization.
if len(np.unique(flex_pol_idx[flex_pol_idx >= 0])) == 1:
flex_pol_idx[:] = np.unique(flex_pol_idx[flex_pol_idx >= 0])
# Now that we have polarizations sorted out, update metadata attributes
self.flex_spw_polarization_array = self.polarization_array[flex_pol_idx]
self.polarization_array = np.array([0])
self.Npols = 1
# Finally, prep for working w/ data-like attibutes. Start by determining the
# right shape for the new values, modulo future_array_shapes
new_shape = [self.Nblts, 1, self.Nfreqs, 1]
# Pop the spw-axis if using future array shapes
if self.future_array_shapes:
del new_shape[1]
# Now go through one-by-one with data-like parameters and update
for name, param in zip(self._data_params, self.data_like_parameters):
# We can use empty here, since we know that we will be filling all
# values of his array (and empty allows us to forgo the extra overhead
# of setting all the elements to a particular value).
new_param = np.empty(new_shape, dtype=param.dtype)
# Now we have to iterate through each spectral window
for spw, pol_idx in zip(self.spw_array, flex_pol_idx):
spw_screen = self.flex_spw_id_array == spw
# Note that this works because pol_idx is an integer, ergo a simple
# slice (wherease spw_screen is a complex slice, which we can only have
# one of for an array).
if self.future_array_shapes:
new_param[:, spw_screen, 0] = param[:, spw_screen, pol_idx]
else:
new_param[:, :, spw_screen, 0] = param[:, :, spw_screen, pol_idx]
# With the new array defined and filled, set the attribute equal to it
setattr(self, name, new_param)
def convert_to_flex_pol(self):
"""
Convert a regular UVData object into a flex-polarization object.
This effectively combines the frequency and polarization axis with polarization
changing slowest. Saving data to uvh5 files this way can speed up some kinds
of data access.
"""
if self.flex_spw_polarization_array is not None:
raise ValueError("This is already a flex-pol object")
if not self.flex_spw:
self._set_flex_spw()
if not self.future_array_shapes:
self.channel_width = np.full(self.Nfreqs, self.channel_width)
self.flex_spw_id_array = np.zeros(self.Nfreqs, dtype=int)
new_spw_array = self.spw_array
new_flex_pol_array = np.full(self.Nspws, self.polarization_array[0])
new_spw_id_array = np.zeros((self.Nfreqs, self.Npols), dtype=int)
for pol_ind, pol in enumerate(self.polarization_array):
if pol_ind == 0:
new_spw_id_array[:, 0] = self.flex_spw_id_array
else:
for spw in self.spw_array:
new_spw = (
set(range(self.Nspws * self.Npols + 1))
.difference(new_spw_array)
.pop()
)
new_spw_array = np.concatenate((new_spw_array, np.array([new_spw])))
new_flex_pol_array = np.concatenate(
(new_flex_pol_array, np.array([pol]))
)
spw_inds = np.nonzero(self.flex_spw_id_array == spw)[0]
new_spw_id_array[spw_inds, pol_ind] = new_spw
spw_sort = np.argsort(new_spw_array)
self.spw_array = new_spw_array[spw_sort]
self.flex_spw_polarization_array = new_flex_pol_array[spw_sort]
# we use the order="F" parameter here to ensure that polarization is the slowest
# changing axis
self.flex_spw_id_array = new_spw_id_array.reshape(
self.Nfreqs * self.Npols, order="F"
)
self.Nspws = self.spw_array.size
self.polarization_array = np.array([0])
if self.future_array_shapes:
freq_array_use = self.freq_array
else:
freq_array_use = self.freq_array[0, :]
self.freq_array = np.tile(freq_array_use, self.Npols)
if not self.future_array_shapes:
self.freq_array = self.freq_array[np.newaxis, :]
self.channel_width = np.tile(self.channel_width, self.Npols)
if not self.metadata_only:
# we use the order="F" parameter here to ensure that polarization is the
# slowest changing axis
if self.future_array_shapes:
self.data_array = self.data_array.reshape(
self.Nblts, self.Nfreqs * self.Npols, 1, order="F"
)
self.flag_array = self.flag_array.reshape(
self.Nblts, self.Nfreqs * self.Npols, 1, order="F"
)
self.nsample_array = self.nsample_array.reshape(
self.Nblts, self.Nfreqs * self.Npols, 1, order="F"
)
else:
self.data_array = self.data_array.reshape(
self.Nblts, 1, self.Nfreqs * self.Npols, 1, order="F"
)
self.flag_array = self.flag_array.reshape(
self.Nblts, 1, self.Nfreqs * self.Npols, 1, order="F"
)
self.nsample_array = self.nsample_array.reshape(
self.Nblts, 1, self.Nfreqs * self.Npols, 1, order="F"
)
self.Nfreqs = self.Nfreqs * self.Npols
self.Npols = 1
return
def _calc_nants_data(self):
"""Calculate the number of antennas from ant_1_array and ant_2_array arrays."""
return int(np.union1d(self.ant_1_array, self.ant_2_array).size)
def _fix_autos(self):
"""Remove imaginary component of auto-correlations."""
if self.polarization_array is None or (
self.ant_1_array is None or self.ant_2_array is None
):
warnings.warn(
"Cannot use _fix_autos if ant_1_array, ant_2_array, or "
"polarization_array are None. Leaving data_array untouched."
)
return
# Select out the autos
auto_screen = self.ant_1_array == self.ant_2_array
# Only these pols have "true" auto-correlations, that we'd expect
# to be real only. Select on only them
auto_pol_list = ["xx", "yy", "rr", "ll", "pI", "pQ", "pU", "pV"]
pol_screen = np.array(
[
uvutils.POL_NUM2STR_DICT[pol] in auto_pol_list
for pol in self.polarization_array
]
)
# Make sure we actually have work to do here, otherwise skip all of this
if (np.any(pol_screen) and np.any(auto_screen)) and not (
pol_screen is None or auto_screen is None
):
# Select out the relevant data. Need to do this because we have two
# complex slices we need to do
auto_data = self.data_array[auto_screen]
# Set the autos to be real-only by taking the absolute value
if self.future_array_shapes:
auto_data[:, :, pol_screen] = np.abs(auto_data[:, :, pol_screen])
else:
auto_data[:, :, :, pol_screen] = np.abs(auto_data[:, :, :, pol_screen])
# Finally, plug the modified values back into data_array
self.data_array[auto_screen] = auto_data
def check(
self,
check_extra=True,
run_check_acceptability=True,
check_freq_spacing=False,
strict_uvw_antpos_check=False,
allow_flip_conj=False,
check_autos=False,
fix_autos=False,
):
"""
Add some extra checks on top of checks on UVBase class.
Check that required parameters exist. Check that parameters have
appropriate shapes and optionally that the values are acceptable.
Parameters
----------
check_extra : bool
If true, check all parameters, otherwise only check required parameters.
run_check_acceptability : bool
Option to check if values in parameters are acceptable.
check_freq_spacing : bool
Option to check if frequencies are evenly spaced and the spacing is
equal to their channel_width. This is not required for UVData
objects in general but is required to write to uvfits and miriad files.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
allow_flip_conj : bool
If set to True, and the UVW coordinates do not match antenna positions,
check and see if flipping the conjugation of the baselines (i.e, multiplying
the UVWs by -1) resolves the apparent discrepancy -- and if it does, fix
the apparent conjugation error in `uvw_array` and `data_array`. Default is
False.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is False.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is False.
Returns
-------
bool
True if check passes
Raises
------
ValueError
if parameter shapes or types are wrong or do not have acceptable
values (if run_check_acceptability is True)
"""
# first run the basic check from UVBase
logger.debug("Doing UVBase check...")
super(UVData, self).check(
check_extra=check_extra, run_check_acceptability=run_check_acceptability
)
logger.debug("... Done UVBase Check")
# Check internal consistency of numbers which don't explicitly correspond
# to the shape of another array.
if self.Nants_data != self._calc_nants_data():
raise ValueError(
"Nants_data must be equal to the number of unique "
"values in ant_1_array and ant_2_array"
)
if self.Nbls != len(np.unique(self.baseline_array)):
raise ValueError(
"Nbls must be equal to the number of unique "
"baselines in the data_array"
)
if self.Ntimes != len(np.unique(self.time_array)):
raise ValueError(
"Ntimes must be equal to the number of unique "
"times in the time_array"
)
if self.flex_spw:
# Check that all values in flex_spw_id_array are entries in the spw_array
if not np.all(np.isin(self.flex_spw_id_array, self.spw_array)):
raise ValueError(
"All values in the flex_spw_id_array must exist in the spw_array."
)
for val in np.unique(self.phase_center_id_array):
if val not in self.phase_center_catalog.keys():
raise ValueError(
f"Phase center id {val} is does not have an entry in "
"`phase_center_catalog`, which has keys "
f"{self.phase_center_catalog.keys()}. All values in "
"`phase_center_id_array` must be keys in `phase_center_catalog`. "
)
if self.flex_spw_polarization_array is not None:
# Check that usage of flex_spw_polarization_array follows the rule that
# each window only has a single polarization per spectral window.
if self.Npols != 1:
raise ValueError(
"Npols must be equal to 1 if flex_spw_polarization_array is set."
)
if np.any(self.polarization_array != 0):
raise ValueError(
"polarization_array must all be equal to 0 if "
"flex_spw_polarization_array is set."
)
elif np.any(self.polarization_array == 0):
# If flex_spw_polarization_array is not set, then make sure that
# no entries in polarization_array are equal to zero.
raise ValueError(
"polarization_array may not be equal to 0 if "
"flex_spw_polarization_array is not set."
)
# require that all entries in ant_1_array and ant_2_array exist in
# antenna_numbers
logger.debug("Doing Antenna Uniqueness Check...")
if not set(np.unique(self.ant_1_array)).issubset(self.antenna_numbers):
raise ValueError("All antennas in ant_1_array must be in antenna_numbers.")
if not set(np.unique(self.ant_2_array)).issubset(self.antenna_numbers):
raise ValueError("All antennas in ant_2_array must be in antenna_numbers.")
logger.debug("... Done Antenna Uniqueness Check")
# issue warning if extra_keywords keys are longer than 8 characters
for key in self.extra_keywords.keys():
if len(key) > 8:
warnings.warn(
f"key {key} in extra_keywords is longer than 8 "
"characters. It will be truncated to 8 if written "
"to uvfits or miriad file formats."
)
# issue warning if extra_keywords values are lists, arrays or dicts
for key, value in self.extra_keywords.items():
if isinstance(value, (list, dict, np.ndarray)):
warnings.warn(
f"{key} in extra_keywords is a list, array or dict, "
"which will raise an error when writing uvfits or "
"miriad file types"
)
if run_check_acceptability:
# check that the uvws make sense given the antenna positions
# make a metadata only copy of this object to properly calculate uvws
temp_obj = self.copy(metadata_only=True)
logger.info("Temp Obj")
with warnings.catch_warnings():
warnings.simplefilter("ignore")
logger.debug("Setting UVWs from antenna positions...")
temp_obj.set_uvws_from_antenna_positions()
logger.debug("... Done Setting UVWs")
if not np.allclose(temp_obj.uvw_array, self.uvw_array, atol=1):
max_diff = np.max(np.abs(temp_obj.uvw_array - self.uvw_array))
if allow_flip_conj and np.allclose(
-temp_obj.uvw_array, self.uvw_array, atol=1
):
warnings.warn(
"UVW orientation appears to be flipped, attempting to "
"fix by changing conjugation of baselines."
)
self.uvw_array *= -1
self.data_array = np.conj(self.data_array)
logger.info("Flipped Array")
elif not strict_uvw_antpos_check:
warnings.warn(
"The uvw_array does not match the expected values given "
"the antenna positions. The largest discrepancy is "
f"{max_diff} meters. This is a fairly common situation "
"but might indicate an error in the antenna positions, "
"the uvws or the phasing."
)
else:
raise ValueError(
"The uvw_array does not match the expected values given "
"the antenna positions. The largest discrepancy is "
f"{max_diff} meters."
)
# check auto and cross-corrs have sensible uvws
logger.debug("Checking autos...")
autos = np.isclose(self.ant_1_array - self.ant_2_array, 0.0)
if not np.all(
np.isclose(
self.uvw_array[autos],
0.0,
rtol=self._uvw_array.tols[0],
atol=self._uvw_array.tols[1],
)
):
raise ValueError(
"Some auto-correlations have non-zero uvw_array coordinates."
)
logger.debug("... Done Checking Autos")
if (self.data_array is not None and np.any(autos)) and check_autos:
# Verify here that the autos do not have any imaginary components
# Only these pols have "true" auto-correlations, that we'd expect
# to be real only. Select on only them
auto_pol_list = ["xx", "yy", "rr", "ll", "pI", "pQ", "pU", "pV"]
if self.flex_spw_polarization_array is not None:
pol_screen = np.array(
[
uvutils.POL_NUM2STR_DICT[pol] in auto_pol_list
for pol in self.flex_spw_polarization_array
]
)
# There should be a better way...
spw_inds = np.zeros_like(self.flex_spw_id_array)
for spw_ind, spw in enumerate(self.spw_array):
these_freq_inds = np.nonzero(self.flex_spw_id_array == spw)[0]
spw_inds[these_freq_inds] = spw_ind
freq_screen = pol_screen[spw_inds]
else:
pol_screen = np.array(
[
uvutils.POL_NUM2STR_DICT[pol] in auto_pol_list
for pol in self.polarization_array
]
)
# Check autos if they have imag component -- doing iscomplex first and
# then pol select was faster in every case checked in test files.
if not np.any(pol_screen):
# There's no relevant pols to check, just skip the rest of this
auto_imag = False
else:
auto_imag = np.iscomplex(self.data_array[autos])
if np.all(pol_screen):
auto_imag = np.any(auto_imag)
elif self.future_array_shapes:
if self.flex_spw_polarization_array is not None:
auto_imag = np.any(auto_imag[:, freq_screen])
else:
auto_imag = np.any(auto_imag[:, :, pol_screen])
else:
if self.flex_spw_polarization_array is not None:
auto_imag = np.any(auto_imag[:, :, freq_screen])
else:
auto_imag = np.any(auto_imag[:, :, :, pol_screen])
if auto_imag:
if np.all(pol_screen):
temp_data = self.data_array[autos]
else:
auto_data = self.data_array[autos]
if self.future_array_shapes:
if self.flex_spw_polarization_array is not None:
temp_data = auto_data[:, freq_screen]
else:
temp_data = auto_data[:, :, pol_screen]
else:
if self.flex_spw_polarization_array is not None:
temp_data = auto_data[:, :, freq_screen]
else:
temp_data = auto_data[:, :, :, pol_screen]
temp_data = temp_data[temp_data.imag != 0]
max_imag = np.max(np.abs(temp_data.imag))
max_imag_ratio = np.max(np.abs(temp_data.imag / temp_data.real))
if fix_autos:
warnings.warn(
"Fixing auto-correlations to be be real-only, after some "
"imaginary values were detected in data_array. "
f"Largest imaginary component was {max_imag}, largest "
f"imaginary/real ratio was {max_imag_ratio}."
)
self._fix_autos()
else:
raise ValueError(
"Some auto-correlations have non-real values in data_array."
f" Largest imaginary component was {max_imag}, largest "
f"imaginary/real ratio was {max_imag_ratio}."
" You can attempt to fix this by setting fix_autos=True."
)
if np.any(
np.isclose(
# this line used to use np.linalg.norm but it turns out
# squaring and sqrt is slightly more efficient unless the array
# is "very large".
np.sqrt(
self.uvw_array[~autos, 0] ** 2
+ self.uvw_array[~autos, 1] ** 2
+ self.uvw_array[~autos, 2] ** 2
),
0.0,
rtol=self._uvw_array.tols[0],
atol=self._uvw_array.tols[1],
)
):
raise ValueError(
"Some cross-correlations have near-zero uvw_array magnitudes."
)
if check_freq_spacing:
self._check_freq_spacing()
return True
def copy(self, metadata_only=False):
"""
Make and return a copy of the UVData object.
Parameters
----------
metadata_only : bool
If True, only copy the metadata of the object.
Returns
-------
UVData
Copy of self.
"""
if not metadata_only:
return super(UVData, self).copy()
else:
uv = UVData()
# include all attributes, not just UVParameter ones.
for attr in self.__iter__(uvparams_only=False):
# skip properties
if isinstance(getattr(type(self), attr, None), property):
continue
# skip data like parameters
# parameter names have a leading underscore we want to ignore
if attr.lstrip("_") in self._data_params:
continue
setattr(uv, attr, copy.deepcopy(getattr(self, attr)))
if uv.future_array_shapes:
for param_name in uv._data_params:
getattr(uv, "_" + param_name).form = ("Nblts", "Nfreqs", "Npols")
return uv
def baseline_to_antnums(self, baseline):
"""
Get the antenna numbers corresponding to a given baseline number.
Parameters
----------
baseline : int or array_like of int
baseline number
Returns
-------
int or array_like of int
first antenna number(s)
int or array_like of int
second antenna number(s)
"""
return uvutils.baseline_to_antnums(baseline, self.Nants_telescope)
def antnums_to_baseline(self, ant1, ant2, attempt256=False):
"""
Get the baseline number corresponding to two given antenna numbers.
Parameters
----------
ant1 : int or array_like of int
first antenna number
ant2 : int or array_like of int
second antenna number
attempt256 : bool
Option to try to use the older 256 standard used in many uvfits files
(will use 2048 standard if there are more than 256 antennas).
Returns
-------
int or array of int
baseline number corresponding to the two antenna numbers.
"""
return uvutils.antnums_to_baseline(
ant1, ant2, self.Nants_telescope, attempt256=attempt256
)
def antpair2ind(self, ant1, ant2=None, ordered=True):
"""
Get indices along the baseline-time axis for a given antenna pair.
This will search for either the key as specified, or the key and its
conjugate.
Parameters
----------
ant1, ant2 : int
Either an antenna-pair key, or key expanded as arguments,
e.g. antpair2ind( (10, 20) ) or antpair2ind(10, 20)
ordered : bool
If True, search for antpair as provided, else search for it and
its conjugate.
Returns
-------
inds : ndarray of int-64
indices of the antpair along the baseline-time axis.
"""
# check for expanded antpair or key
if ant2 is None:
if not isinstance(ant1, tuple):
raise ValueError(
"antpair2ind must be fed an antpair tuple "
"or expand it as arguments"
)
ant2 = ant1[1]
ant1 = ant1[0]
else:
if not isinstance(ant1, (int, np.integer)):
raise ValueError(
"antpair2ind must be fed an antpair tuple or "
"expand it as arguments"
)
if not isinstance(ordered, (bool, np.bool_)):
raise ValueError("ordered must be a boolean")
# if getting auto-corr, ordered must be True
if ant1 == ant2:
ordered = True
# get indices
inds = np.where((self.ant_1_array == ant1) & (self.ant_2_array == ant2))[0]
if ordered:
return inds
else:
ind2 = np.where((self.ant_1_array == ant2) & (self.ant_2_array == ant1))[0]
inds = np.asarray(np.append(inds, ind2), dtype=np.int64)
return inds
def _key2inds(self, key):
"""
Interpret user specified key as antenna pair and/or polarization.
Parameters
----------
key : tuple of int
Identifier of data. Key can be length 1, 2, or 3:
if len(key) == 1:
if (key < 5) or (type(key) is str): interpreted as a
polarization number/name, return all blts for that pol.
else: interpreted as a baseline number. Return all times and
polarizations for that baseline.
if len(key) == 2: interpreted as an antenna pair. Return all
times and pols for that baseline.
if len(key) == 3: interpreted as antenna pair and pol (ant1, ant2, pol).
Return all times for that baseline, pol. pol may be a string.
Returns
-------
blt_ind1 : ndarray of int
blt indices for antenna pair.
blt_ind2 : ndarray of int
blt indices for conjugate antenna pair.
Note if a cross-pol baseline is requested, the polarization will
also be reversed so the appropriate correlations are returned.
e.g. asking for (1, 2, 'xy') may return conj(2, 1, 'yx'), which
is equivalent to the requesting baseline. See utils.conj_pol() for
complete conjugation mapping.
pol_ind : tuple of ndarray of int
polarization indices for blt_ind1 and blt_ind2
"""
key = uvutils._get_iterable(key)
if type(key) is str:
# Single string given, assume it is polarization
pol_ind1 = np.where(
self.polarization_array
== uvutils.polstr2num(key, x_orientation=self.x_orientation)
)[0]
if len(pol_ind1) > 0:
blt_ind1 = np.arange(self.Nblts, dtype=np.int64)
blt_ind2 = np.array([], dtype=np.int64)
pol_ind2 = np.array([], dtype=np.int64)
pol_ind = (pol_ind1, pol_ind2)
else:
raise KeyError("Polarization {pol} not found in data.".format(pol=key))
elif len(key) == 1:
key = key[0] # For simplicity
if isinstance(key, Iterable):
# Nested tuple. Call function again.
blt_ind1, blt_ind2, pol_ind = self._key2inds(key)
elif key < 5:
# Small number, assume it is a polarization number a la AIPS memo
pol_ind1 = np.where(self.polarization_array == key)[0]
if len(pol_ind1) > 0:
blt_ind1 = np.arange(self.Nblts)
blt_ind2 = np.array([], dtype=np.int64)
pol_ind2 = np.array([], dtype=np.int64)
pol_ind = (pol_ind1, pol_ind2)
else:
raise KeyError(
"Polarization {pol} not found in data.".format(pol=key)
)
else:
# Larger number, assume it is a baseline number
inv_bl = self.antnums_to_baseline(
self.baseline_to_antnums(key)[1], self.baseline_to_antnums(key)[0]
)
blt_ind1 = np.where(self.baseline_array == key)[0]
blt_ind2 = np.where(self.baseline_array == inv_bl)[0]
if len(blt_ind1) + len(blt_ind2) == 0:
raise KeyError("Baseline {bl} not found in data.".format(bl=key))
if len(blt_ind1) > 0:
pol_ind1 = np.arange(self.Npols)
else:
pol_ind1 = np.array([], dtype=np.int64)
if len(blt_ind2) > 0:
try:
pol_ind2 = uvutils.reorder_conj_pols(self.polarization_array)
except ValueError as err:
if len(blt_ind1) == 0:
raise KeyError(
f"Baseline {key} not found for polarization "
"array in data."
) from err
else:
pol_ind2 = np.array([], dtype=np.int64)
blt_ind2 = np.array([], dtype=np.int64)
else:
pol_ind2 = np.array([], dtype=np.int64)
pol_ind = (pol_ind1, pol_ind2)
elif len(key) == 2:
# Key is an antenna pair
blt_ind1 = self.antpair2ind(key[0], key[1])
blt_ind2 = self.antpair2ind(key[1], key[0])
if len(blt_ind1) + len(blt_ind2) == 0:
raise KeyError("Antenna pair {pair} not found in data".format(pair=key))
if len(blt_ind1) > 0:
pol_ind1 = np.arange(self.Npols)
else:
pol_ind1 = np.array([], dtype=np.int64)
if len(blt_ind2) > 0:
try:
pol_ind2 = uvutils.reorder_conj_pols(self.polarization_array)
except ValueError as err:
if len(blt_ind1) == 0:
raise KeyError(
f"Baseline {key} not found for polarization array in data."
) from err
else:
pol_ind2 = np.array([], dtype=np.int64)
blt_ind2 = np.array([], dtype=np.int64)
else:
pol_ind2 = np.array([], dtype=np.int64)
pol_ind = (pol_ind1, pol_ind2)
elif len(key) == 3:
# Key is an antenna pair + pol
blt_ind1 = self.antpair2ind(key[0], key[1])
blt_ind2 = self.antpair2ind(key[1], key[0])
if len(blt_ind1) + len(blt_ind2) == 0:
raise KeyError(
"Antenna pair {pair} not found in "
"data".format(pair=(key[0], key[1]))
)
if type(key[2]) is str:
# pol is str
if len(blt_ind1) > 0:
pol_ind1 = np.where(
self.polarization_array
== uvutils.polstr2num(key[2], x_orientation=self.x_orientation)
)[0]
else:
pol_ind1 = np.array([], dtype=np.int64)
if len(blt_ind2) > 0:
pol_ind2 = np.where(
self.polarization_array
== uvutils.polstr2num(
uvutils.conj_pol(key[2]), x_orientation=self.x_orientation
)
)[0]
else:
pol_ind2 = np.array([], dtype=np.int64)
else:
# polarization number a la AIPS memo
if len(blt_ind1) > 0:
pol_ind1 = np.where(self.polarization_array == key[2])[0]
else:
pol_ind1 = np.array([], dtype=np.int64)
if len(blt_ind2) > 0:
pol_ind2 = np.where(
self.polarization_array == uvutils.conj_pol(key[2])
)[0]
else:
pol_ind2 = np.array([], dtype=np.int64)
pol_ind = (pol_ind1, pol_ind2)
if len(blt_ind1) * len(pol_ind[0]) + len(blt_ind2) * len(pol_ind[1]) == 0:
raise KeyError(
"Polarization {pol} not found in data.".format(pol=key[2])
)
# Catch autos
if np.array_equal(blt_ind1, blt_ind2):
blt_ind2 = np.array([], dtype=np.int64)
return (blt_ind1, blt_ind2, pol_ind)
def _smart_slicing(
self, data, ind1, ind2, indp, squeeze="default", force_copy=False
):
"""
Quickly get the relevant section of a data-like array.
Used in get_data, get_flags and get_nsamples.
Parameters
----------
data : ndarray
4-dimensional array shaped like self.data_array
ind1 : array_like of int
blt indices for antenna pair (e.g. from self._key2inds)
ind2 : array_like of int
blt indices for conjugate antenna pair. (e.g. from self._key2inds)
indp : tuple array_like of int
polarization indices for ind1 and ind2 (e.g. from self._key2inds)
squeeze : str
string specifying how to squeeze the returned array. Options are:
'default': squeeze pol and spw dimensions if possible;
'none': no squeezing of resulting numpy array;
'full': squeeze all length 1 dimensions.
force_copy : bool
Option to explicitly make a copy of the data.
Returns
-------
ndarray
copy (or if possible, a read-only view) of relevant section of data
"""
p_reg_spaced = [False, False]
p_start = [0, 0]
p_stop = [0, 0]
dp = [1, 1]
for i, pi in enumerate(indp):
if len(pi) == 0:
continue
if len(set(np.ediff1d(pi))) <= 1:
p_reg_spaced[i] = True
p_start[i] = pi[0]
p_stop[i] = pi[-1] + 1
if len(pi) != 1:
dp[i] = pi[1] - pi[0]
if len(ind2) == 0:
# only unconjugated baselines
if len(set(np.ediff1d(ind1))) <= 1:
blt_start = ind1[0]
blt_stop = ind1[-1] + 1
if len(ind1) == 1:
dblt = 1
else:
dblt = ind1[1] - ind1[0]
if p_reg_spaced[0]:
if self.future_array_shapes:
out = data[
blt_start:blt_stop:dblt, :, p_start[0] : p_stop[0] : dp[0]
]
else:
out = data[
blt_start:blt_stop:dblt,
:,
:,
p_start[0] : p_stop[0] : dp[0],
]
else:
if self.future_array_shapes:
out = data[blt_start:blt_stop:dblt, :, indp[0]]
else:
out = data[blt_start:blt_stop:dblt, :, :, indp[0]]
else:
out = data[ind1]
if p_reg_spaced[0]:
if self.future_array_shapes:
out = out[:, :, p_start[0] : p_stop[0] : dp[0]]
else:
out = out[:, :, :, p_start[0] : p_stop[0] : dp[0]]
else:
if self.future_array_shapes:
out = out[:, :, indp[0]]
else:
out = out[:, :, :, indp[0]]
elif len(ind1) == 0:
# only conjugated baselines
if len(set(np.ediff1d(ind2))) <= 1:
blt_start = ind2[0]
blt_stop = ind2[-1] + 1
if len(ind2) == 1:
dblt = 1
else:
dblt = ind2[1] - ind2[0]
if p_reg_spaced[1]:
if self.future_array_shapes:
out = np.conj(
data[
blt_start:blt_stop:dblt,
:,
p_start[1] : p_stop[1] : dp[1],
]
)
else:
out = np.conj(
data[
blt_start:blt_stop:dblt,
:,
:,
p_start[1] : p_stop[1] : dp[1],
]
)
else:
if self.future_array_shapes:
out = np.conj(data[blt_start:blt_stop:dblt, :, indp[1]])
else:
out = np.conj(data[blt_start:blt_stop:dblt, :, :, indp[1]])
else:
out = data[ind2]
if p_reg_spaced[1]:
if self.future_array_shapes:
out = np.conj(out[:, :, p_start[1] : p_stop[1] : dp[1]])
else:
out = np.conj(out[:, :, :, p_start[1] : p_stop[1] : dp[1]])
else:
if self.future_array_shapes:
out = np.conj(out[:, :, indp[1]])
else:
out = np.conj(out[:, :, :, indp[1]])
else:
# both conjugated and unconjugated baselines
out = (data[ind1], np.conj(data[ind2]))
if p_reg_spaced[0] and p_reg_spaced[1]:
if self.future_array_shapes:
out = np.append(
out[0][:, :, p_start[0] : p_stop[0] : dp[0]],
out[1][:, :, p_start[1] : p_stop[1] : dp[1]],
axis=0,
)
else:
out = np.append(
out[0][:, :, :, p_start[0] : p_stop[0] : dp[0]],
out[1][:, :, :, p_start[1] : p_stop[1] : dp[1]],
axis=0,
)
else:
if self.future_array_shapes:
out = np.append(
out[0][:, :, indp[0]], out[1][:, :, indp[1]], axis=0
)
else:
out = np.append(
out[0][:, :, :, indp[0]], out[1][:, :, :, indp[1]], axis=0
)
if squeeze == "full":
out = np.squeeze(out)
elif squeeze == "default":
if self.future_array_shapes:
if out.shape[2] == 1:
# one polarization dimension
out = np.squeeze(out, axis=2)
else:
if out.shape[3] == 1:
# one polarization dimension
out = np.squeeze(out, axis=3)
if out.shape[1] == 1:
# one spw dimension
out = np.squeeze(out, axis=1)
elif squeeze != "none":
raise ValueError(
'"' + str(squeeze) + '" is not a valid option for squeeze.'
'Only "default", "none", or "full" are allowed.'
)
if force_copy:
out = np.array(out)
elif out.base is not None:
# if out is a view rather than a copy, make it read-only
out.flags.writeable = False
return out
def get_ants(self):
"""
Get the unique antennas that have data associated with them.
Returns
-------
ndarray of int
Array of unique antennas with data associated with them.
"""
return np.unique(np.append(self.ant_1_array, self.ant_2_array))
def get_baseline_nums(self):
"""
Get the unique baselines that have data associated with them.
Returns
-------
ndarray of int
Array of unique baselines with data associated with them.
"""
return np.unique(self.baseline_array)
def get_antpairs(self):
"""
Get the unique antpair tuples that have data associated with them.
Returns
-------
list of tuples of int
list of unique antpair tuples (ant1, ant2) with data associated with them.
"""
return list(zip(*self.baseline_to_antnums(self.get_baseline_nums())))
def get_pols(self):
"""
Get the polarizations in the data.
Returns
-------
list of str
list of polarizations (as strings) in the data.
"""
return uvutils.polnum2str(
self.polarization_array, x_orientation=self.x_orientation
)
def get_antpairpols(self):
"""
Get the unique antpair + pol tuples that have data associated with them.
Returns
-------
list of tuples of int
list of unique antpair + pol tuples (ant1, ant2, pol) with data
associated with them.
"""
pols = self.get_pols()
bls = self.get_antpairs()
return [(bl) + (pol,) for bl in bls for pol in pols]
def get_feedpols(self):
"""
Get the unique antenna feed polarizations in the data.
Returns
-------
list of str
list of antenna feed polarizations (e.g. ['X', 'Y']) in the data.
Raises
------
ValueError
If any pseudo-Stokes visibilities are present
"""
if np.any(self.polarization_array > 0):
raise ValueError(
"Pseudo-Stokes visibilities cannot be interpreted as feed polarizations"
)
else:
return list(set("".join(self.get_pols())))
def get_data(self, key1, key2=None, key3=None, squeeze="default", force_copy=False):
"""
Get the data corresonding to a baseline and/or polarization.
Parameters
----------
key1, key2, key3 : int or tuple of ints
Identifier of which data to get, can be passed as 1, 2, or 3 arguments
or as a single tuple of length 1, 2, or 3. These are collectively
called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, get all data for
that pol.
else:
interpreted as a baseline number, get all data for that baseline.
if key is length 2: interpreted as an antenna pair, get all data
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2, pol),
get all data for that baseline, pol. pol may be a string or int.
squeeze : str
string specifying how to squeeze the returned array. Options are:
'default': squeeze pol and spw dimensions if possible;
'none': no squeezing of resulting numpy array;
'full': squeeze all length 1 dimensions.
force_copy : bool
Option to explicitly make a copy of the data.
Returns
-------
ndarray
copy (or if possible, a read-only view) of relevant section of data.
If data exists conjugate to requested antenna pair, it will be conjugated
before returning.
"""
key = []
for val in [key1, key2, key3]:
if isinstance(val, str):
key.append(val)
elif val is not None:
key += list(uvutils._get_iterable(val))
if len(key) > 3:
raise ValueError("no more than 3 key values can be passed")
ind1, ind2, indp = self._key2inds(key)
out = self._smart_slicing(
self.data_array, ind1, ind2, indp, squeeze=squeeze, force_copy=force_copy
)
return out
def get_flags(
self, key1, key2=None, key3=None, squeeze="default", force_copy=False
):
"""
Get the flags corresonding to a baseline and/or polarization.
Parameters
----------
key1, key2, key3 : int or tuple of ints
Identifier of which data to get, can be passed as 1, 2, or 3 arguments
or as a single tuple of length 1, 2, or 3. These are collectively
called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, get all flags for
that pol.
else:
interpreted as a baseline number, get all flags for that baseline.
if key is length 2: interpreted as an antenna pair, get all flags
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2, pol),
get all flags for that baseline, pol. pol may be a string or int.
squeeze : str
string specifying how to squeeze the returned array. Options are:
'default': squeeze pol and spw dimensions if possible;
'none': no squeezing of resulting numpy array;
'full': squeeze all length 1 dimensions.
force_copy : bool
Option to explicitly make a copy of the data.
Returns
-------
ndarray
copy (or if possible, a read-only view) of relevant section of flags.
"""
key = []
for val in [key1, key2, key3]:
if isinstance(val, str):
key.append(val)
elif val is not None:
key += list(uvutils._get_iterable(val))
if len(key) > 3:
raise ValueError("no more than 3 key values can be passed")
ind1, ind2, indp = self._key2inds(key)
# When we select conjugated baselines, there is a call to np.conj()
# inside of _smart_slicing to correct the data array. This has the
# unintended consequence of promoting the dtype of an array of np.bool_
# to np.int8. Rather than having a bunch of special handling for this
# ~corner case, we instead explicitly cast back to np.bool_ before we
# hand back to the user.
out = self._smart_slicing(
self.flag_array, ind1, ind2, indp, squeeze=squeeze, force_copy=force_copy
).astype(np.bool_)
return out
def get_nsamples(
self, key1, key2=None, key3=None, squeeze="default", force_copy=False
):
"""
Get the nsamples corresonding to a baseline and/or polarization.
Parameters
----------
key1, key2, key3 : int or tuple of ints
Identifier of which data to get, can be passed as 1, 2, or 3 arguments
or as a single tuple of length 1, 2, or 3. These are collectively
called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, get all nsamples for
that pol.
else:
interpreted as a baseline number, get all nsamples for that
baseline.
if key is length 2: interpreted as an antenna pair, get all nsamples
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2, pol),
get all nsamples for that baseline, pol. pol may be a string or int.
squeeze : str
string specifying how to squeeze the returned array. Options are:
'default': squeeze pol and spw dimensions if possible;
'none': no squeezing of resulting numpy array;
'full': squeeze all length 1 dimensions.
force_copy : bool
Option to explicitly make a copy of the data.
Returns
-------
ndarray
copy (or if possible, a read-only view) of relevant section of
nsample_array.
"""
key = []
for val in [key1, key2, key3]:
if isinstance(val, str):
key.append(val)
elif val is not None:
key += list(uvutils._get_iterable(val))
if len(key) > 3:
raise ValueError("no more than 3 key values can be passed")
ind1, ind2, indp = self._key2inds(key)
out = self._smart_slicing(
self.nsample_array, ind1, ind2, indp, squeeze=squeeze, force_copy=force_copy
)
return out
def get_times(self, key1, key2=None, key3=None):
"""
Get the times for a given antpair or baseline number.
Meant to be used in conjunction with get_data function.
Parameters
----------
key1, key2, key3 : int or tuple of ints
Identifier of which data to get, can be passed as 1, 2, or 3 arguments
or as a single tuple of length 1, 2, or 3. These are collectively
called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, get all times.
else:
interpreted as a baseline number, get all times for that baseline.
if key is length 2: interpreted as an antenna pair, get all times
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2, pol),
get all times for that baseline.
Returns
-------
ndarray
times from the time_array for the given antpair or baseline.
"""
key = []
for val in [key1, key2, key3]:
if isinstance(val, str):
key.append(val)
elif val is not None:
key += list(uvutils._get_iterable(val))
if len(key) > 3:
raise ValueError("no more than 3 key values can be passed")
inds1, inds2, indp = self._key2inds(key)
return self.time_array[np.append(inds1, inds2)]
def get_lsts(self, key1, key2=None, key3=None):
"""
Get the LSTs for a given antpair or baseline number.
Meant to be used in conjunction with get_data function.
Parameters
----------
key1, key2, key3 : int or tuple of ints
Identifier of which data to get, can be passed as 1, 2, or 3 arguments
or as a single tuple of length 1, 2, or 3. These are collectively
called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, get all times.
else:
interpreted as a baseline number, get all times for that baseline.
if key is length 2: interpreted as an antenna pair, get all times
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2, pol),
get all times for that baseline.
Returns
-------
ndarray
LSTs from the lst_array for the given antpair or baseline.
"""
key = []
for val in [key1, key2, key3]:
if isinstance(val, str):
key.append(val)
elif val is not None:
key += list(uvutils._get_iterable(val))
if len(key) > 3:
raise ValueError("no more than 3 key values can be passed")
inds1, inds2, indp = self._key2inds(key)
return self.lst_array[np.append(inds1, inds2)]
def get_ENU_antpos(self, center=False, pick_data_ants=False):
"""
Get antenna positions in ENU (topocentric) coordinates in units of meters.
Parameters
----------
center : bool
If True, subtract median of array position from antpos
pick_data_ants : bool
If True, return only antennas found in data
Returns
-------
antpos : ndarray
Antenna positions in ENU (topocentric) coordinates in units of
meters, shape=(Nants, 3)
ants : ndarray
Antenna numbers matching ordering of antpos, shape=(Nants,)
"""
antpos = uvutils.ENU_from_ECEF(
(self.antenna_positions + self.telescope_location),
*self.telescope_location_lat_lon_alt,
frame=self._telescope_location.frame,
)
ants = self.antenna_numbers
if pick_data_ants:
data_ants = np.unique(np.concatenate([self.ant_1_array, self.ant_2_array]))
telescope_ants = self.antenna_numbers
select = np.in1d(telescope_ants, data_ants)
antpos = antpos[select, :]
ants = telescope_ants[select]
if center is True:
antpos -= np.median(antpos, axis=0)
return antpos, ants
def _set_method_helper(self, dshape, key1, key2=None, key3=None):
"""
Extract the indices for setting data, flags, or nsample arrays.
This is a helper method designed to work with set_data, set_flags, and
set_nsamples. Given the shape of the data-like array and the keys
corresponding to where the data should end up, it finds the indices
that are needed for the `_index_dset` method.
Parameters
----------
dshape : tuple of int
The shape of the data-like array. This is used to ensure the array
is compatible with the indices selected.
key1, key2, key3 : int or tuple of ints
Identifier of which flags to set, can be passed as 1, 2, or 3 arguments
or as a single tuple of length 1, 2, or 3. These are collectively
called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, set all flags for
that pol.
else:
interpreted as a baseline number, set all flags for that baseline.
if key is length 2: interpreted as an antenna pair, set all flags
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2, pol),
set all flags for that baseline, pol. pol may be a string or int.
Returns
-------
inds : tuple of int
The indices in the data-like array to slice into.
Raises
------
ValueError:
If more than 3 keys are passed, if the requested indices are
conjugated in the data, if the data array shape is not compatible
with the indices.
"""
key = []
for val in [key1, key2, key3]:
if isinstance(val, str):
key.append(val)
elif val is not None:
key += list(uvutils._get_iterable(val))
if len(key) > 3:
raise ValueError("no more than 3 key values can be passed")
ind1, ind2, indp = self._key2inds(key)
if len(ind2) != 0:
raise ValueError(
"the requested key is present on the object, but conjugated. Please "
"conjugate data and keys appropriately and try again"
)
if self.future_array_shapes:
expected_shape = (len(ind1), self.Nfreqs, len(indp[0]))
else:
expected_shape = (len(ind1), 1, self.Nfreqs, len(indp[0]))
if dshape != expected_shape:
raise ValueError(
"the input array is not compatible with the shape of the destination. "
f"Input array shape is {dshape}, expected shape is {expected_shape}."
)
blt_slices, blt_sliceable = uvutils._convert_to_slices(
ind1, max_nslice_frac=0.1
)
pol_slices, pol_sliceable = uvutils._convert_to_slices(
indp[0], max_nslice_frac=0.5
)
if self.future_array_shapes:
inds = [ind1, np.s_[:], indp[0]]
else:
inds = [ind1, np.s_[:], np.s_[:], indp[0]]
if blt_sliceable:
inds[0] = blt_slices
if pol_sliceable:
inds[-1] = pol_slices
return tuple(inds)
def set_data(self, data, key1, key2=None, key3=None):
"""
Set the data array to some values provided by the user.
Parameters
----------
data : ndarray of complex
The data to overwrite into the data_array. Must be the same shape as
the target indices.
key1, key2, key3 : int or tuple of ints
Identifier of which data to set, can be passed as 1, 2, or 3 arguments
or as a single tuple of length 1, 2, or 3. These are collectively
called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, get all data for
that pol.
else:
interpreted as a baseline number, get all data for that baseline.
if key is length 2: interpreted as an antenna pair, get all data
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2, pol),
get all data for that baseline, pol. pol may be a string or int.
Returns
-------
None
Raises
------
ValueError:
If more than 3 keys are passed, if the requested indices are
conjugated in the data, if the data array shape is not compatible
with the indices.
"""
dshape = data.shape
inds = self._set_method_helper(dshape, key1, key2, key3)
uvutils._index_dset(self.data_array, inds, data)
return
def set_flags(self, flags, key1, key2=None, key3=None):
"""
Set the flag array to some values provided by the user.
Parameters
----------
flag : ndarray of boolean
The flags to overwrite into the fkag_array. Must be the same shape
as the target indices.
key1, key2, key3 : int or tuple of ints
Identifier of which flags to set, can be passed as 1, 2, or 3 arguments
or as a single tuple of length 1, 2, or 3. These are collectively
called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, set all flags for
that pol.
else:
interpreted as a baseline number, set all flags for that baseline.
if key is length 2: interpreted as an antenna pair, set all flags
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2, pol),
set all flags for that baseline, pol. pol may be a string or int.
Returns
-------
None
Raises
------
ValueError:
If more than 3 keys are passed, if the requested indices are
conjugated in the data, if the data array shape is not compatible
with the indices.
"""
dshape = flags.shape
inds = self._set_method_helper(dshape, key1, key2, key3)
uvutils._index_dset(self.flag_array, inds, flags)
return
def set_nsamples(self, nsamples, key1, key2=None, key3=None):
"""
Set the nsamples array to some values provided by the user.
Parameters
----------
nsamples : ndarray of float
The nsamples to overwrite into the nsample_array. Must be the same
shape as the target indices.
key1, key2, key3 : int or tuple of ints
Identifier of which nsamples to set, can be passed as 1, 2, or 3
arguments or as a single tuple of length 1, 2, or 3. These are
collectively called the key.
If key is length 1:
if (key < 5) or (type(key) is str):
interpreted as a polarization number/name, set all data for
that pol.
else:
interpreted as a baseline number, set all nsamples for that
baseline.
if key is length 2: interpreted as an antenna pair, set all nsamples
for that baseline.
if key is length 3: interpreted as antenna pair and pol (ant1, ant2,
pol), set all nsamples for that baseline, pol. pol may be a
string or int.
Returns
-------
None
Raises
------
ValueError:
If more than 3 keys are passed, if the requested indices are
conjugated in the data, if the data array shape is not compatible
with the indices.
"""
dshape = nsamples.shape
inds = self._set_method_helper(dshape, key1, key2, key3)
uvutils._index_dset(self.nsample_array, inds, nsamples)
return
def antpairpol_iter(self, squeeze="default"):
"""
Iterate the data for each antpair, polarization combination.
Parameters
----------
squeeze : str
string specifying how to squeeze the returned array. Options are:
'default': squeeze pol and spw dimensions if possible;
'none': no squeezing of resulting numpy array;
'full': squeeze all length 1 dimensions.
Yields
------
key : tuple
antenna1, antenna2, and polarization string
data : ndarray of complex
data for the ant pair and polarization specified in key
"""
antpairpols = self.get_antpairpols()
for key in antpairpols:
yield (key, self.get_data(key, squeeze=squeeze))
def conjugate_bls(self, convention="ant1<ant2", use_enu=True, uvw_tol=0.0):
"""
Conjugate baselines according to one of the supported conventions.
This will fail if only one of the cross pols is present (because
conjugation requires changing the polarization number for cross pols).
Parameters
----------
convention : str or array_like of int
A convention for the directions of the baselines, options are:
'ant1<ant2', 'ant2<ant1', 'u<0', 'u>0', 'v<0', 'v>0' or an
index array of blt indices to conjugate.
use_enu : bool
Use true antenna positions to determine uv location (as opposed to
uvw array). Only applies if `convention` is 'u<0', 'u>0', 'v<0', 'v>0'.
Set to False to use uvw array values.
uvw_tol : float
Defines a tolerance on uvw coordinates for setting the
u>0, u<0, v>0, or v<0 conventions. Defaults to 0m.
Raises
------
ValueError
If convention is not an allowed value or if not all conjugate pols exist.
"""
if isinstance(convention, (np.ndarray, list, tuple)):
convention = np.array(convention)
if (
np.max(convention) >= self.Nblts
or np.min(convention) < 0
or convention.dtype not in [int, np.int_, np.int32, np.int64]
):
raise ValueError(
"If convention is an index array, it must "
"contain integers and have values greater "
"than zero and less than NBlts"
)
else:
if convention not in ["ant1<ant2", "ant2<ant1", "u<0", "u>0", "v<0", "v>0"]:
raise ValueError(
"convention must be one of 'ant1<ant2', "
"'ant2<ant1', 'u<0', 'u>0', 'v<0', 'v>0' or "
"an index array with values less than NBlts"
)
if isinstance(convention, str):
if convention in ["u<0", "u>0", "v<0", "v>0"]:
if use_enu is True:
enu, anum = self.get_ENU_antpos()
anum = anum.tolist()
uvw_array_use = np.zeros_like(self.uvw_array)
for i in range(self.baseline_array.size):
a1, a2 = self.ant_1_array[i], self.ant_2_array[i]
i1, i2 = anum.index(a1), anum.index(a2)
uvw_array_use[i, :] = enu[i2] - enu[i1]
else:
uvw_array_use = copy.copy(self.uvw_array)
if convention == "ant1<ant2":
index_array = np.asarray(self.ant_1_array > self.ant_2_array).nonzero()
elif convention == "ant2<ant1":
index_array = np.asarray(self.ant_2_array > self.ant_1_array).nonzero()
elif convention == "u<0":
index_array = np.asarray(
(uvw_array_use[:, 0] > uvw_tol)
| (uvw_array_use[:, 1] > uvw_tol)
& np.isclose(uvw_array_use[:, 0], 0, atol=uvw_tol)
| (uvw_array_use[:, 2] > uvw_tol)
& np.isclose(uvw_array_use[:, 0], 0, atol=uvw_tol)
& np.isclose(uvw_array_use[:, 1], 0, atol=uvw_tol)
).nonzero()
elif convention == "u>0":
index_array = np.asarray(
(uvw_array_use[:, 0] < -uvw_tol)
| (
(uvw_array_use[:, 1] < -uvw_tol)
& np.isclose(uvw_array_use[:, 0], 0, atol=uvw_tol)
)
| (
(uvw_array_use[:, 2] < -uvw_tol)
& np.isclose(uvw_array_use[:, 0], 0, atol=uvw_tol)
& np.isclose(uvw_array_use[:, 1], 0, atol=uvw_tol)
)
).nonzero()
elif convention == "v<0":
index_array = np.asarray(
(uvw_array_use[:, 1] > uvw_tol)
| (uvw_array_use[:, 0] > uvw_tol)
& np.isclose(uvw_array_use[:, 1], 0, atol=uvw_tol)
| (uvw_array_use[:, 2] > uvw_tol)
& np.isclose(uvw_array_use[:, 0], 0, atol=uvw_tol)
& np.isclose(uvw_array_use[:, 1], 0, atol=uvw_tol)
).nonzero()
elif convention == "v>0":
index_array = np.asarray(
(uvw_array_use[:, 1] < -uvw_tol)
| (uvw_array_use[:, 0] < -uvw_tol)
& np.isclose(uvw_array_use[:, 1], 0, atol=uvw_tol)
| (uvw_array_use[:, 2] < -uvw_tol)
& np.isclose(uvw_array_use[:, 0], 0, atol=uvw_tol)
& np.isclose(uvw_array_use[:, 1], 0, atol=uvw_tol)
).nonzero()
else:
index_array = convention
if index_array[0].size > 0:
new_pol_inds = uvutils.reorder_conj_pols(self.polarization_array)
self.uvw_array[index_array] *= -1
if not self.metadata_only:
orig_data_array = copy.copy(self.data_array)
for pol_ind in np.arange(self.Npols):
if self.future_array_shapes:
self.data_array[
index_array, :, new_pol_inds[pol_ind]
] = np.conj(orig_data_array[index_array, :, pol_ind])
else:
self.data_array[
index_array, :, :, new_pol_inds[pol_ind]
] = np.conj(orig_data_array[index_array, :, :, pol_ind])
ant_1_vals = self.ant_1_array[index_array]
ant_2_vals = self.ant_2_array[index_array]
self.ant_1_array[index_array] = ant_2_vals
self.ant_2_array[index_array] = ant_1_vals
self.baseline_array[index_array] = self.antnums_to_baseline(
self.ant_1_array[index_array], self.ant_2_array[index_array]
)
self.Nbls = np.unique(self.baseline_array).size
def reorder_pols(
self,
order="AIPS",
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
):
"""
Arrange polarization axis according to desired order.
Parameters
----------
order : str
Either a string specifying a canonical ordering ('AIPS' or 'CASA')
or an index array of length Npols that specifies how to shuffle the
data (this is not the desired final pol order).
CASA ordering has cross-pols in between (e.g. XX,XY,YX,YY)
AIPS ordering has auto-pols followed by cross-pols (e.g. XX,YY,XY,YX)
Default ('AIPS') will sort by absolute value of pol values.
run_check : bool
Option to check for the existence and proper shapes of parameters
after reordering.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reordering.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
Raises
------
ValueError
If the order is not one of the allowed values.
"""
if isinstance(order, (np.ndarray, list, tuple)):
order = np.array(order)
if (
order.size != self.Npols
or order.dtype not in [int, np.int_, np.int32, np.int64]
or np.min(order) < 0
or np.max(order) >= self.Npols
):
raise ValueError(
"If order is an index array, it must "
"contain integers and be length Npols."
)
index_array = order
elif order == "AIPS":
index_array = np.argsort(np.abs(self.polarization_array))
elif order == "CASA":
casa_order = np.array([1, 2, 3, 4, -1, -3, -4, -2, -5, -7, -8, -6])
pol_inds = []
for pol in self.polarization_array:
pol_inds.append(np.where(casa_order == pol)[0][0])
index_array = np.argsort(pol_inds)
else:
raise ValueError(
"order must be one of: 'AIPS', 'CASA', or an "
"index array of length Npols"
)
self.polarization_array = self.polarization_array[index_array]
if not self.metadata_only:
# data array is special and large, take is faster here
if self.future_array_shapes:
self.data_array = np.take(self.data_array, index_array, axis=2)
self.nsample_array = self.nsample_array[:, :, index_array]
self.flag_array = self.flag_array[:, :, index_array]
else:
self.data_array = np.take(self.data_array, index_array, axis=3)
self.nsample_array = self.nsample_array[:, :, :, index_array]
self.flag_array = self.flag_array[:, :, :, index_array]
# check if object is self-consistent
if run_check:
self.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
def reorder_blts(
self,
order="time",
minor_order=None,
autos_first=False,
conj_convention=None,
uvw_tol=0.0,
conj_convention_use_enu=True,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
):
"""
Arrange baseline-times axis according to desired order.
If the specified order is an index type, it will be the slowest changing index
along the baseline-time axis (the minor order will be the next slowest).
If the `conj_convention` is set, this method can also conjugate some baselines
using the `conjugate_bls` method.
Parameters
----------
order : str or array_like of int
A string describing the desired order along the blt axis or an
index array of length Nblts that specifies the new order.
If a string, the options are: `time`, `baseline`, `ant1`, `ant2`, `bda`.
If this is `time`, `baseline`, `ant1`, `ant2` then this will be the
slowest changing index.
minor_order : str
Optionally specify a secondary ordering.
Default depends on how order is set:
if order is 'time', this defaults to `baseline`,
if order is `ant1`, or `ant2` this defaults to the other antenna,
if order is `baseline` the only allowed value is `time`.
Ignored if order is `bda` or an integer array.
If this is the same as order, it is reset to the default.
This will be the next slowest changing index.
autos_first : bool
If True, sort the autos before all the crosses. The autos and crosses will
each be sorted according to the order and minor order keywords. Ignored if
order is an integer array.
conj_convention : str or array_like of int
Optionally conjugate baselines to make the baselines have the
desired orientation. See conjugate_bls for allowed values and details.
uvw_tol : float
If conjugating baselines, sets a tolerance for determining the signs
of u,v, and w, and whether or not they are zero.
See conjugate_bls for details.
conj_convention_use_enu: bool
If `conj_convention` is set, this is passed to conjugate_bls, see that
method for details.
run_check : bool
Option to check for the existence and proper shapes of parameters
after reordering.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reordering.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
Raises
------
ValueError
If parameter values are inappropriate
"""
if isinstance(order, (np.ndarray, list, tuple)):
order = np.array(order)
if order.size != self.Nblts or order.dtype not in [
int,
np.int_,
np.int32,
np.int64,
]:
raise ValueError(
"If order is an index array, it must "
"contain integers and be length Nblts."
)
if minor_order is not None:
raise ValueError(
"Minor order cannot be set if order is an index array."
)
else:
if order not in ["time", "baseline", "ant1", "ant2", "bda"]:
raise ValueError(
"order must be one of 'time', 'baseline', "
"'ant1', 'ant2', 'bda' or an index array of "
"length Nblts"
)
if minor_order == order:
minor_order = None
if minor_order is not None:
if minor_order not in ["time", "baseline", "ant1", "ant2"]:
raise ValueError(
"minor_order can only be one of 'time', "
"'baseline', 'ant1', 'ant2'"
)
if isinstance(order, np.ndarray) or order == "bda":
raise ValueError(
"minor_order cannot be specified if order is "
"'bda' or an index array."
)
if order == "baseline":
if minor_order in ["ant1", "ant2"]:
raise ValueError("minor_order conflicts with order")
else:
if order == "time":
minor_order = "baseline"
elif order == "ant1":
minor_order = "ant2"
elif order == "ant2":
minor_order = "ant1"
elif order == "baseline":
minor_order = "time"
if conj_convention is not None:
self.conjugate_bls(
convention=conj_convention,
use_enu=conj_convention_use_enu,
uvw_tol=uvw_tol,
)
if isinstance(order, str):
if minor_order is None:
self.blt_order = (order,)
self._blt_order.form = (1,)
else:
self.blt_order = (order, minor_order)
# set it back to the right shape in case it was set differently before
self._blt_order.form = (2,)
else:
self.blt_order = None
if autos_first:
# find the auto indices
auto_inds = np.nonzero(self.ant_1_array == self.ant_2_array)[0]
cross_inds = np.nonzero(self.ant_1_array != self.ant_2_array)[0]
inds_use_list = [auto_inds, cross_inds]
else:
inds_use_list = [np.arange(self.Nblts)]
if not isinstance(order, np.ndarray):
index_array = []
for inds_use in inds_use_list:
if inds_use.size == 0:
continue
# Use lexsort to sort along different arrays in defined order.
if order == "time":
arr1 = self.time_array[inds_use]
if minor_order == "ant1":
arr2 = self.ant_1_array[inds_use]
arr3 = self.ant_2_array[inds_use]
elif minor_order == "ant2":
arr2 = self.ant_2_array[inds_use]
arr3 = self.ant_1_array[inds_use]
else:
# minor_order is baseline
arr2 = self.baseline_array[inds_use]
arr3 = self.baseline_array[inds_use]
elif order == "ant1":
arr1 = self.ant_1_array[inds_use]
if minor_order == "time":
arr2 = self.time_array[inds_use]
arr3 = self.ant_2_array[inds_use]
elif minor_order == "ant2":
arr2 = self.ant_2_array[inds_use]
arr3 = self.time_array[inds_use]
else: # minor_order is baseline
arr2 = self.baseline_array[inds_use]
arr3 = self.time_array[inds_use]
elif order == "ant2":
arr1 = self.ant_2_array[inds_use]
if minor_order == "time":
arr2 = self.time_array[inds_use]
arr3 = self.ant_1_array[inds_use]
elif minor_order == "ant1":
arr2 = self.ant_1_array[inds_use]
arr3 = self.time_array[inds_use]
else:
# minor_order is baseline
arr2 = self.baseline_array[inds_use]
arr3 = self.time_array[inds_use]
elif order == "baseline":
arr1 = self.baseline_array[inds_use]
# only allowed minor order is time
arr2 = self.time_array[inds_use]
arr3 = self.time_array[inds_use]
elif order == "bda":
arr1 = self.integration_time[inds_use]
# only allowed minor order is time
arr2 = self.baseline_array[inds_use]
arr3 = self.time_array[inds_use]
# lexsort uses the listed arrays from last to first
# (so the primary sort is on the last one)
index_array.extend(inds_use[np.lexsort((arr3, arr2, arr1))].tolist())
else:
index_array = order
index_array = np.asarray(index_array, dtype=int)
# actually do the reordering
self.ant_1_array = self.ant_1_array[index_array]
self.ant_2_array = self.ant_2_array[index_array]
self.baseline_array = self.baseline_array[index_array]
self.uvw_array = self.uvw_array[index_array, :]
self.time_array = self.time_array[index_array]
self.lst_array = self.lst_array[index_array]
self.integration_time = self.integration_time[index_array]
self.phase_center_app_ra = self.phase_center_app_ra[index_array]
self.phase_center_app_dec = self.phase_center_app_dec[index_array]
self.phase_center_frame_pa = self.phase_center_frame_pa[index_array]
self.phase_center_id_array = self.phase_center_id_array[index_array]
if not self.metadata_only:
self.data_array = self.data_array[index_array]
self.flag_array = self.flag_array[index_array]
self.nsample_array = self.nsample_array[index_array]
# check if object is self-consistent
if run_check:
self.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
def reorder_freqs(
self,
spw_order=None,
channel_order=None,
select_spw=None,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
):
"""
Arrange frequency axis according to desired order.
Can be applied across the entire frequency axis, or just a subset.
Parameters
----------
spw_order : str or array_like of int
A string describing the desired order of spectral windows along the
frequecy axis. Allowed strings include `number` (sort on spectral window
number) and `freq` (sort on median frequency). A '-' can be prepended
to signify descending order instead of the default ascending order,
e.g., if you have SPW #1 and 2, and wanted them ordered as [2, 1],
you would specify `-number`. Alternatively, one can supply an index array
of length Nspws that specifies how to shuffle the spws (this is not the
desired final spw order). Default is to apply no sorting of spectral
windows.
channel_order : str or array_like of int
A string describing the desired order of frequency channels within a
spectral window. Allowed strings are "freq" and "-freq", which will sort
channels within a spectral window by ascending or descending frequency
respectively. Alternatively, one can supply an index array of length Nfreqs
that specifies the new order. Default is to apply no sorting of channels
within a single spectral window. Note that proving an array_like of ints
will cause the values given to `spw_order` and `select_spw` to be ignored.
select_spw : int or array_like of int
An int or array_like of ints which specifies which spectral windows to
apply sorting. Note that setting this argument will cause the value
given to `spw_order` to be ignored.
run_check : bool
Option to check for the existence and proper shapes of parameters
after reordering.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reordering.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
Returns
-------
None
Raises
------
UserWarning
Raised if providing arguments to select_spw and channel_order (the latter
overrides the former).
ValueError
Raised if select_spw contains values not in spw_array, or if channel_order
is not the same length as freq_array.
"""
index_array = uvutils._sort_freq_helper(
self.Nfreqs,
self.freq_array,
self.Nspws,
self.spw_array,
self.flex_spw,
self.flex_spw_id_array,
self.future_array_shapes,
spw_order,
channel_order,
select_spw,
)
if index_array is None:
# This only happens if no sorting is needed
return
# Now update all of the arrays.
if self.future_array_shapes:
self.freq_array = self.freq_array[index_array]
if not self.metadata_only:
self.data_array = self.data_array[:, index_array, :]
self.flag_array = self.flag_array[:, index_array, :]
self.nsample_array = self.nsample_array[:, index_array, :]
else:
self.freq_array = self.freq_array[:, index_array]
if not self.metadata_only:
self.data_array = self.data_array[:, :, index_array, :]
self.flag_array = self.flag_array[:, :, index_array, :]
self.nsample_array = self.nsample_array[:, :, index_array, :]
if self.flex_spw:
self.flex_spw_id_array = self.flex_spw_id_array[index_array]
self.channel_width = self.channel_width[index_array]
# Reorder the spw-axis items based on their first appearance in the data
unique_index = np.sort(
np.unique(self.flex_spw_id_array, return_index=True)[1]
)
new_spw_array = self.flex_spw_id_array[unique_index]
if self.flex_spw_polarization_array is not None:
spw_sort_inds = np.zeros_like(self.spw_array)
for idx, spw in enumerate(new_spw_array):
spw_sort_inds[idx] = np.nonzero(self.spw_array == spw)[0]
self.flex_spw_polarization_array = self.flex_spw_polarization_array[
spw_sort_inds
]
self.spw_array = new_spw_array
elif self.future_array_shapes:
self.channel_width = self.channel_width[index_array]
if self.eq_coeffs is not None:
self.eq_coeffs = self.eq_coeffs[:, index_array]
# check if object is self-consistent
if run_check:
self.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
def remove_eq_coeffs(self):
"""
Remove equalization coefficients from the data.
Some telescopes, e.g. HERA, apply per-antenna, per-frequency gain
coefficients as part of the signal chain. These are stored in the
`eq_coeffs` attribute of the object. This method will remove them, so
that the data are in "unnormalized" raw units.
Parameters
----------
None
Returns
-------
None
Raises
------
ValueError
Raised if eq_coeffs or eq_coeffs_convention are not defined on the
object, or if eq_coeffs_convention is not one of "multiply" or "divide".
"""
if self.eq_coeffs is None:
raise ValueError(
"The eq_coeffs attribute must be defined on the object to apply them."
)
if self.eq_coeffs_convention is None:
raise ValueError(
"The eq_coeffs_convention attribute must be defined on the object "
"to apply them."
)
if self.eq_coeffs_convention not in ("multiply", "divide"):
raise ValueError(
"Got unknown convention {}. Must be one of: "
'"multiply", "divide"'.format(self.eq_coeffs_convention)
)
# apply coefficients for each baseline
for key in self.get_antpairs():
# get indices for this key
blt_inds = self.antpair2ind(key)
ant1_index = np.asarray(self.antenna_numbers == key[0]).nonzero()[0][0]
ant2_index = np.asarray(self.antenna_numbers == key[1]).nonzero()[0][0]
eq_coeff1 = self.eq_coeffs[ant1_index, :]
eq_coeff2 = self.eq_coeffs[ant2_index, :]
# make sure coefficients are the right size to broadcast
eq_coeff1 = np.repeat(eq_coeff1[:, np.newaxis], self.Npols, axis=1)
eq_coeff2 = np.repeat(eq_coeff2[:, np.newaxis], self.Npols, axis=1)
if self.eq_coeffs_convention == "multiply":
self.data_array[blt_inds] *= eq_coeff1 * eq_coeff2
else:
self.data_array[blt_inds] /= eq_coeff1 * eq_coeff2
return
def _apply_w_proj(self, new_w_vals, old_w_vals, select_mask=None):
"""
Apply corrections based on changes to w-coord.
Adjusts the data to account for a change along the w-axis of a baseline.
Parameters
----------
new_w_vals: float or ndarray of float
New w-coordinates for the baselines, in units of meters. Can either be a
solitary float (helpful for unphasing data, where new_w_vals can be set to
0.0) or an array of shape (Nselect,) (which is Nblts if select_mask=None).
old_w_vals: float or ndarray of float
Old w-coordinates for the baselines, in units of meters. Can either be a
solitary float (helpful for unphasing data, where new_w_vals can be set to
0.0) or an array of shape (Nselect,) (which is Nblts if select_mask=None).
select_mask: ndarray of bool
Array is of shape (Nblts,), where the sum of all enties marked True is
equal to Nselect (mentioned above).
Raises
------
IndexError
If the length of new_w_vals or old_w_vals isn't compatible with
select_mask, or if select mask isn't the right length.
"""
# If we only have metadata, then we have no work to do. W00t!
if self.metadata_only or (self.data_array is None):
return
if select_mask is None:
select_len = self.Nblts
else:
try:
inv_mask = np.ones(self.Nblts, dtype=bool)
inv_mask[select_mask] = False
select_mask = ~inv_mask
select_len = np.sum(select_mask)
except IndexError as err:
raise IndexError(
"select_mask must be an array-like, either of ints with shape "
"(Nblts), or of ints within the range (-Nblts, Nblts)."
) from err
# Promote everything to float64 ndarrays if they aren't already
old_w_vals = np.array(old_w_vals, dtype=np.float64)
old_w_vals.shape += (1,) if (old_w_vals.ndim == 0) else ()
new_w_vals = np.array(new_w_vals, dtype=np.float64)
new_w_vals.shape += (1,) if (new_w_vals.ndim == 0) else ()
# Make sure the lengths of everything make sense
new_val_len = len(new_w_vals)
old_val_len = len(old_w_vals)
if new_val_len not in [1, select_len]:
raise IndexError(
f"The length of new_w_vals is wrong (expected 1 or {select_len}, "
f"got {new_val_len})!"
)
if old_val_len not in [1, select_len]:
raise IndexError(
f"The length of old_w_vals is wrong (expected 1 or {select_len}, "
f"got {old_val_len})!"
)
# Calculate the difference in w terms as a function of freq. Note that the
# 1/c is there to speed of processing (faster to multiply than divide)
delta_w_lambda = (
(new_w_vals - old_w_vals).reshape(-1, 1)
* (1.0 / const.c.to("m/s").value)
* self.freq_array.reshape(1, self.Nfreqs)
)
if select_mask is None or np.all(select_mask):
# If all the w values are changing, it turns out to be twice as fast
# to ditch any sort of selection mask and just do the full multiply.
if self.future_array_shapes:
self.data_array *= np.exp(
(-1j * 2 * np.pi) * delta_w_lambda[:, :, None]
)
else:
self.data_array *= np.exp(
(-1j * 2 * np.pi) * delta_w_lambda[:, None, :, None]
)
elif np.any(select_mask):
# In the case we are _not_ doing all baselines, use a selection mask to
# only update the values we need. In the worse case, it slows down the
# processing by ~2x, but it can save a lot on time and memory if only
# needing to update a select number of baselines.
if self.future_array_shapes:
self.data_array[select_mask] *= np.exp(
(-1j * 2 * np.pi) * delta_w_lambda[:, :, None]
)
else:
self.data_array[select_mask] *= np.exp(
(-1j * 2 * np.pi) * delta_w_lambda[:, None, :, None]
)
def unphase_to_drift(self, phase_frame=None, use_ant_pos=True, use_old_proj=False):
"""
Undo phasing to get back to an `unprojected` state.
Deprecated. Use "unproject_phase" instead.
See the phasing memo under docs/references for more documentation.
Parameters
----------
phase_frame : str
Deprecated, will be removed in version 2.4. Only used if use_old_proj is
True. The astropy frame to phase from Either 'icrs' or 'gcrs'. 'gcrs'
accounts for precession & nutation, 'icrs' also includes abberation.
Defaults to using the "cat_frame" value in the `phase_center_catalog`.
use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions
rather than from the existing uvws. Default is True.
use_old_proj : bool
Deprecates, will be removed in version 2.4. If True, uses the 'old' way of
calculating baseline projections. In the future, if a dataset is known to
be phased using the old way, use the `fix_phase` method to fix the phasing
before calling this method.
Raises
------
ValueError
If the object is alread unprojected.
"""
warnings.warn(
"This method is deprecated in favor of `unproject_phase`, which better "
"describes what is being done. This method will be removed in version 2.4.",
DeprecationWarning,
)
self.unproject_phase(
phase_frame=phase_frame, use_ant_pos=use_ant_pos, use_old_proj=use_old_proj
)
def unproject_phase(
self,
phase_frame=None,
use_ant_pos=True,
use_old_proj=False,
select_mask=None,
cat_name="unprojected",
):
"""
Undo phasing to get back to an `unprojected` state.
See the phasing memo under docs/references for more documentation.
Parameters
----------
phase_frame : str
Deprecated, will be removed in version 2.4. Only used if use_old_proj is
True. The astropy frame to phase from, either 'icrs' or 'gcrs'. 'gcrs'
accounts for precession & nutation, 'icrs' also includes abberation.
Defaults to using the "cat_frame" value in the `phase_center_catalog`.
use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions
rather than from the existing uvws. Default is True.
use_old_proj : bool
Deprecated, will be removed in version 2.4. If True, uses the 'old' way of
calculating baseline projections. If a dataset is known to be phased using
the old way, use the `fix_phase` method to fix the phasing before calling
this method.
select_mask : ndarray of bool
Optional mask for selecting which data to operate on along the blt-axis.
Shape is (Nblts,). Ignored if `use_old_proj` is True.
cat_name : str
Name for the newly unprojected entry in the phase_center_catalog.
Raises
------
ValueError
If the object is alread unprojected.
"""
if not use_old_proj:
# select_mask_use is length Nblts, True means should be unprojected
# only select blts that are actually phased.
if select_mask is not None:
if len(select_mask) != self.Nblts:
raise IndexError("Selection mask must be of length Nblts.")
if not isinstance(select_mask[0], (bool, np.bool_)):
raise ValueError("Selection mask must be a boolean array")
select_mask_use = ~self._check_for_cat_type("unprojected") & select_mask
else:
select_mask_use = ~self._check_for_cat_type("unprojected")
if np.all(~select_mask_use):
warnings.warn("No selected baselines are projected, doing nothing")
new_uvw = uvutils.calc_uvw(
lst_array=self.lst_array,
use_ant_pos=use_ant_pos,
uvw_array=self.uvw_array,
antenna_positions=self.antenna_positions,
antenna_numbers=self.antenna_numbers,
ant_1_array=self.ant_1_array,
ant_2_array=self.ant_2_array,
old_app_ra=self.phase_center_app_ra,
old_app_dec=self.phase_center_app_dec,
old_frame_pa=self.phase_center_frame_pa,
telescope_lat=self.telescope_location_lat_lon_alt[0],
telescope_lon=self.telescope_location_lat_lon_alt[1],
to_enu=True,
)
self._apply_w_proj(0.0, self.uvw_array[select_mask_use, 2], select_mask_use)
self.uvw_array = new_uvw
# remove/update phase center
match_id, match_diffs = self._look_in_catalog(
cat_name=cat_name, cat_type="unprojected"
)
if match_diffs == 0:
self.phase_center_id_array[select_mask_use] = match_id
else:
self.phase_center_id_array[select_mask_use] = self._add_phase_center(
cat_name, cat_type="unprojected"
)
self._clear_unused_phase_centers()
self.phase_center_app_ra[select_mask_use] = self.lst_array[
select_mask_use
].copy()
self.phase_center_app_dec[
select_mask_use
] = self.telescope_location_lat_lon_alt[0]
self.phase_center_frame_pa[select_mask_use] = 0
return
warnings.warn(
"The `use_old_proj` option is deprecated and will be removed in "
"version 2.4. In the future, if a dataset is known to be phased using the "
"old way, use the `fix_phase` method to fix the phasing before calling "
"this method.",
DeprecationWarning,
)
if phase_frame is not None:
warnings.warn(
"The `phase_frame` parameter is deprecated and will be removed in "
"version 2.4 because it only applies if `use_old_proj` is True.",
DeprecationWarning,
)
# If you are a new phase style data set, there's no valid reason to be going
# back to the old phase method. Time to bail!
compatible, reason = self._old_phase_attributes_compatible()
if not compatible:
raise ValueError(
"Objects with " + reason + " are not compatible with the old phasing "
"method, please set use_old_proj=False."
)
# TODO the following code needs to be moved to `fix_phase` in version 2.4
# when the `use_old_proj` parameter is removed
phase_dict = list(self.phase_center_catalog.values())[0]
if phase_dict["cat_type"] == "unprojected":
raise ValueError("Data are already unprojected.")
if phase_frame is None:
phase_frame = phase_dict["cat_frame"]
icrs_coord = SkyCoord(
ra=phase_dict["cat_lon"],
dec=phase_dict["cat_lat"],
unit="radian",
frame="icrs",
)
if phase_frame == "icrs":
frame_phase_center = icrs_coord
else:
# use center of observation for obstime for gcrs
center_time = np.mean([np.max(self.time_array), np.min(self.time_array)])
icrs_coord.obstime = Time(center_time, format="jd")
frame_phase_center = icrs_coord.transform_to("gcrs")
# This promotion is REQUIRED to get the right answer when we
# add in the telescope location for ICRS
# In some cases, the uvws are already float64, but sometimes they're not
self.uvw_array = np.float64(self.uvw_array)
# apply -w phasor
if not self.metadata_only:
w_lambda = (
self.uvw_array[:, 2].reshape(self.Nblts, 1)
/ const.c.to("m/s").value
* self.freq_array.reshape(1, self.Nfreqs)
)
if self.future_array_shapes:
phs = np.exp(-1j * 2 * np.pi * (-1) * w_lambda[:, :, None])
else:
phs = np.exp(-1j * 2 * np.pi * (-1) * w_lambda[:, None, :, None])
self.data_array *= phs
unique_times, _ = np.unique(self.time_array, return_index=True)
telescope_location = EarthLocation.from_geocentric(
*self.telescope_location, unit=units.m
)
obs_times = Time(unique_times, format="jd")
itrs_telescope_locations = telescope_location.get_itrs(obstime=obs_times)
itrs_telescope_locations = SkyCoord(itrs_telescope_locations)
# just calling transform_to(coord.GCRS) will delete the obstime information
# need to re-add obstimes for a GCRS transformation
if phase_frame == "gcrs":
frame_telescope_locations = itrs_telescope_locations.transform_to(
getattr(coord, f"{phase_frame}".upper())(obstime=obs_times)
)
else:
frame_telescope_locations = itrs_telescope_locations.transform_to(
getattr(coord, f"{phase_frame}".upper())
)
frame_telescope_locations.representation_type = "cartesian"
for ind, jd in enumerate(unique_times):
inds = np.where(self.time_array == jd)[0]
obs_time = obs_times[ind]
if use_ant_pos:
ant_uvw = uvutils.phase_uvw(
self.telescope_location_lat_lon_alt[1],
self.telescope_location_lat_lon_alt[0],
self.antenna_positions,
)
# instead of looping through every ind, find the spot in antenna number
# array where ant_num <= ant1 < ant_number and similarly for ant2
# for all baselines in inds
# then find the uvw coordinate for all at the same time
# antenna_numbers does not necessarily need to be in order on the object
# but needs to be in order for the searchsorted to work.
# ant1_index and ant2_index arrays will preserve the order of blts
ant_sort = np.argsort(self.antenna_numbers)
ant1_index = np.searchsorted(
self.antenna_numbers[ant_sort], self.ant_1_array[inds]
)
ant2_index = np.searchsorted(
self.antenna_numbers[ant_sort], self.ant_2_array[inds]
)
self.uvw_array[inds] = (
ant_uvw[ant_sort][ant2_index, :] - ant_uvw[ant_sort][ant1_index, :]
)
else:
frame_telescope_location = frame_telescope_locations[ind]
itrs_lat_lon_alt = self.telescope_location_lat_lon_alt
uvws_use = self.uvw_array[inds, :]
uvw_rel_positions = uvutils.unphase_uvw(
frame_phase_center.ra.rad, frame_phase_center.dec.rad, uvws_use
)
frame_uvw_coord = SkyCoord(
x=uvw_rel_positions[:, 0] * units.m + frame_telescope_location.x,
y=uvw_rel_positions[:, 1] * units.m + frame_telescope_location.y,
z=uvw_rel_positions[:, 2] * units.m + frame_telescope_location.z,
frame=phase_frame,
obstime=obs_time,
representation_type="cartesian",
)
itrs_uvw_coord = frame_uvw_coord.transform_to("itrs")
# now convert them to ENU, which is the space uvws are in
self.uvw_array[inds, :] = uvutils.ENU_from_ECEF(
itrs_uvw_coord.cartesian.get_xyz().value.T, *itrs_lat_lon_alt
)
# remove/add phase center
self.phase_center_id_array[:] = self._add_phase_center(
cat_name, cat_type="unprojected"
)
self._clear_unused_phase_centers()
self.phase_center_app_ra = self.lst_array.copy()
self.phase_center_app_dec[:] = (
np.zeros(self.Nblts) + self.telescope_location_lat_lon_alt[0]
)
self.phase_center_frame_pa = np.zeros(self.Nblts)
def _phase_dict_helper(
self,
lon,
lat,
epoch,
phase_frame,
ephem_times,
cat_type,
pm_ra,
pm_dec,
dist,
vrad,
cat_name,
lookup_name,
time_array,
):
"""
Supplies a dictionary with parameters for the phase method to use.
This method should not be called directly by users; it is instead a function
called by the `phase` method, which packages up phase center information
into a single dictionary to allow for consistent behavior between different
instantiations of `UVData` objects.
"""
cat_id = None
info_source = "user"
name_dict = {
pc_dict["cat_name"]: pc_id
for pc_id, pc_dict in self.phase_center_catalog.items()
}
if lookup_name:
if len(self._look_for_name(cat_name)) > 1:
raise ValueError(
"Name of object has multiple matches in phase center catalog. "
"Set lookup_name=False in order to continue."
)
# TODO I don't understand this comment:
# We only want to use the JPL-Horizons service if using a non-multi-phase-ctr
# instance of a UVData object.
if lookup_name and (cat_name not in name_dict):
if (cat_type is None) or (cat_type == "ephem"):
[
cat_times,
cat_lon,
cat_lat,
cat_dist,
cat_vrad,
] = uvutils.lookup_jplhorizons(
cat_name,
time_array,
telescope_loc=self.telescope_location_lat_lon_alt,
)
cat_type = "ephem"
cat_pm_ra = cat_pm_dec = None
cat_epoch = 2000.0
cat_frame = "icrs"
info_source = "jplh"
else:
raise ValueError(
f"Unable to find {cat_name} in among the existing sources "
"recorded in the catalog. Please supply source "
"information (e.g., RA and Dec coordinates) and "
"set lookup_name=False."
)
elif cat_name in name_dict:
# If the name of the source matches, then verify that all of its
# properties are the same as what is stored in phase_center_catalog.
if lookup_name:
cat_id = name_dict[cat_name]
cat_diffs = 0
else:
cat_id, cat_diffs = self._look_in_catalog(
cat_name,
cat_type=cat_type,
cat_lon=lon,
cat_lat=lat,
cat_frame=phase_frame,
cat_epoch=epoch,
cat_times=ephem_times,
cat_pm_ra=pm_ra,
cat_pm_dec=pm_dec,
cat_dist=dist,
cat_vrad=vrad,
)
# If cat_diffs > 0, it means that the catalog entries dont match
if cat_diffs != 0:
warnings.warn(
f"The entry name {cat_name} is not unique inside the phase "
"center catalog, adding anyways."
)
cat_type = "sidereal" if cat_type is None else cat_type
cat_lon = lon
cat_lat = lat
cat_frame = phase_frame
cat_epoch = epoch
cat_times = ephem_times
cat_pm_ra = pm_ra
cat_pm_dec = pm_dec
cat_dist = dist
cat_vrad = vrad
else:
temp_dict = self.phase_center_catalog[cat_id]
cat_type = temp_dict["cat_type"]
info_source = temp_dict["info_source"]
# Get here will return None if no key found, which we want
cat_lon = temp_dict.get("cat_lon")
cat_lat = temp_dict.get("cat_lat")
cat_frame = temp_dict.get("cat_frame")
cat_epoch = temp_dict.get("cat_epoch")
cat_times = temp_dict.get("cat_times")
cat_pm_ra = temp_dict.get("cat_pm_ra")
cat_pm_dec = temp_dict.get("cat_pm_dec")
cat_dist = temp_dict.get("cat_dist")
cat_vrad = temp_dict.get("cat_vrad")
else:
# The name of the source is unique!
cat_type = "sidereal" if cat_type is None else cat_type
cat_lon = lon
cat_lat = lat
cat_frame = phase_frame
cat_epoch = epoch
cat_times = ephem_times
cat_pm_ra = pm_ra
cat_pm_dec = pm_dec
cat_dist = dist
cat_vrad = vrad
if cat_epoch is None:
cat_epoch = 1950.0 if (cat_frame in ["fk4", "fk4noeterms"]) else 2000.0
if isinstance(cat_epoch, str) or isinstance(cat_epoch, Time):
cat_epoch = Time(cat_epoch).to_value(
"byear" if cat_frame in ["fk4", "fk4noeterms"] else "jyear"
)
# One last check - if we have an ephem phase center, lets make sure that the
# time range of the ephemeris encapsulates the entire range of time_array
check_ephem = False
if cat_type == "ephem":
# Take advantage of this to make sure that lat, lon, and times are all
# ndarray types
cat_lon = np.array(cat_lon, dtype=float)
cat_lat = np.array(cat_lat, dtype=float)
cat_times = np.array(cat_times, dtype=float)
cat_lon.shape += (1,) if (cat_lon.ndim == 0) else ()
cat_lat.shape += (1,) if (cat_lat.ndim == 0) else ()
cat_times.shape += (1,) if (cat_times.ndim == 0) else ()
check_ephem = np.min(time_array) < np.min(cat_times)
check_ephem = check_ephem or (np.max(time_array) > np.max(cat_times))
# If the ephem was supplied by JPL-Horizons, then we can easily expand
# it to cover the requested range.
if check_ephem and (info_source == "jplh"):
# Concat the two time ranges to make sure that we cover both the
# requested time range _and_ the original time range.
[
cat_times,
cat_lon,
cat_lat,
cat_dist,
cat_vrad,
] = uvutils.lookup_jplhorizons(
cat_name,
np.concatenate((np.reshape(time_array, -1), cat_times)),
telescope_loc=self.telescope_location_lat_lon_alt,
)
elif check_ephem:
# The ephem was user-supplied during the call to the phase method,
# raise an error to ask for more ephem data.
raise ValueError(
"Ephemeris data does not cover the entirety of the time range "
"attempted to be phased. Please supply additional ephem data "
"(and if used, set lookup_name=False)."
)
# Time to repackage everything into a dict
phase_dict = {
"cat_name": cat_name,
"cat_type": cat_type,
"cat_lon": cat_lon,
"cat_lat": cat_lat,
"cat_frame": cat_frame,
"cat_epoch": cat_epoch,
"cat_times": cat_times,
"cat_pm_ra": cat_pm_ra,
"cat_pm_dec": cat_pm_dec,
"cat_dist": cat_dist,
"cat_vrad": cat_vrad,
"info_source": info_source,
"cat_id": cat_id,
}
# Finally, make sure everything is a float or an ndarray of floats
for key in phase_dict.keys():
if isinstance(phase_dict[key], np.ndarray):
phase_dict[key] = phase_dict[key].astype(float)
elif (key == "cat_id") and (phase_dict[key] is not None):
# If this is the cat_id, make it an int
phase_dict[key] = int(phase_dict[key])
elif not ((phase_dict[key] is None) or isinstance(phase_dict[key], str)):
phase_dict[key] = float(phase_dict[key])
return phase_dict
def phase(
self,
lon=None,
lat=None,
epoch="J2000",
phase_frame="icrs",
ra=None,
dec=None,
cat_type=None,
ephem_times=None,
pm_ra=None,
pm_dec=None,
dist=None,
vrad=None,
cat_name=None,
lookup_name=False,
use_ant_pos=True,
allow_rephase=None,
orig_phase_frame=None,
select_mask=None,
cleanup_old_sources=True,
use_old_proj=False,
fix_old_proj=None,
):
"""
Phase data to a new direction, supports sidereal, ephemeris and driftscan types.
Can be used to phase all or a subset of the baseline-times. Types of phase
centers (`cat_type`) that are supported include:
- sidereal (fixed RA/Dec)
- ephem (RA/Dec that moves with time)
- driftscan (fixed az/el position)
See the phasing memo under docs/references for more documentation.
Tested against MWA_Tools/CONV2UVFITS/convutils.
Parameters
----------
lon : float
The longitude coordinate (e.g. RA or Azimuth) to phase to in radians.
lat : float
The latitude coordinate (e.g. Dec or Altitude) to phase to in radians.
epoch : astropy.time.Time object or str
The epoch to use for phasing. Either an astropy Time object or the
string "J2000" (which is the default).
Note that the epoch is only used to evaluate the ra & dec values,
if the epoch is not J2000, the ra & dec values are interpreted
as FK5 ra/dec values and transformed to J2000, the data are then
phased to the J2000 ra/dec values.
phase_frame : str
The astropy frame to phase to, any astropy-supported frame is allowed unless
use_old_proj is True, in which case it can only be 'icrs' or 'gcrs'.
'gcrs' accounts for precession & nutation,
'icrs' accounts for precession, nutation & abberation.
ra : float
An alias for `lon`.
dec : float
An alias for `lat`.
cat_type : str
Type of phase center to be added. Must be one of:
"sidereal" (fixed RA/Dec), "ephem" (RA/Dec that moves with time),
"driftscan" (fixed az/el position). Default is "sidereal".
ephem_times : ndarray of float
Only used when `cat_type="ephem"`. Describes the time for which the values
of `cat_lon` and `cat_lat` are caclulated, in units of JD. Shape is (Npts,).
pm_ra : float
Proper motion in RA, in units of mas/year. Only used for sidereal phase
centers.
pm_dec : float
Proper motion in Dec, in units of mas/year. Only used for sidereal phase
centers.
dist : float or ndarray of float
Distance of the source, in units of pc. Only used for sidereal and ephem
phase centers. Expected to be a float for sidereal phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
vrad : float or ndarray of float
Radial velocity of the source, in units of km/s. Only used for sidereal and
ephem phase centers. Expected to be a float for sidereal phase
centers, and an ndarray of floats of shape (Npts,) for ephem phase centers.
cat_name : str
Name of the phase center being phased to, required.
lookup_name : bool
Allows the user to lookup phase center infomation in `phase_center_catalog`
(for the entry matching `cat_name`). Setting this to `True` will ignore the
values supplied to the `ra`, `dec`, `epoch`, `phase_frame`, `pm_ra`,
`pm_dec`, `dist`, `vrad`.
use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions
rather than from the existing uvws.
allow_rephase : bool
Deprecated, will be removed in version 2.4. Only used if `use_old_proj` is
True. If this parameter is True, allow unphasing and rephasing if this
object is already phased.
orig_phase_frame : str
Deprecated, will be removed in version 2.4. The original phase frame of
this object (to use in unphasing). Only used if `use_old_proj` is True,
object is already phased, `allow_rephase` is True and the
phase_center_ra/dec of the object does not match `ra` and `dec`.
Defaults to using the "cat_frame" value in the `phase_center_catalog`.
select_mask : ndarray of bool
Optional mask for selecting which data to operate on along the blt-axis.
Shape is (Nblts,). Ignored if `use_old_proj` is True.
use_old_proj : bool
Deprecated, will be removed in version 2.4. If True, use the "old" method
for calculating baseline uvw-coordinates, which involved using astropy to
move antenna positions (in ITRF) into the requested reference frame
(either GCRS or ICRS). Default is False.
fix_old_proj : bool
Deprecated and has no effect. Detection of old phasing is really only
possible for uvh5 files, and those are now fixed on read by default. If the
object is known to have been phased with the old phasing, use the
`fix_phase` method to correct the phasing.
Raises
------
ValueError
If the `use_old_proj` is True, the object is already phased, and
allow_rephase is False.
"""
if fix_old_proj is not None:
warnings.warn(
"The `fix_old_proj` parameter is deprecated and will be removed in "
"version 2.4. If the object is known to have been phased with the old "
"phasing, use the `fix_phase` method to correct the phasing.",
DeprecationWarning,
)
if cat_name is None:
warnings.warn(
"No cat_name provided, this will become an error in pyuvdata v2.4. "
"Defaulting to naming based on lat/lon.",
DeprecationWarning,
)
# If you are a new phase style data set, there's no valid reason to be going
# back to the old phase method. Time to bail!
compatible, reason = self._old_phase_attributes_compatible()
if use_old_proj and not compatible:
raise ValueError(
"Objects with " + reason + " are not compatible with the old phasing "
"method, please set use_old_proj=False."
)
if cat_type != "unprojected":
if lon is None:
if ra is None:
raise ValueError(
"lon parameter must be set if cat_type is not 'unprojected'"
)
else:
lon = ra
if lat is None:
if dec is None:
raise ValueError(
"lat parameter must be set if cat_type is not 'unprojected'"
)
else:
lat = dec
if cat_name is None:
cat_name = (
"lon_"
+ str(np.rad2deg(lon))
+ "deg_lat_"
+ str(np.rad2deg(lat))
+ "_deg"
)
# Before moving forward with the heavy calculations, we need to do some
# basic housekeeping to make sure that we've got the coordinate data that
# we need in order to proceed.
phase_dict = self._phase_dict_helper(
lon,
lat,
epoch,
phase_frame,
ephem_times,
cat_type,
pm_ra,
pm_dec,
dist,
vrad,
cat_name,
lookup_name,
self.time_array,
)
if phase_dict["cat_type"] not in ["ephem", "unprojected"]:
if np.array(lon).size > 1:
raise ValueError(
"lon parameter must be a single value for cat_type "
f"{phase_dict['cat_type']}"
)
if np.array(lat).size > 1:
raise ValueError(
"lat parameter must be a single value for cat_type "
f"{phase_dict['cat_type']}"
)
# Right up front, we're gonna split off the piece of the code that
# does the phasing using the "new" method, since its a lot more flexible
# and because I think at some point, everything outside of this loop
# can be deprecated
if not use_old_proj:
# Grab all the meta-data we need for the rotations
time_array = self.time_array
lst_array = self.lst_array
uvw_array = self.uvw_array
ant_1_array = self.ant_1_array
ant_2_array = self.ant_2_array
old_w_vals = self.uvw_array[:, 2].copy()
old_w_vals[self._check_for_cat_type("unprojected")] = 0.0
old_app_ra = self.phase_center_app_ra
old_app_dec = self.phase_center_app_dec
old_frame_pa = self.phase_center_frame_pa
# Check and see if we have any unprojected objects, in which case
# their w-values should be zeroed out.
if select_mask is not None:
if len(select_mask) != self.Nblts:
raise IndexError("Selection mask must be of length Nblts.")
if not isinstance(select_mask[0], (bool, np.bool_)):
raise ValueError("Selection mask must be a boolean array")
time_array = time_array[select_mask]
lst_array = lst_array[select_mask]
uvw_array = uvw_array[select_mask, :]
ant_1_array = ant_1_array[select_mask]
ant_2_array = ant_2_array[select_mask]
if isinstance(old_w_vals, np.ndarray):
old_w_vals = old_w_vals[select_mask]
# We got the meta-data, now handle calculating the apparent coordinates.
# First, check if we need to look up the phase center in question
new_app_ra, new_app_dec = uvutils.calc_app_coords(
phase_dict["cat_lon"],
phase_dict["cat_lat"],
coord_frame=phase_dict["cat_frame"],
coord_epoch=phase_dict["cat_epoch"],
coord_times=phase_dict["cat_times"],
coord_type=phase_dict["cat_type"],
time_array=time_array,
lst_array=lst_array,
pm_ra=phase_dict["cat_pm_ra"],
pm_dec=phase_dict["cat_pm_dec"],
vrad=phase_dict["cat_vrad"],
dist=phase_dict["cat_dist"],
telescope_loc=self.telescope_location_lat_lon_alt,
)
# Now calculate position angles.
if not phase_frame == "altaz":
new_frame_pa = uvutils.calc_frame_pos_angle(
time_array,
new_app_ra,
new_app_dec,
self.telescope_location_lat_lon_alt,
phase_frame,
ref_epoch=epoch,
)
else:
new_frame_pa = np.zeros(time_array.shape, dtype=float)
# Now its time to do some rotations and calculate the new coordinates
new_uvw = uvutils.calc_uvw(
app_ra=new_app_ra,
app_dec=new_app_dec,
frame_pa=new_frame_pa,
lst_array=lst_array,
use_ant_pos=use_ant_pos,
uvw_array=uvw_array,
antenna_positions=self.antenna_positions,
antenna_numbers=self.antenna_numbers,
ant_1_array=ant_1_array,
ant_2_array=ant_2_array,
old_app_ra=old_app_ra,
old_app_dec=old_app_dec,
old_frame_pa=old_frame_pa,
telescope_lat=self.telescope_location_lat_lon_alt[0],
telescope_lon=self.telescope_location_lat_lon_alt[1],
)
# With all operations complete, we now start manipulating the UVData object
cat_id = self._add_phase_center(
phase_dict["cat_name"],
phase_dict["cat_type"],
cat_lon=phase_dict["cat_lon"],
cat_lat=phase_dict["cat_lat"],
cat_frame=phase_dict["cat_frame"],
cat_epoch=phase_dict["cat_epoch"],
cat_times=phase_dict["cat_times"],
cat_pm_ra=phase_dict["cat_pm_ra"],
cat_pm_dec=phase_dict["cat_pm_dec"],
cat_dist=phase_dict["cat_dist"],
cat_vrad=phase_dict["cat_vrad"],
info_source=phase_dict["info_source"],
cat_id=phase_dict["cat_id"],
force_update=True,
)
# Now its time to update the raw data. This will return empty if
# metadata_only is set to True.
new_w_vals = 0.0 if (cat_type == "unprojected") else new_uvw[:, 2]
self._apply_w_proj(new_w_vals, old_w_vals, select_mask=select_mask)
# Finally, we now take it upon ourselves to update some metadata. What we
# do here will depend a little bit on whether or not we have a selection
# mask active, since most everything is affected by that.
if select_mask is not None:
self.uvw_array[select_mask] = new_uvw
self.phase_center_app_ra[select_mask] = new_app_ra
self.phase_center_app_dec[select_mask] = new_app_dec
self.phase_center_frame_pa[select_mask] = new_frame_pa
self.phase_center_id_array[select_mask] = cat_id
else:
self.uvw_array = new_uvw
self.phase_center_app_ra = new_app_ra
self.phase_center_app_dec = new_app_dec
self.phase_center_frame_pa = new_frame_pa
self.phase_center_id_array[:] = cat_id
# If not multi phase center, make sure to update the ra/dec values, since
# otherwise we'll have no record of source properties.
if cleanup_old_sources:
self._clear_unused_phase_centers()
# All done w/ the new phase method
return
# use the old method
warnings.warn(
"The `use_old_proj` option is deprecated and will be removed in "
"version 2.4. Note that the old and new phase methods are NOT compatible "
"with one another, so if you have phased using the old method, you should "
"call the `fix_phase` method to undo the old corrections before using the "
"new version of the phase method.",
DeprecationWarning,
)
if allow_rephase is not None:
warnings.warn(
"The `allow_rephase` option is deprecated, and will be removed in "
"version 2.4. It is only used if `use_old_proj` is True.",
DeprecationWarning,
)
else:
# default it to True if it is not set
allow_rephase = True
if orig_phase_frame is not None:
warnings.warn(
"The `allow_rephase` option is deprecated, and will be removed in "
"version 2.4. It is only used if `use_old_proj` is True.",
DeprecationWarning,
)
if phase_dict["cat_type"] != "sidereal":
raise ValueError(
"Old phasing only supports sidereal catalog types, "
"please change the cat_type or use_old_proj=False."
)
if phase_dict["cat_frame"] not in ["icrs", "gcrs"]:
raise ValueError(
"Old phasing only supports 'icrs' and 'gcrs' phase_frames. Either use "
"one of those frames or set use_old_proj=False."
)
ra = lon
dec = lat
# To get here there can only be one phase center
old_phase_dict = list(self.phase_center_catalog.values())[0]
if old_phase_dict["cat_type"] != "unprojected":
if not allow_rephase:
raise ValueError(
"Visibilities are already phased; set allow_rephase to True to "
"unphase and rephase."
)
# To get to this point, allow_rephase has to be true
if old_phase_dict["cat_lon"] != lon or old_phase_dict["cat_lat"] != lat:
self.unproject_phase(
phase_frame=orig_phase_frame,
use_ant_pos=use_ant_pos,
use_old_proj=True,
)
if epoch == "J2000" or epoch == 2000:
icrs_coord = SkyCoord(ra=ra, dec=dec, unit="radian", frame="icrs")
else:
assert isinstance(epoch, Time)
phase_center_coord = SkyCoord(
ra=ra, dec=dec, unit="radian", equinox=epoch, frame=FK5
)
# convert to icrs (i.e. J2000) to write to object
icrs_coord = phase_center_coord.transform_to("icrs")
# This feels yucky, but it is what was done in old phasing because we defined
# ra/dec as being in ICRS
phase_dict["cat_lon"] = icrs_coord.ra.radian
phase_dict["cat_lat"] = icrs_coord.dec.radian
phase_dict["cat_epoch"] = 2000.0
cat_id = self._add_phase_center(
cat_name,
cat_type="sidereal",
cat_lon=phase_dict["cat_lon"],
cat_lat=phase_dict["cat_lat"],
cat_frame=phase_dict["cat_frame"],
cat_epoch=phase_dict["cat_epoch"],
)
self.phase_center_id_array = np.zeros(self.Nblts, dtype=int) + cat_id
self._clear_unused_phase_centers()
self.phase_center_app_ra = None
self.phase_center_app_dec = None
self.phase_center_frame_pa = None
if phase_dict["cat_frame"] == "icrs":
frame_phase_center = icrs_coord
else:
# use center of observation for obstime for gcrs
center_time = np.mean([np.max(self.time_array), np.min(self.time_array)])
icrs_coord.obstime = Time(center_time, format="jd")
frame_phase_center = icrs_coord.transform_to("gcrs")
# This promotion is REQUIRED to get the right answer when we
# add in the telescope location for ICRS
self.uvw_array = np.float64(self.uvw_array)
unique_times, unique_inds = np.unique(self.time_array, return_index=True)
telescope_location = EarthLocation.from_geocentric(
*self.telescope_location, unit=units.m
)
obs_times = Time(unique_times, format="jd")
itrs_telescope_locations = telescope_location.get_itrs(obstime=obs_times)
itrs_telescope_locations = SkyCoord(itrs_telescope_locations)
# just calling transform_to(coord.GCRS) will delete the obstime information
# need to re-add obstimes for a GCRS transformation
if phase_dict["cat_frame"] == "gcrs":
frame_telescope_locations = itrs_telescope_locations.transform_to(
getattr(coord, f"{phase_dict['cat_frame']}".upper())(obstime=obs_times)
)
else:
frame_telescope_locations = itrs_telescope_locations.transform_to(
getattr(coord, f"{phase_dict['cat_frame']}".upper())
)
# set the representation_type to cartensian to get xyz later
frame_telescope_locations.representation_type = "cartesian"
for ind, jd in enumerate(unique_times):
inds = np.where(self.time_array == jd)[0]
obs_time = obs_times[ind]
itrs_lat_lon_alt = self.telescope_location_lat_lon_alt
frame_telescope_location = frame_telescope_locations[ind]
if use_ant_pos:
# This promotion is REQUIRED to get the right answer when we
# add in the telescope location for ICRS
ecef_ant_pos = (
np.float64(self.antenna_positions) + self.telescope_location
)
itrs_ant_coord = SkyCoord(
x=ecef_ant_pos[:, 0] * units.m,
y=ecef_ant_pos[:, 1] * units.m,
z=ecef_ant_pos[:, 2] * units.m,
frame="itrs",
obstime=obs_time,
)
frame_ant_coord = itrs_ant_coord.transform_to(phase_dict["cat_frame"])
frame_ant_rel = (
(frame_ant_coord.cartesian - frame_telescope_location.cartesian)
.get_xyz()
.T.value
)
frame_ant_uvw = uvutils.phase_uvw(
frame_phase_center.ra.rad, frame_phase_center.dec.rad, frame_ant_rel
)
# instead of looping through every ind, find the spot in antenna number
# array where ant_num <= ant1 < ant_number and similarly for ant2
# for all baselines in inds
# then find the uvw coordinate for all at the same time
# antenna_numbers does not necessarily need to be in order on the object
# but needs to be in order for the searchsorted to work.
# ant1_index and ant2_index arrays will preserve the order of blts
ant_sort = np.argsort(self.antenna_numbers)
ant1_index = np.searchsorted(
self.antenna_numbers[ant_sort], self.ant_1_array[inds]
)
ant2_index = np.searchsorted(
self.antenna_numbers[ant_sort], self.ant_2_array[inds]
)
self.uvw_array[inds] = (
frame_ant_uvw[ant_sort][ant2_index, :]
- frame_ant_uvw[ant_sort][ant1_index, :]
)
else:
# Also, uvws should be thought of like ENU, not ECEF (or rotated ECEF)
# convert them to ECEF to transform between frames
uvws_use = self.uvw_array[inds, :]
uvw_ecef = uvutils.ECEF_from_ENU(uvws_use, *itrs_lat_lon_alt)
itrs_uvw_coord = SkyCoord(
x=uvw_ecef[:, 0] * units.m,
y=uvw_ecef[:, 1] * units.m,
z=uvw_ecef[:, 2] * units.m,
frame="itrs",
obstime=obs_time,
)
frame_uvw_coord = itrs_uvw_coord.transform_to(phase_dict["cat_frame"])
# this takes out the telescope location in the new frame,
# so these are vectors again
frame_rel_uvw = (
frame_uvw_coord.cartesian.get_xyz().value.T
- frame_telescope_location.cartesian.get_xyz().value
)
self.uvw_array[inds, :] = uvutils.phase_uvw(
frame_phase_center.ra.rad, frame_phase_center.dec.rad, frame_rel_uvw
)
# calculate data and apply phasor
if not self.metadata_only:
w_lambda = (
self.uvw_array[:, 2].reshape(self.Nblts, 1)
/ const.c.to("m/s").value
* self.freq_array.reshape(1, self.Nfreqs)
)
if self.future_array_shapes:
phs = np.exp(-1j * 2 * np.pi * w_lambda[:, :, None])
else:
phs = np.exp(-1j * 2 * np.pi * w_lambda[:, None, :, None])
self.data_array *= phs
# make sure apparent coords are properly set.
self._set_app_coords_helper()
def phase_to_time(
self,
time,
phase_frame="icrs",
use_ant_pos=True,
use_old_proj=False,
allow_rephase=True,
orig_phase_frame=None,
select_mask=None,
):
"""
Phase to the ra/dec of zenith at a particular time.
See the phasing memo under docs/references for more documentation.
Parameters
----------
time : astropy.time.Time object or float
The time to phase to, an astropy Time object or a float Julian Date
phase_frame : str
The astropy frame to phase to, any astropy-supported frame is allowed unless
use_old_proj is True, in which case it can only be 'icrs' or 'gcrs'.
'gcrs' accounts for precession & nutation,
'icrs' accounts for precession, nutation & abberation.
use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions
rather than from the existing uvws.
use_old_proj : bool
Deprecated, will be removed in version 2.4. If True, use the "old" method
for calculating baseline uvw-coordinates, which involved using astropy to
move antenna positions (in ITRF) into the requested reference frame
(either GCRS or ICRS). Default is False.
allow_rephase : bool
Deprecated, will be removed in version 2.4. Only used if `use_old_proj` is
True. If this parameter is True, allow unphasing and rephasing if this
object is already phased.
orig_phase_frame : str
Deprecated, will be removed in version 2.4. The original phase frame of
this object (to use in unphasing). Only used if `use_old_proj` is True,
object is already phased, `allow_rephase` is True and the
phase_center_ra/dec of the object does not match `ra` and `dec`.
Defaults to using the "cat_frame" value in the `phase_center_catalog`.
select_mask : array_like
Selection mask for which data should be rephased. Any array-like able to be
used as an index is suitable -- the most typical is an array of bool with
length `Nblts`, or an array of ints within the range (-Nblts, Nblts).
Raises
------
ValueError
If the `use_old_proj` is True, the object is already phased, and
allow_rephase is False.
TypeError
If time is not an astropy.time.Time object or Julian Date as a float
"""
if isinstance(time, (float, np.floating)):
time = Time(time, format="jd")
if not isinstance(time, Time):
raise TypeError("time must be an astropy.time.Time object or a float")
# Generate ra/dec of zenith at time in the phase_frame coordinate
# system to use for phasing
telescope_location = EarthLocation.from_geocentric(
*self.telescope_location, unit="m"
)
zenith_coord = SkyCoord(
alt=Angle(90 * units.deg),
az=Angle(0 * units.deg),
obstime=time,
frame="altaz",
location=telescope_location,
)
obs_zenith_coord = zenith_coord.transform_to(phase_frame)
zenith_ra = obs_zenith_coord.ra.rad
zenith_dec = obs_zenith_coord.dec.rad
self.phase(
zenith_ra,
zenith_dec,
epoch="J2000",
phase_frame=phase_frame,
use_ant_pos=use_ant_pos,
use_old_proj=use_old_proj,
allow_rephase=allow_rephase,
orig_phase_frame=orig_phase_frame,
select_mask=select_mask,
cat_name=("zenith_at_jd%f" % time.jd),
)
def set_uvws_from_antenna_positions(
self,
update_vis=True,
allow_phasing=False,
require_phasing=None,
orig_phase_frame=None,
output_phase_frame="icrs",
use_old_proj=False,
):
"""
Calculate UVWs based on antenna_positions.
Parameters
----------
update_vis : bool
Option to update visibilities when recalculating uvws to correct for
phasing errors from using the `old` method. This should only be set to False
in limited circumstances (e.g., when certain metadata like exact times are
not trusted), as misuse can significantly corrupt data. Ignored if
`use_old_proj` is True.
allow_phasing : bool
Option for phased data, only applies if `use_old_proj` is True. If data is
phased and allow_phasing=True, UVWs will be calculated and the visibilities
will be rephased. Default is False. This option is deprecated and will be
removed in version 2.4 with `use_old_proj`.
require_phasing : bool
Deprecated. Use `update_vis` instead. This parameter will be removed in
version 2.4
orig_phase_frame : str
The astropy frame to phase from. Either 'icrs' or 'gcrs'.
Defaults to using the phase_center_catalog["cat_frame"]. Only used if
allow_phasing is True and use_old_proj is True. This option is deprecated
and will be removed in version 2.4 with `use_old_proj`.
output_phase_frame : str
The astropy frame to phase to. Either 'icrs' or 'gcrs'. Only used if
allow_phasing is True, and use_old_proj is True. This option is deprecated
and will be removed in version 2.4 with `use_old_proj`.
use_old_proj : bool
If set to True, uses the 'old' method of calculating baseline vectors.
Default is False, which will instead use the 'new' method. This option is
deprecated and will be removed in version 2.4 after which only the 'new'
method will be supported (because it is more accurate).
Raises
------
ValueError
If use_old_proj is True, all data is phased and allow_phasing is False.
"""
if require_phasing is not None:
warnings.warn(
"The `require_phasing` option is deprecated, and will be removed in "
"pyuvdata v2.4. Use `update_vis` instead",
DeprecationWarning,
)
update_vis = require_phasing
if not use_old_proj:
telescope_location = self.telescope_location_lat_lon_alt
unprojected_blts = self._check_for_cat_type("unprojected")
new_uvw = uvutils.calc_uvw(
app_ra=self.phase_center_app_ra,
app_dec=self.phase_center_app_dec,
frame_pa=self.phase_center_frame_pa,
lst_array=self.lst_array,
use_ant_pos=True,
antenna_positions=self.antenna_positions,
antenna_numbers=self.antenna_numbers,
ant_1_array=self.ant_1_array,
ant_2_array=self.ant_2_array,
telescope_lat=telescope_location[0],
telescope_lon=telescope_location[1],
)
if np.any(~unprojected_blts):
# At least some are phased
if update_vis:
old_w_vals = self.uvw_array[:, 2].copy()
old_w_vals[unprojected_blts] = 0.0
self._apply_w_proj(new_uvw[:, 2], old_w_vals)
else:
warnings.warn(
"Recalculating uvw_array without adjusting visibility "
"phases -- this can introduce significant errors if used "
"incorrectly."
)
# If the data are phased, we've already adjusted the phases. Now we just
# need to update the uvw's and we are home free.
self.uvw_array = new_uvw
return
# datasets phased with new phasing should not use the "old" method
compatible, reason = self._old_phase_attributes_compatible()
if use_old_proj and not compatible:
raise ValueError(
"Objects with " + reason + " are not compatible with the old phasing "
"method, please set use_old_proj=False."
)
# if we got here, there's only one phase center
phase_dict = list(self.phase_center_catalog.values())[0]
# use the old method
warnings.warn(
"The `use_old_proj` option is deprecated, and will be removed in "
"pyuvdata 2.4. Note that the old and new phase methods are NOT compatible "
"with one another, so if you have phased using the old method, you should "
"call the `fix_phase` method or use the unproject_phase method with "
"use_old_proj=True to undo the old corrections before using the new "
"version of the phase method.",
DeprecationWarning,
)
if phase_dict["cat_type"] != "unprojected":
if allow_phasing:
if not self.metadata_only:
warnings.warn(
"Data will be unphased and rephased "
"to calculate UVWs, which might introduce small "
"inaccuracies to the data."
)
if orig_phase_frame not in [None, "icrs", "gcrs"]:
raise ValueError(
"Invalid parameter orig_phase_frame. "
'Options are "icrs", "gcrs", or None.'
)
if output_phase_frame not in ["icrs", "gcrs"]:
raise ValueError(
"Invalid parameter output_phase_frame. "
'Options are "icrs" or "gcrs".'
)
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore", "The `use_old_proj` option is deprecated"
)
self.unproject_phase(
phase_frame=orig_phase_frame, use_old_proj=True
)
else:
raise ValueError(
"UVW calculation with `use_old_proj` requires unprojected data. "
"Use `unproject_phase` or set allow_phasing=True."
)
antenna_locs_ENU, _ = self.get_ENU_antpos(center=False)
# this code used to loop through every bl in the unique,
# find the index into self.antenna_array of ant1 and ant2
# and fill out the self.uvw_array for all matching bls.
# instead, find the indices and reverse inds from the unique,
# create the unique ant1 and ant2 arrays
# use searchsorted to find the index of the antenna numbers into ant1 and ant2
# create the unique uvw array then broadcast to self.uvw_array
bls, unique_inds, reverse_inds = np.unique(
self.baseline_array, return_index=True, return_inverse=True
)
# antenna_numbers does not necessarily need to be in order on the object
# but needs to be in order for the searchsorted to work.
# ant1_index and ant2_index arrays will preserve the order of blts
ant_sort = np.argsort(self.antenna_numbers)
ant1_index = np.searchsorted(
self.antenna_numbers[ant_sort], self.ant_1_array[unique_inds]
)
ant2_index = np.searchsorted(
self.antenna_numbers[ant_sort], self.ant_2_array[unique_inds]
)
_uvw_array = np.zeros((bls.size, 3))
_uvw_array = (
antenna_locs_ENU[ant_sort][ant2_index, :]
- antenna_locs_ENU[ant_sort][ant1_index, :]
)
self.uvw_array = _uvw_array[reverse_inds]
if phase_dict["cat_type"] != "unprojected":
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore", "The `use_old_proj` option is deprecated"
)
self.phase(
lon=phase_dict["cat_lon"],
lat=phase_dict["cat_lat"],
epoch=phase_dict["cat_epoch"],
phase_frame=phase_dict["cat_frame"],
cat_name=phase_dict["cat_name"],
use_old_proj=True,
)
def fix_phase(self, use_ant_pos=True):
"""
Fix the data to be consistent with the new phasing method.
This is a simple utility function for updating UVW coordinates calculated using
the 'old' phasing algorithm with those calculated by the 'new' algorithm. Note
that this step is required for using the new methods with data phased using the
`phase` methiod prior to pyuvdata v2.2.
Parameters
----------
use_ant_pos : bool
Use the antenna positions for determining UVW coordinates. Default is True.
"""
# datasets phased with new phasing should not use the this method
compatible, reason = self._old_phase_attributes_compatible()
if not compatible:
raise ValueError(
"Objects with " + reason + " were not phased with the old "
"method, so no fixing is required."
)
# If we are just using the antenna positions, we don't actually need to do
# anything, since the new baseline vectors will be unaffected by the prior
# phasing method, and the delta_w values get correctly corrected for in
# set_uvws_from_antenna_positions.
if use_ant_pos:
warnings.warn("Fixing phases using antenna positions.")
self.set_uvws_from_antenna_positions()
else:
warnings.warn(
"Attempting to fix residual phasing errors from the old `phase` method "
"using the `unproject_phase` with use_old_proj set to True. This "
"can result in closure errors if the data were not actually phased "
"using the old method -- caution is advised."
)
# Record the old values
# if we get here, we can only have one phase center
phase_dict = list(self.phase_center_catalog.values())[0]
# Bring the UVWs back to ENU/unprojected
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore", "The `use_old_proj` option is deprecated"
)
warnings.filterwarnings(
"ignore", "The `phase_frame` parameter is deprecated"
)
self.unproject_phase(
phase_frame=phase_dict["cat_frame"],
use_ant_pos=False,
use_old_proj=True,
)
# Check for any autos, since their uvws get potentially corrupted
# by the above operation
auto_mask = self.ant_1_array == self.ant_2_array
if any(auto_mask):
self.uvw_array[auto_mask, :] = 0.0
# And rephase the data using the new algorithm
self.phase(
phase_dict["cat_lon"],
phase_dict["cat_lat"],
phase_frame=phase_dict["cat_frame"],
epoch=phase_dict["cat_epoch"],
cat_name=phase_dict["cat_name"],
use_ant_pos=False,
)
def _deprecated_phase_helper(
self,
other,
phase_center_radec,
unphase_to_drift,
phase_frame,
orig_phase_frame,
use_ant_pos,
make_multi_phase,
):
"""
Handle deprecation warnings and phasing for deprecated parameters.
This is a helper method for `__add__`, `fast_concat` and `read` that should be
removed in version 2.4 when all the phase-related deprecated parameters are
removed.
Parameters
----------
other : UVData object or list of UVData objects
UVData object or list of UVData objects. Can be an empty list.
phase_center_radec : array_like of float
The phase center to phase the objects to in radians.
unphase_to_drift : bool
Unproject the objects before combining them.
phase_frame : str
The astropy frame to phase to. Only used if `phase_center_radec` is set.
orig_phase_frame : str
Has no effect.
use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions rather than
from the existing uvws.
make_multi_phase : bool
Has no effect.
"""
if unphase_to_drift is not None:
warnings.warn(
"The `unphase_to_drift` option is deprecated, and will be removed in "
"pyuvdata v2.4. instead call `unproject_phase` on the combined object "
"after this method.",
DeprecationWarning,
)
if unphase_to_drift is None:
unphase_to_drift = False
if phase_frame is not None:
warnings.warn(
"The `phase_frame` parameter is deprecated, and will be removed in "
"pyuvdata v2.4 along with the `phase_center_radec` parameter.",
DeprecationWarning,
)
if use_ant_pos is not None:
warnings.warn(
"The `use_ant_pos` option is deprecated, and will be removed in "
"pyuvdata v2.4 along with the `phase_center_radec` parameter.",
DeprecationWarning,
)
else:
use_ant_pos = True
if phase_center_radec is not None:
warnings.warn(
"The `phase_center_radec` parameter is deprecated, and will be removed "
"in pyuvdata v2.4. instead call `phase` on the combined object "
"after this method.",
DeprecationWarning,
)
if phase_frame is None:
phase_frame = "icrs"
if orig_phase_frame is not None:
warnings.warn(
"The `orig_phase_frame` option is deprecated and has no effect. It "
"will be removed in pyuvdata v2.4.",
DeprecationWarning,
)
if make_multi_phase is not None:
warnings.warn(
"The `make_multi_phase` option is deprecated and has no effect. It "
"will be removed in pyuvdata v2.4.",
DeprecationWarning,
)
if phase_center_radec is not None and unphase_to_drift:
raise ValueError(
"phase_center_radec cannot be set if unphase_to_drift is True."
)
if unphase_to_drift:
if not isinstance(other, (list, tuple, np.ndarray)):
# if this is a UVData object already, stick it in a list
other = [other]
if np.any(~self._check_for_cat_type("unprojected")):
warnings.warn("Unprojecting this UVData object.")
self.unproject_phase(use_ant_pos=use_ant_pos)
for obj in other:
if np.any(~obj._check_for_cat_type("unprojected")):
warnings.warn("Unprojecting other UVData object.")
obj.unproject_phase(use_ant_pos=use_ant_pos)
if phase_center_radec is not None:
if np.array(phase_center_radec).size != 2:
raise ValueError("phase_center_radec should have length 2.")
if not isinstance(other, (list, tuple, np.ndarray)):
# if this is a UVData object already, stick it in a list
other = [other]
# If this object is not phased or has multiple phase centers or one that
# is not phased close to phase_center_radec, (re)phase it.
# Close is defined using the phase_center_ra/dec tolerances.
obj_names = ["this"] + ["other"] * len(other)
for obj_ind, obj in enumerate([self] + other):
rephase = False
if obj.Nphase > 1 or np.any(obj._check_for_cat_type("unprojected")):
rephase = True
else:
# this only has one phase center and is projected
phase_dict = list(obj.phase_center_catalog.values())[0]
if phase_dict["cat_type"] != "sidereal":
rephase = True
elif phase_dict["cat_frame"] not in ["icrs", "gcrs"]:
rephase = True
elif phase_dict["cat_lon"] != phase_center_radec[0]:
rephase = True
elif phase_dict["cat_lat"] != phase_center_radec[1]:
rephase = True
obj_name = obj_names[obj_ind]
if rephase:
warnings.warn(
f"Phasing {obj_name} UVData object to {phase_center_radec}"
)
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore", "No cat_name provided, this will become an error"
)
obj.phase(
phase_center_radec[0],
phase_center_radec[1],
phase_frame=phase_frame,
use_ant_pos=use_ant_pos,
allow_rephase=True,
)
def __add__(
self,
other,
inplace=False,
phase_center_radec=None,
unphase_to_drift=None,
phase_frame=None,
orig_phase_frame=None,
use_ant_pos=None,
verbose_history=False,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
make_multi_phase=None,
ignore_name=False,
):
"""
Combine two UVData objects along frequency, polarization and/or baseline-time.
Parameters
----------
other : UVData object
Another UVData object which will be added to self.
inplace : bool
If True, overwrite self as we go, otherwise create a third object
as the sum of the two.
phase_center_radec : array_like of float
Deprecated and will be removed in version 2.4, instead call `phase` on the
combined object after this method. The phase center to phase the files to
before adding the objects in radians.
unphase_to_drift : bool
Deprecated, instead call `unproject_phase` on the combined object after
this method. Default is False.
phase_frame : str
The astropy frame to phase to. Only used if `phase_center_radec` is set.
Default is "icrs".
Deprecated and will be removed in version 2.4 with `phase_center_radec`.
orig_phase_frame : str
Deprecated, has no effect. Will be removed in version 2.4.
use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions rather than
from the existing uvws. Only used if `unphase_to_drift` or
`phase_center_radec` are set. Deprecated and will be removed in version 2.4
with `unphase_to_drift` and `phase_center_radec`. Default is True.
verbose_history : bool
Option to allow more verbose history. If True and if the histories for the
two objects are different, the combined object will keep all the history of
both input objects (if many objects are combined in succession this can
lead to very long histories). If False and if the histories for the two
objects are different, the combined object will have the history of the
first object and only the parts of the second object history that are unique
(this is done word by word and can result in hard to interpret histories).
run_check : bool
Option to check for the existence and proper shapes of parameters
after combining objects.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
combining objects.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
make_multi_phase : bool
Deprecated and has no effect. Will be removed in version 2.4.
ignore_name : bool
Option to ignore the name of the phase center (`cat_name` in
`phase_center_catalog`) when combining two UVData objects. If set to True,
phase centers that are the same up to their name will be combined with the
name set to the name found in the first UVData object in the sum. If set to
False, phase centers that are the same up to the name will be kept as
separate phase centers. Default is False.
Raises
------
ValueError
If other is not a UVData object, self and other are not compatible
or if data in self and other overlap.
If `phase_center_radec` is not None and is not length 2.
"""
if inplace:
this = self
else:
this = self.copy()
# Check that both objects are UVData and valid
this.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
if not issubclass(other.__class__, this.__class__):
if not issubclass(this.__class__, other.__class__):
raise ValueError(
"Only UVData (or subclass) objects can be "
"added to a UVData (or subclass) object"
)
other.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
# Check to make sure that both objects are consistent w/ use of flex_spw
if this.flex_spw != other.flex_spw:
raise ValueError(
"To combine these data, flex_spw must be set to the same "
"value (True or False) for both objects."
)
# check that both objects have the same array shapes
if this.future_array_shapes != other.future_array_shapes:
raise ValueError(
"Both objects must have the same `future_array_shapes` parameter. "
"Use the `use_future_array_shapes` or `use_current_array_shapes` "
"methods to convert them."
)
this._deprecated_phase_helper(
other,
phase_center_radec,
unphase_to_drift,
phase_frame,
orig_phase_frame,
use_ant_pos,
make_multi_phase,
)
# Define parameters that must be the same to add objects
compatibility_params = [
"_vis_units",
"_telescope_name",
"_instrument",
"_telescope_location",
"_Nants_telescope",
"_antenna_names",
"_antenna_numbers",
"_antenna_positions",
]
if not this.future_array_shapes and not this.flex_spw:
compatibility_params.append("_channel_width")
# Build up history string
history_update_string = " Combined data along "
n_axes = 0
# Create blt arrays for convenience
prec_t = -2 * np.floor(np.log10(this._time_array.tols[-1])).astype(int)
prec_b = 8
this_blts = np.array(
[
"_".join(
["{1:.{0}f}".format(prec_t, blt[0]), str(blt[1]).zfill(prec_b)]
)
for blt in zip(this.time_array, this.baseline_array)
]
)
other_blts = np.array(
[
"_".join(
["{1:.{0}f}".format(prec_t, blt[0]), str(blt[1]).zfill(prec_b)]
)
for blt in zip(other.time_array, other.baseline_array)
]
)
# Check we don't have overlapping data
both_pol, this_pol_ind, other_pol_ind = np.intersect1d(
this.polarization_array, other.polarization_array, return_indices=True
)
# If we have a flexible spectral window, the handling here becomes a bit funky,
# because we are allowed to have channels with the same frequency *if* they
# belong to different spectral windows (one real-life example: you might want
# to preserve guard bands in the correlator, which can have overlaping RF
# frequency channels)
if this.flex_spw:
this_freq_ind = np.array([], dtype=np.int64)
other_freq_ind = np.array([], dtype=np.int64)
both_freq = np.array([], dtype=float)
both_spw = np.intersect1d(this.spw_array, other.spw_array)
for idx in both_spw:
this_mask = np.where(this.flex_spw_id_array == idx)[0]
other_mask = np.where(other.flex_spw_id_array == idx)[0]
if this.future_array_shapes:
both_spw_freq, this_spw_ind, other_spw_ind = np.intersect1d(
this.freq_array[this_mask],
other.freq_array[other_mask],
return_indices=True,
)
else:
both_spw_freq, this_spw_ind, other_spw_ind = np.intersect1d(
this.freq_array[0, this_mask],
other.freq_array[0, other_mask],
return_indices=True,
)
this_freq_ind = np.append(this_freq_ind, this_mask[this_spw_ind])
other_freq_ind = np.append(other_freq_ind, other_mask[other_spw_ind])
both_freq = np.append(both_freq, both_spw_freq)
else:
if this.future_array_shapes:
both_freq, this_freq_ind, other_freq_ind = np.intersect1d(
this.freq_array, other.freq_array, return_indices=True
)
else:
both_freq, this_freq_ind, other_freq_ind = np.intersect1d(
this.freq_array[0, :], other.freq_array[0, :], return_indices=True
)
both_blts, this_blts_ind, other_blts_ind = np.intersect1d(
this_blts, other_blts, return_indices=True
)
if not self.metadata_only and (
len(both_pol) > 0 and len(both_freq) > 0 and len(both_blts) > 0
):
# check that overlapping data is not valid
if this.future_array_shapes:
this_inds = np.ravel_multi_index(
(
this_blts_ind[:, np.newaxis, np.newaxis],
this_freq_ind[np.newaxis, :, np.newaxis],
this_pol_ind[np.newaxis, np.newaxis, :],
),
this.data_array.shape,
).flatten()
other_inds = np.ravel_multi_index(
(
other_blts_ind[:, np.newaxis, np.newaxis],
other_freq_ind[np.newaxis, :, np.newaxis],
other_pol_ind[np.newaxis, np.newaxis, :],
),
other.data_array.shape,
).flatten()
else:
this_inds = np.ravel_multi_index(
(
this_blts_ind[:, np.newaxis, np.newaxis, np.newaxis],
np.zeros((1, 1, 1, 1), dtype=np.int64),
this_freq_ind[np.newaxis, np.newaxis, :, np.newaxis],
this_pol_ind[np.newaxis, np.newaxis, np.newaxis, :],
),
this.data_array.shape,
).flatten()
other_inds = np.ravel_multi_index(
(
other_blts_ind[:, np.newaxis, np.newaxis, np.newaxis],
np.zeros((1, 1, 1, 1), dtype=np.int64),
other_freq_ind[np.newaxis, np.newaxis, :, np.newaxis],
other_pol_ind[np.newaxis, np.newaxis, np.newaxis, :],
),
other.data_array.shape,
).flatten()
this_all_zero = np.all(this.data_array.flatten()[this_inds] == 0)
this_all_flag = np.all(this.flag_array.flatten()[this_inds])
other_all_zero = np.all(other.data_array.flatten()[other_inds] == 0)
other_all_flag = np.all(other.flag_array.flatten()[other_inds])
if this_all_zero and this_all_flag:
# we're fine to overwrite; update history accordingly
history_update_string = " Overwrote invalid data using pyuvdata."
this.history += history_update_string
elif other_all_zero and other_all_flag:
raise ValueError(
"To combine these data, please run the add operation again, "
"but with the object whose data is to be overwritten as the "
"first object in the add operation."
)
else:
raise ValueError(
"These objects have overlapping data and cannot be combined."
)
# find the blt indices in "other" but not in "this"
temp = np.nonzero(~np.in1d(other_blts, this_blts))[0]
if len(temp) > 0:
bnew_inds = temp
new_blts = other_blts[temp]
history_update_string += "baseline-time"
n_axes += 1
else:
bnew_inds, new_blts = ([], [])
# if there's any overlap in blts, check extra params
temp = np.nonzero(np.in1d(other_blts, this_blts))[0]
if len(temp) > 0:
# add metadata to be checked to compatibility params
extra_params = [
"_integration_time",
"_lst_array",
"_phase_center_catalog",
"_phase_center_id_array",
"_phase_center_app_ra",
"_phase_center_app_dec",
"_phase_center_frame_pa",
"_Nphase",
"_uvw_array",
]
compatibility_params.extend(extra_params)
# find the freq indices in "other" but not in "this"
if this.flex_spw:
if (this.flex_spw_polarization_array is None) != (
other.flex_spw_polarization_array is None
):
raise ValueError(
"Cannot add a flex-pol and non-flex-pol UVData objects. Use "
"the `remove_flex_pol` method to convert the objects to "
"have a regular polarization axis."
)
elif this.flex_spw_polarization_array is not None:
this_flexpol_dict = {
spw: pol
for spw, pol in zip(
this.spw_array, this.flex_spw_polarization_array
)
}
other_flexpol_dict = {
spw: pol
for spw, pol in zip(
other.spw_array, other.flex_spw_polarization_array
)
}
for key in other_flexpol_dict.keys():
try:
if this_flexpol_dict[key] != other_flexpol_dict[key]:
raise ValueError(
"Cannot add a flex-pol UVData objects where the same "
"spectral window contains different polarizations. Use "
"the `remove_flex_pol` method to convert the objects "
"to have a regular polarization axis."
)
except KeyError:
this_flexpol_dict[key] = other_flexpol_dict[key]
other_mask = np.ones_like(other.flex_spw_id_array, dtype=bool)
for idx in np.intersect1d(this.spw_array, other.spw_array):
if this.future_array_shapes:
other_mask[other.flex_spw_id_array == idx] = np.isin(
other.freq_array[other.flex_spw_id_array == idx],
this.freq_array[this.flex_spw_id_array == idx],
invert=True,
)
else:
other_mask[other.flex_spw_id_array == idx] = np.isin(
other.freq_array[0, other.flex_spw_id_array == idx],
this.freq_array[0, this.flex_spw_id_array == idx],
invert=True,
)
temp = np.where(other_mask)[0]
else:
if this.future_array_shapes:
temp = np.nonzero(~np.in1d(other.freq_array, this.freq_array))[0]
else:
temp = np.nonzero(
~np.in1d(other.freq_array[0, :], this.freq_array[0, :])
)[0]
if len(temp) > 0:
fnew_inds = temp
if n_axes > 0:
history_update_string += ", frequency"
else:
history_update_string += "frequency"
n_axes += 1
else:
fnew_inds = []
# if channel width is an array and there's any overlap in freqs,
# check extra params
if this.future_array_shapes or this.flex_spw:
if this.future_array_shapes:
temp = np.nonzero(np.in1d(other.freq_array, this.freq_array))[0]
else:
temp = np.nonzero(
np.in1d(other.freq_array[0, :], this.freq_array[0, :])
)[0]
if len(temp) > 0:
# add metadata to be checked to compatibility params
extra_params = ["_channel_width"]
compatibility_params.extend(extra_params)
# find the pol indices in "other" but not in "this"
temp = np.nonzero(~np.in1d(other.polarization_array, this.polarization_array))[
0
]
if len(temp) > 0:
pnew_inds = temp
if n_axes > 0:
history_update_string += ", polarization"
else:
history_update_string += "polarization"
n_axes += 1
else:
pnew_inds = []
# Actually check compatibility parameters
blt_inds_params = [
"_integration_time",
"_lst_array",
"_phase_center_app_ra",
"_phase_center_app_dec",
"_phase_center_frame_pa",
"_phase_center_id_array",
]
for cp in compatibility_params:
if cp in blt_inds_params:
# only check that overlapping blt indices match
this_param = getattr(this, cp)
other_param = getattr(other, cp)
params_match = np.allclose(
this_param.value[this_blts_ind],
other_param.value[other_blts_ind],
rtol=this_param.tols[0],
atol=this_param.tols[1],
)
elif cp == "_uvw_array":
# only check that overlapping blt indices match
params_match = np.allclose(
this.uvw_array[this_blts_ind, :],
other.uvw_array[other_blts_ind, :],
rtol=this._uvw_array.tols[0],
atol=this._uvw_array.tols[1],
)
elif cp == "_channel_width" and this.future_array_shapes or this.flex_spw:
# only check that overlapping freq indices match
params_match = np.allclose(
this.channel_width[this_freq_ind],
other.channel_width[other_freq_ind],
rtol=this._channel_width.tols[0],
atol=this._channel_width.tols[1],
)
else:
params_match = getattr(this, cp) == getattr(other, cp)
if not params_match:
msg = (
"UVParameter " + cp[1:] + " does not match. "
"Cannot combine objects."
)
raise ValueError(msg)
# Begin manipulating the objects.
# First, handle the internal source catalogs, since merging them is kind of a
# weird, one-off process (i.e., nothing is cat'd across a particular axis)
this._consolidate_phase_center_catalogs(other=other, ignore_name=ignore_name)
# Next, we want to make sure that the ordering of the _overlapping_ data is
# the same, so that things can get plugged together in a sensible way.
if len(this_blts_ind) != 0:
this_argsort = np.argsort(this_blts_ind)
other_argsort = np.argsort(other_blts_ind)
if np.any(this_argsort != other_argsort):
temp_ind = np.arange(this.Nblts)
temp_ind[this_blts_ind[this_argsort]] = temp_ind[
this_blts_ind[other_argsort]
]
this.reorder_blts(temp_ind)
if len(this_freq_ind) != 0:
this_argsort = np.argsort(this_freq_ind)
other_argsort = np.argsort(other_freq_ind)
if np.any(this_argsort != other_argsort):
temp_ind = np.arange(this.Nfreqs)
temp_ind[this_freq_ind[this_argsort]] = temp_ind[
this_freq_ind[other_argsort]
]
this.reorder_freqs(channel_order=temp_ind)
if len(this_pol_ind) != 0:
this_argsort = np.argsort(this_pol_ind)
other_argsort = np.argsort(other_pol_ind)
if np.any(this_argsort != other_argsort):
temp_ind = np.arange(this.Npols)
temp_ind[this_pol_ind[this_argsort]] = temp_ind[
this_pol_ind[other_argsort]
]
this.reorder_pols(temp_ind)
# Pad out self to accommodate new data
if len(bnew_inds) > 0:
this_blts = np.concatenate((this_blts, new_blts))
blt_order = np.argsort(this_blts)
if not self.metadata_only:
if this.future_array_shapes:
zero_pad = np.zeros((len(bnew_inds), this.Nfreqs, this.Npols))
else:
zero_pad = np.zeros((len(bnew_inds), 1, this.Nfreqs, this.Npols))
this.data_array = np.concatenate([this.data_array, zero_pad], axis=0)
this.nsample_array = np.concatenate(
[this.nsample_array, zero_pad], axis=0
)
this.flag_array = np.concatenate(
[this.flag_array, 1 - zero_pad], axis=0
).astype(np.bool_)
this.uvw_array = np.concatenate(
[this.uvw_array, other.uvw_array[bnew_inds, :]], axis=0
)[blt_order, :]
this.time_array = np.concatenate(
[this.time_array, other.time_array[bnew_inds]]
)[blt_order]
this.integration_time = np.concatenate(
[this.integration_time, other.integration_time[bnew_inds]]
)[blt_order]
this.lst_array = np.concatenate(
[this.lst_array, other.lst_array[bnew_inds]]
)[blt_order]
this.ant_1_array = np.concatenate(
[this.ant_1_array, other.ant_1_array[bnew_inds]]
)[blt_order]
this.ant_2_array = np.concatenate(
[this.ant_2_array, other.ant_2_array[bnew_inds]]
)[blt_order]
this.baseline_array = np.concatenate(
[this.baseline_array, other.baseline_array[bnew_inds]]
)[blt_order]
this.phase_center_app_ra = np.concatenate(
[this.phase_center_app_ra, other.phase_center_app_ra[bnew_inds]]
)[blt_order]
this.phase_center_app_dec = np.concatenate(
[this.phase_center_app_dec, other.phase_center_app_dec[bnew_inds]]
)[blt_order]
this.phase_center_frame_pa = np.concatenate(
[this.phase_center_frame_pa, other.phase_center_frame_pa[bnew_inds]]
)[blt_order]
this.phase_center_id_array = np.concatenate(
[this.phase_center_id_array, other.phase_center_id_array[bnew_inds]]
)[blt_order]
if len(fnew_inds) > 0:
if this.future_array_shapes:
this.freq_array = np.concatenate(
[this.freq_array, other.freq_array[fnew_inds]]
)
else:
this.freq_array = np.concatenate(
[this.freq_array, other.freq_array[:, fnew_inds]], axis=1
)
if this.flex_spw or this.future_array_shapes:
this.channel_width = np.concatenate(
[this.channel_width, other.channel_width[fnew_inds]]
)
if this.flex_spw:
this.flex_spw_id_array = np.concatenate(
[this.flex_spw_id_array, other.flex_spw_id_array[fnew_inds]]
)
this.spw_array = np.concatenate([this.spw_array, other.spw_array])
# We want to preserve per-spw information based on first appearance
# in the concatenated array.
unique_index = np.sort(
np.unique(this.flex_spw_id_array, return_index=True)[1]
)
this.spw_array = this.flex_spw_id_array[unique_index]
this.Nspws = len(this.spw_array)
if this.flex_spw_polarization_array is not None:
this.flex_spw_polarization_array = np.array(
[this_flexpol_dict[key] for key in this.spw_array]
)
# If we have a flex/multi-spw data set, need to sort out the order of the
# individual windows first.
if this.flex_spw:
f_order = np.concatenate(
[
np.where(this.flex_spw_id_array == idx)[0]
for idx in sorted(this.spw_array)
]
)
# With spectral windows sorted, check and see if channels within
# windows need sorting. If they are ordered in ascending or descending
# fashion, leave them be. If not, sort in ascending order
for idx in this.spw_array:
select_mask = this.flex_spw_id_array[f_order] == idx
check_freqs = (
this.freq_array[f_order[select_mask]]
if this.future_array_shapes
else this.freq_array[0, f_order[select_mask]]
)
if (not np.all(check_freqs[1:] > check_freqs[:-1])) and (
not np.all(check_freqs[1:] < check_freqs[:-1])
):
subsort_order = f_order[select_mask]
f_order[select_mask] = subsort_order[np.argsort(check_freqs)]
else:
if this.future_array_shapes:
f_order = np.argsort(this.freq_array)
else:
f_order = np.argsort(this.freq_array[0, :])
if not self.metadata_only:
if this.future_array_shapes:
zero_pad = np.zeros(
(this.data_array.shape[0], len(fnew_inds), this.Npols)
)
this.data_array = np.concatenate(
[this.data_array, zero_pad], axis=1
)
this.nsample_array = np.concatenate(
[this.nsample_array, zero_pad], axis=1
)
this.flag_array = np.concatenate(
[this.flag_array, 1 - zero_pad], axis=1
).astype(np.bool_)
else:
zero_pad = np.zeros(
(this.data_array.shape[0], 1, len(fnew_inds), this.Npols)
)
this.data_array = np.concatenate(
[this.data_array, zero_pad], axis=2
)
this.nsample_array = np.concatenate(
[this.nsample_array, zero_pad], axis=2
)
this.flag_array = np.concatenate(
[this.flag_array, 1 - zero_pad], axis=2
).astype(np.bool_)
if len(pnew_inds) > 0:
this.polarization_array = np.concatenate(
[this.polarization_array, other.polarization_array[pnew_inds]]
)
p_order = np.argsort(np.abs(this.polarization_array))
if not self.metadata_only:
if this.future_array_shapes:
zero_pad = np.zeros(
(
this.data_array.shape[0],
this.data_array.shape[1],
len(pnew_inds),
)
)
this.data_array = np.concatenate(
[this.data_array, zero_pad], axis=2
)
this.nsample_array = np.concatenate(
[this.nsample_array, zero_pad], axis=2
)
this.flag_array = np.concatenate(
[this.flag_array, 1 - zero_pad], axis=2
).astype(np.bool_)
else:
zero_pad = np.zeros(
(
this.data_array.shape[0],
1,
this.data_array.shape[2],
len(pnew_inds),
)
)
this.data_array = np.concatenate(
[this.data_array, zero_pad], axis=3
)
this.nsample_array = np.concatenate(
[this.nsample_array, zero_pad], axis=3
)
this.flag_array = np.concatenate(
[this.flag_array, 1 - zero_pad], axis=3
).astype(np.bool_)
# Now populate the data
pol_t2o = np.nonzero(
np.in1d(this.polarization_array, other.polarization_array)
)[0]
# Special handling here needed for flex_spw data
this_freqs = this.freq_array
other_freqs = other.freq_array
if not this.future_array_shapes:
this_freqs = this_freqs[0]
other_freqs = other_freqs[0]
if this.flex_spw:
freq_t2o = np.zeros(this_freqs.shape, dtype=bool)
for spw_id in set(this.spw_array).intersection(other.spw_array):
mask = this.flex_spw_id_array == spw_id
freq_t2o[mask] |= np.in1d(
this_freqs[mask], other_freqs[other.flex_spw_id_array == spw_id]
)
freq_t2o = np.nonzero(freq_t2o)[0]
else:
freq_t2o = np.nonzero(np.in1d(this_freqs, other_freqs))[0]
blt_t2o = np.nonzero(np.in1d(this_blts, other_blts))[0]
if not self.metadata_only:
if this.future_array_shapes:
this.data_array[np.ix_(blt_t2o, freq_t2o, pol_t2o)] = other.data_array
this.nsample_array[
np.ix_(blt_t2o, freq_t2o, pol_t2o)
] = other.nsample_array
this.flag_array[np.ix_(blt_t2o, freq_t2o, pol_t2o)] = other.flag_array
else:
this.data_array[
np.ix_(blt_t2o, [0], freq_t2o, pol_t2o)
] = other.data_array
this.nsample_array[
np.ix_(blt_t2o, [0], freq_t2o, pol_t2o)
] = other.nsample_array
this.flag_array[
np.ix_(blt_t2o, [0], freq_t2o, pol_t2o)
] = other.flag_array
if not self.metadata_only:
if this.future_array_shapes:
if len(bnew_inds) > 0:
for name, param in zip(
this._data_params, this.data_like_parameters
):
setattr(this, name, param[blt_order, :, :])
if len(fnew_inds) > 0:
for name, param in zip(
this._data_params, this.data_like_parameters
):
setattr(this, name, param[:, f_order, :])
if len(pnew_inds) > 0:
for name, param in zip(
this._data_params, this.data_like_parameters
):
setattr(this, name, param[:, :, p_order])
else:
if len(bnew_inds) > 0:
for name, param in zip(
this._data_params, this.data_like_parameters
):
setattr(this, name, param[blt_order, :, :, :])
if len(fnew_inds) > 0:
for name, param in zip(
this._data_params, this.data_like_parameters
):
setattr(this, name, param[:, :, f_order, :])
if len(pnew_inds) > 0:
for name, param in zip(
this._data_params, this.data_like_parameters
):
setattr(this, name, param[:, :, :, p_order])
if len(fnew_inds) > 0:
if this.future_array_shapes:
this.freq_array = this.freq_array[f_order]
else:
this.freq_array = this.freq_array[:, f_order]
if this.flex_spw or this.future_array_shapes:
this.channel_width = this.channel_width[f_order]
if this.flex_spw:
this.flex_spw_id_array = this.flex_spw_id_array[f_order]
if len(pnew_inds) > 0:
this.polarization_array = this.polarization_array[p_order]
# Update N parameters (e.g. Npols)
this.Ntimes = len(np.unique(this.time_array))
this.Nbls = len(np.unique(this.baseline_array))
this.Nblts = this.uvw_array.shape[0]
this.Nfreqs = this.freq_array.size
this.Npols = this.polarization_array.shape[0]
this.Nants_data = this._calc_nants_data()
# Update filename parameter
this.filename = uvutils._combine_filenames(this.filename, other.filename)
if this.filename is not None:
this._filename.form = (len(this.filename),)
# Check specific requirements
if this.Nfreqs > 1:
spacing_error, chanwidth_error = this._check_freq_spacing(
raise_errors=False
)
if spacing_error:
warnings.warn(
"Combined frequencies are not evenly spaced or have differing "
"values of channel widths. This will make it impossible to write "
"this data out to some file types."
)
elif chanwidth_error:
warnings.warn(
"Combined frequencies are separated by more than their "
"channel width. This will make it impossible to write this data "
"out to some file types."
)
if n_axes > 0:
history_update_string += " axis using pyuvdata."
histories_match = uvutils._check_histories(this.history, other.history)
this.history += history_update_string
if not histories_match:
if verbose_history:
this.history += " Next object history follows. " + other.history
else:
extra_history = uvutils._combine_history_addition(
this.history, other.history
)
if extra_history is not None:
this.history += (
" Unique part of next object history follows. "
+ extra_history
)
# Check final object is self-consistent
if run_check:
this.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
if not inplace:
return this
def __iadd__(
self,
other,
phase_center_radec=None,
unphase_to_drift=None,
phase_frame=None,
orig_phase_frame=None,
use_ant_pos=None,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
make_multi_phase=None,
ignore_name=False,
):
"""
In place add.
Parameters
----------
other : UVData object
Another UVData object which will be added to self.
phase_center_radec : array_like of float
Deprecated and will be removed in version 2.4, instead call `phase` on the
combined object after this method. The phase center to phase the files to
before adding the objects in radians.
unphase_to_drift : bool
Deprecated, instead call `unproject_phase` on the combined object after
this method. Default is False.
phase_frame : str
The astropy frame to phase to. Only used if `phase_center_radec` is set.
Default is "icrs".
Deprecated and will be removed in version 2.4 with `phase_center_radec`.
orig_phase_frame : str
Deprecated, has no effect. Will be removed in version 2.4.
use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions rather than
from the existing uvws. Only used if `unphase_to_drift` or
`phase_center_radec` are set. Deprecated and will be removed in version 2.4
with `unphase_to_drift` and `phase_center_radec`.
verbose_history : bool
Option to allow more verbose history. If True and if the histories for the
two objects are different, the combined object will keep all the history of
both input objects (if many objects are combined in succession this can
lead to very long histories). If False and if the histories for the two
objects are different, the combined object will have the history of the
first object and only the parts of the second object history that are unique
(this is done word by word and can result in hard to interpret histories).
run_check : bool
Option to check for the existence and proper shapes of parameters
after combining objects.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
combining objects.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
make_multi_phase : bool
Deprecated and has no effect. Will be removed in version 2.4.
ignore_name : bool
Option to ignore the name of the phase center (`cat_name` in
`phase_center_catalog`) when combining two UVData objects. If set to True,
phase centers that are the same up to their name will be combined with the
name set to the name found in the first UVData object in the sum. If set to
False, phase centers that are the same up to the name will be kept as
separate phase centers. Default is False.
Raises
------
ValueError
If other is not a UVData object, self and other are not compatible
or if data in self and other overlap.
If `phase_center_radec` is not None and is not length 2.
"""
self.__add__(
other,
inplace=True,
phase_center_radec=phase_center_radec,
unphase_to_drift=unphase_to_drift,
phase_frame=phase_frame,
orig_phase_frame=orig_phase_frame,
use_ant_pos=use_ant_pos,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
make_multi_phase=make_multi_phase,
ignore_name=ignore_name,
)
return self
def fast_concat(
self,
other,
axis,
inplace=False,
phase_center_radec=None,
unphase_to_drift=None,
phase_frame=None,
orig_phase_frame=None,
use_ant_pos=None,
verbose_history=False,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
ignore_name=None,
):
"""
Concatenate two UVData objects along specified axis with almost no checking.
Warning! This method assumes all the metadata along other axes is sorted
the same way. The __add__ method is much safer, it checks all the metadata,
but it is slower. Some quick checks are run, but this method doesn't
make any guarantees that the resulting object is correct.
Note that if objects have different phasing (different phase_center_catalogs)
they can still be combined and the phasing information will be preserved.
However, the phase center ID numbers may be changed on any of the objects passed
into this method.
Parameters
----------
other : UVData object or list of UVData objects
UVData object or list of UVData objects which will be added to self.
axis : str
Axis to concatenate files along. This enables fast concatenation
along the specified axis without the normal checking that all other
metadata agrees. Allowed values are: 'blt', 'freq', 'polarization'.
inplace : bool
If True, overwrite self as we go, otherwise create a third object
as the sum of the two.
phase_center_radec : array_like of float
Deprecated and will be removed in version 2.4, instead call `phase` on the
combined object after this method. The phase center to phase the files to
before concatenating the objects in radians.
unphase_to_drift : bool
Deprecated, instead call `unproject_phase` on the combined object after
this method. Unproject the objects before combining them.
phase_frame : str
The astropy frame to phase to. Only used if `phase_center_radec` is set.
Deprecated and will be removed in version 2.4 with `phase_center_radec`.
orig_phase_frame : str
Deprecated, has no effect. Will be removed in version 2.4.
use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions rather than
from the existing uvws. Only used if `unphase_to_drift` or
`phase_center_radec` are set. Deprecated and will be removed in version 2.4
with `unphase_to_drift` and `phase_center_radec`.
verbose_history : bool
Option to allow more verbose history. If True and if the histories for the
objects are different, the combined object will keep all the history of
all input objects (if many objects are combined this can lead to very long
histories). If False and if the histories for the objects are different,
the combined object will have the history of the first object and only the
parts of the other object histories that are unique (this is done word by
word and can result in hard to interpret histories).
run_check : bool
Option to check for the existence and proper shapes of parameters
after combining objects.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
combining objects.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
ignore_name : bool
Option to ignore the name of the phase center (`cat_name` in
`phase_center_catalog`) when combining two UVData objects. If set to True,
phase centers that are the same up to their name will be combined with the
name set to the name found in the first UVData object in the sum. If set to
False, phase centers that are the same up to the name will be kept as
separate phase centers. Default is False.
Raises
------
ValueError
If other is not a UVData object, axis is not an allowed value or if
self and other are not compatible.
"""
allowed_axes = ["blt", "freq", "polarization"]
if axis not in allowed_axes:
raise ValueError(
"If axis is specifed it must be one of: " + ", ".join(allowed_axes)
)
if inplace:
this = self
else:
this = self.copy()
if not isinstance(other, (list, tuple, np.ndarray)):
# if this is a UVData object already, stick it in a list
other = [other]
# Check that both objects are UVData and valid
this.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
for obj in other:
if not issubclass(obj.__class__, this.__class__):
if not issubclass(this.__class__, obj.__class__):
raise ValueError(
"Only UVData (or subclass) objects can be "
"added to a UVData (or subclass) object"
)
obj.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
# check that all objects have the same array shapes
for obj in other:
if this.future_array_shapes != obj.future_array_shapes:
raise ValueError(
"All objects must have the same `future_array_shapes` parameter. "
"Use the `use_future_array_shapes` or `use_current_array_shapes` "
"methods to convert them."
)
this._deprecated_phase_helper(
other,
phase_center_radec,
unphase_to_drift,
phase_frame,
orig_phase_frame,
use_ant_pos,
None,
)
# update the phase_center_catalog to make them consistent across objects
# Doing this as a binary tree merge
# The left object in each loop will have its phase center IDs updated.
uv_list = [this] + other
while len(uv_list) > 1:
for uv1, uv2 in zip(uv_list[0::2], uv_list[1::2]):
uv1._consolidate_phase_center_catalogs(
other=uv2, ignore_name=ignore_name
)
uv_list = uv_list[0::2]
# Because self was at the beginning of the list,
# all the phase centers are merged into it at the end of this loop
compatibility_params = [
"_vis_units",
"_telescope_name",
"_instrument",
"_telescope_location",
"_Nants_telescope",
"_antenna_names",
"_antenna_numbers",
"_antenna_positions",
]
if not this.future_array_shapes and not this.flex_spw:
compatibility_params.append("_channel_width")
history_update_string = " Combined data along "
if axis == "freq":
history_update_string += "frequency"
compatibility_params += [
"_polarization_array",
"_ant_1_array",
"_ant_2_array",
"_integration_time",
"_uvw_array",
"_lst_array",
"_phase_center_id_array",
]
elif axis == "polarization":
history_update_string += "polarization"
compatibility_params += [
"_freq_array",
"_ant_1_array",
"_ant_2_array",
"_integration_time",
"_uvw_array",
"_lst_array",
"_phase_center_id_array",
]
elif axis == "blt":
history_update_string += "baseline-time"
compatibility_params += ["_freq_array", "_polarization_array"]
history_update_string += " axis using pyuvdata."
histories_match = []
for obj in other:
histories_match.append(uvutils._check_histories(this.history, obj.history))
this.history += history_update_string
for obj_num, obj in enumerate(other):
if not histories_match[obj_num]:
if verbose_history:
this.history += " Next object history follows. " + obj.history
else:
extra_history = uvutils._combine_history_addition(
this.history, obj.history
)
if extra_history is not None:
this.history += (
" Unique part of next object history follows. "
+ extra_history
)
# Actually check compatibility parameters
for obj in other:
for a in compatibility_params:
params_match = getattr(this, a) == getattr(obj, a)
if not params_match:
msg = (
"UVParameter "
+ a[1:]
+ " does not match. Cannot combine objects."
)
raise ValueError(msg)
if axis == "freq":
this.Nfreqs = sum([this.Nfreqs] + [obj.Nfreqs for obj in other])
if this.future_array_shapes:
this.freq_array = np.concatenate(
[this.freq_array] + [obj.freq_array for obj in other]
)
else:
this.freq_array = np.concatenate(
[this.freq_array] + [obj.freq_array for obj in other], axis=1
)
if this.flex_spw or this.future_array_shapes:
this.channel_width = np.concatenate(
[this.channel_width] + [obj.channel_width for obj in other]
)
if this.flex_spw:
this.flex_spw_id_array = np.concatenate(
[this.flex_spw_id_array] + [obj.flex_spw_id_array for obj in other]
)
this.spw_array = np.concatenate(
[this.spw_array] + [obj.spw_array for obj in other]
)
# We want to preserve per-spw information based on first appearance
# in the concatenated array.
unique_index = np.sort(
np.unique(this.flex_spw_id_array, return_index=True)[1]
)
this.spw_array = this.flex_spw_id_array[unique_index]
this.Nspws = len(this.spw_array)
spacing_error, chanwidth_error = this._check_freq_spacing(
raise_errors=False
)
if spacing_error:
warnings.warn(
"Combined frequencies are not evenly spaced or have differing "
"values of channel widths. This will make it impossible to write "
"this data out to some file types."
)
elif chanwidth_error:
warnings.warn(
"Combined frequencies are separated by more than their "
"channel width. This will make it impossible to write this data "
"out to some file types."
)
if not self.metadata_only:
if this.future_array_shapes:
this.data_array = np.concatenate(
[this.data_array] + [obj.data_array for obj in other], axis=1
)
this.nsample_array = np.concatenate(
[this.nsample_array] + [obj.nsample_array for obj in other],
axis=1,
)
this.flag_array = np.concatenate(
[this.flag_array] + [obj.flag_array for obj in other], axis=1
)
else:
this.data_array = np.concatenate(
[this.data_array] + [obj.data_array for obj in other], axis=2
)
this.nsample_array = np.concatenate(
[this.nsample_array] + [obj.nsample_array for obj in other],
axis=2,
)
this.flag_array = np.concatenate(
[this.flag_array] + [obj.flag_array for obj in other], axis=2
)
elif axis == "polarization":
this.polarization_array = np.concatenate(
[this.polarization_array] + [obj.polarization_array for obj in other]
)
this.Npols = sum([this.Npols] + [obj.Npols for obj in other])
if not uvutils._test_array_constant_spacing(this._polarization_array):
warnings.warn(
"Combined polarizations are not evenly spaced. This will "
"make it impossible to write this data out to some file types."
)
if not self.metadata_only:
if this.future_array_shapes:
this.data_array = np.concatenate(
[this.data_array] + [obj.data_array for obj in other], axis=2
)
this.nsample_array = np.concatenate(
[this.nsample_array] + [obj.nsample_array for obj in other],
axis=2,
)
this.flag_array = np.concatenate(
[this.flag_array] + [obj.flag_array for obj in other], axis=2
)
else:
this.data_array = np.concatenate(
[this.data_array] + [obj.data_array for obj in other], axis=3
)
this.nsample_array = np.concatenate(
[this.nsample_array] + [obj.nsample_array for obj in other],
axis=3,
)
this.flag_array = np.concatenate(
[this.flag_array] + [obj.flag_array for obj in other], axis=3
)
elif axis == "blt":
this.Nblts = sum([this.Nblts] + [obj.Nblts for obj in other])
this.ant_1_array = np.concatenate(
[this.ant_1_array] + [obj.ant_1_array for obj in other]
)
this.ant_2_array = np.concatenate(
[this.ant_2_array] + [obj.ant_2_array for obj in other]
)
this.Nants_data = this._calc_nants_data()
this.uvw_array = np.concatenate(
[this.uvw_array] + [obj.uvw_array for obj in other], axis=0
)
this.time_array = np.concatenate(
[this.time_array] + [obj.time_array for obj in other]
)
this.Ntimes = len(np.unique(this.time_array))
this.lst_array = np.concatenate(
[this.lst_array] + [obj.lst_array for obj in other]
)
this.baseline_array = np.concatenate(
[this.baseline_array] + [obj.baseline_array for obj in other]
)
this.Nbls = len(np.unique(this.baseline_array))
this.integration_time = np.concatenate(
[this.integration_time] + [obj.integration_time for obj in other]
)
this.phase_center_app_ra = np.concatenate(
[this.phase_center_app_ra] + [obj.phase_center_app_ra for obj in other]
)
this.phase_center_app_dec = np.concatenate(
[this.phase_center_app_dec]
+ [obj.phase_center_app_dec for obj in other]
)
this.phase_center_frame_pa = np.concatenate(
[this.phase_center_frame_pa]
+ [obj.phase_center_frame_pa for obj in other]
)
this.phase_center_id_array = np.concatenate(
[this.phase_center_id_array]
+ [obj.phase_center_id_array for obj in other]
)
if not self.metadata_only:
this.data_array = np.concatenate(
[this.data_array] + [obj.data_array for obj in other], axis=0
)
this.nsample_array = np.concatenate(
[this.nsample_array] + [obj.nsample_array for obj in other], axis=0
)
this.flag_array = np.concatenate(
[this.flag_array] + [obj.flag_array for obj in other], axis=0
)
# update filename attribute
for obj in other:
this.filename = uvutils._combine_filenames(this.filename, obj.filename)
if this.filename is not None:
this._filename.form = len(this.filename)
# Check final object is self-consistent
if run_check:
this.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
if not inplace:
return this
def sum_vis(
self,
other,
inplace=False,
difference=False,
verbose_history=False,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
override_params=None,
):
"""
Sum visibilities between two UVData objects.
By default requires that all UVParameters are the same on the two objects
except for `history`, `data_array`, and `extra_keywords`.
If keys in `extra_keywords` have different values the values from the first
object are taken.
Parameters
----------
other : UVData object
Another UVData object which will be added to self.
difference : bool
If True, differences the visibilities of the two UVData objects
rather than summing them.
inplace : bool
If True, overwrite self as we go, otherwise create a third object
as the sum of the two.
verbose_history : bool
Option to allow more verbose history. If True and if the histories for the
two objects are different, the combined object will keep all the history of
both input objects (this can lead to long histories). If False and if the
histories for the two objects are different, the combined object will have
the history of the first object and only the parts of the second object
history that are unique (this is done word by word and can result in hard
to interpret histories).
run_check : bool
Option to check for the existence and proper shapes of parameters
after combining objects.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
combining objects.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
override_params : array_like of strings
List of object UVParameters to omit from compatibility check. Overridden
parameters will not be compared between the objects, and the values
for these parameters will be taken from the first object.
Returns
-------
UVData Object
If inplace parameter is False.
Raises
------
ValueError
If other is not a UVData object, or if self and other
are not compatible.
"""
if inplace:
this = self
else:
this = self.copy()
# Check that both objects are UVData and valid
this.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
if not issubclass(other.__class__, this.__class__):
if not issubclass(this.__class__, other.__class__):
raise ValueError(
"Only UVData (or subclass) objects can be "
"added to a UVData (or subclass) object"
)
other.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
# check that both objects have the same array shapes
if this.future_array_shapes != other.future_array_shapes:
raise ValueError(
"Both objects must have the same `future_array_shapes` parameter. "
"Use the `use_future_array_shapes` or `use_current_array_shapes` "
"methods to convert them."
)
compatibility_params = list(this.__iter__())
remove_params = ["_history", "_data_array", "_extra_keywords"]
# Add underscores to override_params to match list from __iter__()
# Add to parameters to be removed
if override_params and all(isinstance(param, str) for param in override_params):
for param in override_params:
if param[0] != "_":
param = "_" + param
if param not in compatibility_params:
msg = (
"Provided parameter " + param[1:] + " is not a recognizable "
"UVParameter."
)
raise ValueError(msg)
remove_params.append(param)
# compatibility_params should define the parameters that need to
# be the same for objects to be summed or diffed
compatibility_params = list(set(compatibility_params) - set(remove_params))
# Check each UVParameter in compatibility_params
for param in compatibility_params:
params_match = getattr(this, param) == getattr(other, param)
if not params_match:
msg = (
"UVParameter " + param[1:] + " does not match. Cannot "
"combine objects."
)
raise ValueError(msg)
# Merge extra keywords
for intersection in set(this.extra_keywords.keys()) & set(
other.extra_keywords.keys()
):
if this.extra_keywords[intersection] != other.extra_keywords[intersection]:
warnings.warn(
"Keyword " + intersection + " in _extra_keywords is different "
"in the two objects. Taking the first object's entry."
)
# Merge extra_keywords lists, taking values from the first object
this.extra_keywords = dict(
list(other.extra_keywords.items()) + list(this.extra_keywords.items())
)
# Do the summing / differencing
if difference:
this.data_array = this.data_array - other.data_array
history_update_string = " Visibilities differenced using pyuvdata."
else:
this.data_array = this.data_array + other.data_array
history_update_string = " Visibilities summed using pyuvdata."
histories_match = uvutils._check_histories(this.history, other.history)
this.history += history_update_string
if not histories_match:
if verbose_history:
this.history += " Second object history follows. " + other.history
else:
extra_history = uvutils._combine_history_addition(
this.history, other.history
)
if extra_history is not None:
this.history += (
" Unique part of second object history follows. "
+ extra_history
)
# merge file names
this.filename = uvutils._combine_filenames(this.filename, other.filename)
# Check final object is self-consistent
if run_check:
this.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
if not inplace:
return this
def diff_vis(
self,
other,
inplace=False,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
override_params=None,
):
"""
Difference visibilities between two UVData objects.
By default requires that all UVParameters are the same on the two objects
except for `history`, `data_array`, and `extra_keywords`.
If keys in `extra_keywords` have different values the values from the first
object are taken.
Parameters
----------
other : UVData object
Another UVData object which will be added to self.
inplace : bool
If True, overwrite self as we go, otherwise create a third object
as the sum of the two.
run_check : bool
Option to check for the existence and proper shapes of parameters
after combining objects.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
combining objects.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
override_params : array_like of strings
List of object UVParameters to omit from compatibility check. Overridden
parameters will not be compared between the objects, and the values
for these parameters will be taken from the first object.
Returns
-------
UVData Object
If inplace parameter is False.
Raises
------
ValueError
If other is not a UVData object, or if self and other
are not compatible.
"""
if inplace:
self.sum_vis(
other,
difference=True,
inplace=inplace,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
override_params=override_params,
)
else:
return self.sum_vis(
other,
difference=True,
inplace=inplace,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
override_params=override_params,
)
def parse_ants(self, ant_str, print_toggle=False):
"""
Get antpair and polarization from parsing an aipy-style ant string.
Used to support the select function. Generates two lists of antenna pair
tuples and polarization indices based on parsing of the string ant_str.
If no valid polarizations (pseudo-Stokes params, or combinations of [lr]
or [xy]) or antenna numbers are found in ant_str, ant_pairs_nums and
polarizations are returned as None.
Parameters
----------
ant_str : str
String containing antenna information to parse. Can be 'all',
'auto', 'cross', or combinations of antenna numbers and polarization
indicators 'l' and 'r' or 'x' and 'y'. Minus signs can also be used
in front of an antenna number or baseline to exclude it from being
output in ant_pairs_nums. If ant_str has a minus sign as the first
character, 'all,' will be added to the beginning of the string.
See the tutorial for examples of valid strings and their behavior.
print_toggle : bool
Boolean for printing parsed baselines for a visual user check.
Returns
-------
ant_pairs_nums : list of tuples of int or None
List of tuples containing the parsed pairs of antenna numbers, or
None if ant_str is 'all' or a pseudo-Stokes polarizations.
polarizations : list of int or None
List of desired polarizations or None if ant_str does not contain a
polarization specification.
"""
return uvutils.parse_ants(
uv=self,
ant_str=ant_str,
print_toggle=print_toggle,
x_orientation=self.x_orientation,
)
def _select_preprocess(
self,
antenna_nums,
antenna_names,
ant_str,
bls,
frequencies,
freq_chans,
times,
time_range,
lsts,
lst_range,
polarizations,
blt_inds,
phase_center_ids,
):
"""
Build up blt_inds, freq_inds, pol_inds and history_update_string for select.
Parameters
----------
antenna_nums : array_like of int, optional
The antennas numbers to keep in the object (antenna positions and
names for the removed antennas will be retained unless
`keep_all_metadata` is False). This cannot be provided if
`antenna_names` is also provided.
antenna_names : array_like of str, optional
The antennas names to keep in the object (antenna positions and
names for the removed antennas will be retained unless
`keep_all_metadata` is False). This cannot be provided if
`antenna_nums` is also provided.
bls : list of tuple or list of int, optional
A list of antenna number tuples (e.g. [(0, 1), (3, 2)]), a list of
baseline 3-tuples (e.g. [(0, 1, 'xx'), (2, 3, 'yy')]), or a list of
baseline numbers (e.g. [67599, 71699, 73743]) specifying baselines
to keep in the object. For length-2 tuples, the ordering of the
numbers within the tuple does not matter. For length-3 tuples, the
polarization string is in the order of the two antennas. If
length-3 tuples are provided, `polarizations` must be None.
ant_str : str, optional
A string containing information about what antenna numbers
and polarizations to keep in the object. Can be 'auto', 'cross', 'all',
or combinations of antenna numbers and polarizations (e.g. '1',
'1_2', '1x_2y'). See tutorial for more examples of valid strings and
the behavior of different forms for ant_str.
If '1x_2y,2y_3y' is passed, both polarizations 'xy' and 'yy' will
be kept for both baselines (1, 2) and (2, 3) to return a valid
pyuvdata object.
An ant_str cannot be passed in addition to any of `antenna_nums`,
`antenna_names`, `bls` args or the `polarizations` parameters,
if it is a ValueError will be raised.
frequencies : array_like of float, optional
The frequencies to keep in the object, each value passed here should
exist in the freq_array.
freq_chans : array_like of int, optional
The frequency channel numbers to keep in the object.
times : array_like of float, optional
The times to keep in the object, each value passed here should exist
in the time_array. Cannot be used with `time_range`, `lsts`, or
`lst_array`.
time_range : array_like of float, optional
The time range in Julian Date to keep in the object, must be length
2. Some of the times in the object should fall between the first and
last elements. Cannot be used with `times`, `lsts`, or `lst_array`.
lsts : array_like of float, optional
The local sidereal times (LSTs) to keep in the object, each value
passed here should exist in the lst_array. Cannot be used with
`times`, `time_range`, or `lst_range`.
lst_range : array_like of float, optional
The local sidereal time (LST) range in radians to keep in the
object, must be of length 2. Some of the LSTs in the object should
fall between the first and last elements. If the second value is
smaller than the first, the LSTs are treated as having phase-wrapped
around LST = 2*pi = 0, and the LSTs kept on the object will run from
the larger value, through 0, and end at the smaller value.
polarizations : array_like of int or str, optional
The polarizations numbers to keep in the object, each value passed
here should exist in the polarization_array. If passing strings, the
canonical polarization strings (e.g. "xx", "rr") are supported and if the
`x_orientation` attribute is set, the physical dipole strings
(e.g. "nn", "ee") are also supported.
blt_inds : array_like of int, optional
The baseline-time indices to keep in the object. This is
not commonly used.
phase_center_ids : array_like of int, optional
Phase center IDs to keep on the object (effectively a selection on
baseline-times).
Returns
-------
blt_inds : list of int
list of baseline-time indices to keep. Can be None (to keep everything).
freq_inds : list of int
list of frequency indices to keep. Can be None (to keep everything).
pol_inds : list of int
list of polarization indices to keep. Can be None (to keep everything).
history_update_string : str
string to append to the end of the history.
"""
# build up history string as we go
history_update_string = " Downselected to specific "
n_selects = 0
if ant_str is not None:
if not (
antenna_nums is None
and antenna_names is None
and bls is None
and polarizations is None
):
raise ValueError(
"Cannot provide ant_str with antenna_nums, antenna_names, "
"bls, or polarizations."
)
else:
bls, polarizations = self.parse_ants(ant_str)
if bls is not None and len(bls) == 0:
raise ValueError(
f"There is no data matching ant_str={ant_str} in this object."
)
# Antennas, times and blt_inds all need to be combined into a set of
# blts indices to keep.
# test for blt_inds presence before adding inds from antennas & times
if blt_inds is not None:
blt_inds = uvutils._get_iterable(blt_inds)
if np.array(blt_inds).ndim > 1:
blt_inds = np.array(blt_inds).flatten()
history_update_string += "baseline-times"
n_selects += 1
if phase_center_ids is not None:
phase_center_ids = np.array(uvutils._get_iterable(phase_center_ids))
pc_blt_inds = np.nonzero(
np.isin(self.phase_center_id_array, phase_center_ids)
)[0]
if blt_inds is not None:
# Use intersection (and) to join phase_center_ids
# with blt_inds
blt_inds = np.array(
list(set(blt_inds).intersection(pc_blt_inds)), dtype=np.int64
)
else:
blt_inds = pc_blt_inds
if n_selects > 0:
history_update_string += ", phase center IDs"
else:
history_update_string += "phase center IDs"
n_selects += 1
if antenna_names is not None:
if antenna_nums is not None:
raise ValueError(
"Only one of antenna_nums and antenna_names can be provided."
)
if not isinstance(antenna_names, (list, tuple, np.ndarray)):
antenna_names = (antenna_names,)
if np.array(antenna_names).ndim > 1:
antenna_names = np.array(antenna_names).flatten()
antenna_nums = []
for s in antenna_names:
if s not in self.antenna_names:
raise ValueError(
"Antenna name {a} is not present in the antenna_names"
" array".format(a=s)
)
antenna_nums.append(
self.antenna_numbers[np.where(np.array(self.antenna_names) == s)][0]
)
if antenna_nums is not None:
antenna_nums = uvutils._get_iterable(antenna_nums)
antenna_nums = np.asarray(antenna_nums)
if antenna_nums.ndim > 1:
antenna_nums = antenna_nums.flatten()
if n_selects > 0:
history_update_string += ", antennas"
else:
history_update_string += "antennas"
n_selects += 1
# Check to make sure that we actually have these antenna nums in the data
ant_check = np.logical_or(
np.isin(antenna_nums, self.ant_1_array),
np.isin(antenna_nums, self.ant_2_array),
)
if not np.all(ant_check):
raise ValueError(
f"Antenna number {antenna_nums[~ant_check]} is not present in the "
"ant_1_array or ant_2_array"
)
ant_blt_inds = np.where(
np.logical_and(
np.isin(self.ant_1_array, antenna_nums),
np.isin(self.ant_2_array, antenna_nums),
)
)[0]
else:
ant_blt_inds = None
if bls is not None:
if isinstance(bls, list) and all(
isinstance(bl_ind, (int, np.integer)) for bl_ind in bls
):
for bl_ind in bls:
if not (bl_ind in self.baseline_array):
raise ValueError(
"Baseline number {i} is not present in the "
"baseline_array".format(i=bl_ind)
)
bls = list(zip(*self.baseline_to_antnums(bls)))
elif isinstance(bls, tuple) and (len(bls) == 2 or len(bls) == 3):
bls = [bls]
if len(bls) == 0 or not all(isinstance(item, tuple) for item in bls):
raise ValueError(
"bls must be a list of tuples of antenna numbers "
"(optionally with polarization) or a list of baseline numbers."
)
if not all(
[isinstance(item[0], (int, np.integer)) for item in bls]
+ [isinstance(item[1], (int, np.integer)) for item in bls]
):
raise ValueError(
"bls must be a list of tuples of antenna numbers "
"(optionally with polarization) or a list of baseline numbers."
)
if all(len(item) == 3 for item in bls):
if polarizations is not None:
raise ValueError(
"Cannot provide length-3 tuples and also specify polarizations."
)
if not all(isinstance(item[2], str) for item in bls):
raise ValueError(
"The third element in each bl must be a polarization string"
)
if ant_str is None:
if n_selects > 0:
history_update_string += ", baselines"
else:
history_update_string += "baselines"
else:
history_update_string += "antenna pairs"
n_selects += 1
bls_blt_inds = np.zeros(0, dtype=np.int64)
bl_pols = set()
for bl in bls:
wh1 = np.where(
np.logical_and(self.ant_1_array == bl[0], self.ant_2_array == bl[1])
)[0]
if len(wh1) > 0:
bls_blt_inds = np.append(bls_blt_inds, list(wh1))
if len(bl) == 3:
bl_pols.add(bl[2])
else:
wh2 = np.where(
np.logical_and(
self.ant_1_array == bl[1], self.ant_2_array == bl[0]
)
)[0]
if len(wh2) > 0:
bls_blt_inds = np.append(bls_blt_inds, list(wh2))
if len(bl) == 3:
# find conjugate polarization
bl_pols.add(uvutils.conj_pol(bl[2]))
else:
raise ValueError(
"Antenna pair {p} does not have any data "
"associated with it.".format(p=bl)
)
if len(bl_pols) > 0:
polarizations = list(bl_pols)
if ant_blt_inds is not None:
# Use intersection (and) to join antenna_names/nums & ant_pairs_nums
ant_blt_inds = np.array(
list(set(ant_blt_inds).intersection(bls_blt_inds))
)
else:
ant_blt_inds = bls_blt_inds
if ant_blt_inds is not None:
if blt_inds is not None:
# Use intersection (and) to join antenna_names/nums/ant_pairs_nums
# with blt_inds
blt_inds = np.array(
list(set(blt_inds).intersection(ant_blt_inds)), dtype=np.int64
)
else:
blt_inds = ant_blt_inds
have_times = times is not None
have_time_range = time_range is not None
have_lsts = lsts is not None
have_lst_range = lst_range is not None
if (
np.count_nonzero([have_times, have_time_range, have_lsts, have_lst_range])
> 1
):
raise ValueError(
"Only one of [times, time_range, lsts, lst_range] may be "
"specified per selection operation."
)
if times is not None:
times = uvutils._get_iterable(times)
if np.array(times).ndim > 1:
times = np.array(times).flatten()
time_blt_inds = np.zeros(0, dtype=np.int64)
for jd in times:
if np.any(
np.isclose(
self.time_array,
jd,
rtol=self._time_array.tols[0],
atol=self._time_array.tols[1],
)
):
time_blt_inds = np.append(
time_blt_inds,
np.where(
np.isclose(
self.time_array,
jd,
rtol=self._time_array.tols[0],
atol=self._time_array.tols[1],
)
)[0],
)
else:
raise ValueError(
"Time {t} is not present in the time_array".format(t=jd)
)
if time_range is not None:
if np.size(time_range) != 2:
raise ValueError("time_range must be length 2.")
time_blt_inds = np.nonzero(
(self.time_array <= time_range[1]) & (self.time_array >= time_range[0])
)[0]
if time_blt_inds.size == 0:
raise ValueError(
f"No elements in time range between {time_range[0]} and "
f"{time_range[1]}."
)
if lsts is not None:
if np.any(np.asarray(lsts) > 2 * np.pi):
warnings.warn(
"The lsts parameter contained a value greater than 2*pi. "
"LST values are assumed to be in radians, not hours."
)
lsts = uvutils._get_iterable(lsts)
if np.array(lsts).ndim > 1:
lsts = np.array(lsts).flatten()
time_blt_inds = np.zeros(0, dtype=np.int64)
for lst in lsts:
if np.any(
np.isclose(
self.lst_array,
lst,
rtol=self._lst_array.tols[0],
atol=self._lst_array.tols[1],
)
):
time_blt_inds = np.append(
time_blt_inds,
np.where(
np.isclose(
self.lst_array,
lst,
rtol=self._lst_array.tols[0],
atol=self._lst_array.tols[1],
)
)[0],
)
else:
raise ValueError(f"LST {lst} is not present in the lst_array")
if lst_range is not None:
if np.size(lst_range) != 2:
raise ValueError("lst_range must be length 2.")
if np.any(np.asarray(lst_range) > 2 * np.pi):
warnings.warn(
"The lst_range contained a value greater than 2*pi. "
"LST values are assumed to be in radians, not hours."
)
if lst_range[1] < lst_range[0]:
# we're wrapping around LST = 2*pi = 0
lst_range_1 = [lst_range[0], 2 * np.pi]
lst_range_2 = [0, lst_range[1]]
time_blt_inds1 = np.nonzero(
(self.lst_array <= lst_range_1[1])
& (self.lst_array >= lst_range_1[0])
)[0]
time_blt_inds2 = np.nonzero(
(self.lst_array <= lst_range_2[1])
& (self.lst_array >= lst_range_2[0])
)[0]
time_blt_inds = np.union1d(time_blt_inds1, time_blt_inds2)
else:
time_blt_inds = np.nonzero(
(self.lst_array <= lst_range[1]) & (self.lst_array >= lst_range[0])
)[0]
if time_blt_inds.size == 0:
raise ValueError(
f"No elements in LST range between {lst_range[0]} and "
f"{lst_range[1]}."
)
if times is not None or time_range is not None:
if n_selects > 0:
history_update_string += ", times"
else:
history_update_string += "times"
n_selects += 1
if blt_inds is not None:
# Use intesection (and) to join
# antenna_names/nums/ant_pairs_nums/blt_inds with times
blt_inds = np.array(
list(set(blt_inds).intersection(time_blt_inds)), dtype=np.int64
)
else:
blt_inds = time_blt_inds
if lsts is not None or lst_range is not None:
if n_selects > 0:
history_update_string += ", lsts"
else:
history_update_string += "lsts"
n_selects += 1
if blt_inds is not None:
# Use intesection (and) to join
# antenna_names/nums/ant_pairs_nums/blt_inds with times
blt_inds = np.array(
list(set(blt_inds).intersection(time_blt_inds)), dtype=np.int64
)
else:
blt_inds = time_blt_inds
if blt_inds is not None:
if len(blt_inds) == 0:
raise ValueError("No baseline-times were found that match criteria")
if max(blt_inds) >= self.Nblts:
raise ValueError("blt_inds contains indices that are too large")
if min(blt_inds) < 0:
raise ValueError("blt_inds contains indices that are negative")
blt_inds = sorted(set(blt_inds))
if freq_chans is not None:
freq_chans = uvutils._get_iterable(freq_chans)
if np.array(freq_chans).ndim > 1:
freq_chans = np.array(freq_chans).flatten()
if frequencies is None:
if self.future_array_shapes:
frequencies = self.freq_array[freq_chans]
else:
frequencies = self.freq_array[0, freq_chans]
else:
frequencies = uvutils._get_iterable(frequencies)
if self.future_array_shapes:
frequencies = np.sort(
list(set(frequencies) | set(self.freq_array[freq_chans]))
)
else:
frequencies = np.sort(
list(set(frequencies) | set(self.freq_array[0, freq_chans]))
)
if frequencies is not None:
frequencies = uvutils._get_iterable(frequencies)
if np.array(frequencies).ndim > 1:
frequencies = np.array(frequencies).flatten()
if n_selects > 0:
history_update_string += ", frequencies"
else:
history_update_string += "frequencies"
n_selects += 1
if self.future_array_shapes:
freq_arr_use = self.freq_array
else:
freq_arr_use = self.freq_array[0, :]
# Check and see that all requested freqs are available
freq_check = np.isin(frequencies, freq_arr_use)
if not np.all(freq_check):
raise ValueError(
"Frequency %g is not present in the freq_array"
% frequencies[np.where(~freq_check)[0][0]]
)
freq_inds = np.where(np.isin(freq_arr_use, frequencies))[0]
if len(frequencies) > 1:
freq_ind_separation = freq_inds[1:] - freq_inds[:-1]
if self.flex_spw:
freq_ind_separation = freq_ind_separation[
np.diff(self.flex_spw_id_array[freq_inds]) == 0
]
if not uvutils._test_array_constant(freq_ind_separation):
warnings.warn(
"Selected frequencies are not evenly spaced. This "
"will make it impossible to write this data out to "
"some file types"
)
elif np.max(freq_ind_separation) > 1:
warnings.warn(
"Selected frequencies are not contiguous. This "
"will make it impossible to write this data out to "
"some file types."
)
freq_inds = sorted(set(freq_inds))
else:
freq_inds = None
if polarizations is not None:
polarizations = uvutils._get_iterable(polarizations)
if np.array(polarizations).ndim > 1:
polarizations = np.array(polarizations).flatten()
if n_selects > 0:
history_update_string += ", polarizations"
else:
history_update_string += "polarizations"
n_selects += 1
pol_inds = np.zeros(0, dtype=np.int64)
spw_inds = np.zeros(0, dtype=np.int64)
for p in polarizations:
if isinstance(p, str):
p_num = uvutils.polstr2num(p, x_orientation=self.x_orientation)
else:
p_num = p
if p_num in self.polarization_array:
pol_inds = np.append(
pol_inds, np.where(self.polarization_array == p_num)[0]
)
elif (
False
if (self.flex_spw_polarization_array is None)
else (p_num in self.flex_spw_polarization_array)
):
spw_inds = np.append(
spw_inds, np.where(self.flex_spw_polarization_array == p_num)[0]
)
else:
raise ValueError(
"Polarization {p} is not present in the "
"polarization_array".format(p=p)
)
if len(spw_inds) > 0:
# Since this is a flex-pol data set, we need to filter on the freq
# axis instead of the pol axis
pol_inds = None
if freq_inds is None:
freq_inds = np.where(
np.isin(self.flex_spw_id_array, self.spw_array[spw_inds])
)[0]
else:
freq_inds = np.array(freq_inds)
freq_inds = freq_inds[
np.isin(
self.flex_spw_id_array[freq_inds], self.spw_array[spw_inds]
)
]
freq_inds = sorted(freq_inds)
# Trap a corner case here where the frequency and polarization selects
# on a flex-pol data set end up with no actual data being selected.
if len(freq_inds) == 0:
raise ValueError(
"No data matching this polarization and frequency selection "
"in this UVData object."
)
if not uvutils._test_array_constant_spacing(
np.unique(self.flex_spw_polarization_array[spw_inds])
):
warnings.warn(
"Selected polarization values are not evenly spaced. This "
"will make it impossible to write this data out to "
"some file types"
)
else:
pol_inds = np.unique(pol_inds)
if len(pol_inds) > 2:
if not uvutils._test_array_constant_spacing(pol_inds):
warnings.warn(
"Selected polarization values are not evenly spaced. This "
"will make it impossible to write this data out to "
"some file types"
)
pol_inds = pol_inds.tolist()
else:
pol_inds = None
history_update_string += " using pyuvdata."
return blt_inds, freq_inds, pol_inds, history_update_string
def _select_by_index(
self,
blt_inds,
freq_inds,
pol_inds,
history_update_string,
keep_all_metadata=True,
):
"""
Perform select based on indexing arrays.
Parameters
----------
blt_inds : list of int
list of baseline-time indices to keep. Can be None (to keep everything).
freq_inds : list of int
list of frequency indices to keep. Can be None (to keep everything).
pol_inds : list of int
list of polarization indices to keep. Can be None (to keep everything).
history_update_string : str
string to append to the end of the history.
keep_all_metadata : bool
Option to keep metadata for antennas that are no longer in the dataset.
"""
# Create a dictionary that we can loop over an update if need be
ind_dict = {"Nblts": blt_inds, "Nfreqs": freq_inds, "Npols": pol_inds}
# During each loop interval, we pop off an element of this dict, so continue
# until the dict is empty.
while len(ind_dict):
# This is an easy way to grab the first key in the dict
key = next(iter(ind_dict))
# Grab the corresponding index array
ind_arr = ind_dict.pop(key)
# If nothing to select on, bail!
if ind_arr is None:
continue
for param in self:
# For each attribute, if the value is None, then bail, otherwise
# attempt to figure out along which axis ind_arr will apply.
attr = getattr(self, param)
if attr.value is not None:
try:
sel_axis = attr.form.index(key)
except (AttributeError, ValueError):
# If form is not a tuple/list (and therefore not
# array-like), it'll throw an AttributeError, and if key is
# not found in the tuple/list, it'll throw a ValueError.
# In both cases, skip!
continue
if isinstance(attr.value, np.ndarray):
# If we're working with an ndarray, use take to slice along
# the axis that we want to grab from.
attr.value = attr.value.take(ind_arr, axis=sel_axis)
elif isinstance(attr.value, list):
# If this is a list, it _should_ always have 1-dimension.
assert sel_axis == 0, (
"Something is wrong, sel_axis != 0 when selecting on a "
"list, which should not be possible. Please file an "
"issue in our GitHub issue log so that we can fix it."
)
attr.value = [attr.value[idx] for idx in ind_arr]
if key == "Nblts":
# Process post blt-specific selection actions, including counting
# unique times antennas/baselines in the data.
self.Nblts = len(ind_arr)
self.Nbls = len(np.unique(self.baseline_array))
self.Ntimes = len(np.unique(self.time_array))
self.Nants_data = self._calc_nants_data()
if not keep_all_metadata:
# If we are dropping metadata and selecting on blts, then add
# evaluate the antenna axis of all parameters
ind_dict["Nants_telescope"] = np.where(
np.isin(
self.antenna_numbers,
list(set(self.ant_1_array).union(self.ant_2_array)),
)
)[0]
elif key == "Nfreqs":
# Process post freq-specific selection actions
self.Nfreqs = len(ind_arr)
if self.flex_spw:
# If we are dropping channels, then evaluate the spw axis
ind_dict["Nspws"] = np.where(
np.isin(self.spw_array, self.flex_spw_id_array)
)[0]
elif key == "Npols":
# Count the number of unique pols after pol-based selection
self.Npols = len(ind_arr)
elif key == "Nspws":
# Count the number of unique pols after spw-based selection
self.Nspws = len(ind_arr)
elif key == "Nants_telescope":
# Count the number of unique ants after ant-based selection
self.Nants_telescope = len(ind_arr)
# Update the history string
self.history += history_update_string
def select(
self,
antenna_nums=None,
antenna_names=None,
ant_str=None,
bls=None,
frequencies=None,
freq_chans=None,
times=None,
time_range=None,
lsts=None,
lst_range=None,
polarizations=None,
blt_inds=None,
phase_center_ids=None,
inplace=True,
keep_all_metadata=True,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
):
"""
Downselect data to keep on the object along various axes.
Axes that can be selected along include antenna names or numbers,
antenna pairs, frequencies, times and polarizations. Specific
baseline-time indices can also be selected, but this is not commonly
used.
The history attribute on the object will be updated to identify the
operations performed.
Parameters
----------
antenna_nums : array_like of int, optional
The antennas numbers to keep in the object (antenna positions and
names for the removed antennas will be retained unless
`keep_all_metadata` is False). This cannot be provided if
`antenna_names` is also provided.
antenna_names : array_like of str, optional
The antennas names to keep in the object (antenna positions and
names for the removed antennas will be retained unless
`keep_all_metadata` is False). This cannot be provided if
`antenna_nums` is also provided.
bls : list of tuple, optional
A list of antenna number tuples (e.g. [(0, 1), (3, 2)]) or a list of
baseline 3-tuples (e.g. [(0, 1, 'xx'), (2, 3, 'yy')]) specifying baselines
to keep in the object. For length-2 tuples, the ordering of the numbers
within the tuple does not matter. For length-3 tuples, the polarization
string is in the order of the two antennas. If length-3 tuples are
provided, `polarizations` must be None.
ant_str : str, optional
A string containing information about what antenna numbers
and polarizations to keep in the object. Can be 'auto', 'cross', 'all',
or combinations of antenna numbers and polarizations (e.g. '1',
'1_2', '1x_2y'). See tutorial for more examples of valid strings and
the behavior of different forms for ant_str.
If '1x_2y,2y_3y' is passed, both polarizations 'xy' and 'yy' will
be kept for both baselines (1, 2) and (2, 3) to return a valid
pyuvdata object.
An ant_str cannot be passed in addition to any of `antenna_nums`,
`antenna_names`, `bls` args or the `polarizations` parameters,
if it is a ValueError will be raised.
frequencies : array_like of float, optional
The frequencies to keep in the object, each value passed here should
exist in the freq_array.
freq_chans : array_like of int, optional
The frequency channel numbers to keep in the object.
times : array_like of float, optional
The times to keep in the object, each value passed here should
exist in the time_array. Cannot be used with `time_range`.
time_range : array_like of float, optional
The time range in Julian Date to keep in the object, must be
length 2. Some of the times in the object should fall between the
first and last elements. Cannot be used with `times`.
lsts : array_like of float, optional
The local sidereal times (LSTs) to keep in the object, each value
passed here should exist in the lst_array. Cannot be used with
`times`, `time_range`, or `lst_range`.
lst_range : array_like of float, optional
The local sidereal time (LST) range in radians to keep in the
object, must be of length 2. Some of the LSTs in the object should
fall between the first and last elements. If the second value is
smaller than the first, the LSTs are treated as having phase-wrapped
around LST = 2*pi = 0, and the LSTs kept on the object will run from
the larger value, through 0, and end at the smaller value.
polarizations : array_like of int or str, optional
The polarizations numbers to keep in the object, each value passed
here should exist in the polarization_array. If passing strings, the
canonical polarization strings (e.g. "xx", "rr") are supported and if the
`x_orientation` attribute is set, the physical dipole strings
(e.g. "nn", "ee") are also supported.
blt_inds : array_like of int, optional
The baseline-time indices to keep in the object. This is
not commonly used.
phase_center_ids : array_like of int, optional
Phase center IDs to keep on the object (effectively a selection on
baseline-times).
inplace : bool
Option to perform the select directly on self or return a new UVData
object with just the selected data (the default is True, meaning the
select will be done on self).
keep_all_metadata : bool
Option to keep all the metadata associated with antennas, even those
that do do not have data associated with them after the select option.
run_check : bool
Option to check for the existence and proper shapes of parameters
after downselecting data on this object (the default is True,
meaning the check will be run).
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
downselecting data on this object (the default is True, meaning the
acceptable range check will be done).
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
Returns
-------
UVData object or None
None is returned if inplace is True, otherwise a new UVData object
with just the selected data is returned
Raises
------
ValueError
If any of the parameters are set to inappropriate values.
"""
if inplace:
uv_obj = self
else:
uv_obj = self.copy()
# Figure out which index positions we want to hold on to.
(
blt_inds,
freq_inds,
pol_inds,
history_update_string,
) = uv_obj._select_preprocess(
antenna_nums,
antenna_names,
ant_str,
bls,
frequencies,
freq_chans,
times,
time_range,
lsts,
lst_range,
polarizations,
blt_inds,
phase_center_ids,
)
# Call the low-level selection method.
uv_obj._select_by_index(
blt_inds, freq_inds, pol_inds, history_update_string, keep_all_metadata
)
# If we have a flex-pol data set, but we only have one pol, then this doesn't
# need to be flex-pol anymore, and we can drop it here
if uv_obj.flex_spw_polarization_array is not None:
if len(np.unique(uv_obj.flex_spw_polarization_array)) == 1:
uv_obj.remove_flex_pol()
# check if object is uv_object-consistent
if run_check:
uv_obj.check(
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
if not inplace:
return uv_obj
def _harmonize_resample_arrays(
self,
inds_to_keep,
temp_baseline,
temp_id_array,
temp_time,
temp_int_time,
temp_data,
temp_flag,
temp_nsample,
):
"""
Make a self-consistent object after up/downsampling.
This function is called by both upsample_in_time and downsample_in_time.
See those functions for more information about arguments.
"""
self.baseline_array = self.baseline_array[inds_to_keep]
self.time_array = self.time_array[inds_to_keep]
self.integration_time = self.integration_time[inds_to_keep]
self.phase_center_id_array = self.phase_center_id_array[inds_to_keep]
self.baseline_array = np.concatenate((self.baseline_array, temp_baseline))
self.time_array = np.concatenate((self.time_array, temp_time))
self.integration_time = np.concatenate((self.integration_time, temp_int_time))
self.phase_center_id_array = np.concatenate(
(self.phase_center_id_array, temp_id_array)
)
if not self.metadata_only:
self.data_array = self.data_array[inds_to_keep]
self.flag_array = self.flag_array[inds_to_keep]
self.nsample_array = self.nsample_array[inds_to_keep]
# concatenate temp array with existing arrays
self.data_array = np.concatenate((self.data_array, temp_data), axis=0)
self.flag_array = np.concatenate((self.flag_array, temp_flag), axis=0)
self.nsample_array = np.concatenate(
(self.nsample_array, temp_nsample), axis=0
)
# set antenna arrays from baseline_array
self.ant_1_array, self.ant_2_array = self.baseline_to_antnums(
self.baseline_array
)
# update metadata
self.Nblts = self.baseline_array.shape[0]
self.Ntimes = np.unique(self.time_array).size
self.uvw_array = np.zeros((self.Nblts, 3))
# set lst array
self.set_lsts_from_time_array()
# update app source coords to new times
self._set_app_coords_helper()
# temporarily store the metadata only to calculate UVWs correctly
uv_temp = self.copy(metadata_only=True)
# properly calculate the UVWs self-consistently
uv_temp.set_uvws_from_antenna_positions(allow_phasing=True)
self.uvw_array = uv_temp.uvw_array
return
def upsample_in_time(
self,
max_int_time,
blt_order="time",
minor_order="baseline",
summing_correlator_mode=False,
allow_drift=False,
):
"""
Resample to a shorter integration time.
This method will resample a UVData object such that all data samples have
an integration time less than or equal to the `max_int_time`. The new
samples are copied from the original samples (not interpolated).
Parameters
----------
max_int_time : float
Maximum integration time to upsample to in seconds.
blt_order : str
Major baseline ordering for output object. Default is "time". See
the documentation on the `reorder_blts` method for more info.
minor_order : str
Minor baseline ordering for output object. Default is "baseline".
summing_correlator_mode : bool
Option to split the flux from the original samples into the new
samples rather than duplicating the original samples in all the new
samples (undoing an integration rather than an average) to emulate
undoing the behavior in some correlators (e.g. HERA).
allow_drift : bool
Option to allow resampling of unprojected or driftscan data. If this is
False, unprojected or driftscan data will be phased to the ra/dec of zenith
before resampling and then unprojected or rephased to a driftscan after
resampling. Note that resampling unprojected or driftscan phased data may
result in unexpected behavior.
Returns
-------
None
"""
# check that max_int_time is sensible given integration_time
min_integration_time = np.amin(self.integration_time)
sensible_min = 1e-2 * min_integration_time
if max_int_time < sensible_min:
raise ValueError(
"Decreasing the integration time by more than a "
"factor of 100 is not supported. Also note that "
"max_int_time should be in seconds."
)
# figure out where integration_time is longer than max_int_time
inds_to_upsample = np.nonzero(
(self.integration_time > max_int_time)
& (
~np.isclose(
self.integration_time,
max_int_time,
rtol=self._integration_time.tols[0],
atol=self._integration_time.tols[1],
)
)
)
if len(inds_to_upsample[0]) == 0:
warnings.warn(
"All values in the integration_time array are already "
"longer than the value specified; doing nothing."
)
return
unprojected_blts = self._check_for_cat_type("unprojected")
driftscan_blts = self._check_for_cat_type("driftscan")
initial_driftscan = np.any(driftscan_blts)
initial_unprojected = np.any(unprojected_blts)
initial_nphase_ids = np.unique(self.phase_center_id_array).size
if initial_driftscan:
initial_phase_catalog = self.phase_center_catalog.copy()
initial_ids = self.phase_center_id_array
phased = False
if initial_unprojected or initial_driftscan:
if allow_drift:
print(
"Data are unprojected or phased as a driftscan and allow_drift is "
"True, so resampling will be done without phasing."
)
else:
phased = True
# phase to RA/dec of zenith
print(
"Data are unprojected or phased as a driftscan, phasing before "
"resampling."
)
phase_time = Time(self.time_array[0], format="jd")
self.phase_to_time(phase_time)
# we want the ceil of this, but we don't want to get the wrong answer
# when the number is very close to an integer but just barely above it.
temp_new_samples = self.integration_time[inds_to_upsample] / max_int_time
mask_close_floor = np.isclose(temp_new_samples, np.floor(temp_new_samples))
temp_new_samples[mask_close_floor] = np.floor(
temp_new_samples[mask_close_floor]
)
n_new_samples = np.asarray(list(map(int, np.ceil(temp_new_samples))))
temp_Nblts = np.sum(n_new_samples)
temp_baseline = np.zeros((temp_Nblts,), dtype=np.int64)
temp_id_array = np.zeros((temp_Nblts,), dtype=int)
if initial_nphase_ids > 1 and initial_driftscan:
temp_initial_ids = np.zeros((temp_Nblts,), dtype=int)
else:
temp_initial_ids = None
if initial_nphase_ids > 1 and initial_unprojected:
temp_unprojected_blts = np.zeros((temp_Nblts,), dtype=bool)
else:
temp_unprojected_blts = None
temp_time = np.zeros((temp_Nblts,))
temp_int_time = np.zeros((temp_Nblts,))
if self.metadata_only:
temp_data = None
temp_flag = None
temp_nsample = None
else:
if self.future_array_shapes:
temp_data = np.zeros(
(temp_Nblts, self.Nfreqs, self.Npols), dtype=self.data_array.dtype
)
temp_flag = np.zeros(
(temp_Nblts, self.Nfreqs, self.Npols), dtype=self.flag_array.dtype
)
temp_nsample = np.zeros(
(temp_Nblts, self.Nfreqs, self.Npols),
dtype=self.nsample_array.dtype,
)
else:
temp_data = np.zeros(
(temp_Nblts, 1, self.Nfreqs, self.Npols),
dtype=self.data_array.dtype,
)
temp_flag = np.zeros(
(temp_Nblts, 1, self.Nfreqs, self.Npols),
dtype=self.flag_array.dtype,
)
temp_nsample = np.zeros(
(temp_Nblts, 1, self.Nfreqs, self.Npols),
dtype=self.nsample_array.dtype,
)
i0 = 0
for i, ind in enumerate(inds_to_upsample[0]):
i1 = i0 + n_new_samples[i]
temp_baseline[i0:i1] = self.baseline_array[ind]
temp_id_array[i0:i1] = self.phase_center_id_array[ind]
if initial_nphase_ids > 1:
if initial_driftscan:
temp_initial_ids[i0:i1] = initial_ids[ind]
if initial_unprojected:
temp_unprojected_blts[i0:i1] = unprojected_blts[ind]
if not self.metadata_only:
if summing_correlator_mode:
temp_data[i0:i1] = self.data_array[ind] / n_new_samples[i]
else:
temp_data[i0:i1] = self.data_array[ind]
temp_flag[i0:i1] = self.flag_array[ind]
temp_nsample[i0:i1] = self.nsample_array[ind]
# compute the new times of the upsampled array
t0 = self.time_array[ind]
dt = self.integration_time[ind] / n_new_samples[i]
# `offset` will be 0.5 or 1, depending on whether n_new_samples for
# this baseline is even or odd.
offset = 0.5 + 0.5 * (n_new_samples[i] % 2)
n2 = n_new_samples[i] // 2
# Figure out the new center for sample ii taking offset into
# account. Because `t0` is the central time for the original time
# sample, `nt` will range from negative to positive so that
# `temp_time` will result in the central time for the new samples.
# `idx2` tells us how to far to shift and in what direction for each
# new sample.
for ii, idx in enumerate(range(i0, i1)):
idx2 = ii + offset + n2 - n_new_samples[i]
nt = ((t0 * units.day) + (dt * idx2 * units.s)).to(units.day).value
temp_time[idx] = nt
temp_int_time[i0:i1] = dt
i0 = i1
# harmonize temporary arrays with existing ones
inds_to_keep = np.nonzero(self.integration_time <= max_int_time)
self._harmonize_resample_arrays(
inds_to_keep,
temp_baseline,
temp_id_array,
temp_time,
temp_int_time,
temp_data,
temp_flag,
temp_nsample,
)
if phased:
print("Undoing phasing.")
if initial_unprojected:
if initial_nphase_ids > 1:
select_mask = unprojected_blts[inds_to_keep]
select_mask = np.concatenate((select_mask, temp_unprojected_blts))
else:
select_mask = None
self.unproject_phase(select_mask=select_mask)
if initial_driftscan:
if initial_nphase_ids > 1:
initial_ids = initial_ids[inds_to_keep]
initial_ids = np.concatenate((initial_ids, temp_initial_ids))
for cat_id, cat_dict in initial_phase_catalog.items():
if cat_dict["cat_type"] != "driftscan":
continue
if initial_nphase_ids > 1:
select_mask = initial_ids == cat_id
if not np.any(select_mask):
select_mask = None
else:
select_mask = None
self.phase(
cat_dict["cat_lon"],
cat_dict["cat_lat"],
cat_name=cat_dict["cat_name"],
cat_type=cat_dict["cat_type"],
phase_frame=cat_dict["cat_frame"],
epoch=cat_dict["cat_epoch"],
select_mask=select_mask,
)
# reorganize along blt axis
self.reorder_blts(order=blt_order, minor_order=minor_order)
# check the resulting object
self.check()
# add to the history
history_update_string = (
" Upsampled data to {:f} second integration time "
"using pyuvdata.".format(max_int_time)
)
self.history = self.history + history_update_string
return
def downsample_in_time(
self,
min_int_time=None,
n_times_to_avg=None,
blt_order="time",
minor_order="baseline",
keep_ragged=True,
summing_correlator_mode=False,
allow_drift=False,
):
"""
Average to a longer integration time.
This method will average a UVData object either by an integer factor
(by setting `n_times_to_avg`) or by a factor that can differ by
baseline-time sample such that after averaging, the samples have an
integration time greater than or equal to the `min_int_time` (up to the
tolerance on the integration_time).
Note that if the integrations for a baseline do not divide evenly by the
`n_times_to_avg` or into the specified `min_int_time`, the final
integrations for that baseline may have integration times less than
`min_int_time` or be composed of fewer input integrations than `n_times_to_avg`.
This behavior can be controlled with the `keep_ragged` argument.
The new samples are averages of the original samples (not interpolations).
Parameters
----------
min_int_time : float
Minimum integration time to average the UVData integration_time to
in seconds.
n_times_to_avg : int
Number of time integrations to average together.
blt_order : str
Major baseline ordering for output object. Default is "time". See the
documentation on the `reorder_blts` method for more details.
minor_order : str
Minor baseline ordering for output object. Default is "baseline".
keep_ragged : bool
When averaging baselines that do not evenly divide into min_int_time,
or that have a number of integrations that do not evenly divide by
n_times_to_avg, keep_ragged controls whether to keep the (averaged)
integrations corresponding to the remaining samples (keep_ragged=True),
or discard them (keep_ragged=False).
summing_correlator_mode : bool
Option to integrate the flux from the original samples rather than
average the flux to emulate the behavior in some correlators (e.g. HERA).
allow_drift : bool
Option to allow resampling of unprojected or driftscan data. If this is
False, unprojected or driftscan data will be phased to the ra/dec of zenith
before resampling and then unprojected or rephased to a driftscan after
resampling. Note that resampling unprojected or driftscan phased data may
result in unexpected behavior.
Returns
-------
None
Raises
------
ValueError
If neither or both of `min_int_time` and `n_times_to_avg` are set.
If there's only one time on the object.
If `min_int_time` is more than 100 time the max integration time.
If `n_times_to_avg` is not an integer.
"""
if min_int_time is None and n_times_to_avg is None:
raise ValueError("Either min_int_time or n_times_to_avg must be set.")
if min_int_time is not None and n_times_to_avg is not None:
raise ValueError("Only one of min_int_time or n_times_to_avg can be set.")
if self.Ntimes == 1:
raise ValueError("Only one time in this object, cannot downsample.")
if min_int_time is not None:
# check that min_int_time is sensible given integration_time
max_integration_time = np.amax(self.integration_time)
sensible_max = 1e2 * max_integration_time
if min_int_time > sensible_max:
raise ValueError(
"Increasing the integration time by more than a "
"factor of 100 is not supported. Also note that "
"min_int_time should be in seconds."
)
# first figure out where integration_time is shorter than min_int_time
inds_to_downsample = np.nonzero(
(self.integration_time < min_int_time)
& (
~np.isclose(
self.integration_time,
min_int_time,
rtol=self._integration_time.tols[0],
atol=self._integration_time.tols[1],
)
)
)
if len(inds_to_downsample[0]) == 0:
warnings.warn(
"All values in the integration_time array are already "
"longer than the value specified; doing nothing."
)
return
else:
if not isinstance(n_times_to_avg, (int, np.integer)):
raise ValueError("n_times_to_avg must be an integer.")
# If we're going to do actual work, reorder the baselines to ensure time is
# monotonically increasing.
# Default of reorder_blts is baseline major, time minor, which is what we want.
self.reorder_blts()
if min_int_time is not None:
# now re-compute inds_to_downsample, in case things have changed
inds_to_downsample = np.nonzero(
(self.integration_time < min_int_time)
& ~np.isclose(
self.integration_time,
min_int_time,
rtol=self._integration_time.tols[0],
atol=self._integration_time.tols[1],
)
)
bls_to_downsample = np.unique(self.baseline_array[inds_to_downsample])
else:
bls_to_downsample = np.unique(self.baseline_array)
# figure out how many baseline times we'll end up with at the end
n_new_samples = 0
for bl in bls_to_downsample:
bl_inds = np.nonzero(self.baseline_array == bl)[0]
int_times = self.integration_time[bl_inds]
if min_int_time is not None:
running_int_time = 0.0
for itime, int_time in enumerate(int_times):
running_int_time += int_time
over_min_int_time = running_int_time > min_int_time or np.isclose(
running_int_time,
min_int_time,
rtol=self._integration_time.tols[0],
atol=self._integration_time.tols[1],
)
last_sample = itime == len(bl_inds) - 1
# We sum up all the samples found so far if we're over the
# target minimum time, or we've hit the end of the time
# samples for this baseline.
if over_min_int_time or last_sample:
if last_sample and not (over_min_int_time or keep_ragged):
# don't do anything -- implicitly drop these integrations
continue
n_new_samples += 1
running_int_time = 0.0
else:
n_bl_times = self.time_array[bl_inds].size
nsample_temp = np.sum(n_bl_times / n_times_to_avg)
if keep_ragged and not np.isclose(nsample_temp, np.floor(nsample_temp)):
n_new_samples += np.ceil(nsample_temp).astype(int)
else:
n_new_samples += np.floor(nsample_temp).astype(int)
# figure out if there are any time gaps in the data
# meaning that the time differences are larger than the integration times
# time_array is in JD, need to convert to seconds for the diff
dtime = np.ediff1d(self.time_array[bl_inds]) * 24 * 3600
int_times = int_times
if len(np.unique(int_times)) == 1:
# this baseline has all the same integration times
if len(np.unique(dtime)) > 1 and not uvutils._test_array_constant(
dtime, self._integration_time.tols
):
warnings.warn(
"There is a gap in the times of baseline {bl}. "
"The output may include averages across long "
"time gaps.".format(bl=self.baseline_to_antnums(bl))
)
elif not np.isclose(
dtime[0],
int_times[0],
rtol=self._integration_time.tols[0],
atol=self._integration_time.tols[1],
):
warnings.warn(
"The time difference between integrations is "
"not the same as the integration time for "
"baseline {bl}. The output may average across "
"longer time intervals than "
"expected".format(bl=self.baseline_to_antnums(bl))
)
else:
# varying integration times for this baseline, need to be more careful
expected_dtimes = (int_times[:-1] + int_times[1:]) / 2
wh_diff = np.nonzero(~np.isclose(dtime, expected_dtimes))
if wh_diff[0].size > 1:
warnings.warn(
"The time difference between integrations is "
"different than the expected given the "
"integration times for baseline {bl}. The "
"output may include averages across long time "
"gaps.".format(bl=self.baseline_to_antnums(bl))
)
temp_Nblts = n_new_samples
unprojected_blts = self._check_for_cat_type("unprojected")
driftscan_blts = self._check_for_cat_type("driftscan")
initial_driftscan = np.any(driftscan_blts)
initial_unprojected = np.any(unprojected_blts)
initial_nphase_ids = np.unique(self.phase_center_id_array).size
if initial_driftscan:
initial_phase_catalog = self.phase_center_catalog.copy()
initial_ids = self.phase_center_id_array
phased = False
if initial_unprojected or initial_driftscan:
if allow_drift:
print(
"Data are unprojected or phased as a driftscan and allow_drift is "
"True, so resampling will be done without phasing."
)
else:
phased = True
# phase to RA/dec of zenith
print(
"Data are unprojected or phased as a driftscan, phasing before "
"resampling."
)
phase_time = Time(self.time_array[0], format="jd")
self.phase_to_time(phase_time)
# make temporary arrays
temp_baseline = np.zeros((temp_Nblts,), dtype=np.int64)
temp_id_array = np.zeros((temp_Nblts,), dtype=int)
temp_time = np.zeros((temp_Nblts,))
temp_int_time = np.zeros((temp_Nblts,))
if initial_nphase_ids > 1 and initial_driftscan:
temp_initial_ids = np.zeros((temp_Nblts,), dtype=int)
else:
temp_initial_ids = None
if initial_nphase_ids > 1 and initial_unprojected:
temp_unprojected_blts = np.zeros((temp_Nblts,), dtype=bool)
else:
temp_unprojected_blts = None
if self.metadata_only:
temp_data = None
temp_flag = None
temp_nsample = None
else:
if self.future_array_shapes:
temp_data = np.zeros(
(temp_Nblts, self.Nfreqs, self.Npols), dtype=self.data_array.dtype
)
temp_flag = np.zeros(
(temp_Nblts, self.Nfreqs, self.Npols), dtype=self.flag_array.dtype
)
temp_nsample = np.zeros(
(temp_Nblts, self.Nfreqs, self.Npols),
dtype=self.nsample_array.dtype,
)
else:
temp_data = np.zeros(
(temp_Nblts, 1, self.Nfreqs, self.Npols),
dtype=self.data_array.dtype,
)
temp_flag = np.zeros(
(temp_Nblts, 1, self.Nfreqs, self.Npols),
dtype=self.flag_array.dtype,
)
temp_nsample = np.zeros(
(temp_Nblts, 1, self.Nfreqs, self.Npols),
dtype=self.nsample_array.dtype,
)
temp_idx = 0
for bl in bls_to_downsample:
bl_inds = np.nonzero(self.baseline_array == bl)[0]
running_int_time = 0.0
summing_idx = 0
n_sum = 0
for itime, int_time in enumerate(self.integration_time[bl_inds]):
running_int_time += int_time
n_sum += 1
if min_int_time is not None:
over_min_int_time = running_int_time > min_int_time or np.isclose(
running_int_time,
min_int_time,
rtol=self._integration_time.tols[0],
atol=self._integration_time.tols[1],
)
else:
over_min_int_time = n_sum >= n_times_to_avg
last_sample = itime == len(bl_inds) - 1
# We sum up all the samples found so far if we're over the
# target minimum time, or we've hit the end of the time
# samples for this baseline.
if over_min_int_time or last_sample:
if last_sample and not (over_min_int_time or keep_ragged):
# don't do anything -- implicitly drop these integrations
continue
# sum together that number of samples
temp_baseline[temp_idx] = bl
# this might be wrong if some of the constituent times are
# *totally* flagged
averaging_idx = bl_inds[summing_idx : summing_idx + n_sum]
if self.Nphase > 1:
unique_phase_centers = np.unique(
self.phase_center_id_array[averaging_idx]
)
if unique_phase_centers.size > 1:
raise ValueError(
"Multiple phase centers included in a downsampling "
"window. Use `phase` to phase to a single phase center "
"or decrease the `min_int_time` or `n_times_to_avg` "
"parameter to avoid multiple phase centers being "
"included in a downsampling window."
)
temp_id_array[temp_idx] = self.phase_center_id_array[
averaging_idx[0]
]
if initial_nphase_ids > 1:
if initial_unprojected:
unique_unprojected = np.unique(
unprojected_blts[averaging_idx]
)
temp_unprojected_blts[temp_idx] = unique_unprojected[0]
if initial_driftscan:
unique_initial_ids = np.unique(initial_ids[averaging_idx])
temp_initial_ids[temp_idx] = unique_initial_ids[0]
# take potential non-uniformity of integration_time into account
temp_time[temp_idx] = np.sum(
self.time_array[averaging_idx]
* self.integration_time[averaging_idx]
) / np.sum(self.integration_time[averaging_idx])
temp_int_time[temp_idx] = running_int_time
if not self.metadata_only:
# if all inputs are flagged, the flag array should be True,
# otherwise it should be False.
# The sum below will be zero if it's all flagged and
# greater than zero otherwise
# Then we use a test against 0 to turn it into a Boolean
temp_flag[temp_idx] = (
np.sum(~self.flag_array[averaging_idx], axis=0) == 0
)
mask = self.flag_array[averaging_idx]
# need to update mask if a downsampled visibility will
# be flagged so that we don't set it to zero
if (temp_flag[temp_idx]).any():
if self.future_array_shapes:
ax1_inds, ax2_inds = np.nonzero(temp_flag[temp_idx])
mask[:, ax1_inds, ax2_inds] = False
else:
ax1_inds, ax2_inds, ax3_inds = np.nonzero(
temp_flag[temp_idx]
)
mask[:, ax1_inds, ax2_inds, ax3_inds] = False
masked_data = np.ma.masked_array(
self.data_array[averaging_idx], mask=mask
)
# nsample array is the fraction of data that we actually kept,
# relative to the amount that went into the sum or average
nsample_dtype = self.nsample_array.dtype.type
# promote nsample dtype if half-precision
if nsample_dtype is np.float16:
masked_nsample_dtype = np.float32
else:
masked_nsample_dtype = nsample_dtype
masked_nsample = np.ma.masked_array(
self.nsample_array[averaging_idx],
mask=mask,
dtype=masked_nsample_dtype,
)
if self.future_array_shapes:
int_time_arr = self.integration_time[
averaging_idx, np.newaxis, np.newaxis
]
else:
int_time_arr = self.integration_time[
averaging_idx, np.newaxis, np.newaxis, np.newaxis
]
masked_int_time = np.ma.masked_array(
np.ones_like(
self.data_array[averaging_idx],
dtype=self.integration_time.dtype,
)
* int_time_arr,
mask=mask,
)
if summing_correlator_mode:
temp_data[temp_idx] = np.sum(masked_data, axis=0)
else:
# take potential non-uniformity of integration_time
# and nsamples into account
weights = masked_nsample * masked_int_time
weighted_data = masked_data * weights
temp_data[temp_idx] = np.sum(
weighted_data, axis=0
) / np.sum(weights, axis=0)
# output of masked array calculation should be coerced
# to the datatype of temp_nsample (which has the same
# precision as the original nsample_array)
temp_nsample[temp_idx] = np.sum(
masked_nsample * masked_int_time, axis=0
) / np.sum(self.integration_time[averaging_idx])
# increment counters and reset values
temp_idx += 1
summing_idx += n_sum
running_int_time = 0.0
n_sum = 0
# make sure we've populated the right number of baseline-times
assert temp_idx == temp_Nblts, (
"Wrong number of baselines. Got {:d}, expected {:d}. This is a bug, "
"please make an issue at https://github.com/RadioAstronomySoftwareGroup/"
"pyuvdata/issues".format(temp_idx, temp_Nblts)
)
# harmonize temporary arrays with existing ones
if min_int_time is not None:
bls_not_downsampled = set(self.baseline_array) - set(bls_to_downsample)
inds_to_keep = []
for bl in bls_not_downsampled:
inds_to_keep += np.nonzero(self.baseline_array == bl)[0].tolist()
inds_to_keep = np.array(inds_to_keep, dtype=np.int64)
else:
inds_to_keep = np.array([], dtype=bool)
self._harmonize_resample_arrays(
inds_to_keep,
temp_baseline,
temp_id_array,
temp_time,
temp_int_time,
temp_data,
temp_flag,
temp_nsample,
)
if phased:
print("Undoing phasing.")
if initial_unprojected:
if initial_nphase_ids > 1:
select_mask = unprojected_blts[inds_to_keep]
select_mask = np.concatenate((select_mask, temp_unprojected_blts))
else:
select_mask = None
self.unproject_phase(select_mask=select_mask)
if initial_driftscan:
if initial_nphase_ids > 1:
initial_ids = initial_ids[inds_to_keep]
initial_ids = np.concatenate((initial_ids, temp_initial_ids))
for cat_id, cat_dict in initial_phase_catalog.items():
if cat_dict["cat_type"] != "driftscan":
continue
if initial_nphase_ids > 1:
select_mask = initial_ids == cat_id
if not np.any(select_mask):
select_mask = None
else:
select_mask = None
self.phase(
cat_dict["cat_lon"],
cat_dict["cat_lat"],
cat_name=cat_dict["cat_name"],
cat_type=cat_dict["cat_type"],
phase_frame=cat_dict["cat_frame"],
epoch=cat_dict["cat_epoch"],
select_mask=select_mask,
)
# reorganize along blt axis
self.reorder_blts(order=blt_order, minor_order=minor_order)
# check the resulting object
self.check()
# add to the history
if min_int_time is not None:
history_update_string = (
" Downsampled data to {:f} second integration "
"time using pyuvdata.".format(min_int_time)
)
else:
history_update_string = (
" Downsampled data by a factor of {} in "
"time using pyuvdata.".format(n_times_to_avg)
)
self.history = self.history + history_update_string
return
def resample_in_time(
self,
target_time,
only_downsample=False,
only_upsample=False,
blt_order="time",
minor_order="baseline",
keep_ragged=True,
summing_correlator_mode=False,
allow_drift=False,
):
"""
Intelligently upsample or downsample a UVData object to the target time.
Parameters
----------
target_time : float
The target integration time to resample to, in seconds.
only_downsample : bool
Option to only call bda_downsample.
only_upsample : bool
Option to only call bda_upsample.
blt_order : str
Major baseline ordering for output object. Default is "time". See the
documentation on the `reorder_blts` method for more details.
minor_order : str
Minor baseline ordering for output object. Default is "baseline".
keep_ragged : bool
When averaging baselines that do not evenly divide into min_int_time,
keep_ragged controls whether to keep the (summed) integrations
corresponding to the remaining samples (keep_ragged=True), or
discard them (keep_ragged=False). Note this option only applies to the
`bda_downsample` method.
summing_correlator_mode : bool
Option to integrate or split the flux from the original samples
rather than average or duplicate the flux from the original samples
to emulate the behavior in some correlators (e.g. HERA).
allow_drift : bool
Option to allow resampling of unprojected or driftscan data. If this is
False, unprojected or driftscan data will be phased to the ra/dec of zenith
before resampling and then unprojected or rephased to a driftscan after
resampling. Note that resampling unprojected or driftscan phased data may
result in unexpected behavior.
Returns
-------
None
"""
# figure out integration times relative to target time
min_int_time = np.amin(self.integration_time)
max_int_time = np.amax(self.integration_time)
if int(np.floor(target_time / min_int_time)) >= 2 and not only_upsample:
downsample = True
else:
downsample = False
if int(np.floor(max_int_time / target_time)) >= 2 and not only_downsample:
upsample = True
else:
upsample = False
if not downsample and not upsample:
warnings.warn(
"No resampling will be done because target time is not "
"a factor of 2 or more off from integration_time. To "
"force resampling set only_upsample or only_downsample "
"keywords or call upsample_in_time or downsample_in_time."
)
return
if downsample:
self.downsample_in_time(
target_time,
blt_order=blt_order,
minor_order=minor_order,
keep_ragged=keep_ragged,
summing_correlator_mode=summing_correlator_mode,
allow_drift=allow_drift,
)
if upsample:
self.upsample_in_time(
target_time,
blt_order=blt_order,
minor_order=minor_order,
summing_correlator_mode=summing_correlator_mode,
allow_drift=allow_drift,
)
return
def frequency_average(
self, n_chan_to_avg, summing_correlator_mode=False, propagate_flags=False
):
"""
Average in frequency.
Does a simple average over an integer number of input channels, leaving
flagged samples out of the average.
In the future, this method will support non-equally spaced channels
and varying channel widths. It will also support setting the frequency
to the true mean of the averaged non-flagged frequencies rather than
the simple mean of the input channel frequencies. For now it does not.
Parameters
----------
n_chan_to_avg : int
Number of channels to average together. If Nfreqs does not divide
evenly by this number, the frequencies at the end of the freq_array
will be dropped to make it evenly divisable. To control which
frequencies are removed, use select before calling this method.
summing_correlator_mode : bool
Option to integrate or split the flux from the original samples
rather than average or duplicate the flux from the original samples
to emulate the behavior in some correlators (e.g. HERA).
propagate_flags: bool
Option to flag an averaged entry even if some of its contributors
are not flagged. The averaged result will still leave the flagged
samples out of the average, except when all contributors are
flagged.
"""
if self.flex_spw:
raise NotImplementedError(
"Frequency averaging not (yet) available for flexible spectral windows"
)
self._check_freq_spacing()
n_final_chan = int(np.floor(self.Nfreqs / n_chan_to_avg))
nfreq_mod_navg = self.Nfreqs % n_chan_to_avg
if nfreq_mod_navg != 0:
# not an even number of final channels
warnings.warn(
"Nfreqs does not divide by `n_chan_to_avg` evenly. "
"The final {} frequencies will be excluded, to "
"control which frequencies to exclude, use a "
"select to control.".format(nfreq_mod_navg)
)
chan_to_keep = np.arange(n_final_chan * n_chan_to_avg)
self.select(freq_chans=chan_to_keep)
if self.future_array_shapes:
self.freq_array = self.freq_array.reshape(
(n_final_chan, n_chan_to_avg)
).mean(axis=1)
self.channel_width = self.channel_width.reshape(
(n_final_chan, n_chan_to_avg)
).sum(axis=1)
else:
self.freq_array = self.freq_array.reshape(
(1, n_final_chan, n_chan_to_avg)
).mean(axis=2)
self.channel_width = self.channel_width * n_chan_to_avg
self.Nfreqs = n_final_chan
if self.eq_coeffs is not None:
eq_coeff_diff = np.diff(self.eq_coeffs, axis=1)
if np.abs(np.max(eq_coeff_diff)) > 0:
warnings.warn(
"eq_coeffs vary by frequency. They should be "
"applied to the data using `remove_eq_coeffs` "
"before frequency averaging."
)
self.eq_coeffs = self.eq_coeffs.reshape(
(self.Nants_telescope, n_final_chan, n_chan_to_avg)
).mean(axis=2)
if not self.metadata_only:
if self.future_array_shapes:
shape_tuple = (self.Nblts, n_final_chan, n_chan_to_avg, self.Npols)
else:
shape_tuple = (self.Nblts, 1, n_final_chan, n_chan_to_avg, self.Npols)
mask = self.flag_array.reshape(shape_tuple)
if propagate_flags:
# if any contributors are flagged, the result should be flagged
if self.future_array_shapes:
self.flag_array = np.any(
self.flag_array.reshape(shape_tuple), axis=2
)
else:
self.flag_array = np.any(
self.flag_array.reshape(shape_tuple), axis=3
)
else:
# if all inputs are flagged, the flag array should be True,
# otherwise it should be False.
# The sum below will be zero if it's all flagged and
# greater than zero otherwise
# Then we use a test against 0 to turn it into a Boolean
if self.future_array_shapes:
self.flag_array = (
np.sum(~self.flag_array.reshape(shape_tuple), axis=2) == 0
)
else:
self.flag_array = (
np.sum(~self.flag_array.reshape(shape_tuple), axis=3) == 0
)
# need to update mask if a downsampled visibility will be flagged
# so that we don't set it to zero
for n_chan in np.arange(n_final_chan):
if self.future_array_shapes:
if (self.flag_array[:, n_chan]).any():
ax0_inds, ax2_inds = np.nonzero(self.flag_array[:, n_chan, :])
# Only if all entries are masked
# May not happen due to propagate_flags keyword
# mask should be left alone otherwise
if np.all(mask[ax0_inds, n_chan, :, ax2_inds]):
mask[ax0_inds, n_chan, :, ax2_inds] = False
else:
if (self.flag_array[:, :, n_chan]).any():
ax0_inds, ax1_inds, ax3_inds = np.nonzero(
self.flag_array[:, :, n_chan, :]
)
# Only if all entries are masked
# May not happen due to propagate_flags keyword
# mask should be left alone otherwise
if np.all(mask[ax0_inds, ax1_inds, n_chan, :, ax3_inds]):
mask[ax0_inds, ax1_inds, n_chan, :, ax3_inds] = False
masked_data = np.ma.masked_array(
self.data_array.reshape(shape_tuple), mask=mask
)
self.nsample_array = self.nsample_array.reshape(shape_tuple)
# promote nsample dtype if half-precision
nsample_dtype = self.nsample_array.dtype.type
if nsample_dtype is np.float16:
masked_nsample_dtype = np.float32
else:
masked_nsample_dtype = nsample_dtype
masked_nsample = np.ma.masked_array(
self.nsample_array, mask=mask, dtype=masked_nsample_dtype
)
if summing_correlator_mode:
if self.future_array_shapes:
self.data_array = np.sum(masked_data, axis=2).data
else:
self.data_array = np.sum(masked_data, axis=3).data
else:
# need to weight by the nsample_array
if self.future_array_shapes:
self.data_array = (
np.sum(masked_data * masked_nsample, axis=2)
/ np.sum(masked_nsample, axis=2)
).data
else:
self.data_array = (
np.sum(masked_data * masked_nsample, axis=3)
/ np.sum(masked_nsample, axis=3)
).data
# nsample array is the fraction of data that we actually kept,
# relative to the amount that went into the sum or average.
# Need to take care to return precision back to original value.
if self.future_array_shapes:
self.nsample_array = (
np.sum(masked_nsample, axis=2) / float(n_chan_to_avg)
).data.astype(nsample_dtype)
else:
self.nsample_array = (
np.sum(masked_nsample, axis=3) / float(n_chan_to_avg)
).data.astype(nsample_dtype)
def get_redundancies(
self,
tol=1.0,
use_antpos=False,
include_conjugates=False,
include_autos=True,
conjugate_bls=False,
):
"""
Get redundant baselines to a given tolerance.
This can be used to identify redundant baselines present in the data,
or find all possible redundant baselines given the antenna positions.
Parameters
----------
tol : float
Redundancy tolerance in meters (default 1m).
use_antpos : bool
Use antenna positions to find all possible redundant groups for this
telescope (default False).
The returned baselines are in the 'u>0' convention.
include_conjugates : bool
Option to include baselines that are redundant under conjugation.
Only used if use_antpos is False.
include_autos : bool
Option to include autocorrelations in the full redundancy list.
Only used if use_antpos is True.
conjugate_bls : bool
If using antenna positions, this will conjugate baselines on this
object to correspond with those in the returned groups.
Returns
-------
baseline_groups : list of lists of int
List of lists of redundant baseline numbers
vec_bin_centers : list of ndarray of float
List of vectors describing redundant group uvw centers
lengths : list of float
List of redundant group baseline lengths in meters
conjugates : list of int, or None, optional
List of indices for baselines that must be conjugated to fit into their
redundant groups.
Will return None if use_antpos is True and include_conjugates is True
Only returned if include_conjugates is True
Notes
-----
If use_antpos is set, then this function will find all redundant baseline groups
for this telescope, under the u>0 antenna ordering convention.
If use_antpos is not set, this function will look for redundant groups
in the data.
"""
if use_antpos:
antpos, numbers = self.get_ENU_antpos(center=False)
result = uvutils.get_antenna_redundancies(
numbers, antpos, tol=tol, include_autos=include_autos
)
if conjugate_bls:
self.conjugate_bls(convention="u>0", uvw_tol=tol)
if include_conjugates:
result = result + (None,)
return result
_, unique_inds = np.unique(self.baseline_array, return_index=True)
unique_inds.sort()
baseline_vecs = np.take(self.uvw_array, unique_inds, axis=0)
baselines = np.take(self.baseline_array, unique_inds)
return uvutils.get_baseline_redundancies(
baselines, baseline_vecs, tol=tol, include_conjugates=include_conjugates
)
def compress_by_redundancy(
self, method="select", tol=1.0, inplace=True, keep_all_metadata=True
):
"""
Downselect or average to only have one baseline per redundant group.
Either select the first baseline in the redundant group or average over
the baselines in the redundant group. When averaging, only unflagged data are
averaged and the nsample_array reflects the amount of unflagged data that was
averaged over. In the case that all the data for a particular visibility to be
averaged is flagged, all the flagged data is averaged (with an nsample value
that represents all the data) but the flag array is set to True for that
visibility.
Parameters
----------
tol : float
Redundancy tolerance in meters, default is 1.0 corresponding to 1 meter.
This specifies what tolerance to use when identifying baselines as
redundant.
method : str
Options are "select", which just keeps the first baseline in each
redundant group or "average" which averages over the baselines in each
redundant group and assigns the average to the first baseline in the group.
inplace : bool
Option to do selection on current object.
keep_all_metadata : bool
Option to keep all the metadata associated with antennas,
even those that do not remain after the select option.
Returns
-------
UVData object or None
if inplace is False, return the compressed UVData object
"""
allowed_methods = ["select", "average"]
if method not in allowed_methods:
raise ValueError(f"method must be one of {allowed_methods}")
red_gps, centers, lengths, conjugates = self.get_redundancies(
tol, include_conjugates=True
)
bl_ants = [self.baseline_to_antnums(gp[0]) for gp in red_gps]
if method == "average":
# do a metadata only select to get all the metadata right
new_obj = self.copy(metadata_only=True)
new_obj.select(bls=bl_ants, keep_all_metadata=keep_all_metadata)
if not self.metadata_only:
# If the baseline is in the "conjugated" list, we will need to
# fix the conjugation on assignment (since the visibilities are
# tabulated assuming that the baseline position is on the opposite
# side of the uvw origin).
fix_conj = np.isin(new_obj.baseline_array, conjugates)
# initalize the data like arrays
if new_obj.future_array_shapes:
temp_data_array = np.zeros(
(new_obj.Nblts, new_obj.Nfreqs, new_obj.Npols),
dtype=self.data_array.dtype,
)
temp_nsample_array = np.zeros(
(new_obj.Nblts, new_obj.Nfreqs, new_obj.Npols),
dtype=self.nsample_array.dtype,
)
temp_flag_array = np.zeros(
(new_obj.Nblts, new_obj.Nfreqs, new_obj.Npols),
dtype=self.flag_array.dtype,
)
else:
temp_data_array = np.zeros(
(new_obj.Nblts, 1, new_obj.Nfreqs, new_obj.Npols),
dtype=self.data_array.dtype,
)
temp_nsample_array = np.zeros(
(new_obj.Nblts, 1, new_obj.Nfreqs, new_obj.Npols),
dtype=self.nsample_array.dtype,
)
temp_flag_array = np.zeros(
(new_obj.Nblts, 1, new_obj.Nfreqs, new_obj.Npols),
dtype=self.flag_array.dtype,
)
for grp_ind, group in enumerate(red_gps):
if len(conjugates) > 0:
conj_group = set(group).intersection(conjugates)
reg_group = list(set(group) - conj_group)
conj_group = list(conj_group)
else:
reg_group = group
conj_group = []
group_times = []
group_inds = []
conj_group_inds = []
conj_group_times = []
for bl in reg_group:
bl_inds = np.where(self.baseline_array == bl)[0]
group_inds.extend(bl_inds)
group_times.extend(self.time_array[bl_inds])
for bl in conj_group:
bl_inds = np.where(self.baseline_array == bl)[0]
conj_group_inds.extend(bl_inds)
conj_group_times.extend(self.time_array[bl_inds])
group_inds = np.array(group_inds, dtype=np.int64)
conj_group_inds = np.array(conj_group_inds, dtype=np.int64)
# now we have to figure out which times are the same to a tolerance
# so we can average over them.
time_inds = np.arange(len(group_times + conj_group_times))
time_gps = uvutils.find_clusters(
time_inds,
np.array(group_times + conj_group_times),
self._time_array.tols[1],
)
# average over the same times
obj_bl = bl_ants[grp_ind]
obj_inds = new_obj._key2inds(obj_bl)[0]
obj_times = new_obj.time_array[obj_inds]
for gp in time_gps:
# Note that this average time is just used for identifying the
# index to use for the blt axis on the averaged data set.
# We do not update the actual time on that data set because it can
# result in confusing behavior -- small numerical rounding errors
# can result in many more unique times in the final data set than
# in the initial data set.
avg_time = np.average(np.array(group_times + conj_group_times)[gp])
obj_time_ind = np.where(
np.abs(obj_times - avg_time) < self._time_array.tols[1]
)[0]
if obj_time_ind.size == 1:
this_obj_ind = obj_inds[obj_time_ind[0]]
else:
warnings.warn(
"Index baseline in the redundant group does not "
"have all the times, compressed object will be "
"missing those times."
)
continue
# time_ind contains indices for both regular and conjugated bls
# because we needed to group them together in time.
# The regular ones are first and extend the length of group_times,
# so we use that to split them back up.
regular_orientation = np.array(
[time_ind for time_ind in gp if time_ind < len(group_times)],
dtype=np.int64,
)
regular_inds = group_inds[np.array(regular_orientation)]
conj_orientation = np.array(
[
time_ind - len(group_times)
for time_ind in gp
if time_ind >= len(group_times)
],
dtype=np.int64,
)
conj_inds = conj_group_inds[np.array(conj_orientation)]
# check that the integration times are all the same
int_times = np.concatenate(
(
self.integration_time[regular_inds],
self.integration_time[conj_inds],
)
)
if not np.all(
np.abs(int_times - new_obj.integration_time[obj_time_ind])
< new_obj._integration_time.tols[1]
):
warnings.warn(
"Integrations times are not identical in a redundant "
"group. Averaging anyway but this may cause unexpected "
"behavior."
)
if not self.metadata_only:
vis_to_avg = np.concatenate(
(
self.data_array[regular_inds],
np.conj(self.data_array[conj_inds]),
)
)
nsample_to_avg = np.concatenate(
(
self.nsample_array[regular_inds],
self.nsample_array[conj_inds],
)
)
flags_to_avg = np.concatenate(
(self.flag_array[regular_inds], self.flag_array[conj_inds])
)
# if all data is flagged, average it all as if it were not
if np.all(flags_to_avg):
mask = np.zeros_like(flags_to_avg)
else:
mask = flags_to_avg
vis_to_avg = np.ma.masked_array(vis_to_avg, mask=mask)
nsample_to_avg = np.ma.masked_array(nsample_to_avg, mask=mask)
avg_vis = np.ma.average(
vis_to_avg, weights=nsample_to_avg, axis=0
)
avg_nsample = np.sum(nsample_to_avg, axis=0)
avg_flag = np.all(flags_to_avg, axis=0)
temp_data_array[this_obj_ind] = avg_vis
# If we need to flip the conjugation, to that on the
# value assignment here.
if fix_conj[this_obj_ind]:
temp_data_array[this_obj_ind] = np.conj(avg_vis)
else:
temp_data_array[this_obj_ind] = avg_vis
temp_nsample_array[this_obj_ind] = avg_nsample
temp_flag_array[this_obj_ind] = avg_flag
if inplace:
self.select(bls=bl_ants, keep_all_metadata=keep_all_metadata)
if not self.metadata_only:
self.data_array = temp_data_array
self.nsample_array = temp_nsample_array
self.flag_array = temp_flag_array
self.check()
return
else:
if not self.metadata_only:
new_obj.data_array = temp_data_array
new_obj.nsample_array = temp_nsample_array
new_obj.flag_array = temp_flag_array
new_obj.check()
return new_obj
else:
return self.select(
bls=bl_ants, inplace=inplace, keep_all_metadata=keep_all_metadata
)
def inflate_by_redundancy(self, tol=1.0, blt_order="time", blt_minor_order=None):
"""
Expand data to full size, copying data among redundant baselines.
Note that this method conjugates baselines to the 'u>0' convention in order
to inflate the redundancies.
Parameters
----------
tol : float
Redundancy tolerance in meters, default is 1.0 corresponding to 1 meter.
blt_order : str
string specifying primary order along the blt axis (see `reorder_blts`)
blt_minor_order : str
string specifying minor order along the blt axis (see `reorder_blts`)
"""
self.conjugate_bls(convention="u>0")
red_gps, centers, lengths = self.get_redundancies(
tol=tol, use_antpos=True, conjugate_bls=True
)
# Stack redundant groups into one array.
group_index, bl_array_full = zip(
*[(i, bl) for i, gp in enumerate(red_gps) for bl in gp]
)
# TODO should be an assert that each baseline only ends up in one group
# Map group index to blt indices in the compressed array.
bl_array_comp = self.baseline_array
uniq_bl = np.unique(bl_array_comp)
group_blti = {}
Nblts_full = 0
for i, gp in enumerate(red_gps):
for bl in gp:
# First baseline in the group that is also in the compressed
# baseline array.
if bl in uniq_bl:
group_blti[i] = np.where(bl == bl_array_comp)[0]
# add number of blts for this group
Nblts_full += group_blti[i].size * len(gp)
break
blt_map = np.zeros(Nblts_full, dtype=int)
full_baselines = np.zeros(Nblts_full, dtype=int)
missing = []
counter = 0
for bl, gi in zip(bl_array_full, group_index):
try:
# this makes the time the fastest axis
blt_map[counter : counter + group_blti[gi].size] = group_blti[gi]
full_baselines[counter : counter + group_blti[gi].size] = bl
counter += group_blti[gi].size
except KeyError:
missing.append(bl)
pass
if np.any(missing):
warnings.warn("Missing some redundant groups. Filling in available data.")
# blt_map is an index array mapping compressed blti indices to uncompressed
self.data_array = self.data_array[blt_map, ...]
self.nsample_array = self.nsample_array[blt_map, ...]
self.flag_array = self.flag_array[blt_map, ...]
self.time_array = self.time_array[blt_map]
self.lst_array = self.lst_array[blt_map]
self.integration_time = self.integration_time[blt_map]
self.uvw_array = self.uvw_array[blt_map, ...]
self.baseline_array = full_baselines
self.ant_1_array, self.ant_2_array = self.baseline_to_antnums(
self.baseline_array
)
self.Nants_data = self._calc_nants_data()
self.Nbls = np.unique(self.baseline_array).size
self.Nblts = Nblts_full
self.phase_center_app_ra = self.phase_center_app_ra[blt_map]
self.phase_center_app_dec = self.phase_center_app_dec[blt_map]
self.phase_center_frame_pa = self.phase_center_frame_pa[blt_map]
self.phase_center_id_array = self.phase_center_id_array[blt_map]
self.reorder_blts(order=blt_order, minor_order=blt_minor_order)
self.check()
def _convert_from_filetype(self, other):
"""
Convert from a file-type specific object to a UVData object.
Used in reads.
Parameters
----------
other : object that inherits from UVData
File type specific object to convert to UVData
"""
for p in other:
param = getattr(other, p)
setattr(self, p, param)
def _convert_to_filetype(self, filetype):
"""
Convert from a UVData object to a file-type specific object.
Used in writes.
Parameters
----------
filetype : str
Specifies what file type object to convert to. Options are: 'uvfits',
'fhd', 'miriad', 'uvh5', 'mir', 'ms'
Raises
------
ValueError
if filetype is not a known type
"""
if filetype == "uvfits":
from . import uvfits
other_obj = uvfits.UVFITS()
elif filetype == "fhd":
from . import fhd
other_obj = fhd.FHD()
elif filetype == "miriad":
from . import miriad
other_obj = miriad.Miriad()
elif filetype == "uvh5":
from . import uvh5
other_obj = uvh5.UVH5()
elif filetype == "mir":
from . import mir
other_obj = mir.Mir()
elif filetype == "ms":
from . import ms
other_obj = ms.MS()
else:
raise ValueError("filetype must be uvfits, mir, miriad, ms, fhd, or uvh5")
for p in self:
param = getattr(self, p)
setattr(other_obj, p, param)
return other_obj
def read_fhd(
self,
filelist,
use_model=False,
read_data=True,
background_lsts=True,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
check_autos=True,
fix_autos=True,
):
"""
Read in data from a list of FHD files.
Parameters
----------
filelist : array_like of str
The list/array of FHD save files to read from. Must include at
least one polarization file, a params file, a layout file and a flag file.
An obs file is also required if `read_data` is False.
use_model : bool
Option to read in the model visibilities rather than the dirty
visibilities (the default is False, meaning the dirty visibilities
will be read).
read_data : bool
Read in the visibility, nsample and flag data. If set to False, only
the metadata will be read in. Setting read_data to False results in
a metadata only object. If read_data is False, an obs file must be
included in the filelist.
background_lsts : bool
When set to True, the lst_array is calculated in a background thread.
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run).
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done).
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
Raises
------
ValueError
If required files are missing or multiple files for any polarization
are included in filelist.
If there is no recognized key for visibility weights in the flags_file.
"""
from . import fhd
if isinstance(filelist[0], (list, tuple, np.ndarray)):
raise ValueError(
"Reading multiple files from class specific "
"read functions is no longer supported. "
"Use the generic `uvdata.read` function instead."
)
fhd_obj = fhd.FHD()
fhd_obj.read_fhd(
filelist,
use_model=use_model,
background_lsts=background_lsts,
read_data=read_data,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
self._convert_from_filetype(fhd_obj)
del fhd_obj
def read_mir(
self,
filepath,
isource=None,
irec=None,
isb=None,
corrchunk=None,
pseudo_cont=False,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
allow_flex_pol=True,
check_autos=True,
fix_autos=True,
rechunk=None,
):
"""
Read in data from an SMA MIR file.
Note that with the exception of filepath, the rest of the parameters are
used to sub-select a range of data.
Parameters
----------
filepath : str
The file path to the MIR folder to read from.
isource : int
Source code for MIR dataset. The default is None, which selects all sources.
irec : int
Receiver code for MIR dataset. The default is None, which selects all
receivers.
isb : int
Sideband code for MIR dataset. The default is None, which selects both
sidebands.
corrchunk : int
Correlator chunk code for MIR dataset. The default is None, which selects
all chunks.
pseudo_cont : boolean
Read in only the pseudo-continuuum values. Default is False.
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run). Ignored if read_data is False.
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
Ignored if read_data is False.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done). Ignored if read_data is False.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
allow_flex_pol : bool
If only one polarization per spectral window is read (and the polarization
differs from window to window), allow for the `UVData` object to use
"flexible polarization", which compresses the polarization-axis of various
attributes to be of length 1, sets the `flex_spw_polarization_array`
attribute to define the polarization per spectral window. Default is True.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
"""
from . import mir
mir_obj = mir.Mir()
mir_obj.read_mir(
filepath,
isource=isource,
irec=irec,
isb=isb,
corrchunk=corrchunk,
pseudo_cont=pseudo_cont,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
allow_flex_pol=allow_flex_pol,
check_autos=check_autos,
fix_autos=fix_autos,
rechunk=rechunk,
)
self._convert_from_filetype(mir_obj)
del mir_obj
def read_miriad(
self,
filepath,
antenna_nums=None,
ant_str=None,
bls=None,
polarizations=None,
time_range=None,
read_data=True,
phase_type=None,
projected=None,
correct_lat_lon=True,
background_lsts=True,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
calc_lst=True,
fix_old_proj=False,
fix_use_ant_pos=True,
check_autos=True,
fix_autos=True,
):
"""
Read in data from a miriad file.
Parameters
----------
filepath : str
The miriad root directory to read from.
antenna_nums : array_like of int, optional
The antennas numbers to read into the object.
bls : list of tuple, optional
A list of antenna number tuples (e.g. [(0, 1), (3, 2)]) or a list of
baseline 3-tuples (e.g. [(0, 1, 'xx'), (2, 3, 'yy')]) specifying baselines
to include when reading data into the object. For length-2 tuples,
the ordering of the numbers within the tuple does not matter. For
length-3 tuples, the polarization string is in the order of the two
antennas. If length-3 tuples are provided, `polarizations` must be
None.
ant_str : str, optional
A string containing information about what antenna numbers
and polarizations to include when reading data into the object.
Can be 'auto', 'cross', 'all', or combinations of antenna numbers
and polarizations (e.g. '1', '1_2', '1x_2y'). See tutorial for more
examples of valid strings and the behavior of different forms for ant_str.
If '1x_2y,2y_3y' is passed, both polarizations 'xy' and 'yy' will
be kept for both baselines (1, 2) and (2, 3) to return a valid
pyuvdata object.
An ant_str cannot be passed in addition to any of `antenna_nums`,
`bls` or `polarizations` parameters, if it is a ValueError will be raised.
polarizations : array_like of int or str, optional
List of polarization integers or strings to read-in. e.g. ['xx', 'yy', ...]
time_range : list of float, optional
len-2 list containing min and max range of times in Julian Date to
include when reading data into the object. e.g. [2458115.20, 2458115.40]
read_data : bool
Read in the visibility and flag data. If set to false,
only the metadata will be read in. Setting read_data to False
results in an incompletely defined object (check will not pass).
phase_type : str, optional
Deprecated, use the `projected` parameter insted. Option to specify the
phasing status of the data. Options are 'drift', 'phased' or None. 'drift'
is the same as setting the projected parameter to False, 'phased' is the
same as setting the projected parameter to True. Ignored if `projected`
is set.
projected : bool or None
Option to force the dataset to be labelled as projected or unprojected
regardless of the evidence in the file. The default is None which means that
the projection will be set based on the file contents. Be careful setting
this keyword unless you are confident about the contents of the file.
correct_lat_lon : bool
Option to update the latitude and longitude from the known_telescopes
list if the altitude is missing.
background_lsts : bool
When set to True, the lst_array is calculated in a background thread.
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run). Ignored if read_data is False.
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
Ignored if read_data is False.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done). Ignored if read_data is False.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
calc_lst : bool
Recalculate the LST values that are present within the file, useful in
cases where the "online" calculate values have precision or value errors.
Default is True.
fix_old_proj : bool
Applies a fix to uvw-coordinates and phasing, assuming that the old `phase`
method was used prior to writing the data, which had errors of the order of
one part in 1e4 - 1e5. See the phasing memo for more details. Default is
False.
fix_use_ant_pos : bool
If setting `fix_old_proj` to True, use the antenna positions to derive the
correct uvw-coordinates rather than using the baseline vectors. Default is
True.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
Raises
------
IOError
If root file directory doesn't exist.
ValueError
If incompatible select keywords are set (e.g. `ant_str` with other
antenna selectors, `times` and `time_range`) or select keywords
exclude all data or if keywords are set to the wrong type.
If the data are multi source or have multiple
spectral windows.
If the metadata are not internally consistent.
"""
from . import miriad
if isinstance(filepath, (list, tuple, np.ndarray)):
raise ValueError(
"Reading multiple files from class specific "
"read functions is no longer supported. "
"Use the generic `uvdata.read` function instead."
)
miriad_obj = miriad.Miriad()
miriad_obj.read_miriad(
filepath,
correct_lat_lon=correct_lat_lon,
read_data=read_data,
phase_type=phase_type,
projected=projected,
antenna_nums=antenna_nums,
ant_str=ant_str,
bls=bls,
polarizations=polarizations,
time_range=time_range,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
calc_lst=calc_lst,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
check_autos=check_autos,
fix_autos=fix_autos,
)
self._convert_from_filetype(miriad_obj)
del miriad_obj
def read_ms(
self,
filepath,
data_column="DATA",
pol_order="AIPS",
background_lsts=True,
ignore_single_chan=True,
raise_error=True,
read_weights=True,
allow_flex_pol=True,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
check_autos=True,
fix_autos=True,
):
"""
Read in data from a measurement set.
Parameters
----------
filepath : str
The measurement set root directory to read from.
data_column : str
name of CASA data column to read into data_array. Options are:
'DATA', 'MODEL', or 'CORRECTED_DATA'
pol_order : str
Option to specify polarizations order convention, options are
'CASA' or 'AIPS'.
background_lsts : bool
When set to True, the lst_array is calculated in a background thread.
ignore_single_chan : bool
Some measurement sets (e.g., those from ALMA) use single channel spectral
windows for recording pseudo-continuum channels or storing other metadata
in the track when the telescopes are not on source. Because of the way
the object is strutured (where all spectral windows are assumed to be
simultaneously recorded), this can significantly inflate the size and memory
footprint of UVData objects. By default, single channel windows are ignored
to avoid this issue, although they can be included if setting this parameter
equal to True.
raise_error : bool
The measurement set format allows for different spectral windows and
polarizations to have different metdata for the same time-baseline
combination, but UVData objects do not. If detected, by default the reader
will throw an error. However, if set to False, the reader will simply give
a warning, and will use the first value read in the file as the "correct"
metadata in the UVData object.
read_weights : bool
Read in the weights from the MS file, default is True. If false, the method
will set the `nsamples_array` to the same uniform value (namely 1.0).
allow_flex_pol : bool
If only one polarization per spectral window is read (and the polarization
differs from window to window), allow for the `UVData` object to use
"flexible polarization", which compresses the polarization-axis of various
attributes to be of length 1, sets the `flex_spw_polarization_array`
attribute to define the polarization per spectral window. Default is True.
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run).
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done).
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
Raises
------
IOError
If root file directory doesn't exist.
ValueError
If the `data_column` is not set to an allowed value.
If the data are have multiple subarrays or are multi source or have
multiple spectral windows.
If the data have multiple data description ID values.
"""
if isinstance(filepath, (list, tuple, np.ndarray)):
raise ValueError(
"Reading multiple files from class specific "
"read functions is no longer supported. "
"Use the generic `uvdata.read` function instead."
)
from . import ms
ms_obj = ms.MS()
ms_obj.read_ms(
filepath,
data_column=data_column,
pol_order=pol_order,
background_lsts=background_lsts,
ignore_single_chan=ignore_single_chan,
raise_error=raise_error,
read_weights=read_weights,
allow_flex_pol=allow_flex_pol,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
self._convert_from_filetype(ms_obj)
del ms_obj
def read_mwa_corr_fits(
self,
filelist,
use_aoflagger_flags=None,
use_cotter_flags=None,
remove_dig_gains=True,
remove_coarse_band=True,
correct_cable_len=None,
correct_van_vleck=False,
cheby_approx=True,
flag_small_auto_ants=True,
flag_small_sig_ants=None,
propagate_coarse_flags=True,
flag_init=True,
edge_width=80e3,
start_flag="goodtime",
end_flag=0.0,
flag_dc_offset=True,
remove_flagged_ants=True,
phase_to_pointing_center=False,
read_data=True,
data_array_dtype=np.complex64,
nsample_array_dtype=np.float32,
background_lsts=True,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
check_autos=True,
fix_autos=True,
):
"""
Read in MWA correlator gpu box files.
The default settings remove some of the instrumental effects in the bandpass
by dividing out the coarse band shape (for legacy data only) and the digital
gains, and applying a cable length correction.
If the desired output is raw correlator data, set remove_dig_gains=False,
remove_coarse_band=False, correct_cable_len=False, and
phase_to_pointing_center=False.
Parameters
----------
filelist : list of str
The list of MWA correlator files to read from. Must include at
least one fits file and only one metafits file per data set.
use_aoflagger_flags : bool
Option to use aoflagger mwaf flag files. Defaults to true if aoflagger
flag files are submitted.
use_cotter_flags : bool
Being replaced by use_aoflagger_flags and will be removed in v2.4.
remove_dig_gains : bool
Option to divide out digital gains.
remove_coarse_band : bool
Option to divide out coarse band shape.
correct_cable_len : bool
Option to apply a cable delay correction.
correct_van_vleck : bool
Option to apply a van vleck correction.
cheby_approx : bool
Only used if correct_van_vleck is True. Option to implement the van
vleck correction with a chebyshev polynomial approximation.
flag_small_auto_ants : bool
Only used if correct_van_vleck is True. Option to completely flag any
antenna for which the autocorrelation falls below a threshold found by
the Van Vleck correction to indicate bad data. Specifically, the
threshold used is 0.5 * integration_time * channel_width. If set to False,
only the times and frequencies at which the auto is below the
threshold will be flagged for the antenna.
flag_small_sig_ants : bool
Being replaced with flag_small_auto_ants and will be removed in v2.4.
propagate_coarse_flags : bool
Option to propagate flags for missing coarse channel integrations
across frequency.
flag_init: bool
Set to True in order to do routine flagging of coarse channel edges,
start or end integrations, or the center fine channel of each coarse
channel. See associated keywords.
edge_width: float
Only used if flag_init is True. Set to the width to flag on the edge
of each coarse channel, in hz. Errors if not equal to integer
multiple of channel_width. Set to 0 for no edge flagging.
start_flag: float or str
Only used if flag_init is True. The number of seconds to flag at the
beginning of the observation. Set to 0 for no flagging. Default is
'goodtime', which uses information in the metafits file to determine
the length of time that should be flagged. Errors if input is not a
float or 'goodtime'. Errors if float input is not equal to an
integer multiple of the integration time.
end_flag: floats
Only used if flag_init is True. Set to the number of seconds to flag
at the end of the observation. Set to 0 for no flagging. Errors if
not an integer multiple of the integration time.
flag_dc_offset: bool
Only used if flag_init is True. Set to True to flag the center fine
channel of each coarse channel. Only used if file_type is
'mwa_corr_fits'.
remove_flagged_ants : bool
Option to perform a select to remove antennas flagged in the metafits
file. If correct_van_vleck and flag_small_auto_ants are both True then
antennas flagged by the Van Vleck correction are also removed.
phase_to_pointing_center : bool
Option to phase to the observation pointing center.
read_data : bool
Read in the visibility and flag data. If set to false, only the
basic header info and metadata read in. Setting read_data to False
results in a metdata only object.
data_array_dtype : numpy dtype
Datatype to store the output data_array as. Must be either
np.complex64 (single-precision real and imaginary) or np.complex128
(double-precision real and imaginary).
nsample_array_dtype : numpy dtype
Datatype to store the output nsample_array as. Must be either
np.float64 (double-precision), np.float32 (single-precision), or
np.float16 (half-precision). Half-precision is only recommended for
cases where no sampling or averaging of baselines will occur,
because round-off errors can be quite large (~1e-3).
background_lsts : bool
When set to True, the lst_array is calculated in a background thread.
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run).
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done).
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
Raises
------
ValueError
If required files are missing or multiple files metafits files are
included in filelist.
If files from different observations are included in filelist.
If files in fileslist have different fine channel widths
If file types other than fits, metafits, and mwaf files are included
in filelist.
"""
from . import mwa_corr_fits
if isinstance(filelist[0], (list, tuple, np.ndarray)):
raise ValueError(
"Reading multiple files from class specific "
"read functions is no longer supported. "
"Use the generic `uvdata.read` function instead."
)
if use_cotter_flags is not None:
use_aoflagger_flags = use_cotter_flags
warnings.warn(
"Use `use_aoflagger_flags` instead of `use_cotter_flags`."
"`use_cotter_flags` is deprecated, and will be removed in "
"pyuvdata v2.4.",
DeprecationWarning,
)
if flag_small_sig_ants is not None:
flag_small_auto_ants = flag_small_sig_ants
warnings.warn(
"Use `flag_small_auto_ants` instead of `flag_small_sig_ants`."
"`flag_small_sig_ants` is deprecated, and will be removed in "
"pyuvdata v2.4.",
DeprecationWarning,
)
corr_obj = mwa_corr_fits.MWACorrFITS()
corr_obj.read_mwa_corr_fits(
filelist,
use_aoflagger_flags=use_aoflagger_flags,
remove_dig_gains=remove_dig_gains,
remove_coarse_band=remove_coarse_band,
correct_cable_len=correct_cable_len,
correct_van_vleck=correct_van_vleck,
cheby_approx=cheby_approx,
flag_small_auto_ants=flag_small_auto_ants,
propagate_coarse_flags=propagate_coarse_flags,
flag_init=flag_init,
edge_width=edge_width,
start_flag=start_flag,
end_flag=end_flag,
flag_dc_offset=flag_dc_offset,
remove_flagged_ants=remove_flagged_ants,
phase_to_pointing_center=phase_to_pointing_center,
read_data=read_data,
data_array_dtype=data_array_dtype,
nsample_array_dtype=nsample_array_dtype,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
self._convert_from_filetype(corr_obj)
del corr_obj
def read_uvfits(
self,
filename,
antenna_nums=None,
antenna_names=None,
ant_str=None,
bls=None,
frequencies=None,
freq_chans=None,
times=None,
time_range=None,
lsts=None,
lst_range=None,
polarizations=None,
blt_inds=None,
phase_center_ids=None,
keep_all_metadata=True,
read_data=True,
background_lsts=True,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
fix_old_proj=False,
fix_use_ant_pos=True,
check_autos=True,
fix_autos=True,
):
"""
Read in header, metadata and data from a single uvfits file.
Parameters
----------
filename : str
The uvfits file to read from.
antenna_nums : array_like of int, optional
The antennas numbers to include when reading data into the object
(antenna positions and names for the removed antennas will be retained
unless `keep_all_metadata` is False). This cannot be provided if
`antenna_names` is also provided. Ignored if read_data is False.
antenna_names : array_like of str, optional
The antennas names to include when reading data into the object
(antenna positions and names for the removed antennas will be retained
unless `keep_all_metadata` is False). This cannot be provided if
`antenna_nums` is also provided. Ignored if read_data is False.
bls : list of tuple, optional
A list of antenna number tuples (e.g. [(0, 1), (3, 2)]) or a list of
baseline 3-tuples (e.g. [(0, 1, 'xx'), (2, 3, 'yy')]) specifying baselines
to include when reading data into the object. For length-2 tuples,
the ordering of the numbers within the tuple does not matter. For
length-3 tuples, the polarization string is in the order of the two
antennas. If length-3 tuples are provided, `polarizations` must be
None. Ignored if read_data is False.
ant_str : str, optional
A string containing information about what antenna numbers
and polarizations to include when reading data into the object.
Can be 'auto', 'cross', 'all', or combinations of antenna numbers
and polarizations (e.g. '1', '1_2', '1x_2y'). See tutorial for more
examples of valid strings and the behavior of different forms for ant_str.
If '1x_2y,2y_3y' is passed, both polarizations 'xy' and 'yy' will
be kept for both baselines (1, 2) and (2, 3) to return a valid
pyuvdata object.
An ant_str cannot be passed in addition to any of `antenna_nums`,
`antenna_names`, `bls` args or the `polarizations` parameters,
if it is a ValueError will be raised. Ignored if read_data is False.
frequencies : array_like of float, optional
The frequencies to include when reading data into the object, each
value passed here should exist in the freq_array. Ignored if
read_data is False.
freq_chans : array_like of int, optional
The frequency channel numbers to include when reading data into the
object. Ignored if read_data is False.
times : array_like of float, optional
The times to include when reading data into the object, each value
passed here should exist in the time_array in the file.
Cannot be used with `time_range`.
time_range : array_like of float, optional
The time range in Julian Date to include when reading data into
the object, must be length 2. Some of the times in the file should
fall between the first and last elements.
Cannot be used with `times`.
lsts : array_like of float, optional
The local sidereal times (LSTs) to keep in the object, each value
passed here should exist in the lst_array. Cannot be used with
`times`, `time_range`, or `lst_range`.
lst_range : array_like of float, optional
The local sidereal time (LST) range in radians to keep in the
object, must be of length 2. Some of the LSTs in the object should
fall between the first and last elements. If the second value is
smaller than the first, the LSTs are treated as having phase-wrapped
around LST = 2*pi = 0, and the LSTs kept on the object will run from
the larger value, through 0, and end at the smaller value.
polarizations : array_like of int, optional
The polarizations numbers to include when reading data into the
object, each value passed here should exist in the polarization_array.
Ignored if read_data is False.
blt_inds : array_like of int, optional
The baseline-time indices to include when reading data into the
object. This is not commonly used. Ignored if read_data is False.
phase_center_ids : array_like of int, optional
Phase center IDs to include when reading data into the object (effectively
a selection on baseline-times).
keep_all_metadata : bool
Option to keep all the metadata associated with antennas, even those
that do not have data associated with them after the select option.
read_data : bool
Read in the visibility and flag data. If set to false, only the
basic header info and metadata read in. Setting read_data to False
results in a metdata only object.
background_lsts : bool
When set to True, the lst_array is calculated in a background thread.
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run). Ignored if read_data is False.
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
Ignored if read_data is False.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done). Ignored if read_data is False.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
fix_old_proj : bool
Applies a fix to uvw-coordinates and phasing, assuming that the old `phase`
method was used prior to writing the data, which had errors of the order of
one part in 1e4 - 1e5. See the phasing memo for more details. Default is
False.
fix_use_ant_pos : bool
If setting `fix_old_proj` to True, use the antenna positions to derive the
correct uvw-coordinates rather than using the baseline vectors. Default is
True.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
Raises
------
IOError
If filename doesn't exist.
ValueError
If incompatible select keywords are set (e.g. `ant_str` with other
antenna selectors, `times` and `time_range`) or select keywords
exclude all data or if keywords are set to the wrong type.
If the data are multi source or have multiple
spectral windows.
If the metadata are not internally consistent or missing.
"""
from . import uvfits
if isinstance(filename, (list, tuple, np.ndarray)):
raise ValueError(
"Reading multiple files from class specific "
"read functions is no longer supported. "
"Use the generic `uvdata.read` function instead."
)
uvfits_obj = uvfits.UVFITS()
uvfits_obj.read_uvfits(
filename,
antenna_nums=antenna_nums,
antenna_names=antenna_names,
ant_str=ant_str,
bls=bls,
frequencies=frequencies,
freq_chans=freq_chans,
times=times,
time_range=time_range,
lsts=lsts,
lst_range=lst_range,
polarizations=polarizations,
blt_inds=blt_inds,
keep_all_metadata=keep_all_metadata,
read_data=read_data,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
check_autos=check_autos,
fix_autos=fix_autos,
)
self._convert_from_filetype(uvfits_obj)
del uvfits_obj
def read_uvh5(
self,
filename,
antenna_nums=None,
antenna_names=None,
ant_str=None,
bls=None,
frequencies=None,
freq_chans=None,
times=None,
time_range=None,
lsts=None,
lst_range=None,
polarizations=None,
blt_inds=None,
phase_center_ids=None,
keep_all_metadata=True,
read_data=True,
data_array_dtype=np.complex128,
multidim_index=False,
remove_flex_pol=True,
background_lsts=True,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
fix_old_proj=None,
fix_use_ant_pos=True,
check_autos=True,
fix_autos=True,
):
"""
Read a UVH5 file.
Parameters
----------
filename : str
The UVH5 file to read from.
antenna_nums : array_like of int, optional
The antennas numbers to include when reading data into the object
(antenna positions and names for the removed antennas will be retained
unless `keep_all_metadata` is False). This cannot be provided if
`antenna_names` is also provided. Ignored if read_data is False.
antenna_names : array_like of str, optional
The antennas names to include when reading data into the object
(antenna positions and names for the removed antennas will be retained
unless `keep_all_metadata` is False). This cannot be provided if
`antenna_nums` is also provided. Ignored if read_data is False.
bls : list of tuple, optional
A list of antenna number tuples (e.g. [(0, 1), (3, 2)]) or a list of
baseline 3-tuples (e.g. [(0, 1, 'xx'), (2, 3, 'yy')]) specifying baselines
to include when reading data into the object. For length-2 tuples,
the ordering of the numbers within the tuple does not matter. For
length-3 tuples, the polarization string is in the order of the two
antennas. If length-3 tuples are provided, `polarizations` must be
None. Ignored if read_data is False.
ant_str : str, optional
A string containing information about what antenna numbers
and polarizations to include when reading data into the object.
Can be 'auto', 'cross', 'all', or combinations of antenna numbers
and polarizations (e.g. '1', '1_2', '1x_2y'). See tutorial for more
examples of valid strings and the behavior of different forms for ant_str.
If '1x_2y,2y_3y' is passed, both polarizations 'xy' and 'yy' will
be kept for both baselines (1, 2) and (2, 3) to return a valid
pyuvdata object.
An ant_str cannot be passed in addition to any of `antenna_nums`,
`antenna_names`, `bls` args or the `polarizations` parameters,
if it is a ValueError will be raised. Ignored if read_data is False.
frequencies : array_like of float, optional
The frequencies to include when reading data into the object, each
value passed here should exist in the freq_array. Ignored if
read_data is False.
freq_chans : array_like of int, optional
The frequency channel numbers to include when reading data into the
object. Ignored if read_data is False.
times : array_like of float, optional
The times to include when reading data into the object, each value
passed here should exist in the time_array in the file.
Cannot be used with `time_range`.
time_range : array_like of float, optional
The time range in Julian Date to include when reading data into
the object, must be length 2. Some of the times in the file should
fall between the first and last elements.
Cannot be used with `times`.
lsts : array_like of float, optional
The local sidereal times (LSTs) to keep in the object, each value
passed here should exist in the lst_array. Cannot be used with
`times`, `time_range`, or `lst_range`.
lst_range : array_like of float, optional
The local sidereal time (LST) range in radians to keep in the
object, must be of length 2. Some of the LSTs in the object should
fall between the first and last elements. If the second value is
smaller than the first, the LSTs are treated as having phase-wrapped
around LST = 2*pi = 0, and the LSTs kept on the object will run from
the larger value, through 0, and end at the smaller value.
polarizations : array_like of int, optional
The polarizations numbers to include when reading data into the
object, each value passed here should exist in the polarization_array.
Ignored if read_data is False.
blt_inds : array_like of int, optional
The baseline-time indices to include when reading data into the
object. This is not commonly used. Ignored if read_data is False.
phase_center_ids : array_like of int, optional
Phase center IDs to include when reading data into the object (effectively
a selection on baseline-times).
keep_all_metadata : bool
Option to keep all the metadata associated with antennas, even those
that do not have data associated with them after the select option.
read_data : bool
Read in the visibility and flag data. If set to false, only the
basic header info and metadata will be read in. Setting read_data to
False results in an incompletely defined object (check will not pass).
data_array_dtype : numpy dtype
Datatype to store the output data_array as. Must be either
np.complex64 (single-precision real and imaginary) or np.complex128 (double-
precision real and imaginary). Only used if the datatype of the visibility
data on-disk is not 'c8' or 'c16'.
multidim_index : bool
[Only for HDF5] If True, attempt to index the HDF5 dataset
simultaneously along all data axes. Otherwise index one axis at-a-time.
This only works if data selection is sliceable along all but one axis.
If indices are not well-matched to data chunks, this can be slow.
remove_flex_pol : bool
If True and if the file is a flex_pol file, convert back to a standard
UVData object.
background_lsts : bool
When set to True, the lst_array is calculated in a background thread.
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run). Ignored if read_data is False.
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
Ignored if read_data is False.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done). Ignored if read_data is False.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
fix_old_proj : bool
Applies a fix to uvw-coordinates and phasing, assuming that the old `phase`
method was used prior to writing the data, which had errors of the order of
one part in 1e4 - 1e5. See the phasing memo for more details. Default is
to apply the correction if the attributes `phase_center_app_ra` and
`phase_center_app_dec` are missing (as they were introduced alongside the
new phasing method).
fix_use_ant_pos : bool
If setting `fix_old_proj` to True, use the antenna positions to derive the
correct uvw-coordinates rather than using the baseline vectors. Default is
True.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
Raises
------
IOError
If filename doesn't exist.
ValueError
If the data_array_dtype is not a complex dtype.
If incompatible select keywords are set (e.g. `ant_str` with other
antenna selectors, `times` and `time_range`) or select keywords
exclude all data or if keywords are set to the wrong type.
"""
from . import uvh5
if isinstance(filename, (list, tuple, np.ndarray)):
raise ValueError(
"Reading multiple files from class specific "
"read functions is no longer supported. "
"Use the generic `uvdata.read` function instead."
)
uvh5_obj = uvh5.UVH5()
uvh5_obj.read_uvh5(
filename,
antenna_nums=antenna_nums,
antenna_names=antenna_names,
ant_str=ant_str,
bls=bls,
frequencies=frequencies,
freq_chans=freq_chans,
times=times,
time_range=time_range,
lsts=lsts,
lst_range=lst_range,
polarizations=polarizations,
blt_inds=blt_inds,
phase_center_ids=phase_center_ids,
data_array_dtype=data_array_dtype,
keep_all_metadata=keep_all_metadata,
read_data=read_data,
multidim_index=multidim_index,
remove_flex_pol=remove_flex_pol,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
check_autos=check_autos,
fix_autos=fix_autos,
)
self._convert_from_filetype(uvh5_obj)
del uvh5_obj
def read(
self,
filename,
axis=None,
file_type=None,
read_data=True,
skip_bad_files=False,
background_lsts=True,
ignore_name=False,
# phasing parameters
allow_rephase=None,
phase_center_radec=None,
unphase_to_drift=None,
phase_frame=None,
phase_epoch=None,
orig_phase_frame=None,
phase_use_ant_pos=None,
fix_old_proj=None,
fix_use_ant_pos=True,
make_multi_phase=False,
# selecting parameters
antenna_nums=None,
antenna_names=None,
ant_str=None,
bls=None,
frequencies=None,
freq_chans=None,
times=None,
time_range=None,
lsts=None,
lst_range=None,
polarizations=None,
blt_inds=None,
phase_center_ids=None,
keep_all_metadata=True,
# checking parameters
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
check_autos=True,
fix_autos=True,
# file-type specific parameters
# miriad
phase_type=None,
projected=None,
correct_lat_lon=True,
calc_lst=True,
# FHD
use_model=False,
# MS
data_column="DATA",
pol_order="AIPS",
ignore_single_chan=True,
raise_error=True,
read_weights=True,
# MS & MIR
allow_flex_pol=True,
# uvh5
multidim_index=False,
remove_flex_pol=True,
# uvh5 & mwa_corr_fits
data_array_dtype=np.complex128,
# mwa_corr_fits
use_aoflagger_flags=None,
use_cotter_flags=None,
remove_dig_gains=True,
remove_coarse_band=True,
correct_cable_len=None,
correct_van_vleck=False,
cheby_approx=True,
flag_small_auto_ants=True,
flag_small_sig_ants=None,
propagate_coarse_flags=True,
flag_init=True,
edge_width=80e3,
start_flag="goodtime",
end_flag=0.0,
flag_dc_offset=True,
remove_flagged_ants=True,
phase_to_pointing_center=False,
nsample_array_dtype=np.float32,
# MIR
isource=None,
irec=None,
isb=None,
corrchunk=None,
pseudo_cont=False,
rechunk=None,
):
"""
Read a generic file into a UVData object.
This method supports a number of different types of files.
Universal parameters (required and optional) are listed directly below,
followed by parameters used by all file types related to phasing, selecting on
read (partial read) and checking. Each file type also has its own set of
optional parameters that are listed at the end of this docstring.
Parameters
----------
filename : str or array_like of str
The file(s) or list(s) (or array(s)) of files to read from.
file_type : str
One of ['uvfits', 'miriad', 'ms', 'uvh5', 'fhd', 'mwa_corr_fits', 'mir']
or None. If None, the code attempts to guess what the file type is.
For miriad and ms types, this is based on the standard directory
structure. For FHD, uvfits and uvh5 files it's based on file
extensions (FHD: .sav, .txt; uvfits: .uvfits; uvh5: .uvh5).
Note that if a list of datasets is passed, the file type is
determined from the first dataset.
axis : str
Axis to concatenate files along. This enables fast concatenation
along the specified axis without the normal checking that all other
metadata agrees. This method does not guarantee correct resulting
objects. Please see the docstring for fast_concat for details.
Allowed values are: 'blt', 'freq', 'polarization'. Only used if
multiple files are passed.
read_data : bool
Read in the data. Not used if file_type is 'ms' or 'mir'.
If set to False, only the metadata will be read in. Setting read_data to
False results in a metdata only object.
skip_bad_files : bool
Option when reading multiple files to catch read errors such that
the read continues even if one or more files are corrupted. Files
that produce errors will be printed. Default is False (files will
not be skipped).
background_lsts : bool
When set to True, the lst_array is calculated in a background thread.
ignore_name : bool
Only relevant when reading in multiple files, which are concatenated into a
single UVData object. Option to ignore the name of the phase center when
combining multiple files, which would otherwise result in an error being
raised because of attributes not matching. Doing so effectively adopts the
name found in the first file read in. Default is False.
Phasing
-------
allow_rephase : bool
Deprecated, will have no effect starting in version 2.4 and will be removed
in version 2.6 because data with different phasing can now be combined.
Allow rephasing of phased file data. Currently defaults to True, which will
cause data to be rephased (to preserve the previous behavior). Set to False
to prevent rephasing and mimic the future behavior.
phase_center_radec : array_like of float
Deprecated and will be removed in version 2.4, instead call `phase` on the
combined object after this method. The phase center to phase the files to
before combining the objects in radians (in the ICRS frame). If set to None
and allow_rephase is True and multiple files are read with different phase
centers, the phase center of the first file will be used.
unphase_to_drift : bool
Deprecated, instead call `unproject_phase` on the combined object after
this method. Unproject the data from the files before combining them.
phase_frame : str
The astropy frame to phase to.Only used if `phase_center_radec` is set.
Deprecated and will be removed in version 2.4 with `phase_center_radec`.
phase_epoch : astropy.time.Time object or str
The epoch to use for phasing. Either an astropy Time object or the
string "J2000" (which is the default).
Note that the epoch is only used to evaluate the ra & dec values,
if the epoch is not J2000, the ra & dec values are interpreted
as FK5 ra/dec values and transformed to J2000, the data are then
phased to the J2000 ra/dec values.
Deprecated and will be removed in version 2.4 with `phase_center_radec`.
orig_phase_frame : str
Deprecated, has no effect. Will be removed in version 2.4.
phase_use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions rather than
from the existing uvws. Only used if `unphase_to_drift` or
`phase_center_radec` are set. Deprecated and will be removed in version 2.4
with `unphase_to_drift` and `phase_center_radec`.
fix_old_proj : bool
Applies a fix to uvw-coordinates and phasing, assuming that the old `phase`
method was used prior to writing the data, which had errors of the order of
one part in 1e4 - 1e5. See the phasing memo for more details. Default is
False, unless reading a UVH5 file that is missing the `phase_center_app_ra`
and `phase_center_app_dec` attributes (as these were introduced at the same
time as the new `phase` method), in which case the default is True.
fix_use_ant_pos : bool
If setting `fix_old_proj` to True, use the antenna positions to derive the
correct uvw-coordinates rather than using the baseline vectors. Default is
True.
make_multi_phase : bool
Deprecated, will be removed in version 2.4. Since all objects now support
multiple phase centers, this has no effect if set to False. The only effect
of setting this to True is to prevent rephasing. A better approach is to set
`allow_rephase=False`.
Selecting
---------
antenna_nums : array_like of int, optional
The antennas numbers to include when reading data into the object
(antenna positions and names for the removed antennas will be retained
unless `keep_all_metadata` is False). This cannot be provided if
`antenna_names` is also provided.
antenna_names : array_like of str, optional
The antennas names to include when reading data into the object
(antenna positions and names for the removed antennas will be retained
unless `keep_all_metadata` is False). This cannot be provided if
`antenna_nums` is also provided.
ant_str : str, optional
A string containing information about what antenna numbers
and polarizations to include when reading data into the object.
Can be 'auto', 'cross', 'all', or combinations of antenna numbers
and polarizations (e.g. '1', '1_2', '1x_2y'). See tutorial for more
examples of valid strings and the behavior of different forms for ant_str.
If '1x_2y,2y_3y' is passed, both polarizations 'xy' and 'yy' will
be kept for both baselines (1, 2) and (2, 3) to return a valid
pyuvdata object.
An ant_str cannot be passed in addition to any of `antenna_nums`,
`antenna_names`, `bls` args or the `polarizations` parameters,
if it is a ValueError will be raised.
bls : list of tuple, optional
A list of antenna number tuples (e.g. [(0, 1), (3, 2)]) or a list of
baseline 3-tuples (e.g. [(0, 1, 'xx'), (2, 3, 'yy')]) specifying baselines
to include when reading data into the object. For length-2 tuples,
the ordering of the numbers within the tuple does not matter. For
length-3 tuples, the polarization string is in the order of the two
antennas. If length-3 tuples are provided, `polarizations` must be
None.
frequencies : array_like of float, optional
The frequencies to include when reading data into the object, each
value passed here should exist in the freq_array.
freq_chans : array_like of int, optional
The frequency channel numbers to include when reading data into the
object. Ignored if read_data is False.
times : array_like of float, optional
The times to include when reading data into the object, each value
passed here should exist in the time_array in the file.
Cannot be used with `time_range`.
time_range : array_like of float, optional
The time range in Julian Date to include when reading data into
the object, must be length 2. Some of the times in the file should
fall between the first and last elements.
Cannot be used with `times`.
lsts : array_like of float, optional
The local sidereal times (LSTs) to keep in the object, each value
passed here should exist in the lst_array. Cannot be used with
`times`, `time_range`, or `lst_range`.
lst_range : array_like of float, optional
The local sidereal time (LST) range in radians to keep in the
object, must be of length 2. Some of the LSTs in the object should
fall between the first and last elements. If the second value is
smaller than the first, the LSTs are treated as having phase-wrapped
around LST = 2*pi = 0, and the LSTs kept on the object will run from
the larger value, through 0, and end at the smaller value.
polarizations : array_like of int, optional
The polarizations numbers to include when reading data into the
object, each value passed here should exist in the polarization_array.
blt_inds : array_like of int, optional
The baseline-time indices to include when reading data into the
object. This is not commonly used.
phase_center_ids : array_like of int, optional
Phase center IDs to include when reading data into the object (effectively
a selection on baseline-times).
keep_all_metadata : bool
Option to keep all the metadata associated with antennas, even those
that do not have data associated with them after the select option.
Checking
--------
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run). Ignored if read_data is False.
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
Ignored if read_data is False.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done). Ignored if read_data is False.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
Miriad
------
phase_type : str, optional
Deprecated, use the `projected` parameter insted. Option to specify the
phasing status of the data. Options are 'drift', 'phased' or None. 'drift'
is the same as setting the projected parameter to False, 'phased' is the
same as setting the projected parameter to True. Ignored if `projected`
is set.
projected : bool or None
Option to force the dataset to be labelled as projected or unprojected
regardless of the evidence in the file. The default is None which means that
the projection will be set based on the file contents. Be careful setting
this keyword unless you are confident about the contents of the file.
correct_lat_lon : bool
Option to update the latitude and longitude from the known_telescopes
list if the altitude is missing.
calc_lst : bool
Recalculate the LST values that are present within the file, useful in
cases where the "online" calculate values have precision or value errors.
Default is True.
FHD
---
use_model : bool
Option to read in the model visibilities rather than the dirty
visibilities (the default is False, meaning the dirty visibilities
will be read).
MS
--
data_column : str
name of CASA data column to read into data_array. Options are:
'DATA', 'MODEL', or 'CORRECTED_DATA'.
pol_order : str
Option to specify polarizations order convention, options are
'CASA' or 'AIPS'.
ignore_single_chan : bool
Option to ignore single channel spectral windows in measurement sets to
limit object size. Some measurement sets (e.g., those from ALMA) use single
channel spectral windows for recording pseudo-continuum channels or storing
other metadata in the track when the telescopes are not on source. Because
of the way the object is strutured (where all spectral windows are assumed
to be simultaneously recorded), this can significantly inflate the size and
memory footprint of UVData objects. By default, single channel windows are
ignored to avoid this issue, they can be included by setting this parameter
to True.
raise_error : bool
The measurement set format allows for different spectral windows and
polarizations to have different metdata for the same time-baseline
combination, but UVData objects do not. If detected, by default the reader
will throw an error. However, if set to False, the reader will simply give
a warning, and will use the first value read in the file as the "correct"
metadata in the UVData object.
read_weights : bool
Read in the weights from the MS file. If false, the method
will set the `nsamples_array` to the same uniform value (namely 1.0).
allow_flex_pol : bool
If only one polarization per spectral window is read (and the polarization
differs from window to window), allow for the `UVData` object to use
"flexible polarization", which compresses the polarization-axis of various
attributes to be of length 1, sets the `flex_spw_polarization_array`
attribute to define the polarization per spectral window. Default is True.
UVH5
----
multidim_index : bool
If True, attempt to index the HDF5 dataset simultaneously along all data
axes. Otherwise index one axis at-a-time. This only works if data selection
is sliceable along all but one axis. If indices are not well-matched to
data chunks, this can be slow.
remove_flex_pol : bool
If True and if the file is a flex_pol file, convert back to a standard
UVData object.
data_array_dtype : numpy dtype
Datatype to store the output data_array as. Must be either
np.complex64 (single-precision real and imaginary) or np.complex128 (double-
precision real and imaginary). Only used if the datatype of the visibility
data on-disk is not 'c8' or 'c16'.
MWA FITS
--------
use_aoflagger_flags : bool
Option to use aoflagger mwaf flag files. Defaults to true if aoflagger
flag files are submitted.
use_cotter_flags : bool
Being replaced by use_aoflagger_flags and will be removed in v2.4.
remove_dig_gains : bool
Option to divide out digital gains.
remove_coarse_band : bool
Option to divide out coarse band shape.
correct_cable_len : bool
Flag to apply cable length correction.
correct_van_vleck : bool
Flag to apply a van vleck correction.
cheby_approx : bool
Option to implement the van vleck correction with a chebyshev polynomial
approximation. Set to False to run the integral version of the correction.
Only used if correct_van_vleck is True.
flag_small_auto_ants : bool
Option to completely flag any antenna for which the autocorrelation falls
below a threshold found by the Van Vleck correction to indicate bad data.
Specifically, the threshold used is 0.5 * integration_time * channel_width.
If set to False, only the times and frequencies at which the auto is below
the threshold will be flagged for the antenna.
Only used if correct_van_vleck is True.
flag_small_sig_ants : bool
Deprecated, replaced by flag_small_auto_ants and will be removed in v2.4.
propogate_coarse_flags : bool
Option to propogate flags for missing coarse channel integrations
across frequency.
flag_init: bool
Set to True in order to do routine flagging of coarse channel edges, start
or end integrations, or the center fine channel of each coarse
channel. See associated keywords below.
edge_width: float
Only used if flag_init is True. Set to the width to flag on the edge of
each coarse channel, in hz. Errors if not equal to integer multiple of
channel_width. Set to 0 for no edge flagging.
start_flag: float or str
Only used if flag_init is True. The number of seconds to flag at the
beginning of the observation. Set to 0 for no flagging. Default is
'goodtime', which uses information in the metafits file to determine
the length of time that should be flagged. Errors if input is not a
float or 'goodtime'. Errors if float input is not equal to an
integer multiple of the integration time.
end_flag: floats
Only used if flag_init is True. Set to the number of seconds to flag at the
end of the observation. Set to 0 for no flagging. Errors if not an integer
multiple of the integration time.
flag_dc_offset: bool
Only used if flag_init is True. Set to True to flag the center fine channel
of each coarse channel.
remove_flagged_ants : bool
Option to perform a select to remove antennas flagged in the metafits
file. If correct_van_vleck and flag_small_auto_ants are both True then
antennas flagged by the Van Vleck correction are also removed.
phase_to_pointing_center : bool
Flag to phase to the pointing center. Cannot be set if phase_center_radec
is set to a value.
data_array_dtype : numpy dtype
Datatype to store the output data_array as. Must be either
np.complex64 (single-precision real and imaginary) or np.complex128 (double-
precision real and imaginary).
nsample_array_dtype : numpy dtype
Datatype to store the output nsample_array as. Must be either
np.float64 (double-precision), np.float32 (single-precision), or
np.float16 (half-precision). Half-precision is only recommended for
cases where no sampling or averaging of baselines will occur,
because round-off errors can be quite large (~1e-3).
MIR
---
isource : int
Source code for MIR dataset.
irec : int
Receiver code for MIR dataset.
isb : int
Sideband code for MIR dataset.
corrchunk : int
Correlator chunk code for MIR dataset.
pseudo_cont : boolean
Read in only pseudo-continuuum values in MIR dataset.
allow_flex_pol : bool
If only one polarization per spectral window is read (and the polarization
differs from window to window), allow for the `UVData` object to use
"flexible polarization", which compresses the polarization-axis of various
attributes to be of length 1, sets the `flex_spw_polarization_array`
attribute to define the polarization per spectral window. Default is True.
Raises
------
ValueError
If the file_type is not set and cannot be determined from the file name.
If incompatible select keywords are set (e.g. `ant_str` with other
antenna selectors, `times` and `time_range`) or select keywords
exclude all data or if keywords are set to the wrong type.
If phase_center_radec is not None and is not length 2.
"""
if allow_rephase:
warnings.warn(
"The `allow_rephase` option is deprecated. It will have no effect "
"starting in pyuvdata v2.4 and will be removed in pyuvdata v2.6."
"It is no longer needed as data with different phase centers can now "
"be combined on one object. Starting in version 2.4, the data will be "
"combined on one object without rephasing. To get that behavior now, "
"set `allow_rephase=False`. The final object can be rephased if "
"desired by calling `phase` after this method.",
DeprecationWarning,
)
if make_multi_phase:
warnings.warn(
"The `make_multi_phase` option is deprecated. Since all objects now "
"support multiple phase centers, the only effect of setting this to "
"True is to prevent rephasing. A better approach is to set "
"`allow_rephase=False`.",
DeprecationWarning,
)
if isinstance(filename, (list, tuple, np.ndarray)):
# this is either a list of separate files to read or a list of
# FHD files or MWA correlator FITS files
if isinstance(filename[0], (list, tuple, np.ndarray)):
if file_type is None:
# this must be a list of lists of FHD or MWA correlator FITS
_, extension = os.path.splitext(filename[0][0])
if extension == ".sav" or extension == ".txt":
file_type = "fhd"
elif (
extension == ".fits"
or extension == ".metafits"
or extension == ".mwaf"
):
file_type = "mwa_corr_fits"
multi = True
else:
if file_type is None:
_, extension = os.path.splitext(filename[0])
if extension == ".sav" or extension == ".txt":
file_type = "fhd"
elif (
extension == ".fits"
or extension == ".metafits"
or extension == ".mwaf"
):
file_type = "mwa_corr_fits"
if file_type == "fhd" or file_type == "mwa_corr_fits":
multi = False
else:
multi = True
else:
multi = False
if file_type is None:
if multi:
file_test = filename[0]
else:
file_test = filename
if os.path.isdir(file_test):
# it's a directory, so it's either miriad, mir, or ms file type
if os.path.exists(os.path.join(file_test, "vartable")):
# It's miriad.
file_type = "miriad"
elif os.path.exists(os.path.join(file_test, "OBSERVATION")):
# It's a measurement set.
file_type = "ms"
elif os.path.exists(os.path.join(file_test, "sch_read")):
# It's Submillimeter Array mir format.
file_type = "mir"
else:
_, extension = os.path.splitext(file_test)
if extension == ".uvfits":
file_type = "uvfits"
elif extension == ".uvh5":
file_type = "uvh5"
if file_type is None:
raise ValueError(
"File type could not be determined, use the "
"file_type keyword to specify the type."
)
if time_range is not None:
if times is not None:
raise ValueError("Only one of times and time_range can be provided.")
if antenna_names is not None and antenna_nums is not None:
raise ValueError(
"Only one of antenna_nums and antenna_names can " "be provided."
)
if multi:
file_num = 0
file_warnings = ""
unread = True
f = filename[file_num]
while unread and file_num < len(filename):
try:
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore", "The `allow_rephase` option is deprecated"
)
warnings.filterwarnings(
"ignore", "The `make_multi_phase` option is deprecated"
)
self.read(
filename[file_num],
file_type=file_type,
allow_rephase=allow_rephase,
phase_center_radec=phase_center_radec,
phase_frame=phase_frame,
phase_epoch=phase_epoch,
phase_use_ant_pos=phase_use_ant_pos,
unphase_to_drift=unphase_to_drift,
antenna_nums=antenna_nums,
antenna_names=antenna_names,
ant_str=ant_str,
bls=bls,
frequencies=frequencies,
freq_chans=freq_chans,
times=times,
polarizations=polarizations,
blt_inds=blt_inds,
phase_center_ids=phase_center_ids,
time_range=time_range,
lsts=lsts,
lst_range=lst_range,
keep_all_metadata=keep_all_metadata,
read_data=read_data,
phase_type=phase_type,
projected=projected,
correct_lat_lon=correct_lat_lon,
use_model=use_model,
data_column=data_column,
pol_order=pol_order,
data_array_dtype=data_array_dtype,
nsample_array_dtype=nsample_array_dtype,
skip_bad_files=skip_bad_files,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
isource=isource,
irec=irec,
isb=isb,
corrchunk=corrchunk,
pseudo_cont=pseudo_cont,
calc_lst=calc_lst,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
make_multi_phase=make_multi_phase,
allow_flex_pol=allow_flex_pol,
check_autos=check_autos,
fix_autos=fix_autos,
)
unread = False
except KeyError as err:
file_warnings = (
file_warnings + f"Failed to read {f} due to KeyError: {err}\n"
)
file_num += 1
if skip_bad_files is False:
raise
except ValueError as err:
file_warnings = (
file_warnings + f"Failed to read {f} due to ValueError: {err}\n"
)
file_num += 1
if skip_bad_files is False:
raise
except OSError as err: # pragma: nocover
file_warnings = (
file_warnings + f"Failed to read {f} due to OSError: {err}\n"
)
file_num += 1
if skip_bad_files is False:
raise
if unread is True:
warnings.warn(
"########################################################\n"
"ALL FILES FAILED ON READ - NO READABLE FILES IN FILENAME\n"
"########################################################"
)
return
any_unprojected = np.any(~self._check_for_cat_type("unprojected"))
phase_dict = None
if allow_rephase is None and not make_multi_phase and any_unprojected:
warnings.warn(
"The default behavior is to rephase data from the files to make "
"them have the same phasing. This is no longer needed as data with "
"different phase centers can now be combined on one object. "
"Starting in version 2.4, the data will be combined on one object "
"without rephasing. To get that behavior now, set "
"`allow_rephase=False`. The final object can be rephased if "
"desired by calling `phase` after this method.",
DeprecationWarning,
)
allow_rephase = True
if (
allow_rephase
and phase_center_radec is None
and not unphase_to_drift
and any_unprojected
and not make_multi_phase
):
if self.Nphase > 1:
warnings.warn(
"Allow_rephase is True but `phase_center_radec` is not set and "
"the first file has multiple phase centers, so no rephasing "
"will be done."
)
else:
phase_dict = list(self.phase_center_catalog.values())[0]
uv_list = []
if len(filename) > file_num + 1:
for f in filename[file_num + 1 :]:
uv2 = UVData()
try:
with warnings.catch_warnings():
warnings.filterwarnings(
"ignore", "The `allow_rephase` option is deprecated"
)
warnings.filterwarnings(
"ignore", "The `make_multi_phase` option is deprecated"
)
warnings.filterwarnings(
"ignore",
"The `phase_center_radec` parameter is deprecated",
)
warnings.filterwarnings(
"ignore",
"The `unphase_to_drift` parameter is deprecated",
)
warnings.filterwarnings(
"ignore", "The `allow_rephase` option is deprecated"
)
uv2.read(
f,
file_type=file_type,
allow_rephase=allow_rephase,
phase_center_radec=phase_center_radec,
phase_frame=phase_frame,
phase_epoch=phase_epoch,
phase_use_ant_pos=phase_use_ant_pos,
unphase_to_drift=unphase_to_drift,
antenna_nums=antenna_nums,
antenna_names=antenna_names,
ant_str=ant_str,
bls=bls,
frequencies=frequencies,
freq_chans=freq_chans,
times=times,
polarizations=polarizations,
blt_inds=blt_inds,
phase_center_ids=phase_center_ids,
time_range=time_range,
lsts=lsts,
lst_range=lst_range,
keep_all_metadata=keep_all_metadata,
read_data=read_data,
phase_type=phase_type,
projected=projected,
correct_lat_lon=correct_lat_lon,
use_model=use_model,
data_column=data_column,
pol_order=pol_order,
data_array_dtype=data_array_dtype,
nsample_array_dtype=nsample_array_dtype,
skip_bad_files=skip_bad_files,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
isource=isource,
irec=irec,
isb=isb,
corrchunk=corrchunk,
pseudo_cont=pseudo_cont,
calc_lst=calc_lst,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
make_multi_phase=make_multi_phase,
allow_flex_pol=allow_flex_pol,
check_autos=check_autos,
fix_autos=fix_autos,
rechunk=rechunk,
)
if phase_dict is not None:
uv2.phase(
lon=phase_dict.get("cat_lon"),
lat=phase_dict.get("cat_lat"),
epoch=phase_dict.get("cat_epoch"),
phase_frame=phase_dict.get("cat_frame"),
cat_type=phase_dict.get("cat_type"),
ephem_times=phase_dict.get("cat_times"),
pm_ra=phase_dict.get("cat_pm_ra"),
pm_dec=phase_dict.get("cat_pm_dec"),
dist=phase_dict.get("cat_dist"),
vrad=phase_dict.get("cat_vrad"),
cat_name=phase_dict.get("cat_name"),
use_ant_pos=phase_use_ant_pos,
allow_rephase=allow_rephase,
)
uv_list.append(uv2)
except KeyError as err:
file_warnings = (
file_warnings
+ f"Failed to read {f} due to KeyError: {err}\n"
)
if skip_bad_files:
continue
else:
raise
except ValueError as err:
file_warnings = (
file_warnings
+ f"Failed to read {f} due to ValueError: {err}\n"
)
if skip_bad_files:
continue
else:
raise
except OSError as err: # pragma: nocover
file_warnings = (
file_warnings
+ f"Failed to read {f} due to OSError: {err}\n"
)
if skip_bad_files:
continue
else:
raise
if len(file_warnings) > 0:
warnings.warn(file_warnings)
# Concatenate once at end
if axis is not None:
# Rewrote fast_concat to operate on lists
self.fast_concat(
uv_list,
axis,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
inplace=True,
ignore_name=ignore_name,
)
else:
# Too much work to rewrite __add__ to operate on lists
# of files, so instead doing a binary tree merge
uv_list = [self] + uv_list
while len(uv_list) > 1:
for uv1, uv2 in zip(uv_list[0::2], uv_list[1::2]):
uv1.__iadd__(
uv2,
use_ant_pos=phase_use_ant_pos,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
ignore_name=ignore_name,
)
uv_list = uv_list[0::2]
# Because self was at the beginning of the list,
# everything is merged into it at the end of this loop
else:
if file_type in ["fhd", "ms", "mwa_corr_fits", "mir"]:
if (
antenna_nums is not None
or antenna_names is not None
or ant_str is not None
or bls is not None
or frequencies is not None
or freq_chans is not None
or times is not None
or lsts is not None
or time_range is not None
or lst_range is not None
or polarizations is not None
or blt_inds is not None
or phase_center_ids is not None
):
select = True
warnings.warn(
"Warning: select on read keyword set, but "
'file_type is "{ftype}" which does not support select '
"on read. Entire file will be read and then select "
"will be performed".format(ftype=file_type)
)
# these file types do not have select on read, so set all
# select parameters
select_antenna_nums = antenna_nums
select_antenna_names = antenna_names
select_ant_str = ant_str
select_bls = bls
select_frequencies = frequencies
select_freq_chans = freq_chans
select_times = times
select_lsts = lsts
select_time_range = time_range
select_lst_range = lst_range
select_polarizations = polarizations
select_blt_inds = blt_inds
select_phase_center_ids = phase_center_ids
else:
select = False
elif file_type in ["uvfits", "uvh5"]:
select = False
elif file_type in ["miriad"]:
if (
antenna_names is not None
or frequencies is not None
or freq_chans is not None
or times is not None
or blt_inds is not None
or phase_center_ids is not None
):
if blt_inds is not None:
if (
antenna_nums is not None
or ant_str is not None
or bls is not None
or time_range is not None
):
warnings.warn(
"Warning: blt_inds is set along with select "
"on read keywords that are supported by "
"read_miriad and may downselect blts. "
"This may result in incorrect results "
"because the select on read will happen "
"before the blt_inds selection so the indices "
"may not match the expected locations."
)
else:
warnings.warn(
"Warning: a select on read keyword is set that is "
"not supported by read_miriad. This select will "
"be done after reading the file."
)
select = True
# these are all done by partial read, so set to None
select_antenna_nums = None
select_ant_str = None
select_bls = None
select_time_range = None
select_polarizations = None
# these aren't supported by partial read, so do it in select
select_antenna_names = antenna_names
select_frequencies = frequencies
select_freq_chans = freq_chans
select_times = times
select_lsts = lsts
select_lst_range = lst_range
select_blt_inds = blt_inds
select_phase_center_ids = phase_center_ids
else:
select = False
# reading a single "file". Call the appropriate file-type read
if file_type == "uvfits":
self.read_uvfits(
filename,
antenna_nums=antenna_nums,
antenna_names=antenna_names,
ant_str=ant_str,
bls=bls,
frequencies=frequencies,
freq_chans=freq_chans,
times=times,
time_range=time_range,
lsts=lsts,
lst_range=lst_range,
polarizations=polarizations,
blt_inds=blt_inds,
phase_center_ids=phase_center_ids,
read_data=read_data,
keep_all_metadata=keep_all_metadata,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
check_autos=check_autos,
fix_autos=fix_autos,
)
elif file_type == "mir":
self.read_mir(
filename,
isource=isource,
irec=irec,
isb=isb,
corrchunk=corrchunk,
pseudo_cont=pseudo_cont,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
allow_flex_pol=allow_flex_pol,
check_autos=check_autos,
fix_autos=fix_autos,
rechunk=rechunk,
)
elif file_type == "miriad":
self.read_miriad(
filename,
antenna_nums=antenna_nums,
ant_str=ant_str,
bls=bls,
polarizations=polarizations,
time_range=time_range,
read_data=read_data,
phase_type=phase_type,
projected=projected,
correct_lat_lon=correct_lat_lon,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
calc_lst=calc_lst,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
check_autos=check_autos,
fix_autos=fix_autos,
)
elif file_type == "mwa_corr_fits":
self.read_mwa_corr_fits(
filename,
use_aoflagger_flags=use_aoflagger_flags,
use_cotter_flags=use_cotter_flags,
remove_dig_gains=remove_dig_gains,
remove_coarse_band=remove_coarse_band,
correct_cable_len=correct_cable_len,
correct_van_vleck=correct_van_vleck,
cheby_approx=cheby_approx,
flag_small_auto_ants=flag_small_auto_ants,
flag_small_sig_ants=flag_small_sig_ants,
propagate_coarse_flags=propagate_coarse_flags,
flag_init=flag_init,
edge_width=edge_width,
start_flag=start_flag,
end_flag=end_flag,
flag_dc_offset=flag_dc_offset,
remove_flagged_ants=remove_flagged_ants,
phase_to_pointing_center=phase_to_pointing_center,
read_data=read_data,
data_array_dtype=data_array_dtype,
nsample_array_dtype=nsample_array_dtype,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
elif file_type == "fhd":
self.read_fhd(
filename,
use_model=use_model,
background_lsts=background_lsts,
read_data=read_data,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
elif file_type == "ms":
self.read_ms(
filename,
data_column=data_column,
pol_order=pol_order,
background_lsts=background_lsts,
ignore_single_chan=ignore_single_chan,
raise_error=raise_error,
read_weights=read_weights,
allow_flex_pol=allow_flex_pol,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
elif file_type == "uvh5":
self.read_uvh5(
filename,
antenna_nums=antenna_nums,
antenna_names=antenna_names,
ant_str=ant_str,
bls=bls,
frequencies=frequencies,
freq_chans=freq_chans,
times=times,
time_range=time_range,
lsts=lsts,
lst_range=lst_range,
polarizations=polarizations,
blt_inds=blt_inds,
phase_center_ids=phase_center_ids,
read_data=read_data,
data_array_dtype=data_array_dtype,
keep_all_metadata=keep_all_metadata,
multidim_index=multidim_index,
remove_flex_pol=remove_flex_pol,
background_lsts=background_lsts,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
check_autos=check_autos,
fix_autos=fix_autos,
)
select = False
if select:
self.select(
antenna_nums=select_antenna_nums,
antenna_names=select_antenna_names,
ant_str=select_ant_str,
bls=select_bls,
frequencies=select_frequencies,
freq_chans=select_freq_chans,
times=select_times,
lsts=select_lsts,
time_range=select_time_range,
lst_range=select_lst_range,
polarizations=select_polarizations,
blt_inds=select_blt_inds,
phase_center_ids=select_phase_center_ids,
keep_all_metadata=keep_all_metadata,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
)
self._deprecated_phase_helper(
[],
phase_center_radec,
unphase_to_drift,
phase_frame,
orig_phase_frame,
phase_use_ant_pos,
None,
)
@classmethod
def from_file(
cls,
filename,
axis=None,
file_type=None,
read_data=True,
skip_bad_files=False,
background_lsts=True,
ignore_name=False,
# phasing parameters
allow_rephase=None,
phase_center_radec=None,
unphase_to_drift=None,
phase_frame=None,
phase_epoch=None,
orig_phase_frame=None,
phase_use_ant_pos=None,
fix_old_proj=None,
fix_use_ant_pos=True,
make_multi_phase=False,
# selecting parameters
antenna_nums=None,
antenna_names=None,
ant_str=None,
bls=None,
frequencies=None,
freq_chans=None,
times=None,
time_range=None,
lsts=None,
lst_range=None,
polarizations=None,
blt_inds=None,
phase_center_ids=None,
keep_all_metadata=True,
# checking parameters
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
check_autos=True,
fix_autos=True,
# file-type specific parameters
# miriad
phase_type=None,
projected=None,
correct_lat_lon=True,
calc_lst=True,
# FHD
use_model=False,
# MS
data_column="DATA",
pol_order="AIPS",
ignore_single_chan=True,
raise_error=True,
read_weights=True,
# MS & MIR
allow_flex_pol=True,
# uvh5
multidim_index=False,
remove_flex_pol=True,
# uvh5 & mwa_corr_fits
data_array_dtype=np.complex128,
# mwa_corr_fits
use_aoflagger_flags=None,
use_cotter_flags=None,
remove_dig_gains=True,
remove_coarse_band=True,
correct_cable_len=None,
correct_van_vleck=False,
cheby_approx=True,
flag_small_auto_ants=True,
flag_small_sig_ants=None,
propagate_coarse_flags=True,
flag_init=True,
edge_width=80e3,
start_flag="goodtime",
end_flag=0.0,
flag_dc_offset=True,
remove_flagged_ants=True,
phase_to_pointing_center=False,
nsample_array_dtype=np.float32,
# MIR
isource=None,
irec=None,
isb=None,
corrchunk=None,
pseudo_cont=False,
rechunk=None,
):
"""
Initialize a new UVData object by reading the input file.
This method supports a number of different types of files.
Universal parameters (required and optional) are listed directly below,
followed by parameters used by all file types related to phasing, selecting on
read (partial read) and checking. Each file type also has its own set of
optional parameters that are listed at the end of this docstring.
Parameters
----------
filename : str or array_like of str
The file(s) or list(s) (or array(s)) of files to read from.
file_type : str
One of ['uvfits', 'miriad', 'ms', 'uvh5', 'fhd', 'mwa_corr_fits', 'mir']
or None. If None, the code attempts to guess what the file type is.
For miriad and ms types, this is based on the standard directory
structure. For FHD, uvfits and uvh5 files it's based on file
extensions (FHD: .sav, .txt; uvfits: .uvfits; uvh5: .uvh5).
Note that if a list of datasets is passed, the file type is
determined from the first dataset.
axis : str
Axis to concatenate files along. This enables fast concatenation
along the specified axis without the normal checking that all other
metadata agrees. This method does not guarantee correct resulting
objects. Please see the docstring for fast_concat for details.
Allowed values are: 'blt', 'freq', 'polarization'. Only used if
multiple files are passed.
read_data : bool
Read in the data. Not used if file_type is 'ms' or 'mir'.
If set to False, only the metadata will be read in. Setting read_data to
False results in a metdata only object.
skip_bad_files : bool
Option when reading multiple files to catch read errors such that
the read continues even if one or more files are corrupted. Files
that produce errors will be printed. Default is False (files will
not be skipped).
background_lsts : bool
When set to True, the lst_array is calculated in a background thread.
ignore_name : bool
Only relevant when reading in multiple files, which are concatenated into a
single UVData object. Option to ignore the name of the phase center when
combining multiple files, which would otherwise result in an error being
raised because of attributes not matching. Doing so effectively adopts the
name found in the first file read in. Default is False.
Phasing
-------
allow_rephase : bool
Deprecated, will have no effect starting in version 2.4 and will be removed
in version 2.6 because data with different phasing can now be combined.
Allow rephasing of phased file data. Currently defaults to True, which will
cause data to be rephased (to preserve the previous behavior). Set to False
to prevent rephasing and mimic the future behavior.
phase_center_radec : array_like of float
Deprecated and will be removed in version 2.4, instead call `phase` on the
combined object after this method. The phase center to phase the files to
before combining the objects in radians (in the ICRS frame). If set to None
and allow_rephase is True and multiple files are read with different phase
centers, the phase center of the first file will be used.
unphase_to_drift : bool
Deprecated, instead call `unproject_phase` on the combined object after
this method. Unproject the data from the files before combining them.
phase_frame : str
The astropy frame to phase to.Only used if `phase_center_radec` is set.
Deprecated and will be removed in version 2.4 with `phase_center_radec`.
phase_epoch : astropy.time.Time object or str
The epoch to use for phasing. Either an astropy Time object or the
string "J2000" (which is the default).
Note that the epoch is only used to evaluate the ra & dec values,
if the epoch is not J2000, the ra & dec values are interpreted
as FK5 ra/dec values and transformed to J2000, the data are then
phased to the J2000 ra/dec values.
Deprecated and will be removed in version 2.4 with `phase_center_radec`.
orig_phase_frame : str
Deprecated, has no effect. Will be removed in version 2.4.
phase_use_ant_pos : bool
If True, calculate the uvws directly from the antenna positions rather than
from the existing uvws. Only used if `unphase_to_drift` or
`phase_center_radec` are set. Deprecated and will be removed in version 2.4
with `unphase_to_drift` and `phase_center_radec`.
fix_old_proj : bool
Applies a fix to uvw-coordinates and phasing, assuming that the old `phase`
method was used prior to writing the data, which had errors of the order of
one part in 1e4 - 1e5. See the phasing memo for more details. Default is
False, unless reading a UVH5 file that is missing the `phase_center_app_ra`
and `phase_center_app_dec` attributes (as these were introduced at the same
time as the new `phase` method), in which case the default is True.
fix_use_ant_pos : bool
If setting `fix_old_proj` to True, use the antenna positions to derive the
correct uvw-coordinates rather than using the baseline vectors. Default is
True.
make_multi_phase : bool
Deprecated, will be removed in version 2.4. Since all objects now support
multiple phase centers, this has no effect if set to False. The only effect
of setting this to True is to prevent rephasing. A better approach is to set
`allow_rephase=False`.
Selecting
---------
antenna_nums : array_like of int, optional
The antennas numbers to include when reading data into the object
(antenna positions and names for the removed antennas will be retained
unless `keep_all_metadata` is False). This cannot be provided if
`antenna_names` is also provided.
antenna_names : array_like of str, optional
The antennas names to include when reading data into the object
(antenna positions and names for the removed antennas will be retained
unless `keep_all_metadata` is False). This cannot be provided if
`antenna_nums` is also provided.
ant_str : str, optional
A string containing information about what antenna numbers
and polarizations to include when reading data into the object.
Can be 'auto', 'cross', 'all', or combinations of antenna numbers
and polarizations (e.g. '1', '1_2', '1x_2y'). See tutorial for more
examples of valid strings and the behavior of different forms for ant_str.
If '1x_2y,2y_3y' is passed, both polarizations 'xy' and 'yy' will
be kept for both baselines (1, 2) and (2, 3) to return a valid
pyuvdata object.
An ant_str cannot be passed in addition to any of `antenna_nums`,
`antenna_names`, `bls` args or the `polarizations` parameters,
if it is a ValueError will be raised.
bls : list of tuple, optional
A list of antenna number tuples (e.g. [(0, 1), (3, 2)]) or a list of
baseline 3-tuples (e.g. [(0, 1, 'xx'), (2, 3, 'yy')]) specifying baselines
to include when reading data into the object. For length-2 tuples,
the ordering of the numbers within the tuple does not matter. For
length-3 tuples, the polarization string is in the order of the two
antennas. If length-3 tuples are provided, `polarizations` must be
None.
frequencies : array_like of float, optional
The frequencies to include when reading data into the object, each
value passed here should exist in the freq_array.
freq_chans : array_like of int, optional
The frequency channel numbers to include when reading data into the
object. Ignored if read_data is False.
times : array_like of float, optional
The times to include when reading data into the object, each value
passed here should exist in the time_array in the file.
Cannot be used with `time_range`.
time_range : array_like of float, optional
The time range in Julian Date to include when reading data into
the object, must be length 2. Some of the times in the file should
fall between the first and last elements.
Cannot be used with `times`.
lsts : array_like of float, optional
The local sidereal times (LSTs) to keep in the object, each value
passed here should exist in the lst_array. Cannot be used with
`times`, `time_range`, or `lst_range`.
lst_range : array_like of float, optional
The local sidereal time (LST) range in radians to keep in the
object, must be of length 2. Some of the LSTs in the object should
fall between the first and last elements. If the second value is
smaller than the first, the LSTs are treated as having phase-wrapped
around LST = 2*pi = 0, and the LSTs kept on the object will run from
the larger value, through 0, and end at the smaller value.
polarizations : array_like of int, optional
The polarizations numbers to include when reading data into the
object, each value passed here should exist in the polarization_array.
blt_inds : array_like of int, optional
The baseline-time indices to include when reading data into the
object. This is not commonly used.
phase_center_ids : array_like of int, optional
Phase center IDs to include when reading data into the object (effectively
a selection on baseline-times).
keep_all_metadata : bool
Option to keep all the metadata associated with antennas, even those
that do not have data associated with them after the select option.
Checking
--------
run_check : bool
Option to check for the existence and proper shapes of parameters
after after reading in the file (the default is True,
meaning the check will be run). Ignored if read_data is False.
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
Ignored if read_data is False.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters after
reading in the file (the default is True, meaning the acceptable
range check will be done). Ignored if read_data is False.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is True.
Miriad
------
phase_type : str, optional
Deprecated, use the `projected` parameter insted. Option to specify the
phasing status of the data. Options are 'drift', 'phased' or None. 'drift'
is the same as setting the projected parameter to False, 'phased' is the
same as setting the projected parameter to True. Ignored if `projected`
is set.
projected : bool or None
Option to force the dataset to be labelled as projected or unprojected
regardless of the evidence in the file. The default is None which means that
the projection will be set based on the file contents. Be careful setting
this keyword unless you are confident about the contents of the file.
correct_lat_lon : bool
Option to update the latitude and longitude from the known_telescopes
list if the altitude is missing.
calc_lst : bool
Recalculate the LST values that are present within the file, useful in
cases where the "online" calculate values have precision or value errors.
Default is True.
FHD
---
use_model : bool
Option to read in the model visibilities rather than the dirty
visibilities (the default is False, meaning the dirty visibilities
will be read).
MS
--
data_column : str
name of CASA data column to read into data_array. Options are:
'DATA', 'MODEL', or 'CORRECTED_DATA'.
pol_order : str
Option to specify polarizations order convention, options are
'CASA' or 'AIPS'.
ignore_single_chan : bool
Option to ignore single channel spectral windows in measurement sets to
limit object size. Some measurement sets (e.g., those from ALMA) use single
channel spectral windows for recording pseudo-continuum channels or storing
other metadata in the track when the telescopes are not on source. Because
of the way the object is strutured (where all spectral windows are assumed
to be simultaneously recorded), this can significantly inflate the size and
memory footprint of UVData objects. By default, single channel windows are
ignored to avoid this issue, they can be included by setting this parameter
to True.
raise_error : bool
The measurement set format allows for different spectral windows and
polarizations to have different metdata for the same time-baseline
combination, but UVData objects do not. If detected, by default the reader
will throw an error. However, if set to False, the reader will simply give
a warning, and will use the first value read in the file as the "correct"
metadata in the UVData object.
read_weights : bool
Read in the weights from the MS file. If false, the method
will set the `nsamples_array` to the same uniform value (namely 1.0).
allow_flex_pol : bool
If only one polarization per spectral window is read (and the polarization
differs from window to window), allow for the `UVData` object to use
"flexible polarization", which compresses the polarization-axis of various
attributes to be of length 1, sets the `flex_spw_polarization_array`
attribute to define the polarization per spectral window. Default is True.
UVH5
----
multidim_index : bool
If True, attempt to index the HDF5 dataset simultaneously along all data
axes. Otherwise index one axis at-a-time. This only works if data selection
is sliceable along all but one axis. If indices are not well-matched to
data chunks, this can be slow.
remove_flex_pol : bool
If True and if the file is a flex_pol file, convert back to a standard
UVData object.
data_array_dtype : numpy dtype
Datatype to store the output data_array as. Must be either
np.complex64 (single-precision real and imaginary) or np.complex128 (double-
precision real and imaginary). Only used if the datatype of the visibility
data on-disk is not 'c8' or 'c16'.
MWA FITS
--------
use_aoflagger_flags : bool
Option to use aoflagger mwaf flag files. Defaults to true if aoflagger
flag files are submitted.
use_cotter_flags : bool
Being replaced by use_aoflagger_flags and will be removed in v2.4.
remove_dig_gains : bool
Option to divide out digital gains.
remove_coarse_band : bool
Option to divide out coarse band shape.
correct_cable_len : bool
Flag to apply cable length correction.
correct_van_vleck : bool
Flag to apply a van vleck correction.
cheby_approx : bool
Option to implement the van vleck correction with a chebyshev polynomial
approximation. Set to False to run the integral version of the correction.
Only used if correct_van_vleck is True.
flag_small_auto_ants : bool
Option to completely flag any antenna for which the autocorrelation falls
below a threshold found by the Van Vleck correction to indicate bad data.
Specifically, the threshold used is 0.5 * integration_time * channel_width.
If set to False, only the times and frequencies at which the auto is below
the threshold will be flagged for the antenna.
Only used if correct_van_vleck is True.
flag_small_sig_ants : bool
Deprecated, replaced by flag_small_auto_ants and will be removed in v2.4.
propogate_coarse_flags : bool
Option to propogate flags for missing coarse channel integrations
across frequency.
flag_init: bool
Set to True in order to do routine flagging of coarse channel edges, start
or end integrations, or the center fine channel of each coarse
channel. See associated keywords below.
edge_width: float
Only used if flag_init is True. Set to the width to flag on the edge of
each coarse channel, in hz. Errors if not equal to integer multiple of
channel_width. Set to 0 for no edge flagging.
start_flag: float or str
Only used if flag_init is True. The number of seconds to flag at the
beginning of the observation. Set to 0 for no flagging. Default is
'goodtime', which uses information in the metafits file to determine
the length of time that should be flagged. Errors if input is not a
float or 'goodtime'. Errors if float input is not equal to an
integer multiple of the integration time.
end_flag: floats
Only used if flag_init is True. Set to the number of seconds to flag at the
end of the observation. Set to 0 for no flagging. Errors if not an integer
multiple of the integration time.
flag_dc_offset: bool
Only used if flag_init is True. Set to True to flag the center fine channel
of each coarse channel.
remove_flagged_ants : bool
Option to perform a select to remove antennas flagged in the metafits
file. If correct_van_vleck and flag_small_auto_ants are both True then
antennas flagged by the Van Vleck correction are also removed.
phase_to_pointing_center : bool
Flag to phase to the pointing center. Cannot be set if phase_center_radec
is set to a value.
data_array_dtype : numpy dtype
Datatype to store the output data_array as. Must be either
np.complex64 (single-precision real and imaginary) or np.complex128 (double-
precision real and imaginary).
nsample_array_dtype : numpy dtype
Datatype to store the output nsample_array as. Must be either
np.float64 (double-precision), np.float32 (single-precision), or
np.float16 (half-precision). Half-precision is only recommended for
cases where no sampling or averaging of baselines will occur,
because round-off errors can be quite large (~1e-3).
MIR
---
isource : int
Source code for MIR dataset.
irec : int
Receiver code for MIR dataset.
isb : int
Sideband code for MIR dataset.
corrchunk : int
Correlator chunk code for MIR dataset.
pseudo_cont : boolean
Read in only pseudo-continuuum values in MIR dataset.
allow_flex_pol : bool
If only one polarization per spectral window is read (and the polarization
differs from window to window), allow for the `UVData` object to use
"flexible polarization", which compresses the polarization-axis of various
attributes to be of length 1, sets the `flex_spw_polarization_array`
attribute to define the polarization per spectral window. Default is True.
Raises
------
ValueError
If the file_type is not set and cannot be determined from the file name.
If incompatible select keywords are set (e.g. `ant_str` with other
antenna selectors, `times` and `time_range`) or select keywords
exclude all data or if keywords are set to the wrong type.
If phase_center_radec is not None and is not length 2.
"""
uvd = cls()
uvd.read(
filename,
axis=axis,
file_type=file_type,
read_data=read_data,
skip_bad_files=skip_bad_files,
background_lsts=background_lsts,
ignore_name=ignore_name,
# phasing parameters
allow_rephase=allow_rephase,
phase_center_radec=phase_center_radec,
unphase_to_drift=unphase_to_drift,
phase_frame=phase_frame,
phase_epoch=phase_epoch,
orig_phase_frame=orig_phase_frame,
phase_use_ant_pos=phase_use_ant_pos,
fix_old_proj=fix_old_proj,
fix_use_ant_pos=fix_use_ant_pos,
make_multi_phase=make_multi_phase,
# select parameters
antenna_nums=antenna_nums,
antenna_names=antenna_names,
ant_str=ant_str,
bls=bls,
frequencies=frequencies,
freq_chans=freq_chans,
times=times,
time_range=time_range,
lsts=lsts,
lst_range=lst_range,
polarizations=polarizations,
blt_inds=blt_inds,
phase_center_ids=phase_center_ids,
keep_all_metadata=keep_all_metadata,
# checking parameters
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
# start file-type specific parameters
# miriad
phase_type=phase_type,
projected=projected,
correct_lat_lon=correct_lat_lon,
calc_lst=calc_lst,
# FHD
use_model=use_model,
# MS
data_column=data_column,
pol_order=pol_order,
ignore_single_chan=ignore_single_chan,
raise_error=raise_error,
read_weights=read_weights,
# MS & MIR
allow_flex_pol=allow_flex_pol,
# uvh5
multidim_index=multidim_index,
remove_flex_pol=remove_flex_pol,
# uvh5 & mwa_corr_fits
data_array_dtype=data_array_dtype,
nsample_array_dtype=nsample_array_dtype,
# mwa_corr_fits
use_aoflagger_flags=use_aoflagger_flags,
use_cotter_flags=use_cotter_flags,
remove_dig_gains=remove_dig_gains,
remove_coarse_band=remove_coarse_band,
correct_cable_len=correct_cable_len,
correct_van_vleck=correct_van_vleck,
cheby_approx=cheby_approx,
flag_small_auto_ants=flag_small_auto_ants,
flag_small_sig_ants=flag_small_sig_ants,
propagate_coarse_flags=propagate_coarse_flags,
flag_init=flag_init,
edge_width=edge_width,
start_flag=start_flag,
end_flag=end_flag,
flag_dc_offset=flag_dc_offset,
remove_flagged_ants=remove_flagged_ants,
phase_to_pointing_center=phase_to_pointing_center,
# MIR
isource=isource,
irec=irec,
isb=isb,
corrchunk=corrchunk,
pseudo_cont=pseudo_cont,
rechunk=rechunk,
)
return uvd
def write_miriad(
self,
filepath,
clobber=False,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
no_antnums=False,
calc_lst=False,
check_autos=True,
fix_autos=False,
):
"""
Write the data to a miriad file.
Parameters
----------
filename : str
The miriad root directory to write to.
clobber : bool
Option to overwrite the filename if the file already exists.
run_check : bool
Option to check for the existence and proper shapes of parameters
after before writing the file (the default is True,
meaning the check will be run).
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
run_check_acceptability : bool
Option to check acceptable range of the values of parameters before
writing the file (the default is True, meaning the acceptable
range check will be done).
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
no_antnums : bool
Option to not write the antnums variable to the file.
Should only be used for testing purposes.
calc_lst : bool
Recalculate the LST values upon writing the file. This is done to perform
higher-precision accounting for the difference in MIRAD timestamps vs
pyuvdata (the former marks the beginning of an integration, the latter
marks the midpoint). Default is False, which instead uses a simple formula
for correcting the LSTs, expected to be accurate to approximately 0.1 µsec
precision.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is False.
Raises
------
ValueError
If the frequencies are not evenly spaced or are separated by more
than their channel width or if the UVData object is a metadata only object.
TypeError
If any entry in extra_keywords is not a single string or number.
"""
if self.metadata_only:
raise ValueError("Cannot write out metadata only objects to a miriad file.")
miriad_obj = self._convert_to_filetype("miriad")
# If we have a flex-pol dataset, remove it so that we can pass the data to the
# writer without issue. We create a copy because otherwise we run the risk of
# messing w/ the metadata of the original object, and because the overhead of
# doing so is generally smaller than that from removing flex-pol.
# TODO: reconsider this. The overhead is not smaller if `convert_to_flex_pol`
# was used.
if self.flex_spw_polarization_array is not None:
miriad_obj = miriad_obj.copy()
miriad_obj.remove_flex_pol()
miriad_obj.write_miriad(
filepath,
clobber=clobber,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
no_antnums=no_antnums,
calc_lst=calc_lst,
check_autos=check_autos,
fix_autos=fix_autos,
)
del miriad_obj
def write_mir(self, filepath):
"""
Write the data to a mir file.
Parameters
----------
filename : str
The mir root directory to write to.
Raises
------
ValueError
If the UVData object is a metadata only object.
NotImplementedError
Method is not fully implemented yet, and thus will raise an error
"""
if self.metadata_only:
raise ValueError("Cannot write out metadata only objects to a mir file.")
mir_obj = self._convert_to_filetype("mir")
mir_obj.write_mir(filepath)
del mir_obj # pragma: nocover
def write_ms(
self,
filename,
force_phase=False,
clobber=False,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
check_autos=True,
fix_autos=False,
):
"""
Write a CASA measurement set (MS).
Parameters
----------
filename : str
The measurement set file path to write to (a measurement set is really
a folder with many files).
force_phase : bool
Option to automatically phase unprojected data to zenith of the first
timestamp.
clobber : bool
Option to overwrite the file if it already exists.
run_check : bool
Option to check for the existence and proper shapes of parameters
before writing the file.
check_extra : bool
Option to check optional parameters as well as required ones.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters before
writing the file.
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is False.
Raises
------
ValueError
If the UVData object is a metadata only object.
"""
if self.metadata_only:
raise ValueError(
"Cannot write out metadata only objects to a measurement set file."
)
ms_obj = self._convert_to_filetype("ms")
ms_obj.write_ms(
filename,
force_phase=force_phase,
clobber=clobber,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
del ms_obj
def write_uvfits(
self,
filename,
spoof_nonessential=False,
write_lst=True,
force_phase=False,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
check_autos=True,
fix_autos=False,
):
"""
Write the data to a uvfits file.
If using this method to write out a data set for import into CASA, users should
be aware that the `importuvifts` task does not currently support reading in
data sets where the number of antennas is > 255. If writing out such a data set
for use in CASA, we suggest using the measurement set writer (`UVData.write_ms`)
instead.
Parameters
----------
filename : str
The uvfits file to write to.
spoof_nonessential : bool
Deprecated and has no effect. Values are automatically set to their best
known values if not previously set.
write_lst : bool
Option to write the LSTs to the metadata (random group parameters).
force_phase: : bool
Option to automatically phase unprojected data to zenith of the first
timestamp.
run_check : bool
Option to check for the existence and proper shapes of parameters
after before writing the file (the default is True,
meaning the check will be run).
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
run_check_acceptability : bool
Option to check acceptable range of the values of parameters before
writing the file (the default is True, meaning the acceptable
range check will be done).
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is False.
Raises
------
ValueError
Any blts are unprojected and `force_phase` keyword is not set.
If the frequencies are not evenly spaced or are separated by more
than their channel width.
The polarization values are not evenly spaced.
If the `timesys` parameter is not set to "UTC".
If the UVData object is a metadata only object.
TypeError
If any entry in extra_keywords is not a single string or number.
"""
if self.metadata_only:
raise ValueError("Cannot write out metadata only objects to a uvfits file.")
uvfits_obj = self._convert_to_filetype("uvfits")
# If we have a flex-pol dataset, remove it so that we can pass the data to the
# writer without issue. We create a copy because otherwise we run the risk of
# messing w/ the metadata of the original object, and because the overhead of
# doing so is generally smaller than that from removing flex-pol.
# TODO: reconsider this. The overhead is not smaller if `convert_to_flex_pol`
# was used.
if self.flex_spw_polarization_array is not None:
uvfits_obj = uvfits_obj.copy()
uvfits_obj.remove_flex_pol()
uvfits_obj.write_uvfits(
filename,
spoof_nonessential=spoof_nonessential,
write_lst=write_lst,
force_phase=force_phase,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
del uvfits_obj
def write_uvh5(
self,
filename,
clobber=False,
chunks=True,
data_compression=None,
flags_compression="lzf",
nsample_compression="lzf",
data_write_dtype=None,
run_check=True,
check_extra=True,
run_check_acceptability=True,
strict_uvw_antpos_check=False,
check_autos=True,
fix_autos=False,
):
"""
Write a completely in-memory UVData object to a UVH5 file.
Parameters
----------
filename : str
The UVH5 file to write to.
clobber : bool
Option to overwrite the file if it already exists.
chunks : tuple or bool
h5py.create_dataset chunks keyword. Tuple for chunk shape,
True for auto-chunking, None for no chunking. Default is True.
data_compression : str
HDF5 filter to apply when writing the data_array. Default is None
(no filter/compression). In addition to the normal HDF5 filter values, the
user may specify "bitshuffle" which will set the compression to `32008` for
bitshuffle and will set the `compression_opts` to `(0, 2)` to allow
bitshuffle to automatically determine the block size and to use the LZF
filter after bitshuffle. Using `bitshuffle` requires having the
`hdf5plugin` package installed. Dataset must be chunked to use compression.
flags_compression : str
HDF5 filter to apply when writing the flags_array. Default is "lzf"
for the LZF filter. Dataset must be chunked.
nsample_compression : str
HDF5 filter to apply when writing the nsample_array. Default is "lzf"
for the LZF filter. Dataset must be chunked.
data_write_dtype : numpy dtype
datatype of output visibility data. If 'None', then the same datatype
as data_array will be used. Otherwise, a numpy dtype object must be
specified with an 'r' field and an 'i' field for real and imaginary
parts, respectively. See uvh5.py for an example of defining such a datatype.
run_check : bool
Option to check for the existence and proper shapes of parameters
after before writing the file (the default is True,
meaning the check will be run).
check_extra : bool
Option to check optional parameters as well as required ones (the
default is True, meaning the optional parameters will be checked).
run_check_acceptability : bool
Option to check acceptable range of the values of parameters before
writing the file (the default is True, meaning the acceptable
range check will be done).
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
check_autos : bool
Check whether any auto-correlations have non-zero imaginary values in
data_array (which should not mathematically exist). Default is True.
fix_autos : bool
If auto-correlations with imaginary values are found, fix those values so
that they are real-only in data_array. Default is False.
Raises
------
ValueError
If the UVData object is a metadata only object.
"""
if self.metadata_only:
raise ValueError(
"Cannot write out metadata only objects to a uvh5 file. To initialize "
"a uvh5 file for partial writing, use the `initialize_uvh5_file` "
"method."
)
uvh5_obj = self._convert_to_filetype("uvh5")
uvh5_obj.write_uvh5(
filename,
clobber=clobber,
chunks=chunks,
data_compression=data_compression,
flags_compression=flags_compression,
nsample_compression=nsample_compression,
data_write_dtype=data_write_dtype,
run_check=run_check,
check_extra=check_extra,
run_check_acceptability=run_check_acceptability,
strict_uvw_antpos_check=strict_uvw_antpos_check,
check_autos=check_autos,
fix_autos=fix_autos,
)
del uvh5_obj
def initialize_uvh5_file(
self,
filename,
clobber=False,
chunks=True,
data_compression=None,
flags_compression="lzf",
nsample_compression="lzf",
data_write_dtype=None,
):
"""
Initialize a UVH5 file on disk with the header metadata and empty data arrays.
Parameters
----------
filename : str
The UVH5 file to write to.
clobber : bool
Option to overwrite the file if it already exists.
chunks : tuple or bool
h5py.create_dataset chunks keyword. Tuple for chunk shape,
True for auto-chunking, None for no chunking. Default is True.
data_compression : str
HDF5 filter to apply when writing the data_array. Default is None
(no filter/compression). In addition to the normal HDF5 filter values, the
user may specify "bitshuffle" which will set the compression to `32008` for
bitshuffle and will set the `compression_opts` to `(0, 2)` to allow
bitshuffle to automatically determine the block size and to use the LZF
filter after bitshuffle. Using `bitshuffle` requires having the
`hdf5plugin` package installed. Dataset must be chunked to use compression.
flags_compression : str
HDF5 filter to apply when writing the flags_array. Default is "lzf"
for the LZF filter. Dataset must be chunked.
nsample_compression : str
HDF5 filter to apply when writing the nsample_array. Default is "lzf"
for the LZF filter. Dataset must be chunked.
data_write_dtype : numpy dtype
datatype of output visibility data. If 'None', then the same datatype
as data_array will be used. Otherwise, a numpy dtype object must be
specified with an 'r' field and an 'i' field for real and imaginary
parts, respectively. See uvh5.py for an example of defining such a datatype.
Notes
-----
When partially writing out data, this function should be called first
to initialize the file on disk. The data is then actually written by
calling the write_uvh5_part method, with the same filename as the one
specified in this function. See the tutorial for a worked example.
"""
uvh5_obj = self._convert_to_filetype("uvh5")
uvh5_obj.initialize_uvh5_file(
filename,
clobber=clobber,
chunks=chunks,
data_compression=data_compression,
flags_compression=flags_compression,
nsample_compression=nsample_compression,
data_write_dtype=data_write_dtype,
)
del uvh5_obj
def write_uvh5_part(
self,
filename,
data_array,
flags_array,
nsample_array,
check_header=True,
antenna_nums=None,
antenna_names=None,
ant_str=None,
bls=None,
frequencies=None,
freq_chans=None,
times=None,
time_range=None,
lsts=None,
lst_range=None,
polarizations=None,
blt_inds=None,
phase_center_ids=None,
add_to_history=None,
run_check_acceptability=True,
fix_autos=False,
):
"""
Write data to a UVH5 file that has already been initialized.
Parameters
----------
filename : str
The UVH5 file to write to. It must already exist, and is assumed to
have been initialized with initialize_uvh5_file.
data_array : ndarray
The data to write to disk. A check is done to ensure that the
dimensions of the data passed in conform to the ones specified by
the "selection" arguments.
flags_array : ndarray
The flags array to write to disk. A check is done to ensure that the
dimensions of the data passed in conform to the ones specified by
the "selection" arguments.
nsample_array : ndarray
The nsample array to write to disk. A check is done to ensure that the
dimensions of the data passed in conform to the ones specified by
the "selection" arguments.
check_header : bool
Option to check that the metadata present in the header on disk
matches that in the object.
antenna_nums : array_like of int, optional
The antennas numbers to include when writing data into the file
(antenna positions and names for the removed antennas will be retained).
This cannot be provided if `antenna_names` is also provided.
antenna_names : array_like of str, optional
The antennas names to include when writing data into the file
(antenna positions and names for the removed antennas will be retained).
This cannot be provided if `antenna_nums` is also provided.
bls : list of tuple, optional
A list of antenna number tuples (e.g. [(0, 1), (3, 2)]) or a list of
baseline 3-tuples (e.g. [(0, 1, 'xx'), (2, 3, 'yy')]) specifying baselines
to include when writing data into the file. For length-2 tuples,
the ordering of the numbers within the tuple does not matter. For
length-3 tuples, the polarization string is in the order of the two
antennas. If length-3 tuples are provided, `polarizations` must be
None.
ant_str : str, optional
A string containing information about what antenna numbers
and polarizations to include writing data into the file.
Can be 'auto', 'cross', 'all', or combinations of antenna numbers
and polarizations (e.g. '1', '1_2', '1x_2y'). See tutorial for more
examples of valid strings and the behavior of different forms for ant_str.
If '1x_2y,2y_3y' is passed, both polarizations 'xy' and 'yy' will
be kept for both baselines (1, 2) and (2, 3) to return a valid
pyuvdata object.
An ant_str cannot be passed in addition to any of `antenna_nums`,
`antenna_names`, `bls` args or the `polarizations` parameters,
if it is a ValueError will be raised.
frequencies : array_like of float, optional
The frequencies to include when writing data into the file, each
value passed here should exist in the freq_array.
freq_chans : array_like of int, optional
The frequency channel numbers to include writing data into the file.
times : array_like of float, optional
The times to include when writing data into the file, each value
passed here should exist in the time_array.
time_range : array_like of float, optional
The time range in Julian Date to include when writing data to the
file, must be length 2. Some of the times in the object should fall
between the first and last elements. Cannot be used with `times`.
lsts : array_like of float, optional
The local sidereal times (LSTs) to keep in the object, each value
passed here should exist in the lst_array. Cannot be used with
`times`, `time_range`, or `lst_range`.
lst_range : array_like of float, optional
The local sidereal time (LST) range in radians to keep in the
object, must be of length 2. Some of the LSTs in the object should
fall between the first and last elements. If the second value is
smaller than the first, the LSTs are treated as having phase-wrapped
around LST = 2*pi = 0, and the LSTs kept on the object will run from
the larger value, through 0, and end at the smaller value.
polarizations : array_like of int, optional
The polarizations numbers to include when writing data into the file,
each value passed here should exist in the polarization_array.
blt_inds : array_like of int, optional
The baseline-time indices to include when writing data into the file.
This is not commonly used.
phase_center_ids : array_like of int, optional
Phase center IDs to include when writing data into the file (effectively
a selection on baseline-times).
add_to_history : str
String to append to history before write out. Default is no appending.
run_check_acceptability : bool
Option to check acceptable range of the values of parameters before
writing the file (the default is True, meaning the acceptable
range check will be done).
strict_uvw_antpos_check : bool
Option to raise an error rather than a warning if the check that
uvws match antenna positions does not pass.
fix_autos : bool
Force the auto-correlations to be real-only values in data_array.
Default is False.
"""
if fix_autos:
self._fix_autos()
uvh5_obj = self._convert_to_filetype("uvh5")
uvh5_obj.write_uvh5_part(
filename,
data_array,
flags_array,
nsample_array,
check_header=check_header,
antenna_nums=antenna_nums,
antenna_names=antenna_names,
bls=bls,
ant_str=ant_str,
frequencies=frequencies,
freq_chans=freq_chans,
times=times,
time_range=time_range,
lsts=lsts,
lst_range=lst_range,
polarizations=polarizations,
blt_inds=blt_inds,
phase_center_ids=phase_center_ids,
add_to_history=add_to_history,
run_check_acceptability=run_check_acceptability,
)
del uvh5_obj
def normalize_by_autos(self, skip_autos=True, invert=False):
"""
Normalize cross-correlations by auto-correlation data.
Normalizes the cross-correlations by the geometric mean of the autocorrelations
that make up the antenna pair for a given baseline. Useful for converting
arbitrarily-scaled data into correlation coefficients (which can sometimes be
more readily converted to a flux scale).
Parameters
----------
skip_autos : bool
If set to True, the method will skip over records which correspond to the
auto-correlations (which would otherwise simply be equal to all ones), which
can be useful if intending to undo or redo the normalization at some point
later. Default is True.
invert : bool
If set to True, will multiply by the geometric mean of the autos instead of
dividing by it. Useful for undoing previous normalizations (which also
requires setting `skip_autos=True` on previous calls).
"""
# First up, let's double-check to make sure that we can actually normalize this
# data set, and if not, bail early.
if not np.any(self.ant_1_array == self.ant_2_array):
raise ValueError(
"No autos available in this data set to do normalization with."
)
# Now figure out how the different polarizations map to the autos
pol_groups = []
pol_list = list(self.polarization_array)
for pol in pol_list:
try:
feed_pols = uvutils.POL_TO_FEED_DICT[uvutils.POL_NUM2STR_DICT[pol]]
pol_groups.append(
[
pol_list.index(uvutils.POL_STR2NUM_DICT[item + item])
for item in feed_pols
]
)
except KeyError:
# If we run into a key error, it means that one of the dicts above does
# not have a match to the given polarization, in which case assume that
# this pol its it's own auto.
pol_groups.append([pol_list.index(pol)] * 2)
except ValueError as err:
# If we have an value error, it means that the pol that _would_ be the
# auto is not found in the data, in which case we throw an error.
raise ValueError(
"Cannot normalize {pol}, matching pols for autos not found.".format(
pol=uvutils.POL_NUM2STR_DICT[pol]
)
) from err
# Each pol group contains the index positions for the "auto" polarizations,
# so we can grab all of the auto pols by searching for the unique values
auto_pols = list(np.unique(pol_groups))
# Grab references to data and flags, to manipulate later
data_arr = self.data_array
flag_arr = self.flag_array
if not self.future_array_shapes:
data_arr = np.squeeze(data_arr, 1)
flag_arr = np.squeeze(flag_arr, 1)
# We need to match baselines in a single integration, so figure out how to
# group the data by time.
time_ordered = not np.any(self.time_array[:-1] > self.time_array[1:])
blt_idx = np.arange(self.Nblts) if time_ordered else np.argsort(self.time_array)
ordered_time = self.time_array[blt_idx if time_ordered else ...]
# Normally time has a fixed atol, but verify that this is the case
time_tol = self._time_array.tols[1]
assert self._time_array.tols[0] == 0
# Start searching through time, keeping in mind that the data are time ordered.
time_groups = []
start_idx = end_idx = 0
while self.Nblts != start_idx:
# Search sorted will find where one would insert the value of the current
# time + the time tol, which will give us the ending index of the group.
end_idx += np.searchsorted(
ordered_time[start_idx:], ordered_time[start_idx] + time_tol, "right"
)
# Create this new grouping of blts
time_groups.append(blt_idx[start_idx:end_idx])
start_idx = end_idx
# Not start the heavy lifting
for group in time_groups:
# Now start going through the groups
ant_1_arr = self.ant_1_array[group]
ant_2_arr = self.ant_2_array[group]
norm_dict = {}
flag_dict = {}
for grp_idx, ant1, ant2 in zip(group, ant_1_arr, ant_2_arr):
# Tabulate up front the normalization for each auto-correlation
# spectrum, which will save some work downstream
if ant1 != ant2:
continue
norm_dict[ant1] = {}
flag_dict[ant1] = {}
for pol in auto_pols:
# Autos _should_ be real only, extracting them out like this will
# make the multipliction later a bit faster.
auto_data = data_arr[grp_idx, :, pol].real
auto_flag = flag_arr[grp_idx, :, pol] | ~(auto_data > 0)
norm_data = np.zeros_like(auto_data)
if invert:
norm_data = np.sqrt(auto_data, where=~auto_flag, out=norm_data)
else:
norm_data = np.reciprocal(
auto_data, where=~auto_flag, out=norm_data
)
norm_data = np.sqrt(norm_data, out=norm_data)
norm_dict[ant1][pol] = norm_data
flag_dict[ant1][pol] = auto_flag
# Now that we have the autos "normalization-ready", we can get to
# actually normalizing the crosses.
for grp_idx, ant1, ant2 in zip(group, ant_1_arr, ant_2_arr):
try:
for jdx, (pol1, pol2) in enumerate(pol_groups):
# Proceed pol by pol
if skip_autos and (ant1 == ant2) and (pol1 == pol2):
continue
data_arr[grp_idx, :, jdx] *= (
norm_dict[ant1][pol1] * norm_dict[ant2][pol2]
)
flag_arr[grp_idx, :, jdx] |= (
flag_dict[ant1][pol1] | flag_dict[ant2][pol2]
)
except KeyError:
# If no data found for this antenna, then flag the whole blt
flag_arr[grp_idx] = True
Computing file changes ...
