shapelib 0.7.2 → 0.8.0

data/ext/shapelib_ext/extconf.rb

@@ -3,17 +3,21 @@ require 'mkmf'
 
 HEADER_DIRS = [
   '/opt/local/include',
+  '/opt/homebrew/include',
   '/usr/local/include',
   '/usr/include',
 ]
 
 LIB_DIRS = [
   '/opt/local/lib',
+  '/opt/homebrew/lib',
   '/usr/local/lib',
   '/usr/lib',
 ]
 
 dir_config('shapelib', HEADER_DIRS, LIB_DIRS)
-have_header('shapefil.h') or have_header('libshp/shapefil.h') or raise 'shapefil.h not found'
+header = %w[shapefil.h libshp/shapefil.h].find { |name| have_header(name) }
+raise 'shapefil.h not found' unless header
 have_library('shp') or raise 'libshp not found'
+have_const('FTDate', header)
 create_makefile('shapelib_ext')

data/ext/shapelib_ext/main.c

@@ -45,6 +45,7 @@ static void shape_free(void *ptr)
 static void shapefile_free(void *ptr)
 {
     shapefile_close(ptr);
+    xfree(ptr);
 }
 
 static const rb_data_type_t shape_data_type = {
@@ -55,6 +56,8 @@ static const rb_data_type_t shapefile_data_type = {
     "Shapelib::ShapeFile", { 0, shapefile_free, 0 }
 };
 
+static VALUE WrapShape(VALUE klass, shape_t *shape);
+
 #include "valconv.h"
 #include "sfcode.h"
 #include "sflist.h"

data/ext/shapelib_ext/sfcode.h

@@ -19,7 +19,7 @@ static VALUE sf_add_field(int argc, VALUE *args, VALUE obj)
 	rb_raise(rb_eArgError, "specify decimal places for Float attribute");
     }
     TypedData_Get_Struct(obj, shapefile_t, &shapefile_data_type, self);
-    r = shapefile_add_field(self, RSTRING_PTR(name), atype, width, wdeci);
+    r = shapefile_add_field(self, StringValueCStr(name), atype, width, wdeci);
     return obj;
 }
 
@@ -44,7 +44,7 @@ static int Ruby2FieldIndex(shapefile_t *sfile, VALUE field)
     if (FIXNUM_P(field)) {
 	return FIX2INT(field);
     } else {
-	return shapefile_field_index(sfile, StringValuePtr(field));
+	return shapefile_field_index(sfile, StringValueCStr(field));
     }
 }
 
@@ -129,7 +129,7 @@ static VALUE sf_maxbound(VALUE obj)
     z = bound[2];
     bound[2] = bound[3];
     bound[3] = z;
-    return DoubleUnpack(bound, 4);
+    return DoubleUnpackM(bound, 4, 2);
 }
 
 static VALUE sf_minbound(VALUE obj)
@@ -142,14 +142,14 @@ static VALUE sf_minbound(VALUE obj)
     z = bound[2];
     bound[2] = bound[3];
     bound[3] = z;
-    return DoubleUnpack(bound, 4);
+    return DoubleUnpackM(bound, 4, 2);
 }
 
 static VALUE sf_read(int argc, VALUE *args, VALUE obj)
 {
     shapefile_t *self;
     shape_t *s;
-    VALUE ishape, r;
+    VALUE ishape;
     int irec;
     rb_scan_args(argc, args, "01", &ishape);
     TypedData_Get_Struct(obj, shapefile_t, &shapefile_data_type, self);
@@ -157,8 +157,7 @@ static VALUE sf_read(int argc, VALUE *args, VALUE obj)
     s = shapefile_read(self, irec);
     if (s == NULL)
 	return Qnil;
-    r = TypedData_Wrap_Struct(cShape, &shape_data_type, s);
-    return r;
+    return WrapShape(cShape, s);
 }
 
 static VALUE sf_rewind(VALUE obj)
@@ -210,7 +209,7 @@ static VALUE sf_write(int argc, VALUE *args, VALUE obj)
     }
     TypedData_Get_Struct(obj, shapefile_t, &shapefile_data_type, self);
     TypedData_Get_Struct(args[0], shape_t, &shape_data_type, shape);
-    irec = ((argc == 2) ? NUM2INT(args[1]) : -1);
+    irec = (argc == 2 && !NIL_P(args[1])) ? NUM2INT(args[1]) : -1;
     r = shapefile_write(self, irec, shape);
     return INT2FIX(r);
 }
@@ -219,12 +218,41 @@ static VALUE sf_write(int argc, VALUE *args, VALUE obj)
  * constructor
  */
 
+struct shapefile_wrap_args {
+    VALUE klass;
+    shapefile_t *sfile;
+};
+
+static VALUE WrapShapeFileProtected(VALUE ptr)
+{
+    struct shapefile_wrap_args *args = (struct shapefile_wrap_args *)ptr;
+    return TypedData_Wrap_Struct(args->klass, &shapefile_data_type, args->sfile);
+}
+
+static VALUE WrapShapeFile(VALUE klass, shapefile_t *sfile)
+{
+    struct shapefile_wrap_args args = { klass, sfile };
+    int state;
+    VALUE sf = rb_protect(WrapShapeFileProtected, (VALUE)&args, &state);
+    if (state) {
+	shapefile_free(sfile);
+	rb_jump_tag(state);
+    }
+    return sf;
+}
+
+static VALUE CallShapeFileBlock(VALUE args)
+{
+    rb_funcall(rb_ary_entry(args, 0), rb_intern("call"), 1,
+	rb_ary_entry(args, 1));
+    return Qnil;
+}
+
 static VALUE YieldIfPossible(VALUE proc, VALUE sf)
 {
     if (!NIL_P(proc)) {
-	rb_funcall(proc, rb_intern("call"), 1, sf);
-	sf_close(sf);
-	return Qnil;
+	return rb_ensure(CallShapeFileBlock, rb_ary_new_from_args(2, proc, sf),
+	    sf_close, sf);
     } else {
 	return sf;
     }
@@ -235,10 +263,14 @@ static VALUE sf_s_open(int argc, VALUE *argv, VALUE klass)
     VALUE fnam, mode, sf, proc;
     shapefile_t *sfile;
     rb_scan_args(argc, argv, "11&", &fnam, &mode, &proc);
-    sfile = shapefile_open(StringValuePtr(fnam), 
-	(NIL_P(mode) ? "rb" : StringValuePtr(mode))
+    /* Finish Ruby callbacks before taking pointers into either string. */
+    StringValue(fnam);
+    if (!NIL_P(mode))
+	StringValue(mode);
+    sfile = shapefile_open(StringValueCStr(fnam),
+	(NIL_P(mode) ? "rb" : StringValueCStr(mode))
 	);
-    sf = TypedData_Wrap_Struct(klass, &shapefile_data_type, sfile);
+    sf = WrapShapeFile(klass, sfile);
     return YieldIfPossible(proc, sf);
 }
 
@@ -250,7 +282,7 @@ static VALUE sf_s_new(int argc, VALUE *argv, VALUE klass)
     int i;
     if (argc > 3 || argc < 2) {
 	rb_raise(rb_eArgError,
-	"usage: ShapeLib::Shapefile.new filename, shapetype, attrs = []");
+	"usage: Shapelib::ShapeFile.new filename, shapetype, attrs = []");
 	return Qnil;
     }
     rb_scan_args(argc, argv, "21&", &fnam, &shapetype, &attrs, &proc);
@@ -262,11 +294,11 @@ static VALUE sf_s_new(int argc, VALUE *argv, VALUE klass)
 	    rb_raise(rb_eArgError, "attrs must be Array");
 	}
     }
-    sfile = shapefile_new(StringValuePtr(fnam), stype);
+    sfile = shapefile_new(StringValueCStr(fnam), stype);
     if (sfile == NULL) {
 	rb_raise(rb_eRuntimeError, "shapefile_new failed");
     }
-    sf = TypedData_Wrap_Struct(klass, &shapefile_data_type, sfile);
+    sf = WrapShapeFile(klass, sfile);
     if (argc > 2) {
 	for (i = 0; i < RARRAY_LEN(attrs); i++) {
 	    VALUE p = RARRAY_PTR(attrs)[i];

data/ext/shapelib_ext/sflist.h

@@ -1,21 +1,21 @@
 static method_table_entry sf_methods[] = {
-    { "add_field", sf_add_field, -1 },
-    { "close", sf_close, 0 },
-    { "delete_shape", sf_delete_shape, 1 },
-    { "each", sf_each, 0 },
-    { "field_count", sf_field_count, 0 },
-    { "field_index", sf_field_index, 1 },
-    { "field_name", sf_field_name, 1 },
-    { "field_type", sf_field_type, 1 },
-    { "field_width", sf_field_width, 1 },
-    { "field_decimals", sf_field_decimals, 1 },
-    { "fields", sf_fields, 0 },
-    { "minbound", sf_minbound, 0 },
-    { "maxbound", sf_maxbound, 0 },
-    { "read", sf_read, -1 },
-    { "rewind", sf_rewind, 0 },
-    { "shape_type", sf_shape_type, 0 },
-    { "size", sf_size, 0 },
-    { "write", sf_write, -1 },
+    { "add_field", RUBY_METHOD_FUNC(sf_add_field), -1 },
+    { "close", RUBY_METHOD_FUNC(sf_close), 0 },
+    { "delete_shape", RUBY_METHOD_FUNC(sf_delete_shape), 1 },
+    { "each", RUBY_METHOD_FUNC(sf_each), 0 },
+    { "field_count", RUBY_METHOD_FUNC(sf_field_count), 0 },
+    { "field_index", RUBY_METHOD_FUNC(sf_field_index), 1 },
+    { "field_name", RUBY_METHOD_FUNC(sf_field_name), 1 },
+    { "field_type", RUBY_METHOD_FUNC(sf_field_type), 1 },
+    { "field_width", RUBY_METHOD_FUNC(sf_field_width), 1 },
+    { "field_decimals", RUBY_METHOD_FUNC(sf_field_decimals), 1 },
+    { "fields", RUBY_METHOD_FUNC(sf_fields), 0 },
+    { "minbound", RUBY_METHOD_FUNC(sf_minbound), 0 },
+    { "maxbound", RUBY_METHOD_FUNC(sf_maxbound), 0 },
+    { "read", RUBY_METHOD_FUNC(sf_read), -1 },
+    { "rewind", RUBY_METHOD_FUNC(sf_rewind), 0 },
+    { "shape_type", RUBY_METHOD_FUNC(sf_shape_type), 0 },
+    { "size", RUBY_METHOD_FUNC(sf_size), 0 },
+    { "write", RUBY_METHOD_FUNC(sf_write), -1 },
     { NULL, NULL, 0 }
 };