libpng 1bit Grayscale画像の読み込み

#pragma once

#include <cstdlib>
#include <iostream>
#include <bitset>

#include <png.h>

#pragma comment(lib,"libpng16d.lib")

//エラーの時強制終了
void abort_(const char* c) {
  printf(c);
  abort();
}

//! @brief pngファイル読み込み関数
//! @param [in] file_name ファイル名
//! @param [out] width 画像幅（ピクセル）
//! @param [out] height 画像高さ（ピクセル）
//! @param [out] color_type RGBかRGBAか...等
//! @param [out] bit_depth チャンネルのビット数
//! @param [out] row_pointers 画像データへのポインタのポインタ
void read_png(
  const char* file_name,
  int* width,
  int* height,
  png_byte* color_type,
  png_byte* bit_depth,
  png_bytep** row_pointers

) {

  png_byte  header[8];    // 8 is the maximum size that can be checked

  FILE* fp = fopen(file_name, "rb");
  if (!fp) {
    abort_("[read_png_file] File could not be opened for reading");
  }
  fread(header, 1, 8, fp);
  if (png_sig_cmp(header, 0, 8)) {
    abort_("[read_png_file] File is not recognized as a PNG file");
  }


  png_structp png_ptr;
  png_infop info_ptr;

  /* initialize stuff */
  png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);

  if (!png_ptr)
    abort_("[read_png_file] png_create_read_struct failed");

  info_ptr = png_create_info_struct(png_ptr);
  if (!info_ptr)
    abort_("[read_png_file] png_create_info_struct failed");

  if (setjmp(png_jmpbuf(png_ptr)))
    abort_("[read_png_file] Error during init_io");

  png_init_io(png_ptr, fp);
  png_set_sig_bytes(png_ptr, 8);

  png_read_info(png_ptr, info_ptr);

  /////////////////////////////////////////
  // 画像情報の取得
  *width = png_get_image_width(png_ptr, info_ptr);
  *height = png_get_image_height(png_ptr, info_ptr);
  *color_type = png_get_color_type(png_ptr, info_ptr);
  *bit_depth = png_get_bit_depth(png_ptr, info_ptr);
  //
  /////////////////////////////////////////

  int number_of_passes;
  number_of_passes = png_set_interlace_handling(png_ptr);
  png_read_update_info(png_ptr, info_ptr);

  /* read file */
  if (setjmp(png_jmpbuf(png_ptr)))
    abort_("[read_png_file] Error during read_image");

  /////////////////////////////////////////
  // 画像の読み込み
  *row_pointers = (png_bytep*)malloc(sizeof(png_bytep) * *height);

  for (int y = 0; y < *height; y++)
    (*row_pointers)[y] = (png_byte*)malloc(png_get_rowbytes(png_ptr, info_ptr));

  png_read_image(png_ptr, *row_pointers);
  //
  /////////////////////////////////////////


  fclose(fp);
}

int main()
{
  int width;
  int height;
  png_byte color_type;
  png_byte bit_depth;

  png_bytep* row_pointers;

  ////////////////////////////////
  // 画像読み込み
  read_png(
    R"(c:\test\test_bitdepth_1_notpalette_1x16px.png)",
    &width,
    &height,
    &color_type,
    &bit_depth,
    &row_pointers
  );
  //
  ////////////////////////////////

  std::cout << "width : " << width << std::endl;
  std::cout << "height: " << height << std::endl;
  std::cout << "colortype: " << (int)color_type << std::endl;
  std::cout << "bitdepth: " << (int)bit_depth << std::endl;

  ////////////////////////////////
  //今回は１ビットグレイスケール画像だけを対象にする
  if ((color_type != PNG_COLOR_TYPE_GRAY) ||
    (bit_depth != 1)
    ) {
    return -1;
  }
  //
  ////////////////////////////////

  ////////////////////////////////
  // 内容を表示
  png_bytep yhead = row_pointers[0];
  png_bytep xpix = yhead;
  std::cout << "[0]" << std::bitset<8>(xpix[0]) << std::endl;
  std::cout << "[1]" << std::bitset<8>(xpix[1]) << std::endl;
  //
  ///////////////////////////////////////////////

  ///////////////////////////////////////////////
  // メモリの解放
  for (size_t i = 0; i < (size_t)height; i++) {
    png_bytep yhead = row_pointers[i];
    free(yhead);
  }
  free(row_pointers);
  //
  ///////////////////////////////////////////////

  int i;
  std::cin >> i;

}