jfmcarreira
7ccf255a94
Improve InputStream and LibAvHandle to detect the stream duration in order to present it on the side bar Improvements in the LibAvHandle to return NOFMT if format is unknown
100 lines
2.3 KiB
C++
Executable File
100 lines
2.3 KiB
C++
Executable File
/* This file is a part of plaYUVer project
|
|
* Copyright (C) 2014 by Luis Lucas (luisfrlucas@gmail.com)
|
|
* Joao Carreira (jfmcarreira@gmail.com)
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
/**
|
|
* \file LibAvContextHandle.h
|
|
* \brief Interface with libav libs
|
|
*/
|
|
|
|
#ifndef __LIBAVCONTEXTHANDLE_H__
|
|
#define __LIBAVCONTEXTHANDLE_H__
|
|
|
|
#include <inttypes.h>
|
|
|
|
#ifndef __PRI64_PREFIX
|
|
#define __PRI64_PREFIX "l"
|
|
#endif
|
|
|
|
#ifndef PRId64
|
|
#define PRId64 __PRI64_PREFIX "d"
|
|
#endif
|
|
|
|
extern "C"
|
|
{
|
|
#include <libavutil/avutil.h>
|
|
#include <libavutil/imgutils.h>
|
|
#include <libavutil/samplefmt.h>
|
|
#include <libavutil/timestamp.h>
|
|
#include <libavformat/avformat.h>
|
|
}
|
|
|
|
#include "TypeDef.h"
|
|
|
|
namespace plaYUVer
|
|
{
|
|
|
|
class LibAvContextHandle
|
|
{
|
|
public:
|
|
LibAvContextHandle() :
|
|
m_bHasStream( false )
|
|
{
|
|
}
|
|
Bool initAvFormat( char* filename, UInt& width, UInt& height, Int& pixel_format, UInt& frame_rate, UInt64& num_frames );
|
|
Void closeAvFormat();
|
|
Bool decodeAvFormat();
|
|
|
|
Void seekAvFormat( UInt64 frame_num );
|
|
|
|
Bool getStatus()
|
|
{
|
|
return m_bHasStream;
|
|
}
|
|
Char* getCodecName()
|
|
{
|
|
return m_acCodecName;
|
|
}
|
|
UInt getStreamDuration( )
|
|
{
|
|
return m_uiSecs;
|
|
}
|
|
|
|
uint8_t *video_dst_data[4];
|
|
int video_dst_linesize[4];
|
|
int video_dst_bufsize;
|
|
|
|
private:
|
|
AVFormatContext *fmt_ctx;
|
|
AVCodecContext *video_dec_ctx;
|
|
AVStream *video_stream;
|
|
Int video_stream_idx;
|
|
AVFrame *frame;
|
|
AVPacket pkt;
|
|
|
|
Bool m_bHasStream;
|
|
|
|
Char m_acCodecName[20];
|
|
UInt m_uiSecs;
|
|
UInt m_uiMicroSec;
|
|
};
|
|
|
|
} // NAMESPACE
|
|
|
|
#endif // __LIBAVCONTEXTHANDLE_H__
|