需要检查的代码

#include <iostream>
#include <stdio.h>
#include <iomanip>
#include <time.h>
#include <signal.h>
#include <opencv2/opencv.hpp>
#pragma warning( disable : 4996 )
#include <libfreenect2/libfreenect2.hpp>
#include <libfreenect2/frame_listener_impl.h>
#include <libfreenect2/registration.h>
#include <libfreenect2/packet_pipeline.h>
#include <libfreenect2/logger.h>

using namespace std;
using namespace cv;
using namespace libfreenect2;

enum
{
	Processor_cl,
	Processor_gl,
	Processor_cpu
};

bool protonect_shutdown = false; // Whether the running application should shut down.
bool save_images = false;        // Flag to indicate when to save images.
std::chrono::steady_clock::time_point last_save_time;

void sigint_handler(int s)
{
	protonect_shutdown = true;
}

std::string getCurrentTimeString() {
	time_t now = time(0);
	tm *ltm = localtime(&now);
	char buffer[80];
	strftime(buffer, sizeof(buffer), "%Y%m%d_%H%M%S", ltm);
	return std::string(buffer);
}

int main()
{
	// Define variables
	std::cout << "Hello World!" << std::endl;
	libfreenect2::Freenect2 freenect2;
	std::vector<libfreenect2::Freenect2Device*> devices;
	std::vector<libfreenect2::PacketPipeline*> pipelines;
	std::vector<libfreenect2::SyncMultiFrameListener*> listeners;
	std::vector<libfreenect2::FrameMap> frames;

	// Search and initialize sensors
	int numDevices = freenect2.enumerateDevices();
	if (numDevices < 2)
	{
		std::cerr << "Error: Not enough Kinect devices connected!" << std::endl;
		return -1;
	}

	// Configure transfer format
	int depthProcessor = Processor_cl;
	for (int i = 0; i < 2; ++i)
	{
		string serial = freenect2.getDeviceSerialNumber(i);
		std::cout << "SERIAL " << i + 1 << ": " << serial << std::endl;

		libfreenect2::PacketPipeline* pipeline = nullptr;
		if (depthProcessor == Processor_cpu)
		{
			pipeline = new libfreenect2::CpuPacketPipeline();
		}
		else if (depthProcessor == Processor_gl) // if support gl
		{
#ifdef LIBFREENECT2_WITH_OPENGL_SUPPORT
			pipeline = new libfreenect2::OpenGLPacketPipeline();
#else
			std::cout << "OpenGL pipeline is not supported!" << std::endl;
#endif
		}
		else if (depthProcessor == Processor_cl) // if support cl
		{
#ifdef LIBFREENECT2_WITH_OPENCL_SUPPORT
			pipeline = new libfreenect2::OpenCLPacketPipeline();
#else
			std::cout << "OpenCL pipeline is not supported!" << std::endl;
#endif
		}

		if (pipeline)
		{
			libfreenect2::Freenect2Device* dev = freenect2.openDevice(serial, pipeline);
			if (dev == nullptr)
			{
				std::cerr << "Failure opening device " << i + 1 << "!" << std::endl;
				return -1;
			}
			devices.push_back(dev);
			pipelines.push_back(pipeline);

			libfreenect2::SyncMultiFrameListener* listener = new libfreenect2::SyncMultiFrameListener(
				libfreenect2::Frame::Color |
				libfreenect2::Frame::Depth |
				libfreenect2::Frame::Ir);
			listeners.push_back(listener);
			frames.push_back(libfreenect2::FrameMap());

			dev->setColorFrameListener(listener);
			dev->setIrAndDepthFrameListener(listener);
			dev->start();
		}
	}

	signal(SIGINT, sigint_handler);
	protonect_shutdown = false;

	// Create windows
	cv::namedWindow("rgb1", WND_PROP_ASPECT_RATIO);
	cv::namedWindow("rgb2", WND_PROP_ASPECT_RATIO);
	cv::namedWindow("ir1", WND_PROP_ASPECT_RATIO);
	cv::namedWindow("ir2", WND_PROP_ASPECT_RATIO);
	cv::namedWindow("depth1", WND_PROP_ASPECT_RATIO);
	cv::namedWindow("depth2", WND_PROP_ASPECT_RATIO);

	// Define save directory
	std::string save_directory = "C:\\A0fangfang\\shiyan\\9.2\\";

	last_save_time = std::chrono::steady_clock::now(); // Initialize the last save time

													   // Loop to receive frames
	while (!protonect_shutdown)
	{
		for (int i = 0; i < devices.size(); ++i)
		{
			listeners[i]->waitForNewFrame(frames[i]);
			libfreenect2::Frame* rgb = frames[i][libfreenect2::Frame::Color];
			libfreenect2::Frame* ir = frames[i][libfreenect2::Frame::Ir];
			libfreenect2::Frame* depth = frames[i][libfreenect2::Frame::Depth];

			cv::Mat rgbmat((int)rgb->height, (int)rgb->width, CV_8UC4, rgb->data);
			cv::Mat irmat((int)ir->height, (int)ir->width, CV_32FC1, ir->data);
			cv::Mat depthmat((int)depth->height, (int)depth->width, CV_32FC1, depth->data);

			cv::imshow("rgb" + std::to_string(i + 1), rgbmat);
			cv::imshow("ir" + std::to_string(i + 1), irmat / 4500.0f);
			cv::imshow("depth" + std::to_string(i + 1), depthmat / 4500.0f);

			// Check for keypress to trigger image saving
			int key = cv::waitKey(30);
			if (key == 'q')
			{
				save_images = true;
			}
			else if (key == 27) // Escape key
			{
				protonect_shutdown = true;
			}

			if (save_images)
			{
				auto current_time = std::chrono::steady_clock::now();
				std::string timestamp = getCurrentTimeString();

				for (int j = 0; j < devices.size(); ++j)
				{
					std::string filename_rgb = save_directory + "rgb" + std::to_string(j + 1) + "_" + timestamp + ".png";
					std::string filename_ir = save_directory + "ir" + std::to_string(j + 1) + "_" + timestamp + ".png";
					std::string filename_depth = save_directory + "depth" + std::to_string(j + 1) + "_" + timestamp + ".png";

					cv::imwrite(filename_rgb, rgbmat);
					cv::imwrite(filename_ir, irmat / 4500.0f * 255); // Convert to 8-bit
					cv::imwrite(filename_depth, depthmat / 4500.0f * 255); // Convert to 8-bit

					std::cout << "Images saved to " << save_directory << std::endl;
				}

				save_images = false; // Reset flag after saving
			}

			listeners[i]->release(frames[i]);
		}
	}

	// Close devices
	for (auto dev : devices)
	{
		dev->stop();
		dev->close();
	}

	// Clean up resources
	for (auto pipeline : pipelines)
	{
		delete pipeline;
	}
	for (auto listener : listeners)
	{
		delete listener;
	}

	std::cout << "Goodbye World!" << std::endl;
	return 0;
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值