diff --git a/igneous/task_creation/mesh.py b/igneous/task_creation/mesh.py index 3856765b..6dfdc4e2 100644 --- a/igneous/task_creation/mesh.py +++ b/igneous/task_creation/mesh.py @@ -161,7 +161,8 @@ def create_meshing_tasks( mesh_dir=None, cdn_cache=False, dust_threshold=None, object_ids=None, progress=False, fill_missing=False, encoding='precomputed', spatial_index=True, sharded=False, - compress='gzip', closed_dataset_edges=True, dust_global=False + compress='gzip', closed_dataset_edges=True, dust_global=False, + remap_table=None, ): shape = Vec(*shape) @@ -208,6 +209,7 @@ def task(self, shape, offset): sharded=sharded, compress=compress, closed_dataset_edges=closed_dataset_edges, + remap_table=remap_table, ) def on_finish(self): @@ -230,6 +232,7 @@ def on_finish(self): 'compress': compress, 'closed_dataset_edges': closed_dataset_edges, 'dust_global': bool(dust_global), + 'remap_table': remap_table, }, 'by': operator_contact(), 'date': strftime('%Y-%m-%d %H:%M %Z'), diff --git a/igneous/tasks/mesh/mesh.py b/igneous/tasks/mesh/mesh.py index 74fc1014..dced51e6 100644 --- a/igneous/tasks/mesh/mesh.py +++ b/igneous/tasks/mesh/mesh.py @@ -291,7 +291,8 @@ def _remap(self, data): } remap = self.options['remap_table'] - remap[0] = 0 + if 0 not in remap: + remap[0] = 0 data = fastremap.mask_except(data, list(remap.keys()), in_place=True) return fastremap.remap(data, remap, in_place=True)