780 lines
98 KiB
Plaintext
780 lines
98 KiB
Plaintext
|
{
|
||
|
|
"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
|
||
|
|
}
|