def candplots(fil_file,source_name,snr_cut,filter_cut,maxCandSec,noplot,minMem,kill_chans,kill_time_range,nogpu,gcands,fl,fh,tint,Ttot,nchan): if(nogpu is not True): #os.chdir(basedir) #os.system("cd %s" % (basedir)) #print "Inside : %s" % (basedir) os.system("rm *_all.cand") os.system("rm *.ar *.norm") os.system("coincidencer *.cand") os.system("trans_gen_overview.py -cands_file *_all.cand") os.system("mv overview_1024x768.tmp.png %s.overview.png" % (source_name)) os.system("frb_detector_bl.py -gdm 6 -cands_file *_all.cand -filter_cut %d -snr_cut %f -max_cands_per_sec %f -min_members_cut %f -verbose" % (filter_cut,snr_cut,maxCandSec,minMem)) os.system("frb_detector_bl.py -gdm 6 -cands_file *_all.cand -filter_cut %d -snr_cut %f -max_cands_per_sec %f -min_members_cut %f > FRBcand" % (filter_cut,snr_cut,maxCandSec,minMem)) if(os.stat("FRBcand").st_size is not 0): frb_cands = np.loadtxt("FRBcand",dtype={'names': ('snr','time','samp_idx','dm','filter','prim_beam'),'formats': ('f4', 'f4', 'i4','f4','i4','i4')}) else: print "No candidate found" return #print frb_cands['time'],frb_cands['dm'] else: if(gcands is not ""): os.system("rm *.ar *.norm") dt = np.dtype(dtype={'names': ('snr','time','samp_idx','dm','filter','prim_beam'),'formats': ('f4', 'f4', 'i4','f4','i4','i4')}) frb_cands = np.zeros(len(gcands),dt) for i,dd in enumerate(gcands): frb_cands[i] = np.array([(dd.sigma,dd.time,dd.sample,dd.dm,dd.dfact,0)][0],dt) else: print "No candidate found" return extractPlotCand(fil_file,frb_cands,noplot,fl,fh,tint,Ttot,kill_time_range,kill_chans,source_name,nchan)
def candplots(fil_file, source_name, snr_cut, filter_cut, maxCandSec, noplot, minMem, kill_chans, kill_time_range, nogpu, gcands, fl, fh, tint, Ttot, nchan, mask_file, smooth, zerodm, csv_file, ml_model): if (nogpu is not True): #os.chdir(basedir) #os.system("cd %s" % (basedir)) #print "Inside : %s" % (basedir) os.system("rm *_all.cand") os.system("rm *.ar *.norm") run_in_env("coincidencer *.cand") os.system('pwd') os.system("trans_gen_overview.py -cands_file *_all.cand") os.system("mv overview_1024x768.tmp.png %s.overview.png" % (source_name)) os.system( "%s/frb_detector_bl.py -gdm 6 -cands_file *_all.cand -filter_cut %d -snr_cut %f -max_cands_per_sec %f -min_members_cut %f -verbose" % (SPANDAK_PATH, filter_cut, snr_cut, maxCandSec, minMem)) os.system( "%s/frb_detector_bl.py -gdm 6 -cands_file *_all.cand -filter_cut %d -snr_cut %f -max_cands_per_sec %f -min_members_cut %f > FRBcand" % (SPANDAK_PATH, filter_cut, snr_cut, maxCandSec, minMem)) if ml_model: print("ML model given") FRBcand = os.path.abspath("FRBcand") os.system( "python /home/vgajjar/hey-aliens/simulateFRBclassification/predict.py %s %s %s" % (ml_model, fil_file, FRBcand)) if (os.stat("FRBcand").st_size is not 0): if ml_model and os.stat("FRBcand_prob.txt").st_size is not 0: frb_cands = np.loadtxt("FRBcand_prob.txt", dtype={ 'names': ('snr', 'time', 'samp_idx', 'dm', 'filter', 'prim_beam', 'FRBprob'), 'formats': ('f4', 'f4', 'i4', 'f4', 'i4', 'i4', 'f4') }) else: frb_cands = np.loadtxt("FRBcand", dtype={ 'names': ('snr', 'time', 'samp_idx', 'dm', 'filter', 'prim_beam'), 'formats': ('f4', 'f4', 'i4', 'f4', 'i4', 'i4') }) else: print("No candidate found") return #print frb_cands['time'],frb_cands['dm'] else: if (gcands is not ""): os.system("rm *.ar *.norm") dt = np.dtype( dtype={ 'names': ('snr', 'time', 'samp_idx', 'dm', 'filter', 'prim_beam'), 'formats': ('f4', 'f4', 'i4', 'f4', 'i4', 'i4') }) frb_cands = np.zeros(len(gcands), dt) for i, dd in enumerate(gcands): frb_cands[i] = np.array([(dd.sigma, dd.time, dd.sample, dd.dm, dd.dfact, 0)][0], dt) else: print("No candidate found") return extractPlotCand(fil_file, frb_cands, noplot, fl, fh, tint, Ttot, kill_time_range, kill_chans, source_name, nchan, mask_file, smooth, zerodm, csv_file)