mitkNodePredicates.cpp

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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

#include "mitkNodePredicates.h"

// mitk core
#include <mitkNodePredicateDataType.h>
#include <mitkNodePredicateNot.h>
#include <mitkNodePredicateOr.h>
#include <mitkNodePredicateProperty.h>

// multi label module
#include <mitkLabelSetImage.h>

mitk::NodePredicateAnd::Pointer mitk::NodePredicates::GetImagePredicate()
{
  TNodePredicateDataType<Image>::Pointer isImage = TNodePredicateDataType<Image>::New();

  NodePredicateOr::Pointer validImages = NodePredicateOr::New();
  validImages->AddPredicate(isImage);

  NodePredicateAnd::Pointer imagePredicate = NodePredicateAnd::New();
  imagePredicate->AddPredicate(validImages);
  imagePredicate->AddPredicate(NodePredicateNot::New(GetSegmentationPredicate()));
  imagePredicate->AddPredicate(NodePredicateNot::New(NodePredicateProperty::New("helper object")));

  return imagePredicate;
}

mitk::NodePredicateAnd::Pointer mitk::NodePredicates::GetSegmentationPredicate()
{
  NodePredicateAnd::Pointer segmentationPredicate = NodePredicateAnd::New();

  NodePredicateProperty::Pointer isBinary = NodePredicateProperty::New("binary", BoolProperty::New(true));
  TNodePredicateDataType<LabelSetImage>::Pointer isLabelSetImage = TNodePredicateDataType<LabelSetImage>::New();
  NodePredicateOr::Pointer allSegmentations = NodePredicateOr::New(isBinary, isLabelSetImage);

  segmentationPredicate->AddPredicate(allSegmentations);
  segmentationPredicate->AddPredicate(NodePredicateNot::New(NodePredicateProperty::New("helper object")));

  return segmentationPredicate;
}