mitkRigidRegistrationPreset.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef MITKRIGIDREGISTRATIONPRESET_H_HEADER
#define MITKRIGIDREGISTRATIONPRESET_H_HEADER

#include "MitkRigidRegistrationExports.h"
#include <vtkXMLParser.h>

#include <list>
#include <map>
#include <string>

#include "mitkCommon.h"
#include <itkArray.h>

namespace mitk
{
  class MITKRIGIDREGISTRATION_EXPORT RigidRegistrationPreset : public vtkXMLParser
  {
  public:
    RigidRegistrationPreset();
    ~RigidRegistrationPreset();

    bool LoadPreset();
    bool LoadPreset(std::string fileName);
    itk::Array<double> getTransformValues(std::string name);
    itk::Array<double> getMetricValues(std::string name);
    itk::Array<double> getOptimizerValues(std::string name);
    itk::Array<double> getInterpolatorValues(std::string name);

    std::list<std::string> &getAvailablePresets();

    std::map<std::string, itk::Array<double>> &getTransformValuesPresets();
    std::map<std::string, itk::Array<double>> &getMetricValuesPresets();
    std::map<std::string, itk::Array<double>> &getOptimizerValuesPresets();
    std::map<std::string, itk::Array<double>> &getInterpolatorValuesPresets();
    bool newPresets(std::map<std::string, itk::Array<double>> newTransformValues,
                    std::map<std::string, itk::Array<double>> newMetricValues,
                    std::map<std::string, itk::Array<double>> newOptimizerValues,
                    std::map<std::string, itk::Array<double>> newInterpolatorValues,
                    std::string fileName = "");

  private:
    void StartElement(const char *elementName, const char **atts) override;

    // bool saveXML(mitk::XMLWriter& xmlWriter);
    bool save();
    // void saveTransformValues(mitk::XMLWriter& xmlWriter, std::string item);
    // void saveMetricValues(mitk::XMLWriter& xmlWriter, std::string item);
    // void saveOptimizerValues(mitk::XMLWriter& xmlWriter, std::string item);
    // void saveInterpolatorValues(mitk::XMLWriter& xmlWriter, std::string item);

    itk::Array<double> loadTransformValues(itk::Array<double> transformValues, double transform, const char **atts);
    itk::Array<double> loadMetricValues(itk::Array<double> metricValues, double metric, const char **atts);
    itk::Array<double> loadOptimizerValues(itk::Array<double> optimizerValues, double optimizer, const char **atts);
    itk::Array<double> loadInterpolatorValues(
      itk::Array<double> interpolatorValues /*, double interpolator, const char **atts*/);

    std::string ReadXMLStringAttribut(std::string name, const char **atts);

    std::string m_Name;
    std::map<std::string, itk::Array<double>> m_TransformValues;
    std::map<std::string, itk::Array<double>> m_MetricValues;
    std::map<std::string, itk::Array<double>> m_OptimizerValues;
    std::map<std::string, itk::Array<double>> m_InterpolatorValues;
    std::list<std::string> m_LoadedPresets;

    std::string m_XmlFileName;
  };
}
#endif