flexlay/netpanzer/level.rb

##  $Id$
## 
##  Flexlay - A Generic 2D Game Editor
##  Copyright (C) 2002 Ingo Ruhnke <grumbel@gmx.de>
##
##  This program is free software: you can redistribute it and/or modify
##  it under the terms of the GNU General Public License as published by
##  the Free Software Foundation, either version 3 of the License, or
##  (at your option) any later version.
##  
##  This program is distributed in the hope that it will be useful,
##  but WITHOUT ANY WARRANTY; without even the implied warranty of
##  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
##  GNU General Public License for more details.
##  
##  You should have received a copy of the GNU General Public License
##  along with this program.  If not, see <http://www.gnu.org/licenses/>.

require_relative "gameobjects.rb"

class Level
  attr_accessor :filename, :data, :editormap, :objects, :tilemap

  def initialize(*params)
    if params.length == 2 then
      (width, height) = params
      @data = NetPanzerFileStruct.new($tileset, width, height)
      @filename = nil
      generate_random_grass()
    elsif params.length == 1 then
      (@filename,) = params
      @data = NetPanzerFileStruct.new($tileset, @filename)
    end      

    @editormap = EditorMap.new(true)
    @objects   = ObjectLayer.new()
    @editormap.add_layer(@data.get_tilemap().to_layer())
    @editormap.add_layer(@objects.to_layer())
    @tilemap = @data.get_tilemap()

    if @filename then
      load_optfile(@filename[0..-5] + ".opt")
      load_spnfile(@filename[0..-5] + ".spn")
    end

    # FIXME: Data might not get freed since its 'recursively' refcounted
    @editormap.set_data(self)
  end

  def width()
    return @data.get_tilemap().get_width()
  end

  def height()
    return @data.get_tilemap().get_height()
  end

  def generate_random_grass()
    @data.get_tilemap().set_data(Array.new(width()*height()) {|i| 8097 + rand(16) })
  end

  def name()
    return @data.get_name()
  end

  def name=(name)
    @data.set_name(name)
  end

  def load_optfile(filename)
    f = File.new(filename)
    count = /ObjectiveCount: ([0-9]+)/.match(f.readline().chop)[1].to_i
    count.times{|i|
      f.readline() # Skip empty line
      name = /Name: (.+)/.match(f.readline().chop)
      loc  = /Location: ([0-9]+) ([0-9]+)/.match(f.readline().chop)
      GameObjects::Outpost.create(@objects,
                                  name,
                                  loc[1].to_i, loc[2].to_i)
    }
    f.close()
  end

  def load_spnfile(filename)
    f = File.new(filename)
    count = /SpawnCount: ([0-9]+)/.match(f.readline().chop)[1].to_i
    count.times{|i|
      loc  = /Location: ([0-9]+) ([0-9]+)/.match(f.readline().chop)
      GameObjects::SpawnPoint.create(@objects,
                                     loc[1].to_i, loc[2].to_i)
    }
    f.close()
  end

  def save_optfile(filename)
    outposts = @objects.get_objects().
      find_all {|obj| obj.get_data().is_a? GameObjects::Outpost }.
      map {|obj| obj.get_data() }    

    f = open(filename, "w")
    f.write("ObjectiveCount: %d\n\n" % outposts.length)
    outposts.each {|obj|
      f.write("Name: %s\n" % obj.name)
      f.write("Location: %d %d\n\n" % [obj.x, obj.y])
    }
    f.close()
  end

  def save_spnfile(filename)
    spawnpoints = @objects.get_objects().
      find_all {|obj| obj.get_data().is_a? GameObjects::SpawnPoint }.
      map {|obj| obj.get_data() }

    f = open(filename, "w")

    f.write("SpawnCount: %d\n" % spawnpoints.length)
    spawnpoints.each {|obj|
      f.print("Location: %d %d\n" % [obj.x, obj.y])
    }
    f.close()
  end

  def save(filename)
    flatten()

    if filename[-4..-1] == ".npm"
      data.save(filename)
      save_optfile(filename[0..-5] + ".opt")
      save_spnfile(filename[0..-5] + ".spn")
    else
      raise "Fileextension not valid, must be .npm!"
    end
  end

  def activate(workspace)
    puts "Level activate"
    workspace.set_map(@editormap)
    TilemapLayer.set_current(@data.get_tilemap())
    ObjectLayer.set_current(@objects)
  end


  def flatten()
    # Converts all objects to the Tilemap
    @objects.get_objects().each { |obj|
      obj.get_data().draw_to_tilemap(@tilemap)
    }
  end
  
  def unflatten()
    # Reconstructs objects from the Tilemap
    data   = @tilemap.get_data()
    width  = @tilemap.get_width()
    height = @tilemap.get_height()
    first_tiles = {}
    $brushes.each_with_index{|i,index| first_tiles[i[0]] = index }

    (0..height-1).each{|y|
      (0..width-1).each{|x|
        tile = data[width*y + x]
        if tile != 0 && first_tiles.has_key?(tile) then
          # Insert checking for dups here
          GameObjects::TileObject.create(@objects, first_tiles[data[width*y + x]], x, y)
        end
      }
    }
  end
end

# EOF #