-
Notifications
You must be signed in to change notification settings - Fork 0
/
adbscreenshotthread.cpp
95 lines (84 loc) · 2.72 KB
/
adbscreenshotthread.cpp
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
/*
* adb screenshots implementation of visualdatainterface for libvisualdata
* Copyright (c) 2013-2014, Intel Corporation.
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU Lesser General Public License,
* version 2.1, as published by the Free Software Foundation.
*
* This program is distributed in the hope it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
*
*/
#include <stdio.h>
#include "adbscreenshotthread.hpp"
using boost::thread;
using boost::bind;
using std::string;
static const string ADBSCRCAP = "adb shell screencap -p /sdcard/screen.png";
static const string ADBGETSCR = "adb pull /sdcard/screen.png /tmp/screen.png";
static const string SCRSHOT = "/tmp/screen.png";
static const unsigned int REFRESHDELAY = 500000;
adbscreenshotthread::adbscreenshotthread(
boost::function<void(cv::Mat&)> callback) {
visualdatainterface::callback_ = callback;
startthread();
}
adbscreenshotthread::~adbscreenshotthread() {
stopthread();
}
inline void adbscreenshotthread::detach() {
thread_->detach();
}
inline void adbscreenshotthread::operator()() {
capture();
}
void adbscreenshotthread::startthread() {
if (!shootscreen_) {
shootscreen_ = true;
thread_ = new thread(bind(&adbscreenshotthread::capture, this));
} else {
shootscreen_ = false;
}
return;
}
void adbscreenshotthread::stopthread() {
shootscreen_ = false;
thread_->join();
delete thread_;
thread_ = NULL;
}
inline void adbscreenshotthread::capture() {
while (shootscreen_) {
if (!runcmd(ADBSCRCAP) || !runcmd(ADBGETSCR)) {
shootscreen_ = false;
return;
}
cv::Mat image = cv::imread(SCRSHOT, CV_LOAD_IMAGE_COLOR);
if (image.empty())
throw std::runtime_error("adbscreenshotthread - Image empty");
if (framesize_.width == 0)
visualdatainterface::framesize_ = image.size();
frame_ = image;
callback_(frame_);
usleep(REFRESHDELAY);
}
}
inline bool adbscreenshotthread::runcmd(const std::string& cmd) {
FILE* pf = popen(cmd.c_str(), "r");
if(!pf)
return false;
int status = pclose(pf);
if(status == 0) {
return true;
} else {
std::cerr << "ERROR: runcmd [" << cmd << "]:" << status << std::endl;
return false;
}
}