; Skript pro SpectrumLab pro automatickou detekci meteoru pro projekt Bolidozor
; ver. 2.0 - 23. 11. 2013

;PRIKAZY JEN 255 ZNAKU
;PODMINKY JEN 80 ZNAKU

;Casovace
;timer0 - casovac maximalni mezery mezi 2 meteory K_max_time_btw_met
;timer3 - casovac pro odpocet screenshotu bez meteoru K_max_time_to_scr
;timer4 - csovac pro odecet screenshotu s meteorem po odvaleni nekolika malo sekund aby nebyl meteor moc nahore

;Promenne: 
;n   noise background 
;n_1   sum v okamziku detekce meteoru
;f   frequency of maximum amplitude 
;aver_amp  average amplitude +- 100 Hz  
;aver_amp_1 je average amplitude aktualizovana v prubehu mereni meteoru
;aver_amp_max je maximalni hodnota z hodnot v prubehu mereni meteoru
;mag   magnitude
;hour_count   count of meteors in hour
;dur_count  délka meteoru (count po dobu retekce meteoru)
;T_0 cas posledniho vypoctu FFT v internim formatu SpectrumLabu (aktualizace kazdych 50ms)
;T  cas posledniho vypoctu FFT v textove forme ve formatu "YYYYMMDDhhmmss.sss"
;t3 aktualni hodina a minuta
;id_met identifikator meteoru do screenshotu

;Konstanty:
;K_station_name="JOSY-A"  station name (device code - 20 characters maximum, get your unique station code on http://home.robozor.cz)
;K_path  lokalni adresar pro ukladane soubory
;K_max_time_btw_met=2 minimalni doba mezi 2 meteory v sekundach (pokud po tuto dobu proleti dva, tak se pocitaji jako jeden)
;K_max_time_to_scr=100 maximalni doba bez meteoru, ktera pokud vyprsi, udela se screenshot (max doba za kterou odroluje cely waterfall)
;K_init_cond=50 SpectrumLab evaulation interval (50ms)
;K_time_to_start_up_waterf=4 vteřiny od okraje screenshot Kolik sekund se má počkat do screenshotu po detekci metreoru (jak daleko od okraje)
;K=7 - citlivost detekce meteoru
;K_min_dur_count_bolid=50 detekce bolidu pro audio (2,5 vteriny) 

;Udalosti
;A_measurement (kdykoliv po 50 ms)
;A_still_detect (udalost trvani meteoru nebo bolidu) 
;A_start_detect (udalost detekce zacatku meteoru nebo bolidu)
;A_start_detect_bolid (udalost 2,5 vteriny od detekce meteoru)
;A_end_detect (udalost detekce konce meteoru nebo bolidu)
;A_end_detect_end (konec cyklu udalosti detekce konce meteoru nebo bolidu)
;A_write_waterf (udalost zaznamu popisu detekce do waterfallu)
;A_update_SDR_data_file (udalost generovani datoveho souboru pri zaznamu meteoru)
;A_met_screenshot (udalost generovani screenshotu waterfallu pokud je zaznamenan meteor)
;A_no_met_screenshot (udalost generovani screenshotu waterfallu pokud neni dlouho zaznamenan meteor)
;A_once_hour_20s_before (udalost 1x za hodinu pred ukoncenim hodiny 20 vterin)
;A_once_hour
;A_once_hour_10s_before (udalost 1x za hodinu pred ukoncenim hodiny 10 vterin)
;A_once_hour_30s_after (udalost 1x za hodinu po ukonceni hodiny 30 vterin)


;-----------------A_init - Initialisation of variables and constants--------------------
if( initialising ) then A_still_detect=0:n=0:n_1=0:f=0:f_1=0:aver_amp=0:a=0:mag=0:mag_max=0:hour_count=0:dur_count=0:aver_amp_1=999:aver_amp_max=999:T="0000":T_0=0:t1_0=0:t3=0:bolide=0:bolide2=0
if( initialising ) then id_met="no":id_met2="no":K_station_name="SVAKOV-R2":K_path="/media/sd/meteors/"
if( initialising ) then K_max_time_btw_met=2:K_max_time_to_scr=100:K_init_cond=50:K_time_to_start_up_waterf=4:K=7:K_min_dur_count_bolid=50
if( initialising ) then A_measurement=0:A_start_detect=0:A_end_detect=0:A_end_detect_end=0:A_write_waterf=0:A_update_SDR_data_file=0:A_update_RMOB_Dur_file=0:A_met_screenshot=0
if( initialising ) then A_no_met_screenshot=0:A_once_hour_2s_before=0:A_once_hour=0:A_once_hour_1s_before=0:timer3.restart(K_max_time_to_scr):dur2=0:dur3=0
if( initialising ) then K_path_data=K_path+"/"+K_station_name+"/data/":K_path_audio=K_path+"/"+K_station_name+"/audio/":K_path_scr=K_path+"/"+K_station_name+"/capture/"
if( initialising ) then fopen1(K_path+"/"+K_station_name+"/"+"station.cfg",w):fp1("K_station_name="+K_station_name):fp1("K_max_time_btw_met="+str(K_max_time_btw_met)):fp1("K_max_time_to_scr="+str(K_max_time_to_scr)):fp1("K_init_cond="+str(K_init_cond))
if( initialising ) then fp1("K_time_to_start_up_waterf="+str(K_time_to_start_up_waterf)):fp1("K="+str(K)):fp1("K_min_dur_count_bolid="+str(K_min_dur_count_bolid):fp1("#end")
if( initialising ) then fclose1

;----------------A_measurement - (po 50 ms)-------------------------------------
;n=noise(9800,10200) aktualizace a výpočet hodnoty šumu (záleží na nastavení FFT)
;f=peak_f(10300,10900) max intenzita signálu v daném intervalu frekvencí
;aver_amp=avrg(f-100,f+100) vrací prům. intenzitu signálu 100 Hz kolem maxima 
;-------------------------------------------------------------------------------
if( never ) then REM --- Measurement - (each 50 ms) ----------------------------
if( always ) then n=noise(9800,10200):f=peak_f(10300,10900):aver_amp=avrg(f-100,f+100):a=(n+K):T_0=time:dur2=dur2+1
if( always ) then T=str("YYYYMMDDhhmmss.sss",T_0):t1=str("00000000000000000",val(T)*1000):t3=str("mmss",T_0)
if(A_still_detect=0) then dur2=1

;-----------------A_still_detect (an event of meteor or bolide continue)-----------
if( aver_amp>a ) then A_still_detect=A_still_detect+1:timer0.restart(K_max_time_btw_met):timer3.restart(K_max_time_to_scr):dur_count=dur_count+1:aver_amp_1=a-aver_amp:dur3=dur2

;-----------------A_aver_amp (maximal amplitude calculation)---
if( aver_amp_1<aver_amp_max ) then aver_amp_max=aver_amp_1

;-----------------Some characteristics 100, 400, 800 ms after detection---
if(dur2=2) then f=peak_f(10300,10900):aCt1=round(a-avrg(f-10,f+10)):aRt1=round(a-avrg(f+20,f+50)):aLt1=round(a-avrg(f-50,f-20)):f_1=round(f)
if(dur2=8) then f=peak_f(10300,10900):aCt2=round(a-avrg(f-10,f+10)):aRt2=round(a-avrg(f+20,f+50)):aLt2=round(a-avrg(f-50,f-20)):f_2=round(f)
if(dur2=16) then f=peak_f(10300,10900):aCt3=round(a-avrg(f-10,f+10)):aRt3=round(a-avrg(f+20,f+50)):aLt3=round(a-avrg(f-50,f-20)):f_3=round(f)

;-----------------A_start_detect (an event of begining of meteor)----------------------
if( never ) then REM --- A_start_detect (an event of begin of meteor) --------------
if( A_still_detect=1 ) then A_still_detect=2:A_start_detect=1:id_met=(t1+"_"+K_station_name):id_met0=t1:sp.print("`               ^ t+"+str("ss",time))
if( A_start_detect=1 ) then A_start_detect=2:n_1=n:f_0=round(f)
if( A_start_detect=2 ) then A_start_detect=0

;------------A_end_detect (an event of detection meteor or fireball-----------                                           
if( never ) then REM --- A_end_detect (end of meteor) ---
if( timer0.expired(1) ) then A_end_detect=1:id_met2=id_met:hour_count=hour_count+1:timer4.restart(K_time_to_start_up_waterf)
if( A_end_detect=1 ) then A_end_detect=2:s1="`                 "+id_met0+" hCnt"+str(hour_count)+" nb"+str(round(n_1))+" f"+str(f_0)+" hit"+str(dur_count*K_init_cond)+" dur"+str(dur3*K_init_cond)+" mag"+str(round(aver_amp_max)))
if( A_end_detect=2 ) then A_end_detect=3:s2=" 1f"+str(f_1)+" 1L"+str(aLt1)+" 1C"+str(aCt1)+" 1R"+str(aRt1)+" 2f"+str(f_2)+" 2L"+str(aLt2)+" 2C"+str(aCt2)+" 2R"+str(aRt2)+" 3f"+str(f_3)+" 3L"+str(aLt3)+" 3C"+str(aCt3)+" 3R"+str(aRt3)
if( A_end_detect=3 ) then A_end_detect=4:
if( A_end_detect=4 ) then A_end_detect=5:sp.print(s1+s2)
if( A_end_detect=5 ) then A_end_detect=6:A_update_SDR_data_file=1:fft.export.close
if( A_end_detect=6 ) then A_end_detect=0
 
;-----------------A_update_SDR_data_file----------------------------------------
;Record to a data file (*.dat) (project Bolidozor) 
if( never ) then REM --- write data to the file .dat ---
if( A_update_SDR_data_file=1 ) then A_update_SDR_data_file=2:fopen1(K_path_data+str("YYYYMMDDhh",T_0)+"_"+K_station_name+".dat",a)
if((A_update_SDR_data_file=2)&&(bolide2=1)) then A_update_SDR_data_file=3:fw1(id_met2+"_fb;"+str(round(n_1))+";"+str(f_0)+";"+str(dur_count*K_init_cond)+";"+str(dur3*K_init_cond)+";"+str(round(aver_amp_max)) ):bolide2=0
if( A_update_SDR_data_file=2 ) then A_update_SDR_data_file=3:fw1(id_met2+"_met;"+str(round(n_1))+";"+str(f_0)+";"+str(dur_count*K_init_cond)+";"+str(dur3*K_init_cond)+";"+str(round(aver_amp_max)) )
if( A_update_SDR_data_file=3 ) then A_update_SDR_data_file=4:fp1(";"+str(f_1)+";"+str(aLt1)+";"+str(aCt1)+";"+str(aRt1)+";"+str(f_2)+";"+str(aLt2)+";"+str(aCt2)+";"+str(aRt2)+";"+str(f_3)+";"+str(aLt3)+";"+str(aCt3)+";"+str(aRt3))
if( A_update_SDR_data_file=4 ) then A_update_SDR_data_file=5:fclose1:A_end_detect_end=1
if( A_update_SDR_data_file=5 ) then A_update_SDR_data_file=0
    
;--------------A_end_detect_end-------------------------------------------------
;zametení
if( never ) then REM --- Cleaning ----------------------------------------------
if( A_end_detect_end=1 ) then A_end_detect_end=2:dur_count=0:aver_amp_1=999:aver_amp_max=999:rec.trigger=0:A_still_detect=0
if( A_end_detect_end=2 ) then A_end_detect_end=0

;-----------------A_start_detect_bolid (událost start bolidu)-------------------
if( never ) then REM --- A_start_detect_bolid (begin of fireball) --------------
if( dur_count=K_min_dur_count_bolid ) then id_met_bolide=id_met:rec.filename=K_path_audio+id_met_bolide+".wav":rec.trigger=1:bolide=1:bolide2=1

;---A_met_screenshot (událost generování screenshotu waterfallu pokud je zaznamenán meteor)---
if( never ) then REM --- A_met_screenshot (capture a screenshot with meteor) ---
if( timer4.expired(1) ) then A_met_screenshot=1
if((A_met_screenshot=1)&&(bolide=1)) then capture (K_path_scr+id_met_bolide+"_fb.jpg"):A_met_screenshot=2
if( A_met_screenshot=1 ) then capture (K_path_scr+id_met2+"_met.jpg"):A_met_screenshot=2
if( A_met_screenshot=2 ) then A_met_screenshot=0:bolide=0

;---A_no_met_screenshot (udalost generovani screenshotu waterfallu pokud neni dlouho zaznamenan meteor)---
if( never ) then REM --- A_no_met_screenshot (capture a screenshot without meteor) ---
if( timer3.expired ) then A_no_met_screenshot=1:capture (K_path_scr+t1+"_"+K_station_name+"_none.jpg")
if( A_no_met_screenshot=1 ) then A_no_met_screenshot=2:fopen1(K_path_data+str("YYYYMMDDhh",T_0)+"_"+K_station_name+".dat",a)
if( A_no_met_screenshot=2 ) then A_no_met_screenshot=3:fp1(t1+"_"+K_station_name+"_none;"+str(round(n))+";;;;;;;;;;;;;;;;")
if( A_no_met_screenshot=3 ) then A_no_met_screenshot=4:fclose1
if( A_no_met_screenshot=4 ) then A_no_met_screenshot=0:timer3.restart(K_max_time_to_scr)
    
;-A_once_hour_2s_before -
if( never ) then REM --- A_once_hour_20s_before (an event 1x per hour - 2 seconds) ---
if( val(t3,"####")=5958 ) then A_once_hour_2s_before=A_once_hour_2s_before+1
if( A_once_hour_2s_before=1 ) then A_once_hour=1
if( A_once_hour=1 ) then A_once_hour=2:sp.print("Last hour =",hour_count):hour_count=0
if( A_once_hour=2 ) then A_once_hour=0

;-A_once_hour_1s_before -
if( never ) then REM --- A_once_hour_1s_before (an event 1x per hour - 1 second) ---
if( val(t3,"####")=5959 ) then A_once_hour_2s_before=0: