Xaymar
aeaa8dd90e
The encoder is now capable of delayed or immediate threading (using fixed latency) which allows for a much better performance and works better with older VFW encoders that assume they are being used for transcoding instead of live-encoding. In addition to the above, the MPEG-2 rewrite code for Matrox MPEG-2 encoders is completely done now and will fix up the bitstream Matroxs encoder sends out to be mostly correct, allowing for better seeking. The last change this encompasses is the removal of the need to flip a RGBA frame to BGRA. This doesn't break any support and should work better with what we have. This exposed several bugs in OBS Studio: - Color Range, Space & Format are ignored by OBS Studio if set by the encoder. - The ffmpeg-mux process freezes when writing the header for a correct MPEG-2 stream that doesn't match what OBS thinks it should be. ffmpeg itself has no issues writing this header, so this is an OBS Studio issue. And exposed the following bugs in Matrox VFW: - The MPEG-2 bitstream is written as interlaced Top-Field-First instead of Progressive - the content itself is Progressive. - The encoder ignores FastCompress mode and just behaves identically in either mode. - Some parameters that should be set are not being set by Matrox (extended Framerate for example).
125 lines
3.3 KiB
C++
125 lines
3.3 KiB
C++
#pragma once
|
|
#include "plugin.h"
|
|
#include "libobs/obs-encoder.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <fstream>
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <condition_variable>
|
|
#include <queue>
|
|
|
|
// VFW
|
|
#define COMPMAN
|
|
#define VIDEO
|
|
#define MMREG
|
|
#include <windows.h>
|
|
extern "C" {
|
|
#include <Vfw.h>
|
|
#include <vfwext.h>
|
|
#include <vfwmsgs.h>
|
|
};
|
|
|
|
namespace VFW {
|
|
struct Info {
|
|
std::string Id;
|
|
std::string Name;
|
|
std::string Path;
|
|
ICINFO icInfo;
|
|
ICINFO icInfo2;
|
|
size_t index;
|
|
obs_encoder_info obsInfo;
|
|
|
|
std::string FourCC, FourCC2;
|
|
std::vector<uint8_t> stateInfo;
|
|
|
|
int32_t defaultQuality;
|
|
int32_t defaultKeyframeRate;
|
|
bool hasConfigure, hasAbout;
|
|
};
|
|
bool Initialize();
|
|
bool Finalize();
|
|
|
|
class Encoder {
|
|
public:
|
|
|
|
static const char* get_name(void* type_data);
|
|
static void get_defaults(obs_data_t *settings);
|
|
static obs_properties_t* get_properties(void *data);
|
|
static bool cb_configure(obs_properties_t *pr, obs_property_t *p, void *data);
|
|
static bool cb_about(obs_properties_t *pr, obs_property_t *p, void *data);
|
|
static bool cb_modified(obs_properties_t *pr, obs_property_t *p, obs_data_t *data);
|
|
|
|
static void* create(obs_data_t *settings, obs_encoder_t *encoder);
|
|
Encoder(obs_data_t *settings, obs_encoder_t *encoder);
|
|
|
|
static void destroy(void* data);
|
|
~Encoder();
|
|
|
|
static bool encode(void *data, struct encoder_frame *frame, struct encoder_packet *packet, bool *received_packet);
|
|
bool encode(struct encoder_frame *frame, struct encoder_packet *packet, bool *received_packet);
|
|
|
|
static bool update(void *data, obs_data_t *settings);
|
|
bool update(obs_data_t* settings);
|
|
|
|
static bool get_extra_data(void *data, uint8_t **extra_data, size_t *size);
|
|
bool get_extra_data(uint8_t** extra_data, size_t* size);
|
|
|
|
static bool get_sei_data(void *data, uint8_t **sei_data, size_t *size);
|
|
bool get_sei_data(uint8_t** sei_data, size_t* size);
|
|
|
|
static void get_video_info(void *data, struct video_scale_info *info);
|
|
void get_video_info(struct video_scale_info *info);
|
|
|
|
static void threadMain(void *data, int32_t flag);
|
|
void threadLocal(int32_t flag);
|
|
void preProcessLocal(std::unique_lock<std::mutex>& ul);
|
|
void encodeLocal(std::unique_lock<std::mutex>& ul);
|
|
void postProcessLocal(std::unique_lock<std::mutex>& ul);
|
|
|
|
private:
|
|
VFW::Info* myInfo;
|
|
HIC hIC;
|
|
COMPVARS cv;
|
|
std::vector<char>
|
|
m_bufferInput,
|
|
m_bufferInputBitmapInfo,
|
|
m_bufferPrevInput,
|
|
m_bufferPrevInputBitmapInfo,
|
|
m_bufferOutput,
|
|
m_bufferOutputBitmapInfo;
|
|
BITMAPINFO
|
|
*m_inputBitmapInfo,
|
|
*m_prevInputBitmapInfo,
|
|
*m_outputBitmapInfo;
|
|
|
|
uint32_t
|
|
m_width, m_height,
|
|
m_fpsNum, m_fpsDen,
|
|
m_keyframeInterval,
|
|
m_bitrate, m_quality,
|
|
m_latency, m_maxQueueSize;
|
|
bool
|
|
m_useNormalCompress,
|
|
m_useTemporalFlag,
|
|
m_useBitrateFlag,
|
|
m_useQualityFlag,
|
|
m_forceKeyframes;
|
|
|
|
struct thread_data {
|
|
std::thread worker;
|
|
std::mutex lock;
|
|
std::condition_variable cv;
|
|
// Data Vector, Frame, Keyframe
|
|
std::queue<std::tuple<std::shared_ptr<std::vector<char>>, int64_t, bool>> data;
|
|
} m_preProcessData,
|
|
m_encodeData,
|
|
m_postProcessData;
|
|
std::mutex m_finalPacketsLock;
|
|
std::queue<std::tuple<std::shared_ptr<std::vector<char>>, int64_t, bool>> m_finalPackets;
|
|
bool m_threadShutdown;
|
|
std::shared_ptr<std::vector<char>> m_donotuse_datastor;
|
|
};
|
|
};
|