{ "cells": [ { "cell_type": "markdown", "id": "1c498e42-5c04-49ba-91a9-05fc65ba32cc", "metadata": {}, "source": [ "# LOFAR 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": "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:18:36,446\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:18:36,447\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/small_lofar.ms \n" ] } ], "source": [ "import toolviper\n", "\n", "toolviper.utils.data.download(file=\"small_lofar.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:18:36,738\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', 'OBS_MODE', 'OBSERVATION_ID'] \n", "[\u001b[38;2;128;05;128m2026-04-20 15:18:36,739\u001b[0m] \u001b[38;2;50;50;205m INFO\u001b[0m\u001b[38;2;112;128;144m toolviper: \u001b[0m Number of partitions: 1 \n", "[\u001b[38;2;128;05;128m2026-04-20 15:18:36,739\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 [0], FIELD [0], SCAN [0], EPHEMERIS [None] \n", "[\u001b[38;2;128;05;128m2026-04-20 15:18:36,805\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 = \"small_lofar.ms\"\n", "main_chunksize = {\"frequency\": 1, \"time\": 20} # baseline, polarization\n", "outfile = \"small_lofar.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", "small_lofar_0 | \n", "[] | \n", "(5, 2775, 15, 4) | \n", "--- | \n", "[XX, XY, YX, YY] | \n", "[0] | \n", "SB-349_0 | \n", "[UNSPECIFIED] | \n", "[BEAM_4_0] | \n", "[Unknown] | \n", "[] | \n", "[fk5, 14h00m00.00s, 86d00m00.00s] | \n", "--- | \n", "--- | \n", "LC9_007 | \n", "1.270615e+08 | \n", "1.272324e+08 | \n", "
<xarray.DataTree 'small_lofar_0'>\n",
"Group: /small_lofar_0\n",
"│ Dimensions: (time: 5, baseline_id: 2775, frequency: 15,\n",
"│ polarization: 4, uvw_label: 3)\n",
"│ Coordinates:\n",
"│ * time (time) float64 40B 1.52e+09 ... 1.52e+09\n",
"│ field_name (time) <U28 560B dask.array<chunksize=(5,), meta=np.ndarray>\n",
"│ scan_name (time) <U21 420B dask.array<chunksize=(5,), meta=np.ndarray>\n",
"│ * baseline_id (baseline_id) int64 22kB 0 1 2 ... 2773 2774\n",
"│ baseline_antenna1_name (baseline_id) <U9 100kB dask.array<chunksize=(2775,), meta=np.ndarray>\n",
"│ baseline_antenna2_name (baseline_id) <U9 100kB dask.array<chunksize=(2775,), meta=np.ndarray>\n",
"│ * frequency (frequency) float64 120B 1.271e+08 ... 1.272e+08\n",
"│ * polarization (polarization) <U2 32B 'XX' 'XY' 'YX' 'YY'\n",
"│ * uvw_label (uvw_label) <U1 12B 'u' 'v' 'w'\n",
"│ Data variables:\n",
"│ EFFECTIVE_INTEGRATION_TIME (time, baseline_id) float64 111kB dask.array<chunksize=(5, 2775), meta=np.ndarray>\n",
"│ FLAG (time, baseline_id, frequency, polarization) bool 832kB dask.array<chunksize=(5, 2775, 1, 4), meta=np.ndarray>\n",
"│ TIME_CENTROID (time, baseline_id) float64 111kB dask.array<chunksize=(5, 2775), meta=np.ndarray>\n",
"│ UVW (time, baseline_id, uvw_label) float64 333kB dask.array<chunksize=(5, 2775, 3), meta=np.ndarray>\n",
"│ VISIBILITY (time, baseline_id, frequency, polarization) complex64 7MB dask.array<chunksize=(5, 2775, 1, 4), meta=np.ndarray>\n",
"│ WEIGHT (time, baseline_id, frequency, polarization) float32 3MB dask.array<chunksize=(5, 2775, 1, 4), meta=np.ndarray>\n",
"│ Attributes:\n",
"│ creation_date: 2026-04-20T21:18:36.751211+00:00\n",
"│ creator: {'software_name': 'xradio', 'version': '1.1.3'}\n",
"│ data_groups: {'base': {'correlated_data': 'VISIBILITY', 'date': '20...\n",
"│ observation_info: {'observer': ['unknown'], 'observing_log': '[]', 'proj...\n",
"│ processor_info: {'sub_type': 'LOFAR-COBALT', 'type': 'CORRELATOR'}\n",
"│ schema_version: 4.0.0\n",
"│ type: visibility\n",
"├── Group: /small_lofar_0/antenna_xds\n",
"│ Dimensions: (antenna_name: 74, cartesian_pos_label: 3,\n",
"│ receptor_label: 2)\n",
"│ Coordinates:\n",
"│ * antenna_name (antenna_name) <U9 3kB 'CS001HBA0' ... 'IE613HBA'\n",
"│ mount (antenna_name) <U6 2kB dask.array<chunksize=(74,), meta=np.ndarray>\n",
"│ station_name (antenna_name) <U5 1kB dask.array<chunksize=(74,), meta=np.ndarray>\n",
"│ telescope_name (antenna_name) <U5 1kB dask.array<chunksize=(74,), 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 592B dask.array<chunksize=(74, 2), meta=np.ndarray>\n",
"│ Data variables:\n",
"│ ANTENNA_DISH_DIAMETER (antenna_name) float64 592B dask.array<chunksize=(74,), meta=np.ndarray>\n",
"│ ANTENNA_POSITION (antenna_name, cartesian_pos_label) float64 2kB dask.array<chunksize=(74, 3), meta=np.ndarray>\n",
"│ ANTENNA_RECEPTOR_ANGLE (antenna_name, receptor_label) float64 1kB dask.array<chunksize=(74, 2), meta=np.ndarray>\n",
"│ Attributes:\n",
"│ overall_telescope_name: LOFAR\n",
"│ relocatable_antennas: False\n",
"│ type: antenna\n",
"├── Group: /small_lofar_0/field_and_source_base_xds\n",
"│ Dimensions: (field_name: 1, sky_dir_label: 2)\n",
"│ Coordinates:\n",
"│ * field_name (field_name) <U28 112B 'BEAM_4_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\n",
"└── Group: /small_lofar_0/pointing_xds\n",
" Dimensions: (time_pointing: 1, antenna_name: 74,\n",
" local_sky_dir_label: 2)\n",
" Coordinates:\n",
" * time_pointing (time_pointing) float64 8B 1.52e+09\n",
" * antenna_name (antenna_name) <U9 3kB 'CS001HBA0' ... 'IE613HBA'\n",
" * local_sky_dir_label (local_sky_dir_label) <U3 24B 'az' 'alt'\n",
" Data variables:\n",
" POINTING_BEAM (time_pointing, antenna_name, local_sky_dir_label) float64 1kB dask.array<chunksize=(1, 74, 2), meta=np.ndarray>\n",
" Attributes:\n",
" type: pointing<xarray.DatasetView> Size: 23kB\n",
"Dimensions: (field_name: 1, sky_dir_label: 2,\n",
" baseline_id: 2775, frequency: 15,\n",
" polarization: 4, time: 5, uvw_label: 3)\n",
"Coordinates:\n",
" * field_name (field_name) <U28 112B 'BEAM_4_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 22kB 0 1 2 ... 2773 2774\n",
" * frequency (frequency) float64 120B 1.271e+08 ... 1.27...\n",
" * polarization (polarization) <U2 32B 'XX' 'XY' 'YX' 'YY'\n",
" * time (time) float64 40B 1.52e+09 ... 1.52e+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<xarray.DataTree 'antenna_xds'>\n",
"Group: /small_lofar_0/antenna_xds\n",
" Dimensions: (time: 5, baseline_id: 2775, frequency: 15,\n",
" polarization: 4, uvw_label: 3, antenna_name: 74,\n",
" cartesian_pos_label: 3, receptor_label: 2)\n",
" Coordinates:\n",
" * antenna_name (antenna_name) <U9 3kB 'CS001HBA0' ... 'IE613HBA'\n",
" mount (antenna_name) <U6 2kB dask.array<chunksize=(74,), meta=np.ndarray>\n",
" station_name (antenna_name) <U5 1kB dask.array<chunksize=(74,), meta=np.ndarray>\n",
" telescope_name (antenna_name) <U5 1kB dask.array<chunksize=(74,), 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 592B dask.array<chunksize=(74, 2), meta=np.ndarray>\n",
" Inherited coordinates:\n",
" * baseline_id (baseline_id) int64 22kB 0 1 2 3 ... 2772 2773 2774\n",
" * frequency (frequency) float64 120B 1.271e+08 ... 1.272e+08\n",
" * polarization (polarization) <U2 32B 'XX' 'XY' 'YX' 'YY'\n",
" * time (time) float64 40B 1.52e+09 1.52e+09 ... 1.52e+09\n",
" * uvw_label (uvw_label) <U1 12B 'u' 'v' 'w'\n",
" Data variables:\n",
" ANTENNA_DISH_DIAMETER (antenna_name) float64 592B dask.array<chunksize=(74,), meta=np.ndarray>\n",
" ANTENNA_POSITION (antenna_name, cartesian_pos_label) float64 2kB dask.array<chunksize=(74, 3), meta=np.ndarray>\n",
" ANTENNA_RECEPTOR_ANGLE (antenna_name, receptor_label) float64 1kB dask.array<chunksize=(74, 2), meta=np.ndarray>\n",
" Attributes:\n",
" overall_telescope_name: LOFAR\n",
" relocatable_antennas: False\n",
" type: antenna<xarray.DataTree 'pointing_xds'>\n",
"Group: /small_lofar_0/pointing_xds\n",
" Dimensions: (time: 5, baseline_id: 2775, frequency: 15,\n",
" polarization: 4, uvw_label: 3, time_pointing: 1,\n",
" antenna_name: 74, local_sky_dir_label: 2)\n",
" Coordinates:\n",
" * time_pointing (time_pointing) float64 8B 1.52e+09\n",
" * antenna_name (antenna_name) <U9 3kB 'CS001HBA0' ... 'IE613HBA'\n",
" * local_sky_dir_label (local_sky_dir_label) <U3 24B 'az' 'alt'\n",
" Inherited coordinates:\n",
" * baseline_id (baseline_id) int64 22kB 0 1 2 3 ... 2772 2773 2774\n",
" * frequency (frequency) float64 120B 1.271e+08 ... 1.272e+08\n",
" * polarization (polarization) <U2 32B 'XX' 'XY' 'YX' 'YY'\n",
" * time (time) float64 40B 1.52e+09 1.52e+09 ... 1.52e+09\n",
" * uvw_label (uvw_label) <U1 12B 'u' 'v' 'w'\n",
" Data variables:\n",
" POINTING_BEAM (time_pointing, antenna_name, local_sky_dir_label) float64 1kB dask.array<chunksize=(1, 74, 2), meta=np.ndarray>\n",
" Attributes:\n",
" type: pointingsmall_lofar.ms ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00\n\n", "text/plain": "small_lofar.ms \u001b[38;2;114;156;31m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[35m100%\u001b[0m \u001b[36m0:00:00\u001b[0m\n" }, "metadata": {}, "output_type": "display_data" } ] } }, "6d15c0d5925b4eabbeb49451a97d1ecb": { "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 }