"""Plotting for manuscript figures
----------------------------------

This script needs to be executed after `ms_figures_simulations.py`.
"""

import os
from mesocircuit import mesocircuit_framework as mesoframe
from mesocircuit.plotting import ms_figures as figures


################################################################################
# Load MesocircuitExperiments and pass the respective circuits to the functions
# for generating figures.

output_dir = 'ms_figures'
if not os.path.isdir(output_dir):
    os.makedirs(output_dir)

################################################################################
# Parameters.

if 0:
    meso_exp_reference = mesoframe.MesocircuitExperiment(
        'reference', load=True)
    meso_exp_upscaled = mesoframe.MesocircuitExperiment(
        'upscaled_1mm2', load=True)
    figures.parameters(
        output_dir,
        meso_exp_reference.circuits[0], meso_exp_upscaled.circuits[0])

################################################################################
# Reference vs. upscaled (1mm2 sampled): spike raster and statistics overview.

if 0:
    meso_exp_reference = mesoframe.MesocircuitExperiment(
        'reference', load=True)
    meso_exp_upscaled_1mm2 = mesoframe.MesocircuitExperiment(
        'upscaled_1mm2', load=True)
    figures.reference_vs_upscaled(
        output_dir,
        meso_exp_reference.circuits[0], meso_exp_upscaled_1mm2.circuits[0],
        plot_rasters=True, plot_statistics=True)

################################################################################
# Upscaled (1mm2 sampled): correlation.
if 1:
    meso_exp_upscaled_1mm2 = mesoframe.MesocircuitExperiment(
        'upscaled_1mm2', load=True)
    figures.correlation(
        output_dir,
        meso_exp_upscaled_1mm2.circuits[0])


################################################################################
# Evoked activity.

if 0:
    meso_exp_evoked = mesoframe.MesocircuitExperiment(
        'evoked', load=True)
    figures.evoked_activity(output_dir, meso_exp_evoked.circuits[0])