forked from seanavery/viam-csi
-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.cpp
49 lines (40 loc) · 1.35 KB
/
utils.cpp
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
#include <iostream>
#include <string>
#define DEVICE_PATH "/proc/device-tree/model"
enum class device_type {
unknown,
jetson,
pi
};
struct os_info {
std::string os;
std::string version;
std::string kernel;
};
device_type get_device_type() {
std::ifstream device_name(DEVICE_PATH);
if (device_name.is_open()) {
std::string line;
while (std::getline(device_name, line)) {
std::string lowercase_line = line;
std::transform(lowercase_line.begin(), lowercase_line.end(), lowercase_line.begin(), ::tolower);
// Check for specific terms in a case-insensitive manner
if (lowercase_line.find("nvidia") != std::string::npos &&
(lowercase_line.find("orin") != std::string::npos ||
lowercase_line.find("nano") != std::string::npos ||
lowercase_line.find("agx") != std::string::npos ||
lowercase_line.find("jetson") != std::string::npos)) {
return device_type::jetson;
} else if (lowercase_line.find("raspberry") != std::string::npos &&
lowercase_line.find("pi") != std::string::npos) {
return device_type::pi;
}
}
device_name.close();
}
return device_type::unknown;
}
os_info get_os() {
// TODO
return os_info{};
}