#!/usr/bin/env python
########################################################################
#     HISTORY
#
#     APRIL      2024 HUANG               MDL    SCRIPT CREATED
#     MAY        2024 SHAFER              MDL    MODIFIED LOCATION OF
#                                                DATA DIRECTORY
#
########################################################################

import os
import sys
from get_station_header import *
from get_flight_cat import *

data = os.environ.get('DATA')
#comout = os.environ.get('COMOUT')
STA = sys.argv[1]
HH = sys.argv[2]
MN = sys.argv[3]

sta_header=get_station_header(STA)
print(sta_header)

path=data+"/itdlp_uncert."+HH+MN
#path=comout+"/itdlp_uncert."+HH+MN
#path=comout+"/itdlp_test."+HH+MN

# Read Valid Times
timefile=path+"/validtimes.txt"
s = open(timefile, "r")

n = sum(1 for line in open(timefile))

current_line=1;

validtimes=[]
validHH=[]
validMM=[]

for line in s.readlines():

    validtimes.append(line)

    tempHH=line[8:10]
    tempHH.strip()
    tempMM=line[-3:]
    tempMM.strip()

    print(tempHH,tempMM)

    validHH.append(tempHH)
    validMM.append(tempMM)
    current_line += 1

# extract date and write to MM/DD/YYYY format
date=validtimes[0]
year=date[0:4]
month=date[4:6]
day=date[6:8]

if float(month) < 10:
   month=" "+month[1]

str_date=month+"/"+day+"/"+year
print(str_date)

####################################3

outfile=STA+"_15min_cv_cats_"+HH+MN+"z.txt"
out = open(outfile, "w")

#if STA == "KSMB":
#   out.write("SFOA"+"   "+sta_header+"GFS LAMP "+HH+MN+" UTC  "+str_date+"\n")
#else:
out.write(STA+"   "+sta_header+"GFS LAMP "+HH+MN+" UTC  "+str_date+"\n")

out.write("UTC   "+validHH[0].strip()+" "+validHH[1].strip()+" "+validHH[2].strip()+" "+validHH[3].strip()+" "+validHH[4].strip()+" "+validHH[5].strip()+" "+validHH[6].strip()+" "+validHH[7].strip()+" "+validHH[8].strip()+" "+validHH[9].strip()+" "+validHH[10].strip()+" "+validHH[11].strip()+" "+validHH[12].strip()+" "+validHH[13].strip()+" "+validHH[14].strip()+" "+validHH[15].strip()+" "+validHH[16].strip()+" "+validHH[17].strip()+" "+validHH[18].strip()+" "+validHH[19].strip()+" "+validHH[20].strip()+" "+validHH[21].strip()+" "+validHH[22].strip()+" "+validHH[23].strip()+"\n" )

out.write("MIN   "+validMM[0].strip()+" "+validMM[1].strip()+" "+validMM[2].strip()+" "+validMM[3].strip()+" "+validMM[4].strip()+" "+validMM[5].strip()+" "+validMM[6].strip()+" "+validMM[7].strip()+" "+validMM[8].strip()+" "+validMM[9].strip()+" "+validMM[10].strip()+" "+validMM[11].strip()+" "+validMM[12].strip()+" "+validMM[13].strip()+" "+validMM[14].strip()+" "+validMM[15].strip()+" "+validMM[16].strip()+" "+validMM[17].strip()+" "+validMM[18].strip()+" "+validMM[19].strip()+" "+validMM[20].strip()+" "+validMM[21].strip()+" "+validMM[22].strip()+" "+validMM[23].strip()+"\n" )

###################################

# Get CIG and VIS categorical forecasts

#data=comout+"/itdlp_uncert."+HH+MN+"/"

#infile=data+"u660_out_"+STA+".txt"
infile=path+"/u660_out_"+STA+".txt"

#Read from file:
f = open(infile, "r")

num_lines = sum(1 for line in open(infile))

print(num_lines);

current_line=1;
current_line_grp=1;

lmpcig_list=[]
lmpvis_list=[]

## Loop over each line in the file.
for line in f.readlines():

#    # Strip the line to remove whitespace.
    line=" ".join(line.split());     
    lisp = line.split(" ");

    sta = lisp[1];

#    lmpcig = lisp[11];
#    lmpvis = lisp[12];
    lmpcig = lisp[2];
    lmpvis = lisp[3];
    if float(lmpcig) > 9.:
        lmpcig = 99
    if float(lmpvis) > 9.:
        lmpvis = 99

    lmpcig = float(lmpcig);
    lmpcig = int(lmpcig);
    lmpcig = str(lmpcig);
    if float(lmpcig) < 10.:
        lmpcig = ' ' + lmpcig
    lmpcig_list.append(lmpcig)

    lmpvis = float(lmpvis);
    lmpvis = int(lmpvis);
    lmpvis = str(lmpvis);
    if float(lmpvis) < 10.:
        lmpvis = ' ' + lmpvis
    lmpvis_list.append(lmpvis)

###########################################################################

#out.write("CIG    "+lmpcig_list[0].strip()+"  "+lmpcig_list[1].strip()+"  "+lmpcig_list[2].strip()+"  "+lmpcig_list[3].strip()+"  "+lmpcig_list[4].strip()+"  "+lmpcig_list[5].strip()+"  "+lmpcig_list[6].strip()+"  "+lmpcig_list[7].strip()+"  "+lmpcig_list[8].strip()+"  "+lmpcig_list[9].strip()+"  "+lmpcig_list[10].strip()+"  "+lmpcig_list[11].strip()+"  "+lmpcig_list[12].strip()+"  "+lmpcig_list[13].strip()+"  "+lmpcig_list[14].strip()+"  "+lmpcig_list[15].strip()+"  "+lmpcig_list[16].strip()+"  "+lmpcig_list[17].strip()+"  "+lmpcig_list[18].strip()+"  "+lmpcig_list[19].strip()+"  "+lmpcig_list[20].strip()+"  "+lmpcig_list[21].strip()+"  "+lmpcig_list[22].strip()+"  "+lmpcig_list[23].strip()+"\n" )
out.write("CIG   "+lmpcig_list[0]+" "+lmpcig_list[1]+" "+lmpcig_list[2]+" "+lmpcig_list[3]+" "+lmpcig_list[4]+" "+lmpcig_list[5]+" "+lmpcig_list[6]+" "+lmpcig_list[7]+" "+lmpcig_list[8]+" "+lmpcig_list[9]+" "+lmpcig_list[10]+" "+lmpcig_list[11]+" "+lmpcig_list[12]+" "+lmpcig_list[13]+" "+lmpcig_list[14]+" "+lmpcig_list[15]+" "+lmpcig_list[16]+" "+lmpcig_list[17]+" "+lmpcig_list[18]+" "+lmpcig_list[19]+" "+lmpcig_list[20]+" "+lmpcig_list[21]+" "+lmpcig_list[22]+" "+lmpcig_list[23]+"\n" )

#out.write("VIS    "+lmpvis_list[0].strip()+"  "+lmpvis_list[1].strip()+"  "+lmpvis_list[2].strip()+"  "+lmpvis_list[3].strip()+"  "+lmpvis_list[4].strip()+"  "+lmpvis_list[5].strip()+"  "+lmpvis_list[6].strip()+"  "+lmpvis_list[7].strip()+"  "+lmpvis_list[8].strip()+"  "+lmpvis_list[9].strip()+"  "+lmpvis_list[10].strip()+"  "+lmpvis_list[11].strip()+"  "+lmpvis_list[12].strip()+"  "+lmpvis_list[13].strip()+"  "+lmpvis_list[14].strip()+"  "+lmpvis_list[15].strip()+"  "+lmpvis_list[16].strip()+"  "+lmpvis_list[17].strip()+"  "+lmpvis_list[18].strip()+"  "+lmpvis_list[19].strip()+"  "+lmpvis_list[20].strip()+"  "+lmpvis_list[21].strip()+"  "+lmpvis_list[22].strip()+"  "+lmpvis_list[23].strip()+"\n" )
out.write("VIS   "+lmpvis_list[0]+" "+lmpvis_list[1]+" "+lmpvis_list[2]+" "+lmpvis_list[3]+" "+lmpvis_list[4]+" "+lmpvis_list[5]+" "+lmpvis_list[6]+" "+lmpvis_list[7]+" "+lmpvis_list[8]+" "+lmpvis_list[9]+" "+lmpvis_list[10]+" "+lmpvis_list[11]+" "+lmpvis_list[12]+" "+lmpvis_list[13]+" "+lmpvis_list[14]+" "+lmpvis_list[15]+" "+lmpvis_list[16]+" "+lmpvis_list[17]+" "+lmpvis_list[18]+" "+lmpvis_list[19]+" "+lmpvis_list[20]+" "+lmpvis_list[21]+" "+lmpvis_list[22]+" "+lmpvis_list[23]+"\n" )

out.write("\n")

out.close()
print("here!");