#define DIB_HEADER_MARKER ((WORD) ('M' << 8) | 'B')

BOOL IppDib::Load(const char* filename)
{
    const char* ext = strrchr(filename, '.');
    if (!_strcmpi(ext, ".bmp"))
        return LoadBMP(filename);
    else
        return FALSE;
}

BOOL IppDib::Save(const char* filename)
{
    const char* ext = strrchr(filename, '.');
    if (!_strcmpi(ext, ".bmp"))
        return SaveBMP(filename);
    else
        return FALSE;
}

BOOL IppDib::LoadBMP(const char* filename)
{
    FILE* fp = NULL;
    fopen_s(&fp, filename, "rb");
    if (!fp)
        return false;

    BITMAPFILEHEADER bmfh;
    BITMAPINFOHEADER bmih;

    // BITMAPFILEHEADER 읽기
    if (fread(&bmfh, sizeof(BITMAPFILEHEADER), 1, fp) != 1) {
        fclose(fp);
        return false;
    }

    // BMP 파일임을 나타내는 "BM" 마커가 있는지 확인
    if (bmfh.bfType != DIB_HEADER_MARKER) {
        fclose(fp);
        return false;
    }

    // BITMAPINFOHEADER 읽기
    if (fread(&bmih, sizeof(BITMAPINFOHEADER), 1, fp) != 1) {
        fclose(fp);
        return false;
    }

    m_nWidth    = bmih.biWidth;
    m_nHeight   = bmih.biHeight;
    m_nBitCount = bmih.biBitCount;

    // 픽셀 데이터 공간 계산
    DWORD dwWidthStep = (DWORD)((m_nWidth * m_nBitCount / 8 + 3) & ~3);
    DWORD dwSizeImage = m_nHeight * dwWidthStep;

    if (m_nBitCount == 24)
        m_nDibSize = sizeof(BITMAPINFOHEADER) + dwSizeImage;
    else
        m_nDibSize = sizeof(BITMAPINFOHEADER) + sizeof(RGBQUAD) * (1 << m_nBitCount) + dwSizeImage;

    // Packed-DIB 저장할 메모리 공간 동적 할당
    if (m_pDib)
        DestroyBitmap();

    m_pDib = new BYTE[m_nDibSize];
    if (m_pDib == NULL)
    {
        fclose(fp);
        return FALSE;
    }

    // 파일로부터 Packed-DIB 크기만큼을 읽기
    fseek(fp, sizeof(BITMAPFILEHEADER), SEEK_SET);
    if (fread(m_pDib, sizeof(BYTE), m_nDibSize, fp) != m_nDibSize)
    {
        delete[] m_pDib;
        m_pDib = NULL;
        fclose(fp);
        return FALSE;
    }

    // 파일 닫기
    fclose(fp);

    return TRUE;
}

BOOL IppDib::SaveBMP(const char* filename)
{
    if (!IsValid())
        return FALSE;

    FILE* fp;
    fopen_s(&fp, filename, "wb");
    if (!fp)
        return FALSE;

    BITMAPFILEHEADER bmfh;
    bmfh.bfType      = DIB_HEADER_MARKER;
    bmfh.bfSize      = (DWORD)sizeof(BITMAPFILEHEADER) + m_nDibSize;
    bmfh.bfReserved1 = 0;
    bmfh.bfReserved2 = 0;
    bmfh.bfOffBits   = (DWORD)(sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER)
        + (sizeof(RGBQUAD) * GetPaletteNums()));

    fwrite(&bmfh, sizeof(BITMAPFILEHEADER), 1, fp);
    fwrite(GetBitmapInfoAddr(), m_nDibSize, 1, fp);

    fclose(fp);

    return TRUE;
}