blob: 396a41e4643f5601a81030bf99c41e2f085562c9 (
plain)
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
|
#include "dynamic_depth/depth_jpeg.h"
#include <fstream>
#include <sstream>
#include "android-base/logging.h"
#include "dynamic_depth/container.h"
#include "dynamic_depth/device.h"
#include "dynamic_depth/dynamic_depth.h"
#include "dynamic_depth/item.h"
#include "image_io/gcontainer/gcontainer.h"
#include "xmpmeta/xmp_data.h"
#include "xmpmeta/xmp_parser.h"
#include "xmpmeta/xmp_writer.h"
using ::dynamic_depth::xmpmeta::XmpData;
namespace dynamic_depth {
int32_t ValidateAndroidDynamicDepthBuffer(const char* buffer, size_t buffer_length) {
XmpData xmp_data;
const string image_data(buffer, buffer_length);
ReadXmpFromMemory(image_data, /*XmpSkipExtended*/ false, &xmp_data);
// Check device presence
std::unique_ptr<Device> device = Device::FromXmp(xmp_data);
if (device == nullptr) {
LOG(ERROR) << "Dynamic depth device element not present!";
return -1;
}
// Check profiles
const Profiles* profiles = device->GetProfiles();
if (profiles == nullptr) {
LOG(ERROR) << "No Profile found in the dynamic depth metadata";
return -1;
}
const std::vector<const Profile*> profile_list = profiles->GetProfiles();
// Stop at the first depth photo profile found.
bool depth_photo_profile_found = false;
int camera_index = 0;
for (auto profile : profile_list) {
depth_photo_profile_found = !profile->GetType().compare("DepthPhoto");
if (depth_photo_profile_found) {
// Use the first one if available.
auto indices = profile->GetCameraIndices();
if (!indices.empty()) {
camera_index = indices[0];
} else {
camera_index = -1;
}
break;
}
}
if (!depth_photo_profile_found || camera_index < 0) {
LOG(ERROR) << "No dynamic depth profile found";
return -1;
}
auto cameras = device->GetCameras();
if (cameras == nullptr || camera_index > cameras->GetCameras().size() ||
cameras->GetCameras()[camera_index] == nullptr) {
LOG(ERROR) << "No camera or depth photo data found";
return -1;
}
auto camera = cameras->GetCameras()[camera_index];
auto depth_map = camera->GetDepthMap();
if (depth_map == nullptr) {
LOG(ERROR) << "No depth map found";
return -1;
}
auto depth_uri = depth_map->GetDepthUri();
if (depth_uri.empty()) {
LOG(ERROR) << "Invalid depth map URI";
return -1;
}
auto depth_units = depth_map->GetUnits();
if (depth_units != dynamic_depth::DepthUnits::kMeters) {
LOG(ERROR) << "Unexpected depth map units";
return -1;
}
auto depth_format = depth_map->GetFormat();
if (depth_format != dynamic_depth::DepthFormat::kRangeInverse) {
LOG(ERROR) << "Unexpected depth map format";
return -1;
}
auto near = depth_map->GetNear();
auto far = depth_map->GetFar();
if ((near < 0.f) || (far < 0.f) || (near > far) || (near == far)) {
LOG(ERROR) << "Unexpected depth map near and far values";
return -1;
}
auto confidence_uri = depth_map->GetConfidenceUri();
if (confidence_uri.empty()) {
LOG(ERROR) << "No confidence URI";
return -1;
}
std::istringstream input_jpeg_stream(std::string(buffer, buffer_length));
std::string depth_payload;
if (!GetItemPayload(device->GetContainer(), depth_uri, input_jpeg_stream, &depth_payload)) {
LOG(ERROR) << "Unable to retrieve depth map";
return -1;
}
if (depth_payload.empty()) {
LOG(ERROR) << "Invalid depth map";
return -1;
}
return 0;
}
} // namespace dynamic_depth
|
