def setup_coh_inspiral(ifo_name,ifo_char,insp_job,runSplitBank,calibrated,\ runSpinChecker,chunk,dag,bank,scSpinBank,scNoSpinBank,sbOutBanks,scNodes,sb_node,usertag): # make one single inspiral job for the master chunk insp = inspiral.PTFInspiralNode(insp_job) if not opts.disable_dag_categories: insp.set_category('inspiral1') insp.set_priority(2) insp.set_user_tag(usertag) if runSpinChecker: insp.set_spin_bank(scSpinBank[bank]) insp.set_no_spin_bank(scNoSpinBank[bank]) insp.add_parent(scNodes[bank]) else: insp.set_no_spin_bank(sbOutBanks[bank]) if runSplitBank: insp.add_parent(sb_node) insp.set_seed(bank) insp.set_start(chunk.start()) insp.set_end(chunk.end()) insp.set_trig_start(chunk.trig_start()) insp.set_trig_end(chunk.trig_end()) insp.set_ifo(ifo_char) if runSplitBank: insp.set_ifo_tag("FIRST_" + str(bank)) else: insp.set_ifo_tag("FIRST") insp.set_output() insp.set_vds_group(ifo_name[0] + str(chunk.start())) if not calibrated: insp.calibration() output = insp.get_output() if opts.inspiral: dag.add_node(insp) return insp
bank.add_parent(df) dag.add_node(bank) # split the template bank up into smaller banks split = inspiral.SplitBankNode(split_job) split.set_bank(bank.get_output()) split.set_num_banks(cp.get('splitbank', 'number-of-banks')) split.add_parent(bank) dag.add_node(split) # create the inspiral jobs to do the analysis sub_insp = [] for i, subbank in enumerate(split.get_output()): if doCohPTF: insp = inspiral.PTFInspiralNode(insp_job) else: insp = inspiral.InspiralNode(insp_job) insp.set_start(seg[0]) insp.set_end(seg[1]) insp.set_trig_start('trig-start-time') insp.set_trig_end('trig-end-time') insp.set_ifo(ifo) if doCohPTF: insp.set_no_spin_bank(subbank) insp.set_ifo_tag('FIRST') else: insp.set_cache(df.get_output()) insp.set_bank(subbank)