1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
/************************************************************************
*
* Copyright (C) 2021-2023 IRCAD France
* Copyright (C) 2021 IHU Strasbourg
*
* This file is part of Sight.
*
* Sight 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 3 of the License, or
* (at your option) any later version.
*
* Sight is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with Sight. If not, see <https://www.gnu.org/licenses/>.
*
***********************************************************************/
#include "mesh.hpp"
#include <io/vtk/helper/mesh.hpp>
#include <vtkCenterOfMass.h>
#include <vtkPolyData.h>
#include <vtkSmartPointer.h>
namespace sight::geometry::vtk
{
//------------------------------------------------------------------------------
data::point::sptr compute_center_of_mass(const data::mesh::csptr _mesh, const bool _use_scalar_as_weights)
{
vtkSmartPointer<vtkPolyData> poly_data = vtkSmartPointer<vtkPolyData>::New();
sight::io::vtk::helper::mesh::to_vtk_mesh(_mesh, poly_data);
// Compute the center of mass
vtkSmartPointer<vtkCenterOfMass> center_of_mass_filter =
vtkSmartPointer<vtkCenterOfMass>::New();
center_of_mass_filter->SetInputData(poly_data);
center_of_mass_filter->SetUseScalarsAsWeights(_use_scalar_as_weights);
center_of_mass_filter->Update();
std::array<double, 3> center_of_mass {};
center_of_mass_filter->GetCenter(center_of_mass.data());
data::point::sptr center = std::make_shared<data::point>(center_of_mass[0], center_of_mass[1], center_of_mass[2]);
return center;
}
//------------------------------------------------------------------------------
} // namespace sight::geometry::vtk
|
