Ejemplo n.º 1
0
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()
Ejemplo n.º 2
0
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()
Ejemplo n.º 3
0
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()