| /* |
| * Microsoft Windows ICO muxer |
| * Copyright (c) 2012 Michael Bradshaw <mjbshaw gmail com> |
| * |
| * This file is part of FFmpeg. |
| * |
| * FFmpeg is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * FFmpeg is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with FFmpeg; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| */ |
| |
| /** |
| * @file |
| * Microsoft Windows ICO muxer |
| */ |
| |
| #include "libavutil/intreadwrite.h" |
| #include "libavutil/pixdesc.h" |
| #include "avformat.h" |
| |
| typedef struct { |
| int offset; |
| int size; |
| unsigned char width; |
| unsigned char height; |
| short bits; |
| } IcoImage; |
| |
| typedef struct { |
| int current_image; |
| int nb_images; |
| IcoImage *images; |
| } IcoMuxContext; |
| |
| static int ico_check_attributes(AVFormatContext *s, const AVCodecParameters *p) |
| { |
| if (p->codec_id == AV_CODEC_ID_BMP) { |
| if (p->format == AV_PIX_FMT_PAL8 && AV_PIX_FMT_RGB32 != AV_PIX_FMT_BGRA) { |
| av_log(s, AV_LOG_ERROR, "Wrong endianness for bmp pixel format\n"); |
| return AVERROR(EINVAL); |
| } else if (p->format != AV_PIX_FMT_PAL8 && |
| p->format != AV_PIX_FMT_RGB555LE && |
| p->format != AV_PIX_FMT_BGR24 && |
| p->format != AV_PIX_FMT_BGRA) { |
| av_log(s, AV_LOG_ERROR, "BMP must be 1bit, 4bit, 8bit, 16bit, 24bit, or 32bit\n"); |
| return AVERROR(EINVAL); |
| } |
| } else if (p->codec_id == AV_CODEC_ID_PNG) { |
| if (p->format != AV_PIX_FMT_RGBA) { |
| av_log(s, AV_LOG_ERROR, "PNG in ico requires pixel format to be rgba\n"); |
| return AVERROR(EINVAL); |
| } |
| } else { |
| const AVCodecDescriptor *codesc = avcodec_descriptor_get(p->codec_id); |
| av_log(s, AV_LOG_ERROR, "Unsupported codec %s\n", codesc ? codesc->name : ""); |
| return AVERROR(EINVAL); |
| } |
| |
| if (p->width > 256 || |
| p->height > 256) { |
| av_log(s, AV_LOG_ERROR, "Unsupported dimensions %dx%d (dimensions cannot exceed 256x256)\n", p->width, p->height); |
| return AVERROR(EINVAL); |
| } |
| |
| return 0; |
| } |
| |
| static int ico_write_header(AVFormatContext *s) |
| { |
| IcoMuxContext *ico = s->priv_data; |
| AVIOContext *pb = s->pb; |
| int ret; |
| int i; |
| |
| if (!pb->seekable) { |
| av_log(s, AV_LOG_ERROR, "Output is not seekable\n"); |
| return AVERROR(EINVAL); |
| } |
| |
| ico->current_image = 0; |
| ico->nb_images = s->nb_streams; |
| |
| avio_wl16(pb, 0); // reserved |
| avio_wl16(pb, 1); // 1 == icon |
| avio_skip(pb, 2); // skip the number of images |
| |
| for (i = 0; i < s->nb_streams; i++) { |
| if (ret = ico_check_attributes(s, s->streams[i]->codecpar)) |
| return ret; |
| |
| // Fill in later when writing trailer... |
| avio_skip(pb, 16); |
| } |
| |
| ico->images = av_mallocz_array(ico->nb_images, sizeof(IcoMuxContext)); |
| if (!ico->images) |
| return AVERROR(ENOMEM); |
| |
| avio_flush(pb); |
| |
| return 0; |
| } |
| |
| static int ico_write_packet(AVFormatContext *s, AVPacket *pkt) |
| { |
| IcoMuxContext *ico = s->priv_data; |
| IcoImage *image; |
| AVIOContext *pb = s->pb; |
| AVCodecParameters *par = s->streams[pkt->stream_index]->codecpar; |
| int i; |
| |
| if (ico->current_image >= ico->nb_images) { |
| av_log(s, AV_LOG_ERROR, "ICO already contains %d images\n", ico->current_image); |
| return AVERROR(EIO); |
| } |
| |
| image = &ico->images[ico->current_image++]; |
| |
| image->offset = avio_tell(pb); |
| image->width = (par->width == 256) ? 0 : par->width; |
| image->height = (par->height == 256) ? 0 : par->height; |
| |
| if (par->codec_id == AV_CODEC_ID_PNG) { |
| image->bits = par->bits_per_coded_sample; |
| image->size = pkt->size; |
| |
| avio_write(pb, pkt->data, pkt->size); |
| } else { // BMP |
| if (AV_RL32(pkt->data + 14) != 40) { // must be BITMAPINFOHEADER |
| av_log(s, AV_LOG_ERROR, "Invalid BMP\n"); |
| return AVERROR(EINVAL); |
| } |
| |
| image->bits = AV_RL16(pkt->data + 28); // allows things like 1bit and 4bit images to be preserved |
| image->size = pkt->size - 14 + par->height * (par->width + 7) / 8; |
| |
| avio_write(pb, pkt->data + 14, 8); // Skip the BITMAPFILEHEADER header |
| avio_wl32(pb, AV_RL32(pkt->data + 22) * 2); // rewrite height as 2 * height |
| avio_write(pb, pkt->data + 26, pkt->size - 26); |
| |
| for (i = 0; i < par->height * (par->width + 7) / 8; ++i) |
| avio_w8(pb, 0x00); // Write bitmask (opaque) |
| } |
| |
| return 0; |
| } |
| |
| static int ico_write_trailer(AVFormatContext *s) |
| { |
| IcoMuxContext *ico = s->priv_data; |
| AVIOContext *pb = s->pb; |
| int i; |
| |
| avio_seek(pb, 4, SEEK_SET); |
| |
| avio_wl16(pb, ico->current_image); |
| |
| for (i = 0; i < ico->nb_images; i++) { |
| avio_w8(pb, ico->images[i].width); |
| avio_w8(pb, ico->images[i].height); |
| |
| if (s->streams[i]->codecpar->codec_id == AV_CODEC_ID_BMP && |
| s->streams[i]->codecpar->format == AV_PIX_FMT_PAL8) { |
| avio_w8(pb, (ico->images[i].bits >= 8) ? 0 : 1 << ico->images[i].bits); |
| } else { |
| avio_w8(pb, 0); |
| } |
| |
| avio_w8(pb, 0); // reserved |
| avio_wl16(pb, 1); // color planes |
| avio_wl16(pb, ico->images[i].bits); |
| avio_wl32(pb, ico->images[i].size); |
| avio_wl32(pb, ico->images[i].offset); |
| } |
| |
| av_freep(&ico->images); |
| |
| return 0; |
| } |
| |
| AVOutputFormat ff_ico_muxer = { |
| .name = "ico", |
| .long_name = NULL_IF_CONFIG_SMALL("Microsoft Windows ICO"), |
| .priv_data_size = sizeof(IcoMuxContext), |
| .mime_type = "image/vnd.microsoft.icon", |
| .extensions = "ico", |
| .audio_codec = AV_CODEC_ID_NONE, |
| .video_codec = AV_CODEC_ID_BMP, |
| .write_header = ico_write_header, |
| .write_packet = ico_write_packet, |
| .write_trailer = ico_write_trailer, |
| .flags = AVFMT_NOTIMESTAMPS, |
| }; |