mitkDataCollectionUtility.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 mitkDataCollectionUtilit_h
#define mitkDataCollectionUtilit_h


#include <mitkDataCollection.h>

namespace mitk
{
  class DataCollectionUtility
  {
  public:
    template <class OriginImageType, class TargetImageType>
    static void EnsureImageInCollection(mitk::DataCollection::Pointer collection, std::string origin, std::string target);

    static void Compile()
    {
      mitk::DataCollection::Pointer col = mitk::DataCollection::New();
      std::string blub = "haha";
      typedef itk::Image<double, 3> ImType;
      EnsureImageInCollection<ImType, ImType>(col, blub, blub);
  };
} // end namespace

template <class OriginImageType, class TargetImageType>
void
mitk::DataCollectionUtility::EnsureDataImageInCollection(mitk::DataCollection::Pointer collection, std::string origin, std::string target)
{
    typedef typename TargetImageType FeatureImage;
    typedef typename OriginImageType LabelImage;

    if (collection->HasElement(origin))
    {
        LabelImage::Pointer originImage = dynamic_cast<LabelImage*>(collection->GetData(origin).GetPointer());
        if (!collection->HasElement(target) && originImage.IsNotNull())
        {
            FeatureImage::Pointer image = FeatureImage::New();
            image->SetRegions(originImage->GetLargestPossibleRegion());
            image->SetSpacing(originImage->GetSpacing());
            image->SetOrigin(originImage->GetOrigin());
            image->SetDirection(originImage->GetDirection());
            image->Allocate();

            collection->AddData(dynamic_cast<itk::DataObject*>(image.GetPointer()),target,"");
        }
    }
    for (std::size_t i = 0; i < collection->Size();++i)
    {
        mitk::DataCollection* newCol = dynamic_cast<mitk::DataCollection*>(collection->GetData(i).GetPointer());
        if (newCol != 0)
        {
            EnsureDataImageInCollection(newCol, origin, target);
        }
    }
};


#include <Iterators/mitkDataCollectionImageIterator.cxx>
#endif //mitkDataCollectionUtilit_h