main.py

import os
import platform

if platform.system() == "Linux":
    os.environ["ARCADE_HEADLESS"] = "True"

import time
import numpy as np
import pandas as pd
from src.sim.simulation import sim
import pyglet

"""
File to run simulation
"""
DEFAULT_PARAM_NAMES = ["k_s","k_d","k1","k2","k3","k4","k5","k6","tau"]
INDEP_PARAM_NAMES = ["ks_aux","kd_aux","ks_pinu","kd_pinu","kd_pinloc","ks_auxlax","kd_auxlax","k1","k2","k3","k4","k5","k6","tau"]

def make_default_param_series_default():
    ks_range = 0.0553636
    kd_range = 0.0278859
    k1_range = 60
    k2_range = 64
    k3_range = 35
    k4_range = 61
    k5_range = 0.464545
    k6_range = 0.959596
    tau_range = 18
    # ks_range = .3 #np.linspace(0.001, 0.3, 100).astype(float)
    # kd_range = .003 #np.linspace(0.0001, 0.03, 100).astype(float)
    # k1_range = 10 #np.round(np.linspace(10, 160, 100)).astype(int)
    # k2_range = 50 #np.round(np.linspace(50, 100, 100)).astype(int)
    # k3_range = 10 #np.round(np.linspace(10, 75, 100)).astype(int)
    # k4_range = 50 #np.round(np.linspace(50, 100, 100)).astype(int)
    # k5_range = .08 #np.linspace(0.07, 1, 100).astype(float)#np.linspace(0.07, 20, 100).astype(float) # kal
    # k6_range = .3 #np.linspace(0.2, 1, 100).astype(float)#np.linspace(0.2, 20, 100).astype(float) # kpin
    # tau_range = 60 #np.round(np.linspace(60, 7200, 100)).astype(int)
    param_vals =  [ks_range, kd_range, k1_range, k2_range, k3_range, k4_range, k5_range, k6_range, tau_range]
    return pd.Series(param_vals, index=DEFAULT_PARAM_NAMES)

def make_indep_param_series():
    ks_aux = 0.266778
    kd_aux = 0.0224495
    ks_pinu = 0.523434
    kd_pinu = .0176172
    kd_pinloc = 0.00191212
    ks_auxlax = 0.0040202
    kd_auxlax = 0.0257717
    k1_range = 60
    k2_range = 60
    k3_range = 33
    k4_range = 58
    k5_range = 1
    k6_range = 0.975758
    tau_range =3
    param_vals = [ks_aux, kd_aux, ks_pinu, kd_pinu, kd_pinloc, ks_auxlax, kd_auxlax, k1_range, k2_range, k3_range, k4_range, k5_range, k6_range, tau_range]
    return pd.Series(param_vals, index=INDEP_PARAM_NAMES)

if __name__ == '__main__':
    timestep = 1
    vis = True
    start_time = time.time()
    sim.main(timestep, root_midpoint_x, vis, cell_val_file="default", v_file="default", gparam_series=make_indep_param_series())
    end_time = time.time()
    elapsed_time = end_time - start_time
    print(f"Elapsed Time: {elapsed_time} seconds")