import matplotlib
matplotlib.use('Agg')
from grid_cell_model.submitting import flagparse
import noisefigs
from noisefigs.env import MplEnvironment
from noisefigs.plotters.base import SeparateMultipageSaver

import config

parser = flagparse.FlagParser()
parser.add_flag('--param_exploration')
parser.add_argument('--data_root', type=str, help='Data root directory',
                    default='simulation_data/network_test/150pA')
args = parser.parse_args()

env = MplEnvironment(config=config.get_config())


if args.param_exploration or args.all:
    for file_name in os.listdir(args.data_root):
        if 'rasters' in file_name:
            print(file_name)
            env.register_class(
                noisefigs.plotters.PopulationActivityPlotter,
                config={
                    'data_root'     : args.data_root,
                    'data_file_name': file_name,

                    'PopulationActivityPlotter': {
                        'fig_size': (6, 5),
                        't_limits': (5e3, 6e3),
                                 0.98,
                                 'cbar_kw':
                                 dict(ticks=ti.MultipleLocator(0.2), ),
                             },
                         })
    env.plot()

if args.rasters or args.all:
    shape = (31, 31)

    output_dir_0 = join('simulation_data', 'ee_connections_ei_flat',
                        'standard_sweep_g_EE_3060_pEE_sigma_0_0833',
                        'gamma_bump', '0pA')
    sp_0 = JobTrialSpace2D(shape, output_dir_0)
    new_config = deepcopy(config.get_config())
    env = MplEnvironment(config=new_config)
    env.register_class(noisefigs.plotters.PopulationActivityPlotter,
                       config={
                           'data_root': output_dir_0,
                           'data_file_name': sp_0[1][22].file_name_base,
                           'output_dir': 'panels_standard_gEE_3060',
                           'PopulationActivityPlotter': {
                               'fname_prefix': '0pA_r1_c22_',
                               'raster_rect': (.075, 0.35, 0.93, 0.97),
                               'fig_saver':
                               SeparateMultipageSaver(None, 'pdf'),
                               'fig_size': (8, 6),
                               't_limits': (0, 5e3),
                               'snapshot_tstep': 4,
                               'e_snapshots_rect': (.075, .15, 0.93, 0.25),
                               'i_snapshots_rect': (.075, .02, 0.93, 0.12),
parser.add_argument('--shapeCols',    type=int, required=True)
parser.add_argument("--output_dir",   type=str, required=True)
parser.add_argument("--figure_dir",   type=str, required=True)
parser.add_argument("--job_num",      type=int) # unused
parser.add_argument("--type",         type=str, choices=common.allowed_types,
                    required=True, nargs="+")
o = parser.parse_args()

###############################################################################
startT = time.time()

shape = (o.shapeRows, o.shapeCols)
sp = JobTrialSpace2D(shape, o.output_dir)

if common.pop_type in o.type:
    env = MplEnvironment(config=config.get_config())
    env.register_class(
        noisefigs.plotters.PopulationActivityPlotter,
        config={
            'data_root'     : o.output_dir,
            'data_file_name': sp[o.row][o.col].file_name_base,
            'output_dir'    : o.figure_dir,

            'PopulationActivityPlotter': {
                'fname_prefix': 'r%03d_c%03d_' % (o.row, o.col),
                'raster_rect': (.075, 0.35, 0.95, 0.97),
                'fig_saver': SeparateMultipageSaver(None, 'pdf'),
                'fig_size': (10, 6),
                't_limits': (0, 5e3),

                'snapshot_tstep': 4,
from grid_cell_model.submitting import flagparse
import noisefigs
from noisefigs.env import MplEnvironment
from noisefigs.plotters.base import SeparateMultipageSaver

import config

parser = flagparse.FlagParser()
parser.add_flag('--param_exploration')
parser.add_argument('--data_root',
                    type=str,
                    help='Data root directory',
                    default='simulation_data/network_test/150pA')
args = parser.parse_args()

env = MplEnvironment(config=config.get_config())

if args.param_exploration or args.all:
    for file_name in os.listdir(args.data_root):
        if 'job' in file_name:
            print(file_name)
            env.register_class(noisefigs.plotters.PopulationActivityPlotter,
                               config={
                                   'data_root': args.data_root,
                                   'data_file_name': file_name,
                                   'PopulationActivityPlotter': {
                                       'fname_prefix':
                                       'test_%s_' % file_name,
                                       'fig_size': (4, 6),
                                       't_limits': (0, 2.5e3),
                                       'fig_saver':
Example #5
0
parser.add_argument("--job_num", type=int)  # unused
parser.add_argument("--type",
                    type=str,
                    choices=common.allowed_types,
                    required=True,
                    nargs="+")
o = parser.parse_args()

###############################################################################
startT = time.time()

shape = (o.shapeRows, o.shapeCols)
sp = JobTrialSpace2D(shape, o.output_dir)

if common.pop_type in o.type:
    env = MplEnvironment(config=config.get_config())
    env.register_class(noisefigs.plotters.PopulationActivityPlotter,
                       config={
                           'data_root': o.output_dir,
                           'data_file_name': sp[o.row][o.col].file_name_base,
                           'output_dir': o.figure_dir,
                           'PopulationActivityPlotter': {
                               'fname_prefix': 'r%03d_c%03d_' % (o.row, o.col),
                               'raster_rect': (.075, 0.35, 0.95, 0.97),
                               'fig_saver':
                               SeparateMultipageSaver(None, 'pdf'),
                               'fig_size': (10, 6),
                               't_limits': (0, 5e3),
                               'snapshot_tstep': 4,
                               'e_snapshots_rect': (.075, .15, 0.95, 0.25),
                               'i_snapshots_rect': (.075, .02, 0.95, 0.12),
#!/usr/bin/env python
from __future__ import absolute_import, print_function

import matplotlib; matplotlib.use('Agg')
from grid_cell_model.submitting import flagparse
import noisefigs
from noisefigs.env import MplEnvironment

import config

parser = flagparse.FlagParser()
parser.add_flag('--rasters_and_bumps')
parser.add_flag('--rasters_and_bumps_test')
args = parser.parse_args()

env = MplEnvironment(config=config.get_config())

if args.rasters_and_bumps or args.all:
    env.register_class(noisefigs.plotters.PopulationActivityPlotter)

if args.rasters_and_bumps_test or args.all:
    env.register_class(
        noisefigs.plotters.PopulationActivityPlotter,
        config={
            'data_root'     : 'simulation_data_local/network_test/150pA',

            'PopulationActivityPlotter': {
                'fname_prefix': 'test_',
                'fig_size': (4, 6),
                't_limits': (0, 2.5e3),
            },
from grid_cell_model.submitting import flagparse
import noisefigs
from noisefigs.env import MplEnvironment
from noisefigs.plotters.base import SeparateMultipageSaver

import config

parser = flagparse.FlagParser()
parser.add_flag('--param_exploration')
parser.add_argument('--data_root',
                    type=str,
                    help='Data root directory',
                    default='simulation_data/network_test/150pA')
args = parser.parse_args()

env = MplEnvironment(config=config.get_config())

if args.param_exploration or args.all:
    for file_name in os.listdir(args.data_root):
        if 'rasters' in file_name:
            print(file_name)
            env.register_class(noisefigs.plotters.PopulationActivityPlotter,
                               config={
                                   'data_root': args.data_root,
                                   'data_file_name': file_name,
                                   'PopulationActivityPlotter': {
                                       'fig_size': (6, 5),
                                       't_limits': (5e3, 6e3),
                                       'raster_rect': (.125, 0.35, 0.99, 0.97),
                                       'snapshot_tstep':
                                       1,