137 lines
4.7 KiB
C++
137 lines
4.7 KiB
C++
/*
|
|
* Copyright (c) 2009-2011 Intel Corporation. All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
|
|
#ifndef OMX_VIDEO_ENCODER_AVC_H_
|
|
#define OMX_VIDEO_ENCODER_AVC_H_
|
|
|
|
|
|
#include "OMXVideoEncoderBase.h"
|
|
#include <utils/List.h>
|
|
#include <IntelMetadataBuffer.h>
|
|
#include <OMX_VideoExt.h>
|
|
enum {
|
|
F_UNKNOWN = 0x00, // Unknown
|
|
F_I = 0x01, // General I-frame type
|
|
F_P = 0x02, // General P-frame type
|
|
F_B = 0x03, // General B-frame type
|
|
F_SI = 0x04, // H.263 SI-frame type
|
|
F_SP = 0x05, // H.263 SP-frame type
|
|
F_EI = 0x06, // H.264 EI-frame type
|
|
F_EP = 0x07, // H.264 EP-frame type
|
|
F_S = 0x08, // MPEG-4 S-frame type
|
|
F_IDR = 0x09, // IDR-frame type
|
|
};
|
|
|
|
enum {
|
|
CACHE_NONE = 0x00, //nothing to be done
|
|
CACHE_PUSH = 0x01, //push this frame into cache
|
|
CACHE_POP = 0x02, //pop all from cache into queue head by STACK rule
|
|
CACHE_RESET = 0x03, //reset cache, clear all cached frames
|
|
};
|
|
|
|
#define ENC_NSTOP 0x02000000
|
|
|
|
#define GET_FT(x) ( (x & 0xF0000000 ) >> 28 ) //get frame type
|
|
#define GET_CO(x) ( (x & 0x0C000000 ) >> 26 ) //get cache operation
|
|
#define GET_FC(x) ( (x & 0x01FFFFFF ) ) //get frame count
|
|
|
|
#define SET_FT(x, y) { x = ((x & ~0xF0000000) | (y << 28)); }
|
|
#define SET_CO(x, y) { x = ((x & ~0x0C000000) | (y << 26 )); }
|
|
#define SET_FC(x, y) { x = ((x & ~0x01FFFFFF) | (y & 0x01FFFFFF )); }
|
|
|
|
const char* FrameTypeStr[10] = {"UNKNOWN", "I", "P", "B", "SI", "SP", "EI", "EP", "S", "IDR"};
|
|
const char* CacheOperationStr[4]= {"NONE", "PUSH", "POP", "RESET"};
|
|
|
|
typedef struct {
|
|
uint32_t FrameType;
|
|
uint32_t CacheOperation;
|
|
bool NotStopFrame;
|
|
uint32_t FrameCount;
|
|
}Encode_Info;
|
|
|
|
#define MAX_H264_PROFILE 3
|
|
|
|
class OMXVideoEncoderAVC : public OMXVideoEncoderBase {
|
|
public:
|
|
OMXVideoEncoderAVC();
|
|
virtual ~OMXVideoEncoderAVC();
|
|
|
|
protected:
|
|
virtual OMX_ERRORTYPE InitOutputPortFormatSpecific(OMX_PARAM_PORTDEFINITIONTYPE *paramPortDefinitionOutput);
|
|
virtual OMX_ERRORTYPE ProcessorInit(void);
|
|
virtual OMX_ERRORTYPE ProcessorDeinit(void);
|
|
virtual OMX_ERRORTYPE ProcessorStop(void);
|
|
virtual OMX_ERRORTYPE ProcessorProcess(
|
|
OMX_BUFFERHEADERTYPE **buffers,
|
|
buffer_retain_t *retains,
|
|
OMX_U32 numberBuffers);
|
|
virtual OMX_ERRORTYPE ProcessorPreEmptyBuffer(OMX_BUFFERHEADERTYPE* buffer);
|
|
|
|
virtual OMX_ERRORTYPE BuildHandlerList(void);
|
|
virtual OMX_ERRORTYPE SetVideoEncoderParam();
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ParamVideoAvc);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ParamNalStreamFormat);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ParamNalStreamFormatSupported);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ParamNalStreamFormatSelect);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ConfigVideoAVCIntraPeriod);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ConfigVideoNalSize);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ParamIntelAVCVUI);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ParamVideoBytestream);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ConfigIntelSliceNumbers);
|
|
DECLARE_HANDLER(OMXVideoEncoderAVC, ParamVideoProfileLevelQuerySupported);
|
|
|
|
|
|
private:
|
|
enum {
|
|
// OMX_PARAM_PORTDEFINITIONTYPE
|
|
OUTPORT_MIN_BUFFER_COUNT = 1,
|
|
OUTPORT_ACTUAL_BUFFER_COUNT = 6,
|
|
OUTPORT_BUFFER_SIZE = 2000000,
|
|
NUM_REFERENCE_FRAME = 4,
|
|
};
|
|
|
|
OMX_VIDEO_PARAM_AVCTYPE mParamAvc;
|
|
OMX_NALSTREAMFORMATTYPE mNalStreamFormat;
|
|
OMX_VIDEO_CONFIG_AVCINTRAPERIOD mConfigAvcIntraPeriod;
|
|
OMX_VIDEO_CONFIG_NALSIZE mConfigNalSize;
|
|
OMX_VIDEO_PARAM_INTEL_AVCVUI mParamIntelAvcVui;
|
|
OMX_VIDEO_CONFIG_INTEL_SLICE_NUMBERS mConfigIntelSliceNumbers;
|
|
VideoParamsAVC *mAVCParams;
|
|
|
|
OMX_U32 mInputPictureCount;
|
|
OMX_U32 mFrameEncodedCount;
|
|
|
|
List<OMX_BUFFERHEADERTYPE*> mBFrameList;
|
|
|
|
OMX_BOOL ProcessCacheOperation(OMX_BUFFERHEADERTYPE **buffers);
|
|
OMX_ERRORTYPE ProcessDataRetrieve(
|
|
OMX_BUFFERHEADERTYPE **buffers,
|
|
OMX_BOOL *outBufReturned);
|
|
|
|
struct ProfileLevelTable {
|
|
OMX_U32 profile;
|
|
OMX_U32 level;
|
|
};
|
|
|
|
ProfileLevelTable mPLTable[MAX_H264_PROFILE];
|
|
OMX_U32 mPLTableCount;
|
|
|
|
OMX_BOOL mEmptyEOSBuf;
|
|
OMX_BOOL mCSDOutputted;
|
|
};
|
|
|
|
#endif /* OMX_VIDEO_ENCODER_AVC_H_ */
|