@@ -29,6 +29,7 @@ | |||||
#include "internal.h" | #include "internal.h" | ||||
#include "riff.h" | #include "riff.h" | ||||
#include "isom.h" | #include "isom.h" | ||||
#include "mov_chan.h" | |||||
#include "libavutil/intreadwrite.h" | #include "libavutil/intreadwrite.h" | ||||
#include "libavutil/intfloat.h" | #include "libavutil/intfloat.h" | ||||
#include "libavutil/dict.h" | #include "libavutil/dict.h" | ||||
@@ -266,6 +267,11 @@ static int read_header(AVFormatContext *s) | |||||
found_data = 1; | found_data = 1; | ||||
break; | break; | ||||
case MKBETAG('c','h','a','n'): | |||||
if ((ret = ff_mov_read_chan(s, st, size)) < 0) | |||||
return ret; | |||||
break; | |||||
/* magic cookie chunk */ | /* magic cookie chunk */ | ||||
case MKBETAG('k','u','k','i'): | case MKBETAG('k','u','k','i'): | ||||
if (read_kuki_chunk(s, size)) | if (read_kuki_chunk(s, size)) | ||||
@@ -575,10 +575,6 @@ static int mov_read_dec3(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
static int mov_read_chan(MOVContext *c, AVIOContext *pb, MOVAtom atom) | static int mov_read_chan(MOVContext *c, AVIOContext *pb, MOVAtom atom) | ||||
{ | { | ||||
AVStream *st; | AVStream *st; | ||||
uint8_t av_unused version; | |||||
uint32_t av_unused flags; | |||||
uint32_t layout_tag, bitmap, num_descr, label_mask; | |||||
int i; | |||||
if (c->fc->nb_streams < 1) | if (c->fc->nb_streams < 1) | ||||
return 0; | return 0; | ||||
@@ -587,40 +583,7 @@ static int mov_read_chan(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
if (atom.size < 16) | if (atom.size < 16) | ||||
return 0; | return 0; | ||||
version = avio_r8(pb); | |||||
flags = avio_rb24(pb); | |||||
layout_tag = avio_rb32(pb); | |||||
bitmap = avio_rb32(pb); | |||||
num_descr = avio_rb32(pb); | |||||
if (atom.size < 16ULL + num_descr * 20ULL) | |||||
return 0; | |||||
av_dlog(c->fc, "chan: size=%ld version=%u flags=%u layout=%u bitmap=%u num_descr=%u\n", | |||||
atom.size, version, flags, layout_tag, bitmap, num_descr); | |||||
label_mask = 0; | |||||
for (i = 0; i < num_descr; i++) { | |||||
uint32_t label; | |||||
label = avio_rb32(pb); // mChannelLabel | |||||
avio_rb32(pb); // mChannelFlags | |||||
avio_rl32(pb); // mCoordinates[0] | |||||
avio_rl32(pb); // mCoordinates[1] | |||||
avio_rl32(pb); // mCoordinates[2] | |||||
if (layout_tag == 0) { | |||||
uint32_t mask_incr = ff_mov_get_channel_label(label); | |||||
if (mask_incr == 0) { | |||||
label_mask = 0; | |||||
break; | |||||
} | |||||
label_mask |= mask_incr; | |||||
} | |||||
} | |||||
if (layout_tag == 0) | |||||
st->codec->channel_layout = label_mask; | |||||
else | |||||
st->codec->channel_layout = ff_mov_get_channel_layout(layout_tag, bitmap); | |||||
ff_mov_read_chan(c->fc, st, atom.size - 4); | |||||
return 0; | return 0; | ||||
} | } | ||||
@@ -477,7 +477,7 @@ uint64_t ff_mov_get_channel_layout(uint32_t tag, uint32_t bitmap) | |||||
return layout_map[i].layout; | return layout_map[i].layout; | ||||
} | } | ||||
uint32_t ff_mov_get_channel_label(uint32_t label) | |||||
static uint32_t mov_get_channel_label(uint32_t label) | |||||
{ | { | ||||
if (label == 0) | if (label == 0) | ||||
return 0; | return 0; | ||||
@@ -542,3 +542,47 @@ uint32_t ff_mov_get_channel_layout_tag(enum CodecID codec_id, | |||||
return tag; | return tag; | ||||
} | } | ||||
int ff_mov_read_chan(AVFormatContext *s, AVStream *st, int64_t size) | |||||
{ | |||||
AVIOContext *pb = s->pb; | |||||
uint32_t layout_tag, bitmap, num_descr, label_mask; | |||||
int i; | |||||
if (size < 12) | |||||
return AVERROR_INVALIDDATA; | |||||
layout_tag = avio_rb32(pb); | |||||
bitmap = avio_rb32(pb); | |||||
num_descr = avio_rb32(pb); | |||||
av_dlog(s, "chan: layout=%u bitmap=%u num_descr=%u\n", | |||||
layout_tag, bitmap, num_descr); | |||||
if (size < 12ULL + num_descr * 20ULL) | |||||
return 0; | |||||
label_mask = 0; | |||||
for (i = 0; i < num_descr; i++) { | |||||
uint32_t label; | |||||
label = avio_rb32(pb); // mChannelLabel | |||||
avio_rb32(pb); // mChannelFlags | |||||
avio_rl32(pb); // mCoordinates[0] | |||||
avio_rl32(pb); // mCoordinates[1] | |||||
avio_rl32(pb); // mCoordinates[2] | |||||
if (layout_tag == 0) { | |||||
uint32_t mask_incr = mov_get_channel_label(label); | |||||
if (mask_incr == 0) { | |||||
label_mask = 0; | |||||
break; | |||||
} | |||||
label_mask |= mask_incr; | |||||
} | |||||
} | |||||
if (layout_tag == 0) | |||||
st->codec->channel_layout = label_mask; | |||||
else | |||||
st->codec->channel_layout = ff_mov_get_channel_layout(layout_tag, bitmap); | |||||
return 0; | |||||
} |
@@ -29,6 +29,7 @@ | |||||
#include <stdint.h> | #include <stdint.h> | ||||
#include "libavcodec/avcodec.h" | #include "libavcodec/avcodec.h" | ||||
#include "avformat.h" | |||||
/** | /** | ||||
* Get the channel layout for the specified channel layout tag. | * Get the channel layout for the specified channel layout tag. | ||||
@@ -39,14 +40,6 @@ | |||||
*/ | */ | ||||
uint64_t ff_mov_get_channel_layout(uint32_t tag, uint32_t bitmap); | uint64_t ff_mov_get_channel_layout(uint32_t tag, uint32_t bitmap); | ||||
/** | |||||
* Get the channel layout for the specified channel label. | |||||
* | |||||
* @param[in] label channel label | |||||
* @return channel layout mask fragment | |||||
*/ | |||||
uint32_t ff_mov_get_channel_label(uint32_t label); | |||||
/** | /** | ||||
* Get the channel layout tag for the specified codec id and channel layout. | * Get the channel layout tag for the specified codec id and channel layout. | ||||
* If the layout tag was not found, use a channel bitmap if possible. | * If the layout tag was not found, use a channel bitmap if possible. | ||||
@@ -60,4 +53,14 @@ uint32_t ff_mov_get_channel_layout_tag(enum CodecID codec_id, | |||||
uint64_t channel_layout, | uint64_t channel_layout, | ||||
uint32_t *bitmap); | uint32_t *bitmap); | ||||
/** | |||||
* Read 'chan' tag from the input stream. | |||||
* | |||||
* @param s AVFormatContext | |||||
* @param st The stream to set codec values for | |||||
* @param size Remaining size in the 'chan' tag | |||||
* @return 0 if ok, or negative AVERROR code on failure | |||||
*/ | |||||
int ff_mov_read_chan(AVFormatContext *s, AVStream *st, int64_t size); | |||||
#endif /* AVFORMAT_MOV_CHAN_H */ | #endif /* AVFORMAT_MOV_CHAN_H */ |