From 51d11b0fccde1c73defeac487e24619b56391eec Mon Sep 17 00:00:00 2001
From: Sebastien Jourdain <sebastien.jourdain@kitware.com>
Date: Wed, 2 Oct 2024 09:00:27 -0600
Subject: [PATCH] python pickle registration with override
---
Wrapping/Python/vtkmodules/util/data_model.py | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/Wrapping/Python/vtkmodules/util/data_model.py b/Wrapping/Python/vtkmodules/util/data_model.py
index 6014b4036bc..1758b20c6a5 100644
--- a/Wrapping/Python/vtkmodules/util/data_model.py
+++ b/Wrapping/Python/vtkmodules/util/data_model.py
@@ -503,3 +503,15 @@ class CompositeDataSetBase(object):
class vtkPartitionedDataSet(CompositeDataSetBase, vtkPartitionedDataSet):
def append(self, dataset):
self.SetPartition(self.GetNumberOfPartitions(), dataset)
+
+# -----------------------------------------------------------------------------
+# Handle pickle registration
+# -----------------------------------------------------------------------------
+with suppress(ImportError):
+ import copyreg
+ from vtkmodules.util.pickle_support import serialize_VTK_data_object
+
+ copyreg.pickle(vtkPolyData, serialize_VTK_data_object)
+ copyreg.pickle(vtkUnstructuredGrid, serialize_VTK_data_object)
+ copyreg.pickle(vtkImageData, serialize_VTK_data_object)
+ copyreg.pickle(vtkPartitionedDataSet, serialize_VTK_data_object)
--
GitLab