forked from mia/Aegisub
86 lines
2.9 KiB
C++
86 lines
2.9 KiB
C++
// Copyright (c) 2011 Thomas Goyne <tgoyne@gmail.com>
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
// THE SOFTWARE.
|
|
|
|
#include "audiosource.h"
|
|
#include <cassert>
|
|
|
|
FFLAVFAudio::FFLAVFAudio(const char *SourceFile, int Track, FFMS_Index &Index, int DelayMode)
|
|
: FFMS_AudioSource(SourceFile, Index, Track)
|
|
, FormatContext(NULL)
|
|
{
|
|
LAVFOpenFile(SourceFile, FormatContext);
|
|
|
|
CodecContext.reset(FormatContext->streams[TrackNumber]->codec);
|
|
assert(CodecContext);
|
|
|
|
AVCodec *Codec = avcodec_find_decoder(CodecContext->codec_id);
|
|
try {
|
|
if (!Codec)
|
|
throw FFMS_Exception(FFMS_ERROR_DECODING, FFMS_ERROR_CODEC,
|
|
"Audio codec not found");
|
|
|
|
if (avcodec_open(CodecContext, Codec) < 0)
|
|
throw FFMS_Exception(FFMS_ERROR_DECODING, FFMS_ERROR_CODEC,
|
|
"Could not open audio codec");
|
|
}
|
|
catch (...) {
|
|
av_close_input_file(FormatContext);
|
|
throw;
|
|
}
|
|
|
|
if (Frames.back().PTS == Frames.front().PTS)
|
|
SeekOffset = -1;
|
|
else
|
|
SeekOffset = 10;
|
|
Init(Index, DelayMode);
|
|
}
|
|
|
|
FFLAVFAudio::~FFLAVFAudio() {
|
|
av_close_input_file(FormatContext);
|
|
}
|
|
|
|
void FFLAVFAudio::Seek() {
|
|
size_t TargetPacket = GetSeekablePacketNumber(Frames, PacketNumber);
|
|
|
|
if (av_seek_frame(FormatContext, TrackNumber, Frames[TargetPacket].PTS, AVSEEK_FLAG_BACKWARD) < 0)
|
|
av_seek_frame(FormatContext, TrackNumber, Frames[TargetPacket].PTS, AVSEEK_FLAG_BACKWARD | AVSEEK_FLAG_ANY);
|
|
|
|
if (TargetPacket != PacketNumber) {
|
|
// Decode until the PTS changes so we know where we are
|
|
int64_t LastPTS = Frames[PacketNumber].PTS;
|
|
while (LastPTS == Frames[PacketNumber].PTS) DecodeNextBlock();
|
|
}
|
|
}
|
|
|
|
bool FFLAVFAudio::ReadPacket(AVPacket *Packet) {
|
|
InitNullPacket(*Packet);
|
|
|
|
while (av_read_frame(FormatContext, Packet) >= 0) {
|
|
if (Packet->stream_index == TrackNumber) {
|
|
while (Frames[PacketNumber].PTS < Packet->pts) ++PacketNumber;
|
|
return true;
|
|
}
|
|
av_free_packet(Packet);
|
|
}
|
|
return false;
|
|
}
|
|
void FFLAVFAudio::FreePacket(AVPacket *Packet) {
|
|
av_free_packet(Packet);
|
|
}
|