89 lines
2.7 KiB
C++
89 lines
2.7 KiB
C++
// Copyright 2019 Citra Emulator Project
|
|
// Licensed under GPLv2 or any later version
|
|
// Refer to the license.txt file included.
|
|
#pragma once
|
|
|
|
// AAC decoder related APIs are only available with WIN7+
|
|
#define WINVER _WIN32_WINNT_WIN7
|
|
|
|
#include <optional>
|
|
#include <string>
|
|
#include <tuple>
|
|
#include <vector>
|
|
#include <comdef.h>
|
|
#include <mfapi.h>
|
|
#include <mferror.h>
|
|
#include <mfidl.h>
|
|
#include <mftransform.h>
|
|
|
|
#include "adts.h"
|
|
|
|
enum class MFOutputState { FatalError, OK, NeedMoreInput, NeedReconfig, HaveMoreData };
|
|
enum class MFInputState { FatalError, OK, NotAccepted };
|
|
|
|
// utility functions / templates
|
|
template <class T>
|
|
struct MFRelease {
|
|
void operator()(T* pointer) const {
|
|
pointer->Release();
|
|
};
|
|
};
|
|
|
|
template <>
|
|
struct MFRelease<IMFTransform> {
|
|
void operator()(IMFTransform* pointer) const {
|
|
MFShutdownObject(pointer);
|
|
pointer->Release();
|
|
};
|
|
};
|
|
|
|
// wrapper facilities for dealing with pointers
|
|
template <typename T>
|
|
using unique_mfptr = std::unique_ptr<T, MFRelease<T>>;
|
|
|
|
template <typename SmartPtr, typename RawPtr>
|
|
class AmpImpl {
|
|
public:
|
|
AmpImpl(SmartPtr& smart_ptr) : smart_ptr(smart_ptr) {}
|
|
~AmpImpl() {
|
|
smart_ptr.reset(raw_ptr);
|
|
}
|
|
|
|
operator RawPtr*() {
|
|
return &raw_ptr;
|
|
}
|
|
|
|
private:
|
|
SmartPtr& smart_ptr;
|
|
RawPtr raw_ptr = nullptr;
|
|
};
|
|
|
|
template <typename SmartPtr>
|
|
auto Amp(SmartPtr& smart_ptr) {
|
|
return AmpImpl<SmartPtr, decltype(smart_ptr.get())>(smart_ptr);
|
|
}
|
|
|
|
// convient function for formatting error messages
|
|
void ReportError(std::string msg, HRESULT hr);
|
|
|
|
// data type for transferring ADTS metadata between functions
|
|
struct ADTSMeta {
|
|
ADTSData ADTSHeader;
|
|
u8 AACTag[14];
|
|
};
|
|
|
|
// exported functions
|
|
unique_mfptr<IMFTransform> MFDecoderInit(GUID audio_format = MFAudioFormat_AAC);
|
|
unique_mfptr<IMFSample> CreateSample(const void* data, DWORD len, DWORD alignment = 1,
|
|
LONGLONG duration = 0);
|
|
bool SelectInputMediaType(IMFTransform* transform, int in_stream_id, const ADTSData& adts,
|
|
const UINT8* user_data, UINT32 user_data_len,
|
|
GUID audio_format = MFAudioFormat_AAC);
|
|
std::optional<ADTSMeta> DetectMediaType(char* buffer, std::size_t len);
|
|
bool SelectOutputMediaType(IMFTransform* transform, int out_stream_id,
|
|
GUID audio_format = MFAudioFormat_PCM);
|
|
void MFFlush(IMFTransform* transform);
|
|
MFInputState SendSample(IMFTransform* transform, DWORD in_stream_id, IMFSample* in_sample);
|
|
std::tuple<MFOutputState, unique_mfptr<IMFSample>> ReceiveSample(IMFTransform* transform,
|
|
DWORD out_stream_id);
|
|
std::optional<std::vector<f32>> CopySampleToBuffer(IMFSample* sample);
|