params.py

import pathlib
import string
import matplotlib
import numpy as np

# Global params
from monkey import defs as monkey_defs
from mouse import defs as mouse_defs
from random_walk import defs as random_walk_defs
from rnn import defs as rnn_defs

repoPath = pathlib.Path.cwd()
root = pathlib.Path("/data")
# root = pathlib.Path(repoPath/"data")
figPath = repoPath / 'figures'
LargeFig = (7,8)
MedFig = (4.3,4.3)

rng = np.random.default_rng(np.random.SeedSequence(12345))
n_iter = 100
Behav_corr_TH = 0.5
annotate_stats = False

def set_rc_params(dictArg:dict ={}):
    matplotlib.rcParams['backend']   = 'PDF'
    matplotlib.rcParams['lines.markersize'] = 4
    matplotlib.rcParams['xtick.major.pad']  = 1
    matplotlib.rcParams['ytick.major.pad']  = 1
    matplotlib.rcParams['axes.labelpad']    = 2
    matplotlib.rcParams['axes.titlepad']    = 5
    matplotlib.rcParams['font.size']        = 8
    matplotlib.rcParams['axes.titlesize']   = 8
    matplotlib.rcParams['axes.labelsize']   = 8
    matplotlib.rcParams['xtick.labelsize']  = 6.5
    matplotlib.rcParams['ytick.labelsize']  = 6.5
    matplotlib.rcParams['legend.fontsize']  = 6.5
    matplotlib.rcParams['legend.title_fontsize']   = 6.5
    matplotlib.rcParams['text.usetex']      = True
    matplotlib.rcParams['font.family']      = 'sans-serif'
    matplotlib.rcParams['font.serif']       = 'Helvetica'
    matplotlib.rcParams['legend.frameon']   = False
    matplotlib.rcParams['legend.fancybox']  = False
    matplotlib.rcParams['figure.dpi']       = 600
    
    
    for key,val in dictArg.items():
        matplotlib.rcParams[key] = val


def add_panel_caption(axes: tuple, offsetX: tuple, offsetY: tuple, **kwargs):
    """
    This function adds letter captions (a,b,c,d) to Axes in axes
    at top left, with the specified offset, in RELATIVE figure coordinates
    """
    assert len(axes)==len(offsetX)==len(offsetY), 'Bad input!'

    fig=axes[0].get_figure()
    fbox=fig.bbox
    for ax,dx,dy,s in zip(axes,offsetX,offsetY,string.ascii_uppercase):
        axbox=ax.get_window_extent()
        try:
            ax.text(x=(axbox.x0/fbox.xmax)-abs(dx), y=(axbox.y1/fbox.ymax)+abs(dy),
                    s=s, fontweight='extra bold', fontsize=10, ha='left', va='center',
                    transform=fig.transFigure,**kwargs)
        except: #to cover 3D axes
            ax.text2D(x=(axbox.x0/fbox.xmax)-abs(dx), y=(axbox.y1/fbox.ymax)+abs(dy),
                    s=s, fontweight='extra bold', fontsize=10, ha='left', va='center',
                    transform=fig.transFigure,**kwargs)

class panels:
    "sizes of different panels in the paper"
    TinyH = 0.5
    SmallH = 1.2
    MedH = SmallH + TinyH
    BigH = 3
    
    schmatic = (2,MedH)
    raster = (2.4,SmallH)
    velocity = (LargeFig[0]-schmatic[0],TinyH)
    proj_3d_align = (LargeFig[0],SmallH)
    cca = (1.5, SmallH)
    cca_hist = (2,SmallH)

    rnn_raster = (2,SmallH)
    rnn_velocity = (2, TinyH)
    rnn_cca = (1.5 , MedH -0.2)
    rnn_cca_hist = (2 ,MedH-0.2)

class colors:
    "colors for different data types in the paper"
    MouseM1 = 'tab:blue'
    MouseStr = 'tab:orange'
    MainCC = 'r'
    LowerCC = 'k'
    UpperCC = 'cornflowerblue'
    Sim1CC ='cornflowerblue'
    Sim2CC = 'palevioletred'
    SimAcrossCC = 'indigo'
    MonkeyPts = 'xkcd:brown'
    MousePts = 'xkcd:violet'
    LeftTrial = (0,1,0,1)
    RightTrial = (1,0,0,1)