cxStlMeshReader.cpp

Go to the documentation of this file.

/*=========================================================================
This file is part of CustusX, an Image Guided Therapy Application.

Copyright (c) SINTEF Department of Medical Technology.
All rights reserved.

CustusX is released under a BSD 3-Clause license.

See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.
=========================================================================*/

#include "cxStlMeshReader.h"

#include <QFileInfo>
#include <vtkSTLReader.h>
#include "cxTypeConversions.h"
#include <vtkPolyData.h>
#include "cxMesh.h"
#include "cxErrorObserver.h"
#include <ctkPluginContext.h>
#include <vtkSTLWriter.h>
#include "cxLogger.h"

namespace cx
{

StlMeshReader::StlMeshReader(PatientModelServicePtr patientModelService) :
        FileReaderWriterImplService("StlMeshReader", Mesh::getTypeName(), Mesh::getTypeName(), "stl", patientModelService)
{
}

bool StlMeshReader::canRead(const QString &type, const QString &filename)
{
        QString fileType = QFileInfo(filename).suffix();
        return (fileType.compare("stl", Qt::CaseInsensitive) == 0);
}

bool StlMeshReader::readInto(DataPtr data, QString filename)
{
        return this->readInto(boost::dynamic_pointer_cast<Mesh>(data), filename);
}

bool StlMeshReader::readInto(MeshPtr mesh, QString filename)
{
        if (!mesh)
                return false;
        vtkPolyDataPtr raw = this->loadVtkPolyData(filename);
        if(!raw)
                return false;
        mesh->setVtkPolyData(raw);
        return true;
}

QString StlMeshReader::canReadDataType() const
{
        return Mesh::getTypeName();
}

vtkPolyDataPtr StlMeshReader::loadVtkPolyData(QString fileName)
{
        vtkSTLReaderPtr reader = vtkSTLReaderPtr::New();
        reader->SetFileName(cstring_cast(fileName));

        if (!ErrorObserver::checkedRead(reader, fileName))
                return vtkPolyDataPtr();

        vtkPolyDataPtr polyData = reader->GetOutput();
        return polyData;
}

DataPtr StlMeshReader::read(const QString& uid, const QString& filename)
{
        MeshPtr mesh(new Mesh(uid));
        this->readInto(mesh, filename);
        return mesh;
}

std::vector<DataPtr> StlMeshReader::read(const QString &filename)
{
        std::vector<DataPtr> retval;
        MeshPtr mesh = boost::dynamic_pointer_cast<Mesh>(this->createData(Mesh::getTypeName(), filename));

        vtkPolyDataPtr raw = this->loadVtkPolyData(filename);
        if(!raw)
                return retval;
        mesh->setVtkPolyData(raw);

        retval.push_back(mesh);
        return retval;
}

void StlMeshReader::write(DataPtr data, const QString &filename)
{
        MeshPtr mesh = boost::dynamic_pointer_cast<Mesh>(data);
        if(!mesh)
                reportError("Could not cast data to mesh");
        vtkSTLWriterPtr writer = vtkSTLWriterPtr::New();
        writer->SetInputData(mesh->getVtkPolyData());
        writer->SetFileName(cstring_cast(filename));

        writer->Update();
        writer->Write();
}

}

QString cx::StlMeshReader::canWriteDataType() const
{
        return Mesh::getTypeName();
}

bool cx::StlMeshReader::canWrite(const QString &type, const QString &filename) const
{
        return this->canWriteInternal(type, filename);
}