|
|
#include <iostream>
|
|
|
#include <opencv2/opencv.hpp>
|
|
|
#include <zmq.hpp>
|
|
|
#include <chrono>
|
|
|
#include <thread>
|
|
|
#include <sys/time.h>
|
|
|
|
|
|
int main() {
|
|
|
// 创建ZeroMQ上下文
|
|
|
zmq::context_t context(1);
|
|
|
|
|
|
// 创建套接字并绑定到接收方
|
|
|
zmq::socket_t socket(context, zmq::socket_type::pull);
|
|
|
socket.bind("ipc:///tmp/video_socket");
|
|
|
|
|
|
// 创建窗口用于显示视频
|
|
|
cv::namedWindow("Received Video", cv::WINDOW_NORMAL);
|
|
|
|
|
|
// 统计变量
|
|
|
int frameCount = 0;
|
|
|
double totalTime = 0.0;
|
|
|
double avgFps = 0.0;
|
|
|
double avgInterval = 0.0;
|
|
|
std::chrono::steady_clock::time_point startTime = std::chrono::steady_clock::now();
|
|
|
struct timeval tv1, tv2;
|
|
|
gettimeofday(&tv1, NULL);
|
|
|
int lastFrameCount = 0;
|
|
|
double lastTotalTime = 0.0;
|
|
|
float tspan = 0.0;
|
|
|
|
|
|
// 接收并实时播放视频帧
|
|
|
while (true) {
|
|
|
// 接收图像数据
|
|
|
zmq::message_t message;
|
|
|
socket.recv(&message);
|
|
|
|
|
|
// 解析接收到的图像数据
|
|
|
int width, height, type;
|
|
|
memcpy(&width, message.data(), sizeof(int));
|
|
|
memcpy(&height, static_cast<char*>(message.data()) + sizeof(int), sizeof(int));
|
|
|
memcpy(&type, static_cast<char*>(message.data()) + sizeof(int) * 2, sizeof(int));
|
|
|
|
|
|
// 计算图像数据的大小
|
|
|
size_t imageDataSize = message.size() - sizeof(int) * 3 - sizeof(double);
|
|
|
|
|
|
// 创建图像矩阵
|
|
|
cv::Mat frame(height, width, type, static_cast<char*>(message.data()) + sizeof(int) * 3);
|
|
|
|
|
|
// 获取帧率信息
|
|
|
double fps;
|
|
|
memcpy(&fps, static_cast<char*>(message.data()) + sizeof(int) * 3 + imageDataSize, sizeof(double));
|
|
|
|
|
|
// 为了保证周期的准确，一下代码应该写到一起，关键点在于：算当前时间、等待剩余时间、重置时间三者的关系必须是依次紧挨这，中间不要加入额外操作，重置时间以后可以加入其他操作，不用着急imshow
|
|
|
{
|
|
|
// 获取当前时间点
|
|
|
std::chrono::steady_clock::time_point currentTime = std::chrono::steady_clock::now();
|
|
|
|
|
|
// 计算已经过去的时间
|
|
|
std::chrono::duration<double> elapsedSeconds = currentTime - startTime;
|
|
|
|
|
|
// 计算应该等待的时间间隔
|
|
|
double expectedInterval = 1000 / fps;
|
|
|
|
|
|
// 计算实际需要等待的时间间隔
|
|
|
double actualInterval = expectedInterval - elapsedSeconds.count() * 1000;
|
|
|
|
|
|
// 等待剩余时间，以实现与原视频完全一致的帧率
|
|
|
if (actualInterval > 0) {
|
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(static_cast<int>(std::round(actualInterval))));
|
|
|
} else {
|
|
|
printf("Error! Accept and process the image timeout, if it is the first frame please ignore!\n");
|
|
|
}
|
|
|
|
|
|
// 更新起始时间点和计时器
|
|
|
startTime = std::chrono::steady_clock::now();
|
|
|
|
|
|
// 打印帧率和时间间隔信息
|
|
|
// std::cout << "Received FPS: " << fps << std::endl;
|
|
|
// std::cout << "Expected Interval: " << expectedInterval << " ms" << std::endl;
|
|
|
// std::cout << "Elapsed Seconds: " << elapsedSeconds.count() * 1000 << " ms" << std::endl;
|
|
|
// std::cout << "Actual Interval: " << actualInterval << " ms" << std::endl;
|
|
|
}
|
|
|
|
|
|
// 调试，在imshow前引入time.h的时间来统计帧率，看是否已经实时播放
|
|
|
{
|
|
|
gettimeofday(&tv2, NULL);
|
|
|
tspan = (tv2.tv_sec - tv1.tv_sec) * 1000.0 + (tv2.tv_usec - tv1.tv_usec) / 1000.0;
|
|
|
//printf("****************************the tspan=%f ms****************************\n", tspan);
|
|
|
gettimeofday(&tv1, NULL);
|
|
|
}
|
|
|
|
|
|
// 显示图像
|
|
|
cv::imshow("Received Video", frame);
|
|
|
|
|
|
// 按下 ESC 键退出循环
|
|
|
if (cv::waitKey(1) == 27)
|
|
|
break;
|
|
|
|
|
|
// 更新统计变量
|
|
|
frameCount++;
|
|
|
totalTime += tspan/1000;
|
|
|
|
|
|
// 每秒钟打印一次统计结果
|
|
|
if (totalTime - lastTotalTime >= 1.0) {
|
|
|
int currentFrameCount = frameCount - lastFrameCount;
|
|
|
double currentTotalTime = totalTime - lastTotalTime;
|
|
|
avgFps = currentFrameCount / currentTotalTime;
|
|
|
avgInterval = currentTotalTime * 1000 / currentFrameCount;
|
|
|
std::cout << "Average FPS (last 1 second): " << avgFps << std::endl;
|
|
|
std::cout << "Average Interval (last 1 second): " << avgInterval << " ms" << std::endl;
|
|
|
|
|
|
// 更新上次统计的帧数和时间
|
|
|
lastFrameCount = frameCount;
|
|
|
lastTotalTime = totalTime;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 关闭套接字
|
|
|
socket.close();
|
|
|
|
|
|
// 关闭 ZeroMQ 上下文
|
|
|
context.close();
|
|
|
|
|
|
return 0;
|
|
|
}
|