import sys
import numpy as np
import h5py
import os
import linecache
import nest
from mpi4py import MPI
from scaffold import Scaffold
from scaffold.config import JSONConfig, from_hdf5

comm = MPI.COMM_WORLD
size = comm.Get_size()
rank = comm.Get_rank()

nest.set_verbosity('M_ERROR')
nest.Install("extracerebmodule")

filename_h5 = "scaffold_4_4.hdf5"

hdf_config = from_hdf5(filename_h5)
scaffold_inverse = Scaffold(hdf_config, from_file=filename_h5)
print("OK SCAFFOLD" + str(rank), flush=True)

adapter_inverse = scaffold_inverse.get_simulation("FCN_2019")
print("OK ADAPTER" + str(rank), flush=True)

with h5py.File(filename_h5, "a") as hdf_handle:
    adapter_inverse.prepare(hdf_handle)
    print("OK PREPARE" + str(rank), flush=True)

comm.Barrier()
print("Created", flush=True)

adapter_inverse.simulate(nest)
print("Simulated", flush=True)