Commit 9a277808 authored by Mats Peterson's avatar Mats Peterson Committed by Michael Niedermayer

lavf/movenc: Add palette to video sample description

Signed-off-by: 's avatarMichael Niedermayer <michael@niedermayer.cc>
parent 42c5e1cc
...@@ -1711,10 +1711,30 @@ static int mov_write_video_tag(AVIOContext *pb, MOVMuxContext *mov, MOVTrack *tr ...@@ -1711,10 +1711,30 @@ static int mov_write_video_tag(AVIOContext *pb, MOVMuxContext *mov, MOVTrack *tr
avio_write(pb, compressor_name, 31); avio_write(pb, compressor_name, 31);
if (track->mode == MODE_MOV && track->enc->bits_per_coded_sample) if (track->mode == MODE_MOV && track->enc->bits_per_coded_sample)
avio_wb16(pb, track->enc->bits_per_coded_sample); avio_wb16(pb, track->enc->bits_per_coded_sample |
(track->enc->pix_fmt == AV_PIX_FMT_GRAY8 ? 0x20 : 0));
else else
avio_wb16(pb, 0x18); /* Reserved */ avio_wb16(pb, 0x18); /* Reserved */
avio_wb16(pb, 0xffff); /* Reserved */
if (track->is_unaligned_qt_rgb && track->enc->pix_fmt == AV_PIX_FMT_PAL8) {
int i;
avio_wb16(pb, 0); /* Color table ID */
avio_wb32(pb, 0); /* Color table seed */
avio_wb16(pb, 0x8000); /* Color table flags */
avio_wb16(pb, 255); /* Color table size (zero-relative) */
for (i = 0; i < 256; i++) {
uint32_t rgb = AV_RL32(&track->palette[i]);
uint16_t r = (rgb >> 16) & 0xff;
uint16_t g = (rgb >> 8) & 0xff;
uint16_t b = rgb & 0xff;
avio_wb16(pb, 0);
avio_wb16(pb, (r << 8) | r);
avio_wb16(pb, (g << 8) | g);
avio_wb16(pb, (b << 8) | b);
}
} else
avio_wb16(pb, 0xffff); /* Reserved */
if (track->tag == MKTAG('m','p','4','v')) if (track->tag == MKTAG('m','p','4','v'))
mov_write_esds_tag(pb, track); mov_write_esds_tag(pb, track);
else if (track->enc->codec_id == AV_CODEC_ID_H263) else if (track->enc->codec_id == AV_CODEC_ID_H263)
...@@ -4703,6 +4723,7 @@ static int mov_write_packet(AVFormatContext *s, AVPacket *pkt) ...@@ -4703,6 +4723,7 @@ static int mov_write_packet(AVFormatContext *s, AVPacket *pkt)
} else { } else {
int i; int i;
MOVMuxContext *mov = s->priv_data; MOVMuxContext *mov = s->priv_data;
MOVTrack *trk = &mov->tracks[pkt->stream_index];
if (!pkt->size) if (!pkt->size)
return mov_write_single_packet(s, pkt); /* Passthrough. */ return mov_write_single_packet(s, pkt); /* Passthrough. */
...@@ -4739,6 +4760,31 @@ static int mov_write_packet(AVFormatContext *s, AVPacket *pkt) ...@@ -4739,6 +4760,31 @@ static int mov_write_packet(AVFormatContext *s, AVPacket *pkt)
} }
} }
if (trk->is_unaligned_qt_rgb) {
const uint8_t *data = pkt->data;
int size = pkt->size;
int64_t bpc = trk->enc->bits_per_coded_sample != 15 ? trk->enc->bits_per_coded_sample : 16;
int expected_stride = ((trk->enc->width * bpc + 15) >> 4)*2;
int ret = ff_reshuffle_raw_rgb(s, &pkt, trk->enc, expected_stride);
if (ret < 0)
return ret;
if (ret == CONTAINS_PAL && !trk->pal_done) {
int pal_size = 1 << trk->enc->bits_per_coded_sample;
memset(trk->palette, 0, AVPALETTE_SIZE);
memcpy(trk->palette, data + size - 4*pal_size, 4*pal_size);
trk->pal_done++;
} else if (trk->enc->pix_fmt == AV_PIX_FMT_GRAY8 ||
trk->enc->pix_fmt == AV_PIX_FMT_MONOBLACK) {
for (i = 0; i < pkt->size; i++)
pkt->data[i] = ~pkt->data[i];
}
if (ret) {
ret = mov_write_single_packet(s, pkt);
av_packet_free(&pkt);
return ret;
}
}
return mov_write_single_packet(s, pkt); return mov_write_single_packet(s, pkt);
} }
} }
...@@ -5249,6 +5295,20 @@ static int mov_write_header(AVFormatContext *s) ...@@ -5249,6 +5295,20 @@ static int mov_write_header(AVFormatContext *s)
"WARNING codec timebase is very high. If duration is too long,\n" "WARNING codec timebase is very high. If duration is too long,\n"
"file may not be playable by quicktime. Specify a shorter timebase\n" "file may not be playable by quicktime. Specify a shorter timebase\n"
"or choose different container.\n"); "or choose different container.\n");
if (track->mode == MODE_MOV &&
track->enc->codec_id == AV_CODEC_ID_RAWVIDEO &&
track->tag == MKTAG('r','a','w',' ')) {
enum AVPixelFormat pix_fmt = track->enc->pix_fmt;
if (pix_fmt == AV_PIX_FMT_NONE && track->enc->bits_per_coded_sample == 1)
pix_fmt = AV_PIX_FMT_MONOWHITE;
track->is_unaligned_qt_rgb =
pix_fmt == AV_PIX_FMT_RGB24 ||
pix_fmt == AV_PIX_FMT_BGR24 ||
pix_fmt == AV_PIX_FMT_PAL8 ||
pix_fmt == AV_PIX_FMT_GRAY8 ||
pix_fmt == AV_PIX_FMT_MONOWHITE ||
pix_fmt == AV_PIX_FMT_MONOBLACK;
}
} else if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) { } else if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
track->timescale = st->codec->sample_rate; track->timescale = st->codec->sample_rate;
if (!st->codec->frame_size && !av_get_bits_per_sample(st->codec->codec_id)) { if (!st->codec->frame_size && !av_get_bits_per_sample(st->codec->codec_id)) {
......
...@@ -152,6 +152,11 @@ typedef struct MOVTrack { ...@@ -152,6 +152,11 @@ typedef struct MOVTrack {
void *eac3_priv; void *eac3_priv;
MOVMuxCencContext cenc; MOVMuxCencContext cenc;
uint32_t palette[AVPALETTE_COUNT];
int pal_done;
int is_unaligned_qt_rgb;
} MOVTrack; } MOVTrack;
typedef enum { typedef enum {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment