CRNImageGradient.h

Go to the documentation of this file.

/* Copyright 2006-2016 Yann LEYDIER, CoReNum, INSA-Lyon, ENS-Lyon
 * 
 * This file is part of libcrn.
 * 
 * libcrn is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * libcrn is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with libcrn.  If not, see <http://www.gnu.org/licenses/>.
 * 
 * file: CRNImageGradient.h
 * \author Yann LEYDIER
 */

#ifndef CRNIMAGEGRADIENT_HEADER
#define CRNIMAGEGRADIENT_HEADER

#include <CRNImage/CRNImage2D.h>

namespace crn
{
    /****************************************************************************/
    class ImageGradient : public Image<pixel::Polar2D<unsigned int, Angle<ByteAngle>>>
    {
        public:
#ifdef _MSC_VER

            ImageGradient(): Image<pixel::Polar2D<unsigned int, Angle<ByteAngle>>>() {}
            ImageGradient(size_t w, size_t h, pixel_type val = pixel_type(0)) :
                Image<pixel::Polar2D<unsigned int, Angle<ByteAngle>>>(w, h, val) {}
            ImageGradient(size_t w, size_t h, const pixel_type *data):
                Image<pixel::Polar2D<unsigned int, Angle<ByteAngle>>>(w, h, data) {}
            template<typename Y> explicit ImageGradient(const Image<Y> &img) :
                Image<pixel::Polar2D<unsigned int, Angle<ByteAngle>>>(img) {}
            template<typename Y> ImageGradient(const Image<Y> &img, const Rect &bbox) :
                Image<pixel::Polar2D<unsigned int, Angle<ByteAngle>>>(img, bbox) {}
#else
            using Image<pixel::Polar2D<unsigned int, Angle<ByteAngle>>>::Image;
#endif

            ImageGradient(const ImageGradient&) = default;
            ImageGradient(ImageGradient&&) = default;

            ImageGradient& operator=(const ImageGradient&) = default;
            ImageGradient& operator=(ImageGradient&&) = default;

            virtual ~ImageGradient() override {}

            unsigned int GetMinModule() const noexcept { return thresh; }
            void SetMinModule(unsigned int m) { thresh = m; }
            unsigned int AutoMinModule(const ImageGray &img);

            inline bool IsSignificant(size_t i) const { return At(i).rho >= thresh; }
            inline bool IsSignificant(size_t x, size_t y) const { return At(x, y).rho >= thresh; }

            ImageGray MakeImageGray() const;
            ImageRGB MakeImageRGB(bool thres = false) const;
            ImageGray MakeCurvature() const;
            ImageBW MakeMask() const;

            double GetHRun() const noexcept;
            double GetVRun() const noexcept;
            
        private:
            unsigned int thresh;
    };

    CRN_ALIAS_SMART_PTR(ImageGradient)
}
#endif