[params] fix subfolder
This commit is contained in:
+10
-7
@@ -374,6 +374,7 @@ class Plotter(Aggregator, BaseProcessor):
|
|||||||
for run in runs:
|
for run in runs:
|
||||||
clip = ImageSequenceClip(filenames[run], fps=movie_fps)
|
clip = ImageSequenceClip(filenames[run], fps=movie_fps)
|
||||||
movie_filename = self._find_filename(name_full, run=run, ext=".mp4")
|
movie_filename = self._find_filename(name_full, run=run, ext=".mp4")
|
||||||
|
os.makedirs(os.path.dirname(movie_filename), exist_ok=True)
|
||||||
clip.write_videofile(movie_filename)
|
clip.write_videofile(movie_filename)
|
||||||
|
|
||||||
return datafiles
|
return datafiles
|
||||||
@@ -390,6 +391,7 @@ class Plotter(Aggregator, BaseProcessor):
|
|||||||
plt.tight_layout(pad=1)
|
plt.tight_layout(pad=1)
|
||||||
|
|
||||||
if self.params.out.save:
|
if self.params.out.save:
|
||||||
|
os.makedirs(os.path.dirname(plot_filename), exist_ok=True)
|
||||||
plt.savefig(plot_filename)
|
plt.savefig(plot_filename)
|
||||||
self._log("{} plotted".format(plot_filename), "SUCCESS")
|
self._log("{} plotted".format(plot_filename), "SUCCESS")
|
||||||
else:
|
else:
|
||||||
@@ -408,8 +410,12 @@ class Plotter(Aggregator, BaseProcessor):
|
|||||||
Determine a filename based on rule name, run, output and parameters
|
Determine a filename based on rule name, run, output and parameters
|
||||||
"""
|
"""
|
||||||
tag_name = self.params.out.tag
|
tag_name = self.params.out.tag
|
||||||
|
|
||||||
|
if ext is None:
|
||||||
|
ext = self.params.out.ext
|
||||||
|
|
||||||
if self.params.out.ext_subfolder:
|
if self.params.out.ext_subfolder:
|
||||||
subfolder = "/{ext[1:]}/"
|
subfolder = f"/{ext[1:]}"
|
||||||
else:
|
else:
|
||||||
subfolder = ""
|
subfolder = ""
|
||||||
|
|
||||||
@@ -418,11 +424,11 @@ class Plotter(Aggregator, BaseProcessor):
|
|||||||
tag_name = "_" + tag_name
|
tag_name = "_" + tag_name
|
||||||
|
|
||||||
if run is not None and num is not None:
|
if run is not None and num is not None:
|
||||||
fmt = "{out}/{run}/{subfolder}{name}{tag}_{run}_{num:05}{ext}"
|
fmt = "{out}/{run}{subfolder}/{name}{tag}_{run}_{num:05}{ext}"
|
||||||
elif run is not None:
|
elif run is not None:
|
||||||
fmt = "{out}/{run}/{subfolder}{name}{tag}_{run}{ext}"
|
fmt = "{out}/{run}{subfolder}/{name}{tag}_{run}{ext}"
|
||||||
else:
|
else:
|
||||||
fmt = "{out}/{name}{subfolder}{tag}{ext}"
|
fmt = "{out}{subfolder}/{name}{tag}{ext}"
|
||||||
elif fmt is None:
|
elif fmt is None:
|
||||||
fmt = self.params.out.fmt
|
fmt = self.params.out.fmt
|
||||||
|
|
||||||
@@ -430,9 +436,6 @@ class Plotter(Aggregator, BaseProcessor):
|
|||||||
if run is not None:
|
if run is not None:
|
||||||
nml = self.study.namelist[run]
|
nml = self.study.namelist[run]
|
||||||
|
|
||||||
if ext is None:
|
|
||||||
ext = self.params.out.ext
|
|
||||||
|
|
||||||
return fmt.format(
|
return fmt.format(
|
||||||
run=run,
|
run=run,
|
||||||
name=name_full,
|
name=name_full,
|
||||||
|
|||||||
@@ -312,21 +312,21 @@ class SnapshotProcessor(HDF5Container):
|
|||||||
else:
|
else:
|
||||||
tag_name = ""
|
tag_name = ""
|
||||||
|
|
||||||
if self.params.out.subfolder_ext:
|
if self.params.out.ext_subfolder:
|
||||||
subfolder = "/h5/"
|
subfolder = "/h5"
|
||||||
else:
|
else:
|
||||||
subfolder = ""
|
subfolder = ""
|
||||||
|
|
||||||
self.filename = f"{self.path_out}/{subfolder}postproc_{tag_name}{num:05}.h5"
|
self.filename = f"{self.path_out}{subfolder}/postproc_{tag_name}{num:05}.h5"
|
||||||
self.cells_filename = f"{self.path_out}/{subfolder}cells_{tag_name}{num:05}.h5"
|
self.cells_filename = f"{self.path_out}{subfolder}/cells_{tag_name}{num:05}.h5"
|
||||||
self.parts_filename = f"{self.path_out}/{subfolder}parts_{tag_name}{num:05}.h5"
|
self.parts_filename = f"{self.path_out}{subfolder}/parts_{tag_name}{num:05}.h5"
|
||||||
self.pspec_filename = f"{self.path_out}/{subfolder}pspec_{tag_name}{num:05}.h5"
|
self.pspec_filename = f"{self.path_out}{subfolder}/pspec_{tag_name}{num:05}.h5"
|
||||||
self.filaments_filename = (
|
self.filaments_filename = (
|
||||||
f"{self.path_out}/{subfolder}filaments_{tag_name}{num:05}.h5"
|
f"{self.path_out}/{subfolder}filaments_{tag_name}{num:05}.h5"
|
||||||
)
|
)
|
||||||
|
|
||||||
if not os.path.exists(self.path_out):
|
if not os.path.exists(f"{self.path_out}{subfolder}"):
|
||||||
os.makedirs(self.path_out)
|
os.makedirs(f"{self.path_out}{subfolder}")
|
||||||
|
|
||||||
self.path = path
|
self.path = path
|
||||||
self.run = os.path.basename(path)
|
self.run = os.path.basename(path)
|
||||||
|
|||||||
+4
-2
@@ -44,9 +44,11 @@ class StudyProcessor(Aggregator, HDF5Container):
|
|||||||
tag_name = ""
|
tag_name = ""
|
||||||
|
|
||||||
if self.params.out.ext_subfolder:
|
if self.params.out.ext_subfolder:
|
||||||
self.filename = f"{self.path_out}/study{tag_name}.h5"
|
|
||||||
else:
|
|
||||||
self.filename = f"{self.path_out}/h5/study{tag_name}.h5"
|
self.filename = f"{self.path_out}/h5/study{tag_name}.h5"
|
||||||
|
else:
|
||||||
|
self.filename = f"{self.path_out}/study{tag_name}.h5"
|
||||||
|
os.makedirs(os.path.dirname(self.filename), exist_ok=True)
|
||||||
|
|
||||||
# Select runs
|
# Select runs
|
||||||
if selector is None:
|
if selector is None:
|
||||||
selector = RunSelector(
|
selector = RunSelector(
|
||||||
|
|||||||
Reference in New Issue
Block a user