vtkDataArrayTemplate: Add a safe method to change an element with vtk…

…Variant. This fix is in reference to bug report 0013143.
Kitware · Aug 18, 2015 · c7ebe5b · c7ebe5b
1 parent 2e0acc8
commit c7ebe5b
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/Common/Core/vtkDataArrayTemplate.h b/Common/Core/vtkDataArrayTemplate.h
@@ -212,6 +212,10 @@ class VTKCOMMONCORE_EXPORT vtkDataArrayTemplate:
   // Set a value in the array from a vtkVariant.
   void SetVariantValue(vtkIdType id, vtkVariant value);
 
+  // Description:
+  // Insert a value in the array from a vtkVariant.
+  void InsertVariantValue(vtkIdType id, vtkVariant value);
+
   // Description:
   // Insert data at the end of the array. Return its location in the array.
   vtkIdType InsertNextValue(T f);

diff --git a/Common/Core/vtkDataArrayTemplate.txx b/Common/Core/vtkDataArrayTemplate.txx
@@ -979,6 +979,22 @@ void vtkDataArrayTemplate<T>::SetVariantValue(vtkIdType id, vtkVariant value)
     }
 }
 
+//----------------------------------------------------------------------------
+template <class T>
+void vtkDataArrayTemplate<T>::InsertVariantValue(vtkIdType id, vtkVariant value)
+{
+  bool valid;
+  T toInsert = vtkVariantCast<T>(value, &valid);
+  if (valid)
+    {
+    this->InsertValue(id, toInsert);
+    }
+  else
+    {
+    vtkErrorMacro("unable to set value of type " << value.GetType());
+    }
+}
+
 //----------------------------------------------------------------------------
 template <class T>
 vtkIdType vtkDataArrayTemplate<T>::InsertNextValue(T f)