rixmap 0.1.0

data/src/rixmap/image.hxx

@@ -0,0 +1,371 @@
+// -*- coding: utf-8 -*-
+/**
+ * 画像情報クラスデータ定義ヘッダ
+ */
+#pragma once
+#include "common.hxx"
+#include "mode.hxx"
+#include "color.hxx"
+#include "palette.hxx"
+
+namespace Rixmap {
+    /**
+     * Ruby用画像情報クラスデータ
+     */
+    class ImageData {
+    private:        // 非公開メンバ
+        ModeInfo _mode;     // 画像形式
+        int32_t  _width;    // 横幅
+        int32_t  _height;   // 高さ
+        bool     _ready;    // 準備完了しているか
+
+        // ピクセルデータ系
+        std::map<Channel, uint8_t**> _data;
+        VALUE _palette;
+
+    public:         // コンストラクタ
+        /**
+         * デフォルトコンストラクタ
+         */
+        ImageData() : _mode(Mode::NONE), _width(0), _height(0), _ready(false), _palette(Qnil) {}
+
+        /**
+         * 一応初期データつき
+         */
+        ImageData(Mode mode, int32_t width, int32_t height) : _mode(Mode::NONE), _width(0), _height(0), _ready(false), _palette(Qnil) {
+            this->init(mode, width, height);
+        }
+
+        /**
+         * デストラクタ
+         */
+        ~ImageData() {
+            // TODO ピクセルデータの解放処理
+            if (this->_ready) {
+                for (auto it = this->_data.begin(); it != this->_data.end(); it++) {
+                    uint8_t** plane = it->second;
+                    if (plane != NULL) {
+                        for (int32_t h = 0; h < this->_height; h++) {
+                            if (plane[h] != NULL) {
+                                ruby_xfree(plane[h]);
+                            }
+                        }
+                        ruby_xfree(plane);
+                        it->second = NULL;
+                    }
+                }
+                this->_data.clear();
+            }
+        }
+
+    public:         // プロパティメンバ関数
+        /**
+         * 画像形式定数を取得します.
+         */
+        inline Mode getMode() const { return this->_mode.getMode(); }
+
+        /**
+         * 画像形式情報を取得します.
+         */
+        //inline ModeInfo& getModeInfo() { return this->_mode; }
+        inline const ModeInfo& getModeInfo() const { return this->_mode; }
+
+        /**
+         * 横幅を取得します.
+         */
+        inline int32_t getWidth() const { return this->_width; }
+
+        /**
+         * 高さを取得します.
+         */
+        inline int32_t getHeight() const { return this->_height; }
+
+        /**
+         * 画像サイズ (横幅と高さ) をタプル形式で取得します.
+         */
+        inline std::tuple<int32_t, int32_t> getSize() const { return std::make_tuple(this->_width, this->_height); }
+
+        /**
+         * パレットオブジェクトを取得します.
+         */
+        inline VALUE getPalette() const { return this->_palette; }
+
+        /**
+         * パレットオブジェクトの内部データポインタを返します.
+         * パレットがnilの場合はNULLが返されます.
+         */
+        inline const PaletteData* getPaletteData() const {
+            if (NIL_P(this->_palette)) {
+                return NULL;
+            } else {
+                return rixmap_unwrap<Rixmap::PaletteData>(this->_palette);
+            }
+        }
+
+        /**
+         * パレットを設定します.
+         */
+        inline void setPalette(VALUE obj) {
+            // TODO 型チェック
+            // TODO INDEXED以外で設定を許容するか
+            this->_palette = obj;
+        }
+
+    public:         // 公開メンバ関数
+        /**
+         * 指定座標が画像範囲内かどうかを返します.
+         */
+        inline bool isInside(int32_t x, int32_t y) const {
+            return ((0 <= x && x < this->_width) && (0 <= y && y < this->_height));
+        }
+
+        /**
+         * 指定座標が画像の範囲外かどうかを返します.
+         */
+        inline bool isOutside(int32_t x, int32_t y) const {
+            return ((x < 0 || this->_width <= x) || (y < 0 || this->_height <= y));
+        }
+
+        /**
+         * 指定チャンネルデータを格納するデータポインタから取得します.
+         * 対象チャンネルを持っていない場合はNULLになります.
+         */
+        inline uint8_t** get(Channel channel) {
+            auto found = this->_data.find(channel);
+            if (found != this->_data.end()) {
+                return this->_data[channel];
+            } else {
+                return NULL;
+            }
+        }
+
+        /**
+         * 指定チャンネルの指定行番号に対応するバイト列へのポインタから取得します.
+         * 行番号が範囲外の場合か、非対応のチャンネルの場合はNULLになります.
+         */
+        inline uint8_t* get(Channel channel, int32_t lineno) {
+            if (0 <= lineno && lineno < this->_height) {
+                uint8_t** plane = this->get(channel);
+                if (plane != NULL) {
+                    return plane[lineno];
+                }
+            }
+            return NULL;
+        }
+
+        /**
+         * 指定座標の指定チャンネルのバイトデータを取得します.
+         *
+         * Notice: 範囲外だったりチャンネル持ってないときは0になるので、
+         *         実はバイトデータが0の時と判別できないんですよコレ
+         */
+        inline uint8_t get(Channel channel, int32_t x, int32_t y) {
+            if (0 <= x && x < this->_width && 0 <= y && y < this->_height) {
+                uint8_t** plane = this->get(channel);
+                if (plane != NULL) {
+                    return plane[y][x];
+                }
+            }
+            return 0;
+        }
+
+        /**
+         * 指定チャンネルの指定座標のバイトデータを更新します.
+         */
+        inline bool set(Channel channel, int32_t x, int32_t y, uint8_t value) {
+            if (0 <= x && x < this->_width && 0 <= y && y < this->_height) {
+                uint8_t** plane = this->get(channel);
+                if (plane != NULL) {
+                    plane[y][x] = value;
+                    return true;
+                }
+            }
+            return false;
+        }
+
+        /**
+         * 画像データを初期化します.
+         */
+        inline bool init(Mode mode, int32_t width, int32_t height) {
+            if (this->_ready) {
+                // 準備完了済みの場合はなにもしない.
+                // 別のメソッド使ってくれ('A`)
+                return false;
+            }
+
+            // 画像サイズチェック
+            if (width <= 0 || height <= 0) {
+                std::ostringstream msg;
+                msg << "image size must be positive: " << width << " x " << height;
+                throw std::invalid_argument(msg.str());
+            }
+
+            // 解析用にモードを設置
+            ModeInfo info(mode);
+            const ChannelArray& channels = info.getChannels();
+            for (auto it = channels.begin(); it != channels.end(); it++) {
+                Channel channel = *it;
+                this->_data[channel] = NULL;
+
+                // 領域を確保
+                uint8_t** plane = ALLOC_N(uint8_t*, height);
+                std::memset(plane, 0, sizeof(uint8_t*) * height);
+                for (int32_t h = 0; h < height; h++) {
+                    plane[h] = ALLOC_N(uint8_t, width);
+                    if (channel == Channel::ALPHA) {
+                        std::memset(&(plane[h][0]), 255, sizeof(uint8_t) * width);
+                    } else {
+                        std::memset(&(plane[h][0]), 0, sizeof(uint8_t) * width);
+                    }
+                }
+
+                this->_data[channel] = plane;
+            }
+
+            this->_mode   = info;
+            this->_width  = width;
+            this->_height = height;
+            this->_ready  = true;
+            return true;
+        }
+
+        /**
+         * 内部オブジェクトにGC用マークを付けます.
+         */
+        inline void gcmark() {
+            rb_gc_mark_maybe(this->_palette);
+        }
+
+    public:         // 演算子オーバーロード
+        bool operator==(const ImageData& data) const {
+            if (this->_ready && data._ready && (this->_mode == data._mode) && (this->_width == data._width) && (this->_height == data._height)) {
+                if (!(NIL_P(this->_palette) && NIL_P(data._palette))) {
+                    if (!RTEST(rb_equal(this->_palette, data._palette))) {
+                        return false;
+                    }
+                }
+
+                // ピクセルを比較
+                const ChannelArray& channels = this->_mode.getChannels();
+                for (auto chiter = channels.begin(); chiter != channels.end(); chiter++) {
+                    Channel channel = *chiter;
+                    uint8_t** lhsPlane = this->_data.at(channel);
+                    uint8_t** rhsPlane = data._data.at(channel);
+                    for (int32_t h = 0; h < this->_height; h++) {
+                        for (int32_t w = 0; w < this->_width; w++) {
+                            uint8_t lhs = lhsPlane[h][w];
+                            uint8_t rhs = rhsPlane[h][w];
+                            if (lhs != rhs) {
+                                return false;
+                            }
+                        }
+                    }
+                }
+                return true;
+            } else {
+                return false;
+            }
+        }
+        bool operator!=(const ImageData& data) const { return !(*this == data); }
+    };
+
+    /**
+     * Ruby用スキャンラインデータクラス
+     */
+    class ImageLineData {
+    private:        // 非公開メンバ
+        VALUE   _image;     // 親画像
+        int32_t _lineno;    // 行番号
+
+    public:         // コンストラクタ
+        /**
+         * デフォルトコンストラクタ.
+         */
+        ImageLineData(VALUE image = Qnil, int32_t lineno = 0) : _image(image), _lineno(lineno) {}
+
+        /**
+         * デストラクタ
+         */
+        ~ImageLineData() {}
+
+    public:         // プロパティメンバ関数
+        /**
+         * 参照している親画像を返します.
+         */
+        inline VALUE getImage() const { return this->_image; }
+
+        /**
+         * 参照先の画像を設定します.
+         */
+        inline void setImage(VALUE image) { this->_image = image; }
+
+        /**
+         * 参照している親画像の内部データへのポインタを返します.
+         * 参照画像がnilの場合はNULLを返します.
+         */
+        inline ImageData* getImageData() const {
+            if (NIL_P(this->_image)) {
+                return NULL;
+            } else {
+                return rixmap_unwrap<ImageData>(this->_image);
+            }
+        }
+
+        /**
+         * 参照先の行番号を返します.
+         */
+        inline int32_t getLineNumber() const { return this->_lineno; }
+
+        /**
+         * 参照先の行番号を設定します.
+         */
+        inline void setLineNumber(int32_t lineno) { this->_lineno = lineno; }
+
+    public:         // メンバ関数
+        /**
+         * スキャンラインの参照が正しいかを返します.
+         */
+        inline bool isValid() const {
+            ImageData* parent = this->getImageData();
+            if (parent != NULL && 0 <= this->_lineno && this->_lineno < parent->getHeight()) {
+                return true;
+            } else {
+                return false;
+            }
+        }
+
+        /**
+         * このスキャンラインの長さを返します.
+         * 親画像がnilである等、無効なスキャンラインの場合は0になります.
+         */
+        inline int32_t getLength() const {
+            ImageData* image = this->getImageData();
+            if (this->isValid()) {
+                return image->getWidth();
+            } else {
+                return 0;
+            }
+        }
+
+        /**
+         * GCでのマーク処理を行います.
+         */
+        inline void gcmark() {
+            rb_gc_mark_maybe(this->_image);
+        }
+
+    public:         // 演算子オーバーロード
+        /**
+         * 等値比較演算子.
+         */
+        inline bool operator==(const ImageLineData& data) {
+            return ((this->_image == data._image) && (this->_lineno == data._lineno));
+        }
+        inline bool operator!=(const ImageLineData& data) { return !((*this) == data); }
+    };
+}
+
+
+//============================================================================//
+// $Id: image.hxx,v c10e98a1e0fa 2014/04/20 08:17:37 chikuchikugonzalez $
+// vim: set sts=4 ts=4 sw=4 expandtab foldmethod=marker: