mitkDiffusionPropertyHelper.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 MITKDIFFUSIONPROPERTYHELPER_H
#define MITKDIFFUSIONPROPERTYHELPER_H

#include <MitkDiffusionCoreExports.h>

#include <mitkImage.h>
#include <mitkBValueMapProperty.h>
#include <mitkGradientDirectionsProperty.h>
#include <mitkMeasurementFrameProperty.h>
#include <mitkDataNode.h>

namespace mitk
{
  class MITKDIFFUSIONCORE_EXPORT DiffusionPropertyHelper
  {
  public:

    typedef short                                                         DiffusionPixelType;

    typedef mitk::BValueMapProperty::BValueMap                            BValueMapType;
    typedef GradientDirectionsProperty::GradientDirectionType             GradientDirectionType;
    typedef GradientDirectionsProperty::GradientDirectionsContainerType   GradientDirectionsContainerType;
    typedef mitk::MeasurementFrameProperty::MeasurementFrameType          MeasurementFrameType;
    typedef itk::VectorImage< DiffusionPixelType, 3>                      ImageType;

    static const std::string GRADIENTCONTAINERPROPERTYNAME;
    static const std::string ORIGINALGRADIENTCONTAINERPROPERTYNAME;
    static const std::string MEASUREMENTFRAMEPROPERTYNAME;
    static const std::string REFERENCEBVALUEPROPERTYNAME;
    static const std::string BVALUEMAPPROPERTYNAME;
    static const std::string MODALITY;

    DiffusionPropertyHelper( mitk::Image* inputImage );
    ~DiffusionPropertyHelper();

    static bool IsDiffusionWeightedImage(const mitk::Image *);
    static bool IsDiffusionWeightedImage(const mitk::DataNode* node);

    static ImageType::Pointer GetItkVectorImage(Image *image);

    static const BValueMapType & GetBValueMap(const mitk::Image *);
    static float GetReferenceBValue(const mitk::Image *);
    static const MeasurementFrameType & GetMeasurementFrame(const mitk::Image *);
    static GradientDirectionsContainerType::Pointer GetOriginalGradientContainer(const mitk::Image *);
    static GradientDirectionsContainerType::Pointer GetGradientContainer(const mitk::Image *);

    const BValueMapType & GetBValueMap() const;
    float GetReferenceBValue() const;
    const MeasurementFrameType & GetMeasurementFrame() const;
    GradientDirectionsContainerType::Pointer GetOriginalGradientContainer() const;
    GradientDirectionsContainerType::Pointer GetGradientContainer() const;

    bool IsDiffusionWeightedImage() const;

    void AverageRedundantGradients(double precision);

    void InitializeImage();

    GradientDirectionsContainerType::Pointer CalcAveragedDirectionSet(double precision, GradientDirectionsContainerType::Pointer directions);

  protected:

    DiffusionPropertyHelper();

    void ApplyMeasurementFrame();

    void UpdateBValueMap();

    bool AreAlike(GradientDirectionType g1, GradientDirectionType g2, double precision);

    float GetB_Value(unsigned int i);

    mitk::Image* m_Image;

  };
}
#endif