image.hpp

#pragma once

#include <vector>
#include <string>

struct pixel{
    bool gray;
    double r, g, b;
    double y;
    pixel(double red, double grn, double blu):
        r(red), g(grn), b(blu)
        {
            gray = false;
            //RGB to Luma realtion per ITU BT.601
            //https://stackoverflow.com/a/596241
            y = (0.299 * r) + (0.587 * g) + (0.114 * b);
        }
    pixel(double luma): y(luma), gray(true) {
        r = y;
        g = y;
        b = y;
    }
    pixel():r(0), g(0), b(0), y(0), gray(true) {}
};

class image{
    private:
        std::vector<std::vector<pixel> > data;
        int rows, cols;
        std::string format;
    public:
        image();
        image(int r, int c);
        image(std::vector<std::vector<pixel> >);
        image(const image&);
        int c() const;
        int r() const;
        std::string get_format() const;
        void set_format(std::string);
        const std::vector<pixel>& operator[](size_t i) const;
        std::vector<pixel>& operator[](size_t i);
};