gecko-dev/media/gmp-clearkey/0.1/WMFUtils.h

244 строки
5.7 KiB
C++

/*
* Copyright 2013, Mozilla Foundation and contributors
*
* 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 __WMFUtils_h__
#define __WMFUtils_h__
#include <cstdint>
#include <string>
#include <assert.h>
#include <mfapi.h>
#include <mferror.h>
#include <mfobjects.h>
#include <mftransform.h>
#include <wmcodecdsp.h>
#include "mozilla/Attributes.h"
#include "VideoLimits.h"
#include "gmp-platform.h"
void LOG(const char* format, ...);
#ifdef LOG_SAMPLE_DECODE
# define SAMPLE_LOG LOG
#else
# define SAMPLE_LOG(...)
#endif
#ifndef CLSID_CMSAACDecMFT
# define WMF_MUST_DEFINE_AAC_MFT_CLSID
extern "C" const CLSID CLSID_CMSAACDecMFT;
#endif
extern "C" const CLSID CLSID_CMSH264DecMFT;
namespace wmf {
// Reimplementation of CComPtr to reduce dependence on system
// shared libraries.
template <class T>
class CComPtr {
public:
CComPtr(CComPtr&& aOther) : mPtr(aOther.Detach()) {}
CComPtr& operator=(CComPtr&& other) { mPtr = other.Detach(); }
CComPtr(const CComPtr& aOther) : mPtr(nullptr) { Set(aOther.Get()); }
CComPtr() : mPtr(nullptr) {}
MOZ_IMPLICIT CComPtr(T* const& aPtr) : mPtr(nullptr) { Set(aPtr); }
MOZ_IMPLICIT CComPtr(const std::nullptr_t& aNullPtr) : mPtr(aNullPtr) {}
T** operator&() { return &mPtr; }
T* operator->() { return mPtr; }
operator T*() { return mPtr; }
T* operator=(T* const& aPtr) { return Set(aPtr); }
T* operator=(const std::nullptr_t& aPtr) { return mPtr = aPtr; }
T* Get() const { return mPtr; }
T* Detach() {
T* tmp = mPtr;
mPtr = nullptr;
return tmp;
}
~CComPtr() {
if (mPtr) {
mPtr->Release();
}
mPtr = nullptr;
}
private:
T* Set(T* aPtr) {
if (mPtr == aPtr) {
return aPtr;
}
if (mPtr) {
mPtr->Release();
}
mPtr = aPtr;
if (mPtr) {
mPtr->AddRef();
}
return mPtr;
}
T* mPtr;
};
class IntRect {
public:
IntRect(int32_t _x, int32_t _y, int32_t _w, int32_t _h)
: x(_x), y(_y), width(_w), height(_h) {}
IntRect() : x(0), y(0), width(0), height(0) {}
int32_t x;
int32_t y;
int32_t width;
int32_t height;
};
typedef int64_t Microseconds;
#ifdef ENSURE
# undef ENSURE
#endif
#define ENSURE(condition, ret) \
{ \
if (!(condition)) { \
LOG("##condition## FAILED %S:%d\n", __FILE__, __LINE__); \
return ret; \
} \
}
#define STATUS_SUCCEEDED(x) ((x) == Status::kSuccess)
#define STATUS_FAILED(x) ((x) != Status::kSuccess)
#define MFPLAT_FUNC(_func, _dllname) extern decltype(::_func)* _func;
#include "WMFSymbols.h"
#undef MFPLAT_FUNC
bool EnsureLibs();
HRESULT
GetPictureRegion(IMFMediaType* aMediaType, IntRect& aOutPictureRegion);
HRESULT
GetDefaultStride(IMFMediaType* aType, uint32_t* aOutStride);
// Converts from microseconds to hundreds of nanoseconds.
// We use microseconds for our timestamps, whereas WMF uses
// hundreds of nanoseconds.
inline int64_t UsecsToHNs(int64_t aUsecs) { return aUsecs * 10; }
// Converts from hundreds of nanoseconds to microseconds.
// We use microseconds for our timestamps, whereas WMF uses
// hundreds of nanoseconds.
inline int64_t HNsToUsecs(int64_t hNanoSecs) { return hNanoSecs / 10; }
inline std::string narrow(std::wstring& wide) {
std::string ns(wide.begin(), wide.end());
return ns;
}
inline std::wstring widen(std::string& narrow) {
std::wstring ws(narrow.begin(), narrow.end());
return ws;
}
#define ARRAY_LENGTH(array_) (sizeof(array_) / sizeof(array_[0]))
template <class Type>
class AutoPtr {
public:
AutoPtr() : mPtr(nullptr) {}
AutoPtr(AutoPtr<Type>& aPtr) : mPtr(aPtr.Forget()) {}
explicit AutoPtr(Type* aPtr) : mPtr(aPtr) {}
~AutoPtr() {
if (mPtr) {
delete mPtr;
}
}
Type* Forget() {
Type* rv = mPtr;
mPtr = nullptr;
return rv;
}
AutoPtr<Type>& operator=(Type* aOther) {
Assign(aOther);
return *this;
}
AutoPtr<Type>& operator=(AutoPtr<Type>& aOther) {
Assign(aOther.Forget());
return *this;
}
Type* Get() const { return mPtr; }
Type* operator->() const {
assert(mPtr);
return Get();
}
operator Type*() const { return Get(); }
Type** Receive() { return &mPtr; }
private:
void Assign(Type* aPtr) {
if (mPtr) {
delete mPtr;
}
mPtr = aPtr;
}
Type* mPtr;
};
// Video frame microseconds are (currently) in 90kHz units, as used by RTP.
// Use this to convert to microseconds...
inline Microseconds RTPTimeToMicroseconds(int64_t rtptime) {
return (rtptime * 1000000) / 90000;
}
inline uint32_t MicrosecondsToRTPTime(Microseconds us) {
return uint32_t(0xffffffff & (us * 90000) / 1000000);
}
void dump(const uint8_t* data, uint32_t len, const char* filename);
HRESULT
CreateMFT(const CLSID& clsid, const char* aDllName,
CComPtr<IMFTransform>& aOutMFT);
// Returns the name of the DLL that is needed to decode H.264 on
// the given windows version we're running on.
const char* WMFDecoderDllName();
// Returns the maximum number of threads we want WMF to use for decoding
// given the number of logical processors available.
int32_t GetNumThreads(int32_t aCoreCount);
} // namespace wmf
#endif // __WMFUtils_h__