import virgo
import datetime
from gpiozero import LED

led1 = LED(23)
led2 = LED(24)
ReadQty = 288
lat=-36.4 # Detail removed
lon=-174.5 # Detail removed
alt=90
az=0
height=86
diameter=2.3
freq=1420.4057517667e6

led1.on()

# Define observation parameters
obs = {
    'dev_args': '',
    'rf_gain': 30,
    'if_gain': 0,
    'bb_gain': 0,
    'frequency': 1420e6,
    'bandwidth': 2.4e6,
    'channels': 2048,
    't_sample': 1,
    'duration': 240
}




# Begin data acquisition
led2.blink()

i=0
while i < ReadQty :
    
    
    x = datetime.datetime.now()
    print(x.strftime("%Y-%m-%d %H:%M:%S"))
    prefix = 'data/'+x.strftime("%Y-%m-%d %H:%M:%S")
    
    
    RA, DEC = virgo.equatorial(alt, az, lat, lon, height)
    print ('RA : '+ str(RA))
    print ('DEC : '+ str(DEC))
    print ('Beamwidth : '+ str(virgo.beamwidth(diameter, freq)))
    print ('Gain : '+str(virgo.gain(diameter, freq)))
    
    # Create Header File for Observation Details
    filename = prefix + '-Obs Header.txt'
    print(filename)
    f = open(filename, "w")
    f.write ("Observation Details \n")
    f.write ('lat : '+ str(lat)+'\n')
    f.write ('lon : '+ str(lon)+'\n')
    f.write ('height : '+ str(height)+'\n')
   
    f.write ('RA : '+ str(RA)+'\n')
    f.write ('DEC : '+ str(DEC)+'\n')
    f.write ('Beamwidth : '+ str(virgo.beamwidth(diameter, freq))+'\n')
    f.write ('Gain : '+str(virgo.gain(diameter, freq)))

    f.close()
    
    virgo.observe(obs_parameters=obs, obs_file=prefix+'_observation.dat')

    # Analyze data, mitigate RFI and export the data as a CSV file
    virgo.plot(
           obs_parameters=obs, 
           n=20, m=35, 
           f_rest=freq,
           obs_file=prefix+'_observation.dat',
           cal_file='calibration.dat',           
           power_csv=prefix+'_time_series.csv',
           dB=True, 
           spectra_csv=prefix+'_spectrum.csv', 
           plot_file=prefix+'_plot.png')
    
    virgo.map_hi(
        RA,
        DEC ,
        plot_file = prefix+'_hi_map_plot.png')
    
    i+=1

led1.off()
led2.off()