Rename ebml size functions to indicate that they can be used for more kinds of numbers

Originally committed as revision 10386 to svn://svn.ffmpeg.org/ffmpeg/trunk
pull/126/head
David Conrad 18 years ago
parent 8b65b9a313
commit 0f651e8cf3
  1. 42
      libavformat/matroskaenc.c

@ -111,27 +111,27 @@ static void put_ebml_size_unknown(ByteIOContext *pb, int bytes)
} }
/** /**
* Calculate how many bytes are needed to represent a given size in EBML. * Calculate how many bytes are needed to represent a given number in EBML.
*/ */
static int ebml_size_bytes(uint64_t size) static int ebml_num_size(uint64_t num)
{ {
int bytes = 1; int bytes = 1;
while ((size+1) >> bytes*7) bytes++; while ((num+1) >> bytes*7) bytes++;
return bytes; return bytes;
} }
/** /**
* Write a size in EBML variable length format. * Write a number in EBML variable length format.
* *
* @param bytes The number of bytes that need to be used to write the size. * @param bytes The number of bytes that need to be used to write the number.
* If zero, any number of bytes can be used. * If zero, any number of bytes can be used.
*/ */
static void put_ebml_size(ByteIOContext *pb, uint64_t size, int bytes) static void put_ebml_num(ByteIOContext *pb, uint64_t num, int bytes)
{ {
int i, needed_bytes = ebml_size_bytes(size); int i, needed_bytes = ebml_num_size(num);
// sizes larger than this are currently undefined in EBML // sizes larger than this are currently undefined in EBML
assert(size < (1ULL<<56)-1); assert(num < (1ULL<<56)-1);
if (bytes == 0) if (bytes == 0)
// don't care how many bytes are used, so use the min // don't care how many bytes are used, so use the min
@ -140,9 +140,9 @@ static void put_ebml_size(ByteIOContext *pb, uint64_t size, int bytes)
// that we need to use, so write unknown size. This shouldn't happen. // that we need to use, so write unknown size. This shouldn't happen.
assert(bytes >= needed_bytes); assert(bytes >= needed_bytes);
size |= 1ULL << bytes*7; num |= 1ULL << bytes*7;
for (i = bytes - 1; i >= 0; i--) for (i = bytes - 1; i >= 0; i--)
put_byte(pb, size >> i*8); put_byte(pb, num >> i*8);
} }
static void put_ebml_uint(ByteIOContext *pb, unsigned int elementid, uint64_t val) static void put_ebml_uint(ByteIOContext *pb, unsigned int elementid, uint64_t val)
@ -151,7 +151,7 @@ static void put_ebml_uint(ByteIOContext *pb, unsigned int elementid, uint64_t va
while (val >> bytes*8) bytes++; while (val >> bytes*8) bytes++;
put_ebml_id(pb, elementid); put_ebml_id(pb, elementid);
put_ebml_size(pb, bytes, 0); put_ebml_num(pb, bytes, 0);
for (i = bytes - 1; i >= 0; i--) for (i = bytes - 1; i >= 0; i--)
put_byte(pb, val >> i*8); put_byte(pb, val >> i*8);
} }
@ -159,7 +159,7 @@ static void put_ebml_uint(ByteIOContext *pb, unsigned int elementid, uint64_t va
static void put_ebml_float(ByteIOContext *pb, unsigned int elementid, double val) static void put_ebml_float(ByteIOContext *pb, unsigned int elementid, double val)
{ {
put_ebml_id(pb, elementid); put_ebml_id(pb, elementid);
put_ebml_size(pb, 8, 0); put_ebml_num(pb, 8, 0);
put_be64(pb, av_dbl2int(val)); put_be64(pb, av_dbl2int(val));
} }
@ -167,7 +167,7 @@ static void put_ebml_binary(ByteIOContext *pb, unsigned int elementid,
const uint8_t *buf, int size) const uint8_t *buf, int size)
{ {
put_ebml_id(pb, elementid); put_ebml_id(pb, elementid);
put_ebml_size(pb, size, 0); put_ebml_num(pb, size, 0);
put_buffer(pb, buf, size); put_buffer(pb, buf, size);
} }
@ -193,15 +193,15 @@ static void put_ebml_void(ByteIOContext *pb, uint64_t size)
// size we need to reserve so 2 cases, we use 8 bytes to store the // size we need to reserve so 2 cases, we use 8 bytes to store the
// size if possible, 1 byte otherwise // size if possible, 1 byte otherwise
if (size < 10) if (size < 10)
put_ebml_size(pb, size-1, 0); put_ebml_num(pb, size-1, 0);
else else
put_ebml_size(pb, size-9, 8); put_ebml_num(pb, size-9, 8);
url_fseek(pb, currentpos + size, SEEK_SET); url_fseek(pb, currentpos + size, SEEK_SET);
} }
static ebml_master start_ebml_master(ByteIOContext *pb, unsigned int elementid, uint64_t expectedsize) static ebml_master start_ebml_master(ByteIOContext *pb, unsigned int elementid, uint64_t expectedsize)
{ {
int bytes = expectedsize ? ebml_size_bytes(expectedsize) : 8; int bytes = expectedsize ? ebml_num_size(expectedsize) : 8;
put_ebml_id(pb, elementid); put_ebml_id(pb, elementid);
put_ebml_size_unknown(pb, bytes); put_ebml_size_unknown(pb, bytes);
return (ebml_master){ url_ftell(pb), bytes }; return (ebml_master){ url_ftell(pb), bytes };
@ -216,7 +216,7 @@ static void end_ebml_master(ByteIOContext *pb, ebml_master master)
return; return;
url_fseek(pb, master.pos - master.sizebytes, SEEK_SET); url_fseek(pb, master.pos - master.sizebytes, SEEK_SET);
put_ebml_size(pb, pos - master.pos, master.sizebytes); put_ebml_num(pb, pos - master.pos, master.sizebytes);
url_fseek(pb, pos, SEEK_SET); url_fseek(pb, pos, SEEK_SET);
} }
@ -304,7 +304,7 @@ static offset_t mkv_write_seekhead(ByteIOContext *pb, mkv_seekhead *seekhead)
seekentry = start_ebml_master(pb, MATROSKA_ID_SEEKENTRY, MAX_SEEKENTRY_SIZE); seekentry = start_ebml_master(pb, MATROSKA_ID_SEEKENTRY, MAX_SEEKENTRY_SIZE);
put_ebml_id(pb, MATROSKA_ID_SEEKID); put_ebml_id(pb, MATROSKA_ID_SEEKID);
put_ebml_size(pb, ebml_id_size(entry->elementid), 0); put_ebml_num(pb, ebml_id_size(entry->elementid), 0);
put_ebml_id(pb, entry->elementid); put_ebml_id(pb, entry->elementid);
put_ebml_uint(pb, MATROSKA_ID_SEEKPOSITION, entry->segmentpos); put_ebml_uint(pb, MATROSKA_ID_SEEKPOSITION, entry->segmentpos);
@ -692,10 +692,10 @@ static int mkv_block_size(AVPacket *pkt)
static int mkv_blockgroup_size(AVPacket *pkt) static int mkv_blockgroup_size(AVPacket *pkt)
{ {
int size = mkv_block_size(pkt); int size = mkv_block_size(pkt);
size += ebml_size_bytes(size); size += ebml_num_size(size);
size += 2; // EBML ID for block and block duration size += 2; // EBML ID for block and block duration
size += 8; // max size of block duration size += 8; // max size of block duration
size += ebml_size_bytes(size); size += ebml_num_size(size);
size += 1; // blockgroup EBML ID size += 1; // blockgroup EBML ID
return size; return size;
} }
@ -709,7 +709,7 @@ static void mkv_write_block(AVFormatContext *s, unsigned int blockid, AVPacket *
"pts %" PRId64 ", dts %" PRId64 ", duration %d, flags %d\n", "pts %" PRId64 ", dts %" PRId64 ", duration %d, flags %d\n",
url_ftell(pb), pkt->size, pkt->pts, pkt->dts, pkt->duration, flags); url_ftell(pb), pkt->size, pkt->pts, pkt->dts, pkt->duration, flags);
put_ebml_id(pb, blockid); put_ebml_id(pb, blockid);
put_ebml_size(pb, mkv_block_size(pkt), 0); put_ebml_num(pb, mkv_block_size(pkt), 0);
put_byte(pb, 0x80 | (pkt->stream_index + 1)); // this assumes stream_index is less than 126 put_byte(pb, 0x80 | (pkt->stream_index + 1)); // this assumes stream_index is less than 126
put_be16(pb, pkt->pts - mkv->cluster_pts); put_be16(pb, pkt->pts - mkv->cluster_pts);
put_byte(pb, flags); put_byte(pb, flags);

Loading…
Cancel
Save