psd 0.3.2

data/lib/psd/layer_info.rb

@@ -0,0 +1,22 @@
+class PSD
+  # Parent class for all of the extra layer info.
+  class LayerInfo
+    attr_reader :data
+
+    # The value of the key as used in the PSD format.
+    class << self; attr_accessor :key; end
+    @key = ""
+
+    def initialize(file, length)
+      @file = file
+      @length = length
+      @section_end = @file.tell + @length
+      @data = {}
+    end
+
+    # Override this - default seeks to end of section
+    def parse
+      @file.seek @section_end
+    end
+  end
+end

data/lib/psd/layer_info/fill_opacity.rb

@@ -0,0 +1,13 @@
+require_relative '../layer_info'
+
+class PSD
+  class FillOpacity < LayerInfo
+    @key = 'iOpa'
+
+    attr_reader :enabled
+
+    def parse
+      @enabled = @file.read_boolean
+    end
+  end
+end

data/lib/psd/layer_info/layer_id.rb

@@ -0,0 +1,13 @@
+require_relative '../layer_info'
+
+class PSD
+  class LayerID < LayerInfo
+    @key = 'lyid'
+
+    attr_reader :id
+
+    def parse
+      @id = @file.read_int
+    end
+  end
+end

data/lib/psd/layer_info/layer_name_source.rb

@@ -0,0 +1,12 @@
+require_relative '../layer_info'
+
+class PSD
+  class LayerNameSource < LayerInfo
+    @key = 'lnsr'
+    
+    def parse
+      @data = @file.read_int
+      return self
+    end
+  end
+end

data/lib/psd/layer_info/layer_section_divider.rb

@@ -0,0 +1,35 @@
+require_relative '../layer_info'
+
+class PSD
+  class LayerSectionDivider < LayerInfo
+    @key = 'lsct'
+
+    attr_reader :layer_type, :is_folder, :is_hidden
+
+    SECTION_DIVIDER_TYPES = [
+      "other",
+      "open folder",
+      "closed folder",
+      "bounding section divider"
+    ]
+
+    def initialize(file, length)
+      super
+
+      @is_folder = false
+      @is_hidden = false
+    end
+
+    def parse
+      code = @file.read_int
+      @layer_type = SECTION_DIVIDER_TYPES[code]
+
+      case code
+      when 1, 2 then @is_folder = true
+      when 3 then @is_hidden = true
+      end
+
+      return self
+    end
+  end
+end

data/lib/psd/layer_info/legacy_typetool.rb

@@ -0,0 +1,88 @@
+require_relative '../layer_info'
+require_relative 'typetool'
+
+class PSD
+  class LegacyTypeTool < TypeTool
+    @key = 'tySh'
+    
+    def parse      
+      version = @file.read_short
+      parse_transform_info
+
+      # Font Information
+      version = @file.read_short
+
+      faces_count = @file.read_short
+      @data[:face] = []
+
+      faces_count.times do |i|
+        @data[:face][i] = {}
+        @data[:face][i][:mark] = @file.read_short
+        @data[:face][i][:font_type] = @file.read_int
+        @data[:face][i][:font_name] = PascalString.read(@file)
+        @data[:face][i][:font_family_name] = PascalString.read(@file)
+        @data[:face][i][:font_style_name] = PascalString.read(@file)
+        @data[:face][i][:script] = @file.read_short
+        @data[:face][i][:number_axes_vector] = @file.read_int
+        @data[:face][i][:vector] = []
+
+        @data[:face][i][:number_axes_vector].times do |j|
+          @data[:face][i][:vector] << @file.read_int
+        end
+      end
+
+      # Style Information
+      styles_count = @file.read_short
+      @data[:style] = []
+
+      styles_count.times do |i|
+        @data[:style][i] = {}
+        @data[:style][i][:mark] = @file.read_short
+        @data[:style][i][:face_mark] = @file.read_short
+        @data[:style][i][:size] = @file.read_int
+        @data[:style][i][:tracking] = @file.read_int
+        @data[:style][i][:kerning] = @file.read_int
+        @data[:style][i][:leading] = @file.read_int
+        @data[:style][i][:base_shift] = @file.read_int
+        @data[:style][i][:auto_kern] = @file.read_boolean
+
+        # Bleh
+        @file.read 1
+
+        @data[:style][i][:rotate] = @file.read_boolean
+      end
+
+      # Text information
+      @data[:type] = @file.read_short
+      @data[:scaling_factor] = @file.read_int
+      @data[:character_count] = @file.read_int
+      @data[:horz_place] = @file.read_int
+      @data[:vert_place] = @file.read_int
+      @data[:select_start] = @file.read_int
+      @data[:select_end] = @file.read_int
+
+      lines_count = @file.read_short
+      @data[:line] = []
+
+      lines_count.times do |i|
+        @data[:line][i] = {}
+        @data[:line][i][:char_count] = @file.read_int
+        @data[:line][i][:orientation] = @file.read_short
+        @data[:line][i][:alignment] = @file.read_short
+        @data[:line][i][:actual_char] = @file.read_short
+        @data[:line][i][:style] = @file.read_short
+      end
+
+      # Color information
+      @data[:color] = @file.read_space_color
+      @data[:antialias] = @file.read_boolean
+
+      return self
+    end
+
+    # Not sure where this is stored right now
+    def text_value
+      ""
+    end
+  end
+end

data/lib/psd/layer_info/object_effects.rb

@@ -0,0 +1,16 @@
+require_relative '../layer_info'
+
+class PSD
+  class ObjectEffects < LayerInfo
+    @key = 'lfx2'
+    
+    def parse
+      version = @file.read_int
+      descriptor_version = @file.read_int
+
+      @data = Descriptor.new(@file).parse
+
+      return self
+    end
+  end
+end

data/lib/psd/layer_info/placed_layer.rb

@@ -0,0 +1,14 @@
+require_relative '../layer_info'
+
+class PSD
+  class PlacedLayer < LayerInfo
+    @key = 'SoLd'
+
+    def parse
+      # Useless id/version info
+      @file.seek 12, IO::SEEK_CUR
+      
+      @data = Descriptor.new(@file).parse
+    end
+  end
+end

data/lib/psd/layer_info/reference_point.rb

@@ -0,0 +1,16 @@
+require_relative '../layer_info'
+
+class PSD
+  class ReferencePoint < LayerInfo
+    @key = 'fxrp'
+
+    attr_reader :x, :y
+
+    def parse
+      @x = @file.read_double
+      @y = @file.read_double
+
+      return self
+    end
+  end
+end

data/lib/psd/layer_info/typetool.rb

@@ -0,0 +1,127 @@
+require_relative '../layer_info'
+
+class PSD
+  # Parses and provides information about text areas within layers in
+  # the document.
+  class TypeTool < LayerInfo
+    @key = 'TySh'
+    
+    # Parse all of the text data in the layer.
+    def parse
+      version = @file.read_short
+      parse_transform_info
+
+      text_version = @file.read_short
+      descriptor_version = @file.read_int
+
+      @data[:text] = Descriptor.new(@file).parse
+      @data[:text]['EngineData']
+        .encode!('UTF-8', 'MacRoman')
+        .delete!("\000")
+
+      @data[:engine_data] = nil
+      begin
+        parser.parse!
+        @data[:engine_data] = parser.result
+      rescue Exception => e
+        puts e.message
+      end
+
+      warpVersion = @file.read_short
+      descriptor_version = @file.read_int
+
+      @data[:warp] = Descriptor.new(@file).parse
+      [:left, :top, :right, :bottom].each do |pos|
+        @data[pos] = @file.read_int
+      end
+
+      return self
+    end
+
+    # Extracts the text within the text area. In the event that psd-enginedata fails
+    # for some reason, we attempt to extract the text using some rough regex.
+    def text_value
+      if engine_data.nil?
+        # Something went wrong, lets hack our way through.
+        /\/Text \(˛ˇ(.*)\)$/.match(@data[:text]['EngineData'])[1].gsub /\r/, "\n"
+      else
+        engine_data.EngineDict.Editor.Text
+      end
+    end
+    alias :to_s :text_value
+
+    # Gets all of the basic font information for this text area. This assumes that
+    # the first font is the only one you want.
+    def font
+      {
+        name: fonts.first,
+        sizes: sizes,
+        colors: colors,
+        css: parser.to_css
+      }
+    end
+
+    # Returns all fonts listed for this layer, since fonts are defined on a 
+    # per-character basis.
+    def fonts
+      return [] if engine_data.nil?
+      engine_data.ResourceDict.FontSet.map(&:Name)
+    end
+
+    # Return all font sizes for this layer.
+    def sizes
+      return [] if engine_data.nil?
+      engine_data.EngineDict.StyleRun.RunArray.map do |r|
+        r.StyleSheet.StyleSheetData.FontSize
+      end.uniq
+    end
+
+    # Return all colors used for text in this layer. The colors are returned in RGB
+    # format as an array of arrays.
+    #
+    # => [[255, 0, 0], [0, 0, 255]]
+    def colors
+      return [] if engine_data.nil?
+      engine_data.EngineDict.StyleRun.RunArray.map do |r|
+        next unless r.StyleSheet.StyleSheetData.key?('FillColor')
+        r.StyleSheet.StyleSheetData.FillColor.Values.map do |v|
+          (v * 255).round
+        end
+      end.uniq
+    end
+
+    def engine_data
+      @data[:engine_data]
+    end
+
+    def parser
+      @parser ||= PSD::EngineData.new(@data[:text]['EngineData'])
+    end
+
+    def to_hash
+      {
+        value:      text_value,
+        font:       font,
+        left:       left,
+        top:        top,
+        right:      right,
+        bottom:     bottom,
+        transform:  transform
+      }
+    end
+
+    def method_missing(method, *args, &block)
+      return @data[method] if @data.has_key?(method)
+      return super
+    end
+
+    private
+
+    def parse_transform_info
+      @data[:transform] = {}
+      [:xx, :xy, :yx, :yy, :tx, :ty].each do |t|
+        @data[:transform][t] = @file.read_double
+      end
+    end
+  end
+end

data/lib/psd/layer_info/unicode_name.rb

@@ -0,0 +1,18 @@
+require_relative '../layer_info'
+
+class PSD
+  class UnicodeName < LayerInfo
+    @key = 'luni'
+
+    def parse
+      pos = @file.tell
+      len = @file.read_int * 2
+      @data = @file.read(len).unpack("A#{len}")[0].encode('UTF-8').delete("\000")
+
+      # The name seems to be padded with null bytes. This is the easiest solution.
+      @file.seek pos + @length
+
+      return self
+    end
+  end
+end

data/lib/psd/layer_info/vector_mask.rb

@@ -0,0 +1,25 @@
+require_relative '../layer_info'
+
+class PSD
+  class VectorMask < LayerInfo
+    @key = 'vmsk'
+
+    attr_reader :invert, :not_link, :disable, :paths
+
+    def parse
+      version = @file.read_int
+      tag = @file.read_int
+
+      @invert = tag & 0x01
+      @not_link = (tag & (0x01 << 1)) > 0
+      @disable = (tag & (0x01 << 2)) > 0
+
+      num_records = (@length - 8) / 26
+      
+      @paths = []
+      num_records.times do
+        @paths << PathRecord.new(@file)
+      end
+    end
+  end
+end

data/lib/psd/layer_mask.rb

@@ -0,0 +1,106 @@
+class PSD
+  # Covers parsing the global mask and controls parsing of all the
+  # layers/folders in the document.
+  class LayerMask
+    include Section
+
+    attr_reader :layers, :global_mask
+
+    # Store a reference to the file and the header and initialize the defaults.
+    def initialize(file, header)
+      @file = file
+      @header = header
+
+      @layers = []
+      @merged_alpha = false
+      @global_mask = nil
+      @extras = []
+    end
+
+    # Allows us to skip this section because it starts with the length of the section
+    # stored as an integer.
+    def skip
+      @file.seek @file.read_int, IO::SEEK_CUR
+      return self
+    end
+
+    # Parse this section, including all of the layers and folders. Once implemented, this
+    # will also trigger parsing of the channel images for each layer.
+    def parse
+      start_section
+
+      mask_size = @file.read_int
+      finish = @file.tell + mask_size
+
+      return self if mask_size <= 0
+
+      layer_info_size = Util.pad2(@file.read_int)
+
+      if layer_info_size > 0
+        layer_count = @file.read_short
+
+        if layer_count < 0
+          layer_count = layer_count.abs
+          @merged_alpha = true
+        end
+
+        if layer_count * (18 + 6 * @header.channels) > layer_info_size
+          raise "Unlikely number of layers parsed: #{layer_count}"
+        end
+
+        @layer_section_start = @file.tell
+        layer_count.times do
+          @layers << PSD::Layer.new(@file).parse
+        end
+
+        layers.each do |layer|
+          @file.seek 8, IO::SEEK_CUR and next if layer.folder? || layer.folder_end?
+
+          layer.parse_channel_image!(@header)
+        end
+      end
+
+      # Layers are parsed in reverse order
+      layers.reverse!
+      group_layers
+
+      # Temporarily seek to the end of this section
+      @file.seek finish
+      end_section
+
+      return self
+    end
+
+    # Export the mask and all the children layers to a file.
+    def export(outfile)
+      if @layers.size == 0
+        # No data, just read whatever's here.
+        return outfile.write @file.read(@section_end[:all] - start_of_section)
+      end
+
+      # Read the initial mask data since it won't change
+      outfile.write @file.read(@layer_section_start - @file.tell)
+
+      @layers.reverse.each do |layer|
+        layer.export(outfile)
+      end
+
+      outfile.write @file.read(end_of_section - @file.tell)
+    end
+
+    private
+
+    def group_layers
+      group_layer = nil
+      layers.each do |layer|
+        if layer.folder?
+          group_layer = layer
+        elsif layer.folder_end?
+          group_layer = nil
+        else
+          layer.group_layer = layer
+        end
+      end
+    end
+  end
+end