Format uvc-wmf.cc
This commit is contained in:
John Zhao
parent
58fdc4c01b
commit
49771d5b8e
|
|
@ -1,7 +1,5 @@
|
|||
#include "uvc/uvc.h" // NOLINT
|
||||
|
||||
#include <glog/logging.h>
|
||||
|
||||
#include <chrono>
|
||||
#include <iostream>
|
||||
|
||||
|
|
@ -9,8 +7,6 @@
|
|||
#include <usbioctl.h>
|
||||
#include <sstream>
|
||||
|
||||
#include "uvc.h"
|
||||
|
||||
#include <Shlwapi.h> // For QISearch, etc.
|
||||
#include <mfapi.h> // For MFStartup, etc.
|
||||
#include <mfidl.h> // For MF_DEVSOURCE_*, etc.
|
||||
|
|
@ -47,6 +43,11 @@
|
|||
|
||||
#include <strsafe.h>
|
||||
|
||||
#include <glog/logging.h>
|
||||
|
||||
// #define VLOG_INFO VLOG(2)
|
||||
#define VLOG_INFO LOG(INFO)
|
||||
|
||||
MYNTEYE_BEGIN_NAMESPACE
|
||||
|
||||
namespace uvc {
|
||||
|
|
@ -81,7 +82,7 @@ static void check(const char *call, HRESULT hr) {
|
|||
throw_error() << call << "(...) returned 0x" << std::hex
|
||||
<< static_cast<uint32_t>(hr);
|
||||
} else {
|
||||
LOG(INFO) << call << " SUCCESSED";
|
||||
VLOG_INFO << call << " SUCCESSED";
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -130,8 +131,7 @@ public:
|
|||
}
|
||||
};
|
||||
|
||||
static std::string win_to_utf(const WCHAR * s)
|
||||
{
|
||||
static std::string win_to_utf(const WCHAR *s) {
|
||||
int len = WideCharToMultiByte(CP_UTF8, 0, s, -1, nullptr, 0, NULL, NULL);
|
||||
if (len == 0) throw_error() << "WideCharToMultiByte(...) returned 0 and GetLastError() is " << GetLastError();
|
||||
std::string buffer(len - 1, ' ');
|
||||
|
|
@ -140,8 +140,7 @@ static std::string win_to_utf(const WCHAR * s)
|
|||
return buffer;
|
||||
}
|
||||
|
||||
std::vector<std::string> tokenize(std::string string, char separator)
|
||||
{
|
||||
std::vector<std::string> tokenize(std::string string, char separator) {
|
||||
std::vector<std::string> tokens;
|
||||
std::string::size_type i1 = 0;
|
||||
while (true) {
|
||||
|
|
@ -156,19 +155,20 @@ std::vector<std::string> tokenize(std::string string, char separator)
|
|||
}
|
||||
|
||||
static void print_guid(const char *call, GUID guid) {
|
||||
std::cout << call << ":";
|
||||
std::cout << " Data1: " << std::hex << guid.Data1 << ",";
|
||||
std::cout << " Data2: " << std::hex << guid.Data2 << ",";
|
||||
std::cout << " Data3: " << std::hex << guid.Data3 << ",";
|
||||
std::cout << " Data4: [ ";
|
||||
std::ostringstream ss;
|
||||
ss << call << ":";
|
||||
ss << " Data1: " << std::hex << guid.Data1 << ",";
|
||||
ss << " Data2: " << std::hex << guid.Data2 << ",";
|
||||
ss << " Data3: " << std::hex << guid.Data3 << ",";
|
||||
ss << " Data4: [ ";
|
||||
for (int j = 0; j < 8; j++) {
|
||||
std::cout << std::hex << (int)guid.Data4[j] << " ";
|
||||
ss << std::hex << (int)guid.Data4[j] << " ";
|
||||
}
|
||||
std::cout << "]" << std::endl;
|
||||
ss << "]";
|
||||
LOG(INFO) << ss.str();
|
||||
}
|
||||
|
||||
bool parse_usb_path(int & vid, int & pid, int & mi, std::string & unique_id, const std::string & path)
|
||||
{
|
||||
bool parse_usb_path(int &vid, int &pid, int &mi, std::string &unique_id, const std::string &path) {
|
||||
auto name = path;
|
||||
std::transform(begin(name), end(name), begin(name), ::tolower);
|
||||
auto tokens = tokenize(name, '#');
|
||||
|
|
@ -203,8 +203,7 @@ bool parse_usb_path(int & vid, int & pid, int & mi, std::string & unique_id, con
|
|||
return true;
|
||||
}
|
||||
|
||||
bool parse_usb_path_from_device_id(int & vid, int & pid, int & mi, std::string & unique_id, const std::string & device_id)
|
||||
{
|
||||
bool parse_usb_path_from_device_id(int &vid, int &pid, int &mi, std::string &unique_id, const std::string &device_id) {
|
||||
auto name = device_id;
|
||||
std::transform(begin(name), end(name), begin(name), ::tolower);
|
||||
auto tokens = tokenize(name, '\\');
|
||||
|
|
@ -247,8 +246,7 @@ struct context {
|
|||
}
|
||||
};
|
||||
|
||||
class reader_callback : public IMFSourceReaderCallback
|
||||
{
|
||||
class reader_callback : public IMFSourceReaderCallback {
|
||||
std::weak_ptr<device> owner; // The device holds a reference to us, so use weak_ptr to prevent a cycle
|
||||
ULONG ref_count;
|
||||
volatile bool streaming = false;
|
||||
|
|
@ -324,7 +322,6 @@ struct device {
|
|||
check("get_NumNodes", ks_topology_info->get_NumNodes(&numberOfNodes));
|
||||
for (int i = 0; i < numberOfNodes; i++) {
|
||||
check("get_NodeType", ks_topology_info->get_NodeType(i, &node_type));
|
||||
std::cout << "node" << i << " ";
|
||||
print_guid("node_type", node_type);
|
||||
}
|
||||
*/
|
||||
|
|
@ -337,7 +334,7 @@ struct device {
|
|||
|
||||
com_ptr<IKsControl> ks_control;
|
||||
check("QueryInterface", unknown->QueryInterface(__uuidof(IKsControl), (void **)&ks_control));
|
||||
LOG(INFO) << "Obtained KS control node : " << xu.node;
|
||||
VLOG_INFO << "Obtained KS control node : " << xu.node;
|
||||
return ks_controls[xu.node] = ks_control;
|
||||
}
|
||||
|
||||
|
|
@ -381,8 +378,7 @@ struct device {
|
|||
|
||||
};
|
||||
|
||||
HRESULT reader_callback::OnReadSample(HRESULT hrStatus, DWORD dwStreamIndex, DWORD dwStreamFlags, LONGLONG llTimestamp, IMFSample * sample)
|
||||
{
|
||||
HRESULT reader_callback::OnReadSample(HRESULT hrStatus, DWORD dwStreamIndex, DWORD dwStreamFlags, LONGLONG llTimestamp, IMFSample *sample) {
|
||||
if (auto owner_ptr = owner.lock()) {
|
||||
if (sample) {
|
||||
com_ptr<IMFMediaBuffer> buffer = NULL;
|
||||
|
|
@ -415,13 +411,11 @@ HRESULT reader_callback::OnReadSample(HRESULT hrStatus, DWORD dwStreamIndex, DWO
|
|||
return S_OK;
|
||||
}
|
||||
|
||||
std::shared_ptr<context> create_context()
|
||||
{
|
||||
std::shared_ptr<context> create_context() {
|
||||
return std::make_shared<context>();
|
||||
}
|
||||
|
||||
std::vector<std::shared_ptr<device>> query_devices(std::shared_ptr<context> context)
|
||||
{
|
||||
std::vector<std::shared_ptr<device>> query_devices(std::shared_ptr<context> context) {
|
||||
IMFAttributes *pAttributes = NULL;
|
||||
check("MFCreateAttributes", MFCreateAttributes(&pAttributes, 1));
|
||||
check("IMFAttributes::SetGUID", pAttributes->SetGUID(
|
||||
|
|
@ -465,9 +459,8 @@ std::vector<std::shared_ptr<device>> query_devices(std::shared_ptr<context> cont
|
|||
dev = std::make_shared<device>(context, vid, pid, unique_id, name);
|
||||
devices.push_back(dev);
|
||||
} catch (const std::exception &e) {
|
||||
VLOG(2) << "Not a USB video device: " << e.what();
|
||||
VLOG_INFO << "Not a USB video device: " << e.what();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
dev->reader_callback = new reader_callback(dev);
|
||||
|
|
@ -480,28 +473,23 @@ std::vector<std::shared_ptr<device>> query_devices(std::shared_ptr<context> cont
|
|||
return devices;
|
||||
}
|
||||
|
||||
int get_vendor_id(const device &device)
|
||||
{
|
||||
int get_vendor_id(const device &device) {
|
||||
return device.vid;
|
||||
}
|
||||
|
||||
int get_product_id(const device &device)
|
||||
{
|
||||
int get_product_id(const device &device) {
|
||||
return device.pid;
|
||||
}
|
||||
|
||||
std::string get_name(const device &device)
|
||||
{
|
||||
std::string get_name(const device &device) {
|
||||
return device.name;
|
||||
}
|
||||
|
||||
std::string get_video_name(const device &device)
|
||||
{
|
||||
std::string get_video_name(const device &device) {
|
||||
return device.name;
|
||||
}
|
||||
|
||||
static long get_cid(Option option)
|
||||
{
|
||||
static long get_cid(Option option) {
|
||||
switch (option) {
|
||||
case Option::GAIN:
|
||||
return VideoProcAmp_Gain;
|
||||
|
|
@ -516,8 +504,7 @@ static long get_cid(Option option)
|
|||
|
||||
bool pu_control_range(
|
||||
const device &device, Option option, int32_t *min, int32_t *max,
|
||||
int32_t *def)
|
||||
{
|
||||
int32_t *def) {
|
||||
const_cast<uvc::device &>(device).get_media_source();
|
||||
long minVal = 0, maxVal = 0, steppingDelta = 0, defVal = 0, capsFlag = 0;
|
||||
check("IAMVideoProcAmp::GetRange", const_cast<uvc::device &>(device).am_video_proc_amp->GetRange(get_cid(option), &minVal, &maxVal, &steppingDelta, &defVal, &capsFlag));
|
||||
|
|
@ -527,22 +514,19 @@ bool pu_control_range(
|
|||
return true;
|
||||
}
|
||||
|
||||
void get_pu_control(const device &device, long property, int32_t *value)
|
||||
{
|
||||
void get_pu_control(const device &device, long property, int32_t *value) {
|
||||
long data, flags = 0;
|
||||
check("IAMVideoProcAmp::Get", const_cast<uvc::device &>(device).am_video_proc_amp->Get(property, &data, &flags));
|
||||
*value = data;
|
||||
}
|
||||
|
||||
void set_pu_control(const device &device, long property, int32_t *value)
|
||||
{
|
||||
void set_pu_control(const device &device, long property, int32_t *value) {
|
||||
long data = *value;
|
||||
check("IAMVideoProcAmp::Set", const_cast<uvc::device &>(device).am_video_proc_amp->Set(property, data, VideoProcAmp_Flags_Auto));
|
||||
}
|
||||
|
||||
bool pu_control_query(
|
||||
const device &device, Option option, pu_query query, int32_t *value)
|
||||
{
|
||||
const device &device, Option option, pu_query query, int32_t *value) {
|
||||
CHECK_NOTNULL(value);
|
||||
const_cast<uvc::device &>(device).get_media_source();
|
||||
switch (query) {
|
||||
|
|
@ -557,70 +541,10 @@ bool pu_control_query(
|
|||
return false;
|
||||
}
|
||||
}
|
||||
/*
|
||||
void get_extension_control_range(const device &device, const xu &xu, uint8_t selector, xu_query query, uint8_t *data)
|
||||
{
|
||||
CHECK_NOTNULL(data);
|
||||
int offset = 0;
|
||||
auto ks_control = const_cast<uvc::device &>(device).get_ks_control(xu);
|
||||
|
||||
// get step, min and max values
|
||||
KSP_NODE node;
|
||||
memset(&node, 0, sizeof(KSP_NODE));
|
||||
node.Property.Set = reinterpret_cast<const GUID &>(xu.id);
|
||||
node.Property.Id = selector;
|
||||
node.NodeId = xu.node;
|
||||
|
||||
switch (query) {
|
||||
case XU_QUERY_MIN:
|
||||
offset = 1;
|
||||
node.Property.Flags = KSPROPERTY_TYPE_BASICSUPPORT | KSPROPERTY_TYPE_TOPOLOGY;
|
||||
break;
|
||||
case XU_QUERY_MAX:
|
||||
offset = 2;
|
||||
node.Property.Flags = KSPROPERTY_TYPE_BASICSUPPORT | KSPROPERTY_TYPE_TOPOLOGY;
|
||||
break;
|
||||
case XU_QUERY_DEF:
|
||||
offset = 0;
|
||||
node.Property.Flags = KSPROPERTY_TYPE_DEFAULTVALUES | KSPROPERTY_TYPE_TOPOLOGY;
|
||||
break;
|
||||
default:
|
||||
LOG(ERROR) << "xu request code is unaccepted";
|
||||
break;
|
||||
}
|
||||
|
||||
KSPROPERTY_DESCRIPTION description;
|
||||
unsigned long bytes_received = 0;
|
||||
check("IKsControl::KsProperty", ks_control->KsProperty(
|
||||
(PKSPROPERTY)&node,
|
||||
sizeof(node),
|
||||
&description,
|
||||
sizeof(KSPROPERTY_DESCRIPTION),
|
||||
&bytes_received));
|
||||
|
||||
unsigned long size = description.DescriptionSize;
|
||||
std::vector<BYTE> buffer((long)size);
|
||||
|
||||
check("IKsControl::KsProperty", ks_control->KsProperty(
|
||||
(PKSPROPERTY)&node,
|
||||
sizeof(node),
|
||||
buffer.data(),
|
||||
size,
|
||||
&bytes_received));
|
||||
|
||||
if (bytes_received != size) {
|
||||
throw std::runtime_error("wrong data");
|
||||
}
|
||||
|
||||
BYTE * pRangeValues = buffer.data() + sizeof(KSPROPERTY_MEMBERSHEADER) + sizeof(KSPROPERTY_DESCRIPTION);
|
||||
* data = (uint8_t)*(pRangeValues + offset);
|
||||
|
||||
}
|
||||
*/
|
||||
bool xu_control_query(
|
||||
const device &device, const xu &xu, uint8_t selector, xu_query query,
|
||||
uint16_t size, uint8_t *data)
|
||||
{
|
||||
uint16_t size, uint8_t *data) {
|
||||
CHECK_NOTNULL(data);
|
||||
int offset = 0;
|
||||
int range_offset = sizeof(KSPROPERTY_MEMBERSHEADER) + sizeof(KSPROPERTY_DESCRIPTION);
|
||||
|
|
@ -663,8 +587,7 @@ bool xu_control_query(
|
|||
return true;
|
||||
}
|
||||
|
||||
void set_device_mode(device & device, int width, int height, int fourcc, int fps, video_channel_callback callback)
|
||||
{
|
||||
void set_device_mode(device &device, int width, int height, int fourcc, int fps, video_channel_callback callback) {
|
||||
if (!device.mf_source_reader) {
|
||||
com_ptr<IMFAttributes> pAttributes;
|
||||
check("MFCreateAttributes", MFCreateAttributes(&pAttributes, 1));
|
||||
|
|
@ -701,12 +624,11 @@ void set_device_mode(device & device, int width, int height, int fourcc, int fps
|
|||
throw_error() << "no matching media type for pixel format " << std::hex << fourcc;
|
||||
}
|
||||
|
||||
void start_streaming(device & device, int num_transfer_bufs)
|
||||
{
|
||||
void start_streaming(device &device, int num_transfer_bufs) {
|
||||
device.start_streaming();
|
||||
}
|
||||
void stop_streaming(device & device)
|
||||
{
|
||||
|
||||
void stop_streaming(device &device) {
|
||||
device.stop_streaming();
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user