cxCgeoReaderWriter.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 "cxCgeoReaderWriter.h"
 
#include <QDir>
#include "vtkPolyData.h"
#include <vtkCellArray.h>
#include "vtkIdList.h"
#include "cxMesh.h"
#include "cxLogger.h"
 
namespace cx
{
 
CgeoReaderWriter::CgeoReaderWriter(PatientModelServicePtr patientModelService) :
        FileReaderWriterImplService("CgeoReaderWriter", "", Mesh::getTypeName(), "cgeo", patientModelService)
{
}
 
bool CgeoReaderWriter::isNull()
{
        return false;
}
 
bool CgeoReaderWriter::canRead(const QString &type, const QString &filename)
{
        return false;
}
 
std::vector<DataPtr> CgeoReaderWriter::read(const QString &filename)
{
        return std::vector<DataPtr>();
}
 
DataPtr CgeoReaderWriter::read(const QString &uid, const QString &filename)
{
        return DataPtr();
}
 
QString CgeoReaderWriter::canReadDataType() const
{
        return "";
}
 
bool CgeoReaderWriter::readInto(DataPtr data, QString path)
{
        return false;
}
 
void CgeoReaderWriter::write(DataPtr data, const QString &filename)
{
        QFile exportFile(filename);
        exportFile.open(QIODevice::WriteOnly);
        QDataStream out(&exportFile);
        this->writeToStream(data, out);
}
 
QByteArray CgeoReaderWriter::convertToQByteArray(DataPtr data)
{
        QByteArray retval;
        QDataStream out(&retval, QIODevice::WriteOnly);
        this->writeToStream(data, out);
 
        return retval;
}
 
void CgeoReaderWriter::writeToStream(DataPtr data, QDataStream &out)
{
        MeshPtr mesh = boost::dynamic_pointer_cast<Mesh>(data);
        if(!mesh)
        {
                CX_LOG_ERROR() << "Couldn't find mesh.";
                return;
        }
        vtkPolyDataPtr polyData = mesh->getTransformedPolyDataCopy(mesh->get_rMd());
        vtkCellArrayPtr polys = polyData->GetPolys();
 
        out.setByteOrder(QDataStream::LittleEndian);
        out.setFloatingPointPrecision(QDataStream::SinglePrecision);
 
        out << (qint32) 12072001;        // Magic (12072001)
        out << (qint32) 0;  // TextureCount
        out << (qint32) 1;  // PartCount
 
        out << (qint32) 0; // Component ID
 
        if(mesh->getColor().rgba())
        {
                int intRed = mesh->getColor().red();
                int intGreen = mesh->getColor().green();
                int intBlue = mesh->getColor().blue();
                int intAlpha = mesh->getColor().alpha();
                int intColor = intRed<<24 | intGreen<<16 | intBlue<<8 | intAlpha;
                out << (qint32) intColor; // Color
        }
        else
                out << (qint32) 0xFFFFFFFF;//53672537; // Color
 
        out << (qint32) polyData->GetNumberOfPoints(); // # vertices
        out << (qint32) 0; // always 0
        out << (qint32) -1; // always -1
        out << (qint32) polys->GetNumberOfCells(); // # primitives
        out << (qint32) 3; // 3 nodes per primitive (triangle)
 
        for (int i=0; i<polyData->GetNumberOfPoints(); i++)
        {
                double p[3];
                polyData->GetPoint(i,p);
                out << p[0]; // x
                out << p[1]; // y
                out << p[2]; // z
        }
 
        vtkIdType n_ids;
        vtkSmartPointer<vtkIdList> idlist = vtkSmartPointer<vtkIdList>::New();
        polys->InitTraversal();
 
        while(polys->GetNextCell(idlist))
        {
                n_ids = idlist->GetNumberOfIds();
                if (n_ids == 3)
                {
                         out << (qint32)  idlist->GetId(0); // First vertex of triangle
                         out << (qint32)  idlist->GetId(1); // Second vertex of triangle
                         out << (qint32)  idlist->GetId(2); // Third vertex of triangle
                }
                else
                        std::cout << "Warning in .cgeo export: Skipped polygon not containing exactly 3 points." << std::endl;
        }
}
 
QString CgeoReaderWriter::canWriteDataType() const
{
        return Mesh::getTypeName();
}
 
bool CgeoReaderWriter::canWrite(const QString &type, const QString &filename) const
{
        return this->canWriteInternal(type, filename);
}
}//cx