EtienneLescot
ef5855f1f4
Record browser webcam sidecar when native Windows capture is active. Add native webcam sidecar output and DirectShow NV12/YUY2 fallback. Sample exported webcam frames by source timestamp.
62 lines
1.6 KiB
C++
62 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#include "dshow_webcam_capture.h"
|
|
|
|
#include <Windows.h>
|
|
#include <mfidl.h>
|
|
#include <mfreadwrite.h>
|
|
#include <wrl/client.h>
|
|
|
|
#include <atomic>
|
|
#include <cstdint>
|
|
#include <mutex>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <vector>
|
|
|
|
class WebcamCapture {
|
|
public:
|
|
WebcamCapture() = default;
|
|
~WebcamCapture();
|
|
|
|
WebcamCapture(const WebcamCapture&) = delete;
|
|
WebcamCapture& operator=(const WebcamCapture&) = delete;
|
|
|
|
bool initialize(
|
|
const std::wstring& deviceId,
|
|
const std::wstring& deviceName,
|
|
const std::wstring& directShowClsid,
|
|
int requestedWidth,
|
|
int requestedHeight,
|
|
int requestedFps);
|
|
bool start();
|
|
void stop();
|
|
bool copyLatestFrame(WebcamFrameSnapshot& destination);
|
|
|
|
int width() const;
|
|
int height() const;
|
|
int fps() const;
|
|
const std::wstring& selectedDeviceName() const;
|
|
|
|
private:
|
|
bool selectDevice(const std::wstring& deviceId, const std::wstring& deviceName);
|
|
bool configureReader(int requestedWidth, int requestedHeight, int requestedFps);
|
|
void captureLoop();
|
|
|
|
Microsoft::WRL::ComPtr<IMFMediaSource> mediaSource_;
|
|
Microsoft::WRL::ComPtr<IMFSourceReader> sourceReader_;
|
|
DirectShowWebcamCapture directShowCapture_;
|
|
std::thread thread_;
|
|
std::atomic<bool> stopRequested_ = false;
|
|
std::mutex frameMutex_;
|
|
std::vector<BYTE> latestFrame_;
|
|
uint64_t latestFrameSequence_ = 0;
|
|
int width_ = 0;
|
|
int height_ = 0;
|
|
int fps_ = 30;
|
|
bool mfStarted_ = false;
|
|
bool usingDirectShow_ = false;
|
|
int selectedMatchScore_ = 0;
|
|
std::wstring selectedDeviceName_;
|
|
};
|