overlays converted into render primitives

overviewer_core/src/primitives/overlay-mineral.c

@@ -0,0 +1,150 @@
+/*
+ * This file is part of the Minecraft Overviewer.
+ *
+ * Minecraft Overviewer is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as published
+ * by the Free Software Foundation, either version 3 of the License, or (at
+ * your option) any later version.
+ *
+ * Minecraft Overviewer is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
+ * Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with the Overviewer.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "overlay.h"
+
+typedef struct {
+    /* inherits from overlay */
+    RenderPrimitiveOverlay parent;
+    
+    void *minerals;
+} RenderPrimitiveMineral;
+
+struct MineralColor {
+    unsigned char blockid;
+    unsigned char r, g, b;
+};
+
+/* put more valuable ores first -- they take precedence */
+static struct MineralColor default_minerals[] = {
+    {48 /* Mossy Stone  */, 31, 153, 9},
+    
+    {56 /* Diamond Ore  */, 32, 230, 220},
+
+    {21 /* Lapis Lazuli */, 0, 23, 176},
+    {14 /* Gold Ore     */, 255, 234, 0},
+
+    {15 /* Iron Ore     */, 204, 204, 204},
+    {73 /* Redstone     */, 186, 0, 0},
+    {74 /* Lit Redstone */, 186, 0, 0},
+    {16 /* Coal Ore     */, 54, 54, 54},
+    
+    /* end of list marker */
+    {0, 0, 0, 0}
+};
+
+static void get_color(void *data, RenderState *state,
+                      unsigned char *r, unsigned char *g, unsigned char *b, unsigned char *a) {
+    
+    int x = state->x, y = state->y, z_max = state->z + 1, z;
+    int max_i = -1;
+    RenderPrimitiveMineral* self = (RenderPrimitiveMineral *)data;
+    struct MineralColor *minerals = (struct MineralColor *)(self->minerals);
+    *a = 0;
+    
+    for (z = 0; z <= z_max; z++) {
+        int i, tmp;
+        unsigned char blockid = getArrayByte3D(state->blocks, x, y, z);
+        
+        for (i = 0; (max_i == -1 || i < max_i) && minerals[i].blockid != 0; i++) {
+            if (minerals[i].blockid == blockid) {
+                *r = minerals[i].r;
+                *g = minerals[i].g;
+                *b = minerals[i].b;
+                
+                tmp = (128 - z_max + z) * 2 - 40;
+                *a = MIN(MAX(0, tmp), 255);
+                
+                max_i = i;
+                break;
+            }
+        }
+    }
+}
+
+static int
+overlay_mineral_start(void *data, RenderState *state, PyObject *support) {
+    PyObject *opt;
+    RenderPrimitiveMineral* self;
+
+    /* first, chain up */
+    int ret = primitive_overlay.start(data, state, support);
+    if (ret != 0)
+        return ret;
+    
+    /* now do custom initializations */
+    self = (RenderPrimitiveMineral *)data;
+    
+    if (!render_mode_parse_option(support, "minerals", "O", &(opt)))
+        return 1;
+    if (opt && opt != Py_None) {
+        struct MineralColor *minerals = NULL;
+        Py_ssize_t minerals_size = 0, i;
+        /* create custom minerals */
+        
+        if (!PyList_Check(opt)) {
+            PyErr_SetString(PyExc_TypeError, "'minerals' must be a list");
+            return 1;
+        }
+        
+        minerals_size = PyList_GET_SIZE(opt);
+        minerals = self->minerals = calloc(minerals_size + 1, sizeof(struct MineralColor));
+        if (minerals == NULL) {
+            return 1;
+        }
+        
+        for (i = 0; i < minerals_size; i++) {
+            PyObject *mineral = PyList_GET_ITEM(opt, i);
+            if (!PyArg_ParseTuple(mineral, "b(bbb)", &(minerals[i].blockid), &(minerals[i].r), &(minerals[i].g), &(minerals[i].b))) {
+                free(minerals);
+                self->minerals = NULL;
+                return 1;
+            }
+        }
+    } else {
+        self->minerals = default_minerals;
+    }
+    Py_XDECREF(opt);
+    
+    /* setup custom color */
+    self->parent.get_color = get_color;
+    
+    return 0;
+}
+
+static void
+overlay_mineral_finish(void *data, RenderState *state) {
+    /* first free all *our* stuff */
+    RenderPrimitiveMineral* self = (RenderPrimitiveMineral *)data;
+    
+    if (self->minerals && self->minerals != default_minerals) {
+        free(self->minerals);
+    }
+    
+    /* now, chain up */
+    primitive_overlay.finish(data, state);
+}
+
+RenderPrimitiveInterface primitive_overlay_mineral = {
+    "overlay-mineral",
+    sizeof(RenderPrimitiveMineral),
+    overlay_mineral_start,
+    overlay_mineral_finish,
+    overlay_occluded,
+    NULL,
+    overlay_draw,
+};

overviewer_core/src/primitives/overlay-spawn.c

@@ -0,0 +1,108 @@
+/*
+ * This file is part of the Minecraft Overviewer.
+ *
+ * Minecraft Overviewer is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as published
+ * by the Free Software Foundation, either version 3 of the License, or (at
+ * your option) any later version.
+ *
+ * Minecraft Overviewer is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
+ * Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with the Overviewer.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "overlay.h"
+#include <math.h>
+
+typedef struct {
+    /* inherits from overlay */
+    RenderPrimitiveOverlay parent;
+    
+    PyObject *skylight, *blocklight;
+} RenderPrimitiveSpawn;
+
+static void get_color(void *data, RenderState *state,
+                      unsigned char *r, unsigned char *g, unsigned char *b, unsigned char *a) {
+    
+    RenderPrimitiveSpawn* self = (RenderPrimitiveSpawn *)data;
+    int x = state->x, y = state->y, z = state->z;
+    int z_light = z + 1;
+    unsigned char blocklight, skylight;
+    PyObject *block_py;
+    
+    /* set a nice, pretty red color */
+    *r = 229;
+    *g = 36;
+    *b = 38;
+    
+    /* default to no overlay, until told otherwise */
+    *a = 0;
+    
+    if (block_has_property(state->block, NOSPAWN)) {
+        /* nothing can spawn on this */
+        return;
+    }
+    
+    blocklight = getArrayByte3D(self->blocklight, x, y, MIN(127, z_light));
+    
+    /* if we're at the top, force 15 (brightest!) skylight */
+    if (z_light == 128) {
+        skylight = 15;
+    } else {
+        skylight = getArrayByte3D(self->skylight, x, y, z_light);
+    }
+    
+    if (MAX(blocklight, skylight) <= 7) {
+        /* hostile mobs spawn in daylight */
+        *a = 240;
+    } else if (MAX(blocklight, skylight - 11) <= 7) {
+        /* hostile mobs spawn at night */
+        *a = 150;
+    }
+}
+
+static int
+overlay_spawn_start(void *data, RenderState *state, PyObject *support) {
+    RenderPrimitiveSpawn* self;
+
+    /* first, chain up */
+    int ret = primitive_overlay.start(data, state, support);
+    if (ret != 0)
+        return ret;
+    
+    /* now do custom initializations */
+    self = (RenderPrimitiveSpawn *)data;
+    self->blocklight = get_chunk_data(state, CURRENT, BLOCKLIGHT, 1);
+    self->skylight = get_chunk_data(state, CURRENT, SKYLIGHT, 1);
+    
+    /* setup custom color */
+    self->parent.get_color = get_color;
+    
+    return 0;
+}
+
+static void
+overlay_spawn_finish(void *data, RenderState *state) {
+    /* first free all *our* stuff */
+    RenderPrimitiveSpawn* self = (RenderPrimitiveSpawn *)data;
+    
+    Py_DECREF(self->blocklight);
+    Py_DECREF(self->skylight);
+    
+    /* now, chain up */
+    primitive_overlay.finish(data, state);
+}
+
+RenderPrimitiveInterface primitive_overlay_spawn = {
+    "overlay-spawn",
+    sizeof(RenderPrimitiveSpawn),
+    overlay_spawn_start,
+    overlay_spawn_finish,
+    overlay_occluded,
+    NULL,
+    overlay_draw,
+};

overviewer_core/src/primitives/overlay.c

@@ -0,0 +1,118 @@
+/* 
+ * This file is part of the Minecraft Overviewer.
+ *
+ * Minecraft Overviewer is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as published
+ * by the Free Software Foundation, either version 3 of the License, or (at
+ * your option) any later version.
+ *
+ * Minecraft Overviewer is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
+ * Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with the Overviewer.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "overlay.h"
+
+static void get_color(void *data, RenderState *state,
+                      unsigned char *r, unsigned char *g, unsigned char *b, unsigned char *a) {
+    *r = 200;
+    *g = 200;
+    *b = 255;
+    *a = 155;
+}
+
+static int
+overlay_start(void *data, RenderState *state, PyObject *support) {
+    PyObject *facemasks_py;
+    RenderPrimitiveOverlay *self = (RenderPrimitiveOverlay *)data;
+    
+    self->facemask_top = PyObject_GetAttrString(support, "facemask_top");
+    self->white_color = PyObject_GetAttrString(support, "whitecolor");
+    self->get_color = get_color;
+    
+    return 0;
+}
+
+static void
+overlay_finish(void *data, RenderState *state) {
+    RenderPrimitiveOverlay *self = (RenderPrimitiveOverlay *)data;
+    
+    Py_DECREF(self->facemask_top);
+    Py_DECREF(self->white_color);
+}
+
+int
+overlay_occluded(void *data, RenderState *state, int x, int y, int z) {
+    if ( (x != 0) && (y != 15) && (z != 127) &&
+         !render_mode_hidden(state->rendermode, x-1, y, z) &&
+         !render_mode_hidden(state->rendermode, x, y, z+1) &&
+         !render_mode_hidden(state->rendermode, x, y+1, z) &&
+         !is_transparent(getArrayByte3D(state->blocks, x-1, y, z)) &&
+         !is_transparent(getArrayByte3D(state->blocks, x, y, z+1)) &&
+         !is_transparent(getArrayByte3D(state->blocks, x, y+1, z))) {
+        return 1;
+    }
+    
+    return 0;
+}
+
+void
+overlay_draw(void *data, RenderState *state, PyObject *src, PyObject *mask, PyObject *mask_light) {
+    RenderPrimitiveOverlay *self = (RenderPrimitiveOverlay *)data;
+    unsigned char r, g, b, a;
+
+    // exactly analogous to edge-line code for these special blocks
+    int increment=0;
+    if (state->block == 44)  // half-step
+        increment=6;
+    else if (state->block == 78) // snow
+        increment=9;
+    
+    /* clear the draw space -- set alpha to 0 within mask */
+    tint_with_mask(state->img, 255, 255, 255, 0, mask, state->imgx, state->imgy, 0, 0);
+
+    /* skip rendering the overlay if we can't see it */
+    if (state->z != 127) {
+        unsigned char top_block = getArrayByte3D(state->blocks, state->x, state->y, state->z+1);
+        if (!is_transparent(top_block)) {
+            return;
+        }
+        
+        /* check to be sure this block is solid/fluid */
+        if (block_has_property(top_block, SOLID) || block_has_property(top_block, FLUID)) {
+            
+            /* top block is fluid or solid, skip drawing */
+            return;
+        }
+    }
+    
+    /* check to be sure this block is solid/fluid */
+    if (!block_has_property(state->block, SOLID) && !block_has_property(state->block, FLUID)) {
+        
+        /* not fluid or solid, skip drawing the overlay */
+        return;
+    }
+
+    /* get our color info */
+    self->get_color(data, state, &r, &g, &b, &a);
+    
+    /* do the overlay */
+    if (a > 0) {
+        alpha_over(state->img, self->white_color, self->facemask_top, state->imgx, state->imgy + increment, 0, 0);
+        tint_with_mask(state->img, r, g, b, a, self->facemask_top, state->imgx, state->imgy + increment, 0, 0);
+    }
+}
+
+RenderPrimitiveInterface primitive_overlay = {
+    "overlay",
+    sizeof(RenderPrimitiveOverlay),
+    overlay_start,
+    overlay_finish,
+    overlay_occluded,
+    NULL,
+    overlay_draw,
+};

overviewer_core/src/primitives/overlay.h

@@ -0,0 +1,32 @@
+/* 
+ * This file is part of the Minecraft Overviewer.
+ *
+ * Minecraft Overviewer is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as published
+ * by the Free Software Foundation, either version 3 of the License, or (at
+ * your option) any later version.
+ *
+ * Minecraft Overviewer is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
+ * Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with the Overviewer.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "../overviewer.h"
+
+typedef struct {
+    /* top facemask and white color image, for drawing overlays */
+    PyObject *facemask_top, *white_color;
+    /* can be overridden in derived classes to control
+       overlay alpha and color
+       last four vars are r, g, b, a out */
+    void (*get_color)(void *, RenderState *,
+                      unsigned char *, unsigned char *, unsigned char *, unsigned char *);
+} RenderPrimitiveOverlay;
+extern RenderPrimitiveInterface primitive_overlay;
+
+void overlay_draw(void *data, RenderState *state, PyObject *src, PyObject *mask, PyObject *mask_light);
+int overlay_occluded(void *data, RenderState *state, int x, int y, int z);