{ "cells": [ { "cell_type": "markdown", "id": "d398b960-06bd-4e61-a3a8-2e4944b48170", "metadata": {}, "source": [ "# ngEHT conversion guide" ] }, { "cell_type": "code", "execution_count": 1, "id": "14a699d5-bf6e-4ebe-bfa5-dcccbec0ec2e", "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "XRADIO version 1.1.3 already installed.\n" ] } ], "source": [ "from importlib.metadata import version\n", "import os\n", "\n", "try:\n", " import xradio\n", "\n", " print(\"XRADIO version\", version(\"xradio\"), \"already installed.\")\n", "except ImportError as e:\n", " print(e)\n", " print(\"Installing XRADIO\")\n", "\n", " os.system(\"pip install xradio\")\n", "\n", " import xradio\n", "\n", " print(\"xradio version\", version(\"xradio\"), \" installed.\")" ] }, { "cell_type": "markdown", "id": "a939f97e-6c76-47b0-aef5-256df2e908f9", "metadata": {}, "source": [ "## Download dataset" ] }, { "cell_type": "markdown", "id": "805daf71", "metadata": {}, "source": [ "fitsidi from https://almascience.eso.org/almadata/ec/eht/2016.1.01114.V/group.uid___A001_X87c_X245.ec_jlgomez.e17a10-7-hi-oj287-3C279-fits.tgz\n", "\n", "```python\n", "importfitsidi('E17A10.0.bin0000.source0000.FITS',vis='E17A10.0.bin0000.source0000.ms')\n", "mstransform(vis='E17A10.0.bin0000.source0000.ms',outputvis='ngEHT_E17A10.0.bin0000.source0000_split_lsrk.ms',spw='8:55~61,29:20~27', regridms=True,outframe='lsrk',datacolumn='all')\n", "```" ] }, { "cell_type": "code", "execution_count": 2, "id": "134a99e3-c5a2-443c-96d4-f44fae59555a", "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[\u001b[38;2;128;05;128m2026-04-20 15:19:36,129\u001b[0m] \u001b[38;2;50;50;205m INFO\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m Initializing download... \n", "[\u001b[38;2;128;05;128m2026-04-20 15:19:36,130\u001b[0m] \u001b[38;2;50;50;205m INFO\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m File already exists: /Users/vdesouza/work/xradio/docs/source/measurement_set/guides/ngEHT_E17A10.0.bin0000.source0000_split.ms \n" ] } ], "source": [ "import toolviper\n", "\n", "toolviper.utils.data.download(file=\"ngEHT_E17A10.0.bin0000.source0000_split.ms\")" ] }, { "cell_type": "markdown", "id": "f8435c96-7010-4b79-8be6-51f7f6993b5f", "metadata": {}, "source": [ "## Convert to Processing Set" ] }, { "cell_type": "code", "execution_count": 3, "id": "ee52d124-2c17-450b-879f-1f86f0ae265c", "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[\u001b[38;2;128;05;128m2026-04-20 15:19:36,642\u001b[0m] \u001b[38;2;50;50;205m INFO\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m Updated partition scheme used: ['DATA_DESC_ID', 'OBSERVATION_ID'] \n", "[\u001b[38;2;128;05;128m2026-04-20 15:19:36,643\u001b[0m] \u001b[38;2;50;50;205m INFO\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m Number of partitions: 2 \n", "[\u001b[38;2;128;05;128m2026-04-20 15:19:36,643\u001b[0m] \u001b[38;2;50;50;205m INFO\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m OBSERVATION_ID [0], DDI [0], STATE [None], FIELD [0], SCAN [0], EPHEMERIS [None] \n", "[\u001b[38;2;128;05;128m2026-04-20 15:19:36,717\u001b[0m] \u001b[38;2;255;160;0m WARNING\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m Source_id is -1. No source information will be included in the field_and_source_xds. \n", "[\u001b[38;2;128;05;128m2026-04-20 15:19:37,145\u001b[0m] \u001b[38;2;50;50;205m INFO\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m OBSERVATION_ID [0], DDI [1], STATE [None], FIELD [0], SCAN [0], EPHEMERIS [None] \n", "[\u001b[38;2;128;05;128m2026-04-20 15:19:37,217\u001b[0m] \u001b[38;2;255;160;0m WARNING\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m Source_id is -1. No source information will be included in the field_and_source_xds. \n" ] } ], "source": [ "from xradio.measurement_set import convert_msv2_to_processing_set\n", "\n", "ms_file = \"ngEHT_E17A10.0.bin0000.source0000_split.ms\"\n", "\n", "main_chunksize = {\"frequency\": 1, \"time\": 20} # baseline, polarization\n", "outfile = \"ngEHT_E17A10.0.bin0000.source0000_split_lsrk.ps.zarr\"\n", "convert_msv2_to_processing_set(\n", " in_file=ms_file,\n", " out_file=outfile,\n", " parallel_mode=\"none\",\n", " persistence_mode='w',\n", " main_chunksize=main_chunksize,\n", ")" ] }, { "cell_type": "markdown", "id": "fbd02679-0df8-4fa5-8036-6f22f534e386", "metadata": {}, "source": [ "## Processing Set" ] }, { "cell_type": "code", "execution_count": 4, "id": "dab986ca-55f8-4a4a-ba59-9c97fcc2ca84", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
| \n", " | name | \n", "scan_intents | \n", "shape | \n", "execution_block_UID | \n", "polarization | \n", "scan_name | \n", "spw_name | \n", "spw_intents | \n", "field_name | \n", "source_name | \n", "line_name | \n", "field_coords | \n", "session_reference_UID | \n", "scheduling_block_UID | \n", "project_UID | \n", "start_frequency | \n", "end_frequency | \n", "
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 0 | \n", "ngEHT_E17A10.0.bin0000.source0000_split_0 | \n", "[None] | \n", "(2250, 20, 7, 4) | \n", "--- | \n", "[RR, RL, LR, LL] | \n", "[0] | \n", "spw_0 | \n", "[UNSPECIFIED] | \n", "[3C279_0] | \n", "[Unknown] | \n", "[] | \n", "[fk5, 12h56m11.17s, -5d47m21.52s] | \n", "--- | \n", "--- | \n", "E17A10 | \n", "2.286593e+11 | \n", "2.286623e+11 | \n", "
| 1 | \n", "ngEHT_E17A10.0.bin0000.source0000_split_1 | \n", "[None] | \n", "(2250, 20, 8, 4) | \n", "--- | \n", "[RR, RL, LR, LL] | \n", "[0] | \n", "spw_1 | \n", "[UNSPECIFIED] | \n", "[3C279_0] | \n", "[Unknown] | \n", "[] | \n", "[fk5, 12h56m11.17s, -5d47m21.52s] | \n", "--- | \n", "--- | \n", "E17A10 | \n", "2.298723e+11 | \n", "2.298758e+11 | \n", "
<xarray.DataTree 'ngEHT_E17A10.0.bin0000.source0000_split_0'>\n",
"Group: /ngEHT_E17A10.0.bin0000.source0000_split_0\n",
"│ Dimensions: (time: 2250, baseline_id: 20, frequency: 7,\n",
"│ polarization: 4, uvw_label: 3)\n",
"│ Coordinates:\n",
"│ * time (time) float64 18kB 1.492e+09 ... 1.492e+09\n",
"│ field_name (time) <U27 243kB dask.array<chunksize=(2250,), meta=np.ndarray>\n",
"│ scan_name (time) <U21 189kB dask.array<chunksize=(2250,), meta=np.ndarray>\n",
"│ * baseline_id (baseline_id) int64 160B 0 1 2 3 ... 16 17 18 19\n",
"│ baseline_antenna1_name (baseline_id) <U2 160B dask.array<chunksize=(20,), meta=np.ndarray>\n",
"│ baseline_antenna2_name (baseline_id) <U2 160B dask.array<chunksize=(20,), meta=np.ndarray>\n",
"│ * frequency (frequency) float64 56B 2.287e+11 ... 2.287e+11\n",
"│ * polarization (polarization) <U2 32B 'RR' 'RL' 'LR' 'LL'\n",
"│ * uvw_label (uvw_label) <U1 12B 'u' 'v' 'w'\n",
"│ Data variables:\n",
"│ EFFECTIVE_INTEGRATION_TIME (time, baseline_id) float64 360kB dask.array<chunksize=(20, 20), meta=np.ndarray>\n",
"│ FLAG (time, baseline_id, frequency, polarization) bool 1MB dask.array<chunksize=(20, 20, 1, 4), meta=np.ndarray>\n",
"│ TIME_CENTROID (time, baseline_id) float64 360kB dask.array<chunksize=(20, 20), meta=np.ndarray>\n",
"│ UVW (time, baseline_id, uvw_label) float64 1MB dask.array<chunksize=(20, 20, 3), meta=np.ndarray>\n",
"│ VISIBILITY (time, baseline_id, frequency, polarization) complex64 10MB dask.array<chunksize=(20, 20, 1, 4), meta=np.ndarray>\n",
"│ WEIGHT (time, baseline_id, frequency, polarization) float32 5MB dask.array<chunksize=(20, 20, 1, 4), meta=np.ndarray>\n",
"│ Attributes:\n",
"│ creation_date: 2026-04-20T21:19:36.659187+00:00\n",
"│ creator: {'software_name': 'xradio', 'version': '1.1.3'}\n",
"│ data_groups: {'base': {'correlated_data': 'VISIBILITY', 'date': '20...\n",
"│ observation_info: {'observer': ['PLUTO'], 'observing_log': '[]', 'projec...\n",
"│ processor_info: {'sub_type': '', 'type': ''}\n",
"│ schema_version: 4.0.0\n",
"│ type: visibility\n",
"├── Group: /ngEHT_E17A10.0.bin0000.source0000_split_0/antenna_xds\n",
"│ Dimensions: (antenna_name: 6, cartesian_pos_label: 3,\n",
"│ receptor_label: 2)\n",
"│ Coordinates:\n",
"│ * antenna_name (antenna_name) <U2 48B 'AA' 'AP' 'AZ' 'LM' 'PV' 'SP'\n",
"│ mount (antenna_name) <U16 384B dask.array<chunksize=(6,), meta=np.ndarray>\n",
"│ station_name (antenna_name) <U2 48B dask.array<chunksize=(6,), meta=np.ndarray>\n",
"│ telescope_name (antenna_name) <U4 96B dask.array<chunksize=(6,), meta=np.ndarray>\n",
"│ * cartesian_pos_label (cartesian_pos_label) <U1 12B 'x' 'y' 'z'\n",
"│ * receptor_label (receptor_label) <U5 40B 'pol_0' 'pol_1'\n",
"│ polarization_type (antenna_name, receptor_label) <U1 48B dask.array<chunksize=(6, 2), meta=np.ndarray>\n",
"│ Data variables:\n",
"│ ANTENNA_DISH_DIAMETER (antenna_name) float64 48B dask.array<chunksize=(6,), meta=np.ndarray>\n",
"│ ANTENNA_POSITION (antenna_name, cartesian_pos_label) float64 144B dask.array<chunksize=(6, 3), meta=np.ndarray>\n",
"│ ANTENNA_RECEPTOR_ANGLE (antenna_name, receptor_label) float64 96B dask.array<chunksize=(6, 2), meta=np.ndarray>\n",
"│ Attributes:\n",
"│ overall_telescope_name: VLBA\n",
"│ relocatable_antennas: False\n",
"│ type: antenna\n",
"└── Group: /ngEHT_E17A10.0.bin0000.source0000_split_0/field_and_source_base_xds\n",
" Dimensions: (field_name: 1, sky_dir_label: 2)\n",
" Coordinates:\n",
" * field_name (field_name) <U27 108B '3C279_0'\n",
" source_name (field_name) <U7 28B dask.array<chunksize=(1,), meta=np.ndarray>\n",
" * sky_dir_label (sky_dir_label) <U3 24B 'ra' 'dec'\n",
" Data variables:\n",
" FIELD_PHASE_CENTER_DIRECTION (field_name, sky_dir_label) float64 16B dask.array<chunksize=(1, 2), meta=np.ndarray>\n",
" Attributes:\n",
" type: field_and_source<xarray.DataTree 'antenna_xds'>\n",
"Group: /\n",
" Dimensions: (time: 2250, baseline_id: 20, frequency: 7,\n",
" polarization: 4, uvw_label: 3, antenna_name: 6,\n",
" cartesian_pos_label: 3, receptor_label: 2)\n",
" Coordinates:\n",
" * time (time) float64 18kB 1.492e+09 ... 1.492e+09\n",
" * baseline_id (baseline_id) int64 160B 0 1 2 3 4 ... 16 17 18 19\n",
" * frequency (frequency) float64 56B 2.287e+11 ... 2.287e+11\n",
" * polarization (polarization) <U2 32B 'RR' 'RL' 'LR' 'LL'\n",
" * uvw_label (uvw_label) <U1 12B 'u' 'v' 'w'\n",
" * antenna_name (antenna_name) <U2 48B 'AA' 'AP' 'AZ' 'LM' 'PV' 'SP'\n",
" mount (antenna_name) <U16 384B 'ALT-AZ' ... 'ALT-AZ'\n",
" station_name (antenna_name) <U2 48B 'AA' 'AP' 'AZ' 'LM' 'PV' 'SP'\n",
" telescope_name (antenna_name) <U4 96B 'VLBA' 'VLBA' ... 'VLBA'\n",
" * cartesian_pos_label (cartesian_pos_label) <U1 12B 'x' 'y' 'z'\n",
" * receptor_label (receptor_label) <U5 40B 'pol_0' 'pol_1'\n",
" polarization_type (antenna_name, receptor_label) <U1 48B 'R' ... 'L'\n",
" Data variables:\n",
" ANTENNA_DISH_DIAMETER (antenna_name) float64 48B 12.0 12.0 ... 30.0 10.0\n",
" ANTENNA_POSITION (antenna_name, cartesian_pos_label) float64 144B ...\n",
" ANTENNA_RECEPTOR_ANGLE (antenna_name, receptor_label) float64 96B 0.0 .....\n",
" Attributes:\n",
" overall_telescope_name: VLBA\n",
" relocatable_antennas: False\n",
" type: antenna<xarray.DatasetView> Size: 18kB\n",
"Dimensions: (field_name: 1, sky_dir_label: 2,\n",
" baseline_id: 20, frequency: 7,\n",
" polarization: 4, time: 2250, uvw_label: 3)\n",
"Coordinates:\n",
" * field_name (field_name) <U27 108B '3C279_0'\n",
" source_name (field_name) <U7 28B dask.array<chunksize=(1,), meta=np.ndarray>\n",
" * sky_dir_label (sky_dir_label) <U3 24B 'ra' 'dec'\n",
" * baseline_id (baseline_id) int64 160B 0 1 2 3 ... 17 18 19\n",
" * frequency (frequency) float64 56B 2.287e+11 ... 2.287...\n",
" * polarization (polarization) <U2 32B 'RR' 'RL' 'LR' 'LL'\n",
" * time (time) float64 18kB 1.492e+09 ... 1.492e+09\n",
" * uvw_label (uvw_label) <U1 12B 'u' 'v' 'w'\n",
"Data variables:\n",
" FIELD_PHASE_CENTER_DIRECTION (field_name, sky_dir_label) float64 16B dask.array<chunksize=(1, 2), meta=np.ndarray>\n",
"Attributes:\n",
" type: field_and_source"
],
"text/plain": [
"ngEHT_E17A10.0.bin0000.s ... ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00\n\n", "text/plain": "ngEHT_E17A10.0.bin0000.s ... \u001b[38;2;114;156;31m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[35m100%\u001b[0m \u001b[36m0:00:00\u001b[0m\n" }, "metadata": {}, "output_type": "display_data" } ] } }, "bb5a5c3158474547aadc445a997eb526": { "model_module": "@jupyter-widgets/base", "model_module_version": "2.0.0", "model_name": "LayoutModel", "state": {} } }, "version_major": 2, "version_minor": 0 } } }, "nbformat": 4, "nbformat_minor": 5 }