Calculations/Time Series Analyzer/Intensity Noise Analysis.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "6c616b70",
   "metadata": {},
   "outputs": [],
   "source": [
    "from Analyzer import *\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "96bd598e",
   "metadata": {},
   "outputs": [],
   "source": [
    "\"\"\"Synthetic data to validate analysis\"\"\"\n",
    "# # Generate time series data with values against time in seconds\n",
    "# num_seconds = 10  # Total number of seconds\n",
    "# time        = np.arange(0, num_seconds, 1/30)  # Create timestamp index\n",
    "# voltage     = np.sin(2*np.pi*4*time) + np.sin(2*np.pi*7*time) + np.random.randn(len(time))*0.2  # Generate 4 Hz and 7 Hz sine wave\n",
    "# data = np.column_stack((time, voltage))\n",
    "\n",
    "\"\"\"Real data\"\"\"    \n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_bkg = \"Arm 1/PID_Active_SP_7V_Mod_1.0_20khz\"\n",
    "filepath_bkg = dir + filename_bkg + \".csv\"\n",
    "\n",
    "background_data = extract_data(filepath_bkg)\n",
    "\n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_data = \"Arm 1/PID_Inactive_SP_7V_Mod_1.0_20khz\"\n",
    "filepath_data = dir + filename_data + \".csv\"\n",
    "\n",
    "data = extract_data(filepath_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "354ff9ed",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time_bkg    = background_data[:, 0]\n",
    "voltage_bkg = background_data[:, 1]\n",
    "  \n",
    "processed_data_bkg, Sxx_bkg = compute_psd(background_data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "bad46fa8",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time     = data[:, 0]\n",
    "voltages = data[:, 1]\n",
    "\n",
    "processed_data, Sxx = compute_psd(data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "e11977b3",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABKUAAAMWCAYAAAAgRDUeAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAAEAAElEQVR4nOydd3hU1dbG35n0BEjoKXQRRLqCgOgFLwIiRaQE5Ar4IXoVkSZ2L4JXbBCKKKjIpYhUARVEQERQFCx0EGlCgGTSk0mZzKTt748xQ8ZkkrNPzznr9zx5InPWWmfNnvfNubPv2ftYGGMMBEEQBEEQBEEQBEEQBKEiVq0bIAiCIAiCIAiCIAiCIMwHTUoRBEEQBEEQBEEQBEEQqkOTUgRBEARBEARBEARBEITq0KQUQRAEQRAEQRAEQRAEoTo0KUUQBEEQBEEQBEEQBEGoDk1KEQRBEARBEARBEARBEKpDk1IEQRAEQRAEQRAEQRCE6tCkFEEQBEEQBEEQBEEQBKE6/lo3UB0oKSlBYmIiatasCYvFonU7BEEQBEEQBEEQBEEQuoUxhpycHERHR8Nq9X0/FE1KCSAxMRGNGzfWug2CIAiCIAiCIAiCIIhqw7Vr19CoUSOfx2lSSgA1a9YE4B7MWrVqVRhzyy23wGazISoqCn/88Yea7RGELiAPEGaG9E+YHfIAYWZI/4SZIf0TvsjOzkbjxo098yk+YUSV2O12BoBduHDBZ0xUVBQDwKKiolTsTH2SkpLYO++8w5KSkgzdh5z1pdQSk8uTIzRWSBx5wDg9kP7540j/xuqDPMAfZwYPkP7VrUX61x/kAXVrVRcPkP6N1Yec9UvnUex2e6VxtNE5B6GhoT6Ple41ZfQ9p8LCwtC9e3eEhYUZug8560upJSaXJ0dorJA48oBxeiD988eR/o3VB3mAP84MHiD9q1uL9K8/yAPq1qouHiD9G6sPLd6nhTHGVDtbNSU7Oxvh4eGw2+0+l+81atQICQkJiImJwfXr11XukCC0hzxAmBnSP2F2yAOEmSH9E2aG9E/4Qsg8CgDQnVIcFBQUaN2C5hQUFODq1auaj4XSfchZX0otMbk8OUJj9fK56wE9jAXpX54c0j8/ehkL8oA8OeQBPvQyDqR/eXJI//zoZSzIA/LkkAf40Ms4VCf9C4UmpTjIyMjweaykpMTrt1FJT0/HypUrkZ6ebug+5KwvpZaYXJ4cobFC4sgDxumB9M8fR/o3Vh/kAf44M3iA9K9uLdK//iAPqFuruniA9G+sPrR4n7R8TwClt52lp6ejTp06FcaY5bbFoqIiZGVlISIiAv7+2j28Uek+5KwvpZaYXJ4cobFC4sgDxumB9M8fR/o3Vh/kAf44M3iA9K9uLdK//iAPqFuruniA9G+sPuSsL3T5Hk1KCYD2lCKIqiEPEGaG9E+YHfIAYWZI/4SZIf0TvqA9pRQgOzvb57HSuT2jz/FlZ2dj9+7dlY6FEfqQs76UWmJyeXKExgqJIw8YpwfSP38c6d9YfZAH+OPM4AHSv7q1SP/6gzygbq3q4gHSv7H60OJ90qQUB5Vt9mUWM7pcLly6dAkul8vQfchZX0otMbk8OUJjhcSRB4zTA+mfP470b6w+yAP8cWbwAOlf3Vqkf/1BHlC3VnXxAOnfWH1o8T5p+Z4AaPkeQVQNeYAwM6R/wuyQBwgzQ/onzAzpn/AFLd8jCIIgCIIgCIIgCIIgdAtNSnGQmprq85hZHoWZkpKCRYsWISUlxdB9yFlfSi0xuTw5QmOFxJEHjNMD6Z8/jvRvrD7IA/xxZvAA6V/dWqR//UEeULdWdfEA6d9YfWjxPmlSioPg4GCfxywWi9dvoxISEoIOHTogJCTE0H3IWV9KLTG5PDlCY4XEkQeM0wPpnz+O9G+sPsgD/HFm8ADpX91apH/9QR5Qt1Z18QDp31h9aPE+aU8pAdCeUgRRNeQBwsyQ/gmzQx4gzAzpnzAzpH/CF7SnlAIUFhZq3YLmFBYWwmazaT4WSvchZ30ptcTk8uQIjdXL564H9DAWpH95ckj//OhlLMgD8uSQB/jQyziQ/uXJIf3zo5exIA/Ik0Me4EMv41Cd9C8UmpTiID093ecxs6ylTUtLw0cffYS0tDRD9yFnfSm1xOTy5AiNFRJHHjBOD6R//jjSv7H6IA/wx5nBA6R/dWuR/vUHeUDdWtXFA6R/Y/Whxfuk5XsCKL3tLC0tDXXr1q0wxiy3LRYWFiItLQ316tVDQECAYfuQs76UWmJyeXKExgqJIw8YpwfSP38c6d9YfZAH+OPM4AHSv7q1SP/6gzygbq3q4gHSv7H6kLO+0OV7NCklANpTiiCqhjxAmBnSP2F2yAOEmSH9E2aG9E/4gvaUUoCcnByfx0rn9ow+x5eTk4N9+/ZVOhZG6EPO+lJqicnlyREaKySOPGCcHkj//HGkf2P1QR7gjzODB0j/6tYi/esP8oC6taqLB0j/xupDi/dJk1IcOJ1On8fMYsb8/HycPHkS+fn5hu5DzvpSaonJ5ckRGiskjjxgnB5I//xxpH9j9UEe4I8zgwdI/+rWIv3rD/KAurWqiwdI/8bqQ4v3Scv3BEDL9wiiasgDhJkh/RNmhzxAmBnSP2FmSP+EL0yxfG/IkCFo0qQJgoODERUVhbFjxyIxMdEr5urVqxg8eDDCwsJQr149TJkyBQUFBRp1TBAEQRAEQRCElrz55pvo2rUratasiQYNGmDo0KE4d+6cVwxjDLNnz0Z0dDRCQkLQu3dvnDlzxivG5XLh6aefhs1mA+B+Uvffv5RnZmZi7NixCA8PR3h4OMaOHYusrCyvGDHfV0rPXa9ePYSFhWHIkCFVTgg88sgjGDp0aLnX9+/fD4vFUq4vgiAINajWk1L33HMPNm3ahHPnzmHLli24dOkSRowY4TleXFyMgQMHIi8vDwcPHsSGDRuwZcsWPPPMM6LOR4+CBVJTU7F06VKkpqYaug8560upJSaXJ0dorJA48oBxeiD988eR/o3VB3mAP84MHiD9q1tLSf0fOHAA48aNw9SpU7Fx40YUFRWhX79+yMvL88S88847WLBgAebOnYtnnnkGtWvXRt++fb32WZk2bRq2bduGiIgIAG79Dxo0CMXFxZ6YMWPG4Pjx49i1axd27dqF48ePY+zYsZ7jYr+vlJ57w4YNOHjwIHJzc8udW27IA+rWqi7XADP8/QdI/0pSrSelpk+fju7du6Np06a488478cILL+Dw4cMoLCwEAOzZswe///471q5di86dO+Pee+9FXFwcli9fjuzsbO7zBQYG+jxmsVi8fhuVoKAg3HTTTQgKCjJ0H3LWl1JLTC5PjtBYIXHkAeP0QPrnjyP9G6sP8gB/nBk8QPpXt5aS+t+1axfGjx+Pu+++G7fffjtWrlyJq1ev4siRIwDcd0ktWrQIL7/8MoYPH467774bH330ERwOB9atWwcAsNvtWLFiBeLi4hAcHAwAqF27Nk6dOoW9e/cCAM6ePYtdu3bh448/Ro8ePdCjRw8sX74cO3bs8NyZJeb7Stlz33vvvejcuTPWrl3rdW4p9O7dGxaLpdxPUlISeUDFWtXlGmCGv/8AXQMUhRmE9PR0Fhsby3r27Ol57T//+Q/r0KGDV1xGRgYDwPbt2ye4tt1uZwCY3W73GRMTE8MAsJiYGP7mCcIAkAcIM0P6J8wOeYCozly4cIEBYKdOnWKMMXbp0iUGgB09etQrbsiQIWzcuHGMMca+/fZbBoBlZGR46b9Dhw5s1qxZjDHGVqxYwcLDw8udLzw8nP3vf/9jjIn7vlL23GUpe+6KGD9+PHvggQfKvf7dd98xACwzM5Mx5v5eZbPZPD/Dhg1jrVu3Zg6Hw2dtwrzQ33/CF0LmURhjrFrfKQUAzz//PMLCwlC3bl1cvXoVX3zxhedYUlISGjZs6BVfu3ZtBAYGIikpyWdNl8uF7Oxsrx8AKCoqUuZNVCOKioqQlpam+Vgo3Yec9aXUEpPLkyM0Vi+fux7Qw1iQ/uXJIf3zo5exIA/Ik0Me4EMv40D6lyenNLawsBAzZszAXXfdhXbt2gGA53tCw4Y
      "text/plain": [
       "<Figure size 1200x800 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "plot_analysis(processed_data_bkg, processed_data, Sxx_bkg, Sxx, filename_bkg, filename_data, peak_find_threshold=-60)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2522948d",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "f91c772c",
   "metadata": {},
   "outputs": [],
   "source": [
    "\"\"\"Synthetic data to validate analysis\"\"\"\n",
    "# # Generate time series data with values against time in seconds\n",
    "# num_seconds = 10  # Total number of seconds\n",
    "# time        = np.arange(0, num_seconds, 1/30)  # Create timestamp index\n",
    "# voltage     = np.sin(2*np.pi*4*time) + np.sin(2*np.pi*7*time) + np.random.randn(len(time))*0.2  # Generate 4 Hz and 7 Hz sine wave\n",
    "# data = np.column_stack((time, voltage))\n",
    "\n",
    "\"\"\"Real data\"\"\"    \n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_bkg = \"Arm 1/PID_Active_SP_7V_Mod_1.0_100khz\"\n",
    "filepath_bkg = dir + filename_bkg + \".csv\"\n",
    "\n",
    "background_data = extract_data(filepath_bkg)\n",
    "\n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_data = \"Arm 1/PID_Inactive_SP_7V_Mod_1.0_100khz\"\n",
    "filepath_data = dir + filename_data + \".csv\"\n",
    "\n",
    "data = extract_data(filepath_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "f8ad9496",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time_bkg    = background_data[:, 0]\n",
    "voltage_bkg = background_data[:, 1]\n",
    "  \n",
    "processed_data_bkg, Sxx_bkg = compute_psd(background_data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "1a9afee9",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time     = data[:, 0]\n",
    "voltages = data[:, 1]\n",
    "\n",
    "processed_data, Sxx = compute_psd(data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "3bbcd482",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABKUAAAMWCAYAAAAgRDUeAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAAEAAElEQVR4nOydd5gUxdbG35nNC5vIu0tWxIBIEAVzBAMqoijqVUyoV1FRuVc/A4JZAb2KoqiIGURBUUBBRBEQBCQjkiSzOczmXN8f646s7CxVMz091d3v73nmWXf6nNOnq9/X9dbtqnYJIQQIIYQQQgghhBBCCDERd6gbIIQQQgghhBBCCCHOg5NShBBCCCGEEEIIIcR0OClFCCGEEEIIIYQQQkyHk1KEEEIIIYQQQgghxHQ4KUUIIYQQQgghhBBCTIeTUoQQQgghhBBCCCHEdDgpRQghhBBCCCGEEEJMh5NShBBCCCGEEEIIIcR0wkPdgBWoqanBwYMHERcXB5fLFep2CCGEEEIIIYQQQrRFCIHCwkKkpKTA7fb9PBQnpSQ4ePAg2rVrF+o2CCGEEEIIIYQQQizDvn370LZtW5/HOSklQVxcHIDawYyPj28w5thjj0VaWhqSk5Pxxx9/mNkeIVpADxAnQ/0Tp0MPECdD/RMnQ/0TXxQUFKBdu3be+RSfCHJEPB6PACC2b9/uMyY5OVkAEMnJySZ2Zj7p6enipZdeEunp6bbuw8j6gdTyJ1clRzZWJo4esE8P1L96HPVvrz7oAfU4J3iA+je3FvWvH/SAubWs4gHq3159GFm/bh7F4/E0GseNzhWIjY31eaxurym77znVpEkT9O3bF02aNLF1H0bWD6SWP7kqObKxMnH0gH16oP7V46h/e/VBD6jHOcED1L+5tah//aAHzK1lFQ9Q//bqIxTX6RJCCNPOZlEKCgqQkJAAj8fjc/le27ZtceDAAaSmpmL//v0md0hI6KEHiJOh/onToQeIk6H+iZOh/okvZOZRAIBPSilQUVER6hZCTkVFBfbu3RvysQh2H0bWD6SWP7kqObKxutx3HdBhLKh/Y3Kof3V0GQt6wJgcekANXcaB+jcmh/pXR5exoAeMyaEH1NBlHKykf1k4KaVAbm6uz2M1NTX1ftqVnJwcTJ06FTk5Obbuw8j6gdTyJ1clRzZWJo4esE8P1L96HPVvrz7oAfU4J3iA+je3FvWvH/SAubWs4gHq3159hOI6uXxPgrrHznJyctCsWbMGY5zy2GJVVRXy8/ORmJiI8PDQvbwx2H0YWT+QWv7kquTIxsrE0QP26YH6V4+j/u3VBz2gHucED1D/5tai/vWDHjC3llU8QP3bqw8j68su3+OklATcU4qQI0MPECdD/ROnQw8QJ0P9EydD/RNfcE+pIFBQUODzWN3cnt3n+AoKCjB//vxGx8IOfRhZP5Ba/uSq5MjGysTRA/bpgfpXj6P+7dUHPaAe5wQPUP/m1qL+9YMeMLeWVTxA/durj1BcJyelFGhssy+nmLG8vBw7d+5EeXm5rfswsn4gtfzJVcmRjZWJowfs0wP1rx5H/durD3pAPc4JHqD+za1F/esHPWBuLat4gPq3Vx+huE4u35OAy/cIOTL0AHEy1D9xOvQAcTLUP3Ey1D/xBZfvEUIIIYQQQgghhBBt4aSUAllZWT6POeVVmJmZmfjf//6HzMxMW/dhZP1AavmTq5IjGysTRw/YpwfqXz2O+rdXH/SAepwTPED9m1uL+tcPesDcWlbxAPVvrz5CcZ2clFIgOjra5zGXy1Xvp12JiYlB9+7dERMTY+s+jKwfSC1/clVyZGNl4ugB+/RA/avHUf/26oMeUI9zggeof3NrUf/6QQ+YW8sqHqD+7dVHKK6Te0pJwD2lCDky9ABxMtQ/cTr0AHEy1D9xMtQ/8QX3lAoClZWVoW4h5FRWViItLS3kYxHsPoysH0gtf3JVcmRjdbnvOqDDWFD/xuRQ/+roMhb0gDE59IAauowD9W9MDvWvji5jQQ8Yk0MPqKHLOFhJ/7JwUkqBnJwcn8ecspY2Ozsbb7/9NrKzs23dh5H1A6nlT65KjmysTBw9YJ8eqH/1OOrfXn3QA+pxTvAA9W9uLepfP+gBc2tZxQPUv736CMV1cvmeBHWPnWVnZ6N58+YNxjjlscXKykpkZ2ejRYsWiIiIsG0fRtYPpJY/uSo5srEycfSAfXqg/tXjqH979UEPqMc5wQPUv7m1qH/9oAfMrWUVD1D/9urDyPqyy/c4KSUB95Qi5MjQA8TJUP/E6dADxMlQ/8TJUP/EF9xTKggUFhb6PFY3t2f3Ob7CwkIsWrSo0bGwQx9G1g+klj+5KjmysTJx9IB9eqD+1eOof3v1QQ+oxznBA9S/ubWof/2gB8ytZRUPUP/26iMU18lJKQXKysp8HnOKGUtLS7FhwwaUlpbaug8j6wdSy59clRzZWJk4esA+PVD/6nHUv736oAfU45zgAerf3FrUv37QA+bWsooHqH979RGK6+TyPQm4fI+QI0MPECdD/ROnQw8QJ0P9EydD/RNfcPkeIYQQQgghhBBCCNEWTkopwFfBAllZWZg0aRKysrJs3YeR9QOp5U+uSo5srEwcPWCfHqh/9Tjq31590APqcU7wAPVvbi3qXz/oAXNrWcUD1L+9+gjFdXJSSoHIyEifx1wuV72fdiUqKgpHHXUUoqKibN2HkfUDqeVPrkqObKxMHD1gnx6of/U46t9efdAD6nFO8AD1b24t6l8/6AFza1nFA9S/vfoIxXVyTykJuKcUIUeGHiBOhvonToceIE6G+idOhvonvuCeUkGgqqoq1C2EnKqqKmRnZ4d8LILdh5H1A6nlT65KjmysLvddB3QYC+rfmBzqXx1dxoIeMCaHHlBDl3Gg/o3Jof7V0WUs6AFjcugBNXQZByvpXxZOSinAPaVq15i+8cYbWqylDWYfRtYPpJY/uSo5srEycfSAfXqg/tXjqH979UEPqMc5wQPUv7m1qH/9oAfMrWUVD1D/9uojFNfJ5XsS1D12lpWVhRYtWjQY45THFisqKpCeno42bdo0useW1fswsn4gtfzJVcmRjZWJowfs0wP1rx5H/durD3pAPc4JHqD+za1F/esHPWBuLat4gPq3Vx9G1pddvsdJKQm4pxQhR4YeIE6G+idOhx4gTob6J06G+ie+4J5SQaCoqMjnsbq5PbvP8RUVFWHJkiWNjoUd+jCyfiC1/MlVyZGNlYmjB+zTA/WvHkf926sPekA9zgkeoP7NrUX96wc9YG4tq3iA+rdXH6G4Tk5KKVBSUuLzmFPMWFxcjBUrVqC4uNjWfRhZP5Ba/uSq5MjGysTRA/bpgfpXj6P+7dUHPaAe5wQPUP/m1qL+9YMeMLeWVTxA/durj1BcJ5fvScDle4QcGXqAOBnqnzgdeoA4GeqfOBnqn/iCy/cIIYQQQgghhBBCiLZwUkqBnJwcn8ec8irM7OxsvPPOO8jOzrZ1H0bWD6SWP7kqObKxMnH0gH16oP7V46h/e/VBD6jHOcED1L+5tah//aAHzK1lFQ9Q//bqIxTXyUkpBcLDw30ec7lc9X7alYiICLRp0wYRERG27sPI+oHU8idXJUc2ViaOHrBPD9S/ehz1b68+6AH1OCd4gPo3txb1rx/0gLm1rOIB6t9efYTiOrmnlATcU4qQI0MPECdD/ROnQw8QJ0P9EydD/RNfcE+pIFBdXR3qFkJOdXU1CgoKQj4Wwe7DyPqB1PInVyVHNlaX+64DOowF9W9MDvWvji5jQQ8Yk0MPqKHLOFD/xuRQ/+roMhb0gDE59IAauoyDlfQvCyelFMjKyvJ5zClraTMzM/HKK68gMzPT1n0YWT+QWv7kquTIxsrE0QP26YH6V4+j/u3VBz2gHucED1D/5tai/vXD6h4oLCzEyJEj0aFDB8TExOC0007DqlWr6sVkZGTg+uuvR2pqKpo2bYqLLroI27dvrxezc+dOXHnllWjZsiXi4+NxzTXXICMjo17MmjVrcOGFF6JZs2Zo06YNhg0bhqKionoxP/zwA0477TTExcUhOTkZDz/8MKqqqhq9ziO
      "text/plain": [
       "<Figure size 1200x800 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "plot_analysis(processed_data_bkg, processed_data, Sxx_bkg, Sxx, filename_bkg, filename_data, peak_find_threshold=-60)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d58a4d17",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "960c77bc",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "27ef8df7",
   "metadata": {},
   "outputs": [],
   "source": [
    "\"\"\"Synthetic data to validate analysis\"\"\"\n",
    "# # Generate time series data with values against time in seconds\n",
    "# num_seconds = 10  # Total number of seconds\n",
    "# time        = np.arange(0, num_seconds, 1/30)  # Create timestamp index\n",
    "# voltage     = np.sin(2*np.pi*4*time) + np.sin(2*np.pi*7*time) + np.random.randn(len(time))*0.2  # Generate 4 Hz and 7 Hz sine wave\n",
    "# data = np.column_stack((time, voltage))\n",
    "\n",
    "\"\"\"Real data\"\"\"    \n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_bkg = \"Arm 1/PID_Active_SP_7V_Mod_0.5\"\n",
    "filepath_bkg = dir + filename_bkg + \".csv\"\n",
    "\n",
    "background_data = extract_data(filepath_bkg)\n",
    "\n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_data = \"Arm 1/PID_Inactive_SP_7V_Mod_0.5\"\n",
    "filepath_data = dir + filename_data + \".csv\"\n",
    "\n",
    "data = extract_data(filepath_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "716046fb",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time_bkg    = background_data[:, 0]\n",
    "voltage_bkg = background_data[:, 1]\n",
    "  \n",
    "processed_data_bkg, Sxx_bkg = compute_psd(background_data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "dec099f0",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time     = data[:, 0]\n",
    "voltages = data[:, 1]\n",
    "\n",
    "processed_data, Sxx = compute_psd(data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "c6317732",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABKUAAAMWCAYAAAAgRDUeAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAAEAAElEQVR4nOydd3gVVfrHvzc9JBB6SZAiNhQR7IqKDVwLgjTLiqiI66qLjd/quisLKlbAjgVQsdFRsQEqghVFEKQoKL2kd1Ju2vz+yOaaSG5yzsyZkjnfz/PkCbnzvu+899zvN/PkMOdMwDAMA4QQQgghhBBCCCGEOEiE2w0QQgghhBBCCCGEEP3gpBQhhBBCCCGEEEIIcRxOShFCCCGEEEIIIYQQx+GkFCGEEEIIIYQQQghxHE5KEUIIIYQQQgghhBDH4aQUIYQQQgghhBBCCHEcTkoRQgghhBBCCCGEEMfhpBQhhBBCCCGEEEIIcZwotxtoClRVVeHAgQNo3rw5AoGA2+0QQgghhBBCCCGEeBbDMFBYWIjk5GRERIS/H4qTUgIcOHAAhx12mNttEEIIIYQQQgghhDQZ9u7di86dO4c9zkkpAZo3bw6gejBbtGhRb8wxxxyD1NRUdOrUCb/++quT7RHiCegBojPUP9EdeoDoDPVPdIb6J+EoKCjAYYcdFppPCYtBGiU/P98AYPz2229hYzp16mQAMDp16uRgZ86TlpZmPPHEE0ZaWpqv+1BZ30otM7kyOaKxInH0gH96oP7l46h/f/VBD8jH6eAB6t/ZWtS/96AHnK3VVDxA/furD5X1a+ZR8vPzG4zjRucSNGvWLOyxmr2m/L7nVEJCAk4//XQkJCT4ug+V9a3UMpMrkyMaKxJHD/inB+pfPo7691cf9IB8nA4eoP6drUX9ew96wNlaTcUD1L+/+nDjfQYMwzAcO1sTpaCgAElJScjPzw+7fK9z587Yv38/UlJSsG/fPoc7JMR96AGiM9Q/0R16gOgM9U90hvon4RCZRwEA3iklQVlZmdstuE5ZWRn27Nnj+ljY3YfK+lZqmcmVyRGN9crn7gW8MBbUv5oc6l8er4wFPaAmhx6QwyvjQP2ryaH+5fHKWNADanLoATm8Mg5NSf+icFJKgpycnLDHqqqq6nz3K9nZ2XjttdeQnZ3t6z5U1rdSy0yuTI5orEgcPeCfHqh/+Tjq31990APycTp4gPp3thb17z3oAWdrNRUPUP/+6sON98nlewLU3HaWnZ2N1q1b1xujy22LFRUVyMvLQ8uWLREV5d7DG+3uQ2V9K7XM5MrkiMaKxNED/umB+pePo/791Qc9IB+ngweof2drUf/egx5wtlZT8QD1768+VNYXXb7HSSkBuKcUIY1DDxCdof6J7tADRGeof6Iz1D8JB/eUsoGCgoKwx2rm9vw+x1dQUIBly5Y1OBZ+6ENlfSu1zOTK5IjGisTRA/7pgfqXj6P+/dUHPSAfp4MHqH9na1H/3oMecLZWU/EA9e+vPtx4n5yUkqChzb50MWMwGMT27dsRDAZ93YfK+lZqmcmVyRGNFYmjB/zTA/UvH0f9+6sPekA+TgcPUP/O1qL+vQc94GytpuIB6t9ffbjxPrl8TwAu3yOkcegBojPUP9EdeoDoDPVPdIb6J+Hg8j1CCCGEEEIIIYQQ4lk4KSVBZmZm2GO6PAozIyMDTz/9NDIyMnzdh8r6VmqZyZXJEY0ViaMH/NMD9S8fR/37qw96QD5OBw9Q/87Wov69Bz3gbK2m4gHq3199uPE+OSklQVxcXNhjgUCgzne/Eh8fj969eyM+Pt7Xfaisb6WWmVyZHNFYkTh6wD89UP/ycdS/v/qgB+TjdPAA9e9sLerfe9ADztZqKh6g/v3Vhxvvk3tKCcA9pQhpHHqA6Az1T3SHHiA6Q/0TnaH+STi4p5QNlJeXu92C65SXlyM1NdX1sbC7D5X1rdQykyuTIxrrlc/dC3hhLKh/NTnUvzxeGQt6QE0OPSCHV8aB+leTQ/3L45WxoAfU5NADcnhlHJqS/kXhpJQE2dnZYY/pspY2KysLr7zyCrKysnzdh8r6VmqZyZXJEY0ViaMH/NMD9S8fR/37qw96QD5OBw9Q/87Wov69Bz3gbK2m4gHq3199uPE+uXxPgJrbzrKystCmTZt6Y3S5bbG8vBxZWVlo27YtoqOjfduHyvpWapnJlckRjRWJowf80wP1Lx9H/furD3pAPk4HD1D/ztai/r0HPeBsrabiAerfX32orC+6fI+TUgJwTylCGoceIDpD/RPdoQeIzlD/RGeofxIO7illA4WFhWGP1czt+X2Or7CwECtWrGhwLPzQh8r6VmqZyZXJEY0ViaMH/NMD9S8fR/37qw96QD5OBw9Q/87Wov69Bz3gbK2m4gHq3199uPE+OSklQWlpadhjupixpKQEP//8M0pKSnzdh8r6VmqZyZXJEY0ViaMH/NMD9S8fR/37qw96QD5OBw9Q/87Wov69Bz3gbK2m4gHq3199uPE+uXxPAC7fI6Rx6AGiM9Q/0R16gOgM9U90hvon4eDyPUIIIYQQQgghhBDiWTgpJQEfBQtkZmZi+vTpyMzM9HUfKutbqWUmVyZHNFYkjh7wTw/Uv3wc9e+vPugB+TgdPED9O1uL+vce9ICztZqKB6h/f/XhxvvkpJQEMTExYY8FAoE63/1KbGwsevTogdjYWF/3obK+lVpmcmVyRGNF4ugB//RA/cvHUf/+6oMekI/TwQPUv7O1qH/vQQ84W6upeID691cfbrxP7iklAPeUIqRx6AGiM9Q/0R16gOgM9U90hvon4eCeUjZQUVHhdguuU1FRgaysLNfHwu4+VNa3UstMrkyOaKxXPncv4IWxoP7V5FD/8nhlLOgBNTn0gBxeGQfqX00O9S+PV8aCHlCTQw/I4ZVxaEr6F4WTUhJwT6nqNaYvvPCCJ9bS2tmHyvpWapnJlckRjRWJowf80wP1Lx9H/furD3pAPk4HD1D/ztai/r0HPeBsrabiAerfX3248T65fE+AmtvOMjMz0bZt23pjdLltsaysDGlpaejYsWODe2w19T5U1rdSy0yuTI5orEgcPeCfHqh/+Tjq31990APycTp4gPp3thb17z3oAWdrNRUPUP/+6kNlfdHle5yUEoB7ShHSOPQA0Rnqn+gOPUB0hvonOkP9k3BwTykbOHjwYNhjNXN7fp/jO3jwIL766qsGx8IPfaisb6WWmVyZHNFYkTh6wD89UP/ycdS/v/qgB+TjdPAA9e9sLerfe9ADztZqKh6g/v3Vhxvvk5NSEhQXF4c9posZi4qKsHr1ahQVFfm6D5X1rdQykyuTIxorEkcP+KcH6l8+jvr3Vx/0gHycDh6g/p2tRf17j6pfn0HGT6/RAw7Vaioe0EX/vAbYB5fvCcDle4Q0Dj1AdIb6J7pDDxCd0Ub/H58AtD8HOPk5tzshHkIb/RNpuHyPEEIIIYQQQgghhHgWTkpJkJ2dHfaYLo/CzMrKwowZM5CVleXrPlTWt1LLTK5MjmisSBw94J8eqH/5OOrfX33QA/JxOniA+ne2FvXvPSoqKrF582Z6wKFaTcUDuuif1wD74KSUBFFRUWGPBQKBOt/9SnR0NDp27Ijo6Ghf96GyvpVaZnJlckRjReLoAf/0QP3Lx1H//uqDHpCP08ED1L+ztah/7xEIAM2aNaMHHKrVVDygi/55DbAP7iklAPeUIqRx6AGiM9Q/0R16gOiMNvrnnlKkHrTRP5GGe0rZQGVlpdstuE5lZSUKCgpcHwu7+1BZ30otM7kyOaKxXvncvYAXxoL6V5ND/cvjlbGgB9Tk0ANyeGUcqH81OdS/PIYBlJWVuT4W9ICaHHpADq+MQ1PSvyiclJIgMzMz7DFd1tJmZGTgqaeeQkZGhq/7UFnfSi0zuTI5orEicfSAf3qg/uXjqH9/9UEPyMfp4AHq39la1L/3qKiowPr16+kBh2o1FQ/oon9eA+yDy/cEqLntLCMjA+3atas3RpfbFoPBIPbu3YvDDjsMsbGxvu1DZX0rtczkyuSIxorE0QP+6YH6l4+j/v3VBz0gH6eDB6h
      "text/plain": [
       "<Figure size 1200x800 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "plot_analysis(processed_data_bkg, processed_data, Sxx_bkg, Sxx, filename_bkg, filename_data, peak_find_threshold=-60)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3c4dc198",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "05fa8d30",
   "metadata": {},
   "outputs": [],
   "source": [
    "\"\"\"Synthetic data to validate analysis\"\"\"\n",
    "# # Generate time series data with values against time in seconds\n",
    "# num_seconds = 10  # Total number of seconds\n",
    "# time        = np.arange(0, num_seconds, 1/30)  # Create timestamp index\n",
    "# voltage     = np.sin(2*np.pi*4*time) + np.sin(2*np.pi*7*time) + np.random.randn(len(time))*0.2  # Generate 4 Hz and 7 Hz sine wave\n",
    "# data = np.column_stack((time, voltage))\n",
    "\n",
    "\"\"\"Real data\"\"\"    \n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_bkg = \"Arm 1/PID_Active_SP_7V_Mod_0\"\n",
    "filepath_bkg = dir + filename_bkg + \".csv\"\n",
    "\n",
    "background_data = extract_data(filepath_bkg)\n",
    "\n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_data = \"Arm 1/PID_Inactive_SP_7V_Mod_0\"\n",
    "filepath_data = dir + filename_data + \".csv\"\n",
    "\n",
    "data = extract_data(filepath_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "1f4a1b8b",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time_bkg    = background_data[:, 0]\n",
    "voltage_bkg = background_data[:, 1]\n",
    "  \n",
    "processed_data_bkg, Sxx_bkg = compute_psd(background_data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "a00aa553",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time     = data[:, 0]\n",
    "voltages = data[:, 1]\n",
    "\n",
    "processed_data, Sxx = compute_psd(data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "ebb9c09a",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABKUAAAMWCAYAAAAgRDUeAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAAEAAElEQVR4nOydd5gUVdbG3+6eHGDIM0NOiiKIghEFDGvAgHnVXcPi6rrqrnHX3f0MmHbNCbOYcxZQQVQEQUEQSYKA5DA559Dd9f1RzAwD0zP3Vle4XfX+nodnZrrOOX3q9vva33f33ls+TdM0EEIIIYQQQgghhBBiI36nGyCEEEIIIYQQQggh3oOTUoQQQgghhBBCCCHEdjgpRQghhBBCCCGEEEJsh5NShBBCCCGEEEIIIcR2OClFCCGEEEIIIYQQQmyHk1KEEEIIIYQQQgghxHY4KUUIIYQQQgghhBBCbIeTUoQQQgghhBBCCCHEduKcbiAWCIfDyMnJQXp6Onw+n9PtEEIIIYQQQgghhCiLpmmorKxEdnY2/P7I66E4KSVATk4O+vbt63QbhBBCCCGEEEIIITHDjh070KdPn4jXOSklQHp6OgB9MDt16tRmzLBhw5Cbm4usrCysW7fOzvYIUQJ6gHgZ6p94HXqAeBnqn3gZ6p9EoqKiAn379m2eT4mIRjqkvLxcA6D99ttvEWOysrI0AFpWVpaNndlPXl6e9uCDD2p5eXmu7sPM+tHUMpIrkyMaKxJHD7inB+pfPo76d1cf9IB8nBc8QP3bW4v6Vw96wN5aseIB6t9dfZhZv2kepby8vN04HnQuQUpKSsRrTWdNuf3MqdTUVBx55JFITU11dR9m1o+mlpFcmRzRWJE4esA9PVD/8nHUv7v6oAfk47zgAerf3lrUv3rQA/bWihUPUP/u6sOJ+/RpmqbZ9m4xSkVFBTp37ozy8vKI2/f69OmDXbt2oXfv3ti5c6fNHRLiPPQA8TLUP/E69ADxMtQ/8TLUP4mEyDwKAHCllAQNDQ1Ot+A4DQ0N2L59u+NjYXUfZtaPppaRXJkc0VhVPncVUGEsqH9zcqh/eVQZC3rAnBx6QA5VxoH6NyeH+pdHlbGgB8zJoQfkUGUcYkn/onBSSoKSkpKI18LhcKufbqW4uBivvPIKiouLXd2HmfWjqWUkVyZHNFYkjh5wTw/Uv3wc9e+uPugB+TgveID6t7cW9a8e9IC9tWLFA9S/u/pw4j65fU+ApmVnxcXF6Nq1a5sxXlm2GAwGUVZWhoyMDMTFOffwRqv7MLN+NLWM5MrkiMaKxNED7umB+pePo/7d1Qc9IB/nBQ9Q//bWov7Vgx6wt1aseID6d1cfZtYX3b7HSSkBeKYUIR1DDxAvQ/0Tr0MPEC9D/RMvQ/2TSPBMKQuoqKiIeK1pbs/tc3wVFRX48ssv2x0LN/RhZv1oahnJlckRjRWJowfc0wP1Lx9H/burD3pAPs4LHqD+7a1F/asHPWBvrVjxAPXvrj6cuE9OSknQ3mFfXjFjfX09Nm3ahPr6elf3YWb9aGoZyZXJEY0ViaMH3NMD9S8fR/27qw96QD7OCx6g/u2tRf2rBz1gb61Y8QD1764+nLhPbt8TgNv3COkYeoB4GeqfeB16gHgZ6p94GeqfRILb9wghhBBCCCGEEEKIsnBSSoLCwsKI17zyKMyCggI8/vjjKCgocHUfZtaPppaRXJkc0ViROHrAPT1Q//Jx1L+7+qAH5OO84AHq395a1L960AP21ooVD1D/7urDifvkpJQESUlJEa/5fL5WP91KcnIyRo4cieTkZFf3YWb9aGoZyZXJEY0ViaMH3NMD9S8fR/27qw96QD7OCx6g/u2tRf2rBz1gb61Y8QD1764+nLhPniklAM+UIqRj6AHiZah/4nXoAeJlqH/iZah/EgmeKWUBjY2NTrfgOI2NjcjNzXV8LKzuw8z60dQykiuTIxqryueuAiqMBfVvTg71L48qY0EPmJNDD8ihyjhQ/+bkUP/yqDIW9IA5OfSAHKqMQyzpXxROSklQXFwc8ZpX9tIWFRXhhRdeQFFRkav7MLN+NLWM5MrkiMaKxNED7umB+pePo/7d1Qc9IB/nBQ9Q//bWov7Vgx6wt1aseID6d1cfTtwnt+8J0LTsrKioCN26dWszxivLFhsbG1FUVITu3bsjPj7etX2YWT+aWkZyZXJEY0Xi6AH39ED9y8dR/+7qgx6Qj/OCB6h/e2tR/+pBD9hbK1Y8QP27qw8z64tu3+OklAA8U4qQjqEHiJeh/onXoQeIl6H+iZeh/kkkeKaUBVRWVka81jS35/Y5vsrKSsydO7fdsXBDH2bWj6aWkVyZHNFYkTh6wD09UP/ycdS/u/qgB+TjvOAB6t/eWtS/etAD9taKFQ9Q/+7qw4n75KSUBHV1dRGvecWMtbW1WLVqFWpra13dh5n1o6llJFcmRzRWJI4ecE8P1L98HPXvrj7oAfk4L3iA+re3FvWvHvSAvbVixQPUv7v6cOI+uX1PAG7fI6Rj6AHiZah/4nXoAeJlqH/iZah/Eglu3yOEEEIIIYQQQgghysJJKQn4KFigsLAQzzzzDAoLC13dh5n1o6llJFcmRzRWJI4ecE8P1L98HPXvrj7oAfk4L3iA+re3FvWvHvSAvbVixQPUv7v6cOI+OSklQUJCQsRrPp+v1U+3kpiYiMGDByMxMdHVfZhZP5paRnJlckRjReLoAff0QP3Lx1H/7uqDHpCP84IHqH97a1H/6kEP2FsrVjxA/burDyfuk2dKCcAzpQjpGHqAeBnqn3gdeoB4GeqfeBnqn0SCZ0pZQDAYdLoFxwkGgygqKnJ8LKzuw8z60dQykiuTIxqryueuAiqMBfVvTg71L48qY0EPmJNDD8ihyjhQ/+bkUP/yqDIW9IA5OfSAHKqMQyzpXxROSknAM6X0PaZPP/20EntprezDzPrR1DKSK5MjGisSRw+4pwfqXz6O+ndXH/SAfJwXPED921uL+lcPesDeWrHiAerfXX04cZ/cvidA07KzwsJCdO/evc0YryxbbGhoQF5eHjIzM9s9YyvW+zCzfjS1jOTK5IjGisTRA+7pgfqXj6P+3dUHPSAf5wUPUP/21qL+1YMesLdWrHiA+ndXH2bWF92+x0kpAXimFCEdQw8QL0P9E69DDxAvQ/0TL0P9k0jwTCkLqKqqinitaW7P7XN8VVVVWLBgQbtj4YY+zKwfTS0juTI5orEicfSAe3qg/uXjqH939UEPyMd5wQPUv721qH/1oAfsrRUrHqD+3dWHE/fJSSkJampqWr9QVwRsfAFYcC4Q1D80t5uxuroaixcvRnV1tav7MLN+NLWM5MrkiMaKxHnlC0kFD1D/5uRQ//KooH87+qAH5OO84AHq395a1L960AP21ooVD1D/7urDifvk9j0BWi07S2wEdn4K7JoJ+JOAvucAWSehT/8h2JVXjN49krFz+YdA1kmAP27fYuFGoHgJkPsVUL4aOOIlICHD7lsixHS4dJd4GeqfeB16gHgZ6p94GeqfREJ0+14bsyYkIt9fDKSnAH3OBo56A4hPb7kWSNJ/xmcAFWuB9Y8BXQ8DBv0JgAbkfQXkzwNCtUC3w4HsU4DM44Hl/wCOeNGBmyGEEEIIIYQQQghxDm7fk6BkyH+BY94HBlzUekIKezwKUwNwwC3AcXOAPmcC6x4BNkwFUvoCR74ETPgMGHEH0P1IoOc4IC4dyJntwN0Yo6ioCC+++CKKiopc3YeZ9aOpZSRXJkc0ViTOK4+DVcED1L85OdS/PCro344+6AH5OC94gPq3txb1rx70gL21YsUD1L+7+nDiPrlSSoJAUpeI13w+X6uf8Pn0iafuR7Zf9OB7gflnAN2PAhI6m9WqZcTHxyMzMxPx8fGu7sPM+tHUMpIrkyMaKxK3jwdcigoeoP7NyaH+5VFB/3b0QQ/Ix3nBA9S/vbWof/WgB+ytFSseoP7d1YcT98kzpQQQ2QsZ1V7aggXA1reAw58zoVtCnIH7yYmXof6J16EHiJeh/omXof5JJETPlOL2PQlCoZA1hXseC/gT9cPPFScUCqGiosK
      "text/plain": [
       "<Figure size 1200x800 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "plot_analysis(processed_data_bkg, processed_data, Sxx_bkg, Sxx, filename_bkg, filename_data, peak_find_threshold=-60)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "699f9f52",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "id": "67f9cd80",
   "metadata": {},
   "outputs": [],
   "source": [
    "\"\"\"Synthetic data to validate analysis\"\"\"\n",
    "# # Generate time series data with values against time in seconds\n",
    "# num_seconds = 10  # Total number of seconds\n",
    "# time        = np.arange(0, num_seconds, 1/30)  # Create timestamp index\n",
    "# voltage     = np.sin(2*np.pi*4*time) + np.sin(2*np.pi*7*time) + np.random.randn(len(time))*0.2  # Generate 4 Hz and 7 Hz sine wave\n",
    "# data = np.column_stack((time, voltage))\n",
    "\n",
    "\"\"\"Real data\"\"\"    \n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_bkg = \"Arm 2/PID_Active_SP_9V\"\n",
    "filepath_bkg = dir + filename_bkg + \".csv\"\n",
    "\n",
    "background_data = extract_data(filepath_bkg)\n",
    "\n",
    "dir      = str(globals()['_dh'][0]).replace('\\\\','/') + \"/Time Series Data/\"\n",
    "filename_data = \"Arm 2/PID_Inactive_SP_9V\"\n",
    "filepath_data = dir + filename_data + \".csv\"\n",
    "\n",
    "data = extract_data(filepath_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "be833e35",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time_bkg    = background_data[:, 0]\n",
    "voltage_bkg = background_data[:, 1]\n",
    "  \n",
    "processed_data_bkg, Sxx_bkg = compute_psd(background_data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "id": "a37ab5b3",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Calculating power spectrum...\n"
     ]
    }
   ],
   "source": [
    "# Extract the first and second columns from data_array\n",
    "time     = data[:, 0]\n",
    "voltages = data[:, 1]\n",
    "\n",
    "processed_data, Sxx = compute_psd(data, new_sampling_rate = 1000000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "id": "e5ca8ba3",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAABKUAAAMWCAYAAAAgRDUeAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjUuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/YYfK9AAAACXBIWXMAAA9hAAAPYQGoP6dpAAEAAElEQVR4nOydd3gU1f7G3930hBJCaAGpggpYEBRRENvVnwL2roBi71673iuC/Sr23gFFsIAFBSsqoiBKBwWUFkp62YT0Teb3R8xCJBvOmT07c2bm/TxPHpOd7/e775y8L3nu3JmzPsMwDBBCCCGEEEIIIYQQYiF+uwUQQgghhBBCCCGEEO/Bi1KEEEIIIYQQQgghxHJ4UYoQQgghhBBCCCGEWA4vShFCCCGEEEIIIYQQy+FFKUIIIYQQQgghhBBiObwoRQghhBBCCCGEEEIshxelCCGEEEIIIYQQQojl8KIUIYQQQgghhBBCCLGcWLsFOIG6ujrs2LEDLVu2hM/ns1sOIYQQQgghhBBCiLYYhoHS0lJkZGTA7w9/PxQvSgmwY8cO7LPPPnbLIIQQQgghhBBCCHEMW7duRZcuXcIe50UpAVq2bAmgfjFbtWrVZM3++++PrKwsdOrUCWvXrrVSHiFawAwQL0P/E6/DDBAvQ/8TL0P/k3CUlJRgn332CV1PCYtB9kogEDAAGH/++WfYmk6dOhkAjE6dOlmozHqys7ONxx57zMjOzna1DpXzI5llplemR7RWpI4ZcI8G+l++jv53lw5mQL7OCxmg/62dRf/rBzNg7SynZID+d5cOlfMbrqMEAoFm67jRuQTJyclhjzXsNeX2PadSUlJwxBFHICUlxdU6VM6PZJaZXpke0VqROmbAPRrof/k6+t9dOpgB+TovZID+t3YW/a8fzIC1s5ySAfrfXTrsOE+fYRiGZe/mUEpKStC6dWsEAoGwj+916dIF27dvR+fOnbFt2zaLFRJiP8wA8TL0P/E6zADxMvQ/8TL0PwmHyHUUAOCdUhJUV1fbLcF2qqurkZmZaftaRFuHyvmRzDLTK9MjWqvL710HdFgL+l9ND/0vjy5rwQyo6WEG5NBlHeh/NT30vzy6rAUzoKaHGZBDl3Vwkv9F4UUpCQoLC8Meq6ura/Rft1JQUIC33noLBQUFrtahcn4ks8z0yvSI1orUMQPu0UD/y9fR/+7SwQzI13khA/S/tbPof/1gBqyd5ZQM0P/u0mHHefLxPQEabjsrKChAWlpakzVeuW0xGAyiuLgYqampiI2178Mbo61D5fxIZpnplekRrRWpYwbco4H+l6+j/92lgxmQr/NCBuh/a2fR//rBDFg7yykZoP/dpUPlfNHH93hRSgDuKUXI3mEGiJeh/4nXYQaIl6H/iZeh/0k4uKdUFCgpKQl7rOHantuv8ZWUlODLL79sdi3coEPl/EhmmemV6RGtFaljBtyjgf6Xr6P/3aWDGZCv80IG6H9rZ9H/+sEMWDvLKRmg/92lw47z5EUpCZrb7MsrYayqqsKGDRtQVVXlah0q50cyy0yvTI9orUgdM+AeDfS/fB397y4dzIB8nRcyQP9bO4v+1w9mwNpZTskA/e8uHXacJx/fE4CP7xGyd5gB4mXof+J1mAHiZeh/4mXofxIOPr5HCCGEEEIIIYQQQrSFF6UkyMvLC3vMKx+FmZubi6effhq5ubmu1qFyfiSzzPTK9IjWitQxA+7RQP/L19H/7tLBDMjXeSED9L+1s+h//WAGrJ3llAzQ/+7SYcd58qKUBImJiWGP+Xy+Rv91K0lJSTjooIOQlJTkah0q50cyy0yvTI9orUgdM+AeDfS/fB397y4dzIB8nRcyQP9bO4v+1w9mwNpZTskA/e8uHXacJ/eUEoB7ShGyd5gB4mXof+J1mAHiZeh/4mXofxIO7ikVBWpqauyWYDs1NTXIysqyfS2irUPl/EhmmemV6RGt1eX3rgM6rAX9r6aH/pdHl7VgBtT0MANy6LIO9L+aHvpfHl3WghlQ08MMyKHLOjjJ/6LwopQEBQUFYY955Vna/Px8vPrqq8jPz3e1DpXzI5llplemR7RWpI4ZcI8G+l++jv53lw5mQL7OCxmg/62dRf/rBzNg7SynZID+d5cOO86Tj+8J0HDbWX5+Ptq2bdtkjVduW6ypqUF+fj7S09MRFxfnWh0q50cyy0yvTI9orUgdM+AeDfS/fB397y4dzIB8nRcyQP9bO4v+1w9mwNpZTskA/e8uHSrniz6+x4tSAnBPKUL2DjNAvAz9T7wOM0C8DP1PvAz9T8LBPaWiQGlpadhjDdf23H6Nr7S0FPPmzWt2LdygQ+X8SGaZ6ZXpEa0VqWMG3KOB/pevo//dpYMZkK/zQgbof2tn0f/6wQxYO8spGaD/3aXDjvPkRSkJKisrwx7zShgrKiqwcuVKVFRUuFqHyvmRzDLTK9MjWitSxwy4RwP9L19H/7tLBzMgX+eFDND/1s6i//WDGbB2llMyQP+7S4cd58nH9wTg43uE7B1mgHgZ+p94HWaAeBn6n3gZ+p+Eg4/vEUIIIYQQQgghhBBt4UUpCfhRsEBeXh5efPFF5OXluVqHyvmRzDLTK9MjWitSxwy4RwP9L19H/7tLBzMgX+eFDND/1s6i//WDGbB2llMyQP+7S4cd58mLUhLEx8eHPebz+Rr9160kJCSgV69eSEhIcLUOlfMjmWWmV6ZHtFakjhlwjwb6X76O/neXDmZAvs4LGaD/rZ1F/+sHM2DtLKdkgP53lw47zpN7SgnAPaUI2TvMAPEy9D/xOswA8TL0P/Ey9D8JB/eUigLBYNBuCbYTDAaRn59v+1pEW4fK+ZHMMtMr0yNaq8vvXQd0WAv6X00P/S+PLmvBDKjpYQbk0GUd6H81PfS/PLqsBTOgpocZkEOXdXCS/0XhRSkJuKdU/TOmL7zwghbP0kZTh8r5kcwy0yvTI1orUscMuEcD/S9fR/+7SwczIF/nhQzQ/9bOov/1gxmwdpZTMkD/u0uHHefJx/cEaLjtLC8vD+np6U3WeOW2xerqamRnZ6Njx47N7rHldB0q50cyy0yvTI9orUgdM+AeDfS/fB397y4dzIB8nRcyQP9bO4v+1w9mwNpZTskA/e8uHSrniz6+x4tSAnBPKUL2DjNAvAz9T7wOM0C8DP1PvAz9T8LBPaWiwHWzr8PYj8fivA/Pw6nTT8WJb5+IYW8Nw1FvHoWy6jIAgNuv8e3cuRM//vgjdu7c6WodKudHMstMr0yPaK1IXYP3mQHna6D/5evof3fpYAbk67yQAfrf2ln0v34wA9bOckoG6H936bDjPHlRSoJ3Vr+DqSum4v0172P2+tn4euPXWJC5AD9v/RnFlcUA3B/GsrIyLFq0CGVlZa7WoXJ+JLPM9Mr0iNaK1HnlD5IOGaD/1fTQ//Lo4H8rdDAD8nVeyAD9b+0s+l8/mAFrZzklA/S/u3TYcZ58fE+AhtvOcBeAxDBFTwAoBW9bJJ6Ft+4SL0P/E6/DDBAvQ/8TL0P/k3CIPr4Xa6Emx7Ng3AKkt0lHYmwikuKSkBibiFh/LI54/QiswioAwM5qe2/nI4QQQgghhBBCCHECfHxPgozYDOyXvh+6pXZD+5T2aJXQCslxyXhl5CuhmpKqEmQGMm1UGV3y8/Px2muvIT8/39U6VM6PZJaZXpke0VqROq98HKwOGaD/1fTQ//Lo4H8rdDAD8nVeyAD9b+0s+l8/mAFrZzklA/S/u3TYcZ68KCVBbGzTN5YN2WcIkuOSAdQ/S3v9nOtd+0xtXFwcOnbsiLi4OFfrUDk/kllmemV6RGtF6nw+X6P/uhUdMkD/q+mh/+XRwf9W6GAG5Ou8kAH639pZ9L9+MAPWznJKBuh/d+mw4zy5p5QAIs9Cdu7cGTt27ABaArgVmHXuLJxxwBnWCiXERvg8OfEy9D/xOswA8TL0P/Ey9D8Jh+ieUrxTSoLa2tqwx/55ZfiGuTegtKo02pIsp7a2FiUlJc2uhRt0qJwfySwzvTI9orW6/N51QIe1oP/V9ND/8uiyFsyAmh5mQA5d1oH+V9ND/8ujy1owA2p6mAE5dFk
      "text/plain": [
       "<Figure size 1200x800 with 1 Axes>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "plot_analysis(processed_data_bkg, processed_data, Sxx_bkg, Sxx, filename_bkg, filename_data, peak_find_threshold=-60)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "04388b5c",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.9.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}