天天看點

bmp格式圖像的讀寫函數(對一個開源代碼的封裝)

在網上看到一段讀寫bmp格式圖像的代碼,本文對這段代碼分成兩個函數封裝起來友善使用,一個函數是讀取bmp格式的圖像,一個是向指定檔案寫入bmp格式的圖像。

前提

我們不需要知道這段代碼是如何讀取bmp格式圖像的,不需要知道bmp格式的圖像時如何存儲的,我們隻需要知道有三個參數可以确定圖像的尺寸大小,他們是圖像的寬度、高度、通道數(例如灰階圖像有一個通道,rgb圖像有三個通道(rgb))。圖像包含高度X寬度個像素,每個像素有相同的通道,他們在記憶體中按照一定的順序存儲,例如三通道bmp圖像,在記憶體中圖像行存儲,第一個像素存儲圖像左下角的像素,第二個像素存儲圖像左下角向右移動一個機關後的像素,依次類推。

讀圖像操作

函數定義如下:

bool abReadImage(int &rows, int &cols, int &nChannels, io_byte *&imData, const char *imFileName)
{
  imData = NULL;
  int err_code=0;
  try {
    int n;
    bmp_in in;
    if ((err_code = bmp_in__open(&in,imFileName)) != 0)
      throw err_code;
    cols = in.cols;  rows = in.rows;  nChannels = in.num_components;
    io_byte *dp;
    imData = new io_byte[cols*rows*nChannels];
    for (dp=imData, n=rows; n > 0; n--, dp+=cols*nChannels)
      if ((err_code = bmp_in__get_line(&in,dp)) != 0)
        throw err_code;
    bmp_in__close(&in);
  }
  catch (int exc) {
    if (exc == IO_ERR_NO_FILE)
      fprintf(stderr,"Cannot open input file <%s>.\n", imFileName);
    else if (exc == IO_ERR_FILE_HEADER)
      fprintf(stderr,"Error encountered while parsing BMP file header.\n");
    else if (exc == IO_ERR_UNSUPPORTED)
      fprintf(stderr,"Input uses an unsupported BMP file format.\n  Current "
      "simple example supports only 8-bit and 24-bit data.\n");
    else if (exc == IO_ERR_FILE_TRUNC)
      fprintf(stderr,"Input file <%s> truncated unexpectedly.\n", imFileName);
    else if (exc == IO_ERR_FILE_NOT_OPEN)
      fprintf(stderr,"Trying to access a file which is not open!(?)\n");
    return false;
  }
  return true;
}
           

使用此函數必須要包含頭檔案:io_bmp.h,這個頭檔案以及他聲明的函數或者類型的實作可以在 這裡下載下傳到。

讀圖像函數輸入:

imFileName:輸入圖像的檔案名。

讀圖像函數輸出:

rows:圖像的行數,或者說圖像的高度。

cols:圖像的列數,或者說圖像的寬度。

nChannels:圖像的通道數(1或者3,暫時不支援其他的通道)。

imData:存儲圖像像素的數組,注意,這個數組的記憶體是在函數内部申請的,在使用完圖像後,記得釋放掉這塊記憶體。

寫圖像操作

函數定義如下:

bool abWriteImage(const int rows, const int cols, const int nChannels, io_byte *imData, const char *imFileName)
{
  int err_code=0;
  try {
    bmp_out out;
    if ((err_code = bmp_out__open(&out,imFileName,cols,rows,nChannels)) != 0)
      throw err_code;

    io_byte *dp;
    int n;
    for (dp=imData, n=rows; n > 0; n--, dp+=cols*nChannels)
      bmp_out__put_line(&out,dp);
    bmp_out__close(&out);
  }
  catch (int exc) {
    if (exc == IO_ERR_NO_FILE)
      fprintf(stderr,"Cannot open the output file <%s>.\n", imFileName);
    else if (exc == IO_ERR_FILE_HEADER)
      fprintf(stderr,"Error encountered while parsing BMP file header.\n");
    else if (exc == IO_ERR_UNSUPPORTED)
      fprintf(stderr,"Input uses an unsupported BMP file format.\n  Current "
      "simple example supports only 8-bit and 24-bit data.\n");
    else if (exc == IO_ERR_FILE_TRUNC)
      fprintf(stderr,"output file <%s> truncated unexpectedly.\n", imFileName);
    else if (exc == IO_ERR_FILE_NOT_OPEN)
      fprintf(stderr,"Trying to access a file which is not open!(?)\n");
    return false;
  }
  return true;
}
           

使用此函數必須要包含頭檔案:io_bmp.h,這個頭檔案以及他聲明的函數或者類型的實作可以在這裡下載下傳到。

寫圖像函數輸入:

imFileName:要寫入磁盤的圖像檔案名。

rows:圖像的行數,或者說圖像的高度。

cols:圖像的列數,或者說圖像的寬度。

nChannels:圖像的通道數(1或者3,暫時不支援其他的通道)。

imData:存儲圖像像素的數組。

實驗說明

根據你使用的編譯相關工具不同,給出幾點說明:

1、MSVS。 在你使用這兩個函數的項目中要添加你下載下傳的io_bmp.h和io_bmp.cpp。這是比較簡單的一種使用方法。

2、如果你在linux上編譯。記得将你下載下傳的兩個檔案加入到你的工程中,還要記得對檔案的格式進行下轉換(可以使用dos2unix這樣的小工具)。