1
0
Fork 1
mirror of https://github.com/TES3MP/openmw-tes3mp.git synced 2025-01-16 15:49:55 +00:00
openmw-tes3mp/apps/openmw/mwrender/videoplayer.cpp

995 lines
32 KiB
C++
Raw Normal View History

2012-09-25 00:35:50 +00:00
#include "videoplayer.hpp"
#include "../mwbase/windowmanager.hpp"
#include "../mwbase/environment.hpp"
#include "../mwbase/soundmanager.hpp"
#include "../mwsound/sound_decoder.hpp"
#include "../mwsound/sound.hpp"
2012-09-25 00:35:50 +00:00
2012-09-25 00:35:50 +00:00
namespace MWRender
{
2012-12-13 23:10:54 +00:00
int OgreResource_Read(void *user_data, uint8_t *buf, int buf_size)
2012-09-25 00:35:50 +00:00
{
2012-12-13 23:10:54 +00:00
Ogre::DataStreamPtr stream = static_cast<VideoState*>(user_data)->stream;
return stream->read(buf, buf_size);
2012-09-25 00:35:50 +00:00
}
2012-12-13 23:10:54 +00:00
int OgreResource_Write(void *user_data, uint8_t *buf, int buf_size)
2012-09-25 00:35:50 +00:00
{
2012-12-13 23:10:54 +00:00
Ogre::DataStreamPtr stream = static_cast<VideoState*>(user_data)->stream;
return stream->write(buf, buf_size);
2012-09-25 00:35:50 +00:00
}
2012-12-13 23:10:54 +00:00
int64_t OgreResource_Seek(void *user_data, int64_t offset, int whence)
2012-09-25 00:35:50 +00:00
{
2012-12-13 23:10:54 +00:00
Ogre::DataStreamPtr stream = static_cast<VideoState*>(user_data)->stream;
whence &= ~AVSEEK_FORCE;
if(whence == AVSEEK_SIZE)
return stream->size();
if(whence == SEEK_SET)
stream->seek(offset);
else if(whence == SEEK_CUR)
stream->seek(stream->tell()+offset);
else if(whence == SEEK_END)
stream->seek(stream->size()+offset);
else
return -1;
2012-09-25 00:35:50 +00:00
2012-12-11 22:06:06 +00:00
return stream->tell();
2012-09-25 00:35:50 +00:00
}
2012-12-13 11:37:04 +00:00
void packet_queue_init(PacketQueue *q)
{ memset(q, 0, sizeof(PacketQueue)); }
int packet_queue_put(PacketQueue *q, AVPacket *pkt)
{
2012-12-11 21:49:31 +00:00
AVPacketList *pkt1;
if(av_dup_packet(pkt) < 0)
return -1;
pkt1 = (AVPacketList*)av_malloc(sizeof(AVPacketList));
if(!pkt1) return -1;
pkt1->pkt = *pkt;
pkt1->next = NULL;
2012-12-11 21:49:31 +00:00
q->mutex.lock ();
2012-12-11 21:49:31 +00:00
if (!q->last_pkt)
q->first_pkt = pkt1;
else
q->last_pkt->next = pkt1;
q->last_pkt = pkt1;
q->nb_packets++;
q->size += pkt1->pkt.size;
q->cond.notify_one();
q->mutex.unlock ();
return 0;
}
2012-12-13 11:37:04 +00:00
static int packet_queue_get(PacketQueue *q, AVPacket *pkt, VideoState *is, int block)
{
2012-12-11 21:49:31 +00:00
AVPacketList *pkt1;
int ret;
2012-12-11 21:49:31 +00:00
boost::unique_lock<boost::mutex> lock(q->mutex);
for(;;)
{
2012-12-13 11:37:04 +00:00
if(is->quit)
{
2012-12-11 21:49:31 +00:00
ret = -1;
break;
}
2012-12-11 21:49:31 +00:00
pkt1 = q->first_pkt;
if (pkt1)
{
2012-12-11 21:49:31 +00:00
q->first_pkt = pkt1->next;
if (!q->first_pkt)
2012-12-11 22:06:06 +00:00
q->last_pkt = NULL;
2012-12-11 21:49:31 +00:00
q->nb_packets--;
q->size -= pkt1->pkt.size;
*pkt = pkt1->pkt;
av_free(pkt1);
2012-12-11 21:49:31 +00:00
ret = 1;
break;
}
if (!block)
{
2012-12-11 21:49:31 +00:00
ret = 0;
break;
}
q->cond.wait(lock);
2012-12-11 21:49:31 +00:00
}
2012-12-11 21:49:31 +00:00
return ret;
}
2012-12-13 11:37:04 +00:00
void PacketQueue::flush()
{
2012-12-11 21:49:31 +00:00
AVPacketList *pkt, *pkt1;
this->mutex.lock();
for(pkt = this->first_pkt; pkt != NULL; pkt = pkt1) {
2012-12-11 21:49:31 +00:00
pkt1 = pkt->next;
av_free_packet(&pkt->pkt);
av_freep(&pkt);
}
this->last_pkt = NULL;
this->first_pkt = NULL;
this->nb_packets = 0;
this->size = 0;
this->mutex.unlock ();
2012-12-11 21:49:31 +00:00
}
2012-12-13 11:37:04 +00:00
double get_audio_clock(VideoState *is)
{
return is->AudioTrack->getTimeOffset();
2012-12-11 21:49:31 +00:00
}
double get_video_clock(VideoState *is)
{
2012-12-11 21:49:31 +00:00
double delta;
delta = (av_gettime() - is->video_current_pts_time) / 1000000.0;
return is->video_current_pts + delta;
}
2012-12-13 11:37:04 +00:00
double get_external_clock(VideoState *is)
{
2012-12-11 21:49:31 +00:00
return av_gettime() / 1000000.0;
}
2012-12-13 11:37:04 +00:00
double get_master_clock(VideoState *is)
{
if(is->av_sync_type == AV_SYNC_VIDEO_MASTER)
2012-12-11 21:49:31 +00:00
return get_video_clock(is);
2012-12-13 11:37:04 +00:00
if(is->av_sync_type == AV_SYNC_AUDIO_MASTER)
2012-12-11 21:49:31 +00:00
return get_audio_clock(is);
2012-12-13 11:37:04 +00:00
return get_external_clock(is);
2012-12-11 21:49:31 +00:00
}
class MovieAudioDecoder : public MWSound::Sound_Decoder
{
static void fail(const std::string &str)
{
throw std::runtime_error(str);
}
VideoState *is;
2012-12-14 01:53:22 +00:00
AVFrame *mFrame;
size_t mFramePos;
size_t mFrameSize;
2012-12-11 21:49:31 +00:00
/* Add or subtract samples to get a better sync, return new
* audio buffer size */
int synchronize_audio(uint8_t *samples, int samples_size, double pts)
{
if(is->av_sync_type == AV_SYNC_AUDIO_MASTER)
return samples_size;
double diff, avg_diff, ref_clock;
int wanted_size, min_size, max_size, n;
// int nb_samples;
2012-12-11 21:49:31 +00:00
n = 2 * is->audio_st->codec->channels;
ref_clock = get_master_clock(is);
diff = get_audio_clock(is) - ref_clock;
if(diff < AV_NOSYNC_THRESHOLD)
{
// accumulate the diffs
is->audio_diff_cum = diff + is->audio_diff_avg_coef *
is->audio_diff_cum;
if(is->audio_diff_avg_count < AUDIO_DIFF_AVG_NB)
is->audio_diff_avg_count++;
else
{
avg_diff = is->audio_diff_cum * (1.0 - is->audio_diff_avg_coef);
if(fabs(avg_diff) >= is->audio_diff_threshold)
{
wanted_size = samples_size + ((int)(diff * is->audio_st->codec->sample_rate) * n);
2012-12-14 02:05:27 +00:00
min_size = samples_size/n * (100-SAMPLE_CORRECTION_PERCENT_MAX) / 100 * n;
max_size = samples_size/n * (100+SAMPLE_CORRECTION_PERCENT_MAX) / 100 * n;
if(wanted_size < min_size)
wanted_size = min_size;
else if (wanted_size > max_size)
wanted_size = max_size;
if(wanted_size < samples_size)
{
/* remove samples */
samples_size = wanted_size;
}
else if(wanted_size > samples_size)
{
uint8_t *samples_end, *q;
int nb;
2012-12-14 01:53:22 +00:00
/* add samples by copying final sample*/
nb = (samples_size - wanted_size);
samples_end = samples + samples_size - n;
q = samples_end + n;
2012-12-14 01:53:22 +00:00
while(nb > 0)
{
memcpy(q, samples_end, n);
q += n;
nb -= n;
2012-12-12 04:11:48 +00:00
}
2012-12-14 01:53:22 +00:00
samples_size = wanted_size;
2012-12-12 04:11:48 +00:00
}
2012-12-11 21:49:31 +00:00
}
}
}
else
{
/* difference is TOO big; reset diff stuff */
is->audio_diff_avg_count = 0;
is->audio_diff_cum = 0;
}
2012-12-11 21:49:31 +00:00
return samples_size;
}
2012-12-14 01:53:22 +00:00
int audio_decode_frame(AVFrame *frame, double *pts_ptr)
{
2012-12-11 21:49:31 +00:00
AVPacket *pkt = &is->audio_pkt;
2012-12-14 01:53:22 +00:00
int len1, data_size;
2012-12-11 21:49:31 +00:00
for(;;)
{
2012-12-14 01:53:22 +00:00
while(pkt->size > 0)
{
2012-12-14 01:53:22 +00:00
int got_frame;
2012-12-14 01:53:22 +00:00
len1 = avcodec_decode_audio4(is->audio_st->codec, frame, &got_frame, pkt);
if(len1 < 0 || len1 > pkt->size)
{
2012-12-14 01:53:22 +00:00
/* if error, skip packet */
2012-12-11 22:06:06 +00:00
break;
2012-12-11 21:49:31 +00:00
}
2012-12-14 01:53:22 +00:00
if(len1 <= pkt->size)
{
/* Move the unread data to the front and clear the end bits */
int remaining = pkt->size - len1;
memmove(pkt->data, &pkt->data[len1], remaining);
memset(&pkt->data[remaining], 0, pkt->size - remaining);
pkt->size -= len1;
}
if(!got_frame)
continue;
int smp_size = av_samples_get_buffer_size(NULL, is->audio_st->codec->channels, 1,
is->audio_st->codec->sample_fmt, 1);
data_size = frame->nb_samples * smp_size;
if(data_size <= 0)
{
2012-12-12 04:11:48 +00:00
/* No data yet, get more frames */
2012-12-11 22:06:06 +00:00
continue;
2012-12-11 21:49:31 +00:00
}
2012-12-14 01:53:22 +00:00
*pts_ptr = is->audio_clock;
is->audio_clock += (double)(data_size/smp_size) /
(double)is->audio_st->codec->sample_rate;
2012-12-11 21:49:31 +00:00
/* We have data, return it and come back for more later */
return data_size;
}
if(pkt->data)
av_free_packet(pkt);
if(is->quit)
2012-12-11 21:49:31 +00:00
return -1;
2012-12-11 21:49:31 +00:00
/* next packet */
2012-12-13 11:37:04 +00:00
if(packet_queue_get(&is->audioq, pkt, is, 1) < 0)
2012-12-11 21:49:31 +00:00
return -1;
2012-12-11 21:49:31 +00:00
/* if update, update the audio clock w/pts */
if((uint64_t)pkt->pts != AV_NOPTS_VALUE)
2012-12-11 21:49:31 +00:00
is->audio_clock = av_q2d(is->audio_st->time_base)*pkt->pts;
}
}
void open(const std::string&)
{ fail(std::string("Invalid call to ")+__PRETTY_FUNCTION__); }
void close() { }
std::string getName()
{ return is->stream->getName(); }
void rewind() { }
public:
2012-12-14 01:53:22 +00:00
MovieAudioDecoder(VideoState *_is)
: is(_is)
, mFrame(avcodec_alloc_frame())
, mFramePos(0)
, mFrameSize(0)
{ }
virtual ~MovieAudioDecoder()
{
av_freep(&mFrame);
}
void getInfo(int *samplerate, MWSound::ChannelConfig *chans, MWSound::SampleType * type)
{
if(is->audio_st->codec->sample_fmt == AV_SAMPLE_FMT_U8)
*type = MWSound::SampleType_UInt8;
else if(is->audio_st->codec->sample_fmt == AV_SAMPLE_FMT_S16)
*type = MWSound::SampleType_Int16;
else
fail(std::string("Unsupported sample format: ")+
2012-12-13 11:37:04 +00:00
av_get_sample_fmt_name(is->audio_st->codec->sample_fmt));
if(is->audio_st->codec->channel_layout == AV_CH_LAYOUT_MONO)
*chans = MWSound::ChannelConfig_Mono;
else if(is->audio_st->codec->channel_layout == AV_CH_LAYOUT_STEREO)
*chans = MWSound::ChannelConfig_Stereo;
else if(is->audio_st->codec->channel_layout == AV_CH_LAYOUT_QUAD)
*chans = MWSound::ChannelConfig_Quad;
else if(is->audio_st->codec->channel_layout == AV_CH_LAYOUT_5POINT1)
*chans = MWSound::ChannelConfig_5point1;
else if(is->audio_st->codec->channel_layout == AV_CH_LAYOUT_7POINT1)
*chans = MWSound::ChannelConfig_7point1;
else if(is->audio_st->codec->channel_layout == 0)
{
/* Unknown channel layout. Try to guess. */
if(is->audio_st->codec->channels == 1)
*chans = MWSound::ChannelConfig_Mono;
else if(is->audio_st->codec->channels == 2)
*chans = MWSound::ChannelConfig_Stereo;
else
{
std::stringstream sstr("Unsupported raw channel count: ");
sstr << is->audio_st->codec->channels;
fail(sstr.str());
2012-12-11 21:49:31 +00:00
}
}
else
{
char str[1024];
av_get_channel_layout_string(str, sizeof(str), is->audio_st->codec->channels,
is->audio_st->codec->channel_layout);
fail(std::string("Unsupported channel layout: ")+str);
}
*samplerate = is->audio_st->codec->sample_rate;
2012-12-11 21:49:31 +00:00
}
size_t read(char *stream, size_t len)
{
size_t total = 0;
2012-12-11 21:49:31 +00:00
while(total < len)
{
2012-12-14 01:53:22 +00:00
if(mFramePos >= mFrameSize)
{
int audio_size;
double pts;
2012-12-14 01:53:22 +00:00
2012-12-11 21:49:31 +00:00
/* We have already sent all our data; get more */
2012-12-14 01:53:22 +00:00
audio_size = audio_decode_frame(mFrame, &pts);
if(audio_size < 0)
{
/* If error, we're done */
break;
2012-12-11 21:49:31 +00:00
}
2012-12-14 01:53:22 +00:00
mFrameSize = synchronize_audio(mFrame->data[0], audio_size, pts);
mFramePos = 0;
2012-12-11 21:49:31 +00:00
}
2012-12-14 01:53:22 +00:00
size_t len1 = std::min<size_t>(len - total, mFrameSize-mFramePos);
memcpy(stream, mFrame->data[0]+mFramePos, len1);
total += len1;
2012-12-11 21:49:31 +00:00
stream += len1;
2012-12-14 01:53:22 +00:00
mFramePos += len1;
2012-12-11 21:49:31 +00:00
}
return total;
2012-12-11 21:49:31 +00:00
}
size_t getSampleOffset()
{
return (size_t)(is->audio_clock*is->audio_st->codec->sample_rate);
}
};
2012-12-14 02:24:57 +00:00
void timer_callback(boost::system_time t, VideoState* is)
2012-09-25 00:35:50 +00:00
{
2012-12-13 11:37:04 +00:00
boost::this_thread::sleep(t);
2012-12-14 02:24:57 +00:00
is->refresh = true;
2012-09-25 00:35:50 +00:00
}
2012-12-11 21:49:31 +00:00
/* schedule a video refresh in 'delay' ms */
static void schedule_refresh(VideoState *is, int delay)
2012-09-25 00:35:50 +00:00
{
boost::system_time t = boost::get_system_time() + boost::posix_time::milliseconds(delay);
2012-12-13 11:37:04 +00:00
boost::thread(boost::bind(&timer_callback, t, is)).detach();
2012-12-11 21:49:31 +00:00
}
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
void video_display(VideoState *is)
{
VideoPicture *vp;
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
vp = &is->pictq[is->pictq_rindex];
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
if (is->video_st->codec->width != 0 && is->video_st->codec->height != 0)
{
Ogre::TexturePtr texture = Ogre::TextureManager::getSingleton ().getByName("VideoTexture");
if (texture.isNull () || static_cast<int>(texture->getWidth()) != is->video_st->codec->width
|| static_cast<int>(texture->getHeight()) != is->video_st->codec->height)
2012-12-11 21:49:31 +00:00
{
Ogre::TextureManager::getSingleton ().remove ("VideoTexture");
texture = Ogre::TextureManager::getSingleton().createManual(
"VideoTexture",
Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME,
Ogre::TEX_TYPE_2D,
is->video_st->codec->width, is->video_st->codec->height,
0,
Ogre::PF_BYTE_RGBA,
Ogre::TU_DYNAMIC_WRITE_ONLY_DISCARDABLE);
}
Ogre::PixelBox pb(is->video_st->codec->width, is->video_st->codec->height, 1, Ogre::PF_BYTE_RGBA, vp->data);
Ogre::HardwarePixelBufferSharedPtr buffer = texture->getBuffer();
buffer->blitFromMemory(pb);
is->display_ready = 1;
2012-12-11 21:49:31 +00:00
}
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
free(vp->data);
}
2012-09-25 00:35:50 +00:00
void video_refresh_timer(void *userdata)
{
2012-12-11 21:49:31 +00:00
VideoState *is = (VideoState *)userdata;
VideoPicture *vp;
double actual_delay, delay, sync_threshold, ref_clock, diff;
2012-09-25 00:35:50 +00:00
2012-12-14 02:12:17 +00:00
if(!is->video_st)
2012-12-13 11:05:37 +00:00
{
2012-12-14 02:12:17 +00:00
schedule_refresh(is, 100);
return;
}
if(is->pictq_size == 0)
{
2012-12-14 02:24:57 +00:00
is->refresh = true;
2012-12-14 02:12:17 +00:00
return;
}
2012-09-25 00:35:50 +00:00
2012-12-14 02:12:17 +00:00
vp = &is->pictq[is->pictq_rindex];
2012-09-25 00:35:50 +00:00
2012-12-14 02:12:17 +00:00
is->video_current_pts = vp->pts;
is->video_current_pts_time = av_gettime();
2012-12-11 21:49:31 +00:00
2012-12-14 02:12:17 +00:00
delay = vp->pts - is->frame_last_pts; /* the pts from last time */
if(delay <= 0 || delay >= 1.0) {
/* if incorrect delay, use previous one */
delay = is->frame_last_delay;
}
/* save for next time */
is->frame_last_delay = delay;
is->frame_last_pts = vp->pts;
2012-12-11 22:06:06 +00:00
2012-12-14 02:12:17 +00:00
/* update delay to sync to audio if not master source */
if(is->av_sync_type != AV_SYNC_VIDEO_MASTER)
{
ref_clock = get_master_clock(is);
diff = vp->pts - ref_clock;
2012-09-25 00:35:50 +00:00
2012-12-14 02:12:17 +00:00
/* Skip or repeat the frame. Take delay into account
FFPlay still doesn't "know if this is the best guess." */
sync_threshold = (delay > AV_SYNC_THRESHOLD) ? delay : AV_SYNC_THRESHOLD;
if(fabs(diff) < AV_NOSYNC_THRESHOLD)
{
if(diff <= -sync_threshold)
delay = 0;
else if(diff >= sync_threshold)
delay = 2 * delay;
2012-12-11 21:49:31 +00:00
}
2012-12-11 22:06:06 +00:00
}
2012-12-14 02:12:17 +00:00
is->frame_timer += delay;
/* computer the REAL delay */
actual_delay = is->frame_timer - (av_gettime() / 1000000.0);
if(actual_delay < 0.010)
{
/* Really it should skip the picture instead */
actual_delay = 0.010;
}
schedule_refresh(is, (int)(actual_delay * 1000 + 0.5));
/* show the picture! */
video_display(is);
/* update queue for next picture! */
if(++is->pictq_rindex == VIDEO_PICTURE_QUEUE_SIZE)
is->pictq_rindex = 0;
is->pictq_mutex.lock();
is->pictq_size--;
is->pictq_cond.notify_one ();
is->pictq_mutex.unlock ();
2012-12-11 21:49:31 +00:00
}
2012-09-25 00:35:50 +00:00
int queue_picture(VideoState *is, AVFrame *pFrame, double pts)
{
2012-12-11 21:49:31 +00:00
VideoPicture *vp;
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
/* wait until we have a new pic */
2012-09-25 00:35:50 +00:00
{
2012-12-11 21:49:31 +00:00
boost::unique_lock<boost::mutex> lock(is->pictq_mutex);
2012-12-13 11:05:37 +00:00
while(is->pictq_size >= VIDEO_PICTURE_QUEUE_SIZE && !is->quit)
2012-12-11 21:49:31 +00:00
is->pictq_cond.timed_wait(lock, boost::posix_time::milliseconds(1));
2012-09-25 00:35:50 +00:00
}
2012-12-11 21:49:31 +00:00
if(is->quit)
return -1;
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
// windex is set to 0 initially
vp = &is->pictq[is->pictq_windex];
// Convert the image into YUV format that SDL uses
2012-12-13 11:05:37 +00:00
if(is->sws_context == NULL)
{
2012-12-11 21:49:31 +00:00
int w = is->video_st->codec->width;
int h = is->video_st->codec->height;
2012-12-12 04:11:48 +00:00
is->sws_context = sws_getContext(w, h, is->video_st->codec->pix_fmt,
w, h, PIX_FMT_RGBA, SWS_BICUBIC,
NULL, NULL, NULL);
2012-12-11 21:49:31 +00:00
if(is->sws_context == NULL)
throw std::runtime_error("Cannot initialize the conversion context!\n");
}
2012-12-03 15:44:41 +00:00
2012-12-11 21:49:31 +00:00
vp->data =(uint8_t*) malloc(is->video_st->codec->width * is->video_st->codec->height * 4);
2012-12-03 15:44:41 +00:00
2012-12-11 21:49:31 +00:00
sws_scale(is->sws_context, pFrame->data, pFrame->linesize,
2012-12-12 04:11:48 +00:00
0, is->video_st->codec->height, &vp->data, is->rgbaFrame->linesize);
2012-12-03 15:44:41 +00:00
2012-12-11 21:49:31 +00:00
vp->pts = pts;
2012-12-03 15:44:41 +00:00
2012-12-11 21:49:31 +00:00
// now we inform our display thread that we have a pic ready
2012-12-13 11:05:37 +00:00
if(++is->pictq_windex == VIDEO_PICTURE_QUEUE_SIZE)
2012-12-11 21:49:31 +00:00
is->pictq_windex = 0;
is->pictq_mutex.lock();
is->pictq_size++;
is->pictq_mutex.unlock();
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
return 0;
}
2012-12-03 15:44:41 +00:00
double synchronize_video(VideoState *is, AVFrame *src_frame, double pts)
{
2012-12-11 21:49:31 +00:00
double frame_delay;
2012-12-13 11:05:37 +00:00
if(pts != 0)
{
2012-12-11 21:49:31 +00:00
/* if we have pts, set video clock to it */
is->video_clock = pts;
2012-12-13 11:05:37 +00:00
}
else
{
2012-12-11 21:49:31 +00:00
/* if we aren't given a pts, set it to the clock */
pts = is->video_clock;
2012-09-25 00:35:50 +00:00
}
2012-12-11 21:49:31 +00:00
/* update the video clock */
frame_delay = av_q2d(is->video_st->codec->time_base);
/* if we are repeating a frame, adjust clock accordingly */
frame_delay += src_frame->repeat_pict * (frame_delay * 0.5);
is->video_clock += frame_delay;
return pts;
}
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
uint64_t global_video_pkt_pts = AV_NOPTS_VALUE;
/* These are called whenever we allocate a frame
* buffer. We use this to store the global_pts in
* a frame at the time it is allocated.
*/
int our_get_buffer(struct AVCodecContext *c, AVFrame *pic)
{
2012-12-11 21:49:31 +00:00
int ret = avcodec_default_get_buffer(c, pic);
uint64_t *pts = (uint64_t*)av_malloc(sizeof(uint64_t));
*pts = global_video_pkt_pts;
pic->opaque = pts;
return ret;
}
void our_release_buffer(struct AVCodecContext *c, AVFrame *pic)
{
2012-12-11 21:49:31 +00:00
if(pic) av_freep(&pic->opaque);
avcodec_default_release_buffer(c, pic);
}
2012-09-25 00:35:50 +00:00
2012-12-13 14:11:32 +00:00
int video_thread(void *arg)
{
2012-12-11 21:49:31 +00:00
VideoState *is = (VideoState *)arg;
AVPacket pkt1, *packet = &pkt1;
int frameFinished;
2012-12-11 21:49:31 +00:00
AVFrame *pFrame;
double pts;
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
pFrame = avcodec_alloc_frame();
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
is->rgbaFrame = avcodec_alloc_frame();
2012-12-13 11:05:37 +00:00
avpicture_alloc((AVPicture*)is->rgbaFrame, PIX_FMT_RGBA, is->video_st->codec->width, is->video_st->codec->height);
2012-09-25 00:35:50 +00:00
2012-12-13 11:05:37 +00:00
for(;;)
{
2012-12-13 11:37:04 +00:00
if(packet_queue_get(&is->videoq, packet, is, 1) < 0)
2012-12-13 11:05:37 +00:00
{
2012-12-11 21:49:31 +00:00
// means we quit getting packets
break;
}
pts = 0;
// Save global pts to be stored in pFrame
global_video_pkt_pts = packet->pts;
// Decode video frame
2012-12-13 11:05:37 +00:00
if (avcodec_decode_video2(is->video_st->codec, pFrame, &frameFinished, packet) < 0)
throw std::runtime_error("Error decoding video frame");
2012-12-13 11:05:37 +00:00
if((uint64_t)packet->dts == AV_NOPTS_VALUE &&
pFrame->opaque && *(uint64_t*)pFrame->opaque != AV_NOPTS_VALUE)
2012-12-11 21:49:31 +00:00
pts = *(uint64_t *)pFrame->opaque;
else if((uint64_t)packet->dts != AV_NOPTS_VALUE)
pts = packet->dts;
2012-12-13 11:05:37 +00:00
else
2012-12-11 21:49:31 +00:00
pts = 0;
pts *= av_q2d(is->video_st->time_base);
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
// Did we get a video frame?
2012-12-13 11:05:37 +00:00
if(frameFinished)
{
2012-12-11 21:49:31 +00:00
pts = synchronize_video(is, pFrame, pts);
2012-12-13 11:05:37 +00:00
if(queue_picture(is, pFrame, pts) < 0)
2012-12-11 22:06:06 +00:00
break;
2012-12-11 21:49:31 +00:00
}
av_free_packet(packet);
}
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
av_free(pFrame);
2012-09-25 00:54:29 +00:00
2012-12-11 21:49:31 +00:00
avpicture_free((AVPicture *)is->rgbaFrame);
av_free(is->rgbaFrame);
2012-12-11 21:49:31 +00:00
return 0;
2012-09-25 00:35:50 +00:00
}
2012-12-11 21:49:31 +00:00
int stream_component_open(VideoState *is, int stream_index, AVFormatContext *pFormatCtx)
2012-09-25 00:35:50 +00:00
{
MWSound::DecoderPtr decoder;
2012-12-11 21:49:31 +00:00
AVCodecContext *codecCtx;
AVCodec *codec;
2012-09-25 00:35:50 +00:00
2012-12-13 11:05:37 +00:00
if(stream_index < 0 || stream_index >= static_cast<int>(pFormatCtx->nb_streams))
2012-12-11 21:49:31 +00:00
return -1;
// Get a pointer to the codec context for the video stream
codecCtx = pFormatCtx->streams[stream_index]->codec;
codec = avcodec_find_decoder(codecCtx->codec_id);
if(!codec || (avcodec_open2(codecCtx, codec, NULL) < 0))
{
2012-12-11 21:49:31 +00:00
fprintf(stderr, "Unsupported codec!\n");
return -1;
}
switch(codecCtx->codec_type)
{
2012-12-11 21:49:31 +00:00
case AVMEDIA_TYPE_AUDIO:
is->audioStream = stream_index;
is->audio_st = pFormatCtx->streams[stream_index];
/* averaging filter for audio sync */
is->audio_diff_avg_coef = exp(log(0.01 / AUDIO_DIFF_AVG_NB));
is->audio_diff_avg_count = 0;
/* Correct audio only if larger error than this */
is->audio_diff_threshold = 2.0 * 0.1/* 100 ms */;
2012-12-11 21:49:31 +00:00
memset(&is->audio_pkt, 0, sizeof(is->audio_pkt));
packet_queue_init(&is->audioq);
decoder.reset(new MovieAudioDecoder(is));
is->AudioTrack = MWBase::Environment::get().getSoundManager()->playTrack(decoder);
if(!is->AudioTrack)
{
is->audioStream = -1;
avcodec_close(is->audio_st->codec);
is->audio_st = NULL;
return -1;
}
2012-12-11 21:49:31 +00:00
break;
2012-12-11 21:49:31 +00:00
case AVMEDIA_TYPE_VIDEO:
is->videoStream = stream_index;
is->video_st = pFormatCtx->streams[stream_index];
is->frame_timer = (double)av_gettime() / 1000000.0;
is->frame_last_delay = 40e-3;
is->video_current_pts_time = av_gettime();
packet_queue_init(&is->videoq);
2012-12-11 21:49:31 +00:00
codecCtx->get_buffer = our_get_buffer;
codecCtx->release_buffer = our_release_buffer;
is->video_thread = boost::thread(video_thread, is);
2012-12-11 21:49:31 +00:00
break;
2012-12-11 21:49:31 +00:00
default:
break;
}
return 0;
2012-09-25 00:35:50 +00:00
}
int decode_thread(void *arg)
{
2012-12-11 21:49:31 +00:00
VideoState *is = (VideoState *)arg;
try
{
AVFormatContext *pFormatCtx = is->format_ctx;
AVPacket pkt1, *packet = &pkt1;
2012-12-11 21:49:31 +00:00
if(is->videoStream >= 0 /*|| is->audioStream < 0*/)
{
// main decode loop
for(;;)
{
if(is->quit)
break;
if((is->audioStream >= 0 && is->audioq.size > MAX_AUDIOQ_SIZE) ||
is->videoq.size > MAX_VIDEOQ_SIZE)
{
boost::this_thread::sleep(boost::posix_time::milliseconds(10));
continue;
}
if(av_read_frame(pFormatCtx, packet) < 0)
break;
// Is this a packet from the video stream?
if(packet->stream_index == is->videoStream)
packet_queue_put(&is->videoq, packet);
else if(packet->stream_index == is->audioStream)
packet_queue_put(&is->audioq, packet);
else
av_free_packet(packet);
2012-12-11 21:49:31 +00:00
}
/* all done - wait for it */
while(!is->quit)
{
// EOF reached, all packets processed, we can exit now
if(is->audioq.nb_packets == 0 && is->videoq.nb_packets == 0)
break;
boost::this_thread::sleep(boost::posix_time::milliseconds(100));
2012-12-11 21:49:31 +00:00
}
}
2012-09-25 00:35:50 +00:00
is->quit = 1;
}
catch (std::runtime_error& e)
{
std::cerr << "An error occured playing the video: " << e.what () << std::endl;
is->quit = 1;
}
catch (Ogre::Exception& e)
{
std::cerr << "An error occured playing the video: " << e.getFullDescription () << std::endl;
is->quit = 1;
}
2012-12-11 21:49:31 +00:00
return 0;
}
2012-09-25 00:35:50 +00:00
void init_state(VideoState *is, const std::string& resourceName)
{
try
{
int video_index = -1;
int audio_index = -1;
unsigned int i;
is->av_sync_type = DEFAULT_AV_SYNC_TYPE;
is->videoStream = -1;
is->audioStream = -1;
2012-12-14 02:24:57 +00:00
is->refresh = false;
is->quit = 0;
is->stream = Ogre::ResourceGroupManager::getSingleton ().openResource(resourceName);
if(is->stream.isNull())
throw std::runtime_error("Failed to open video resource");
is->format_ctx = avformat_alloc_context();
is->format_ctx->pb = avio_alloc_context(NULL, 0, 0, is, OgreResource_Read, OgreResource_Write, OgreResource_Seek);
if(!is->format_ctx->pb)
{
avformat_free_context(is->format_ctx);
throw std::runtime_error("Failed to allocate ioContext ");
}
// Open video file
/// \todo leak here, ffmpeg or valgrind bug ?
if (avformat_open_input(&is->format_ctx, resourceName.c_str(), NULL, NULL))
{
// "Note that a user-supplied AVFormatContext will be freed on failure."
is->format_ctx = NULL;
throw std::runtime_error("Failed to open video input");
}
// Retrieve stream information
if(avformat_find_stream_info(is->format_ctx, NULL) < 0)
throw std::runtime_error("Failed to retrieve stream information");
// Dump information about file onto standard error
av_dump_format(is->format_ctx, 0, resourceName.c_str(), 0);
for(i = 0;i < is->format_ctx->nb_streams;i++)
{
if(is->format_ctx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO && video_index < 0)
video_index = i;
if(is->format_ctx->streams[i]->codec->codec_type == AVMEDIA_TYPE_AUDIO && audio_index < 0)
audio_index = i;
}
if(audio_index >= 0)
stream_component_open(is, audio_index, is->format_ctx);
if(video_index >= 0)
stream_component_open(is, video_index, is->format_ctx);
}
catch (std::runtime_error& e)
{
is->quit = 1;
throw;
}
catch (Ogre::Exception& e)
{
is->quit = 1;
throw;
}
}
void deinit_state(VideoState *is)
{
is->audioq.cond.notify_one ();
is->videoq.cond.notify_one ();
is->parse_thread.join();
is->video_thread.join();
if(is->audioStream >= 0)
avcodec_close(is->audio_st->codec);
if(is->videoStream >= 0)
avcodec_close(is->video_st->codec);
if (is->sws_context)
sws_freeContext(is->sws_context);
if (is->format_ctx)
{
AVIOContext *ioContext = is->format_ctx->pb;
avformat_close_input(&is->format_ctx);
av_free(ioContext);
}
}
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
VideoPlayer::VideoPlayer(Ogre::SceneManager* sceneMgr)
: mState(NULL)
, mSceneMgr(sceneMgr)
{
mVideoMaterial = Ogre::MaterialManager::getSingleton ().create("VideoMaterial", "General");
mVideoMaterial->getTechnique(0)->getPass(0)->setDepthWriteEnabled(false);
mVideoMaterial->getTechnique(0)->getPass(0)->setDepthCheckEnabled(false);
mVideoMaterial->getTechnique(0)->getPass(0)->setLightingEnabled(false);
mVideoMaterial->getTechnique(0)->getPass(0)->createTextureUnitState();
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
mRectangle = new Ogre::Rectangle2D(true);
mRectangle->setCorners(-1.0, 1.0, 1.0, -1.0);
mRectangle->setMaterial("VideoMaterial");
mRectangle->setRenderQueueGroup(Ogre::RENDER_QUEUE_OVERLAY+1);
// Use infinite AAB to always stay visible
Ogre::AxisAlignedBox aabInf;
aabInf.setInfinite();
mRectangle->setBoundingBox(aabInf);
// Attach background to the scene
Ogre::SceneNode* node = sceneMgr->getRootSceneNode()->createChildSceneNode();
node->attachObject(mRectangle);
mRectangle->setVisible(false);
mRectangle->setVisibilityFlags (0x1);
2012-09-25 00:35:50 +00:00
}
2012-12-11 21:49:31 +00:00
VideoPlayer::~VideoPlayer ()
2012-09-25 00:35:50 +00:00
{
2012-12-11 21:49:31 +00:00
if (mState)
close();
2012-09-25 00:35:50 +00:00
}
2012-12-11 21:49:31 +00:00
void VideoPlayer::playVideo (const std::string &resourceName)
2012-09-25 00:35:50 +00:00
{
2012-12-13 11:13:44 +00:00
// Register all formats and codecs
av_register_all();
2012-12-11 21:49:31 +00:00
if (mState)
close();
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
mRectangle->setVisible(true);
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
MWBase::Environment::get().getWindowManager ()->pushGuiMode (MWGui::GM_Video);
2012-09-25 00:35:50 +00:00
// Turn off rendering except the GUI
mSceneMgr->clearSpecialCaseRenderQueues();
// SCRQM_INCLUDE with RENDER_QUEUE_OVERLAY does not work.
for (int i = 0; i < Ogre::RENDER_QUEUE_MAX; ++i)
{
if (i > 0 && i < 96)
mSceneMgr->addSpecialCaseRenderQueue(i);
}
mSceneMgr->setSpecialCaseRenderQueueMode(Ogre::SceneManager::SCRQM_EXCLUDE);
MWBase::Environment::get().getSoundManager()->pauseAllSounds();
2012-12-11 21:49:31 +00:00
mState = new VideoState;
2012-09-25 00:35:50 +00:00
2012-12-13 11:13:44 +00:00
init_state(mState, resourceName);
2012-12-11 21:49:31 +00:00
schedule_refresh(mState, 40);
mState->parse_thread = boost::thread(decode_thread, mState);
}
2012-12-11 21:49:31 +00:00
void VideoPlayer::update ()
{
if(mState)
{
if(mState->quit)
close();
else if(mState->refresh)
{
2012-12-14 02:24:57 +00:00
mState->refresh = false;
video_refresh_timer(mState);
}
2012-12-11 21:49:31 +00:00
}
if (mState && mState->display_ready && !Ogre::TextureManager::getSingleton ().getByName ("VideoTexture").isNull ())
2012-12-11 22:06:06 +00:00
mVideoMaterial->getTechnique(0)->getPass(0)->getTextureUnitState (0)->setTextureName ("VideoTexture");
else
mVideoMaterial->getTechnique(0)->getPass(0)->getTextureUnitState (0)->setTextureName ("black.png");
2012-12-11 21:49:31 +00:00
}
void VideoPlayer::close()
{
mState->quit = 1;
deinit_state(mState);
2012-12-11 21:49:31 +00:00
delete mState;
mState = NULL;
MWBase::Environment::get().getSoundManager()->resumeAllSounds();
2012-12-11 21:49:31 +00:00
mRectangle->setVisible (false);
MWBase::Environment::get().getWindowManager ()->removeGuiMode (MWGui::GM_Video);
mSceneMgr->clearSpecialCaseRenderQueues();
mSceneMgr->setSpecialCaseRenderQueueMode(Ogre::SceneManager::SCRQM_EXCLUDE);
}
2012-09-25 00:35:50 +00:00
2012-12-11 21:49:31 +00:00
bool VideoPlayer::isPlaying ()
{
return mState != NULL;
}
}