noe/pipeline
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[study] some renaming

Browse Source
This commit is contained in:
Noe Brucy
2021-07-22 23:34:28 +02:00
parent c1960b1742
commit 4d284c271a
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
studyprocessor.py
+13 -13
View File
@@ -79,7 +79,7 @@ class StudyProcessor(Aggregator, HDF5Container):
self.info = selector.info[run0][self.nums[run0][0]]
self.namelist = selector.namelist
# log info
self.log_id = "[comp {}] ".format(self.params.out.tag)
self.log_id = "[study {}] ".format(self.params.out.tag)
# Define rules
self.def_rules()
@@ -118,7 +118,7 @@ class StudyProcessor(Aggregator, HDF5Container):
series[run] = np.array(series[run])
return series
def _comp(self, getter, use_num=True):
def _compare(self, getter, use_num=True):
prop = []
for i, run in enumerate(self.runs):
if use_num:
@@ -193,33 +193,33 @@ class StudyProcessor(Aggregator, HDF5Container):
}
def get_attr(self, attr_name, run, num, node_name="/", arg=None):
pp = self.snaps[run][num]
snap = self.snaps[run][num]
if arg is not None:
node_name = node_name + "_" + str(arg)
return pp.get_attribute(node_name, attr_name)
return snap.get_attribute(node_name, attr_name)
def get_snap_value(self, name, run, num, arg=None):
pp = self.snaps[run][num]
snap = self.snaps[run][num]
if arg is not None:
name = name + "_" + str(arg)
return pp.get_value(name)
return snap.get_value(name)
def get_global(self, node_name, run, num, arg=None, unload_cells=False):
if arg is not None:
node_name = node_name + "_" + str(arg)
pp = self.snaps[run][num]
snap = self.snaps[run][num]
if unload_cells:
pp.unload_cells()
value = pp.get_value(node_name)
snap.unload_cells()
value = snap.get_value(node_name)
return value
def get_nml(self, nml_key, run):
return self.namelist[run][nml_key]
def get_pdf_slope(self, name, run, num):
pp = self.snaps[run][num]
pp.process(["fit_pdf_" + name], ["z"], overwrite=self.overwrite_dep)
slope = pp.get_attribute("/hist/pdf_" + name + "_z", "slope")
snap = self.snaps[run][num]
snap.process(["fit_pdf_" + name], ["z"], overwrite=self.overwrite_dep)
slope = snap.get_attribute("/hist/pdf_" + name + "_z", "slope")
return slope
def _extract_sinks_from_log(self, series, log_filename, run):
@@ -708,7 +708,7 @@ class StudyProcessor(Aggregator, HDF5Container):
# namelist
"nml": Rule(
self,
lambda nml_key: self._comp(
lambda nml_key: self._compare(
partial(self.get_nml, nml_key), use_num=False
),
group="/comp",