cxCustomMetaImage.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 "cxCustomMetaImage.h"

#include <QFile>
#include <QTextStream>
#include <QStringList>
#include "cxLogger.h"

#include "cxTypeConversions.h"

namespace cx
{

CustomMetaImage::CustomMetaImage(QString filename) :
    mFilename(filename)
{}

QString CustomMetaImage::readKey(QString key)
{
        QFile file(mFilename);

        QString line;
        if (file.open(QIODevice::ReadOnly))
        {
                QTextStream t(&file);
                while (!t.atEnd())
                {
                        line.clear();
                        line = t.readLine();
                        // do something with the line
                        if (line.startsWith(key, Qt::CaseInsensitive))
                        {
                                QStringList list = line.split("=", QString::SkipEmptyParts);
                                if (list.size() >= 2)
                                {
                                        list = list.mid(1);
                                        return list.join("=");
                                }
                        }
                }
                file.close();
        }

        return "";
}

QString CustomMetaImage::readModality()
{
        QString mod = this->readKey("Modality");

        if (mod.contains("CT", Qt::CaseInsensitive))
                return "CT";
        if (mod.contains("MR", Qt::CaseInsensitive))
                return "MR";
        if (mod.contains("US", Qt::CaseInsensitive))
                return "US";
        if (mod.contains("OTHER", Qt::CaseInsensitive))
                return "OTHER";
        mod = mod.remove("MET_MOD_");
        return mod;
}

QString CustomMetaImage::readImageType()
{
        return this->readKey("ImageType3");
}

void CustomMetaImage::remove(QStringList* data, QStringList keys)
{
        QRegExp regexp(QString("(^%1)").arg(keys.join("|^")));
        QStringList removeThese = data->filter(regexp);
        for (int i=0; i<removeThese.size(); ++i)
                data->removeAll(removeThese[i]);
}

void CustomMetaImage::append(QStringList* data, QString key, QString value)
{
        // find index of ElementDataFile - this is the last element according to MHD standard (but we might have appended something else after it).
    int last = data->lastIndexOf(QRegExp("^ElementDataFile.*"));
        data->insert(last, QString("%1 = %2").arg(key).arg(value));
}

void CustomMetaImage::setKey(QString key, QString value)
{
        QFile file(mFilename);

        if (!file.open(QIODevice::ReadWrite))
        {
          reportError("Failed to open file " + mFilename + ".");
          return;
        }

        QStringList data = QTextStream(&file).readAll().split("\n");

        this->remove(&data, QStringList()<<key);
        this->append(&data, key, value);

        file.resize(0);
        file.write(data.join("\n").toLatin1());
}

void CustomMetaImage::setModality(QString value)
{
    QStringList valid_values;
    valid_values << "CT" << "MR" << "NM" << "US" << "OTHER" << "UNKNOWN";
    if(!valid_values.contains(value, Qt::CaseInsensitive))
        value = "UNKNOWN";

        this->setKey("Modality", value);
}

void CustomMetaImage::setImageType(QString value)
{
        if (value.isEmpty())
                return;
        this->setKey("ImageType3", value);
}


Transform3D CustomMetaImage::readTransform()
{
  //read the specific TransformMatrix-tag from the header
  Vector3D p_r(0, 0, 0);
  Vector3D e_x(1, 0, 0);
  Vector3D e_y(0, 1, 0);
  Vector3D e_z(0, 0, 1);

  QFile file(mFilename);

  QString line;
  if (file.open(QIODevice::ReadOnly))
  {
    QTextStream t(&file);
    while (!t.atEnd())
    {
      line.clear();
      line = t.readLine();
      // do something with the line
      if (line.startsWith("Position", Qt::CaseInsensitive) || line.startsWith("Offset", Qt::CaseInsensitive))
      {
        QStringList list = line.split(" ", QString::SkipEmptyParts);
        if (list.size()>=5)
          p_r = Vector3D(list[2].toDouble(), list[3].toDouble(), list[4].toDouble());
      }
      else if (line.startsWith("TransformMatrix", Qt::CaseInsensitive) || line.startsWith("Orientation",
          Qt::CaseInsensitive))
      {
        QStringList list = line.split(" ", QString::SkipEmptyParts);

        if (list.size()>=8)
        {
          e_x = Vector3D(list[2].toDouble(), list[3].toDouble(), list[4].toDouble());
          e_y = Vector3D(list[5].toDouble(), list[6].toDouble(), list[7].toDouble());
          e_z = cross(e_x, e_y);
        }
      }
    }
    file.close();
  }

  Transform3D rMd = Transform3D::Identity();

  // add rotational part
  for (unsigned i = 0; i < 3; ++i)
  {
    rMd(i,0) = e_x[i];
    rMd(i,1) = e_y[i];
    rMd(i,2) = e_z[i];
  }


  // add translational part
  rMd(0,3) = p_r[0];
  rMd(1,3) = p_r[1];
  rMd(2,3) = p_r[2];
  return rMd;
}

void CustomMetaImage::setTransform(const Transform3D M)
{
  QFile file(mFilename);

  if (!file.open(QIODevice::ReadWrite))
  {
    reportWarning("Could not save transform because: Failed to open file " + mFilename);
    return;
  }

  QStringList data = QTextStream(&file).readAll().split("\n");

  this->remove(&data, QStringList()<<"TransformMatrix"<<"Offset"<<"Position"<<"Orientation");

  int dim = 3; // hardcoded - will fail for 2d images
  std::stringstream tmList;
  for (int c=0; c<dim; ++c)
    for (int r=0; r<dim; ++r)
      tmList << " " << M(r,c);
  this->append(&data, "TransformMatrix", qstring_cast(tmList.str()));

  std::stringstream posList;
  for (int r=0; r<dim; ++r)
    posList << " " << M(r,3);
  this->append(&data, "Offset", qstring_cast(posList.str()));

  file.resize(0);
  file.write(data.join("\n").toLatin1());
}

}