/
MPFDetectionComponent.h
225 lines (187 loc) · 8.57 KB
/
MPFDetectionComponent.h
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
/******************************************************************************
* NOTICE *
* *
* This software (or technical data) was produced for the U.S. Government *
* under contract, and is subject to the Rights in Data-General Clause *
* 52.227-14, Alt. IV (DEC 2007). *
* *
* Copyright 2023 The MITRE Corporation. All Rights Reserved. *
******************************************************************************/
/******************************************************************************
* Copyright 2023 The MITRE Corporation *
* *
* Licensed under the Apache License, Version 2.0 (the "License"); *
* you may not use this file except in compliance with the License. *
* You may obtain a copy of the License at *
* *
* http://www.apache.org/licenses/LICENSE-2.0 *
* *
* Unless required by applicable law or agreed to in writing, software *
* distributed under the License is distributed on an "AS IS" BASIS, *
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
* See the License for the specific language governing permissions and *
* limitations under the License. *
******************************************************************************/
#ifndef OPENMPF_CPP_COMPONENT_SDK_DETECTION_BASE_H
#define OPENMPF_CPP_COMPONENT_SDK_DETECTION_BASE_H
#include <map>
#include <string>
#include <utility>
#include <vector>
#include "MPFComponentInterface.h"
#include "MPFDetectionObjects.h"
namespace MPF { namespace COMPONENT {
struct MPFJob {
const std::string job_name;
const std::string data_uri;
const Properties job_properties;
const Properties media_properties;
protected:
MPFJob(std::string job_name,
std::string data_uri,
Properties job_properties,
Properties media_properties)
: job_name(std::move(job_name))
, data_uri(std::move(data_uri))
, job_properties(std::move(job_properties))
, media_properties(std::move(media_properties)) {
}
};
struct MPFVideoJob : MPFJob {
const int start_frame;
const int stop_frame;
const bool has_feed_forward_track;
const MPFVideoTrack feed_forward_track;
MPFVideoJob(std::string job_name,
std::string data_uri,
int start_frame,
int stop_frame,
Properties job_properties,
Properties media_properties)
: MPFJob(std::move(job_name),
std::move(data_uri),
std::move(job_properties),
std::move(media_properties))
, start_frame(start_frame)
, stop_frame(stop_frame)
, has_feed_forward_track(false) {
}
MPFVideoJob(std::string job_name,
std::string data_uri,
int start_frame,
int stop_frame,
MPFVideoTrack track,
Properties job_properties,
Properties media_properties)
: MPFJob(std::move(job_name),
std::move(data_uri),
std::move(job_properties),
std::move(media_properties))
, start_frame(start_frame)
, stop_frame(stop_frame)
, has_feed_forward_track(true)
, feed_forward_track(std::move(track)) {
}
};
struct MPFImageJob : MPFJob {
const bool has_feed_forward_location;
const MPFImageLocation feed_forward_location;
MPFImageJob(std::string job_name,
std::string data_uri,
Properties job_properties,
Properties media_properties)
: MPFJob(std::move(job_name),
std::move(data_uri),
std::move(job_properties),
std::move(media_properties))
, has_feed_forward_location(false) {
}
MPFImageJob(std::string job_name,
std::string data_uri,
MPFImageLocation location,
Properties job_properties,
Properties media_properties)
: MPFJob(std::move(job_name),
std::move(data_uri),
std::move(job_properties),
std::move(media_properties))
, has_feed_forward_location(true)
, feed_forward_location(std::move(location)) {
}
};
struct MPFAudioJob : MPFJob {
const int start_time;
const int stop_time;
const bool has_feed_forward_track;
const MPFAudioTrack feed_forward_track;
MPFAudioJob(std::string job_name,
std::string data_uri,
int start_time,
int stop_time,
Properties job_properties,
Properties media_properties)
: MPFJob(std::move(job_name),
std::move(data_uri),
std::move(job_properties),
std::move(media_properties))
, start_time(start_time)
, stop_time(stop_time)
, has_feed_forward_track(false) {
}
MPFAudioJob(std::string job_name,
std::string data_uri,
int start_time,
int stop_time,
MPFAudioTrack track,
Properties job_properties,
Properties media_properties)
: MPFJob(std::move(job_name),
std::move(data_uri),
std::move(job_properties),
std::move(media_properties))
, start_time(start_time)
, stop_time(stop_time)
, has_feed_forward_track(true)
, feed_forward_track(std::move(track)) {
}
};
struct MPFGenericJob : MPFJob {
const bool has_feed_forward_track;
const MPFGenericTrack feed_forward_track;
MPFGenericJob(std::string job_name,
std::string data_uri,
Properties job_properties,
Properties media_properties)
: MPFJob(std::move(job_name),
std::move(data_uri),
std::move(job_properties),
std::move(media_properties))
, has_feed_forward_track(false) {
}
MPFGenericJob(std::string job_name,
std::string data_uri,
MPFGenericTrack track,
Properties job_properties,
Properties media_properties)
: MPFJob(std::move(job_name),
std::move(data_uri),
std::move(job_properties),
std::move(media_properties))
, has_feed_forward_track(true)
, feed_forward_track(std::move(track)) {
}
};
// Class used for batch processing detection jobs, any type of media.
class MPFDetectionComponent : public MPFComponent {
public:
virtual std::vector<MPFVideoTrack> GetDetections(const MPFVideoJob &job) = 0;
virtual std::vector<MPFImageLocation> GetDetections(const MPFImageJob &job) = 0;
virtual std::vector<MPFAudioTrack> GetDetections(const MPFAudioJob &job) = 0;
virtual std::vector<MPFGenericTrack> GetDetections(const MPFGenericJob &job) = 0;
virtual bool Supports(MPFDetectionDataType data_type) = 0;
MPFComponentType GetComponentType() override { return MPF_DETECTION_COMPONENT; };
protected:
MPFDetectionComponent() = default;
};
}}
#endif //OPENMPF_CPP_COMPONENT_SDK_DETECTION_BASE_H