int main(int argc, char *argv[])
{
CSimpleOpt s(argc, argv, argumentSpecifications);
logger.setDefaultLogLevel(LOG_INFO);
uint16_t vid = 0;
Vector<uint16_t> pids;
String serialNumber;
String dumpFileName;
String type = "raw";
int32_t frameCount = 1;
char *endptr;
while (s.Next())
{
if (s.LastError() != SO_SUCCESS)
{
std::cout << s.GetLastErrorText(s.LastError()) << ": '" << s.OptionText() << "' (use -h to get command line help)" << std::endl;
help();
return -1;
}
//std::cout << s.OptionId() << ": " << s.OptionArg() << std::endl;
Vector<String> splits;
switch (s.OptionId())
{
case VENDOR_ID:
vid = (uint16_t)strtol(s.OptionArg(), &endptr, 16);
break;
case PRODUCT_ID:
split(s.OptionArg(), ',', splits);
for(auto &s1: splits)
pids.push_back((uint16_t)strtol(s1.c_str(), &endptr, 16));
break;
case SERIAL_NUMBER:
serialNumber = s.OptionArg();
break;
case DUMP_FILE:
dumpFileName = s.OptionArg();
break;
case NUM_OF_FRAMES:
frameCount = (int32_t)strtol(s.OptionArg(), &endptr, 10);
break;
case CAPTURE_TYPE:
type = s.OptionArg();
break;
default:
help();
break;
};
}
if(vid == 0 || pids.size() == 0 || pids[0] == 0 || dumpFileName.size() == 0)
{
std::cerr << "Required argument missing." << std::endl;
help();
return -1;
}
if (type != "raw" && type != "raw_processed" && type != "depth" && type != "pointcloud")
{
std::cerr << "Unknown type '" << type << "'" << std::endl;
help();
return -1;
}
std::ofstream f(dumpFileName, std::ios::binary | std::ios::out);
if(!f.good())
{
std::cerr << "Failed to open '" << dumpFileName << "'" << std::endl;
return -1;
}
CameraSystem sys;
// Get all valid detected devices
const Vector<DevicePtr> &devices = sys.scan();
DevicePtr toConnect;
std::cout << "Detected devices: " << std::endl;
for(auto &d: devices)
{
std::cout << d->id() << std::endl;
if(d->interfaceID() == Device::USB)
{
USBDevice &usb = (USBDevice &)*d;
if(usb.vendorID() == vid && (serialNumber.size() == 0 || usb.serialNumber() == serialNumber))
{
for(auto pid: pids)
if(usb.productID() == pid)
toConnect = d;
}
}
}
if(!toConnect)
{
std::cerr << "No valid device found for the specified VID:PID:serialnumber" << std::endl;
return -1;
}
DepthCameraPtr depthCamera = sys.connect(toConnect);
if(!depthCamera)
{
std::cerr << "Could not load depth camera for device " << toConnect->id() << std::endl;
return -1;
}
if(!depthCamera->isInitialized())
{
std::cerr << "Depth camera not initialized for device " << toConnect->id() << std::endl;
return -1;
}
std::cout << "Successfully loaded depth camera for device " << toConnect->id() << std::endl;
int count = 0;
TimeStampType lastTimeStamp = 0;
if (type == "raw")
{
depthCamera->registerCallback(DepthCamera::FRAME_RAW_FRAME_UNPROCESSED, [&](DepthCamera &dc, const Frame &frame, DepthCamera::FrameType c) {
const RawDataFrame *d = dynamic_cast<const RawDataFrame *>(&frame);
if (!d)
{
std::cout << "Null frame captured? or not of type RawDataFrame" << std::endl;
return;
}
std::cout << "Capture frame " << d->id << "@" << d->timestamp;
if (lastTimeStamp != 0)
std::cout << " (" << 1E6 / (d->timestamp - lastTimeStamp) << " fps)";
std::cout << std::endl;
f.write((char *)&d->id, sizeof(d->id));
f.write((char *)&d->timestamp, sizeof(d->timestamp));
lastTimeStamp = d->timestamp;
f.write((char *)d->data.data(), d->data.size());
count++;
if (count >= frameCount)
dc.stop();
});
}
else if (type == "raw_processed")
{
depthCamera->registerCallback(DepthCamera::FRAME_RAW_FRAME_PROCESSED, [&](DepthCamera &dc, const Frame &frame, DepthCamera::FrameType c) {
const ToFRawFrame *d = dynamic_cast<const ToFRawFrame *>(&frame);
if (!d)
{
std::cout << "Null frame captured? or not of type ToFRawFrame" << std::endl;
return;
}
std::cout << "Capture frame " << d->id << "@" << d->timestamp;
if (lastTimeStamp != 0)
std::cout << " (" << 1E6 / (d->timestamp - lastTimeStamp) << " fps)";
std::cout << std::endl;
f.write((char *)&d->id, sizeof(d->id));
f.write((char *)&d->timestamp, sizeof(d->timestamp));
lastTimeStamp = d->timestamp;
if (d->phase())
f.write((char *)d->phase(), d->phaseWordWidth()*d->size.width*d->size.height);
if (d->amplitude())
f.write((char *)d->amplitude(), d->amplitudeWordWidth()*d->size.width*d->size.height);
if (d->ambient())
f.write((char *)d->ambient(), d->ambientWordWidth()*d->size.width*d->size.height);
if (d->flags())
f.write((char *)d->flags(), d->flagsWordWidth()*d->size.width*d->size.height);
count++;
if (count >= frameCount)
dc.stop();
});
}
else if (type == "depth")
{
depthCamera->registerCallback(DepthCamera::FRAME_DEPTH_FRAME, [&](DepthCamera &dc, const Frame &frame, DepthCamera::FrameType c) {
const DepthFrame *d = dynamic_cast<const DepthFrame *>(&frame);
if (!d)
{
std::cout << "Null frame captured? or not of type DepthFrame" << std::endl;
return;
}
std::cout << "Capture frame " << d->id << "@" << d->timestamp;
if (lastTimeStamp != 0)
std::cout << " (" << 1E6 / (d->timestamp - lastTimeStamp) << " fps)";
std::cout << std::endl;
f.write((char *)&d->id, sizeof(d->id));
f.write((char *)&d->timestamp, sizeof(d->timestamp));
lastTimeStamp = d->timestamp;
f.write((char *)d->depth.data(), sizeof(float)*d->size.width*d->size.height);
f.write((char *)d->amplitude.data(), sizeof(float)*d->size.width*d->size.height);
count++;
if (count >= frameCount)
dc.stop();
});
}
else if (type == "pointcloud")
{
depthCamera->registerCallback(DepthCamera::FRAME_XYZI_POINT_CLOUD_FRAME, [&](DepthCamera &dc, const Frame &frame, DepthCamera::FrameType c) {
const XYZIPointCloudFrame *d = dynamic_cast<const XYZIPointCloudFrame *>(&frame);
if (!d)
{
std::cout << "Null frame captured? or not of type XYZIPointCloudFrame" << std::endl;
return;
}
std::cout << "Capture frame " << d->id << "@" << d->timestamp;
if (lastTimeStamp != 0)
std::cout << " (" << 1E6 / (d->timestamp - lastTimeStamp) << " fps)";
std::cout << std::endl;
f.write((char *)&d->id, sizeof(d->id));
f.write((char *)&d->timestamp, sizeof(d->timestamp));
lastTimeStamp = d->timestamp;
f.write((char *)d->points.data(), sizeof(IntensityPoint)*d->points.size());
count++;
if (count >= frameCount)
dc.stop();
});
}
if(depthCamera->start())
{
FrameRate r;
if(depthCamera->getFrameRate(r))
logger(LOG_INFO) << "Capturing at a frame rate of " << r.getFrameRate() << " fps" << std::endl;
depthCamera->wait();
}
else
std::cerr << "Could not start the depth camera " << depthCamera->id() << std::endl;
return 0;
}
int main (int argc, char* argv[])
{
logger.setDefaultLogLevel(LOG_INFO);
CameraSystem sys;
DepthCameraPtr depthCamera;
FrameSize real;
DepthFrame *frm, hand;
float ampGain = 200.0;
float proximity = 0.5;
float area = 0;
uint illum_power = 48U;
ClusterMap cmap(0.8, 0.1, 3);
real.width = XDIM;
real.height = YDIM;
hand.size = real;
if (argc != 5) {
cout << "Usage: glass_gesture <ampGain> <proximity> <illum power> <area> " << endl;
return -1;
}
ampGain = atof(argv[1]);
proximity = atof(argv[2]);
illum_power = atoi(argv[3]);
area = atof(argv[4]);
cout << "ampGain= " << ampGain << " "
<< "proximity= " << proximity << " "
<< "illum_power = " << illum_power
<< "area = " << area << endl;
const Vector<DevicePtr> &devices = sys.scan();
if(devices.size() > 0) {
depthCamera = sys.connect(devices[0]); // Connect to first available device
}
else
{
cerr << "Error: Could not find a compatible device." << endl;
return -1;
}
if(!depthCamera)
{
cerr << "Error: Could not open a depth camera." << endl;
return -1;
}
if (!depthCamera->isInitialized())
{
cerr << "Error: Depth camera not initialized " << endl;
return -1;
}
cout << "Successfully loaded depth camera " << endl;
depthCamera->registerCallback(DepthCamera::FRAME_DEPTH_FRAME, frameCallback);
depthCamera->setFrameSize(real);
cout << "set intg = " << depthCamera->set("intg_duty_cycle", 1U) << endl;
depthCamera->start();
bool done = false;
namedWindow( "AmpThresh", WINDOW_NORMAL );
namedWindow( "DepthThresh", WINDOW_NORMAL );
namedWindow( "Original", WINDOW_NORMAL );
namedWindow( "Filtered", WINDOW_NORMAL );
while (!done) {
char key = getkey();
if ('q' == key) {
done = true;
cout << "Exit." << endl;
}
if (!qFrame.empty()) {
frm = &qFrame.front();
// Clip background (objects > proximity)
//
hand.depth.clear();
hand.amplitude.clear();
for (int i = 0; i < frm->depth.size(); i++) {
hand.depth.push_back((frm->depth[i] < proximity && frm->amplitude[i] > 0.01) ? frm->depth[i] : 0.0);
hand.amplitude.push_back((frm->depth[i] < proximity && frm->amplitude[i] > 0.01) ? ampGain : 0.0);
}
// Create viewable images
//
unsigned char *depth = FloatImageUtils::getVisualImage(hand.depth.data(),
hand.size.width, hand.size.height,
0, MAX_AMPLITUDE,false);
unsigned char *orig = FloatImageUtils::getVisualImage(frm->amplitude.data(),
frm->size.width, frm->size.height,
0, MAX_AMPLITUDE,true);
Mat orig_img = Mat(frm->size.height, frm->size.width, CV_8UC3, orig);
Mat depth_img = Mat(hand.size.height, hand.size.width, CV_8UC3, depth);
Mat amp_img = Mat(hand.size.height, hand.size.width, CV_32FC1, hand.amplitude.data());
Mat filter_img = Mat(hand.size.height, hand.size.width, CV_32FC1, Scalar(0));
// Perform clustering
//
cmap.scan(amp_img);
// Find the largest cluster, which should be the hand
//
int max_cluster_id = 0;
float max_area = 0;
for (int i=0; i < cmap.getClusters().size(); i++) {
if (cmap.getClusters()[i].getArea() > max_area) {
max_area = cmap.getClusters()[i].getArea();
max_cluster_id = i;
}
}
// Find the finger tip
//
POINT tip = POINT(amp_img.cols,amp_img.rows,0);
float dist2, max_dist2 = 0.0;
if (cmap.getClusters().size() > 0) {
// Find tip which point in cluster farthest from bottom-right corner
//
for (int i=0; i < cmap.getClusters()[max_cluster_id].getPoints().size(); i++) {
POINT p = cmap.getClusters()[max_cluster_id].getPoints()[i];
filter_img.at<float>(p.y, p.x) = p.z;
if ((p.y <= cmap.getClusters()[max_cluster_id].getMin().y) ||
(p.x <= cmap.getClusters()[max_cluster_id].getMin().x)) {
dist2 = (filter_img.cols-p.x)*(filter_img.cols-p.x)
+ (filter_img.rows-p.y)*(filter_img.rows-p.y);
if (dist2 > max_dist2) {
max_dist2 = dist2;
tip = p;
}
}
}
POINT p = cmap.getClusters()[max_cluster_id].getCentroid();
cv::Point palm = cv::Point(p.x, p.y);
cv::Point finger_tip = cv::Point(tip.x, tip.y);
float click_dist = depth_img.at<float>(finger_tip) - depth_img.at<float>(palm);
// Draw bounding box and tip
if (max_area > area) {
cv::Point Pmax = cv::Point(cmap.getClusters()[max_cluster_id].getMax().x,
cmap.getClusters()[max_cluster_id].getMax().y);
cv::Point Pmin = cv::Point(cmap.getClusters()[max_cluster_id].getMin().x,
cmap.getClusters()[max_cluster_id].getMin().y);
rectangle(orig_img, Pmin, Pmax, Scalar(255));
circle(orig_img, finger_tip, 3, Scalar(255));
circle(orig_img, palm, 3, Scalar(255));
}
cout << "tip = (" << tip.x << "," << tip.y << ") "
<< "tip depth = " << frm->depth[frm->size.width*tip.y+tip.x]
<< "click_dist2 = " << max_dist2 << endl;
if (max_dist2 < 2000)
cout << " ======== mouse down ========" << endl;
}
int mid = (frm->size.width*frm->size.height + frm->size.width)/2;
cout << "center amp =" << frm->amplitude[mid] << " "
<< "center depth =" << frm->depth[mid]
<< "depth*amp = " << frm->depth[mid]*frm->amplitude[mid] << endl;
imshow("Original", ampGain*orig_img);
imshow("AmpThresh", amp_img);
imshow("DepthThresh", depth_img);
imshow("Filtered", filter_img);
delete depth;
delete orig;
qFrame.pop_front();
}
if (waitKey(30) >= 0) done = true;
} // end while
depthCamera->stop();
return 0;
}