71 lines
1.7 KiB
C++
71 lines
1.7 KiB
C++
/*
|
|
* DoRayMe - a quick and dirty Raytracer
|
|
* Canvas implementation
|
|
*
|
|
* Created by Manoël Trapier
|
|
* Copyright (c) 2020 986-Studio.
|
|
*
|
|
*/
|
|
|
|
#include <canvas.h>
|
|
#include <lodepng.h>
|
|
|
|
#define BPP (24)
|
|
#define BytePP (BPP / 8)
|
|
|
|
#define MIN(_a, _b) ((_a)<(_b)?(_a):(_b))
|
|
#define MAX(_a, _b) ((_a)>(_b)?(_a):(_b))
|
|
|
|
Canvas::Canvas(uint32_t width, uint32_t height) : width(width), height(height)
|
|
{
|
|
this->bitmap = (uint8_t *)calloc(4, width * height);
|
|
this->stride = BytePP * width;
|
|
}
|
|
|
|
Canvas::Canvas(const Canvas &b)
|
|
{
|
|
this->width = b.width;
|
|
this->height = b.height;
|
|
this->stride = b.stride;
|
|
this->bitmap = (uint8_t *)calloc(4, b.width * b.height);
|
|
memcpy(this->bitmap, b.bitmap, 4 * b.width * b.height);
|
|
}
|
|
|
|
Canvas::Canvas(const Canvas *b)
|
|
{
|
|
this->width = b->width;
|
|
this->height = b->height;
|
|
this->stride = b->stride;
|
|
this->bitmap = (uint8_t *)calloc(4, b->width * b->height);
|
|
memcpy(this->bitmap, b->bitmap, 4 * b->width * b->height);
|
|
}
|
|
|
|
Canvas::~Canvas()
|
|
{
|
|
if (this->bitmap != nullptr)
|
|
{
|
|
free(this->bitmap);
|
|
}
|
|
}
|
|
|
|
void Canvas::putPixel(uint32_t x, uint32_t y, Tuple colour)
|
|
{
|
|
uint32_t offset = y * this->stride + x * BytePP;
|
|
this->bitmap[offset + 0] = MAX(MIN(colour.x * 255, 255), 0);
|
|
this->bitmap[offset + 1] = MAX(MIN(colour.y * 255, 255), 0);
|
|
this->bitmap[offset + 2] = MAX(MIN(colour.z * 255, 255), 0);
|
|
}
|
|
|
|
Colour Canvas::getPixel(uint32_t x, uint32_t y)
|
|
{
|
|
uint32_t offset = y * this->stride + x * BytePP;
|
|
return Colour(this->bitmap[offset + 0] / 255.0, this->bitmap[offset + 1] / 255.0, this->bitmap[offset + 2] / 255.0);
|
|
}
|
|
|
|
bool Canvas::SaveAsPNG(const char *filename)
|
|
{
|
|
uint32_t ret = lodepng_encode24_file(filename, this->bitmap, this->width, this->height);
|
|
|
|
return ret == 0;
|
|
}
|