def main(): basename = sys.argv[1] commonDict,analysisDict,likelihoodDict,plotDict,curveDict = readConfig(basename) cwd = os.getcwd() dirs = [dirname for dirname in os.listdir(cwd) if (os.path.isdir(os.path.join(cwd,dirname)) and "quickCurve" in dirname)] params = zip(dirs,[curveDict['model']]*len(dirs),[commonDict['irfs']]*len(dirs),[basename]*len(dirs)) pool = Pool(processes=int(commonDict['multicore'])) pool.map(rundiffResps,params) pool.close()
def main(): basename = sys.argv[1] START_BIN = int(sys.argv[2]) STOP_BIN = int(sys.argv[3]) commonDict,analysisDict,likeDict,plotDict,curveDict = readConfig(basename) cwd = os.getcwd() dirs = [dirname for dirname in os.listdir(cwd) if (os.path.isdir(os.path.join(cwd,dirname)) \ and "quickCurve" in dirname)][START_BIN:STOP_BIN] f = open('lc_results_' + str(START_BIN) + '-' + str(STOP_BIN-1) + '.txt','w') for directory in dirs: f.write(runFit(basename,directory,curveDict['model'],commonDict['irfs'],likeDict['sourcename'], \ analysisDict['emin'],analysisDict['emax']) + '\n') f.close()
def main(): basename = sys.argv[1] commonDict,analysisDict,likeDict,plotDict,curveDict = readConfig(basename) cwd = os.getcwd() dirs = [dirname for dirname in os.listdir(cwd) if (os.path.isdir(os.path.join(cwd,dirname)) \ and "quickCurve" in dirname)] bin_nums = range(len(dirs)) chunk_size = int(math.ceil(len(dirs)/float(commonDict['multicore']))) processes = [] for chunk in splitList(bin_nums,chunk_size): processes.append(subprocess.Popen(['python','qC_fit.py',basename,str(chunk[0]),str(chunk[-1]+1)])) for p in processes: p.wait() writeResults()