Skip to content
Snippets Groups Projects
Commit 99aeda1c authored by Christopher Randolph Rhodes's avatar Christopher Randolph Rhodes
Browse files

Standardize that export record is a dict of pathlib.Path objects

parent 8001be75
No related branches found
No related tags found
No related merge requests found
......@@ -474,7 +474,15 @@ class RoiSet(object):
dfe['patch'] = self._df.apply(lambda r: _make_patch(r), axis=1)
return dfe
def run_exports(self, where, channel, prefix, params: RoiSetExportParams):
def run_exports(self, where: Path, channel, prefix, params: RoiSetExportParams) -> dict:
"""
Export various representations of ROIs, e.g. patches, annotated stacks, and object maps.
:param where: path of directory in which to write all export products
:param channel: color channel of products to export
:param prefix: prefix of the name of each product's file or subfolder
:param params: RoiSetExportParams object describing which products to export and with which parameters
:return: dict of Path objects describing the location of single-file export products
"""
record = {}
if not self.count:
return
......@@ -506,11 +514,10 @@ class RoiSet(object):
if k == 'annotated_zstacks':
self.export_annotated_zstack(subdir, prefix=pr, **kp)
if k == 'object_classes':
# record[k] = {}
for kc, acc in self.object_class_maps.items():
fp = subdir / kc / (pr + '.tif')
write_accessor_data_to_file(fp, acc)
record[f'{k}_{kc}'] = fp.__str__()
record[f'{k}_{kc}'] = fp
if k == 'dataframe':
dfpa = subdir / (pr + '.csv')
dfpa.parent.mkdir(parents=True, exist_ok=True)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment