#ifndef DITHERING_IMAGE_HPP
#define DITHERING_IMAGE_HPP

#include <cstdint>
#include <vector>
#include <string>

namespace image {
    enum class color_type {
        Black,
        Red,
        Green,
        Blue,
        Alpha,
    };

    enum class file_type {
        PBM,
        PGM,
        PPM,
        // TODO PNG support
    };

    class Base {
    public:
        Base() = default;
        virtual ~Base() {}

        Base(const Base &other) = default;
        Base(Base &&other) = default;

        Base& operator=(const Base &other) = default;
        Base& operator=(Base &&other) = default;

        virtual void randomize() = 0;

        virtual int getSize() = 0;
        virtual uint8_t* getData() = 0;
        virtual const uint8_t* getDataC() const = 0;

        virtual int getTypesCount() = 0;
        virtual std::vector<color_type> getTypes() = 0;
        virtual int getTypeStride(color_type type) = 0;

        virtual bool canWriteFile(file_type type) = 0;
        virtual bool writeToFile(file_type type, bool canOverwrite, const char *filename) = 0;
        virtual bool writeToFile(file_type type, bool canOverwrite, const std::string &filename) = 0;
    };

    class Bl : public Base {
    public:
        Bl();
        Bl(int width, int height);
        Bl(const std::vector<uint8_t> &data, int width);
        Bl(std::vector<uint8_t> &&data, int width);
        Bl(const std::vector<float> &data, int width);
        virtual ~Bl() {}

        Bl(const Bl &other) = default;
        Bl(Bl &&other) = default;

        Bl& operator=(const Bl &other) = default;
        Bl& operator=(Bl &&other) = default;

        virtual void randomize() override;

        virtual int getSize() override;
        virtual uint8_t* getData() override;
        virtual const uint8_t* getDataC() const override;

        virtual int getTypesCount() override { return 1; }
        virtual std::vector<color_type> getTypes() override { return { color_type::Black }; }
        virtual int getTypeStride(color_type) override { return 0; }

        virtual bool canWriteFile(file_type type) override;
        virtual bool writeToFile(file_type type, bool canOverwrite, const char *filename) override;
        virtual bool writeToFile(file_type type, bool canOverwrite, const std::string &filename) override;
        virtual bool isValid() const;
    private:
        std::vector<uint8_t> data;
        int width;
        int height;
    };
}

#endif