import numpy
import sys
import zmq
import time
import h5py
import os

path="/home/alex/Downloads/pedestal"
ext=".hdf5"

port ="5556"
if len(sys.argv)>1:
    port = sys.argv[1]
    int(port)

if len(sys.argv)>2:
    port1 = sys.argv[2]
    int(port1)

#Socket to talk to server
context = zmq.Context()
socket  = context.socket(zmq.SUB)

print("Collecting updates from weather server...")
socket.connect("tcp://localhost:%s"%port)

if len(sys.argv)>2:
    socket.connect("tcp://localhost:%s"%port1)

#Subscribe to zipcode, default is NYC,10001
topicfilter = "10001"
socket.setsockopt_string(zmq.SUBSCRIBE,topicfilter)
#Process 5 updates
total_value=0
count= -1
azi= []
elev=[]
time0=[]
#for update_nbr in range(250):
while(True):
    string= socket.recv()
    topic,ang_elev,ang_elev_dec,ang_azi,ang_azi_dec,seconds,seconds_dec= string.split()
    ang_azi =float(ang_azi)+1e-3*float(ang_azi_dec)
    ang_elev =float(ang_elev)+1e-3*float(ang_elev_dec)
    seconds =float(seconds) +1e-6*float(seconds_dec)
    azi.append(ang_azi)
    elev.append(ang_elev)
    time0.append(seconds)
    count +=1
    if count == 100:
        timetuple=time.localtime()
        epoc = time.mktime(timetuple)
         #print(epoc)
        fullpath = path + ("/" if path[-1]!="/" else "")

        if not os.path.exists(fullpath):
            os.mkdir(fullpath)

        azi_array  = numpy.array(azi)
        elev_array = numpy.array(elev)
        time0_array= numpy.array(time0)
        pedestal_array=numpy.array([azi,elev,time0])
        count=0
        azi= []
        elev=[]
        time0=[]
        #print(pedestal_array[0])
        #print(pedestal_array[1])

        meta='PE'
        filex="%s%4.4d%3.3d%10.4d%s"%(meta,timetuple.tm_year,timetuple.tm_yday,epoc,ext)
        filename = os.path.join(fullpath,filex)
        fp = h5py.File(filename,'w')
        #print("Escribiendo HDF5...",epoc)
        #·················· Data·....······································
        grp = fp.create_group("Data")
        dset = grp.create_dataset("azimuth"  , data=pedestal_array[0])
        dset = grp.create_dataset("elevacion", data=pedestal_array[1])
        dset = grp.create_dataset("utc"      , data=pedestal_array[2])
        #·················· Metadata·······································
        grp = fp.create_group("Metadata")
        dset = grp.create_dataset("utctimeInit", data=pedestal_array[2][0])
        timeInterval = pedestal_array[2][1]-pedestal_array[2][0]
        dset = grp.create_dataset("timeInterval", data=timeInterval)
        fp.close()

#print ("Average messagedata value for topic '%s' was %dF" % ( topicfilter,total_value / update_nbr))