COSC-3P93-Project/Step 3/include/engine/image/image.h

84 lines
2.7 KiB
C++

/*
* Created by Brett Terpstra 6920201 on 14/10/22.
* Copyright (c) Brett Terpstra 2022 All Rights Reserved
*/
#ifndef STEP_2_IMAGE_H
#define STEP_2_IMAGE_H
#include "engine/util/std.h"
#include "engine/math/vectors.h"
namespace Raytracing {
// glorified structure to store our image data.
class Image {
private:
unsigned long width;
unsigned long height;
unsigned long _width;
unsigned long _height;
Vec4 *pixelData;
bool m_modified = false;
public:
Image(unsigned long width, unsigned long height);
Image(const Image &image);
Image(const Image&& image) = delete;
std::vector<double> toArray();
void fromArray(double* array, int size, int id);
inline void setPixelColor(unsigned long x, unsigned long y, const Vec4 &color) {
m_modified = true;
pixelData[(x * height) + y] = Vec4{color.r(), color.g(), color.b(), 1.0};
}
[[nodiscard]] inline Vec4 getPixelColor(unsigned long x, unsigned long y) const {
return pixelData[(x * height) + y];
}
[[nodiscard]] inline int getPixelR(int x, int y) const {
return int(255.0 * getPixelColor(x, y).r());
};
[[nodiscard]] inline int getPixelG(int x, int y) const {
return int(255.0 * getPixelColor(x, y).g());
};
[[nodiscard]] inline int getPixelB(int x, int y) const {
return int(255.0 * getPixelColor(x, y).b());
}
[[nodiscard]] inline int getPixelA(int x, int y) const {
return int(255.0 * getPixelColor(x, y).a());
}
[[nodiscard]] inline int getWidth() const { return int(width); }
[[nodiscard]] inline int getHeight() const { return int(height); }
[[nodiscard]] inline bool modified() const { return m_modified; }
~Image();
};
class ImageInput {
private:
int width, height, channels;
unsigned char* data;
public:
explicit ImageInput(const std::string& image);
unsigned long* getImageAsIconBuffer();
unsigned char* getImageData(){return data;}
[[nodiscard]] int getImageSize() const{return width;}
~ImageInput();
};
// image writer class used to output the image to a file.
class ImageOutput {
private:
Image image;
public:
explicit ImageOutput(const Image& image): image(image) {}
virtual void write(const std::string& file, const std::string& formatExtension);
};
}
#endif //STEP_2_IMAGE_H