license.cpp

#include <face_sdk_base/exception.h>
#include <face_sdk_base/license.h>
 
#include <vector>
#include <iostream>
#include <fstream>
 
const std::string output_file_name = "face_sdk.req";
 
std::vector<uint32_t> parse_gpu_list(int argc, char **argv)
{
    std::vector<uint32_t> gpu_list;
 
    for (int i = 1; i < argc; ++i) {
        gpu_list.push_back(std::stoul(argv[i], nullptr, 0));
    }
 
    return gpu_list;
}
 
int main(int argc, char **argv)
{
    std::vector<uint32_t> gpu_list;
    try {
        gpu_list = parse_gpu_list(argc, argv);
    }
    catch (const std::exception &) {
        std::cerr << "Usage: " << argv[0] << " [list of GPU ids]" << std::endl;
        return 1;
    }
 
    if (gpu_list.empty()) {
        std::cout << "Note: making CPU-only license request" << std::endl;
    }
 
    std::vector<uint8_t> request;
    try {
        request = face_sdk::generate_license_request(gpu_list);
    }
    catch (const face_sdk::exception &exp) {
        std::cerr << face_sdk::format_exception(exp) << std::endl;
        return 1;
    }
 
    std::ofstream stream(output_file_name, std::ios::binary);
    stream.write((char*)request.data(), request.size());
    std::cout << "Request is saved in " << output_file_name << std::endl;
 
    return 0;
}