From 1022d0735e66e6ff88bb1ba8bb085050ba937eab Mon Sep 17 00:00:00 2001 From: Andrey Aristov <aaristov@pasteur.fr> Date: Sun, 29 May 2022 13:04:37 +0200 Subject: [PATCH] combine zarrs --- combine.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 combine.py diff --git a/combine.py b/combine.py new file mode 100644 index 0000000..753b50b --- /dev/null +++ b/combine.py @@ -0,0 +1,25 @@ +import dask.array as da +import os +from zarr_tools.convert import to_zarr +import fire + +def combine(BF_zarr_2D_path:str, TRITC_zarr_3D_path:str, out_zarr_path:str): + bd2d = da.from_zarr(os.path.join(BF_zarr_2D_path,'0/')) + print('Opened BF:', bd2d) + fd3d = da.from_zarr(os.path.join(TRITC_zarr_3D_path,'0/')) + print('Opened TRITC:', fd3d) + fd2d = fd3d.max(axis=1) + bd2d = da.stack([bd2d, fd2d], axis=1) + print('Resulting stack:', bd2d) + to_zarr( + bd2d, + path=out_zarr_path, + steps=4, + name=['BF','TRITC'], + colormap=['gray','orange'], + lut=((1000,30000),(440, 600)), + ) + +# combine(snakemake.input[0], snakemake.input[1], snakemake.output[0]) +if __name__ == "__main__": + fire.Fire(combine) \ No newline at end of file -- GitLab