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
|
// Copyright 2023 Intel Corporation.
//
// This software and the related documents are Intel copyrighted materials,
// and your use of them is governed by the express license under which they
// were provided to you ("License"). Unless the License provides otherwise,
// you may not use, modify, copy, publish, distribute, disclose or transmit
// this software or the related documents without Intel's prior written
// permission.
//
// This software and the related documents are provided as is, with no express
// or implied warranties, other than those that are expressly stated in the
// License.
#pragma once
#include <condition_variable>
#include <filesystem>
#include <vector>
#include <semaphore.h>
#include "ILayoutTransform.h"
#include "command_line.h"
using namespace std::chrono_literals;
class RawImage;
class Event {
public:
void Wait() {
std::unique_lock<std::mutex> lock(_signalMutex);
_conditionVariable.wait(lock);
}
void Set() {
std::lock_guard<std::mutex> lock(_signalMutex);
_conditionVariable.notify_all();
}
bool IsSignalled() {
std::unique_lock<std::mutex> lock(_signalMutex);
return (_conditionVariable.wait_for(lock, 0ms) != std::cv_status::timeout);
}
private:
std::mutex _signalMutex;
std::condition_variable _conditionVariable;
};
class ImageStreamingApp {
public:
ImageStreamingApp(int numParams, char* paramValues[]);
void Run();
private:
bool ProgramLayoutTransform();
bool SendNextImage();
bool LoadImageFiles(bool dumpLayoutTransform);
void RunSendImageSignalThread();
static void SigIntHandler(int);
uint32_t GetUintOption(const char* optionName, uint32_t defaultValue);
float GetFloatOption(const char* optionName, float defaultValue);
bool OpenMsgDmaStream();
void CloseMsgDmaStream();
bool WaitForInferenceApp();
CommandLine _commandLine;
std::filesystem::path _imageFilesFolder;
std::string _imageFile;
std::vector<std::shared_ptr<RawImage>> _images;
Event _sendNextImageEvent;
static volatile bool _shutdownEvent;
size_t _nextImageIndex = 0;
uint32_t _numToSend = 0;
uint32_t _sendRate = 30;
uint32_t _sentCount = 0;
bool _dumpTransformedImages = false;
bool _runLayoutTransform = true;
FILE* _msgDmaStream = nullptr;
ILayoutTransform::Configuration _ltConfiguration = {};
};
|
