danabr75-ashton 0.1.5

data/ext/ashton/shader.c

@@ -0,0 +1,9 @@
+#include "shader.h"
+
+VALUE rb_cShader;
+
+// ----------------------------------------
+void Init_Ashton_Shader(VALUE module)
+{
+    rb_cShader = rb_define_class_under(module, "Shader", rb_cObject);
+}

data/ext/ashton/shader.h

@@ -0,0 +1,16 @@
+/*
+ * class Ashton::Shader
+ *
+ *
+ */
+
+
+#ifndef ASHTON_SHADER_H
+#define ASHTON_SHADER_H
+
+#include "common.h"
+
+void Init_Ashton_Shader(VALUE module);
+
+#endif // ASHTON_SHADER_H
+

data/ext/ashton/texture.c

@@ -0,0 +1,442 @@
+#include "texture.h"
+
+VALUE rb_cTexture;
+
+// Helpers
+static VALUE texture_allocate(VALUE klass);
+static void texture_mark(Texture* texture);
+static void texture_free(Texture* texture);
+static void ensure_cache_exists(VALUE self, Texture* texture);
+static void ensure_fbo_exists(Texture* texture);
+
+void Init_Ashton_Texture(VALUE module)
+{
+    rb_cTexture = rb_define_class_under(module, "Texture", rb_cObject);
+
+    rb_define_alloc_func(rb_cTexture, texture_allocate);
+
+    rb_define_protected_method(rb_cTexture, "initialize_", Ashton_Texture_init, 3);
+    rb_define_protected_method(rb_cTexture, "render_", Ashton_Texture_render, 0);
+
+    rb_define_method(rb_cTexture, "cache", Ashton_Texture_get_cache, 0);
+
+    rb_define_method(rb_cTexture, "width", Ashton_Texture_get_width, 0);
+    rb_define_method(rb_cTexture, "height", Ashton_Texture_get_height, 0);
+    rb_define_method(rb_cTexture, "fbo_id", Ashton_Texture_get_fbo_id, 0);
+    rb_define_method(rb_cTexture, "id", Ashton_Texture_get_id, 0);
+
+    rb_define_method(rb_cTexture, "[]", Ashton_Texture_get_pixel, 2);
+    rb_define_method(rb_cTexture, "rgba", Ashton_Texture_get_rgba_array, 2);
+    rb_define_method(rb_cTexture, "red", Ashton_Texture_get_red, 2);
+    rb_define_method(rb_cTexture, "green", Ashton_Texture_get_green, 2);
+    rb_define_method(rb_cTexture, "blue", Ashton_Texture_get_blue, 2);
+    rb_define_method(rb_cTexture, "alpha", Ashton_Texture_get_alpha, 2);
+
+    rb_define_method(rb_cTexture, "transparent?", Ashton_Texture_is_transparent, 2);
+    rb_define_method(rb_cTexture, "refresh_cache", Ashton_Texture_refresh_cache, 0);
+    rb_define_method(rb_cTexture, "to_blob", Ashton_Texture_to_blob, 0);
+    rb_define_method(rb_cTexture, "draw", Ashton_Texture_draw, -1);
+}
+
+//
+VALUE Ashton_Texture_get_width(VALUE self)
+{
+    TEXTURE();
+    return UINT2NUM(texture->width);
+}
+
+//
+VALUE Ashton_Texture_get_height(VALUE self)
+{
+    TEXTURE();
+    return UINT2NUM(texture->height);
+}
+
+//
+static void ensure_fbo_exists(Texture* texture)
+{
+    if(!texture->fbo_id)
+    {
+        glGenFramebuffersEXT(1, &texture->fbo_id);
+
+        // Bind our already existing texture to the FBO.
+        glBindFramebufferEXT(GL_FRAMEBUFFER_EXT, texture->fbo_id);
+        glFramebufferTexture2DEXT(GL_FRAMEBUFFER_EXT, GL_COLOR_ATTACHMENT0_EXT,
+                                  GL_TEXTURE_2D, texture->id, 0);
+        glBindFramebufferEXT(GL_FRAMEBUFFER_EXT, 0);
+    }
+}
+
+VALUE Ashton_Texture_get_fbo_id(VALUE self)
+{
+    TEXTURE();
+    ensure_fbo_exists(texture);
+    return UINT2NUM(texture->fbo_id);
+}
+
+VALUE Ashton_Texture_get_id(VALUE self)
+{
+    TEXTURE();
+    return UINT2NUM(texture->id);
+}
+
+//
+VALUE Ashton_Texture_init(VALUE self, VALUE width, VALUE height, VALUE blob)
+{
+    TEXTURE();
+
+    if(!GL_EXT_framebuffer_object)
+    {
+       rb_raise(rb_eRuntimeError, "Ashton::Texture requires GL_EXT_framebuffer_object, which is not supported by OpenGL");
+    }
+
+    texture->width = NUM2UINT(width);
+    texture->height = NUM2UINT(height);
+
+    texture->rb_cache = Qnil;
+
+    // Create the texture.
+    glGenTextures(1, &texture->id);
+    glBindTexture(GL_TEXTURE_2D, texture->id);
+
+    glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
+    glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
+    // MAG_FILTER set on each draw
+    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
+
+    // Create the texture, either undefined or based on RGBA blob data string.
+    char* data;
+    if(NIL_P(blob))
+    {
+       data = NULL; // Create an empty texture, that might be filled with junk.
+    }
+    else
+    {
+       data = StringValuePtr(blob); // Create from blob data.
+    }
+
+    glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA8, texture->width,
+                        texture->height, 0, GL_RGBA, GL_UNSIGNED_BYTE, data);
+
+    // Ensure the texture was created.
+    GLint created_width;
+    glGetTexLevelParameteriv(GL_TEXTURE_2D, 0, GL_TEXTURE_WIDTH, &created_width);
+    if(created_width == 0) {
+      rb_raise(rb_eArgError, "Unable to create a texture of size %dx%d",
+               texture->width, texture->height);
+    }
+
+    // Be as lazy as possible in creating the actual FBO :)
+
+    return Qnil;
+}
+
+VALUE Ashton_Texture_render(VALUE self)
+{
+    TEXTURE();
+
+    ensure_fbo_exists(texture);
+    glBindFramebufferEXT(GL_FRAMEBUFFER_EXT, texture->fbo_id);
+
+    return Qnil;
+}
+
+//
+static VALUE texture_allocate(VALUE klass)
+{
+    Texture* texture = ALLOC(Texture);
+    memset(texture, 0, sizeof(Texture));
+
+    return Data_Wrap_Struct(klass, texture_mark, texture_free, texture);
+}
+
+//
+static void texture_mark(Texture* texture)
+{
+    if(!NIL_P(texture->rb_cache)) rb_gc_mark(texture->rb_cache);
+}
+
+//
+static void texture_free(Texture* texture)
+{
+    // FBO might never have been created.
+    if(texture->fbo_id) glDeleteFramebuffersEXT(1, &texture->fbo_id);
+
+    glDeleteTextures(1, &texture->id);
+    xfree(texture);
+}
+
+//
+void ensure_cache_exists(VALUE self, Texture* texture)
+{
+    if(NIL_P(texture->rb_cache))
+    {
+        texture->rb_cache = rb_funcall(rb_cPixelCache, rb_intern("new"), 1, self);
+    }
+}
+
+// Returns the cache. Creates it if necessary.
+VALUE Ashton_Texture_get_cache(VALUE self)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return texture->rb_cache;
+}
+
+//
+VALUE Ashton_Texture_get_pixel(VALUE self, VALUE x, VALUE y)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_get_pixel(texture->rb_cache, x, y);
+}
+
+VALUE Ashton_Texture_get_rgba_array(VALUE self, VALUE x, VALUE y)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_get_rgba_array(texture->rb_cache, x, y);
+}
+
+//
+VALUE Ashton_Texture_get_red(VALUE self, VALUE x, VALUE y)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_get_red(texture->rb_cache, x, y);
+}
+
+//
+VALUE Ashton_Texture_get_green(VALUE self, VALUE x, VALUE y)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_get_green(texture->rb_cache, x, y);
+}
+
+//
+VALUE Ashton_Texture_get_blue(VALUE self, VALUE x, VALUE y)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_get_blue(texture->rb_cache, x, y);
+}
+
+//
+VALUE Ashton_Texture_get_alpha(VALUE self, VALUE x, VALUE y)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_get_alpha(texture->rb_cache, x, y);
+}
+
+//
+VALUE Ashton_Texture_is_transparent(VALUE self, VALUE x, VALUE y)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_is_transparent(texture->rb_cache, x, y);
+}
+
+VALUE Ashton_Texture_refresh_cache(VALUE self)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_refresh(texture->rb_cache);
+}
+
+VALUE Ashton_Texture_to_blob(VALUE self)
+{
+    TEXTURE();
+    ensure_cache_exists(self, texture);
+    return Ashton_PixelCache_to_blob(texture->rb_cache);
+}
+
+// ----------------------------------------
+static VALUE draw_block(VALUE yield_value, VALUE parameters, int argc, VALUE argv[])
+{
+    VALUE self = rb_ary_entry(parameters, 0);
+    float x = (float)NUM2DBL(rb_ary_entry(parameters, 1));
+    float y = (float)NUM2DBL(rb_ary_entry(parameters, 2));
+    VALUE blend_mode = rb_ary_entry(parameters, 3);
+    VALUE color = rb_ary_entry(parameters, 4);
+    VALUE shader = rb_ary_entry(parameters, 5);
+    VALUE multitexture = rb_ary_entry(parameters, 6);
+    VALUE pixelated = rb_ary_entry(parameters, 7);
+
+    TEXTURE(); // Uses 'self' value, so can't be at start of function.
+
+    if(!NIL_P(shader))
+    {
+        VALUE options = rb_hash_new();
+        rb_hash_aset(options, SYMBOL("required"), Qfalse);
+
+        VALUE location = rb_funcall(shader, rb_intern("send"), 3, SYMBOL("uniform_location"), SYMBOL("texture_enabled"), options);
+        if(FIX2INT(location) != -1)
+        {
+            rb_funcall(shader, rb_intern("send"), 3, SYMBOL("set_uniform"), location, Qtrue);
+        }
+        rb_funcall(shader, rb_intern("color="), 1, color);
+    }
+
+    glEnable(GL_BLEND);
+    glEnable(GL_TEXTURE_2D);
+    glBindTexture(GL_TEXTURE_2D, texture->id);
+
+    // Pixelate or smooth, depending on preference.
+    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER,
+        RTEST(pixelated) ? GL_NEAREST : GL_LINEAR);
+
+    // Set blending mode.
+    // Don't need an 'else' clause, since we checked before.
+    ID blend_id = SYM2ID(blend_mode);
+    if(blend_id == rb_intern(DRAW_MODE_ALPHA_BLEND))
+    {
+        glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
+    }
+    else if(blend_id == rb_intern(DRAW_MODE_ADD))
+    {
+        glBlendFunc(GL_ONE, GL_ONE);
+    }
+    else if(blend_id == rb_intern(DRAW_MODE_MULTIPLY))
+    {
+        glBlendFunc(GL_DST_COLOR, GL_ZERO);
+    }
+    else if(blend_id == rb_intern(DRAW_MODE_REPLACE))
+    {
+        glBlendFunc(GL_ONE, GL_ZERO);
+    }
+
+    // Enable multitexturing.
+    if(!NIL_P(multitexture))
+    {
+        uint texture_id = NUM2UINT(rb_funcall(multitexture, rb_intern("id"), 0));
+        glActiveTexture(GL_TEXTURE1);
+        glBindTexture(GL_TEXTURE_2D, texture_id);
+    }
+
+    glBegin(GL_QUADS);
+        glTexCoord2d(0.0, 0.0);
+        glMultiTexCoord2d(GL_TEXTURE1, 0.0, 0.0);
+        glVertex2d(x, y + texture->height); // BL
+
+        glTexCoord2d(0.0, 1.0);
+        glMultiTexCoord2d(GL_TEXTURE1, 0.0, 1.0);
+        glVertex2d(x, y); // TL
+
+        glTexCoord2d(1.0, 1.0);
+        glMultiTexCoord2d(GL_TEXTURE1, 1.0, 1.0);
+        glVertex2d(x + texture->width, y); // TR
+
+        glTexCoord2d(1.0, 0.0);
+        glMultiTexCoord2d(GL_TEXTURE1, 1.0, 0.0);
+        glVertex2d(x + texture->width, y + texture->height); // BR
+    glEnd();
+
+    // Disable multitexturing.
+    if(!NIL_P(multitexture))
+    {
+        glBindTexture(GL_TEXTURE_2D, 0);
+        glActiveTexture(GL_TEXTURE0);
+    }
+
+    return Qnil;
+}
+
+// ----------------------------------------
+VALUE Ashton_Texture_draw(int argc, VALUE argv[], VALUE self)
+{
+    // TEXTURE(); // This isn't actually needed in this function.
+
+    VALUE x, y, z, options;
+    VALUE shader, blend_mode, color, multitexture, pixelated;
+
+    if(rb_scan_args(argc, argv, "31", &x, &y, &z, &options) == 4)
+    {
+        // Get :shader
+        shader = rb_hash_aref(options, SYMBOL("shader"));
+        if(!NIL_P(shader) && !rb_obj_is_kind_of(shader, rb_cShader))
+        {
+            rb_raise(rb_eTypeError, "Expected :shader option of type Ashton::Shader");
+        }
+
+        // Get draw :mode
+        blend_mode = rb_hash_aref(options, SYMBOL("mode"));
+        if(NIL_P(blend_mode))
+        {
+            blend_mode = SYMBOL(DRAW_MODE_ALPHA_BLEND);
+        }
+        else
+        {
+            Check_Type(blend_mode, T_SYMBOL);
+
+            ID blend_id = SYM2ID(blend_mode);
+            if(!(blend_id == rb_intern(DRAW_MODE_ALPHA_BLEND) ||
+                 blend_id == rb_intern(DRAW_MODE_ADD) ||
+                 blend_id == rb_intern(DRAW_MODE_MULTIPLY) ||
+                 blend_id == rb_intern(DRAW_MODE_REPLACE)))
+            {
+               rb_raise(rb_eArgError, "Unsupported draw :mode, :%s", rb_id2name(blend_id));
+            }
+        }
+
+        // Get :color
+        color = rb_hash_aref(options, SYMBOL("color"));
+        if(NIL_P(color))
+        {
+            color = UINT2NUM(0xffffffff);
+        }
+        else if(!rb_obj_is_kind_of(color, rb_cColor))
+        {
+            rb_raise(rb_eTypeError, "Expecting :color to be a Gosu::Color");
+        }
+
+        // Get :multitexture
+        multitexture = rb_hash_aref(options, SYMBOL("multitexture"));
+        if(!NIL_P(multitexture) && !rb_obj_is_kind_of(multitexture, rb_cTexture))
+        {
+            rb_raise(rb_eTypeError, "Expected :multitexture option of type Ashton::Texture");
+        }
+
+        // Get :pixelated
+        pixelated = rb_hash_aref(options, SYMBOL("pixelated"));
+        if(NIL_P(pixelated))
+        {
+            pixelated = rb_funcall(rb_cTexture, rb_intern("pixelated?"), 0);
+        }
+    }
+    else
+    {
+       shader = multitexture = Qnil;
+       blend_mode = SYMBOL(DRAW_MODE_ALPHA_BLEND);
+       color = UINT2NUM(0xffffffff);
+       pixelated = rb_funcall(rb_cTexture, rb_intern("pixelated?"), 0);
+    }
+
+    // Enable the shader, if provided.
+    if(!NIL_P(shader)) rb_funcall(shader, rb_intern("enable"), 1, z);
+
+    // Create a GL block to do the actual drawing.
+    VALUE window = rb_gv_get("$window");
+
+    VALUE block_argv[1];
+    block_argv[0] = z;
+
+    VALUE parameters = rb_ary_new();
+    rb_ary_push(parameters, self);
+    rb_ary_push(parameters, x);
+    rb_ary_push(parameters, y);
+    rb_ary_push(parameters, blend_mode);
+    rb_ary_push(parameters, color);
+    rb_ary_push(parameters, shader);
+    rb_ary_push(parameters, multitexture);
+    rb_ary_push(parameters, pixelated);
+
+    rb_block_call(window, rb_intern("gl"), 1, block_argv,
+                  RUBY_METHOD_FUNC(draw_block), parameters);
+
+    // Disable the shader, if provided.
+    if(!NIL_P(shader)) rb_funcall(shader, rb_intern("disable"), 1, z);
+
+    return Qnil;
+}