summaryrefslogtreecommitdiffhomepage
path: root/ftnoir_tracker_pt/camera.h
blob: 32c76465b7e9296ce34ec51d342d0115f2aa9d25 (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
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
113
114
115
116
117
118
119
/* Copyright (c) 2012 Patrick Ruoff
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 */

#ifndef CAMERA_H
#define CAMERA_H

#include <opencv2/opencv.hpp>
#include <opencv2/highgui.hpp>
//#include "videoInput/videoInput.h"

// ----------------------------------------------------------------------------
struct CamInfo
{
	CamInfo() : res_x(0), res_y(0), fps(0), f(1) {}

	int res_x;
	int res_y;
	int fps;
	float f;	// (focal length) / (sensor width)
};

// ----------------------------------------------------------------------------
// base class for cameras
class Camera
{
public:
    Camera() : desired_index(0), active_index(-1), active(false), dt_valid(0), dt_mean(0) {}
	virtual ~Camera() {}

	// start/stop capturing
	virtual void start() = 0;
	virtual void stop() = 0;
	void restart() { stop(); start(); }

	void set_index(int index);
	void set_f(float f);
	void set_fps(int fps);
	void set_res(int x_res, int y_res);

	// gets a frame from the camera, dt: time since last call in seconds
	bool get_frame(float dt, cv::Mat* frame);

	// WARNING: returned references are valid as long as object
	const CamInfo& get_info() const { return cam_info; }
	const CamInfo& get_desired() const { return cam_desired; }

protected:
	// get a frame from the camera
	virtual bool _get_frame(cv::Mat* frame) = 0;

	// update the camera
	virtual void _set_index() = 0;
	virtual void _set_f() = 0;
	virtual void _set_fps() = 0;
	virtual void _set_res() = 0;

    int desired_index;
    int active_index;
	bool active;
    float dt_valid;
    float dt_mean;
	CamInfo cam_info;
	CamInfo cam_desired;
};


// ----------------------------------------------------------------------------
// OpenCV camera

class CVCamera : public Camera
{
public:
	CVCamera() : cap(NULL) {}
	~CVCamera() { stop(); }

	void start();
	void stop();

protected:
	bool _get_frame(cv::Mat* frame);
	void _set_index();
	void _set_f();
	void _set_fps();
	void _set_res();

	cv::VideoCapture* cap;
};


// ----------------------------------------------------------------------------
// videoInput camera
#if 0
class VICamera : public Camera
{
public:
	VICamera();
	~VICamera() { stop(); }

	void start();
	void stop();

protected:
	bool _get_frame(cv::Mat* frame);
	void _set_index();
	void _set_f();
	void _set_fps();
	void _set_res();

	videoInput VI;
	cv::Mat new_frame;
	unsigned char* frame_buffer;
};
#endif

#endif //CAMERA_H