Celestial/celeste_map_reader.rb

154 lines
4.1 KiB
Ruby
Raw Normal View History

2018-05-20 04:29:39 +00:00
require './rom'
# Value Types
# :boolean, :u8, :s16, :s32, :float, :lookup, :bin, :rle
class Element
attr_accessor :package, :name, :attributes, :children, :attributes_value_types;
def initialize
@attributes = {}
@attributes_value_types = {}
@children = []
@package = nil
@name = nil
end
def inspect
disp
end
def disp pre=0
fend = children.size == 0 ? (attributes.size == 0 ? " />" : "/>") : ">"
pres = ' ' * pre
fstart = if attributes.size == 0
"#{pres}<#{name}#{fend}"
else
"#{pres}<#{name}\n#{attributes_disp(pre)}\n#{pres}#{fend}"
end
if children.size > 0
"#{fstart}\n#{children_disp(pre)}#{pres}</#{name}>\n"
else
"#{fstart}\n"
end
end
def attributes_disp pre=0
pres = ' ' * (pre + 2)
attributes.map do |k, v|
value_type = attributes_value_types[k]
case value_type
when :boolean
if v
"#{pres}#{k}"
else
"#{pres}#{k}={#{v}}"
end
when :u8, :s16, :s32, :float
"#{pres}#{k}={#{v}}"
when :lookup
"#{pres}#{k}=\"#{v}\""
when :bin
"#{pres}#{k}={bin#{v.inspect}}"
when :rle
"#{pres}#{k}={rle#{v.inspect}}"
end
end.join("\n")
end
def children_disp pre=0
children.map { |c| c.disp(pre + 2) }.join("\n")
end
def get_children_by_name child_name
children.select { |c| c.name == child_name }
end
def find_child_by_name child_name
children.detect { |c| c.name == child_name }
end
def [] name
attributes[name.to_s]
end
end
class CelesteMapReader
attr_accessor :rom, :package, :string_lookup, :root
def initialize fn
@rom = ROM.from_file(fn)
raise "Not a celeste map" unless rom.read_str_varlen == 'CELESTE MAP'
@package = rom.read_str_varlen
@string_lookup = (0...rom.read_u16_le).map { rom.read_str_varlen }
@root = read_element
@root.package = @package
end
def read_element pre=0
element = Element.new
element.name = string_lookup[rom.read_u16_le]
# puts "#{" " * pre}<#{element.name}"
rom.read_byte.times do |i|
key = string_lookup[rom.read_u16_le]
value_type_enc = rom.read_byte
value = nil
value_type = nil
case value_type_enc
when 0
value = rom.read_bool
if value
# puts "#{" " * (pre + 2)}#{key}"
else
# puts "#{" " * (pre + 2)}#{key}={#{value}}"
end
value_type = :boolean
when 1
value = rom.read_byte
# puts "#{" " * (pre + 2)}#{key}={#{value}}"
value_type = :u8
when 2
value = rom.read_s16_le
# puts "#{" " * (pre + 2)}#{key}={#{value}}"
value_type = :s16
when 3
value = rom.read_s32_le
# puts "#{" " * (pre + 2)}#{key}={#{value}}"
value_type = :s32
when 4
value = rom.read_f32_le
# puts "#{" " * (pre + 2)}#{key}={#{value}}"
value_type = :float
when 5
value = string_lookup[rom.read_u16_le]
# puts "#{" " * (pre + 2)}#{key}=\"#{value}\""
value_type = :lookup
when 6
count = rom.read_varlen_le
base = rom.tell
bin = []
while rom.tell < base + count
bin << rom.read_byte
end
# value = rom.read_str_varlen
# # puts "#{" " * (pre + 2)}#{key}=\"#{value}\""
# puts "#{" " * (pre + 2)}#{key}={bin#{bin.inspect}}"
value_type = :bin
when 7
count = rom.read_u16_le
base = rom.tell
bin = []
while rom.tell < base + count
num = rom.read_byte
val = rom.read_byte
num.times { bin << val }
end
value = bin
# puts "#{" " * (pre + 2)}#{key}={rle#{bin.inspect}}"
value_type = :rle
end
element.attributes[key] = value
element.attributes_value_types[key] = value_type
end
# puts "#{" " * pre}>"
rom.read_u16_le.times do |j|
element.children << read_element(pre + 2)
end
# puts "#{" " * pre}</#{element.name}>"
element
end
def inspect
"<Bin #{package} />"
end
end