file_status_e
avi_reader_c::read_audio(avi_demuxer_t &demuxer) {
AVI_set_audio_track(m_avi, demuxer.m_aid);
while (true) {
int size = AVI_audio_size(m_avi, AVI_get_audio_position_index(m_avi));
// -1 indicates the last chunk.
if (-1 == size)
return flush_packetizer(demuxer.m_ptzr);
// Sanity check. Ignore chunks with obvious wrong size information
// (> 10 MB). Also skip 0-sized blocks. Those are officially
// skipped.
if (!size || (size > AVI_MAX_AUDIO_CHUNK_SIZE)) {
AVI_set_audio_position_index(m_avi, AVI_get_audio_position_index(m_avi) + 1);
continue;
}
auto chunk = memory_c::alloc(size);
size = AVI_read_audio_chunk(m_avi, reinterpret_cast<char *>(chunk->get_buffer()));
if (0 > size)
return flush_packetizer(demuxer.m_ptzr);
if (!size)
continue;
PTZR(demuxer.m_ptzr)->process(new packet_t(chunk));
m_bytes_processed += size;
return AVI_get_audio_position_index(m_avi) < AVI_max_audio_chunk(m_avi) ? FILE_STATUS_MOREDATA : flush_packetizer(demuxer.m_ptzr);
}
}
file_status_e
avi_reader_c::read_subtitles(avi_subs_demuxer_t &demuxer) {
if (!demuxer.m_subs->empty())
demuxer.m_subs->process(PTZR(demuxer.m_ptzr));
return demuxer.m_subs->empty() ? flush_packetizer(demuxer.m_ptzr) : FILE_STATUS_MOREDATA;
}
file_status_e
avi_reader_c::read_audio(avi_demuxer_t &demuxer) {
AVI_set_audio_track(m_avi, demuxer.m_aid);
while (true) {
int size = AVI_audio_size(m_avi, AVI_get_audio_position_index(m_avi));
if (-1 == size)
return flush_packetizer(demuxer.m_ptzr);
if (!size) {
AVI_set_audio_position_index(m_avi, AVI_get_audio_position_index(m_avi) + 1);
continue;
}
auto chunk = memory_c::alloc(size);
size = AVI_read_audio_chunk(m_avi, reinterpret_cast<char *>(chunk->get_buffer()));
if (0 >= size)
continue;
PTZR(demuxer.m_ptzr)->add_avi_block_size(size);
PTZR(demuxer.m_ptzr)->process(new packet_t(chunk));
m_bytes_processed += size;
return AVI_get_audio_position_index(m_avi) < AVI_max_audio_chunk(m_avi) ? FILE_STATUS_MOREDATA : flush_packetizer(demuxer.m_ptzr);
}
}
file_status_e
usf_reader_c::read(generic_packetizer_c *ptzr,
bool) {
auto track_itr = brng::find_if(m_tracks, [&](usf_track_cptr &tr) { return (-1 != tr->m_ptzr) && (PTZR(tr->m_ptzr) == ptzr); });
if (track_itr == m_tracks.end())
return FILE_STATUS_DONE;
auto &track = *track_itr;
if (track->m_entries.end() == track->m_current_entry)
return flush_packetizer(track->m_ptzr);
auto &entry = *track->m_current_entry;
PTZR(track->m_ptzr)->process(new packet_t(memory_c::clone(entry.m_text), entry.m_start, entry.m_end - entry.m_start));
++track->m_current_entry;
if (track->m_entries.end() == track->m_current_entry)
return flush_packetizer(track->m_ptzr);
return FILE_STATUS_MOREDATA;
}
file_status_e
usf_reader_c::read(generic_packetizer_c *ptzr,
bool) {
auto track_itr = brng::find_if(m_tracks, [&](usf_track_cptr &tr) { return (-1 != tr->m_ptzr) && (PTZR(tr->m_ptzr) == ptzr); });
if (track_itr == m_tracks.end())
return FILE_STATUS_DONE;
auto track = *track_itr;
if (track->m_entries.end() == track->m_current_entry)
return flush_packetizer(track->m_ptzr);
const usf_entry_t &entry = *(track->m_current_entry);
// A length of 0 here is OK because the text subtitle packetizer assumes
// that the data is a zero-terminated string.
memory_cptr mem(new memory_c((unsigned char *)entry.m_text.c_str(), 0, false));
PTZR(track->m_ptzr)->process(new packet_t(mem, entry.m_start, entry.m_end - entry.m_start));
++(track->m_current_entry);
if (track->m_entries.end() == track->m_current_entry)
return flush_packetizer(track->m_ptzr);
return FILE_STATUS_MOREDATA;
}
file_status_e
avi_reader_c::read_video() {
if (m_video_frames_read >= m_max_video_frames)
return flush_packetizer(m_vptzr);
memory_cptr chunk;
int key = 0;
int old_video_frames_read = m_video_frames_read;
int size, num_read;
int dropped_frames_here = 0;
do {
size = AVI_frame_size(m_avi, m_video_frames_read);
chunk = memory_c::alloc(size);
num_read = AVI_read_frame(m_avi, reinterpret_cast<char *>(chunk->get_buffer()), &key);
++m_video_frames_read;
if (0 > num_read) {
// Error reading the frame: abort
m_video_frames_read = m_max_video_frames;
return flush_packetizer(m_vptzr);
} else if (0 == num_read)
++dropped_frames_here;
} while ((0 == num_read) && (m_video_frames_read < m_max_video_frames));
if (0 == num_read)
// This is only the case if the AVI contains dropped frames only.
return flush_packetizer(m_vptzr);
size_t i;
for (i = m_video_frames_read; i < m_max_video_frames; ++i) {
if (0 != AVI_frame_size(m_avi, i))
break;
int dummy_key;
AVI_read_frame(m_avi, nullptr, &dummy_key);
++dropped_frames_here;
++m_video_frames_read;
}
int64_t timestamp = static_cast<int64_t>(static_cast<int64_t>(old_video_frames_read) * 1000000000ll / m_fps);
int64_t duration = static_cast<int64_t>(static_cast<int64_t>(dropped_frames_here + 1) * 1000000000ll / m_fps);
m_dropped_video_frames += dropped_frames_here;
// AVC with framed packets (without NALU start codes but with length fields)
// or non-AVC video track?
if (0 >= m_avc_nal_size_size)
PTZR(m_vptzr)->process(new packet_t(chunk, timestamp, duration, key ? VFT_IFRAME : VFT_PFRAMEAUTOMATIC, VFT_NOBFRAME));
else {
// AVC video track without NALU start codes. Re-frame with NALU start codes.
int offset = 0;
while ((offset + m_avc_nal_size_size) < num_read) {
int nalu_size = get_uint_be(chunk->get_buffer() + offset, m_avc_nal_size_size);
offset += m_avc_nal_size_size;
if ((offset + nalu_size) > num_read)
break;
memory_cptr nalu = memory_c::alloc(4 + nalu_size);
put_uint32_be(nalu->get_buffer(), NALU_START_CODE);
memcpy(nalu->get_buffer() + 4, chunk->get_buffer() + offset, nalu_size);
offset += nalu_size;
PTZR(m_vptzr)->process(new packet_t(nalu, timestamp, duration, key ? VFT_IFRAME : VFT_PFRAMEAUTOMATIC, VFT_NOBFRAME));
}
}
m_bytes_processed += num_read;
return m_video_frames_read >= m_max_video_frames ? flush_packetizer(m_vptzr) : FILE_STATUS_MOREDATA;
}
file_status_e
generic_reader_c::flush_packetizer(int num) {
return flush_packetizer(PTZR(num));
}
