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

npr module created

Browse source
This commit is contained in:
Fabian Joswig 2021-10-19 16:15:00 +01:00
parent 01d06506ff
commit 5e98c89411
3 changed files with 23 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
pyerrors/__init__.py
View file

@ -1,7 +1,8 @@
from .pyerrors import * from .pyerrors import *
from . import correlators
from . import fits from . import fits
from . import linalg from . import linalg
from . import misc from . import misc
from . import mpm from . import mpm
from . import correlators from . import npr
from . import roots from . import roots

10
pyerrors/input/hadrons.py
View file

@ -6,6 +6,7 @@ import h5py
import numpy as np import numpy as np
from ..pyerrors import Obs, CObs from ..pyerrors import Obs, CObs
from ..correlators import Corr from ..correlators import Corr
from ..npr import Propagator
def _get_files(path, filestem): def _get_files(path, filestem):
@ -69,7 +70,7 @@ def read_meson_hd5(path, filestem, ens_id, meson='meson_0', tree='meson'):
return Corr(l_obs) return Corr(l_obs)
def read_ExternalLeg_hd5(path, filestem, ens_id): def read_ExternalLeg_hd5(path, filestem, ens_id, order='C'):
"""Read hadrons ExternalLeg hdf5 file and output an array of CObs """Read hadrons ExternalLeg hdf5 file and output an array of CObs
Parameters Parameters
@ -77,6 +78,9 @@ def read_ExternalLeg_hd5(path, filestem, ens_id):
path -- path to the files to read path -- path to the files to read
filestem -- namestem of the files to read filestem -- namestem of the files to read
ens_id -- name of the ensemble, required for internal bookkeeping ens_id -- name of the ensemble, required for internal bookkeeping
order -- order in which the array is to be reshaped,
'C' for the last index changing fastest,
'F' for the first index changing fastest.
""" """
files = _get_files(path, filestem) files = _get_files(path, filestem)
@ -88,6 +92,8 @@ def read_ExternalLeg_hd5(path, filestem, ens_id):
corr_data.append(raw_data) corr_data.append(raw_data)
corr_data = np.array(corr_data) corr_data = np.array(corr_data)
mom = np.array(str(file['ExternalLeg/info'].attrs['pIn'])[3:-2].strip().split(' '), dtype=int)
rolled_array = np.rollaxis(corr_data, 0, 5) rolled_array = np.rollaxis(corr_data, 0, 5)
matrix = np.empty((rolled_array.shape[:-1]), dtype=object) matrix = np.empty((rolled_array.shape[:-1]), dtype=object)
@ -97,4 +103,4 @@ def read_ExternalLeg_hd5(path, filestem, ens_id):
matrix[si, sj, ci, cj] = CObs(real, imag) matrix[si, sj, ci, cj] = CObs(real, imag)
matrix[si, sj, ci, cj].gamma_method() matrix[si, sj, ci, cj].gamma_method()
return matrix return Propagator(matrix.reshape((12,12), order=order), mom)

13
pyerrors/npr.py Normal file
View file

@ -0,0 +1,13 @@
import numpy as np
class Propagator(np.ndarray):
def __new__(cls, input_array, mom=None):
obj = np.asarray(input_array).view(cls)
obj.mom = mom
return obj
def __array_finalize__(self, obj):
if obj is None: return
self.mom = getattr(obj, 'mom', None)