Migration to OpenCASCADE CMake configuration

[modules/gui.git] / src / VTKViewer / VTKViewer_ArcBuilder.cxx

diff --git a/src/VTKViewer/VTKViewer_ArcBuilder.cxx b/src/VTKViewer/VTKViewer_ArcBuilder.cxx
index dd87a91a31919ce991f4c81447787dc0b1ecb61e..e0bf486804bf3ec885c7669500497a0a86ff876a 100644 (file)
--- a/src/VTKViewer/VTKViewer_ArcBuilder.cxx
+++ b/src/VTKViewer/VTKViewer_ArcBuilder.cxx
@@ -1,9 +1,9 @@
-// Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
+// Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
 //
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 // License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
 //
 // This library is distributed in the hope that it will be useful,
 // but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -396,9 +396,13 @@ VTKViewer_ArcBuilder::TransformGrid(vtkUnstructuredGrid* theGrid,
   aTransform->RotateWXYZ(angle, theAxis.GetXYZ().X(), theAxis.GetXYZ().Y(), theAxis.GetXYZ().Z());
   vtkTransformFilter* aTransformFilter  = vtkTransformFilter::New();
   aTransformFilter->SetTransform(aTransform);
-  aTransformFilter->SetInputData(theGrid);
   aTransform->Delete();
-  return aTransformFilter->GetUnstructuredGridOutput();
+  aTransformFilter->SetInputData(theGrid);
+  aTransformFilter->Update();
+  vtkUnstructuredGrid * aGrid = aTransformFilter->GetUnstructuredGridOutput();
+  aGrid->Register(0);
+  aTransformFilter->Delete();
+  return aGrid;
 }