worldgen 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: de4a34b0248c80788bc6ce2f4a12403b49e8df05
+  data.tar.gz: d8210298cd91385beab0cb5366dcc19bfba9eefd
+SHA512:
+  metadata.gz: 2829eab4c21503c6243774487c0121dd84fc41300103759116d97c2eeb378ed0a0b5cce0a2f16061d70e24b81bd8ec226375abda3e50827828e3d5316ab8b26f
+  data.tar.gz: 58b82dcbc654c466ced978d629881bcba395d4577becff34b40017b912bff9a51462668e426033c36ec7106618be51c2e7b1d066fc24054dd55270acf1822691

data/.gitignore

@@ -0,0 +1,19 @@
+# Gem/Bundler stuff
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+Makefile
+mkmf.log
+
+# Vim stuff
+*.sw[op]

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in worldgen.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Rob Britton
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,59 @@
+# Worldgen
+
+Worldgen allows you to generate random worlds.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'worldgen'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install worldgen
+
+## Usage
+
+So far there is not much. You can do square heightmaps:
+
+```ruby
+# Create a 50x50 heightmap - this will be flat
+heightmap = Worldgen::HeightMap.new 50
+```
+
+Then you can do a diamond square fractal to it:
+
+```ruby
+Worldgen::Algorithms.diamond_square! heightmap
+```
+
+Then dump it to an image file:
+
+```ruby
+Worldgen::Render.heightmap heightmap, "output.png"
+```
+
+## Contributing
+
+1. Fork it ( https://github.com/robbrit/worldgen/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create a new Pull Request
+
+### Notes
+
+Heightmaps are purely written in C for performance. If you're going to be
+writing code that interacts with heightmaps, you're going to have a much
+better time doing it in C since Ruby is a fair bit too slow once you get to
+larger maps (512x512 or higher).
+
+### TODO
+
+* Diamond Square output seems to be rougher than it should be

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+
+require "rake/extensiontask"
+
+spec = Gem::Specification.load("worldgen.gemspec")
+Rake::ExtensionTask.new("worldgen", spec)

data/ext/worldgen/common.c

@@ -0,0 +1,31 @@
+#include "common.h"
+
+/**
+ * Convert an array of doubles into normalized doubles between 0 and 1
+ **/
+void normalize(double * values, int size) {
+  double min = values[0],
+         max = values[0];
+
+  int x;
+
+  for (x = 1; x < num_points(size); x++) {
+    if (values[x] > max) {
+      max = values[x];
+    }
+    if (values[x] < min) {
+      min = values[x];
+    }
+  }
+
+  for (x = 0; x < num_points(size); x++) {
+    values[x] = (values[x] - min) / (max - min);
+  }
+}
+
+/** Get the number of points for a heightmap of a given size. This will change
+ * later when we move from beyond just a simple square geometry.
+ */
+int num_points(int size) {
+  return size * size;
+}

data/ext/worldgen/common.h

@@ -0,0 +1,16 @@
+#ifndef COMMON_H__
+#define COMMON_H__
+
+#define ARR(a, x, y) a[(x) * size + (y)]
+
+typedef double * heightmap_points;
+
+typedef struct {
+  int size;
+  heightmap_points heights;
+} heightmap;
+
+int num_points(int);
+void normalize(double *, int);
+
+#endif // COMMON_H__

data/ext/worldgen/diamondsquare.c

@@ -0,0 +1,106 @@
+#include <ruby.h>
+#include <stdlib.h>
+
+#include "common.h"
+
+#define DEFAULT_ROUGHNESS 5
+
+extern heightmap get_heights(VALUE);
+extern void set_heights(VALUE, heightmap_points);
+extern int get_size(VALUE);
+
+double diamond_shift(double roughness) {
+  double r = (double)rand() / (double)RAND_MAX;
+
+  return (r * 2.0 - 1.0) * roughness;
+}
+
+/**
+ * Generate terrain using a diamond square algorithm.
+ * Arguments:
+ * `heightmap` - The heightmap to use for the algorithm
+ * `roughness` (optional) - How "rough" to make the surface
+ *
+ * Return value: nil
+ **/
+VALUE diamond_square(int argc, VALUE *argv, VALUE self) {
+  int x, y;
+  double ratio = 500.0;
+  VALUE heightmap_obj, vroughness;
+  int size, side_size;
+  double roughness;
+  heightmap_points heights;
+
+  rb_scan_args(argc, argv, "11", &heightmap_obj, &vroughness);
+    
+  size = get_size(heightmap_obj);
+  side_size = size - 1;
+
+  roughness = vroughness == Qnil ? DEFAULT_ROUGHNESS : NUM2DBL(vroughness);
+
+  // need to allocate on the heap since Ruby will throw stack size problems
+  // if we don't
+  heights = (double *)malloc(sizeof(double) * num_points(size));
+
+  memset(heights, 0.0, sizeof(double) * num_points(size));
+
+  ARR(heights, 0, 0) = ARR(heights, 0, size - 1) =
+    ARR(heights, size - 1, 0) = ARR(heights, size - 1, size - 1) =
+    diamond_shift(roughness);
+
+  while (side_size >= 2) {
+    int half_side = side_size / 2;
+
+    // Square step
+    for (x = 0; x < size - 1; x += side_size) {
+      for (y = 0; y < size - 1; y += side_size) {
+        double avg = (ARR(heights, x, y) +
+          ARR(heights, x + side_size, y) +
+          ARR(heights, x, y + side_size) +
+          ARR(heights, x + side_size, y + side_size)) / 4.0;
+
+        ARR(heights, x + half_side, y + half_side) = avg + diamond_shift(roughness);
+      }
+    }
+
+    for (x = 0; x < size - 1; x += half_side) {
+      for (y = (x + half_side) % side_size; y < size - 1; y += side_size) {
+        double avg = (ARR(heights, (x - half_side + size - 1) % (size - 1), y) +
+          ARR(heights, (x + half_side) % (size - 1), y) +
+          ARR(heights, x, (y + half_side) % (size - 1)) +
+          ARR(heights, x, (y - half_side + size - 1) % (size - 1))) / 4.0;
+
+        avg += diamond_shift(roughness);
+
+        ARR(heights, x, y) = avg;
+
+        if (x == 0) {
+          ARR(heights, size - 1, y) = avg;
+        }
+        if (y == 0) {
+          ARR(heights, x, size - 1) = avg;
+        }
+      }
+    }
+
+    side_size /= 2.0;
+    ratio /= 2.0;
+  }
+
+  // normalize
+  normalize(heights, size);
+
+  // copy into the array
+  set_heights(heightmap_obj, heights);
+
+  return Qnil;
+}
+
+void load_diamond_square() {
+  VALUE mod, algos;
+
+  mod = rb_define_module("Worldgen");
+  algos = rb_define_module_under(mod, "Algorithms");
+
+  rb_define_singleton_method(algos, "diamond_square!", diamond_square, -1);
+}

data/ext/worldgen/extconf.rb

@@ -0,0 +1,4 @@
+require 'mkmf'
+
+dir_config("worldgen")
+create_makefile("worldgen")

data/ext/worldgen/heightmap.c

@@ -0,0 +1,106 @@
+#include <ruby.h>
+
+#include "common.h"
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <time.h>
+
+VALUE HeightmapData; // conversion between heightmap struct and Ruby
+
+void delete_heightmap(void * ptr) {
+  heightmap *map = (heightmap *)ptr;
+  free(map->heights);
+}
+
+/**
+ * Get the heightmap for a HeightMap object
+ */
+heightmap get_heights(VALUE heightmap_obj) {
+  VALUE heights_ptr = rb_iv_get(heightmap_obj, "@heights_ptr");
+  heightmap *map;
+
+  Data_Get_Struct(heights_ptr, heightmap, map);
+
+  return *map;
+}
+
+int get_size(VALUE heightmap_obj) {
+  return FIX2INT(rb_iv_get(heightmap_obj, "@size"));
+}
+
+void set_heights(VALUE heightmap_obj, heightmap_points map) {
+  // if we already have a height, kill it
+  VALUE heights_ptr;
+  heightmap * hmap;
+
+  hmap = ALLOC(heightmap);
+  hmap->heights = map;
+  hmap->size = get_size(heightmap_obj);
+
+  heights_ptr = Data_Wrap_Struct(HeightmapData, 0, delete_heightmap, hmap);
+
+  rb_iv_set(heightmap_obj, "@heights_ptr", heights_ptr);
+}
+
+/**
+ * Initialize the C side of things for a heightmap
+ */
+VALUE initialize_native(VALUE self, VALUE vsize) {
+  int size = FIX2INT(vsize);
+  int memsize = num_points(size) * sizeof(double);
+
+  heightmap_points map = (heightmap_points)malloc(memsize);
+  memset(map, 0, memsize);
+  set_heights(self, map);
+
+  return Qnil;
+}
+
+/**
+ * Get the number of points within the heightmap. Right now this is a very
+ * simple calculation of size * size.
+ */
+VALUE num_points_wrapped(VALUE self) {
+  return INT2FIX(num_points(FIX2INT(rb_iv_get(self, "@size"))));
+}
+
+/**
+ * Iterate over all the points in the heightmap.
+ * Example:
+ * ```
+ * heightmap.each_height do |x, y, height|
+ *   puts "Height at (#{x}, #{y}) is #{height}"
+ * end
+ */
+VALUE each_height(VALUE self) {
+  heightmap map = get_heights(self);
+  heightmap_points ptr = map.heights;
+  VALUE args = rb_ary_new2(3);
+  int size = get_size(self);
+  int x, y;
+
+  for (x = 0; x < size; x++) {
+    rb_ary_store(args, 0, INT2FIX(x));
+    for (y = 0; y < size; y++) {
+      rb_ary_store(args, 1, INT2FIX(y));
+      rb_ary_store(args, 2, DBL2NUM(*ptr++));
+      rb_yield(args);
+    }
+  }
+  return self;
+}
+
+void load_heightmap() {
+  VALUE mod, height_map;
+
+  mod = rb_define_module("Worldgen");
+
+  height_map = rb_define_class_under(mod, "HeightMap", rb_cObject);
+
+  rb_define_private_method(height_map, "initialize_native", initialize_native, 1);
+  rb_define_method(height_map, "num_points", num_points_wrapped, 0);
+  rb_define_method(height_map, "each_height", each_height, 0);
+
+  HeightmapData = rb_define_class_under(height_map, "HeightmapData", rb_cObject);
+}

data/ext/worldgen/worldgen.c

@@ -0,0 +1,13 @@
+#include <ruby.h>
+#include <stdlib.h>
+#include <time.h>
+
+extern void load_heightmap();
+extern void load_diamond_square();
+
+void Init_worldgen() {
+  srand((unsigned)time(NULL));
+
+  load_heightmap();
+  load_diamond_square();
+}

data/generate.rb

@@ -0,0 +1,51 @@
+require 'optparse'
+require "worldgen"
+
+options = {}
+
+OptionParser.new do |opts|
+  opts.on("--heightmap [FILE]", String, "Output a heightmap to FILE") do |file|
+    options[:heightmap] = file
+  end
+
+  opts.on("--platemap [FILE]", String, "Output a platemap to FILE") do |file|
+    options[:platemap] = file
+  end
+
+  opts.on("--size N", Integer, "Generate a map of size 2^N + 1") do |n|
+    options[:size] = 2**n + 1
+  end
+
+  opts.on("--num-plates [N]", Integer, "Generate N plates") do |n|
+    options[:num_plates] = n
+  end
+end.parse!
+
+if not options[:size]
+  puts "No size specified."
+  exit
+end
+
+if options[:heightmap]
+  puts "Generating heightmap..."
+  heightmap = Worldgen::HeightMap.new(options[:size])
+  Worldgen::Algorithms.diamond_square!(heightmap)
+  Worldgen::Render.heightmap heightmap, options[:heightmap]
+end
+
+if options[:platemap]
+  puts "Generating plate map..."
+  if not options[:num_plates]
+    puts "num_plates not specified."
+    exit
+  end
+
+  platemap = Worldgen::PlateMap.new(options[:size])
+  platemap.generate_plates! options[:num_plates]
+  puts "Converting to height map..."
+  heightmap = platemap.to_height_map
+  render_heightmap heightmap, options[:platemap]
+end
+
+puts "Done."
+

data/lib/worldgen.rb

@@ -0,0 +1,8 @@
+require "worldgen/version"
+require "worldgen/heightmap"
+require "worldgen/render"
+require_relative "worldgen.so"
+
+module Worldgen
+  # Your code goes here...
+end

data/lib/worldgen/heightmap.rb

@@ -0,0 +1,18 @@
+module Worldgen
+  # A square heightmap
+  class HeightMap
+    # A class used internally to manage C-allocated memory
+    class HeightmapData
+    end
+
+    attr_reader :size
+
+    # Create a new square heightmap.
+    # Arguments:
+    # * size - the width/height of the map
+    def initialize size
+      @size = size
+      initialize_native(size)
+    end
+  end
+end

data/lib/worldgen/render.rb

@@ -0,0 +1,44 @@
+require 'RMagick'
+
+module Worldgen::Render
+  # Render a heightmap to a grayscale file.
+  # Arguments:
+  # * map - The heightmap to render
+  # * filename - The filename to use. Image format will be inferred from the filename
+  def self.heightmap map, filename
+    # loading each one in is crazy slow, just throw it into a pixel map
+    image = Magick::Image.new(map.size, map.size) { self.background_color = "black" }
+
+    map.each_height do |x, y, pix_height|
+      grey = ("%2X" % (pix_height * 255).round) * 3
+      image.pixel_color x, y, "##{grey}"
+    end
+
+    #image.display
+    image.write filename
+  end
+
+=begin
+  def self.platemap map, filename
+    image = Magick::Image.new(map.size, map.size) { self.background_color = "black" }
+
+    # draw plates
+    colours = [
+      "#FF0000", "#0000FF", "#FFFF00", "#00FF00",
+      "#FF6600", "#FF00FF", "#00FFFF", "#CCCCCC",
+      "#006600", "#000066", "#660066", "#666600"
+    ]
+    map.each_plate_point do |x, y, plate|
+      begin
+        image.pixel_color x, y, colours[plate]
+      rescue
+        puts "colour fail"
+        puts [x, y, plate, colour[plate]].inspect
+      end
+    end
+
+    #image.display
+    image.write filename
+  end
+=end
+end

data/lib/worldgen/version.rb

@@ -0,0 +1,3 @@
+module Worldgen
+  VERSION = "0.0.2"
+end

data/worldgen.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'worldgen/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "worldgen"
+  spec.version       = Worldgen::VERSION
+  spec.authors       = ["Rob Britton"]
+  spec.email         = ["rob@robbritton.com"]
+  spec.summary       = %q{Gem to handle procedural content generation for worlds.}
+  #spec.description   = %q{TODO: Write a longer description. Optional.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.extensions << "ext/worldgen/extconf.rb"
+
+  spec.add_dependency "rmagick", "~> 2.13.3"
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rake-compiler", "~> 0.9.3"
+end

metadata

@@ -0,0 +1,118 @@
+--- !ruby/object:Gem::Specification
+name: worldgen
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Rob Britton
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-08-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rmagick
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.13.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.13.3
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.3
+description: 
+email:
+- rob@robbritton.com
+executables: []
+extensions:
+- ext/worldgen/extconf.rb
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- ext/worldgen/common.c
+- ext/worldgen/common.h
+- ext/worldgen/diamondsquare.c
+- ext/worldgen/extconf.rb
+- ext/worldgen/heightmap.c
+- ext/worldgen/worldgen.c
+- generate.rb
+- lib/worldgen.rb
+- lib/worldgen/heightmap.rb
+- lib/worldgen/render.rb
+- lib/worldgen/version.rb
+- worldgen.gemspec
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.1.11
+signing_key: 
+specification_version: 4
+summary: Gem to handle procedural content generation for worlds.
+test_files: []