83 lines
2.3 KiB
C++
83 lines
2.3 KiB
C++
|
#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() { return getData(); }
|
||
|
|
|
||
|
|
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(int width, int height);
|
||
|
|
Bl(const std::vector<uint8_t> &data, int width);
|
||
|
|
Bl(std::vector<uint8_t> &&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 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;
|
||
|
|
private:
|
||
|
|
std::vector<uint8_t> data;
|
||
|
|
int width;
|
||
|
|
int height;
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|