image.cpp

#include "image.h"

#define STB_IMAGE_IMPLEMENTATION
#include "stb_image.h"

#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image_write.h"



Image::Image(const std::string& filename) {
    data = stbi_load(filename.c_str(), &width, &height, &channels, 0);
    if (!data) {
        throw std::runtime_error("Could not open or find the image");
    }
}

Image::Image(const Image& other) {
    width = other.width;
    height = other.height;
    channels = other.channels;
    data = new unsigned char[width * height * channels];
    std::copy(other.data, other.data + (width * height * channels), data);
}

Image::~Image() {
    stbi_image_free(data);
}

unsigned char* Image::getData() {
    return data;
}

int Image::getWidth() const {
    return width;
}

int Image::getHeight() const {
    return height;
}

int Image::getChannels() const {
    return channels;
}

void Image::save(const std::string& filename) {
    stbi_write_jpg(filename.c_str(), width, height, channels, data, 100);
}

bool Image::operator==(const Image& other) const {
    if (width != other.width || height != other.height || channels != other.channels) {
        return false;
    }
    for (int i = 0; i < width * height * channels; ++i) {
        if (data[i] != other.data[i]) {
            return false;
        }
    }
    return true;
}

Image& Image::operator=(const Image& other) {
    if (this == &other) {
        return *this;
    }
    width = other.width;
    height = other.height;
    channels = other.channels;
    data = new unsigned char[width * height * channels];
    std::copy(other.data, other.data + (width * height * channels), data);
    return *this;
}

JPGImage::JPGImage(const std::string& filename) : Image(filename) {}

JPGImage::JPGImage(const JPGImage& other) : Image(other) {}