Liver_vesselSegPreprocessing/PreProcess.h at master · rahulpk/Liver_vesselSegPreprocessing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#ifndef _PreProcess_h
#define _PreProcess_h

#include <itkImage.h>
#include <itkImageFileReader.h>
#include <itkImageFileWriter.h>
#include <itkSigmoidImageFilter.h>
#include <itkRescaleIntensityImageFilter.h>
#include <itkCurvatureAnisotropicDiffusionImageFilter.h>
#include <itkGradientAnisotropicDiffusionImageFilter.h>
#include <itkTimeProbe.h>
#include <itkResampleImageFilter.h>
#include <itkAffineTransform.h>
#include <itkLinearInterpolateImageFunction.h>

#include <itkOpenCLUtil.h>
#include <itkGPUImage.h>
#include <itkGPUKernelManager.h>
#include <itkGPUContextManager.h>
#include <itkGPUImageToImageFilter.h>
#include <itkGPUGradientAnisotropicDiffusionImageFilter.h>


class PreProcess
{

public:

    PreProcess();
    ~PreProcess();

    //Main Run function to run all the functions
    void RunPreProcess(std::string inputFilename, std::string outputFilename, float lowerThreshold, float upperThreshold, float alpha, float beta, float conductance, int numberOfInterations);

protected:

private:

    typedef itk::GPUImage<float, 3>  ImageType;

    //Original Input Image
    ImageType::Pointer OriginalImage;

    //Function for non-linearly remapping the image
    ImageType::Pointer nonLinearIntensityRemap(float lowerThreshold, float upperThreshold, float alpha, float beta);

//    //Function to smooth the image
//    ImageType::Pointer curvatureSmoothImage(ImageType::Pointer remappedImage);

    //Function to smooth the image
    ImageType::Pointer gradientSmoothImage(ImageType::Pointer remappedImage, float conductance, int numberOfInterations);

    ImageType::Pointer resampleImage(ImageType::Pointer smoothedImage);


};


#endif