put back missing header and use vtk SafeDownCast, re #11706

mantidproject · May 13, 2015 · 811369e · 811369e
1 parent 782fd86
commit 811369e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Code/Mantid/Vates/VatesSimpleGui/ViewWidgets/src/ColorUpdater.cpp b/Code/Mantid/Vates/VatesSimpleGui/ViewWidgets/src/ColorUpdater.cpp
@@ -22,6 +22,7 @@
 #include <pqServerManagerModel.h>
 #include <pqSMAdaptor.h>
 
+#include <vtkCallbackCommand.h>
 #include <vtkSMDoubleVectorProperty.h>
 #include <vtkSMProxy.h>
 
@@ -311,7 +312,8 @@ void ColorUpdater::observeColorScaleEdited(pqPipelineRepresentation *repr, Color
   ccdata.csel = cs;
   CRChangeCallback->SetClientData(&ccdata);
   // install callback
-  vtkSMDoubleVectorProperty *points = lutProxy->GetProperty("RGBPoints");
+  vtkSMDoubleVectorProperty *points =
+    vtkSMDoubleVectorProperty::SafeDownCast(lutProxy->GetProperty("RGBPoints"));
   points->AddObserver(vtkCommand::ModifiedEvent, CRChangeCallback);
 }