ProcessingUtils.h

Go to the documentation of this file.

#pragma once

#include <vlCore/Image.hpp>


class ProcessingUtils
{
public:
        ProcessingUtils(vl::ref<vl::Image> volume, float sigma=1.0);
        ~ProcessingUtils(void);

        void compute();

        vl::ref<vl::Image> getGradient() { return _gradient; }
        vl::ref<vl::Image> getCurvature() { return _curvature; }
        vl::ref<vl::Image> getSmoothedVolume() { return _smoothedVolume; }

private:
        void computeGradient();
        void computeSecondDerivativeX();
        void computeSecondDerivativeY();
        void computeSecondDerivativeZ();
        void computeCurvature();

        int computeGaussianGPU();
        int computeGradientGPU();
        int computeCurvatureGPU();

        //vl::ref<vl::Image> genGradient(vl::ref<vl::Image> img);

        vl::ref<vl::Image> _volume;
        vl::ref<vl::Image> _smoothedVolume;
        vl::ref<vl::Image> _gradient;
        vl::ref<vl::Image> _curvature;

        float _sigma;
};