rixmap 0.1.0

data/lib/rixmap/format/png/imageio.rb

@@ -0,0 +1,288 @@
+# -*- coding: utf-8 -*-
+require 'zlib'
+require_relative 'chunk'
+
+#
+module Rixmap
+  module Format
+    module PNG
+
+      # スキャンライン用フィルタ実装.
+      #
+      # @see http://www.milk-island.net/document/png/#9Filters
+      class AdaptiveFilter
+        # デフォルトフィルタ実装を初期化します.
+        #
+        # @param [Rixmap::Mode] mode 画像形式
+        def initialize(mode)
+          @prev = nil
+          @mode = mode
+        end
+
+        # フィルタ関数
+        #
+        # @param [Array] line 対象スキャンライン
+        # @param [Integer] method フィルタメソッド
+        # @return [Array] フィルタ処理後スキャンライン
+        def filt(line, method = ADAPTIVEFILTER_NONE)
+          base_line = Rixmap::Binary.new(line)
+          filt_line = Rixmap::Binary.new([method])
+          case method
+          when ADAPTIVEFILTER_NONE
+            filt_line += base_line
+          else
+            raise NotImplementedError.new("Unsupported Filter Method: #{method}")
+          end
+
+          @prev = base_line
+          return filt_line
+        end
+
+        # 復元関数
+        #
+        # @param [Array] line 対象スキャンライン
+        def recon(line)
+          base_line  = Rixmap::Binary.new(line)
+          recon_line = Rixmap::Binary.new
+          method = base_line.shift
+
+          case method
+          when ADAPTIVEFILTER_NONE
+            recon_line += base_line
+          else
+            raise NotImplementedError.new("Unsupported Filter Method: #{method}")
+          end
+
+          @prev = base_line
+          return recon_line
+        end
+      end
+
+      # PNG入出力実装クラス.
+      #
+      # さすがにPNGフル実装はきついですな(´・ω・`)
+      class PNGImageIO < Rixmap::ImageIO::BaseImageIO
+        # 指定マジックデータがPNG画像のものかを判定します.
+        #
+        # @param [String] magic 画像先頭データ
+        # @return [Boolean] PNG形式のシグネチャならtrue
+        def self.readable?(magic)
+          return (magic[0, FILE_SIGNATURE_SIZE] == FILE_SIGNATURE)
+        end
+
+        # 指定画像をPNGとしてエンコードできるかを返します.
+        #
+        # @param [Rixmap::Image] image 対象画像
+        # @return [Boolean] PNGとしてエンコードできる場合はtrue
+        def self.writable?(image)
+          return true
+        end
+
+        # 画像をPNG形式でエンコードします.
+        #
+        # @param [Rixmap::Image] image 対象画像
+        # @param [Hash] options オプションパラメータ
+        # @return [String] エンコードされた画像データ
+        def encode(image, options={})
+          # イメージヘッダ
+          ihdr = Chunk::IHDRChunk.new
+          ihdr.width  = image.width
+          ihdr.height = image.height
+          ihdr.depth  = 8 #image.mode.depth
+
+          # 必要チャンク
+          plte = nil
+
+          # ピクセルデータ
+          pixels = ''.force_encoding(Encoding::BINARY)
+          filter = AdaptiveFilter.new(image.mode)
+
+          # 画像種別による処理
+          case image.mode
+          when Rixmap::INDEXED
+            ihdr.colortype = COLORTYPE_INDEXED
+
+            # パレットを設定
+            plte = Chunk::PLTEChunk.new
+            image.palette.each_with_index do |color, i|
+              plte[i] = [color.red, color.green, color.blue]
+            end
+
+            # ピクセルを作成
+            image.each_line do |line|
+              # ByteArrayはto_bytesできる
+              pixels.concat(filter.filt(line).to_str)
+            end
+          when Rixmap::GRAYSCALE
+            ihdr.colortype = COLORTYPE_GRAYSCALE
+            image.each_line do |line|
+              pixels.concat(filter.filt(line).to_str)
+            end
+          when Rixmap::GRAYALPHA
+            ihdr.colortype = COLORTYPE_GRAYSCALE_WITH_ALPHA
+            image.each_line do |line|
+              pixels.concat(filter.filt(line.to_s('LA')).to_str)
+            end
+          when Rixmap::RGB
+            ihdr.colortype = COLORTYPE_TRUECOLOR
+            image.each_line do |line|
+              pixels.concat(filter.filt(line.to_s('RGB')).to_str)
+            end
+          when Rixmap::RGBA
+            ihdr.colortype = COLORTYPE_TRUECOLOR_WITH_ALPHA
+            image.each_line do |line|
+              pixels.concat(filter.filt(line.to_s('RGBA')).to_str)
+            end
+          else
+            raise ArgumentError.new("Unknown image mode: #{image.mode}")
+          end
+
+          # ピクセルを圧縮
+          deflated_pixels = Zlib.deflate(pixels, Zlib::BEST_COMPRESSION)
+
+          # IDATチャンクを生成
+          idat = Chunk::IDATChunk.new
+          idat.data = deflated_pixels
+
+          # チャンクリスト
+          chunks = []
+          chunks << ihdr
+          chunks << plte unless plte.nil?
+          chunks << idat
+          chunks << Chunk::IENDChunk.new
+
+          # PNGストリームを構築
+          png_data = ''
+          png_data.concat(FILE_SIGNATURE)
+          png_data.force_encoding(Encoding::BINARY)
+          chunks.each do |chunk|
+            chunk_data = chunk.data
+            chunk_size = chunk_data.bytesize
+            chunk_fulldata = chunk.type + chunk_data
+            chunk_crc = Zlib.crc32(chunk_fulldata)
+            #chunk_crc = CRC.crc(chunk_fulldata)   # 自前実装
+            png_data.concat([chunk_size].pack('N'))
+            png_data.concat(chunk_fulldata)
+            png_data.concat([chunk_crc].pack('N'))
+          end
+
+          return png_data
+        end
+
+        # バイトデータをPNG画像として復元します.
+        #
+        # @param [String] data 入力データ
+        # @param [Hash] options オプションパラメータ
+        # @return [Rixmap::Image] 復元された画像
+        def decode(data, options={})
+          if data.byteslice(0, FILE_SIGNATURE_SIZE) != FILE_SIGNATURE
+            raise ArgumentError.new("InvalidSignature detected. Input is not PNG Image Data.")
+          end
+
+          # チャンクを分割
+          offset = FILE_SIGNATURE_SIZE
+          length = data.bytesize
+          chunks = []
+          while offset < length
+            chunk_size = data.byteslice(offset, offset + 4).unpack('N')[0]; offset += 4
+            chunk_data = nil
+            chunk_type = nil
+            chunk_crc  = nil
+            if chunk_size > 0
+              chunk_type, chunk_data, chunk_crc = data.byteslice(offset, offset + (chunk_size + 8)).unpack("a4a#{chunk_size}N")
+            elsif chunk_size == 0
+              chunk_type, chunk_crc = data.byteslice(offset, offset + 8).unpack("a4N")
+            else
+              raise RuntimeError.new("Invalid Chunk Size: #{chunk_size}")
+            end
+            offset += (chunk_size + 8)
+
+            chunk = Chunk.get(chunk_type).new
+            chunk.data = chunk_data
+
+            chunks << chunk
+            break if chunk.type == Chunk::IENDChunk::TYPE
+          end
+
+          # IHDRチャンクを処理
+          ihdr = chunks.shift
+          unless ihdr.type == Chunk::IHDRChunk::TYPE
+            raise ArgumentError.new("Illegal Chunk Order. First chunk must to be IHDR chunk.")
+          end
+
+          image_width  = ihdr.width
+          image_height = ihdr.height
+          image_depth  = ihdr.depth
+          colortype    = ihdr.colortype
+
+          # 残りチャンクを処理
+          image_data = ''
+          palette    = nil
+          chunks.each do |chunk|
+            case chunk
+            when Chunk::IDATChunk
+              image_data.concat(chunk.data)
+            when Chunk::PLTEChunk
+              palette = Rixmap::Palette.new(2 ** image_depth)
+              palette.size.times do |i|
+                palette[i] = chunk[i]
+              end
+            when Chunk::IENDChunk
+              break
+            end
+          end
+
+          # ピクセルを復元
+          pixels = Zlib.inflate(image_data).each_byte
+
+          # ベース画像を作成
+          image = nil
+          case colortype
+          when COLORTYPE_INDEXED
+            image  = Rixmap::Image.new(Rixmap::INDEXED, image_width, image_height, :palette => palette)
+            filter = AdaptiveFilter.new(image.mode)
+            pixels.each_slice(image_width + 1).each_with_index do |bytes, i|
+              image[i] = filter.recon(bytes)
+            end
+          when COLORTYPE_GRAYSCALE
+            image  = Rixmap::Image.new(Rixmap::GRAYSCALE, image_width, image_height)
+            filter = AdaptiveFilter.new(image.mode)
+            pixels.each_slice(image_width + 1).each_with_index do |bytes, i|
+              image[i] = filter.recon(bytes)
+            end
+          when COLORTYPE_GRAYSCALE_WITH_ALPHA
+            image  = Rixmap::Image.new(Rixmap::GRAYALPHA, image_width, image_height)
+            filter = AdaptiveFilter.new(image.mode)
+            pixels.each_slice((image_width * 2) + 1).each_with_index do |bytes, i|
+              image[i] = filter.recon(bytes)
+            end
+          when COLORTYPE_TRUECOLOR
+            image  = Rixmap::Image.new(Rixmap::RGB, image_width, image_height)
+            filter = AdaptiveFilter.new(image.mode)
+            pixels.each_slice((image_width * 3) + 1).each_with_index do |bytes, i|
+              image[i] = filter.recon(bytes)
+            end
+          when COLORTYPE_TRUECOLOR_WITH_ALPHA
+            image  = Rixmap::Image.new(Rixmap::RGBA, image_width, image_height)
+            filter = AdaptiveFilter.new(image.mode)
+            pixels.each_slice((image_width * 4) + 1).each_with_index do |bytes, i|
+              image[i] = filter.recon(bytes)
+            end
+          else
+            raise RuntimeError.new("Unsupported Colo-Type: #{colortype}")
+          end
+
+          return image
+        end
+      end
+
+      Rixmap::ImageIO.register(:PNG, PNGImageIO, [".png"])
+    end
+
+  end
+end
+
+
+#==============================================================================#
+# $Id: imageio.rb,v 57b1fb2cd6a6 2014/04/20 12:21:27 chikuchikugonzalez $
+# vim: set sts=2 ts=2 sw=2 expandtab:

data/lib/rixmap/format/png.rb

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+
+#
+module Rixmap
+  module Format
+
+    # PNGフォーマット対応実装モジュール.
+    module PNG
+
+      # PNGシグネチャ
+      FILE_SIGNATURE      = "\x89PNG\r\n\x1A\n".force_encoding(Encoding::BINARY)
+
+      # PNGシグネチャサイズ
+      FILE_SIGNATURE_SIZE = 8
+
+      # グレースケール形式
+      #
+      # ビット幅は 1, 2, 4, 8, 16
+      COLORTYPE_GRAYSCALE = 0x00
+
+      # RGBトゥルーカラー形式
+      #
+      # ビット幅は 8, 16
+      COLORTYPE_TRUECOLOR = 0x02
+
+      # インデックスカラー形式
+      #
+      # ビット幅は 1, 2, 4, 8
+      COLORTYPE_INDEXED = 0x03
+
+      # グレースケール形式 with 透明度
+      #
+      # ビット幅は 8, 16
+      COLORTYPE_GRAYSCALE_WITH_ALPHA = 0x04
+
+      # RGBトゥルーカラー形式 with 透明度
+      #
+      # ビット幅は 8, 16
+      COLORTYPE_TRUECOLOR_WITH_ALPHA = 0x06
+
+      # Deflate圧縮メソッド
+      COMPRESSION_DEFLATE = 0x00
+
+      # 基本のフィルタタイプ
+      FILTER_ADAPTIVE = 0x00
+
+      # インターレースなし
+      INTERLACE_NONE = 0x00
+
+      # Adam7インターレース
+      INTERLACE_ADAM7 = 0x01
+
+      # なにもしない基本フィルタ
+      ADAPTIVEFILTER_NONE = 0x00
+
+      # 左側差分基本フィルタ
+      ADAPTIVEFILTER_SUB = 0x01
+
+      # 上側差分基本フィルタ
+      ADAPTIVEFILTER_UP = 0x02
+
+      # 平均基本フィルタ
+      ADAPTIVEFILTER_AVERAGE = 0x03
+
+      # 周辺から近い色をとって、その差分をとるようなの
+      ADAPTIVEFILTER_PEATH = 0x04
+    end
+
+  end
+end
+
+require_relative 'png/chunk'
+require_relative 'png/imageio'
+
+
+#==============================================================================#
+# $Id: png.rb,v 57b1fb2cd6a6 2014/04/20 12:21:27 chikuchikugonzalez $
+# vim: set sts=2 ts=2 sw=2 expandtab: