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
|
// 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 <mutex>
#include <string>
#include <vector>
#include "mmd_wrapper.h"
#include "stream_controller_messages.h"
template <class T>
struct Payload : public T {
void* GetPayload() { return this; }
size_t GetSize() { return sizeof(*this); }
};
class BusyFlag {
public:
bool Lock();
void Release();
private:
std::recursive_mutex _mutex;
bool _busy = false;
};
class BusyCheck {
public:
BusyCheck(BusyFlag& busyFlag);
~BusyCheck();
operator bool();
private:
BusyFlag& _busyFlag;
bool _haveLocked;
};
class StreamControllerComms {
public:
StreamControllerComms();
bool IsPresent();
Payload<StatusMessagePayload> GetStatus();
std::string GetStatusString(Payload<StatusMessagePayload>& statusPayload);
bool ScheduleItems(std::vector<Payload<CoreDlaJobPayload>> items);
bool Ping();
bool Initialize(uint32_t sourceBufferSize, uint32_t dropSourceBuffers, uint32_t numInferenceRequests);
private:
bool StatusMessageHandler(uint32_t payloadOffset);
MessageType ReceiveMessage();
bool SendMessage(MessageType, void* pPayload = nullptr, size_t size = 0);
MmdWrapper _mmdWrapper;
uint32_t _lastReceiveSequenceID = 0;
uint32_t _sendSequenceID = 0;
uint32_t _numBadMessages = 0;
const int _streamControllerInstance = 0;
Payload<StatusMessagePayload> _receivedStatusMessage;
BusyFlag _busyFlag;
};
|
