import a calcbstraps = 1 destdir = "/hiskp2/oehm/analysis/baryons/A40.24/" correlator = sys.argv[1] if calcbstraps == 1: x = a.readcor("/hiskp2/oehm/baryon/contractions/A40.24/correlators/"+correlator,48) x = x[0:50,:] x_bstraps = a.boot(x,1500) np.save(os.path.join(destdir,correlator+"_x_bstraps50.npy"),x_bstraps) else: x_bstraps = np.load(os.path.join(destdir,correlator+"_x_bstraps50.npy")) if correlator == "corr1": for t in range(0,48): x_bstraps[:,t] = -x_bstraps[:,t] if correlator == "corr2": for t in range(24,48): x_bstraps[:,t] = -x_bstraps[:,t] _effmass, effmass_error, effmass_nans = a.effmass(x_bstraps)
import a sigma = int(sys.argv[1]) print "sigma "+str(sigma) a.ensure_dir("/hiskp2/oehm/correlators/momentum/A40.24/O012/analysis/sigma"+str(sigma)+"/") numBootstraps = 1500 O012 = a.readcor("/hiskp2/oehm/correlators/momentum/A40.24/O012/sigma"+str(sigma)+"/O012_sigma"+str(sigma),48) xn = a.readcor("/hiskp2/oehm/correlators/momentum/A40.24/x/sigma"+str(sigma)+"/x"+str(sigma),48) O012 = a.asym(O012) xn = a.sym(xn) O012_bstraps = a.boot(O012,numBootstraps) O012_bstraps_mean = np.mean(O012_bstraps,axis=0) xn_bstraps = a.boot(xn,numBootstraps) xn_bstraps_mean = np.mean(xn_bstraps,axis=0) xn_result = np.zeros((21,numBootstraps,4)) effmass, effmass_error, effmass_nans = a.effmass(xn_bstraps) print effmass xnfunc = lambda p,x: p[0]*(np.exp(-x*p[1])+np.exp(-(48-x)*p[1])) O012_func = lambda p,x: p[0] tt = [] for t in range(0,25):
parentdir = os.path.dirname(currentdir) sys.path.insert(0,parentdir) import a import numpy as np numBoots = 200 #O0123_1 = a.readcor("/hiskp2/oehm/correlators/momentum3/A80/O0123/sigma1/O0123_sigma1",48) #O0123_2 = a.readcor("/hiskp2/oehm/correlators/momentum3/A80/O0123/sigma2/O0123_sigma2",48) #O0123_3 = a.readcor("/hiskp2/oehm/correlators/momentum3/A80/O0123/sigma3/O0123_sigma3",48) #O0123_4 = a.readcor("/hiskp2/oehm/correlators/momentum3/A80/O0123/sigma4/O0123_sigma4",48) #O0123_5 = a.readcor("/hiskp2/oehm/correlators/momentum3/A80/O0123/sigma5/O0123_sigma5",48) O0123_5t8 = a.readcor("/hiskp2/oehm/correlators/momentum3/A80/O0123/sigma5-8/O0123_sigma5-8",48) #O0123_1_bstraps = a.boot(O0123_1,numBoots) #O0123_2_bstraps = a.boot(O0123_2,numBoots) #O0123_3_bstraps = a.boot(O0123_3,numBoots) #O0123_4_bstraps = a.boot(O0123_4,numBoots) #O0123_5_bstraps = a.boot(O0123_5,numBoots) O0123_5t8_bstraps = a.boot(O0123_5t8,numBoots) #a.nsettofile2(O0123_1_bstraps[0,:],np.std(O0123_1_bstraps,axis=0),"/hiskp2/oehm/correlators/momentum3/A80/O0123/analysis/O0123_1") #a.nsettofile2(O0123_2_bstraps[0,:],np.std(O0123_2_bstraps,axis=0),"/hiskp2/oehm/correlators/momentum3/A80/O0123/analysis/O0123_2") #a.nsettofile2(O0123_3_bstraps[0,:],np.std(O0123_3_bstraps,axis=0),"/hiskp2/oehm/correlators/momentum3/A80/O0123/analysis/O0123_3") #a.nsettofile2(O0123_4_bstraps[0,:],np.std(O0123_4_bstraps,axis=0),"/hiskp2/oehm/correlators/momentum3/A80/O0123/analysis/O0123_4") #a.nsettofile2(O0123_5_bstraps[0,:],np.std(O0123_5_bstraps,axis=0),"/hiskp2/oehm/correlators/momentum3/A80/O0123/analysis/O0123_5") a.nsettofile2(O0123_5t8_bstraps[0,:],np.std(O0123_5t8_bstraps,axis=0),"/hiskp2/oehm/correlators/momentum3/A80/O0123/analysis/O0123_5t8")
if calcbstraps == 1: x1 = a.readcor("/hiskp2/oehm/baryon/contractions/A40.24/correlators/"+correlator1,T) x2 = a.readcor("/hiskp2/oehm/baryon/contractions/A40.24/correlators/"+correlator2,T) y1 = np.zeros(x1.shape[0],T/2+1) y2 = np.zeros(x1.shape[0],T/2+1) for i in range(0,x.shape[0]): y1[i,0] = x1[i,0] y2[i,0] = x2[i,0] for t in range(1,T/2+1): y1[i,t] = (x1[i,t]+x2[i,T-t])/2 y2[i,t] = (x2[i,t]+x1[i,T-t])/2 y1_bstraps = a.boot(y1,1500) y2_bstraps = a.boot(y2,1500) np.save(os.path.join(destdir,correlator1+"_"+correlator2+"_y1_bstraps.npy"),y1_bstraps) np.save(os.path.join(destdir,correlator1+"_"+correlator2+"_y2_bstraps.npy"),y2_bstraps) else: y1_bstraps = np.load(os.path.join(destdir,correlator1+"_"+correlator2+"_y1_bstraps.npy")) y2_bstraps = np.load(os.path.join(destdir,correlator1+"_"+correlator2+"_y2_bstraps.npy")) #fit 2pt functions to the two correlation functions of the partiy partners def fit2pttoy(xn_bstraps): xnfunc = lambda p,x: p[0]*(np.exp(-x*p[1])+np.exp(-(T-x)*p[1]))
else: data = a.sym(data) errors = a.autocorr_bstrap(data,int(taskr.rparam("start")),int(taskr.rparam("end")),numBstraps) a.plotautocorr_bstrap(errors,os.path.join(destdir,"autocorr_bstrap.pdf")) elif task == "mwerr": data = a.readcor(runr.getopcf(op,sigma),latT) if runr.istrue("wpe"): data = a.sym_wope(data) data = data[:,1:len(data[0])-1] else: data = a.sym(data) databstraps = a.boot(data,numBstraps) a.nsettofile2(databstraps[0],np.std(databstraps,axis=0),os.path.join(destdir,op+"_mean_werr")) elif task == "fitplatval2": doasym = 0 if taskr.istrue("asym"): doasym = 1 wpe = 0 if runr.istrue("wpe"): wpe = 1 print "wpe: "+str(wpe) a.fitplatval2(runr.getopcf(op,sigma),destdir,latT,numBstraps,doasym,wpe) elif task == "plotbstraphists":
numBootstraps = 1500 lattice = sys.argv[1] print "destdir: "+"/hiskp2/oehm/averx/analysis/"+lattice+"/" T = int(sys.argv[2]) print "T: "+str(T) ensure_dir("/hiskp2/oehm/averx/analysis/"+lattice+"/") O44 = a.readcor("/hiskp2/oehm/averx/"+lattice+"/maxmomf2e",T) xn = a.readcor("/hiskp2/oehm/averx/"+lattice+"/maxppcor",T) O44 = a.sym_wope(O44) xn = a.sym(xn) O44_bstraps = a.boot(O44,numBootstraps) O44_bstraps_mean = np.mean(O44_bstraps,axis=0) xn_bstraps = a.boot(xn,numBootstraps) xn_bstraps_mean = np.mean(xn_bstraps,axis=0) xn_result = np.zeros((T/2+1-4,numBootstraps,4)) effmass, effmass_error, effmass_nans = a.effmass(xn_bstraps) print effmass xnfunc = lambda p,x: p[0]*(np.exp(-x*p[1])+np.exp(-(T-x)*p[1])) O44_func = lambda p,x: p[0] tt = [] for t in range(0,T/2+1):