cgal/Mesh_3/demo/Mesh_3/Io_off_plugin.cpp

95 lines
2.3 KiB
C++

#include "config.h"
#include "Scene_polyhedron_item.h"
#include "Scene_polygon_soup.h"
#include "Polyhedron_type.h"
#include <CGAL_demo/Io_plugin_interface.h>
#include <fstream>
class Io_off_plugin :
public QObject,
public Io_plugin_interface
{
Q_OBJECT
Q_INTERFACES(Io_plugin_interface)
public:
QStringList nameFilters() const;
bool canLoad() const;
Scene_item* load(QFileInfo fileinfo);
bool canSave(const Scene_item*);
bool save(const Scene_item*, QFileInfo, QString);
};
QStringList Io_off_plugin::nameFilters() const {
return QStringList() << "OFF files (*.off)";
}
bool Io_off_plugin::canLoad() const {
return true;
}
Scene_item*
Io_off_plugin::load(QFileInfo fileinfo) {
// Open file
std::ifstream in(fileinfo.filePath().toUtf8());
if(!in) {
std::cerr << "Error! Cannot open file " << (const char*)fileinfo.filePath().toUtf8() << std::endl;
return NULL;
}
// Try to read .off in a polyhedron
Scene_polyhedron_item* item = new Scene_polyhedron_item();
item->setName(fileinfo.baseName());
if(!item->load(in))
{
delete item;
// Try to read .off in a polygon soup
Scene_polygon_soup* soup_item = new Scene_polygon_soup;
soup_item->setName(fileinfo.baseName());
in.close();
std::ifstream in2(fileinfo.filePath().toUtf8());
if(!soup_item->load(in2)) {
delete soup_item;
return 0;
}
return soup_item;
}
return item;
}
bool Io_off_plugin::canSave(const Scene_item* item)
{
// This plugin supports polyhedrons and polygon soups
return qobject_cast<const Scene_polyhedron_item*>(item) ||
qobject_cast<const Scene_polygon_soup*>(item);
}
bool Io_off_plugin::save(const Scene_item* item, QFileInfo fileinfo,
QString /* `selecterFilter` is not used: only OFF */)
{
// This plugin supports polyhedrons and polygon soups
const Scene_polyhedron_item* poly_item =
qobject_cast<const Scene_polyhedron_item*>(item);
const Scene_polygon_soup* soup_item =
qobject_cast<const Scene_polygon_soup*>(item);
if(!poly_item && !soup_item)
return false;
std::ofstream out(fileinfo.filePath().toUtf8());
return (poly_item && poly_item->save(out)) ||
(soup_item && soup_item->save(out));
}
#include <QtPlugin>
Q_EXPORT_PLUGIN2(Io_off_plugin, Io_off_plugin)
#include "Io_off_plugin.moc"