|
|
{ "cells": [ { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "# Import supporting package" ] }, { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "import xarray as xr\n", "import numpy as np\n", "\n", "from uncertainties import ufloat\n", "from uncertainties import unumpy as unp\n", "from uncertainties import umath\n", "\n", "import matplotlib.pyplot as plt\n", "\n", "from DataContainer.ReadData import read_hdf5_file\n", "from Analyser.ImagingAnalyser import ImageAnalyser\n", "from Analyser.FitAnalyser import FitAnalyser\n", "from Analyser.FitAnalyser import ThomasFermi2dModel, DensityProfileBEC2dModel, Polylog22dModel\n", "from Analyser.FitAnalyser import NewFitModel\n", "from ToolFunction.ToolFunction import *\n", "\n", "from ToolFunction.HomeMadeXarrayFunction import errorbar, dataarray_plot_errorbar\n", "xr.plot.dataarray_plot.errorbar = errorbar\n", "xr.plot.accessor.DataArrayPlotAccessor.errorbar = dataarray_plot_errorbar\n", "\n", "imageAnalyser = ImageAnalyser()" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [], "source": [ "import numpy as np\n", "\n", "from matplotlib.colors import ListedColormap, LinearSegmentedColormap\n", "\n", "import matplotlib.pyplot as plt\n", "plt.rcParams[\"font.family\"] = \"arial\"" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [], "source": [ "def Ncount_to_atoms():\n", " return 1 / 8.4743e-14 / 0.3725 * 5.86e-6**2 / 0.6606**2" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [], "source": [ "import csv\n", "\n", "colormap = np.zeros((1024, 3))\n", "\n", "with open('smooth-cool-warm-table-float-1024.csv', newline='') as csvfile:\n", " spamreader = csv.reader(csvfile, delimiter=' ', quotechar='|')\n", " i = 0\n", " for row in spamreader:\n", " try:\n", " a = row[0].split(',')\n", " colormap[i, 0] = float(a[1])\n", " colormap[i, 1] = float(a[2])\n", " colormap[i, 2] = float(a[3])\n", " i = i + 1\n", " except:\n", " pass\n", "\n", "colormap = ListedColormap(colormap)" ] }, { "cell_type": "code", "execution_count": 5, "metadata": {}, "outputs": [], "source": [ "data_colors = colormap(np.linspace(0, 1, 7))\n", "plot_blue = data_colors[-3]\n", "plot_red = data_colors[-1]\n", "plot_red_alpha = 1" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## Start a client for parallel computing" ] }, { "cell_type": "code", "execution_count": 6, "metadata": {}, "outputs": [ { "data": { "text/html": [ "<div>\n", " <div style=\"width: 24px; height: 24px; background-color: #e1e1e1; border: 3px solid #9D9D9D; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <h3 style=\"margin-bottom: 0px;\">Client</h3>\n", " <p style=\"color: #9D9D9D; margin-bottom: 0px;\">Client-65877ea1-1120-11ee-bd28-9c7bef43b4fb</p>\n", " <table style=\"width: 100%; text-align: left;\">\n", "\n", " <tr>\n", " \n", " <td style=\"text-align: left;\"><strong>Connection method:</strong> Cluster object</td>\n", " <td style=\"text-align: left;\"><strong>Cluster type:</strong> distributed.LocalCluster</td>\n", " \n", " </tr>\n", "\n", " \n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:8787/status\" target=\"_blank\">http://127.0.0.1:8787/status</a>\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " \n", "\n", " </table>\n", "\n", " \n", " <button style=\"margin-bottom: 12px;\" data-commandlinker-command=\"dask:populate-and-launch-layout\" data-commandlinker-args='{\"url\": \"http://127.0.0.1:8787/status\" }'>\n", " Launch dashboard in JupyterLab\n", " </button>\n", " \n", "\n", " \n", " <details>\n", " <summary style=\"margin-bottom: 20px;\"><h3 style=\"display: inline;\">Cluster Info</h3></summary>\n", " <div class=\"jp-RenderedHTMLCommon jp-RenderedHTML jp-mod-trusted jp-OutputArea-output\">\n", " <div style=\"width: 24px; height: 24px; background-color: #e1e1e1; border: 3px solid #9D9D9D; border-radius: 5px; position: absolute;\">\n", " </div>\n", " <div style=\"margin-left: 48px;\">\n", " <h3 style=\"margin-bottom: 0px; margin-top: 0px;\">LocalCluster</h3>\n", " <p style=\"color: #9D9D9D; margin-bottom: 0px;\">ee7c3b53</p>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard:</strong> <a href=\"http://127.0.0.1:8787/status\" target=\"_blank\">http://127.0.0.1:8787/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Workers:</strong> 6\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads:</strong> 60\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total memory:</strong> 55.88 GiB\n", " </td>\n", " </tr>\n", " \n", " <tr>\n", " <td style=\"text-align: left;\"><strong>Status:</strong> running</td>\n", " <td style=\"text-align: left;\"><strong>Using processes:</strong> True</td>\n", "</tr>\n", "\n", " \n", " </table>\n", "\n", " <details>\n", " <summary style=\"margin-bottom: 20px;\">\n", " <h3 style=\"display: inline;\">Scheduler Info</h3>\n", " </summary>\n", "\n", " <div style=\"\">\n", " <div>\n", " <div style=\"width: 24px; height: 24px; background-color: #FFF7E5; border: 3px solid #FF6132; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <h3 style=\"margin-bottom: 0px;\">Scheduler</h3>\n", " <p style=\"color: #9D9D9D; margin-bottom: 0px;\">Scheduler-f17926fd-5020-4803-9692-4e5947c8d4dd</p>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm:</strong> tcp://127.0.0.1:63894\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Workers:</strong> 6\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard:</strong> <a href=\"http://127.0.0.1:8787/status\" target=\"_blank\">http://127.0.0.1:8787/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads:</strong> 60\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Started:</strong> Just now\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total memory:</strong> 55.88 GiB\n", " </td>\n", " </tr>\n", " </table>\n", " </div>\n", " </div>\n", "\n", " <details style=\"margin-left: 48px;\">\n", " <summary style=\"margin-bottom: 20px;\">\n", " <h3 style=\"display: inline;\">Workers</h3>\n", " </summary>\n", "\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 0</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:63924\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:63932/status\" target=\"_blank\">http://127.0.0.1:63932/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:63897\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\JIANSH~1\\AppData\\Local\\Temp\\dask-worker-space\\worker-oaisnlk0\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 1</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:63928\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:63941/status\" target=\"_blank\">http://127.0.0.1:63941/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:63898\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\JIANSH~1\\AppData\\Local\\Temp\\dask-worker-space\\worker-lgn_00uq\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 2</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:63921\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:63930/status\" target=\"_blank\">http://127.0.0.1:63930/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:63899\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\JIANSH~1\\AppData\\Local\\Temp\\dask-worker-space\\worker-1ozf7xf0\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 3</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:63926\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:63937/status\" target=\"_blank\">http://127.0.0.1:63937/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:63900\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\JIANSH~1\\AppData\\Local\\Temp\\dask-worker-space\\worker-53pv7g96\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 4</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:63927\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:63938/status\" target=\"_blank\">http://127.0.0.1:63938/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:63901\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\JIANSH~1\\AppData\\Local\\Temp\\dask-worker-space\\worker-tr6dxqb7\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", " <div style=\"margin-bottom: 20px;\">\n", " <div style=\"width: 24px; height: 24px; background-color: #DBF5FF; border: 3px solid #4CC9FF; border-radius: 5px; position: absolute;\"> </div>\n", " <div style=\"margin-left: 48px;\">\n", " <details>\n", " <summary>\n", " <h4 style=\"margin-bottom: 0px; display: inline;\">Worker: 5</h4>\n", " </summary>\n", " <table style=\"width: 100%; text-align: left;\">\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Comm: </strong> tcp://127.0.0.1:63925\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Total threads: </strong> 10\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Dashboard: </strong> <a href=\"http://127.0.0.1:63934/status\" target=\"_blank\">http://127.0.0.1:63934/status</a>\n", " </td>\n", " <td style=\"text-align: left;\">\n", " <strong>Memory: </strong> 9.31 GiB\n", " </td>\n", " </tr>\n", " <tr>\n", " <td style=\"text-align: left;\">\n", " <strong>Nanny: </strong> tcp://127.0.0.1:63902\n", " </td>\n", " <td style=\"text-align: left;\"></td>\n", " </tr>\n", " <tr>\n", " <td colspan=\"2\" style=\"text-align: left;\">\n", " <strong>Local directory: </strong> C:\\Users\\JIANSH~1\\AppData\\Local\\Temp\\dask-worker-space\\worker-7qa0mcnj\n", " </td>\n", " </tr>\n", "\n", " \n", "\n", " \n", "\n", " </table>\n", " </details>\n", " </div>\n", " </div>\n", " \n", "\n", " </details>\n", "</div>\n", "\n", " </details>\n", " </div>\n", "</div>\n", " </details>\n", " \n", "\n", " </div>\n", "</div>" ], "text/plain": [ "<Client: 'tcp://127.0.0.1:63894' processes=6 threads=60, memory=55.88 GiB>" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "from dask.distributed import Client\n", "client = Client(n_workers=6, threads_per_worker=10, processes=True, memory_limit='10GB')\n", "client" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## Set global path for experiment" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [], "source": [ "groupList = [\n", " \"images/MOT_3D_Camera/in_situ_absorption\",\n", " \"images/ODT_1_Axis_Camera/in_situ_absorption\",\n", " \"images/ODT_2_Axis_Camera/in_situ_absorption\",\n", "]\n", "\n", "dskey = {\n", " \"images/MOT_3D_Camera/in_situ_absorption\": \"camera_0\",\n", " \"images/ODT_1_Axis_Camera/in_situ_absorption\": \"camera_1\",\n", " \"images/ODT_2_Axis_Camera/in_situ_absorption\": \"camera_2\",\n", "}\n" ] }, { "cell_type": "code", "execution_count": 8, "metadata": {}, "outputs": [], "source": [ "img_dir = '//DyLabNAS/Data/'\n", "SequenceName = \"Repetition_scan\" + \"/\"" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "# MOT optimize" ] }, { "attachments": {}, "cell_type": "markdown", "metadata": {}, "source": [ "## 09.06.2023" ] }, { "cell_type": "code", "execution_count": 10, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "The detected scaning axes and values are: \n", "\n", "{'cmot_initial_current': array([0.1 , 0.12, 0.14, 0.16, 0.18, 0.2 , 0.22, 0.24, 0.26, 0.28]), 'initial_freq': array([100.5, 100.9, 101.3, 101.7, 102.1, 102.5, 102.9, 103.3, 103.7,\n", " 104.1]), 'runs': array([0., 1., 2.])}\n" ] }, { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAAAaYAAAEYCAYAAAAXsVIGAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAAsTAAALEwEAmpwYAAAuCUlEQVR4nO3de7xcVX338c/3JCEBSUQaKKJyDeXRqhQTTdAYAoIBLA1qbRGrFauRSi+CVg2lJra1UAQvfR5qCG2qVYRHQArILTxIaBISMMjNC6bhYhBRCZJwDUnO+T1/rDVkZzhnzp6TPXMmc75vXvvFzNpr7/2b4XB+Z+219lqKCMzMzDpFz3AHYGZmVuTEZGZmHcWJyczMOooTk5mZdRQnJjMz6yijhzuAqu2ksTFOu7b1mhq7U1uvt+Ul7f/PtuUl7b3ezjs/394LAhPHPN32a45WX1uvt/aXe7T1egBj1j3b1usNx0jjp3hiXURs95c764iXxOO/6S1d/457nr8hIo7Z3ut2mq5LTOO0K9NGz2rrNUftt29br/fYm9v/y2XdlPb+Aj3kdQ+19XoAH9p7WduvuXtPe5PhX//zqW29HsAe/7Gqrdfr27ylrdcD+H993/5ZFedZ95tebrvhlaXrj3n5/ROruG6n8a08M7OOEfRGX+mtEUkflLQkbyslbZQ0TdJtkpZLmpfr9UhaIGlFrjspl5euW7WuazGZme2oAuijmluREfE14GsAks4HFgELgHcDDwDXSDoU2B8YFxGHSZoGnAfMbrJupdxiMjPrIH1N/FOGpCnA7wKXAGMj4v5IHXE3AEcB04HrASJiJTBF0oSydSv98JlbTGZmHSIIepsbvDFRUrETb2FELKyrcwbwOWAC8GSh/CnggFy+oVDe20xdSaMjotKOPScmM7MO0uStvHURMWCrRdJuwMERcXNuBY0v7B4PrAd2qSvvISWlUnWrTkq1AMzMrAME0EuU3kqYAdwEEBFPApskHShJwCxgKbAcOA7SgAfg3mbqVvn5a9xiMjPrIFUNfsgOJg1eqDkFuAgYBSyOiNskfR84WtKtgICTh1C3Uk5MZmYdIqDZPqbG54v4Qt37lcC0urI+UhKqP7Z03ao5MZmZdZD2PsremZyYzMw6RJTvO+pqTkxmZh0iAjY7LzkxmZl1DtGLhjuIYefEZGbWIQLoc4vJicnMrJO4xdSiB2wHm4FW0ml51trbCrPWfqYwE+5dkn5ZqPujwr6DWxGzmdlwSw/YqvTWrVrVYjqBAWaglXQA8D5gKmlk5DJJV0TE2cDZuc53gU/lc00GPhARd7QoVjOzjtEX3ZtwympVYtpmBto8u23Nw8AxEdELIGkMsLG2U9K7gCciYnEumgzMlbQXcE1EnFV/MUlzgDkA49ilBR/HzKz1ai2mka5ViWnAGWgjYjOwLs+/9AXgzohYXag7F3hv4f0lwPmkSQWvkPT7EfHd4sXybLoLASb0/Ja7Ds1shxSIXk9h2rJvoH5m2m1moJU0jjQH03jgY4Xy1wDrI2JNfi/gyxGxLiI2AdcAh7YoZjOzYdcXKr11q1YlpgFnoM3J5krg7oj4aO2WXnYUcF3h/QTgh5J2zccdCbivycy6kgc/JK26lXcFdTPQSjodWEOaqfZwYKykY3P9uRGxgjQT7o21k0TEBklnADcDzwM3RcS1LYrZzGyYid7wrbyWJKYBZqC9r/B63ADHndpP2TeAb1QXnZlZZwqgz31MXfiAbQREm+fnfX5zWy/X0zt4nR3dHuOebvs19xq1YfBKFbv12YPaer2X3bdx8EoVC09l0JRuvkVXVvclJjOzHVSE2ByjhjuMYefEZGbWIdLgB9/Kc2IyM+sYHvwATkxmZh3Dgx8SJyYzsw7S28UPzpblxGRm1iE8JVHixGRm1kH63MfkxGRm1ik8Ki9xYjIz6xCB3MeEE5OZWUfxqDwnJjOzjhGBn2OidctemJlZ00RfE9ugZ5PmSloh6Q5JfyZpkqRlkpZK+qqknlxvnqTbJd0q6U25rHTdqjkxmZl1iAA2xejSWyOSZgJvBt5CWmroVcAXgTMj4q2kJYlmS3pD3j8VOJG0YjhN1q2Ub+WZmXWIoNKVaWeRFmm9grTo6t8AHwFuyfuvA94O/BRYHBEBrJU0WtIewOSydSPisaqCBicmM7OO0uRw8YmSVhXeL4yIhbV9wL7A7wP7A1cBPTmpADwFvJSUtB4vnKNWribqOjGZmXWjoOkHbNdFxJQB9j0O3BcRm4CfStpIup1XMx5YDzyZX9eX9zVRt1LuYzIz6xiit4ltEMuAY5TsDbwEuCn3PQEcCywFlgOzJPVI2ofUqloH3NlE3Uq5xWRm1iGG0GIa+FwR35U0A7id1Ag5FXgQuFDSTsBPgMsiolfSUmBFoR7AJ5qoWyknJjOzDlLl0uoR8al+ig/vp958YH5d2eqydavmxGRm1iEi5ElcaVEfU77/uCA/2LVE0qS6/adJui1v83KZJD2S6y+RdFYuP17S9/O5PtKKeM3MOkVv9JTeulWrWkwnAOMi4jBJ04DzgNkAkg4A3kd6QKsPWCbpCuBZ4AcRcXztJJLGAF8C3gg8AyyXdFVE/KpFcZuZDZu0gq0ncW1VYpoOXA8QESslFYczPgwcExG98ELy2Uh6mOsVkm4GngNOA8YCayLiiVx3GTADuLRFcZuZDSN1dUuorFYlpgnAhsL7XkmjI2JLRGwG1kkS8AXgzohYLWkv4KyIuFTSdOCbpORUPE/tYa5tSJoDzAEYxy5EX9RXaa3Nm9t6uZ4tbb1cMrq93+luY55r6/UA9h39fNuvec4TkwavVKGd7vt5W68H0Nv2K+640qg8t5halZjqH8LqiYgXfp1KGgcsIiWaj+XiVcAWgIhYlsfdP0WJh7nyk84LASZo9zZnJTOz6nihwNY9YLscOA4g9zHdW9uRW0pXAndHxEdrt/SAecDHc51DSLf8fgwcJGn3PJZ+Bmn8vJlZ1wnElhhVeutWrWoxXQEcLelW0qy0J0s6HVgDjCKNjR8r6dhcfy5wNvBNSe8gtZw+GBGb83E3kJLoooh4pEUxm5kNq7Qek2/ltSQxRUQfcEpd8X2F1+MGOPQd/ZzrauDqikIzM+to7mMqkZgkvRaYCfwW8GvgpvxEsJmZVSgte+E+pgETk6RXA+eSni+6F/gF8DLgLEmjgTMi4kdtidLMbISockqiHVWjFtMfAydFxIb6HZJeRhrK/dlWBWZmNtJ4uHgyYGLKE/VtI7eU/jAiLsFJycysYr6VByWHi0t6uaTPAQ8Bf9LSiMzMRrA+VHrrVg0HP0g6HPgL4FDSvHaHRcTD7QjMzGyk8XDxpNHghztIi0NdAHwPuMZJycystXwrr3GL6TbgLaQldR8h9cuZmVmLpOHibjENmJoj4mPANOAe0jx0h0n6c0m7tys4M7ORxn1Mgwx+iIjnIuLrEfFW4DDgQFKiMjOzitWGi5fdutWAiUnSG4rvI+K+iPgksH9/+83MbDuF2NI3qvTWrRr1Mb1d0ieAy0itpF8BuwFTJf0xcDfwg5ZHaGY2QngF26TRA7ZnS3ol8OfAJ4GJpOS0BPhkRKxtS4RmZiNIN9+iK6vhc0wR8XPgb9sUi5nZiOYpiZJWrcdkZmZD4MTkxGRm1jH8HFMy6CPGkj5c9/6vWheOmdnI5ueYGk9J9F7gD4AjJB2Zi0cBrwX+pQ2xmZmNLFHtrTxJPwCezG8fJE0x9xVgC7A4Ij4nqQf4V+AQ4HngwxGxRtK0snUrCzhrdCvveuBR0sq1F+SyPuD+qoMwM7NqBz9IGgcoImYWyu4C3g08AFwj6VDSs6njIuKwnIzOA2YDC5qoW6lGw8WfIA0NXyJpT2DcYMd0jOhr7/W2bGnr5Xo2tX/awnbf9t5rp/XtvSCw56hd237NtRt2a+v1dn/qkbZeD2j//487uCYT00RJqwrvF0bEwvz6EGAXSYtJv7fnA2Mj4n4ASTcARwEvJzVEiIiVkqZImlC27tA+ZWODJhlJ5wPvIC2tLlJSf3MrgjEzG8mGMPhhXUQMlByeBc4F/g04CLgOWF/Y/xRwADA "text/plain": [ "<Figure size 432x288 with 2 Axes>" ] }, "metadata": { "needs_background": "light" }, "output_type": "display_data" } ], "source": [ "folderPath = img_dir + SequenceName + '2023/06/09'# get_date()\n", "\n", "shotNum = \"0010\"\n", "filePath = folderPath + \"/\" + shotNum + \"/*.h5\"\n", "\n", "dataSetDict = {\n", " dskey[groupList[i]]: read_hdf5_file(filePath, groupList[i])\n", " for i in range(len(groupList))\n", "}\n", "\n", "dataSet = dataSetDict[\"camera_0\"]\n", "\n", "print_scanAxis(dataSet)\n", "\n", "scanAxis = get_scanAxis(dataSet)\n", "\n", "dataSet = auto_rechunk(dataSet)\n", "\n", "dataSet = imageAnalyser.get_absorption_images(dataSet)\n", "\n", "imageAnalyser.center = (690, 1435)\n", "imageAnalyser.span = (1000, 700)\n", "imageAnalyser.fraction = (0.1, 0.1)\n", "\n", "dataSet_cropOD = imageAnalyser.crop_image(dataSet.OD)\n", "dataSet_cropOD = imageAnalyser.substract_offset(dataSet_cropOD).load()\n", "\n", "Ncount = imageAnalyser.get_Ncount(dataSet_cropOD)\n", "Ncount_mean = calculate_mean(Ncount)\n", "Ncount_std = calculate_std(Ncount)\n", "\n", "fig = plt.figure()\n", "# ax = fig.gca()\n", "Ncount_mean.plot.pcolormesh()\n", "plt.xlabel('MOT AOM Frequency (MHz)')\n", "plt.ylabel('MOT Gradient Coil Current (A)')\n", "plt.tight_layout()\n", "# plt.grid(visible=1)\n", "plt.show()" ] }, { "cell_type": "code", "execution_count": 31, "metadata": {}, "outputs": [], "source": [ "freq = -(Ncount_mean.initial_freq.to_numpy() - 99.969) *2 / 0.136\n", "current = Ncount_mean.cmot_initial_current.to_numpy() * 5.17\n", "data = Ncount_mean.to_numpy() * Ncount_to_atoms()\n", "\n", "X, Y = np.meshgrid(freq, current)" ] }, { "cell_type": "code", "execution_count": 39, "metadata": {}, "outputs": [ { "data": { "image/png": "iVBORw0KGgoAAAANSUhEUgAAAw8AAAKNCAYAAABm01exAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAABJ0AAASdAHeZh94AACXkElEQVR4nOzdd7wcZdn/8c83IYTQey+hS5NOKFIF6V3Ex0ceQQEVFUVEbCAIKor8LCgCFhALqKB0UHpX6WIA6b2XhJaElOv3xz2bnZycs2fLzJZzvm9e85rZ2Xvuuc7JJsy1d1NEYGZmZmZmNpgRnQ7AzMzMzMx6g5MHMzMzMzOri5MHMzMzMzOri5MHMzMzMzOri5MHMzMzMzOri5MHMzMzMzOri5MHMzMzMzOri5MHMzMzM7MBSBon6S+SXpI0VdKL2estOh1bJ8iLxJmZmZmZzU7SR4BzgJHAfcDDwGrA2kAAh0bELzsXYfs5eTAzMzMz60PSgsBTwHzAwRHxq9x7nwROByYDYyPixY4E2QHutmRmZmZmNrstSYnDf/KJA0BEnAE8AMwFbN6B2DrGyYOZmZmZ2exmZPvFJY3OvyFpFLBg9vKVdgbVaU4ezMzMzMxmdzMwAVgcOF/S2pLmkrQG8CdgKeDGrNyw4TEPZmZmZmb9kLQVcC6wdJ+3AjgJOCEiJrU9sA5yy4OZmZmZWf/uJ822NAP4N3AhaayDgIOAnToWWYe45cHMzMzMrA9JywE3kLonfSgiLsm992FSUiHg/RFxY2eibD+3PJiZmZmZze7bwIrAifnEASAiziN1W5oD+EYHYusYJw9mZmZmZrPbIdtfPsD7lYRiozbE0jWcPJiZmZmZzW7BbD91gPenZftR5YfSPZw8mJmZmZnN7v5sv8cA71cGS9/Vhli6hpMHMzMzM7PZ/STbf13Sjvk3JO0KHJO9PKWtUXWYZ1syMzMzM+uHpB8Dh2cv7wEeBVYF3pudOy4iju9AaB3j5MHMzMzMbACSdgEOA8aRxkG8DtwG/CgirutgaB3h5MHMzMzMupYkATcDmwMHRcTZDVy7NPA5YGdgZWAk8GxW3y8i4tbCAx7i5uh0AGZmZmZmNXyVlDg0RNJewG+A+fu8tUq2HSjpNODwiJjeapDDhZMHMzMzM+tKkg4CTmziuu2A80ktDQD3AZcBk4FNgR1Jq0Mflr3/mZaDHSacPJiZmZnZoCTNExFvN3ntvBHxVgPlRVq5+XjSQ34j9xoDnE01cTgRODZyffUlbQ9cCMwDHCbp/OE4fqEZnqrVzMzMzGqStAXwuKStm7h2B+AxSePqLL8kqZXgWzSYOGQOBpbLjq+KiGOizyDfiLgaOCR3quHWjeHKyYOZmZmZDUjSGsDlwGLAZZLe18C12wEXZddeKWm1GmXHSPoG8DBpgDPAm8ANDYZ8YO742zXKnQc8kB1vLmmlBu8zLDl5MDMzM7NaHgFuzI7nAa6QNOgAZknbAJcAY7JT1wGP1bhkf+AEYN7s9V3AZsD19QYqaTFg/ezlBNKsSv3KWiMuz53au977DGdOHszMzMxsQBExFfggcGV2al5SAjFgNyRJWwKXAnNnpy4F9o+IaXXc8jXgi8C4iBjfYLgbUO3qdHsdsyj9I3dcV7eq4c7Jg5mZmZnVFBFTSN/MX52dmh/4m6SN+5bNxkdcTmqlALgC+GCWhNTyImla1pUj4od1Jhp95btF1WrlqHhygGttAE4ezMzMzGxQETEZ2JPqGIQFgL9L2qBSRtJmpGSh0vXoKmCfLPkYrP4rIuKkiJjQQphL5Y6frqP8c7njJVq477Dh5MHMzMzM6hIR7wC7ArdkpxYErpa0nqRNSF2b5sveuw7YM0s62mWB3PE7dZTPl1lgwFI2k5MHMzMzM6tbttbDzlTHCyxE6s70N6qrOd8E7B4Rk9oc3ujccT33zpcZPWApm8nJg5mZmZk1JCLeBHYC7shOLUJqhQC4Fdil2QXlWjSjhWtj8CLm5MHMzMzMGhYRE0mzIvX1hUZWky5Y/r5z1VF+TO64nd2repaTBzMzMzNrmKQ1gfP7eetCSau2O55MPnmYe8BS/ZeZWHAsQ5KTBzMzMzNriKTVgWuAxbNT/wLuyY6XBq6TtHIHQsvPnrR0HeWXyR0/X3AsQ5KTBzMzMzOrW9aqcC2wZHbqTmBHYHvg39m5ZUgJxIptDi+/qNzYOsqvkDt+qNhQhiYnD2ZmZmZWl6w14Vqq3+rfDewQERMi4lVSAvGf7L3lSAnECrPXVJr7qA583liSahUGNssd31lOSEOLkwczMzMzG1TWinAtsGx26l5g+4h4vVImIl4G3g/cn51agZRALNeOGLMF5m7OXi4OzLYCdkWWWOySO3VleZENHU4ezMzMzKwmScuTEofls1P3kRKH1/qWjYiXSAnEf7NTK5ISiGX6li3JH3LHx9Uo9xFg9ez4jogYX6OsZZw8mJmZmdmAJC1LWi16bHZqPPD+iHhloGsi4gVgW6rjCFYmJRBLlRhqxdnAo9nxzpJ+LGmOfAFJ2wNn5E59sw1xDQlzDF7EzMzMzIaxSUBlwbcHSInDy4NdFBHPS9oOuB5YhTSNaulrKUTEZEmHkrohjQIOJyURF2YxjCOtkF0ZD/GLiLi87LiGCrc8mJmZmdmAsoHQOwCXANtFxIsNXPsssF127Qfy4yPKFBHXAvtRXbthVeAo4HjSOIdK4nAm8Ol2xDRUuOXBzMzMzGrKEoY9mrz26WavbUVEXJStR/FZYDfS2IsxwIvArcDpEXF9u+PqdYqIwUuZmZmZmdmw525LZmZmZmZWFycPZmZmZmZWFycPZmZmZmZWFw+Y7iGSPEDFzMxsGIkIDV7KrH2cPJiZmZkZAJKWBhYq+TavR8RzJd/DSuLkwczMzMyQtPQYRjw7iRll3+pNSe9xAtGbnDyYmZmZGcBCk5jBN+ZcmqU1Zyk3eC7e5cR3n5uP1Lrh5KEHOXkwMzMzs5mWmWM0K4wcXUrdmi54t5SqrU2cPPS4rfe9tdMhNG3kqFGdDqEwnzlig06HUIjVv719p0MozOj5x3Q6hMIss9MWnQ6hEK9vtFunQyjM9S+v0+kQCnHVNS93OoTCPHn/E50OoWXX/Wlcp0MAQCPTVkrd5VRrbeTkwczMzMyqRghGlvSY78mjep6TBzMzMzOrGgEqayUwrzDW85w8mJmZmdlMGgEaUU4LQWlJibWNkwczMzMzm0kq7yFf7rXU85w8mJmZmVnVCKWtrLqtpzl5MDMzM7OZVOKYB3db6n1OHszMzMxsJknljXlwv6We5+TBzMzMzGZyy4PV4uTBzMzMzKo85sFqcPJgZmZmZjkqsXuRk4de5+TBzMzMzGZytyWrxcmDmZmZmc0kldfy4AHTvc/Jg5mZmZlVjci2suq2nubkwczMzMxmEuW1ELjdofc5eTAzMzOzmTzmwWpx8mBmZmZmVVLayqrbepqTBzMzMzObyS0PVouTBzMzMzPL8ToPNjAnD2ZmZmY2k6dqtVqcPJiZmZlZladqtRqcPJiZmZnZTG55sFqcPJiZmZnZTJ5syWpx8mBmZmZmVRKMcPZg/XPyYGZmZmYzpZaHsrotlVKttZGTBzMzMzObyd2WrBYnD2ZmZmZWNaLEbktl1Wtt4+TBzMzMzGbhFgIbiJMHMzMzM5vJU7VaLU4ezMzMzKzKgx6sBicPZmZmZjaTcwerpfTkQdKCwJbAOGA1YHlgAWA0MAl4A3gc+C9wG3BzRLxTdlxmZmZmNjuNECppYHNZ9Vr7lJI8SBoN7A98jJQ4jBzkkk1yx9Mk/R34LfDXiJhaRoxmZmZmNgA/49sACk0eJM0LHAF8HliocrrBakYBu2TbC5JOBX4aEW8VFqiZmZmZ9csDpq2WQpIHpU/CYcDxpKQh/8l4ALgHuC87fg2YCLwFzAXMCywHrAhsQGqFWCm7ding28ARkr5NSiJmFBHzAD/DzcDmwEERcXYZ98ndb2/gL9nL0u9nZmZmVg+vMG21tJw8SFoHOAtYn5Q0BHA18Afg7xHxXB3V/KtPnSsBuwL/S0omFgN+CBwk6ZCIuKPVuPvxVVL "text/plain": [ "<Figure size 816x1615.68 with 2 Axes>" ] }, "metadata": { "needs_background": "light" }, "output_type": "display_data" } ], "source": [ "from mpl_toolkits.axes_grid1 import make_axes_locatable\n", "\n", "fig = plt.figure(figsize=(6.8, 6.8/4*3*2*(1 + 16/50)), dpi=120)\n", "grid = fig.add_gridspec(4, 1, height_ratios=[2, 50, 10, 50], wspace=0.4, hspace=0.1)\n", "\n", "filter = ''\n", "edgecolor = None\n", "\n", "colorbar_ticks = np.linspace(0, 2.5e8, 6)\n", "\n", "plot_axes = plt.subplot(grid[1, 0])\n", "\n", "im = plot_axes.pcolormesh(X, Y, data, cmap=colormap, edgecolor=edgecolor, vmin=0)\n", "\n", "colorbar = fig.colorbar(im, ax=plot_axes, ticks=np.array([0, 0.5, 1, 1.5, 2])*1e8)\n", "colorbar.formatter.set_powerlimits((0, 0))\n", "colorbar.formatter.set_useMathText(True)\n", "colorbar.ax.yaxis.set_offset_position('left')\n", "colorbar.ax.tick_params(axis='both', labelsize=20)\n", "colorbar.ax.yaxis.offsetText.set_fontsize(20)\n", "\n", "plot_axes.set_xlabel(\"3D-MOT detuning $\\delta_\\mathrm{3D}$ $(\\Gamma_\\mathrm{626})$\", fontsize=20)\n", "plot_axes.set_ylabel(\"3D-MOT gradient $b'_\\mathrm{3D}$ (G/cm)\", fontsize=20)\n", "colorbar.set_label(\"Loaded atom number $N_\\mathrm{4s}$\", fontsize=20)\n", "\n", "plot_axes.tick_params(axis='both', which='major', labelsize=20)\n", "plot_axes.tick_params(axis='both', which='minor', labelsize=16)\n", "plot_axes.xaxis.offsetText.set_fontsize(20)\n", "plot_axes.yaxis.offsetText.set_fontsize(20)\n", "\n", "plt.setp(plot_axes.spines.values(), linewidth=3)\n", "plot_axes.xaxis.set_tick_params(width=3)\n", "plot_axes.yaxis.set_tick_params(width=3)\n", "plot_axes.tick_params(direction='in', length=10)\n", "\n", "# # plotting.figure[dataExtractor_key].colorbar_min = 0\n", "# # plotting.figure[dataExtractor_key].colorbar_max = 2.5e8\n", "\n", "# plotting.figure[dataExtractor_key].add_pcolormesh_plot(cmap=colormap, edgecolor=edgecolor, grid=plot_axes)\n", "# plotting.figure[dataExtractor_key].add_axes_label()\n", "# # plotting.figure[dataExtractor_key].add_color_bar(ticks=colorbar_ticks)\n", "# plotting.figure[dataExtractor_key].add_color_bar(ticks=np.linspace(0, 8, 5))\n", "# plotting.figure[dataExtractor_key].colorbar.ax.set_yticklabels(['{:.1f}'.format(x) for x in np.linspace(0, 8, 5)])\n", "\n", "# plotting.figure[dataExtractor_key].colorbar.ax.text(3.3, 7.69, '$\\\\times 10^{7}$', va='bottom', weight='bold', fontsize=20)\n", "\n", "# # plotting.figure[dataExtractor_key].set_color_bar_offset_position(0, 1.07)\n", "\n", "# # plotting.figure[dataExtractor_key].plot_axes.set_xlim([-30, -55])\n", "# # plotting.figure[dataExtractor_key].plot_axes.set_ylim([0.15, 0.65])\n", "\n", "# # plt.xticks([-40])\n", "\n", "# plotting.figure[dataExtractor_key].plot_axes.set_xlabel(\"3D-MOT detuning $\\delta_\\mathrm{3D} / \\Gamma_{626}$\", fontsize=20)\n", "# plotting.figure[dataExtractor_key].plot_axes.set_ylabel(\"3D-MOT gradient $b'_\\mathrm{3D}$ (G/cm)\", fontsize=20)\n", "# plotting.figure[dataExtractor_key].colorbar.set_label(\"Loaded atom number $N_\\mathrm{4s}$\", fontsize=20)\n", "\n", "# plot_axes = plt.subplot(grid[0, 0])\n", "# plot_axes.text(-0.17, 1.0, '(a)', va='bottom', weight='bold', fontsize=20)\n", "# plot_axes.set_axis_off()\n", "\n", "fig.savefig('figS3_v1.pdf', bbox_inches = \"tight\")\n", "\n", "plt.show()" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.8.10" }, "orig_nbformat": 4 }, "nbformat": 4, "nbformat_minor": 2}
|
