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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/*****************************************************************************
* $CAMITK_LICENCE_BEGIN$
*
* CamiTK - Computer Assisted Medical Intervention ToolKit
* (c) 2001-2014 UJF-Grenoble 1, CNRS, TIMC-IMAG UMR 5525 (GMCAO)
*
* Visit http://camitk.imag.fr for more information
*
* This file is part of CamiTK.
*
* CamiTK is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 3
* only, as published by the Free Software Foundation.
*
* CamiTK 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 version 3 for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* version 3 along with CamiTK. If not, see <http://www.gnu.org/licenses/>.
*
* $CAMITK_LICENCE_END$
****************************************************************************/
#include "OffComponent.h"
#include "OffExtension.h"
//-- Qt
#include <QFileInfo>
#include <QMessageBox>
//-- VTK
#include <vtkCellType.h>
#include <vtkCell.h>
//-- CamiTK
#include <Log.h>
#include <Application.h>
using namespace camitk;
// declare the plugin
Q_EXPORT_PLUGIN2(OffExtension, OffExtension);
QString OffExtension::getName() const {
return "Off Component";
}
QString OffExtension::getDescription() const {
return "OFF <em>.off</em> files in <b>CamiTK</b>.(very few support!) <br>(c)TIMC-IMAG UJF 2012";
}
QStringList OffExtension::getFileExtensions() const {
QStringList ext;
ext << "off";
return ext;
}
camitk::Component * OffExtension::open(const QString & fileName) throw (camitk::AbortException) {
return new OffComponent(fileName);
}
bool OffExtension::save(Component* component) const {
bool saveOk = true;
vtkSmartPointer<vtkPointSet> ps = component->getPointSet();
if (ps == NULL || ps->GetNumberOfPoints()==0 || ps->GetNumberOfCells() == 0) {
QMessageBox::warning(NULL, "Save As msh", "The selected component does not have any points or cells. This is an invalid mesh.", QMessageBox::Abort);
return false;
}
QString baseFilename = QFileInfo(component->getFileName()).absolutePath() + "/" + QFileInfo(component->getFileName()).completeBaseName();
QString offFilename = baseFilename + ".off";
ofstream offFile ( offFilename.toUtf8() );
double xyz[3];
offFile << "OFF" << endl;
// write the number of points, cells and edges
offFile << ps->GetPoints()->GetNumberOfPoints() << " " << ps->GetNumberOfCells() << " " << ps->GetNumberOfElements(vtkPointSet::EDGE) << endl;
// list the points
for ( int i = 0; i < ps->GetPoints()->GetNumberOfPoints(); i++ ) {
ps->GetPoints()->GetPoint(i, xyz);
// write atom
offFile << xyz[0] << " " << xyz[1] << " " << xyz[2] << endl;
}
for ( vtkIdType i = 0; i < ps->GetNumberOfCells() ; i++ ) {
switch (ps->GetCell(i)->GetCellType()) {
case VTK_TRIANGLE :
offFile << "3 ";
for (vtkIdType j=0;j<ps->GetCell(i)->GetNumberOfPoints();j++)
offFile << ps->GetCell(i)->GetPointId(j) << " ";
offFile << endl;
break;
case VTK_QUAD :
offFile << "4 ";
for (vtkIdType j=0;j<ps->GetCell(i)->GetNumberOfPoints();j++)
offFile << ps->GetCell(i)->GetPointId(j) << " ";
offFile << endl;
break;
default:
saveOk = false;
CAMITK_ERROR("VolumetricMeshExtension", "save", "unsupported element");
break;
}
}
offFile.close();
return saveOk;
}
|
