import numpy as np
from tools.generation.config import DotheLoop, get_config
import os
import sys
from mpi4py import MPI
from tools.connec.comp_cmap import comp_connec
from tools.postprocessK.comp_PostKeff import comp_postKeff
from shutil import copyfile
from tools.solver.comp_Kperm_scale import comp_kperm_sub
from tools.solver.Ndar import PetscP
from tools.generation.fftma_gen import fftmaGenerator

CONFIG_FILE_PATH = (
    "config.ini"
    if "CONFIG_FILE_PATH" not in os.environ
    else os.environ["CONFIG_FILE_PATH"]
)


def realization(job):

    if job == -1:
        return

    conffile = CONFIG_FILE_PATH
    parser, iterables = get_config(conffile)
    start_job = int(parser.get("General", "startJob"))

    if job < start_job:
        return

    rdir = "./" + parser.get("General", "simDir") + "/"
    datadir = rdir + str(job) + "/"
    create_dir(datadir, job)
    if job == 0:
        copyfile(conffile, rdir + "config.ini")

    genera = parser.get("Generation", "genera")
    if genera != "no":
        fftmaGenerator(datadir, job, CONFIG_FILE_PATH)
        # os.system('CONFIG_FILE_PATH=' + CONFIG_FILE_PATH + ' python3 ./tools/generation/fftma_gen.py ' + datadir +' ' + str(job))

    nr = DotheLoop(job, parser, iterables)[3] - iterables["seeds"][0]
    Cconec = parser.get("Connectivity", "conec")
    if Cconec != "no":
        comp_connec(parser, datadir, nr)

    n_p = int(parser.get("Solver", "num_of_cores"))
    ref = int(parser.get("Solver", "ref"))
    solv = parser.get("Solver", "solve")
    Rtol = parser.get("Solver", "rtol")

    if solv != "no":
        if n_p > 1:
            icomm = MPI.COMM_SELF.Spawn(
                sys.executable,
                args=["./tools/solver/Ndar.py", datadir, str(ref), "0", Rtol, "1"],
                maxprocs=n_p,
            )
            icomm.Disconnect()
        else:
            PetscP(datadir, ref, "0", True, float(Rtol), 0)

    compkperm = parser.get("K-Postprocess", "kperm")
    if compkperm != "no":
        # print('start kperm')
        comp_kperm_sub(parser, datadir, nr)
    # print('finished job ' +str(job))

    postP = parser.get("K-Postprocess", "postprocess")
    if postP != "no":
        comp_postKeff(parser, datadir, nr)

    return


def create_dir(datadir, job):

    try:
        os.makedirs(datadir)
    except:
        print("Warning: Unable to create dir job: " + str(job))

    return