#!/usr/bin/python

# program displays CG strikes over Oklahoma. 
# written by K. Kuhlman modifying code originally written by B. Fiedler for 
# plotting and using vplot 

import cPickle, vplot, math, getopt, sys

pickledboundaries='world_us_country.pickle'

epsfilename='cgmap.eps'

def reCoordList(a,maptrans,param):
    m=0
    b=[]
    while m<len(a):
        lon,lat,m=vplot.next2(a,m) # vplot utility
        x,y = maptrans(lon,lat,param)
        b.append(x)
        b.append(y)
    return b

#set the bounds for the map:
lonmin = -103.0
lonmax = -94.5
latmin = 33.0
latmax = 38.0
mapfile = open(pickledboundaries,'r')
thedict = cPickle.load(mapfile) #receives the dict stored in pickled file

print "the keys in thedict",thedict.keys()
lines=thedict['lines']
heads=thedict['heads']
mapfile.close()
# prints data type in pickled file
#print "type(lines)= ",type(lines),"   len(lines)= ",len(lines)
#print "type(heads)= ",type(heads),"   len(heads)= ",len(heads)

#start using vplot
a = vplot.eps_class(epsfilename)
a.scale(lonmin,lonmax,latmin,latmax)  # sets boundary for mercator map
for i in range(0,len(heads)):
    dum1,dum2,linelatmax,linelatmin,linelonmax,linelonmin=[float(x) for x in heads[i]]
    #If no part of the line is in the visible map, don't plot it:
    if linelatmin>latmax or linelatmax<latmin or \
        linelonmin>lonmax or linelonmax<lonmin: continue
    line=lines[i]
    a.color('black')
    a.draw(line[:])

### read in cg data ###
cgflash = 'cgdata_040529.pickle'
cgfile = open(cgflash, 'r')
allcgs = cPickle.load(cgfile)
cgfile.close()

print "\nhere are the keys and types in allcgs:"
for key in allcgs.keys():
    print key,type(allcgs[key])
print "\nthe README:\n", allcgs['Readme']
#print "\nthe cgs:\n", allcgs['cgs']


cg = allcgs.get('cgs')

n=0 
j = 0
p = len(cg)
print len(cg) 

###  determine start and end times of plot, input must be string ###
time_start = input("What time to start plotting?  (format- 'hh:mm') ")
time_end = input("What time to end plotting? (format- 'hh:mm') ")

while j < p:
    time = cg[j][1]
    if time.startswith(time_start):
        n = j
        break
    else:
        j += 1 

t = n
k = 0
while k < p:
    time = cg[k][1]
    if time.startswith(time_end):
        l = k
        break
    else:
        k += 1

###  plots circles for CG strikes, +CG=red, -CG=blue  ###
#####  strikes < 10 kA in pink (+) and lt blue (-)  #####
while n < l:
    kA = float(cg[n][5])
    if cg[n][4].startswith('-'):
        if kA < -10.0:
            a.color('blue')
        else:
            a.color(0.64,0.83,0.93)    
    else:
        if kA > 10.0:
            a.color('red')
        else:
            a.color(1.0,0.7,0.75)
    lat = float(cg[n][2])
    lon = float(cg[n][3])
    a.circle(lon,lat,2,'F') # prints CG location
    n += 1

### text for legend in upper corner of map ###
a.color('black')
a.text(lonmin+0.05,latmax,0,10,cg[n][0])
a.text(lonmin+0.05,latmax-0.15,0,10,cg[t][1][0:8]+ " - " +cg[n][1][0:8]+ " UTC")
a.color('red')
a.text(lonmin+0.05,latmax-0.3,0,10,"+ CG")
a.color(1.0,0.7,0.75)
a.text(lonmin+0.7,latmax-0.3,0,10,"(< 10 kA)")
a.color('blue')
a.text(lonmin+0.05,latmax-0.45,0,10,"- CG")
a.color(0.64,0.83,0.93)
a.text(lonmin+0.7,latmax-0.45,0,10,"(< 10 kA)")


a.close()