yading@10: /* yading@10: * BMP image format encoder yading@10: * Copyright (c) 2006, 2007 Michel Bardiaux yading@10: * Copyright (c) 2009 Daniel Verkamp yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: #include "libavutil/imgutils.h" yading@10: #include "libavutil/avassert.h" yading@10: #include "avcodec.h" yading@10: #include "bytestream.h" yading@10: #include "bmp.h" yading@10: #include "internal.h" yading@10: yading@10: static const uint32_t monoblack_pal[] = { 0x000000, 0xFFFFFF }; yading@10: static const uint32_t rgb565_masks[] = { 0xF800, 0x07E0, 0x001F }; yading@10: static const uint32_t rgb444_masks[] = { 0x0F00, 0x00F0, 0x000F }; yading@10: yading@10: static av_cold int bmp_encode_init(AVCodecContext *avctx){ yading@10: switch (avctx->pix_fmt) { yading@10: case AV_PIX_FMT_BGRA: yading@10: avctx->bits_per_coded_sample = 32; yading@10: break; yading@10: case AV_PIX_FMT_BGR24: yading@10: avctx->bits_per_coded_sample = 24; yading@10: break; yading@10: case AV_PIX_FMT_RGB555: yading@10: case AV_PIX_FMT_RGB565: yading@10: case AV_PIX_FMT_RGB444: yading@10: avctx->bits_per_coded_sample = 16; yading@10: break; yading@10: case AV_PIX_FMT_RGB8: yading@10: case AV_PIX_FMT_BGR8: yading@10: case AV_PIX_FMT_RGB4_BYTE: yading@10: case AV_PIX_FMT_BGR4_BYTE: yading@10: case AV_PIX_FMT_GRAY8: yading@10: case AV_PIX_FMT_PAL8: yading@10: avctx->bits_per_coded_sample = 8; yading@10: break; yading@10: case AV_PIX_FMT_MONOBLACK: yading@10: avctx->bits_per_coded_sample = 1; yading@10: break; yading@10: default: yading@10: av_log(avctx, AV_LOG_INFO, "unsupported pixel format\n"); yading@10: return -1; yading@10: } yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: static int bmp_encode_frame(AVCodecContext *avctx, AVPacket *pkt, yading@10: const AVFrame *pict, int *got_packet) yading@10: { yading@10: int n_bytes_image, n_bytes_per_row, n_bytes, i, n, hsize, ret; yading@10: const uint32_t *pal = NULL; yading@10: uint32_t palette256[256]; yading@10: int pad_bytes_per_row, pal_entries = 0, compression = BMP_RGB; yading@10: int bit_count = avctx->bits_per_coded_sample; yading@10: uint8_t *ptr, *buf; yading@10: AVFrame * const p = (AVFrame *)pict; yading@10: yading@10: p->pict_type= AV_PICTURE_TYPE_I; yading@10: p->key_frame= 1; yading@10: switch (avctx->pix_fmt) { yading@10: case AV_PIX_FMT_RGB444: yading@10: compression = BMP_BITFIELDS; yading@10: pal = rgb444_masks; // abuse pal to hold color masks yading@10: pal_entries = 3; yading@10: break; yading@10: case AV_PIX_FMT_RGB565: yading@10: compression = BMP_BITFIELDS; yading@10: pal = rgb565_masks; // abuse pal to hold color masks yading@10: pal_entries = 3; yading@10: break; yading@10: case AV_PIX_FMT_RGB8: yading@10: case AV_PIX_FMT_BGR8: yading@10: case AV_PIX_FMT_RGB4_BYTE: yading@10: case AV_PIX_FMT_BGR4_BYTE: yading@10: case AV_PIX_FMT_GRAY8: yading@10: av_assert1(bit_count == 8); yading@10: avpriv_set_systematic_pal2(palette256, avctx->pix_fmt); yading@10: pal = palette256; yading@10: break; yading@10: case AV_PIX_FMT_PAL8: yading@10: pal = (uint32_t *)p->data[1]; yading@10: break; yading@10: case AV_PIX_FMT_MONOBLACK: yading@10: pal = monoblack_pal; yading@10: break; yading@10: } yading@10: if (pal && !pal_entries) pal_entries = 1 << bit_count; yading@10: n_bytes_per_row = ((int64_t)avctx->width * (int64_t)bit_count + 7LL) >> 3LL; yading@10: pad_bytes_per_row = (4 - n_bytes_per_row) & 3; yading@10: n_bytes_image = avctx->height * (n_bytes_per_row + pad_bytes_per_row); yading@10: yading@10: // STRUCTURE.field refer to the MSVC documentation for BITMAPFILEHEADER yading@10: // and related pages. yading@10: #define SIZE_BITMAPFILEHEADER 14 yading@10: #define SIZE_BITMAPINFOHEADER 40 yading@10: hsize = SIZE_BITMAPFILEHEADER + SIZE_BITMAPINFOHEADER + (pal_entries << 2); yading@10: n_bytes = n_bytes_image + hsize; yading@10: if ((ret = ff_alloc_packet2(avctx, pkt, n_bytes)) < 0) yading@10: return ret; yading@10: buf = pkt->data; yading@10: bytestream_put_byte(&buf, 'B'); // BITMAPFILEHEADER.bfType yading@10: bytestream_put_byte(&buf, 'M'); // do. yading@10: bytestream_put_le32(&buf, n_bytes); // BITMAPFILEHEADER.bfSize yading@10: bytestream_put_le16(&buf, 0); // BITMAPFILEHEADER.bfReserved1 yading@10: bytestream_put_le16(&buf, 0); // BITMAPFILEHEADER.bfReserved2 yading@10: bytestream_put_le32(&buf, hsize); // BITMAPFILEHEADER.bfOffBits yading@10: bytestream_put_le32(&buf, SIZE_BITMAPINFOHEADER); // BITMAPINFOHEADER.biSize yading@10: bytestream_put_le32(&buf, avctx->width); // BITMAPINFOHEADER.biWidth yading@10: bytestream_put_le32(&buf, avctx->height); // BITMAPINFOHEADER.biHeight yading@10: bytestream_put_le16(&buf, 1); // BITMAPINFOHEADER.biPlanes yading@10: bytestream_put_le16(&buf, bit_count); // BITMAPINFOHEADER.biBitCount yading@10: bytestream_put_le32(&buf, compression); // BITMAPINFOHEADER.biCompression yading@10: bytestream_put_le32(&buf, n_bytes_image); // BITMAPINFOHEADER.biSizeImage yading@10: bytestream_put_le32(&buf, 0); // BITMAPINFOHEADER.biXPelsPerMeter yading@10: bytestream_put_le32(&buf, 0); // BITMAPINFOHEADER.biYPelsPerMeter yading@10: bytestream_put_le32(&buf, 0); // BITMAPINFOHEADER.biClrUsed yading@10: bytestream_put_le32(&buf, 0); // BITMAPINFOHEADER.biClrImportant yading@10: for (i = 0; i < pal_entries; i++) yading@10: bytestream_put_le32(&buf, pal[i] & 0xFFFFFF); yading@10: // BMP files are bottom-to-top so we start from the end... yading@10: ptr = p->data[0] + (avctx->height - 1) * p->linesize[0]; yading@10: buf = pkt->data + hsize; yading@10: for(i = 0; i < avctx->height; i++) { yading@10: if (bit_count == 16) { yading@10: const uint16_t *src = (const uint16_t *) ptr; yading@10: uint16_t *dst = (uint16_t *) buf; yading@10: for(n = 0; n < avctx->width; n++) yading@10: AV_WL16(dst + n, src[n]); yading@10: } else { yading@10: memcpy(buf, ptr, n_bytes_per_row); yading@10: } yading@10: buf += n_bytes_per_row; yading@10: memset(buf, 0, pad_bytes_per_row); yading@10: buf += pad_bytes_per_row; yading@10: ptr -= p->linesize[0]; // ... and go back yading@10: } yading@10: yading@10: pkt->flags |= AV_PKT_FLAG_KEY; yading@10: *got_packet = 1; yading@10: return 0; yading@10: } yading@10: yading@10: AVCodec ff_bmp_encoder = { yading@10: .name = "bmp", yading@10: .type = AVMEDIA_TYPE_VIDEO, yading@10: .id = AV_CODEC_ID_BMP, yading@10: .init = bmp_encode_init, yading@10: .encode2 = bmp_encode_frame, yading@10: .pix_fmts = (const enum AVPixelFormat[]){ yading@10: AV_PIX_FMT_BGRA, AV_PIX_FMT_BGR24, yading@10: AV_PIX_FMT_RGB565, AV_PIX_FMT_RGB555, AV_PIX_FMT_RGB444, yading@10: AV_PIX_FMT_RGB8, AV_PIX_FMT_BGR8, AV_PIX_FMT_RGB4_BYTE, AV_PIX_FMT_BGR4_BYTE, AV_PIX_FMT_GRAY8, AV_PIX_FMT_PAL8, yading@10: AV_PIX_FMT_MONOBLACK, yading@10: AV_PIX_FMT_NONE yading@10: }, yading@10: .long_name = NULL_IF_CONFIG_SMALL("BMP (Windows and OS/2 bitmap)"), yading@10: };