census_shapes 0.2.0

data/lib/generators/census_shapes/setup/templates/models/shape.rb

@@ -0,0 +1,164 @@
+class MODEL_NAME < ActiveRecord::Base
+  
+  alias_attribute :id, :geoid10
+  alias_attribute :name, :name10
+  alias_attribute :region, :region10
+  alias_attribute :division, :division10
+  alias_attribute :state, :state10
+  alias_attribute :statens, :statens10
+  alias_attribute :statefp, :statefp10
+  alias_attribute :countyfp, :countyfp10
+  alias_attribute :tractce, :tractce10
+  alias_attribute :blkgrpce, :blkgrpce10
+  alias_attribute :stusps, :stusps10
+  alias_attribute :lsad, :lsad10
+  alias_attribute :mtcc10, :mtfcc10
+  alias_attribute :funcstat, :funcstat10
+  alias_attribute :land, :aland10
+  alias_attribute :water, :awater10
+  alias_attribute :lat, :intptlat10
+  alias_attribute :lng, :intptlon10
+
+  self.primary_key = :gid
+  
+  default_scope select("geoid10, name10, intptlat10, intptlon10, latlng, geom")
+
+  def self.bbox(z,x,y)
+    return false if z.nil? || x.nil? || y.nil?
+    nw=coordinateLocation(z.to_i,x.to_i,y.to_i)
+    se=coordinateLocation(z.to_i,x.to_i+1,y.to_i+1)
+    return "#{nw[:lng]},#{se[:lat]},#{se[:lng]},#{nw[:lat]}"
+  end
+  
+  def self.coordinateLocation(z,x,y)
+    k = 45 / 2 ** (z - 3.001)
+    return {:lng=> (k * x - 180).to_i,:lat=> y2lat(180 - k * y)}
+  end
+
+  def self.y2lat(y)
+    return (360 / Math::PI * Math.atan(Math.exp(y * Math::PI / 180)) -90).round(2)
+  end
+
+  
+  def self.features(options={})
+    features = []; i =1; 
+    options[:detail]=0.01 if options[:detail].nil?
+    model = options[:type].capitalize.constantize
+    if !options[:detail].nil?
+      geos = model.select("name10, geoid10, ST_AsGeoJson(ST_SimplifyPreserveTopology(geom,#{options[:detail]})) AS geojson");
+    else
+      geos = model.select("name10, geoid10, ST_AsGeoJson(geom) As geojson, geom, ST_Area(geom) AS area, ST_IsValid(geom) AS isvalid");
+    end
+    if(options[:bbox])
+      b = options[:bbox].split(",")
+      ids = model.select("geoid10").where("ST_Contains(ST_AsText(ST_Envelope('LINESTRING(#{b[0]} #{b[1]},#{b[2]} #{b[3]})'::geometry)), ST_AsText(latlng))").map{|id| id['geoid10']}
+      if options[:grid]
+        geobox = model.select("ST_AsGeoJson(ST_Envelope('LINESTRING(#{b[0]} #{b[1]},#{b[2]} #{b[3]})'::geometry)) AS geojson").first
+        features <<  {:type => "Feature", :geometry=> JSON.parse(geobox['geojson']), :properties => {:id=> "1", :name => "Box", :class=> 'box'}} 
+      end
+      if ids.empty?
+        geos = []
+      else
+        geos = geos.where(:geoid10 => ids)
+      end
+    end
+    if(options[:id])
+       geos = geos.where(:geoid10=>options[:id].split(","))
+    end
+    for g in geos
+      if g['isvalid'] != "t" && options[:fix_geom]
+        new_g = g.fix_geometry
+        msg = "Geometry invalid. Rebuilding. Original = #{g['isvalid'] == "t"}, Rebuilt Geometry = #{new_g['isvalid'] == "t"}, Old area (#{g['area']}) == New area (#{new_g['area']})"
+        features <<  {:type => "Feature", :geometry=> JSON.parse(new_g['geojson']), :properties => {:id=> g.geoid10, :name => g.name10 }} 
+      else
+        features <<  {:type => "Feature", :geometry=> JSON.parse(g['geojson']), :properties => {:id=> g.geoid10, :name => g.name10 }} 
+      end
+      i=i+1
+    end
+    return features
+  end
+  
+  def fix_geometry
+    return self if self['isvalid'] == "t"
+    rebuilt_g = MODEL_NAME.rebuild(self.polygons)
+    return rebuilt_g if rebuilt_g['isvalid'] == "t"
+    buffer_g = self.buffer
+    return buffer_g if buffer_g['isvalid'] == "t"
+    puts "ERROR GEOMETRY NOT FIXED"
+    return false
+  end
+  
+  def buffer
+    return self.class.name.constantize.select("geoid10, ST_Buffer(geom,.000001) AS geo, ST_AsGeoJson(ST_Buffer(geom,.000001)) AS geojson, ST_IsValid(ST_Buffer(geom,.000001)) AS isvalid, ST_Area(ST_Buffer(geom,.000001)) AS area").where(:geoid10 => self.geoid10).first
+  end
+  
+  def polygons
+     return self.class.name.constantize.select("ST_AsText((ST_DumpRings(((ST_Dump(geom)).geom))).geom) AS g").where(:geoid10 => self.geoid10).map{|p| p['g'] }
+  end
+  
+  def self.sort(polygons)
+    return polygons.map{|p| 
+      pp = ActiveRecord::Base.connection.execute("SELECT ST_Area('#{p}'), ST_AsText('#{p}') AS geo").first
+      [pp['st_area'].to_f, pp['geo'] ]}.sort {|a, b| a[0] <=> b[0]}.reverse.map{|p| p[1]
+    }
+  end
+  
+  def self.rebuild(polygons)
+    polygons = MODEL_NAME.sort(polygons)
+    polygons.reverse.each do |p1|
+      polygons.each do |p2|
+         if p1 != p2 && ActiveRecord::Base.connection.execute("SELECT ST_Contains('#{p1}', '#{p2}') AS contains").first['contains'] == "t"
+          temp = ActiveRecord::Base.connection.execute("SELECT ST_AsText(ST_Difference('#{p1}', '#{p2}')) AS union")
+          polygons.delete(p1); polygons.delete(p2)
+          polygons << temp.first['union']
+        end
+      end
+    end
+    polygons = MODEL_NAME.merge(polygons)
+    return polygons
+  end
+  
+  def self.merge(polygons)
+    if polygons.count == 1
+      return polygons[0]
+    else
+      array = polygons.map { |p| "ST_GeomFromText('#{p}')" }
+      geo = ActiveRecord::Base.connection.execute("SELECT ST_Collect(ARRAY[#{array.join(',')}]) AS geo").first['geo']
+      sql = "SELECT '#{geo}' AS geo, ST_AsGeoJson('#{geo}') AS geojson, ST_Area('#{geo}') AS area, ST_IsValid('#{geo}') AS isvalid;"
+      return ActiveRecord::Base.connection.execute(sql).first
+    end
+  end
+  
+  def self.available_shapes
+    return @available_shapes if @available_shapes
+    @available_shapes = CENSUS_SHAPES
+    available_types = MODEL_NAME.unscoped.select("DISTINCT(type)").map{|k,v| k['type'].upcase}
+    @available_shapes.each do |k,v|
+      @available_shapes.delete(k) if !available_types.include?(k)
+    end
+    return @available_shapes
+  end
+end
+
+class State < MODEL_NAME; end
+class County < MODEL_NAME; end
+class Cousub < MODEL_NAME; end
+class Submcd < MODEL_NAME; end
+class Block < MODEL_NAME; end
+class Tract < MODEL_NAME; end
+class BG < MODEL_NAME; end
+class Place < MODEL_NAME; end
+class Anrc < MODEL_NAME; end
+class Aiannh < MODEL_NAME; end
+class Aits < MODEL_NAME; end
+class Cbsa < MODEL_NAME; end
+class Metdiv < MODEL_NAME; end
+class Csa < MODEL_NAME; end
+class Cd < MODEL_NAME; end
+class Sldu < MODEL_NAME; end
+class Sldl < MODEL_NAME; end
+class Vtd < MODEL_NAME; end
+class Zcta5 < MODEL_NAME; end
+class Elsd < MODEL_NAME; end
+class Scsd < MODEL_NAME; end
+class Unsd < MODEL_NAME; end

data/lib/generators/census_shapes/setup/templates/views/layouts/shapes.html.erb

@@ -0,0 +1,17 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <meta http-equiv="content-type" content="text/html;charset=utf-8">
+    <title>Census Shapes Polymaps Example</title>
+    <!-- POLYMAPS REFERENCE FILES -->
+    <script type="text/javascript" src="http://polymaps.org/polymaps.min.js?2.5.0"></script>
+    <script type="text/javascript" src="http://polymaps.org/nns.min.js?1.1.0"></script>
+    <link rel="stylesheet" type="text/css" href="http://polymaps.org/screen.css?0.9+1" />
+    <link rel="stylesheet" type="text/css" href="http://polymaps.org/style.css?2.2.0" />
+  </head>
+  <body>
+    <div class="container">
+      <%= yield %>
+    </div>
+  </body>
+</html>

data/lib/generators/census_shapes/setup/templates/views/shapes/index.html.erb

@@ -0,0 +1,89 @@
+<div class="span-18 last top">
+  <a href="http://polymaps.org/">Census-Shapes</a>::
+  <%= select_tag :MODEL_NAME_type, options_for_select(MODEL_NAME.available_shapes.map{|g| [g[1]['name'], g[1]['slug']]}), :onchange => 'loadLayer()'%>
+</div>
+<div id="map" class="span-24 last"></div>
+<style type="text/css">
+  .layer path{
+    fill: lightsteelblue;
+    fill-opacity: .5;
+    stroke: steelblue;
+  	stroke-width: 1px;
+  }
+</style>
+
+<script type="text/javascript">
+
+var po=org.polymaps,
+map=po.map()
+.container(document.getElementById("map")
+.appendChild(po.svg("svg")))
+.add(po.interact())
+.add(po.hash());
+
+map.add(
+  po.image()
+  .url(po.url("http://{S}tile.cloudmade.com/1a1b06b230af4efdbb989ea99e9841af/998/256/{Z}/{X}/{Y}.png")
+  .hosts(["a.","b.","c.",""])));
+
+map.center({lat: 39, lon: -96}).zoom(4);
+loadLayer();
+
+function segment(template) {
+  return function(c) {
+   var max = 1 << c.zoom, column = c.column % max;
+   if (column < 0) column += max;	
+    return template.replace(/{(.)}/g, function(s, v) {
+      switch (v) {
+        case "T":{
+					return document.getElementById('MODEL_NAME_type').value;
+				}
+				case "Z": return c.zoom;
+				case "X": return column;
+				case "Y": return c.row;
+				case "B": {
+				  var nw = map.coordinateLocation({row: c.row, column: column, zoom: c.zoom}),
+				      se = map.coordinateLocation({row: c.row + 1, column: column + 1, zoom: c.zoom}),
+				      pn = Math.ceil(Math.log(c.zoom) / Math.LN2);
+				  return nw.lon.toFixed(pn)
+				      + "," + se.lat.toFixed(pn)
+				      + "," + se.lon.toFixed(pn)
+				      + "," + nw.lat.toFixed(pn);
+				}
+      }
+      return v;
+    });
+  };
+}
+
+function loadLayer(){
+  map_layer = po.geoJson()
+    .url(segment("/{T}/{Z}/{X}/{Y}.json"))
+    .on("load", load)
+    .id("map_layer")
+    .clip(false);
+  map.add(map_layer);
+
+  compass_layer = po.compass().pan("small");
+  map.add(compass_layer);
+}
+
+function load(e) {
+  for (var i = 0; i < e.features.length; i++) {
+    var feature = e.features[i];
+    var properties = feature.data.properties
+    feature.element.setAttribute("id", properties.id)
+		feature.element.setAttribute("name", properties.name)
+		feature.element.setAttribute("class", properties.class)
+    feature.element.appendChild(po.svg("title")
+     .appendChild(document.createTextNode(properties.name+" "+0+")"))
+     .parentNode);
+ 	}
+}
+
+
+
+(function(){function d(){if(b=!b){c.style("position","fixed").style("border-width",0).style("width","100%").style("height","100%").style("top",0).style("left",0);a.style("position","fixed").style("right","16px").style("top","16px");e.attr("transform","translate(16,16)rotate(135)scale(5)translate(-1.85,0)");f.style("visibility","hidden").style("overflow","hidden")}else{c.style("position",null).style("border-width",null).style("width",null).style("height",null).style("top",null).style("left",null);
+a.style("position","absolute").style("right","-16px").style("top","-16px");e.attr("transform","translate(16,16)rotate(-45)scale(5)translate(-1.85,0)");f.style("visibility",null).style("overflow",null)}map.resize()}var f=n$(document.body),c=n$("#map").style("visibility","visible"),b=false,a=c.add("svg:svg").attr("width",32).attr("height",32).style("position","absolute").style("right","-16px").style("top","-16px").style("visibility","visible").on("mousedown",d);a.add("svg:circle").attr("cx",16).attr("cy",
+16).attr("r",14).attr("fill","#fff").attr("stroke","#ccc").attr("stroke-width",4).add("svg:title").text("Toggle fullscreen. (ESC)");var e=a.add("svg:path").attr("transform","translate(16,16)rotate(-45)scale(5)translate(-1.85,0)").attr("d","M0,0L0,.5 2,.5 2,1.5 4,0 2,-1.5 2,-.5 0,-.5Z").attr("pointer-events","none").attr("fill","#aaa");window.addEventListener("keydown",function(g){g.keyCode==27&&b&&d()},false)})();
+</script>

data/spec/spec_helper.rb

@@ -0,0 +1,8 @@
+require 'rubygems'
+require 'bundler/setup'
+
+require 'census_shapes'
+
+RSpec.configure do |config|
+  # some (optional) config here
+end

metadata

@@ -0,0 +1,149 @@
+--- !ruby/object:Gem::Specification
+name: census_shapes
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+  prerelease: 
+platform: ruby
+authors:
+- Ty Rauber
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: postgis_adapter
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: progress_bar
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: generator_spec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Imports all the US Census Geographies into a PostGIS database.
+email:
+- tyrauber@mac.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- census_shapes.gemspec
+- lib/census_shapes.rb
+- lib/census_shapes/version.rb
+- lib/generators/census_shapes/setup/setup_generator.rb
+- lib/generators/census_shapes/setup/templates/.DS_Store
+- lib/generators/census_shapes/setup/templates/config/database_example.yml
+- lib/generators/census_shapes/setup/templates/config/initializers/shapes_globals.rb
+- lib/generators/census_shapes/setup/templates/controllers/shapes_controller.rb
+- lib/generators/census_shapes/setup/templates/db/migrate/create_shapes.rb
+- lib/generators/census_shapes/setup/templates/lib/tasks/census_shapes.rake
+- lib/generators/census_shapes/setup/templates/lib/tasks/postgis_template.rake
+- lib/generators/census_shapes/setup/templates/lib/yaml/us_shapes.yml
+- lib/generators/census_shapes/setup/templates/lib/yaml/us_states.yml
+- lib/generators/census_shapes/setup/templates/models/shape.rb
+- lib/generators/census_shapes/setup/templates/views/layouts/shapes.html.erb
+- lib/generators/census_shapes/setup/templates/views/shapes/index.html.erb
+- spec/spec_helper.rb
+homepage: https://github.com/tyrauber/census_shapes
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: census_shapes
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: A Ruby Gem for importing US Census Shapes into PostGIS
+test_files:
+- spec/spec_helper.rb