//
// Created by Scheaven on 2020/11/2.
//

#include "type_utils.h"
using namespace std;
/**
import  torch

a=torch.Tensor([[1,2,3],[4,5,6]])
b=torch.Tensor([[7,8,9],[10,11,12]])
d=torch.stack( (a,b) ,dim = 1)

print(d)

template<typename T>
unsigned char * ReID_Utils::T2bytes(T u)
{
    int n = sizeof(T);
    unsigned char* b = new unsigned char[n];
    memcpy(b, &u, n);
    return b;
}

template<typename T>
T ReID_Utils::bytes2T(unsigned char *bytes)
{
    T res = 0;
    int n = sizeof(T);
    memcpy(&res, bytes, n);
    return res;
}

*/
unsigned char * ReID_Utils::T2bytes(float u)
{
    int n = sizeof(u);
    unsigned char* b = new unsigned char[n];
    memcpy(b, &u, n);
    return b;
}

float ReID_Utils::bytes2T(unsigned char *bytes)
{
    float res = 0;
    int n = sizeof(res);
    memcpy(&res, bytes, n);
    return res;
}

char* get_subStr(std::string line, const char* split, int which)
{
    char* input = (char*)line.c_str();
    char *p = strtok(input,split);
    char* sub_str;
    int local = 0;
    while( p !=NULL )
    {
//        sscanf(p, "%s", &sub_str);
        if(local == which)
        {
            std::cout <<p<< std::endl;
            return p;
        }
        p = strtok(NULL, split);
        local++;
    }

    if(which>local)
        return "";
}

//int main(int argc, char *argv[])
//{
//    std::string line = "18eadadc-f583-4517-8c5d-e2a8cb605f97___FaceDetect___Yolo";
//    std::cout << get_subStr(line, "_", 0) << std::endl;
//    return 0;
//}