lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-05-14 19:43:41 +02:00
Code Issues Projects Releases Packages Wiki Activity

prange set to default value None

Browse source
This commit is contained in:
Fabian Joswig 2021-10-11 09:37:44 +01:00
parent 4d746802d2
commit b14f430f18
1 changed files with 29 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

86
pyerrors/correlators.py
View file

@ -24,7 +24,7 @@ class Corr:
""" """
def __init__(self, data_input, padding_front=0, padding_back=0,prange=None): def __init__(self, data_input, padding_front=0, padding_back=0, prange=None):
#All data_input should be a list of things at different timeslices. This needs to be verified #All data_input should be a list of things at different timeslices. This needs to be verified
if not (isinstance(data_input, list)): if not (isinstance(data_input, list)):
@ -57,16 +57,13 @@ class Corr:
self.content = [None] * padding_front + self.content + [None] * padding_back self.content = [None] * padding_front + self.content + [None] * padding_back
self.T = len(self.content) #for convenience: will be used a lot self.T = len(self.content) #for convenience: will be used a lot
#The attribute "range" [start,end] marks a range of two timeslices.
#The attribute "range" [start,end] marks a range of two timeslices.
#This is useful for keeping track of plateaus and fitranges. #This is useful for keeping track of plateaus and fitranges.
#The range can be inherited from other Corrs, if the operation should not alter a chosen range eg. multiplication with a constant. #The range can be inherited from other Corrs, if the operation should not alter a chosen range eg. multiplication with a constant.
if not prange is None: self.prange = prange
self.prange=prange
self.gamma_method() self.gamma_method()
def gamma_method(self): def gamma_method(self):
for item in self.content: for item in self.content:
if not(item is None): if not(item is None):
@ -143,8 +140,7 @@ class Corr:
newcontent.append(0.5 * (self.content[t] + self.content[self.T - t])) newcontent.append(0.5 * (self.content[t] + self.content[self.T - t]))
if(all([x is None for x in newcontent])): if(all([x is None for x in newcontent])):
raise Exception("Corr could not be symmetrized: No redundant values") raise Exception("Corr could not be symmetrized: No redundant values")
return Corr(newcontent, prange=self.prange)
return Corr(newcontent,prange= self.prange if hasattr(self,"prange") else None)
def anti_symmetric(self): def anti_symmetric(self):
@ -159,7 +155,7 @@ class Corr:
newcontent.append(0.5 * (self.content[t] - self.content[self.T - t])) newcontent.append(0.5 * (self.content[t] - self.content[self.T - t]))
if(all([x is None for x in newcontent])): if(all([x is None for x in newcontent])):
raise Exception("Corr could not be symmetrized: No redundant values") raise Exception("Corr could not be symmetrized: No redundant values")
return Corr(newcontent,prange= self.prange if hasattr(self,"prange") else None) return Corr(newcontent, prange=self.prange)
#This method will symmetrice the matrices and therefore make them positive definit. #This method will symmetrice the matrices and therefore make them positive definit.
@ -192,12 +188,12 @@ class Corr:
G0=G.content[t0] G0=G.content[t0]
L = mat_mat_op(anp.linalg.cholesky, G0) L = mat_mat_op(anp.linalg.cholesky, G0)
Li = mat_mat_op(anp.linalg.inv, L) Li = mat_mat_op(anp.linalg.inv, L)
LT=L.T LT=L.T
LTi=mat_mat_op(anp.linalg.inv, LT) LTi=mat_mat_op(anp.linalg.inv, LT)
newcontent=[] newcontent=[]
for t in range(self.T): for t in range(self.T):
Gt=G.content[t] Gt=G.content[t]
M=Li@Gt@LTi M=Li@Gt@LTi
eigenvalues = eigh(M)[0] eigenvalues = eigh(M)[0]
#print(eigenvalues) #print(eigenvalues)
eigenvalue=eigenvalues[-state] eigenvalue=eigenvalues[-state]
@ -205,14 +201,6 @@ class Corr:
return Corr(newcontent) return Corr(newcontent)
def roll(self, dt): def roll(self, dt):
return Corr(list(np.roll(np.array(self.content, dtype=object), dt))) return Corr(list(np.roll(np.array(self.content, dtype=object), dt)))
@ -307,16 +295,16 @@ class Corr:
raise Exception("Correlator must be projected before fitting") raise Exception("Correlator must be projected before fitting")
#The default behaviour is: #The default behaviour is:
#1 use explicit fitrange #1 use explicit fitrange
# if none is provided, use the range of the corr # if none is provided, use the range of the corr
# if this is also not set, use the whole length of the corr (This could come with a warning!) # if this is also not set, use the whole length of the corr (This could come with a warning!)
if fitrange is None: if fitrange is None:
if hasattr(self,"prange"): if self.prange:
fitrange=self.prange fitrange = self.prange
else: else:
fitrange=[0, self.T] fitrange = [0, self.T]
xs = [x for x in range(fitrange[0], fitrange[1]) if not self.content[x] is None] xs = [x for x in range(fitrange[0], fitrange[1]) if not self.content[x] is None]
ys = [self.content[x][0] for x in range(fitrange[0], fitrange[1]) if not self.content[x] is None] ys = [self.content[x][0] for x in range(fitrange[0], fitrange[1]) if not self.content[x] is None]
@ -329,12 +317,13 @@ class Corr:
raise Exception('Unexpected fit result.') raise Exception('Unexpected fit result.')
return result return result
#we want to quickly get a plateau #we want to quickly get a plateau
def plateau(self, plateau_range=None, method="fit"): def plateau(self, plateau_range=None, method="fit"):
if not plateau_range: if not plateau_range:
if hasattr(self,"prange"): if self.prange:
plateau_range=self.prange plateau_range = self.prange
else: else:
raise Exception("no plateau range provided") raise Exception("no plateau range provided")
if self.N != 1: if self.N != 1:
raise Exception("Correlator must be projected before getting a plateau.") raise Exception("Correlator must be projected before getting a plateau.")
@ -350,11 +339,10 @@ class Corr:
return returnvalue return returnvalue
else: else:
raise Exception("Unsupported plateau method: " + method) raise Exception("Unsupported plateau method: " + method)
def set_prange(self, prange):
def set_prange(self,prange):
if not len(prange)==2: if not len(prange)==2:
raise Exception("range must be a list or array with two values") raise Exception("range must be a list or array with two values")
if not ((isinstance(prange[0],int)) and (isinstance(prange[1],int))): if not ((isinstance(prange[0],int)) and (isinstance(prange[1],int))):
@ -362,18 +350,10 @@ class Corr:
if not (0<=prange[0]<=self.T and 0<=prange[1]<=self.T and prange[0]<prange[1] ): if not (0<=prange[0]<=self.T and 0<=prange[1]<=self.T and prange[0]<prange[1] ):
raise Exception("start and end point must define a range in the interval 0,T") raise Exception("start and end point must define a range in the interval 0,T")
self.prange=prange self.prange = prange
return return
# plotting function to view Correlator
#quick and dirty plotting function to view Correlator inside Jupyter
#If one would not want to import pyplot, this could easily be replaced by a call to pe.plot_corrs
#This might be a bit more flexible later
def show(self, x_range=None, comp=None, logscale=False, plateau=None, fit_res=None, save=None, ylabel=None): def show(self, x_range=None, comp=None, logscale=False, plateau=None, fit_res=None, save=None, ylabel=None):
"""Plots the correlator, uses tag as label if available. """Plots the correlator, uses tag as label if available.
@ -418,9 +398,9 @@ class Corr:
ax1.axhspan(plateau.value - plateau.dvalue, plateau.value + plateau.dvalue, alpha=0.25, color=plt.rcParams['text.color'], ls='-') ax1.axhspan(plateau.value - plateau.dvalue, plateau.value + plateau.dvalue, alpha=0.25, color=plt.rcParams['text.color'], ls='-')
else: else:
raise Exception('plateau must be an Obs') raise Exception('plateau must be an Obs')
if hasattr(self,"prange"): if self.prange:
ax1.axvline(self.prange[0],0,1) ax1.axvline(self.prange[0], 0, 1, ls='-', marker=',')
ax1.axvline(self.prange[1],0,1) ax1.axvline(self.prange[1], 0, 1, ls='-', marker=',')
if fit_res: if fit_res:
x_samples = np.arange(x_range[0], x_range[1] + 1, 0.05) x_samples = np.arange(x_range[0], x_range[1] + 1, 0.05)
@ -498,7 +478,7 @@ class Corr:
newcontent.append(None) newcontent.append(None)
else: else:
newcontent.append(self.content[t]+y) newcontent.append(self.content[t]+y)
return Corr(newcontent,prange= self.prange if hasattr(self,"prange") else None) return Corr(newcontent, prange=self.prange)
else: else:
raise TypeError("Corr + wrong type") raise TypeError("Corr + wrong type")
@ -521,7 +501,7 @@ class Corr:
newcontent.append(None) newcontent.append(None)
else: else:
newcontent.append(self.content[t]*y) newcontent.append(self.content[t]*y)
return Corr(newcontent,prange= self.prange if hasattr(self,"prange") else None) return Corr(newcontent, prange=self.prange)
else: else:
raise TypeError("Corr * wrong type") raise TypeError("Corr * wrong type")
@ -559,11 +539,7 @@ class Corr:
newcontent.append(None) newcontent.append(None)
else: else:
newcontent.append(self.content[t]/y) newcontent.append(self.content[t]/y)
if hasattr(self,"prange"): return Corr(newcontent, prange=self.prange)
newrange=self.prange
else:
newrange=None
return Corr(newcontent,prange=newrange)
elif isinstance(y, int) or isinstance(y,float): elif isinstance(y, int) or isinstance(y,float):
if y==0: if y==0:
@ -580,11 +556,7 @@ class Corr:
def __neg__(self): def __neg__(self):
newcontent=[None if (item is None) else -1.*item for item in self.content] newcontent=[None if (item is None) else -1.*item for item in self.content]
if hasattr(self,"prange"): return Corr(newcontent,prange=self.prange)
newrange=self.prange
else:
newrange=None
return Corr(newcontent,prange=newrange)
def __sub__(self,y): def __sub__(self,y):
return self +(-y) return self +(-y)
@ -592,7 +564,7 @@ class Corr:
def __pow__(self, y): def __pow__(self, y):
if isinstance(y, Obs) or isinstance(y,int) or isinstance(y,float): if isinstance(y, Obs) or isinstance(y,int) or isinstance(y,float):
newcontent=[None if (item is None) else item**y for item in self.content] newcontent=[None if (item is None) else item**y for item in self.content]
return Corr(newcontent,prange= self.prange if hasattr(self,"prange") else None) return Corr(newcontent,prange=self.prange)
else: else:
raise TypeError("type of exponent not supported") raise TypeError("type of exponent not supported")