Загрузка и сохранение BMP файла - C++ - Ответ 9836737

#include "windows.h"
 
typedef unsigned __int16    FILE_ID;
 
typedef struct
{
    unsigned __int32 size = 54;     //сюда надо будет прибавить размер массива c паддингами
    unsigned __int16 rzrv1 = 0;
    unsigned __int16 rzrv2 = 0;
    unsigned __int32 offset = 54;   //смещение массива данных относительно начала файла = sizeof(file_id) + sizeof(BMPheader)
    BITMAPINFOHEADER dib;
}       BMP_HEADER;

unsigned char* loadBMP(const char *fname, int &x_pix, int &y_pix)
{
    FILE_ID id;
    BMP_HEADER bh;
    size_t res;
    unsigned int x_bytes, bytes, i, j, padding;
    unsigned char *v, tmp;
 
    x_pix = y_pix = -1;
    FILE *f = fopen(fname, "rb");
    if (!f)
    {
        return NULL;
    }
    res = fread(&id, 1, sizeof(FILE_ID), f);
    if (res != sizeof(FILE_ID))
    {
        fclose(f);
        return NULL;
    }
    res = fread(&bh, 1, sizeof(BMP_HEADER), f);
    if (res != sizeof(BMP_HEADER))
    {
        fclose(f);
        return NULL;
    }
 
    /* */
    x_pix = bh.dib.biWidth;
    y_pix = bh.dib.biHeight;
 
    //находим длину строки в байтах и округляем с учетом паддингов
    x_bytes = x_pix * 3;                //каждый пиксель занимает 3 байта в строке
    if ((x_bytes % 4) != 0)
    {
        padding = (4 - (x_bytes % 4));
        bytes = x_bytes + padding;
    }
    else
    {
        bytes = x_bytes;
    }
    v = new unsigned char[bytes * y_pix];
 
    for (i = 0; i < y_pix; i++)
    {
        for (j = 0; j < bytes; j++)
        {
            if (j < x_bytes)
            {
                fread((v + j + 0 + (i * x_bytes)), 1, 1, f);
                fread((v + j + 1 + (i * x_bytes)), 1, 1, f);
                fread((v + j + 2 + (i * x_bytes)), 1, 1, f);
                j += 2;
            }
            else
            {
                fread(&tmp, 1, 1, f);
            }
        }
    }
 
    return v;
}

int saveBMP(const char *fname, unsigned char *v, int x_pix, int y_pix)
{
    unsigned int x_bytes, bytes, i, j, padding;
    unsigned char tmp = 255;
    BMP_HEADER bh;
    memset(&bh, 0, sizeof(bh));
    FILE_ID id = 0x4d42;
    //находим длину строки в байтах и округляем с учетом паддингов
    x_bytes = x_pix * 3;                //каждый пиксель занимает 3 байта в строке
    if ((x_bytes % 4) != 0)
    {
        padding = (4 - (x_bytes % 4));
        bytes = x_bytes + padding;
    }
    else
    {
        bytes = x_bytes;
    }
    //размер файла 54 байта заголовок + массив данных
    bh.size += bytes * y_pix;
    //заполняем DIB заголовок
    bh.dib.biSize = 40;                 //размер заголовка
    bh.dib.biWidth = x_pix;             //ширина картинки
    bh.dib.biHeight = y_pix;                //высота картинки
    bh.dib.biPlanes = 1;
    bh.dib.biBitCount = 24;
    bh.dib.biCompression = 0;
    bh.dib.biSizeImage = bytes * y_pix; //размер массива данных 
    bh.dib.biXPelsPerMeter = 2835;
    bh.dib.biYPelsPerMeter = 2835;
    bh.dib.biClrUsed = 0;
    bh.dib.biClrImportant = 0;
    //открываем файл
    FILE *f = fopen(fname, "wb");
    if (!f) return -1;
    //пишем заголовок
    size_t res;
    res = fwrite(&id, 1, sizeof(FILE_ID), f);
    if (res != sizeof(FILE_ID))
    {
        fclose(f);
        return -1;
    }
    res = fwrite(&bh, 1, sizeof(BMP_HEADER), f);
    if (res != sizeof(BMP_HEADER))
    {
        fclose(f);
        return -1;
    }
    //пишем массив данных
    for (i = 0; i < y_pix; i++)
    {
        for (j = 0; j < bytes; j++)
        {
            if (j < x_bytes)
            {
                fwrite((v + j + 0 + (i * x_bytes)), 1, 1, f);
                fwrite((v + j + 1 + (i * x_bytes)), 1, 1, f);
                fwrite((v + j + 2 + (i * x_bytes)), 1, 1, f);
                j += 2;
            }
            else
            {
                fwrite(&tmp, 1, 1, f);
            }
        }
    }
    fclose(f);
 
    return 0;
}