# - compute (pyTree) -
import Generator.PyTree as G
import Converter.PyTree as C
import Fast.PyTree as Fast
import Converter.Internal as Internal
import Fast.Utils as Utils
# NP: 0: Seq, NP> 0: running with mpi, distributed on NP procs
NP = Utils.getArgs1()
if NP > 0:
    import Converter.Mpi as Cmpi
    import FastS.Mpi as FastS
    rank = Cmpi.rank
else:
    import FastS.PyTree as FastS
    rank=0

NIT = 10000

# Reading input files: t.cgns/restart.cgns and tc.cgns
t,tc,ts,graph = Fast.load('t.cgns', 'tc.cgns', split='single', 
                          restart=False, NP=NP)

it0 = 0; time0 = 0.
first = Internal.getNodeFromName1(t, 'Iteration')
if first is not None: it0 = Internal.getValue(first)
first = Internal.getNodeFromName1(t, 'Time')
if first is not None: time0 = Internal.getValue(first)

# Numerics
numb = {}
numb["temporal_scheme"]    = "explicit"
numb["ss_iteration"]       = 20
numz = {}
numz["time_step"]          = 0.007
numz["time_step_nature"]   = "global"
numz["cfl"]                = 1.
numz["scheme"]             = "senseur"
FastS._setNum2Zones(t, numz) ; FastS._setNum2Base(t, numb)

(t, tc, metrics) = FastS.warmup(t, tc, graph)

# Extraction de zone
zone = Internal.getNodeFromName(tc, 'join.6')
if zone is not None: file = open('density', 'w')

time = time0
time_step = Internal.getNodeFromName(t, 'time_step')
time_step = Internal.getValue(time_step)
for it in xrange(NIT):
    FastS._compute(t, metrics, it, tc, graph)
    if it%100 == 0:
        if rank == 0: print '- %d / %d - %f'%(it+it0, NIT+it0, time)
        if zone is not None:
            density = C.getValue(zone, 'Density', (54,25,12))
            file.write(str(density)+'\n') ; file.flush()
    time += time_step

if zone is not None: file.close()

# time stamp
Internal.createUniqueChild(t, 'Iteration', 'DataArray_t', value=it0+NIT)
Internal.createUniqueChild(t, 'Time', 'DataArray_t', value=time)
Fast.save(t, 'restart.cgns', split='single', NP=NP)