{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "from DataContainer.ReadData import read_hdf5_file" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [], "source": [ "# filepath = \"//DyLabNAS/Data/Evaporative_Cooling/2023/04/18/0003/*.h5\"\n", "# filepath = \"//DyLabNAS/Data/Evaporative_Cooling/2023/04/18/0003/2023-04-18_0003_Evaporative_Cooling_000.h5\"\n", "\n", "filepath = \"//DyLabNAS/Data/Repetition_scan/2023/04/21/0000/*.h5\"\n", "\n", "groupList = [\n", " \"images/MOT_3D_Camera/in_situ_absorption\",\n", " \"images/ODT_1_Axis_Camera/in_situ_absorption\",\n", "]\n", "\n", "dskey = {\n", " \"images/MOT_3D_Camera/in_situ_absorption\": \"camera_1\",\n", " \"images/ODT_1_Axis_Camera/in_situ_absorption\": \"camera_2\",\n", "}\n", "\n", "ds = {\n", " dskey[groupList[i]]: read_hdf5_file(filepath, groupList[i])\n", " for i in range(len(groupList))\n", "}" ] }, { "cell_type": "code", "execution_count": 8, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
<xarray.Dataset>\n",
       "Dimensions:       (initial_freq: 10, runs: 3, x: 1200, y: 1920)\n",
       "Coordinates:\n",
       "  * initial_freq  (initial_freq) float64 100.8 101.0 101.2 ... 102.4 102.6 102.8\n",
       "  * runs          (runs) float64 0.0 1.0 2.0\n",
       "Dimensions without coordinates: x, y\n",
       "Data variables:\n",
       "    atoms         (initial_freq, runs, x, y) uint16 dask.array<chunksize=(1, 1, 1200, 1920), meta=np.ndarray>\n",
       "    background    (initial_freq, runs, x, y) uint16 dask.array<chunksize=(1, 1, 1200, 1920), meta=np.ndarray>\n",
       "    dark          (initial_freq, runs, x, y) uint16 dask.array<chunksize=(1, 1, 1200, 1920), meta=np.ndarray>\n",
       "    shotNum       (initial_freq, runs) <U2 '09' '19' '29' ... '05' '15' '25'\n",
       "Attributes: (12/96)\n",
       "    TOF_free:                          0.02\n",
       "    abs_img_freq:                      110.858\n",
       "    absorption_imaging_flag:           True\n",
       "    backup_data:                       True\n",
       "    blink_off_time:                    nan\n",
       "    blink_on_time:                     nan\n",
       "    ...                                ...\n",
       "    z_offset_img:                      0.189\n",
       "    mot_3d_freq:                       initial_freq\n",
       "    initial_freq:                      [101.67 102.36 102.13 100.98 102.59 10...\n",
       "    runs:                              [0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 1. 1. 1...\n",
       "    scanAxis:                          ['initial_freq', 'runs']\n",
       "    scanAxisLength:                    [30. 30.]
" ], "text/plain": [ "\n", "Dimensions: (initial_freq: 10, runs: 3, x: 1200, y: 1920)\n", "Coordinates:\n", " * initial_freq (initial_freq) float64 100.8 101.0 101.2 ... 102.4 102.6 102.8\n", " * runs (runs) float64 0.0 1.0 2.0\n", "Dimensions without coordinates: x, y\n", "Data variables:\n", " atoms (initial_freq, runs, x, y) uint16 dask.array\n", " background (initial_freq, runs, x, y) uint16 dask.array\n", " dark (initial_freq, runs, x, y) uint16 dask.array\n", " shotNum (initial_freq, runs)