mlClusters.h

Go to the documentation of this file.

/*************************************************************************************
**
** Copyright 2014, MeVis Medical Solutions AG
**
** The user may use this file in accordance with the license agreement provided with
** the Software or, alternatively, in accordance with the terms contained in a
** written agreement between the user and MeVis Medical Solutions AG.
**
** For further information use the contact form at https://www.mevislab.de/contact
**
**************************************************************************************/
 
#ifndef ML_CLUSTERS_H
#define ML_CLUSTERS_H
 
#include "MLClusterAlgorithmSystem.h"
#include "mlModuleIncludes.h"
 
#include "mlClusterRefVolume.h"
#include "mlClusterRefCollection.h"
 
 
#include <boost/shared_ptr.hpp>
 
ML_START_NAMESPACE
 
class MLCLUSTERALGORITHM_EXPORT Clusters
{
public:
 
  Clusters();
  ~Clusters();
 
  MLErrorCode init(MLint extentX, MLint extentY, MLint extentZ);
 
 
  void calcClusterSizes(const ClusterUserDataParameters& userDataParameters);
  void incrementClusterSize(MLuint currentClusterRef);
 
  MLuint merge(MLuint currentClusterRef, MLuint neighborClusterRef);
 
 
 
  std::vector<ClusterInfo> getClusterInfoList() const;
 
  const ClusterInfo& getClusterOfVoxel(MLint x, MLint y, MLint z) const;
  const ClusterInfo& getClusterAt(MLuint rank) const;
  const ClusterInfo& getBackgroundCluster() const;
  MLuint getClusterRef(MLint x, MLint y, MLint z) const;
  void setUserDataAtRank(MLuint rank, MLdouble userData);
 
  MLuint getNewReference(MLdouble imageValue);
  void setClusterRef(MLint xi, MLint yi, MLint zi, MLuint currentClusterRef);
 
  void setImageParameters(const PagedImage* inImage);
 
  MLuint getNumClusters() const;
  MLuint getMaxRank() const;
  MLuint getMaxClusterSize() const;
  ImageVector getImageExtent() const { return _imageProperties.getImageExtent(); }
  const ImageVector& getPageExtent() const { return _pageExtent; }
  const Matrix4& getVoxelToWorldMatrix() const { return _imageProperties.getVoxelToWorldMatrix(); }
  const Matrix4& getWorldToVoxeldMatrix() const { return _imageProperties.getWorldToVoxelMatrix(); }
  MLdouble getVoxelVolumeInMilliliters() const { return _imageProperties.getVoxelVolumeInMilliliters(); }
  MedicalImageProperties getImageProperties() const { return _imageProperties; }
  MLuint getRankOfSizeAt(MLint x, MLint y, MLint z) const;
  void setUse2DNeighborhood() { _using2DNeighborhood = true; }
  bool isUsing2DNeighborhood() const { return _using2DNeighborhood; }
 
 
private:
 
  void computeBoundingBoxes();
 
  void setCTUTo1(ImageVector& v);
 
  ClusterRefVolume*     _clusterRefVolume;
  ClusterRefCollection* _clusterRefCollection;
 
  MedicalImageProperties _imageProperties;
  ImageVector _pageExtent;
 
  bool _using2DNeighborhood;
};
 
 
ML_END_NAMESPACE
 
 
#endif // __mlClusters_H