include/FCam/SharpnessMap.h

Go to the documentation of this file.

#ifndef FCAM_SHARPNESS_MAP_H
#define FCAM_SHARPNESS_MAP_H

#include <vector>

namespace FCam {
    class SharpnessMapConfig {
      public:
        SharpnessMapConfig() : size(0, 0), enabled(false) {}

        Size size;

        bool enabled;

        bool operator==(const SharpnessMapConfig &other) const {
            if (enabled != other.enabled) return false;
            if (enabled && size != other.size) return false;
            return true;
        }

        bool operator!=(const SharpnessMapConfig &other) const {
            return !((*this) == other);
        }
    };    


    class SharpnessMap {
    private:
        Size _size;
        unsigned _channels;
        std::vector<unsigned> _data;

    public:

        SharpnessMap() : _size(0, 0), _channels(0) {
        }

        SharpnessMap(Size s, int channels) : _size(s), _channels(channels) {
            _data.resize(s.width*s.height*channels);            
        }

        unsigned operator()(int x, int y, int c) const {
            return _data[(y*_size.width+x)*_channels + c];
        }

        unsigned &operator()(int x, int y, int c) {
            return _data[(y*_size.width+x)*_channels + c];
        }

        unsigned operator()(int x, int y) const {
            unsigned total = 0;
            for (size_t c = 0; c < _channels; c++) {
                total += (*this)(x, y, c);
            }
            return total;
        }       

        bool valid() const {return _data.size() != 0;}

        unsigned channels() const {return _channels;}

        const Size &size() const {return _size;} 

        int height() const {return _size.height;}

        int width() const {return _size.width;}

        unsigned *data() {return &_data[0];}
    };
}

#endif