【画像処理100本ノック】独自の画像入出力クラスを作る

背景

画像処理100本ノックが面白そうなのでやってみることにしました。
基本的には配列操作しかしないだろうし、不慣れなOpenCVでやるよりも自分が使いやすいものを使いたい、ということで簡単な画像入出力クラスを作ってみました。

PPM形式

作るのが簡単なのでPPM形式の画像を扱うことにします。PPM形式の画像は非常に単純な構造をしています。簡単なので説明は省略します。

コード

100本ノックの画像はJPEGで与えられているので、ImageJを使ってPPMに変換します。PPM形式の仕様を完全に満たすのは面倒なので、ここではImageJでの読み書きに問題がない範囲で実装しました。なので、任意のPPM画像を読み込めるわけではないことに注意が必要です。また、エラーチェックも省略しています。

#include <iostream>
#include <stdio.h>

class PPM
{
private:
    int width;
    int height;
    int* R;
    int* G;
    int* B;
public:
    PPM(int width, int height)
    {
        this->width = width;
        this->height = height;
        R = new int[width * height];
        G = new int[width * height];
        B = new int[width * height];
    }
    PPM(const char* filename)
    {
        FILE* fp = fopen(filename, "rb");
        char tmp[256];
        fscanf(fp, "%s\n", tmp);
        if (fgetc(fp) == '#')//コメントありの場合
        {
            fseek(fp, -1, SEEK_CUR);
            while (fgetc(fp) != '\n');//読み飛ばす
        }
        else fseek(fp, -1, SEEK_CUR);
        fscanf(fp, "%d %d\n", &width,&height);
        fscanf(fp, "%s\n", tmp);
        R = new int[width * height];
        G = new int[width * height];
        B = new int[width * height];
        for (int j = 0; j < height; j++)
            for (int i = 0; i < width; i++)
            {
                R[i + width * j] = fgetc(fp);
                G[i + width * j] = fgetc(fp);
                B[i + width * j] = fgetc(fp);
            }
        fclose(fp);
    }
    int Get_width() const
    {
        return width;
    }
    int Get_height() const
    {
        return height;
    }

    int &operator()(int i, int j, char rgb)
    {
        if(rgb=='r') return R[i+width*j];
        if (rgb == 'g') return G[i + width * j];
        if (rgb == 'b') return B[i + width * j];
    }
    void Flush(const char *filename)
    {
        FILE* fp = fopen(filename, "wb");
        fprintf(fp, "P6\n");
        fprintf(fp, "%d %d\n", width, height);
        fprintf(fp, "255\n");
        for (int j = 0; j < height; j++)
            for (int i = 0; i < width; i++)
            {
                fputc(R[i + width * j], fp);
                fputc(G[i + width * j], fp);
                fputc(B[i + width * j], fp);
            }

        fclose(fp);
    }

    ~PPM()
    {
        delete[] R;
        delete[] G;
        delete[] B;
    }
};

int main()
{
    PPM ppm("imori.pnm");

    int width = ppm.Get_width();
    int height = ppm.Get_height();
    PPM ppm2(width, height);
    for(int j=0; j<height; j++)
        for (int i = 0; i < width; i++)
        {
            ppm2(i, j, 'r') = ppm(i, j, 'r');
            ppm2(i, j, 'g') = ppm(i, j, 'g');
            ppm2(i, j, 'b') = ppm(i, j, 'b');
        }
    ppm2.Flush("iout.ppm");

    return 0;
}

問題があれば随時修正予定です。