/*========================================================================= Program: vv http://www.creatis.insa-lyon.fr/rio/vv Authors belong to: - University of LYON http://www.universite-lyon.fr/ - Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr - CREATIS CNRS laboratory http://www.creatis.insa-lyon.fr This software is distributed WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the copyright notices for more information. It is distributed under dual licence - BSD See included LICENSE.txt file - CeCILL-B http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html ======================================================================-====*/ #ifndef CLITKSEGMENTATIONUTILS_H #define CLITKSEGMENTATIONUTILS_H // clitk #include "clitkCommon.h" #include "clitkAutoCropFilter.h" #include "clitkLabelizeParameters.h" // itk #include /* According to http://answerpot.com/showthread.php?357451-Itk::SmartPointer%20-%20problem%20making%20code%20const-correct it is better to take raw pointer as argument instead of SmartPointer. */ namespace clitk { //-------------------------------------------------------------------- template void ComputeBBFromImageRegion(const ImageType * image, typename ImageType::RegionType region, typename itk::BoundingBox::Pointer bb); //-------------------------------------------------------------------- template void ComputeBBIntersection(typename itk::BoundingBox::Pointer bbo, typename itk::BoundingBox::Pointer bbi1, typename itk::BoundingBox::Pointer bbi2); //-------------------------------------------------------------------- template void ComputeRegionFromBB(const ImageType * image, const typename itk::BoundingBox::Pointer bb, typename ImageType::RegionType & region); //-------------------------------------------------------------------- template typename TInternalImageType::Pointer SetBackground(const TInternalImageType * input, const TMaskInternalImageType * mask, typename TMaskInternalImageType::PixelType maskBG, typename TInternalImageType::PixelType outValue, bool inPlace); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template int GetNumberOfConnectedComponentLabels(const ImageType * input, typename ImageType::PixelType BG, bool isFullyConnected); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename TImageType::Pointer Labelize(const TImageType * input, typename TImageType::PixelType BG, bool isFullyConnected, int minimalComponentSize); template typename TImageType::Pointer LabelizeAndCountNumberOfObjects(const TImageType * input, typename TImageType::PixelType BG, bool isFullyConnected, int minimalComponentSize, int & nb); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename ImageType::Pointer RemoveLabels(const ImageType * input, typename ImageType::PixelType BG, std::vector & labelsToRemove); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename ImageType::Pointer AutoCrop(const ImageType * input, typename ImageType::PixelType BG) { typedef clitk::AutoCropFilter AutoCropFilterType; typename AutoCropFilterType::Pointer autoCropFilter = AutoCropFilterType::New(); autoCropFilter->SetInput(input); autoCropFilter->SetBackgroundValue(BG); autoCropFilter->Update(); return autoCropFilter->GetOutput(); } //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename TImageType::Pointer KeepLabels(const TImageType * input, typename TImageType::PixelType BG, typename TImageType::PixelType FG, typename TImageType::PixelType firstKeep, typename TImageType::PixelType lastKeep, bool useLastKeep); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename TImageType::Pointer LabelizeAndSelectLabels(const TImageType * input, typename TImageType::PixelType BG, typename TImageType::PixelType FG, bool isFullyConnected, int minimalComponentSize, LabelizeParameters * param); //-------------------------------------------------------------------- template typename ImageType::Pointer ResizeImageLike(const ImageType * input, const itk::ImageBase * like, typename ImageType::PixelType BG); //-------------------------------------------------------------------- template typename MaskImageType::Pointer SliceBySliceRelativePosition(const MaskImageType * input, const MaskImageType * object, int direction, double threshold, std::string orientation, bool uniqueConnectedComponent=false, double spacing=-1, bool autocropflag=true, bool singleObjectCCL=true); //-------------------------------------------------------------------- // In a binary image, search for the point belonging to the FG that // is the most exterma in the direction 'direction' (or in the // opposite if notFlag is given). template bool FindExtremaPointInAGivenDirection(const ImageType * input, typename ImageType::PixelType bg, int direction, bool opposite, typename ImageType::PointType & p); //-------------------------------------------------------------------- //-------------------------------------------------------------------- // Same as above but check that the found point is not more than // 'distanceMax' away from 'refPoint' template bool FindExtremaPointInAGivenDirection(const ImageType * input, typename ImageType::PixelType bg, int direction, bool opposite, typename ImageType::PointType refPoint, double distanceMax, typename ImageType::PointType & p); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename ImageType::Pointer CropImageAlongOneAxis(const ImageType * image, int dim, double min, double max, bool autoCrop = false, typename ImageType::PixelType BG=0); template typename ImageType::Pointer CropImageAbove(const ImageType * image, int dim, double min, bool autoCrop = false, typename ImageType::PixelType BG=0); template typename ImageType::Pointer CropImageBelow(const ImageType * image, int dim, double max,bool autoCrop = false, typename ImageType::PixelType BG=0); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template void ComputeCentroids(const ImageType * image, typename ImageType::PixelType BG, std::vector & centroids); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template void ExtractSlices(const ImageType * image, int dim, std::vector< typename itk::Image::Pointer > & slices); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename ImageType::Pointer JoinSlices(std::vector::Pointer > & slices, const ImageType * input, int dim); //-------------------------------------------------------------------- //-------------------------------------------------------------------- // Set of tools to manage 3D points and 2D points in slices template class PointsUtils { typedef typename ImageType::PointType PointType3D; typedef typename ImageType::IndexType IndexType3D; typedef typename ImageType::PixelType PixelType; typedef typename ImageType::Pointer ImagePointer; typedef typename ImageType::ConstPointer ImageConstPointer; typedef itk::Image SliceType; typedef typename SliceType::PointType PointType2D; typedef typename SliceType::IndexType IndexType2D; typedef std::map MapPoint2DType; typedef std::vector VectorPoint3DType; public: static void Convert2DTo3D(const PointType2D & p2D, const ImageType * image, const int slice, PointType3D & p3D); static void Convert2DTo3DList(const MapPoint2DType & map, const ImageType * image, VectorPoint3DType & list); }; //-------------------------------------------------------------------- template void WriteListOfLandmarks(std::vector points, std::string filename); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename ImageType::Pointer Dilate(const ImageType * image, double radiusInMM, typename ImageType::PixelType BG, typename ImageType::PixelType FG, bool extendSupport); template typename ImageType::Pointer Dilate(const ImageType * image, typename ImageType::SizeType radius, typename ImageType::PixelType BG, typename ImageType::PixelType FG, bool extendSupport); template typename ImageType::Pointer Dilate(const ImageType * image, typename ImageType::PointType radiusInMM, typename ImageType::PixelType BG, typename ImageType::PixelType FG, bool extendSupport); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template void ConvertOption(std::string optionName, uint given, ValueType * values, VectorType & p, uint dim, bool required); #define ConvertOptionMacro(OPTIONNAME, VAR, DIM, REQUIRED) \ ConvertOption(#OPTIONNAME, OPTIONNAME##_given, OPTIONNAME##_arg, VAR, DIM, REQUIRED); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template void SliceBySliceSetBackgroundFromLineSeparation(ImageType * input, std::vector & lA, std::vector & lB, typename ImageType::PixelType BG, int mainDirection, double offsetToKeep); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template void AndNot(ImageType * input, const ImageType * object, typename ImageType::PixelType BG=0); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename ImageType::Pointer Binarize(const ImageType * input, typename ImageType::PixelType lower, typename ImageType::PixelType upper, typename ImageType::PixelType BG=0, typename ImageType::PixelType FG=1); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template void GetMinMaxPointPosition(const ImageType * input, typename ImageType::PointType & min, typename ImageType::PointType & max); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template typename ImageType::PointType FindExtremaPointInAGivenLine(const ImageType * input, int dimension, bool inverse, typename ImageType::PointType p, typename ImageType::PixelType BG, double distanceMax); //-------------------------------------------------------------------- //-------------------------------------------------------------------- template bool IsOnTheSameLineSide(PointType C, PointType A, PointType B, PointType like); //-------------------------------------------------------------------- } #include "clitkSegmentationUtils.txx" #endif