Skip to content

Commit

Permalink
Pass reader to ImageConverter.to_tiledb() instead of creating custom …
Browse files Browse the repository at this point in the history
…ImageConverter subclasses
  • Loading branch information
gsakkis committed Feb 9, 2023
1 parent 228171b commit adb5c0d
Showing 1 changed file with 3 additions and 11 deletions.
14 changes: 3 additions & 11 deletions napari_tiledb_bioimg/_writer.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,14 +108,6 @@ def level_metadata(self, level: int) -> Dict[str, Any]:
return {}


class NapariSingleScaleConverter(ImageConverter):
_ImageReaderType = NapariSingleScaleDataReader


class NapariMultiScaleConverter(ImageConverter):
_ImageReaderType = NapariMultiScaleDataReader


def napari_write_image_lossless(path, data, attributes):
return _napari_write_image(path, data, lossless=True)

Expand All @@ -128,8 +120,8 @@ def _napari_write_image(path, data, lossless):
kwargs = dict(compressor=tiledb.WebpFilter(lossless=lossless))
if isinstance(data, MultiScaleData):
kwargs.update(chunked=True, max_workers=os.cpu_count())
converter = NapariMultiScaleConverter
reader = NapariMultiScaleDataReader(data)
else:
converter = NapariSingleScaleConverter
converter.to_tiledb(data, path, **kwargs)
reader = NapariSingleScaleDataReader(data)
ImageConverter.to_tiledb(reader, path, **kwargs)
return [path]

0 comments on commit adb5c0d

Please sign in to comment.