#include "pgm.h"

#include <stdio.h>
#include <stdlib.h>

int pgm_write_header(FILE *fp, unsigned int w, unsigned int h)
{
	return fprintf(fp, "P3\n%d %d\n255\n", w, h);
}

int pgm_write_pixel(FILE *fp, color_t *c)
{
	return fprintf(fp, "%d %d %d\n", c->r, c->g, c->b);
}

color_t *color_set(color_t *c, uint8_t r, uint8_t g, uint8_t b)
{
	if (!c) {
		c = (color_t *) malloc(sizeof(c));
	}

	c->r = r;
	c->g = g;
	c->b = b;

	return c;
}

color_t *color_add(color_t *dest, color_t *a, color_t *b)
{
	if (!dest) {
		dest = (color_t *) malloc(sizeof(dest));
	}

	unsigned int tmp = a->r + b->r;
	dest->r = tmp > 255 ? 255 : tmp;

	tmp = a->g + b->g;
	dest->g = tmp > 255 ? 255 : tmp;

	tmp = a->b + b->b;
	dest->b = tmp > 255 ? 255 : tmp;

	return dest;
}

color_t *color_scale(color_t *dest, color_t *a, float b)
{
	if (!dest) {
		dest = (color_t *) malloc(sizeof(dest));
	}

	unsigned int tmp = a->r * b;
	dest->r = tmp > 255 ? 255 : tmp;

	tmp = a->g * b;
	dest->g = tmp > 255 ? 255 : tmp;

	tmp = a->b * b;
	dest->b = tmp > 255 ? 255 : tmp;

	return dest;
}

color_t *color_scale_vector(color_t *dest, color_t *a, vector_t *v)
{
	if (!dest) {
		dest = (color_t *) malloc(sizeof(dest));
	}

	dest->r = a->r * v->x;
	dest->g = a->g * v->y;
	dest->b = a->b * v->z;

	return dest;
}