c++相关的 常用帮助函数集

//
// Created by DangXS on 2022/4/27.
//

#ifndef CPLUSPLUS_PROJECT1_HELPER_H
#define CPLUSPLUS_PROJECT1_HELPER_H

#include <iostream>
#include <cstring>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "dirent.h"
#include <ctime>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;
#define MICRO_IN_SECOND 1000000
#define NANOS_IN_SECOND 1000000000


// 字符串分割到 vector
static vector<string> string_split(const string &str, const string &delim) {
    vector<string> res;
    if ("" == str) return res;
    //先将要切割的字符串从string类型转换为char*类型
    char *strs = new char[str.length() + 1]; //不要忘了
    strcpy(strs, str.c_str());

    char *d = new char[delim.length() + 1];
    strcpy(d, delim.c_str());

    char *p = strtok(strs, d);
    while (p) {
        string s = p; //分割得到的字符串转换为string类型
        res.push_back(s); //存入结果数组
        p = strtok(NULL, d);
    }
    return res;
}

// 获取文件后缀名
static string getExtension(const string &filename) {
    return filename.substr(filename.find_last_of('.'));
}

// 判断是不是图片
static bool isImage(const string &filename) {
    string ext = getExtension(filename);
    vector<string> formats = {".jpg", ".jpeg", ".png", ".ppm", ".bmp", ".pgm", ".tif", ".tiff", ".webp"};
    auto it = find(formats.begin(), formats.end(), ext);
    return it != formats.end();
}

// 获取文件名，不包含目录
static string getFilename(const string &file_fullpath) {
    return file_fullpath.substr(file_fullpath.find_last_of('/') + 1);
}

//获取文件名，不包含目录和后缀名
static string getFilenameNoExt(const string &filename) {
    string _filename = getFilename(filename);
    return _filename.substr(0, _filename.find_last_of('.'));
}

static string getParentPath(const string &file_fullpath) {
    int idx = file_fullpath.find_last_of('/');
    return file_fullpath.substr(0, idx);
}

// 获取文件绝对路径的父目录名
static string getParentDirname(string file_fullpath) {
    vector<string> groups = string_split(file_fullpath, "/");
    return groups[groups.size() - 2];
}

static bool pathIsExisted(const string &path) {
    return access(path.c_str(), 0) == F_OK;
}

static bool isFolder(const string &path) {
    if (!pathIsExisted(path)) {
        printf_s("%s:%d not existed.\n", __FILE__, __LINE__, path.c_str());
        return false;
    }
    struct stat buffer{};
    return (stat(path.c_str(), &buffer)) == 0 && S_ISDIR(buffer.st_mode);

}


static bool isFile(const string &path) {
    if (!pathIsExisted(path)) {
        printf_s("%s:%d not existed.\n", __FILE__, __LINE__, path.c_str());
        return false;
    }

    struct stat buffer{};
    return (stat(path.c_str(), &buffer)) == 0 && S_ISREG(buffer.st_mode);

}


// 递归创建文件夹
static int mkDirsRecurive(string filename) {
    filename += "/";
    char *fileName = (char *) filename.c_str();
    char *tag;
    int a = 0;
    for (tag = fileName; *tag; tag++, a++) {
        if (*tag == '/') {
            string new_path = filename.substr(0, a + 1);

            if (access(new_path.c_str(), 0) == -1) //如果文件夹不存在
//                mkdir(new_path.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH); //则创建
                mkdir(new_path.c_str());
        }
    }

    return 0;
}

//读取某给定路径下所有文件夹与文件名称，并带完整路径
static void getAllFilesInDirs(const string &path, vector<string> &fileFullpath) {

    DIR *pDir;
    struct dirent *ptr;
    if (!(pDir = opendir(path.c_str())))
        return;
    struct stat buffer{};
    while ((ptr = readdir(pDir)) != 0) {
        string new_path = path + "/" + ptr->d_name;
        stat(new_path.c_str(), &buffer);
        if (S_ISDIR(buffer.st_mode)) {
            if (strcmp(ptr->d_name, ".") != 0 && strcmp(ptr->d_name, "..") != 0) {
                getAllFilesInDirs(new_path, fileFullpath);
            }
        } else {
            fileFullpath.push_back(path + "/" + ptr->d_name);
        }
    }
    closedir(pDir);
}

//读取某给定路径下后缀名为format得文件名称，并带完整路径
static void getAllFilesByformat(const string &path, vector<string> &fileFullpath, const string &format) {
    DIR *pDir;
    struct dirent *ptr;
    if (!(pDir = opendir(path.c_str())))
        return;
    while ((ptr = readdir(pDir)) != 0) {
        string new_path = path + "/" + ptr->d_name;
        if (isFolder(new_path)) {
            if (strcmp(ptr->d_name, ".") != 0 && strcmp(ptr->d_name, "..") != 0) {
                getAllFilesByformat(new_path, fileFullpath, format);
            }
        } else if (isFile(new_path) && getExtension(ptr->d_name) == format) {
            string f = path + string("/") + string(ptr->d_name);
//            std::cout << f << std::endl;
            fileFullpath.push_back(f);
        }
    }
    closedir(pDir);
}

// 文件重命名
static int file_rename(string oldname, string newname) {
    return rename(oldname.c_str(), newname.c_str());
}

// 文件移动
static int file_move(const string &src_filename, const string &dst_path) {
    string command = "mv ";
    command += src_filename;
    command += " ";
    command += dst_path; //mv /home/file1 /root/file2
    system((char *) command.c_str());
    return 0;
}

// 文件拷贝
static int file_copy(string src_filename, string dst_filename) {
    string command = "cp ";
    command += src_filename.c_str();
    command += " ";
    command += dst_filename.c_str(); //cp /home/file1 /root/file2
    // printf("%s\n", command.c_str());
    system((char *) command.c_str());
    return 0;
}

// 打乱vector
static vector<string> vector_shuffle(vector<string> v1) {
    vector<string> v2;
// 复制
    v2.assign(v1.begin(), v1.end());
//打乱
    random_shuffle(v2.begin(), v2.end());

    return v2;
}

///替换指定字符串
static string replace_all(string str, const string &old_value, const string &new_value) {
    string str1;
    while (true) {
        string::size_type pos(0);
        if ((pos = str.find(old_value)) != string::npos)
            str1 = str.replace(pos, old_value.length(), new_value);
        else break;
    }
    return str1;
}

static std::string GetTime() {
    time_t timep;
    time(&timep);
    char tmp[64];
    strftime(tmp, sizeof(tmp), "%Y-%m-%d %H:%M:%S", localtime(&timep));
    return tmp;
}

// 获取当前毫秒数
// 用于测试程序耗时
// elapsed time = (t2-t1)
static double GetCurrentMiliSec() {
    struct timespec res{};
    double ret = 0;
    clock_gettime(CLOCK_MONOTONIC, &res);
    ret = (double) (res.tv_sec * NANOS_IN_SECOND + res.tv_nsec) / MICRO_IN_SECOND;

    return ret;
}

static bool cmp(const pair<int, float> &a, const pair<int, float> &b) {
    return a.second > b.second;
}

#endif //CPLUSPLUS_PROJECT1_HELPER_H