#include "basic/util/resource/ResourcesManager.h"
|
#include <string>
|
#include <iostream>
|
#include <unistd.h>
|
#include <thread>
|
|
#include <basic/pipe/PipeElement.h>
|
#include <basic/pipe_element/ImageShowElement.h>
|
#include <basic/pipe/PipeController.h>
|
#include <basic/pipe_element/ffmpeg/VideoCaptureElement.h>
|
#include "ImageDrawElement.h"
|
#include "YoloRpcElement.h"
|
class Controllor : public PipeController
|
{
|
public:
|
Controllor(const int index,const std::string& rtsp):m_videoCaptureElement(rtsp,25,1000,0),
|
m_YoloRpcElement(std::to_string(index)+"YoloRpc")
|
{
|
m_rtsp = rtsp;
|
m_videoCaptureElement.registerConnector([&]{
|
if(!m_YoloRpcElement.isBusy()){
|
m_YoloRpcElement.setImage(m_videoCaptureElement.getImage());
|
// paElement.setImage(videoCaptureElement.getImage());
|
// paNumElement.setImage(videoCaptureElement.getImage());
|
m_YoloRpcElement.submit();
|
}
|
if(!m_imageDrawElement.isBusy()){
|
m_imageDrawElement.setImage(m_videoCaptureElement.getImage());
|
m_imageDrawElement.submit();
|
}
|
});
|
|
m_YoloRpcElement.registerConnector([&]{
|
m_imageDrawElement.setRects(m_YoloRpcElement.getRects());
|
});
|
|
m_imageDrawElement.registerConnector([&]{
|
ImageShowElement::showImage(m_rtsp,*m_imageDrawElement.getImage());
|
|
});
|
m_videoCaptureElement.setOutPutInterval(2);
|
registerElement(m_videoCaptureElement);
|
registerElement(m_YoloRpcElement);
|
registerElement(m_imageDrawElement);
|
}
|
virtual ~Controllor()
|
{}
|
|
private:
|
std::string m_rtsp;
|
ffmpeg::VideoCaptureElement m_videoCaptureElement;
|
YoloRpcElement m_YoloRpcElement;
|
ImageDrawElement m_imageDrawElement;
|
};
|
|
//using namespace std;
|
int main(int argc, char *argv[])
|
{
|
SAVE_APP_ARGS
|
Controllor _Controllor(0,"/home/basic/video/201804/day0416/01010002606000000-slice1.mp4");
|
_Controllor.start();
|
|
Controllor _Controllor0(1,"/home/basic/video/201804/day0416/01010002606000000-slice2.mp4");
|
_Controllor0.start();
|
cout << "Hello World!" << endl;
|
Controllor _Controllor1(2,"/home/basic/video/201804/day0416/01010002609000000-slice.mp4");
|
_Controllor1.start();
|
Controllor _Controllor2(3,"/home/basic/video/201804/day0416/01010002611000000-slice.mp4");
|
_Controllor2.start();
|
Controllor _Controllor3(4,"/home/basic/video/201804/night0418/02000000652000100-slice.mp4");
|
_Controllor3.start();
|
Controllor _Controllor4(5,"/home/basic/video/201804/night0418/02000000654000100-slice.mp4");
|
_Controllor4.start();
|
Controllor _Controllor5(6,"/home/basic/video/201804/night0418/02000000659000100-slice.mp4");
|
_Controllor5.start();
|
while(true)
|
{sleep(1);}
|
return 0;
|
}
|
