Count_licks_in_Drinking.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Total drinking duration: 415.0\n",
      "File: ILPG_silencing_#2_LickToGetReward_Training_20230413_162800_2023_04_24_Cue0_adjusted_data.mat\n",
      "Average BNC1High count during drinking periods: 6.676470588235294\n",
      "Average frequency: 1.6409638554216868\n",
      "Total drinking duration: 280.0\n",
      "File: ILPG_silencing_#2_LickToGetReward_Training_20230414_153200_2023_04_24_Cue0_adjusted_data.mat\n",
      "Average BNC1High count during drinking periods: 3.8256880733944953\n",
      "Average frequency: 1.4892857142857143\n",
      "Total drinking duration: 375.0\n",
      "File: ILPG_silencing_#2_LickToGetReward_Training_20230411_165753_2023_04_24_Cue0_adjusted_data.mat\n",
      "Average BNC1High count during drinking periods: 8.725274725274724\n",
      "Average frequency: 2.1173333333333333\n",
      "Total drinking duration: 320.0\n",
      "File: ILPG_silencing_#2_LickToGetReward_Training_20230421_132825_2023_04_24_Cue0_adjusted_data.mat\n",
      "Average BNC1High count during drinking periods: 6.433962264150943\n",
      "Average frequency: 2.13125\n",
      "Total drinking duration: 260.0\n",
      "File: ILPG_silencing_#2_LickToGetReward_Training_20230420_140353_2023_04_24_Cue0_adjusted_data.mat\n",
      "Average BNC1High count during drinking periods: 4.84070796460177\n",
      "Average frequency: 2.103846153846154\n",
      "Total drinking duration: 360.0\n",
      "File: ILPG_silencing_#2_LickToGetReward_Training_20230418_151813_2023_04_24_Cue0_adjusted_data.mat\n",
      "Average BNC1High count during drinking periods: 6.592233009708738\n",
      "Average frequency: 1.886111111111111\n",
      "Total drinking duration: 505.0\n",
      "File: ILPG_silencing_#2_LickToGetReward_Training_20230412_172237_2023_04_24_Cue0_adjusted_data.mat\n",
      "Average BNC1High count during drinking periods: 8.481818181818182\n",
      "Average frequency: 1.8475247524752476\n",
      "Total drinking duration: 260.0\n",
      "File: ILPG_silencing_#2_LickToGetReward_Training_20230419_160151_2023_04_24_Cue0_adjusted_data.mat\n",
      "Average BNC1High count during drinking periods: 6.106796116504855\n",
      "Average frequency: 2.419230769230769\n"
     ]
    }
   ],
   "source": [
    "# This script calculates average lick count and average frequency across all trials in a session.\n",
    "\n",
    "import os\n",
    "import numpy as np\n",
    "from scipy.io import loadmat\n",
    "\n",
    "# Set the working directory to the folder containing the .mat files\n",
    "os.chdir('/Users/jun/Documents/Work/Project/Intralingual parasympathetic ganglion neurons/Behavior/ILPG_silencing_#2/SessionData_LickToGetReward/')\n",
    "\n",
    "# Get a list of all .mat files in the current directory that end with \"adjusted_data.mat\"\n",
    "mat_files = [f for f in os.listdir() if f.endswith('adjusted_data.mat')]\n",
    "\n",
    "for mat_file in mat_files:\n",
    "    # Load the original data\n",
    "    mat_contents = loadmat(mat_file)\n",
    "    SessionData = mat_contents['adjusted_SessionData']\n",
    "\n",
    "    n_trials = SessionData['nTrials'][0, 0][0, 0]\n",
    "    raw_events = SessionData['RawEvents'][0, 0]['Trial'][0, 0]\n",
    "\n",
    "    lickometer = {\n",
    "        'BNCHigh': [np.nan] * n_trials,\n",
    "        'BNCLow': [np.nan] * n_trials\n",
    "    }\n",
    "\n",
    "    for i in range(n_trials):\n",
    "        trial = raw_events[0, i]['Events'][0, 0]\n",
    "\n",
    "        if 'BNC1High' not in trial.dtype.names or 'BNC1Low' not in trial.dtype.names:\n",
    "            lickometer['BNCHigh'][i] = np.nan\n",
    "            lickometer['BNCLow'][i] = np.nan\n",
    "        else:\n",
    "            lickometer['BNCHigh'][i] = trial['BNC1High'][0, 0]\n",
    "            lickometer['BNCLow'][i] = trial['BNC1Low'][0, 0]\n",
    "\n",
    "    total_drinking_duration = 0.0\n",
    "    total_BNC1High_count = 0.0\n",
    "\n",
    "    trials = SessionData[\"RawEvents\"][0, 0][\"Trial\"]\n",
    "    num_trials = SessionData['nTrials'][0, 0][0, 0]\n",
    "\n",
    "    for trial in range(num_trials):\n",
    "        state_info = None\n",
    "        if trial < len(trials.item()[0]):\n",
    "            state_info = trials.item()[0, trial][\"States\"][0, 0]\n",
    "\n",
    "        if state_info is not None and \"Drinking\" in state_info.dtype.fields.keys():\n",
    "            drinking_start = state_info[\"Drinking\"][0][0][0][0]\n",
    "            drinking_end = state_info[\"Drinking\"][0][0][0][1]\n",
    "\n",
    "            if not (np.isnan(drinking_start) or np.isnan(drinking_end)):\n",
    "                drinking_duration = drinking_end - drinking_start\n",
    "                total_drinking_duration += drinking_duration\n",
    "                # print(f\"Trial: {trial}, drinking_duration: {drinking_duration}, total_drinking_duration: {total_drinking_duration}\")\n",
    "\n",
    "                if not np.isnan(lickometer[\"BNCHigh\"][trial]).all():\n",
    "                    BNC1High_count = sum([1 for l in lickometer[\"BNCHigh\"][trial][0] if l >= drinking_start and l <= drinking_end])\n",
    "                else:\n",
    "                    BNC1High_count = 0\n",
    "                total_BNC1High_count += BNC1High_count\n",
    "\n",
    "    print(f\"Total drinking duration: {total_drinking_duration}\")\n",
    "\n",
    "    average_BNC1High_count = total_BNC1High_count / num_trials if num_trials > 0 else np.nan\n",
    "    average_frequency = (total_BNC1High_count / total_drinking_duration) if total_drinking_duration > 0 else np.nan\n",
    "\n",
    "    print(\"File:\", mat_file)\n",
    "    print(\"Average BNC1High count during drinking periods:\", average_BNC1High_count)\n",
    "    print(\"Average frequency:\", average_frequency)\n",
    "   \n"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "python37",
   "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.7.12"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}