blob: a0bae6947bc51528e12d46eb75570ed9df9e83e1 (
plain)
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
|
#include "camera.h"
using namespace cv;
// ----------------------------------------------------------------------------
Camera::Camera()
: dt_valid(0), dt_mean(0), cap(NULL)
{}
Camera::~Camera()
{
if (cap) cvReleaseCapture(&cap);
}
void Camera::set_index(int index)
{
if (cap) cvReleaseCapture(&cap);
cap = cvCreateCameraCapture(index);
// extract camera info
if (cap)
{
cam_info.res_x = cvGetCaptureProperty(cap, CV_CAP_PROP_FRAME_WIDTH);
cam_info.res_y = cvGetCaptureProperty(cap, CV_CAP_PROP_FRAME_HEIGHT);
}
// reset fps calculation
dt_mean = 0;
}
cv::Mat Camera::get_frame(float dt)
{
cv::Mat frame;
if (cap && cvGrabFrame(cap) != 0)
{
// retrieve frame
IplImage* _img = cvRetrieveFrame(cap, 0);
if(_img)
{
if(_img->origin == IPL_ORIGIN_TL)
frame = Mat(_img);
else
{
Mat temp(_img);
flip(temp, frame, 0);
}
}
}
// measure fps of valid frames
const float dt_smoothing_const = 0.9;
dt_valid += dt;
if (!frame.empty())
{
dt_mean = dt_smoothing_const * dt_mean + (1.0 - dt_smoothing_const) * dt_valid;
cam_info.fps = 1.0 / dt_mean;
dt_valid = 0;
}
return frame;
}
|
