histgram.cpp

//
//  histgram.cpp
//  histgram
//
//  Copyright (c) 2016年 Ryunosuke Kirikihira. All rights reserved.
//

#include <iostream>
#include <fstream>
#include <cstdio>
#include <iomanip>
#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>

using namespace cv;
using namespace std;

// 64色に減色
// R:4等分、G:4等分、B:4等分で4x4x4=64色
uchar decleaseColor(int value) {
    if (value < 64) {
        return 32;
    } else if (value < 128) {
        return 96;
    } else if (value < 196) {
        return 160;
    } else {
        return 224;
    }
    
    return 0;  // 未到達
}


/**
 * オリジナル画像のRGBからヒストグラムのビン番号を計算
 * @param[in]  red    赤の輝度（0-255）
 * @param[in]  green  緑の輝度（0-255）
 * @param[in]  blue   青の輝度（0-255）
 * @return ヒストグラムのビン番号（64色カラーインデックス）
 */
int rgb2bin(int red, int green, int blue) {
    int redNo = red / 64;
    int greenNo = green / 64;
    int blueNo = blue / 64;
    return 16 * redNo + 4 * greenNo + blueNo;
}


/**
 * ヒストグラムを計算
 * @param[in]  filename   画像ファイル名
 * @param[out] histogram  ヒストグラム
 * @return 正常終了で0、異常終了で-1
 */
int calcHistogram(char *filename, int histogram[64]) {
    // ヒストグラムを初期化
    for (int i = 0; i < 64; i++) {
        histogram[i] = 0;
    }
    
    // 画像のロード
    IplImage *img = cvLoadImage(filename, CV_LOAD_IMAGE_COLOR);
    if (img == NULL) {
        cerr << "cannot open image: " << filename << endl;
        return -1;
    }
    
    // 64色に減色してヒストグラムを計算
    for (int y = 0; y < img->height; y++) {
        // y行目のデータの先頭ポインタを取得
        uchar *pin = (uchar *)(img->imageData + y * img->widthStep);
        for (int x = 0; x < img->width; x++) {
            int blue = pin[3*x+0];
            int green = pin[3*x+1];
            int red = pin[3*x+2];
            
            // カラー値（0-63）をカウント
            int bin = rgb2bin(red, green, blue);
            histogram[bin] += 1;
        }
    }
    
    cvReleaseImage(&img);
    
    return 0;
}


/**
 * ヒストグラムをファイルに出力
 * @param[in]  filename  出力ファイル名
 * @param[in]  histogram ヒストグラム
 * @return 正常終了で0、異常終了で-1
 */
int writeHistogram(char *filename, int histogram[64]) {
    ofstream outFile(filename);
    if (outFile.fail()) {
        cerr << "cannot open file: " << filename << endl;
        return -1;
    }
    
    double sum = 0;
    for (int i=0; i<64; i++) {
        sum += histogram[i];
    }
    
    for (int i = 0; i < 64; i++) {
        outFile << fixed << setprecision(5) << histogram[i]/sum << "," << endl;
    }
    
    outFile.close();
    
    return 0;
}


/**
 * メイン関数
 * @return 正常終了で0、異常終了で-1
 */
int main(int argc, char **argv) {
    for (int i=0; i<70; i++) {
        int ret;
        char imageFile[100];
        sprintf(imageFile, "/Users/Kiri/Desktop/UserModeling/Images/%03d.jpg", i);
        char histFile[100];
        sprintf(histFile, "/Users/Kiri/Desktop/UserModeling/Images/%03d.txt", i);
        
        cout << imageFile << " -> " << histFile;
        
        // ヒストグラムを計算
        int histogram[64];
        ret = calcHistogram(imageFile, histogram);
        if (ret < 0) {
            cerr << "cannot calc histogram" << endl;
            return -1;
        }
        
        // ヒストグラムをファイルに出力
        ret = writeHistogram(histFile, histogram);
        if (ret < 0) {
            cerr << "cannot write histogram" << endl;
            return -1;
        }
        
        cout << " ... OK" << endl;
    }
    return 0;
}