require 'geokit'
require 'geo_ruby'

class Chouette::StopArea < Chouette::TridentActiveRecord
  # FIXME http://jira.codehaus.org/browse/JRUBY-6358
  set_primary_key :id
  include Geokit::Mappable
  has_many :stop_points, :dependent => :destroy
  has_many :access_points, :dependent => :destroy
  has_many :access_links, :dependent => :destroy
  has_and_belongs_to_many :routing_lines, :class_name => 'Chouette::Line', :foreign_key => "stop_area_id", :association_foreign_key => "line_id", :join_table => "routing_constraints_lines", :order => "lines.number"
  has_and_belongs_to_many :routing_stops, :class_name => 'Chouette::StopArea', :foreign_key => "parent_id", :association_foreign_key => "child_id", :join_table => "stop_areas_stop_areas", :order => "stop_areas.name"

  acts_as_tree :foreign_key => 'parent_id',:order => "name"

  attr_accessor :stop_area_type
  attr_accessor :children_ids
  
  attr_accessible :routing_stop_ids, :routing_line_ids, :children_ids, :stop_area_type, :parent_id, :objectid
  attr_accessible :object_version, :creation_time, :creator_id, :name, :comment, :area_type, :registration_number
  attr_accessible :nearest_topic_name, :fare_code, :longitude, :latitude, :long_lat_type, :x, :y, :projection_type
  attr_accessible :country_code, :street_name
  
  # workaround of ruby 1.8 private method y block attribute y reading access
  def y
    read_attribute :y
  end

  validates_uniqueness_of :registration_number, :allow_nil => true, :allow_blank => true
  validates_format_of :registration_number, :with => %r{\A[0-9A-Za-z_-]+\Z}, :allow_blank => true
  validates_presence_of :name
  validates_presence_of :area_type

  validates_presence_of :latitude, :if => :longitude
  validates_presence_of :longitude, :if => :latitude
  validates_numericality_of :latitude, :less_than_or_equal_to => 90, :greater_than_or_equal_to => -90, :allow_nil => true
  validates_numericality_of :longitude, :less_than_or_equal_to => 180, :greater_than_or_equal_to => -180, :allow_nil => true

  validates_presence_of :x, :if => :y
  validates_presence_of :y, :if => :x
  validates_numericality_of :x, :allow_nil => true
  validates_numericality_of :y, :allow_nil => true

  def self.nullable_attributes
    [:registration_number, :street_name, :country_code, :fare_code, :nearest_topic_name, :comment, :projection_type, :long_lat_type]
  end

  after_update :clean_invalid_access_links

  def children_in_depth
    return [] if self.children.empty?

    self.children + self.children.map do |child|
      child.children_in_depth
    end.flatten.compact
  end

  def possible_children
    case area_type
      when "BoardingPosition" then []
      when "Quay" then []
      when "CommercialStopPoint" then Chouette::StopArea.where(:area_type => ['Quay', 'BoardingPosition']) - [self]
      when "StopPlace" then Chouette::StopArea.where(:area_type => ['StopPlace', 'CommercialStopPoint']) - [self]
      when "ITL" then Chouette::StopArea.where(:area_type => ['Quay', 'BoardingPosition', 'StopPlace', 'CommercialStopPoint'])
    end
      
  end

  def possible_parents
    case area_type
      when "BoardingPosition" then Chouette::StopArea.where(:area_type => "CommercialStopPoint")  - [self]
      when "Quay" then Chouette::StopArea.where(:area_type => "CommercialStopPoint") - [self]
      when "CommercialStopPoint" then Chouette::StopArea.where(:area_type => "StopPlace") - [self]
      when "StopPlace" then Chouette::StopArea.where(:area_type => "StopPlace") - [self]
    end
  end

  def lines
    if (area_type == 'CommercialStopPoint')
      self.children.collect(&:stop_points).flatten.collect(&:route).flatten.collect(&:line).flatten.uniq
    else
      self.stop_points.collect(&:route).flatten.collect(&:line).flatten.uniq
    end
  end

  def routes
    self.stop_points.collect(&:route).flatten.uniq
  end

  def self.commercial
    where :area_type => "CommercialStopPoint"
  end

  def to_lat_lng
    Geokit::LatLng.new(latitude, longitude) if latitude and longitude
  end

  def geometry
    GeoRuby::SimpleFeatures::Point.from_lon_lat(longitude, latitude, 4326) if latitude and longitude
  end

  def geometry=(geometry)
    geometry = geometry.to_wgs84
    self.latitude, self.longitude, self.long_lat_type = geometry.lat, geometry.lng, "WGS84"
  end

  def position 
    geometry
  end

  def position=(position)
    position = nil if String === position && position == ""
    position = Geokit::LatLng.normalize(position), 4326 if String === position
    self.latitude = position.lat
    self.longitude = position.lng
  end

  def default_position 
    # for first StopArea ... the bounds is nil :(
    Chouette::StopArea.bounds and Chouette::StopArea.bounds.center
  end

  def self.near(origin, distance = 0.3)
    origin = origin.to_lat_lng

    lat_degree_units = units_per_latitude_degree(:kms)
    lng_degree_units = units_per_longitude_degree(origin.lat, :kms)
    
    where "SQRT(POW(#{lat_degree_units}*(#{origin.lat}-latitude),2)+POW(#{lng_degree_units}*(#{origin.lng}-longitude),2)) <= #{distance}"
  end

  def self.bounds
    # Give something like :
    # [["113.5292500000000000", "22.1127580000000000", "113.5819330000000000", "22.2157050000000000"]]
    min_and_max = connection.select_rows("select min(longitude) as min_lon, min(latitude) as min_lat, max(longitude) as max_lon, max(latitude) as max_lat from #{table_name} where latitude is not null and longitude is not null").first
    return nil unless min_and_max

    # Ignore [nil, nil, nil, nil]
    min_and_max.compact!
    return nil unless min_and_max.size == 4

    min_and_max.collect! { |n| n.to_f }

    # We need something like :
    # [[113.5292500000000000, 22.1127580000000000], [113.5819330000000000, 22.2157050000000000]]
    coordinates = min_and_max.each_slice(2).to_a
    GeoRuby::SimpleFeatures::Envelope.from_coordinates coordinates
  end

  def stop_area_type
    area_type && Chouette::AreaType.new(area_type.underscore)
  end

  def stop_area_type=(stop_area_type)   
    self.area_type = (stop_area_type ? stop_area_type.camelcase : nil)
    if self.area_type == 'Itl'
      self.area_type = 'ITL'
    end
  end

  @@stop_area_types = nil
  def self.stop_area_types
    @@stop_area_types ||= Chouette::AreaType.all.select do |stop_area_type|
      stop_area_type.to_i >= 0
    end
  end

  def children_ids=(children_ids)
    children = children_ids.split(',').uniq
    # remove unset children
    self.children.each do |child|
      if (! children.include? child.id)
        child.update_attribute :parent_id, nil
      end
    end
    # add new children
    Chouette::StopArea.find(children).each do |child|
       child.update_attribute :parent_id, self.id
    end
  end
  
  def routing_stop_ids=(routing_stop_ids)
    stops = routing_stop_ids.split(',').uniq
    self.routing_stops.clear
    Chouette::StopArea.find(stops).each do |stop|
      self.routing_stops << stop
    end
  end

  def routing_line_ids=(routing_line_ids)
    lines = routing_line_ids.split(',').uniq
    self.routing_lines.clear
    Chouette::Line.find(lines).each do |line|
      self.routing_lines << line
    end
  end

  def self.without_geometry
    where("latitude is null or longitude is null")
  end

  def self.with_geometry
    where("latitude is not null and longitude is not null")
  end

  def self.default_geometry!
    count = 0
    scoped.find_each do |stop_area|
      Chouette::StopArea.unscoped do
        count += 1 if stop_area.default_geometry!
      end
    end
    count
  end

  def default_geometry!
    new_geometry = default_geometry
    update_attribute :geometry, new_geometry if new_geometry
  end

  def default_geometry
    children_geometries = children.with_geometry.map(&:geometry).uniq
    GeoRuby::SimpleFeatures::Point.centroid children_geometries if children_geometries.present?
  end
  
  def generic_access_link_matrix
     matrix = Array.new
     access_points.each do |access_point|
       matrix += access_point.generic_access_link_matrix
     end
     matrix
  end

  def detail_access_link_matrix
     matrix = Array.new
     access_points.each do |access_point|
       matrix += access_point.detail_access_link_matrix
     end
     matrix
  end
  
  def children_at_base
    list = Array.new
    children_in_depth.each do |child|
      if child.area_type == 'Quay' || child.area_type == 'BoardingPosition'
        list << child
      end 
    end
    list
  end
  
  def parents
    list = Array.new
    if !parent.nil?
      list << parent
      list += parent.parents
    end
    list
  end
  
  def clean_invalid_access_links
    stop_parents = parents
    access_links.each do |link|
      unless stop_parents.include? link.access_point.stop_area
        link.delete
      end
    end
    children.each do |child|
      child.clean_invalid_access_links
    end
  end
  
end