#!/usr/bin/python3
# service.py driver script to control the assimilation system
# version: 0.1
# July 2012
# Report bugs to: a.srinivasan@tendral.com

import sys
import datetime as dt
import os
import subprocess
#import numpy as NP
from glob import glob
import shutil
from  defs    import *
from  limits  import dolimits
from  hycom   import dohycom
from  clean   import do_tsis_clean
from  assim   import doxprep,doxgmrf


format = '%Y%m%d%H'
dtgref=dt.datetime.strptime(str(1900123100), format)
scr_dir="/p/work1/abozec/HYCOM-TSIS/GOMb0.04/expt_35.1/data/"
dtgs=dt.datetime.strptime(str(2024030400), format)

nsteps=9
npes=32
npes_tsis=1

jdateDiff = dtgs - dtgref
je=jdateDiff.days
jd_start=je
jd_start=jd_start-0.25
adtg=dtgs
sadtg=adtg.strftime('%Y%m%d%H')

for steps in range(0, nsteps): 
    start=jd_start+steps*1
    end=start+1.
    adtg=adtg+dt.timedelta(hours=24)
    sadtg=adtg.strftime('%Y%m%d%H')
    iyear=adtg.strftime('%Y')
    iday=adtg.strftime('%j')
    ihour=adtg.strftime('%H')
    #print (iyear,iday,ihour,start,end,sadtg)
    dolimits(scr_dir,start,end)
    dohycom(scr_dir,npes)
    if os.path.isfile("restart_out.a"):
        #print (sadtg[6:8])
        # save restart on the 15th 
        #if(sadtg[6:8]=="15"):
        #    shutil.copy("restart_out.b","restart_"+sadtg+".b")
        #    shutil.copy("restart_out.a","restart_"+sadtg+".a")
        #dorstc2r(scr_dir,sadtg)
        doxprep(scr_dir,npes_tsis,sadtg)
        doxgmrf(scr_dir,npes_tsis,sadtg)
        do_tsis_clean(scr_dir)
    else:
        sys.exit(0)
sys.exit(0)