module Floorplanner
  class Document

    def to_dae(design_id,out_path,xrefs=false)
      @design = Design.new(@xml,design_id)
      @design.build_geometries
      @design.save_textures(File.dirname(out_path)) unless xrefs
      dae = File.new(out_path,'w')
      dae.write(@design.to_dae(xrefs))
      dae.close
    end

  end

  module ColladaExport
    DESIGN_QUERY   = "/project/floors/floor/designs/design[id='%s']"
    ASSET_QUERY    = DESIGN_QUERY+"/assets/asset[@id='%s']"
    ASSETS_QUERY   = DESIGN_QUERY+"/assets/asset"
    OBJECTS_QUERY  = DESIGN_QUERY+"/objects/object"

    def to_dae(xrefs=false)
      raise "No geometries to export. Call build_geometries first" unless @areas && @walls
      @assets   = assets
      @elements = objects

      # somehow...
      @walls.reverse
      @areas.each {|a| a.reverse}
      @xrefs = xrefs

      template = ERB.new(
        File.read(
          File.join(File.dirname(__FILE__), '..', '..', 'views', 'design.dae.erb')))
      template.result(binding)
    end

    def assets
      return @assets if @assets
      @assets = {}
      @xml.find(ASSETS_QUERY % @design_id).each do |asset_node|
        asset_id = asset_node.attributes['id']
        name  = asset_node.find('name').first.content
        url3d = asset_node.find('url3d').first
        next unless url3d
        url3d = url3d.content

        # TODO: store asset bounding box
        asset = Floorplanner::Asset.get(asset_id,name,url3d)
        next unless asset
        @assets.store(asset_id, asset)
      end
      @assets
    end

    def objects
      result = []
      @xml.find(OBJECTS_QUERY % @design_id).each do |object|
        begin
          refid = object.find('asset').first.attributes['refid']
          next unless assets[refid]

          asset    = assets[refid]
          position = Geom::Number3D.from_str(object.find('points').first.content)
          # correct Flash axis issues
          position.y *= -1.0

          # correct Flash rotation issues
          rotation = unless object.find('rotation').empty?
            object.find('rotation').first.content
          else
            '0 0 0'
          end
          rotation = Geom::Number3D.from_str(rotation)
          rotation.z += 360 if rotation.z < 0
          rotation.z += 180

          # find proper scale for object
          size     = object.find('size').first.content
          scale    = asset.scale_ratio(Geom::Number3D.from_str(size))

          mirrored = object.find('mirrored').first
          reflection = Geom::Matrix3D.reflection(Geom::Plane.new(Geom::Number3D.new(0.0,1.0,0.0), Geom::Number3D.new))
          if mirrored
            mirror   = Geom::Number3D.from_str(mirrored)
            if mirror.x != 0 || mirror.y != 0 || mirror.z != 0
              mirror.x = mirror.x > 0 ? 1 : 0
              mirror.y = mirror.y > 0 ? 1 : 0
              mirror.z = mirror.z > 0 ? 1 : 0

              origin = Geom::Number3D.new
              plane  = Geom::Plane.new(mirror,origin)
              reflection = Geom::Matrix3D.reflection(plane).multiply reflection
            end
          end

          result << {
            :asset => asset,
            :position => position,
            :rotation => rotation,
            :scale    => scale,
            :matrix   => reflection
          }
        rescue
          # TODO: handle text
        end
      end
      result
    end

    def save_textures(root_path)
      img_path = File.join(root_path,'textures')
      FileUtils.mkdir_p img_path
      assets.each_value do |asset|
        asset.save_textures img_path
      end
    end
  end
end