Refactor canvas blender into rasterizer

Signed-off-by: pacien <pacien.trangirard@pacien.net>

2 files changed, 86 insertions, 39 deletions

diff --git a/src/blender/blender.c b/src/blender/blender.c
deleted file mode 100644
index 08cafa4..0000000
--- a/src/blender/blender.c
+++ /dev/null
@@ -1,39 +0,0 @@
-#include "blender/blender.h"
-#include <assert.h>
-#include <math.h>
-
-static inline ColorComponent blend_components(ColorComponent origin, ColorComponent target, TimeVector frame) {
-  // https://www.youtube.com/watch?v=LKnqECcg6Gw
-  return (ColorComponent) sqrt((TIME_UNIT - frame) * pow(origin, 2) + frame * pow(target, 2));
-}
-
-static inline Color blend_colors(Color origin, Color target, TimeVector frame) {
-  return (Color) {{blend_components(origin.rgba.r, target.rgba.r, frame),
-                   blend_components(origin.rgba.g, target.rgba.g, frame),
-                   blend_components(origin.rgba.b, target.rgba.b, frame),
-                   blend_components(origin.rgba.a, target.rgba.a, frame)}};
-}
-
-void blender_blend_canvas(Canvas *canvas, Canvas *source, Canvas *target, Morphing *morphing, TimeVector frame) {
-  IntVector flat_dim;
-  CartesianVector dim, point;
-  CartesianMapping mapping;
-  Color pixel;
-
-  dim = morphing->dim;
-
-  assert(dim.x > 0 && dim.y > 0);
-  assert(vector_equals(dim, canvas_get_dim(canvas)));
-  assert(vector_equals(dim, canvas_get_dim(source)));
-  assert(vector_equals(dim, canvas_get_dim(target)));
-  assert(frame >= TIME_ORIGIN && frame <= TIME_UNIT);
-
-  for (flat_dim = (dim.x - 1) * (dim.y - 1); flat_dim >= 0; --flat_dim) {
-    point.x = flat_dim % dim.y;
-    point.y = flat_dim / dim.y;
-
-    mapping = (CartesianMapping) {point, point};
-    pixel = blend_colors(canvas_get_pixel(source, mapping.origin), canvas_get_pixel(target, mapping.target), frame);
-    canvas_set_pixel(canvas, point, pixel);
-  }
-}
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;
+}