Refs #11814 Change recursion depth to 1 when using top level split

mantidproject · May 20, 2015 · ec6b4b1 · ec6b4b1
1 parent 3d28511
commit ec6b4b1
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Code/Mantid/Vates/ParaviewPlugins/ParaViewSources/MDEWSource/vtkMDEWSource.cxx b/Code/Mantid/Vates/ParaviewPlugins/ParaViewSources/MDEWSource/vtkMDEWSource.cxx
@@ -9,6 +9,7 @@
 #include "vtkUnstructuredGrid.h"
 #include "vtkStreamingDemandDrivenPipeline.h"
 
+#include "MantidAPI/BoxController.h"
 #include "MantidVatesAPI/MDEWInMemoryLoadingPresenter.h"
 #include "MantidVatesAPI/MDLoadingViewAdapter.h"
 #include "MantidVatesAPI/ADSWorkspaceProvider.h"
@@ -230,6 +231,9 @@ int vtkMDEWSource::RequestInformation(vtkInformation *vtkNotUsed(request), vtkIn
     }
     else
     {
+      // If the MDEvent workspace has had top level splitting applied to it, then use the a depth of 1
+      setDepthForWorkspaceWithTopLevelSplitting();
+
       m_presenter->executeLoadMetadata();
       setTimeRange(outputVector);
     }
@@ -325,3 +329,17 @@ const char* vtkMDEWSource::GetWorkspaceName()
 {
   return m_wsName.c_str();
 }
+
+void vtkMDEWSource::setDepthForWorkspaceWithTopLevelSplitting()
+{
+  // Fetch the workspace from the data analysis service
+  ADSWorkspaceProvider<Mantid::API::IMDEventWorkspace> workspaceProvider;
+  auto result = boost::dynamic_pointer_cast<Mantid::API::IMDEventWorkspace>(workspaceProvider.fetchWorkspace(m_wsName));
+  if (result) {
+    auto boxController = result->getBoxController();
+    boost::optional<std::vector<size_t>> topLevelSplits = boxController->getSplitTopInto();
+    if (topLevelSplits) {
+      SetDepth(1);
+    }
+  }
+}
diff --git a/Code/Mantid/Vates/ParaviewPlugins/ParaViewSources/MDEWSource/vtkMDEWSource.h b/Code/Mantid/Vates/ParaviewPlugins/ParaViewSources/MDEWSource/vtkMDEWSource.h
@@ -100,5 +100,6 @@ class VTK_EXPORT vtkMDEWSource : public vtkUnstructuredGridAlgorithm
   vtkMDEWSource(const vtkMDEWSource&);
   void operator = (const vtkMDEWSource&);
   void setTimeRange(vtkInformationVector* outputVector);
+  void setDepthForWorkspaceWithTopLevelSplitting();
 };
 #endif