1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#include "shm-layout.hpp"
#include "shm.hpp"
#include "ps3eye-driver/ps3eye.hpp"
#include <cstdlib>
#ifdef __clang__
# pragma clang diagnostic ignored "-Watomic-implicit-seq-cst"
#endif
#ifdef __GNUG__
# pragma GCC diagnostic ignored "-Wcast-qual"
# pragma GCC diagnostic ignored "-Wformat-security"
# pragma GCC diagnostic ignored "-Wformat-nonliteral"
#endif
#ifdef _MSC_VER
# include <windows.h>
# define FULL_BARRIER [](){ _ReadWriteBarrier(); MemoryBarrier(); }
#else
# define FULL_BARRIER __sync_synchronize
#endif
template<int N, typename... xs>
[[noreturn]]
static void error(volatile ps3eye::shm_out& out, const char (&error)[N], const xs&... args)
{
snprintf((char*)out.error_string, sizeof(ps3eye::shm_out::error_string), error, args...);
std::quick_exit(2);
}
static void update_settings(ps3eye::camera& camera, const volatile ps3eye::shm_in& in)
{
camera.set_framerate(in.framerate);
camera.set_auto_gain(in.auto_gain);
camera.set_gain(in.gain);
camera.set_exposure(in.exposure);
camera.set_test_pattern_status(in.test_pattern);
}
int main(int argc, char** argv)
{
(void)argc; (void)argv;
shm_wrapper mem_("ps3eye-driver-shm", nullptr, sizeof(ps3eye::shm));
volatile auto& ptr_ = *(ps3eye::shm*)mem_.ptr();
const volatile auto& in = ptr_.in;
volatile auto& out = ptr_.out;
auto cameras = ps3eye::list_devices();
if (cameras.empty())
error(out, "no camera found");
auto& camera = cameras[0];
uint8_t* frame;
decltype(out.timecode) timecode = 0;
{
ps3eye::resolution mode;
switch (in.resolution)
{
case ps3eye::shm_in::mode::qvga:
mode = ps3eye::res_QVGA;
frame = (uint8_t*)out.data_320x240;
break;
case ps3eye::shm_in::mode::vga:
mode = ps3eye::res_VGA;
frame = (uint8_t*)out.data_640x480;
break;
default: error(out, "wrong resolution %u", (unsigned)mode);
}
int framerate = in.framerate;
if (!framerate)
framerate = 60;
bool ret = camera->init(mode, framerate) && camera->start();
if (!ret)
error(out, "camera init failed: %s", camera->error_string());
}
update_settings(*camera, in);
for (;;)
{
{
auto cookie = in.settings_updated;
if (cookie != out.settings_updated_ack)
{
update_settings(*camera, in);
out.settings_updated_ack = cookie;
}
}
bool success = true;
success &= camera->get_frame(frame);
if (!success)
error(out, "error %s", camera->error_string());
out.timecode = ++timecode;
if (in.do_exit)
break;
FULL_BARRIER();
}
return 0;
}
|
