Merge remote-tracking branch 'origin/master' into gui

3 files changed, 139 insertions, 0 deletions

diff --git a/src/painter/canvas.c b/src/painter/canvas.c
new file mode 100644
index 0000000..306dc9c
--- /dev/null
+++ b/src/painter/canvas.c
@@ -0,0 +1,33 @@
+#include "painter/canvas.h"
+#include "common/mem.h"
+
+Canvas *canvas_create(IntVector width, IntVector height) {
+  Canvas *c = malloc_or_die(sizeof(Canvas));
+  c->mlv = MLV_create_image(width, height);
+  return c;
+}
+
+Canvas *canvas_create_from_image(const char *fpath) {
+  Canvas *c = malloc_or_die(sizeof(Canvas));
+  c->mlv = MLV_load_image(fpath);
+  return c;
+}
+
+void canvas_destroy(Canvas *c) {
+  MLV_free_image(c->mlv);
+  free(c);
+}
+
+void canvas_set_pixel(Canvas *c, CartesianVector pos, Color color) {
+  MLV_set_pixel_on_image(pos.x, pos.y, color.mlv, c->mlv);
+}
+
+Color canvas_get_pixel(Canvas *c, CartesianVector pos) {
+  int r, g, b, a;
+  MLV_get_pixel_on_image(c->mlv, pos.x, pos.y, &r, &g, &b, &a);
+  return (Color) {{a, b, g, r}};
+}
+
+CartesianVector canvas_get_dim(Canvas *c) {
+  return (CartesianVector) {MLV_get_image_width(c->mlv), MLV_get_image_height(c->mlv)};
+}
diff --git a/src/painter/color.c b/src/painter/color.c
new file mode 100644
index 0000000..65c4f20
--- /dev/null
+++ b/src/painter/color.c
@@ -0,0 +1,20 @@
+#include "painter/color.h"
+#include <math.h>
+
+static inline ColorComponent blend_component(ColorComponent origin, ColorComponent target, TimeVector frame) {
+  return (ColorComponent) round(sqrt((TIME_UNIT - frame) * pow(origin, 2) + frame * pow(target, 2)));
+}
+
+bool color_equals(Color c1, Color c2) {
+  return c1.rgba.r == c2.rgba.r &&
+         c1.rgba.g == c2.rgba.g &&
+         c1.rgba.b == c2.rgba.b &&
+         c1.rgba.a == c2.rgba.a;
+}
+
+Color color_blend(Color origin, Color target, TimeVector distance) {
+  return (Color) {{blend_component(origin.rgba.a, target.rgba.a, distance),
+                   blend_component(origin.rgba.b, target.rgba.b, distance),
+                   blend_component(origin.rgba.g, target.rgba.g, distance),
+                   blend_component(origin.rgba.r, target.rgba.r, distance)}};
+}
diff --git a/src/painter/rasterizer.c b/src/painter/rasterizer.c
new file mode 100644
index 0000000..881920a
--- /dev/null
+++ b/src/painter/rasterizer.c
@@ -0,0 +1,86 @@
+#include "painter/rasterizer.h"
+#include <math.h>
+#include <assert.h>
+
+static inline IntVector i(double (*f)(double, double), RealVector a, RealVector b) {
+  return (IntVector) floor(f(a, b));
+}
+
+static inline CartesianVector vertex_at_frame(CartesianMapping m, TimeVector t) {
+  return (CartesianVector) {(IntVector) round((TIME_UNIT - t) * m.origin.x + t * m.target.x),
+                            (IntVector) round((TIME_UNIT - t) * m.origin.y + t * m.target.y)};
+}
+
+static inline RealVector slope(CartesianVector a, CartesianVector b) {
+  return (((RealVector) b.x) - ((RealVector) a.x)) / (((RealVector) b.y) - ((RealVector) a.y));
+}
+
+static inline int positive_y_vertex_comparator(const void *l, const void *r) {
+  return ((CartesianVector *) l)->y - ((CartesianVector *) r)->y;
+}
+
+static inline Color color_at(Canvas *c, Triangle ref, BarycentricVector b) {
+  CartesianVector v = barycentric_to_cartesian(ref, b);
+  return canvas_get_pixel(c, v);
+}
+
+static inline TriangleContext build_triangle_context(Triangle current, TriangleMap *map) {
+  TriangleContext c;
+  int cursor;
+
+  for (cursor = 0; cursor < 3; ++cursor) {
+    c.current.v[cursor] = current.v[cursor];
+    c.source.v[cursor] = map->vertices[cursor].origin;
+    c.target.v[cursor] = map->vertices[cursor].target;
+  }
+
+  return c;
+}
+
+static inline void draw_pixel(CartesianVector pos, TriangleContext *tctx, RasterizationContext *rctx) {
+  BarycentricVector b = cartesian_to_barycentric(tctx->current, pos);
+  Color c = color_blend(color_at(rctx->source, tctx->source, b), color_at(rctx->target, tctx->target, b), rctx->frame);
+  canvas_set_pixel(rctx->result, pos, c);
+}
+
+static inline void draw_flat_triangle(IntVector top, IntVector bottom,
+                                      RealVector dx1, RealVector dx2, RealVector *x1, RealVector *x2,
+                                      TriangleContext *tctx, RasterizationContext *rctx) {
+
+  IntVector y, l, r;
+  for (y = top; y <= bottom; ++y, *x1 += dx1, *x2 += dx2)
+    for (l = i(fmin, *x1, *x2), r = i(fmax, *x1, *x2); l <= r; ++l)
+      draw_pixel(v(l, y), tctx, rctx);
+}
+
+static inline void draw_triangle(TriangleMap *t, RasterizationContext *rctx) {
+  Triangle triangle = {{vertex_at_frame(t->vertices[0], rctx->frame),
+                        vertex_at_frame(t->vertices[1], rctx->frame),
+                        vertex_at_frame(t->vertices[2], rctx->frame)}};
+
+  TriangleContext tctx = build_triangle_context(triangle, t);
+  CartesianVector *v = triangle.v;
+  qsort(v, 3, sizeof(CartesianVector), positive_y_vertex_comparator);
+
+  {
+    RealVector dx1 = slope(v[0], v[1]), dx2 = slope(v[0], v[2]), dx3 = slope(v[1], v[2]);
+    RealVector x1 = v[0].x, x2 = v[0].x, x3 = v[1].x;
+
+    draw_flat_triangle(v[0].y, v[1].y - 1, dx1, dx2, &x1, &x2, &tctx, rctx);
+    draw_flat_triangle(v[1].y, v[2].y, dx2, dx3, &x2, &x3, &tctx, rctx);
+  }
+}
+
+Canvas *rasterize(Canvas *source, Canvas *target, Morphing *m, TimeVector frame) {
+  RasterizationContext rctx;
+  TriangleMap *t;
+
+  assert(source != NULL && target != NULL && m != NULL);
+  assert(vector_equals(canvas_get_dim(source), m->dim) && vector_equals(canvas_get_dim(target), m->dim));
+  assert(frame >= TIME_ORIGIN && frame <= TIME_UNIT);
+
+  rctx = (RasterizationContext) {canvas_create(m->dim.y, m->dim.x), source, target, frame};
+  for (t = m->first; t != NULL; t = t->next) draw_triangle(t, &rctx);
+
+  return rctx.result;
+}