integration/luprex/core/cpp/util.cpp

#include <string>
#include <vector>
#include <fstream>
#include <cstdlib>
#include "util.hpp"
#include <sys/types.h>
#include <sys/stat.h>
#include <cmath>
#include <iostream>
#include <iomanip>

#ifndef WIN32
#include <unistd.h>
#define stat _stat
#endif

namespace util {
    
IdVector sort_union_id_vectors(const IdVector &v1, const IdVector &v2) {
    IdVector result(v1.size() + v2.size());
    int next = 0;
    for (int64_t id : v1) result[next++] = id;
    for (int64_t id : v2) result[next++] = id;
    std::sort(result.begin(), result.end());
    int64_t prev = -1;
    int64_t count = 0;
    for (int64_t id : result) {
        if (id != prev) {
            prev = id;
            result[count++] = id;
        }
    }
    result.resize(count);
    return result;
}

StringVec split(const std::string &s, char sep) {
    StringVec result;
    int start = 0;
    for (int i = 0; i < int(s.size()); i++) {
        if (s[i] == sep) {
            result.push_back(s.substr(start, i-start));
            start = i+1;
        }
    }
    if (start < int(s.size())) {
        result.push_back(s.substr(start));
    }
    return result;
}

std::string tolower(std::string input) {
    for (int i = 0; i < int(input.size()); i++) {
        input[i] = std::tolower(input[i]);
    }
    return input;
}

std::string toupper(std::string input) {
    for (int i = 0; i < int(input.size()); i++) {
        input[i] = std::toupper(input[i]);
    }
    return input;
}

bool validinteger(const std::string &value) {
    char *endptr;
    if (value.size() == 0) return false;
    strtoll(value.c_str(), &endptr, 10);
    return (endptr == value.c_str() + value.size());
}

int64_t strtoint(const std::string &value, int64_t errval) {
    char *endptr;
    if (value.size() == 0) return errval;
    int64_t result = strtoll(value.c_str(), &endptr, 10);
    if (endptr == value.c_str() + value.size()) {
        return result;
    } else {
        return errval;
    }
}

double strtodouble(const std::string &value) {
    char *endptr;
    if (value.size() == 0) return std::nan("");
    double result = strtod(value.c_str(), &endptr);
    if (endptr == value.c_str() + value.size()) {
        return result;
    } else {
        return std::nan("");
    }
}

std::string ltrim(std::string s) {
    s.erase(s.begin(), std::find_if(s.begin(), s.end(),
            std::not1(std::ptr_fun<int, int>(std::isspace))));
    return s;
}

std::string rtrim(std::string s) {
    s.erase(std::find_if(s.rbegin(), s.rend(),
            std::not1(std::ptr_fun<int, int>(std::isspace))).base(), s.end());
    return s;
}

std::string trim(std::string s) {
    return ltrim(rtrim(s));
}

double distance_squared(double x1, double y1, double x2, double y2) {
    double dx = x1 - x2;
    double dy = y1 - y2;
    return dx*dx + dy*dy;
}

std::string get_file_contents(const std::string &fn) {
    std::ifstream fs(fn);
    std::stringstream buffer;
    buffer << fs.rdbuf();
    return buffer.str();
}

StringVec get_file_lines(const std::string &path) {
    StringVec result;
    std::ifstream f;
    f.open(path);
    if (f) {
        std::string line;
        while (std::getline(f, line)) {
            result.push_back(line);
        }
        f.close();
    }
    return result;
}

std::string get_file_fingerprint(const std::string &fn) {
    struct stat result;
    if(stat(fn.c_str(), &result)==0)
    {
        std::stringstream ss;
        ss << result.st_mtime;
        return ss.str();
    }
    return "";
}

std::string XYZ::debug_string() const {
    std::ostringstream oss;
    oss << "(" << x << "," << y << "," << z << ")";
    return oss.str();
}

} // namespace util

std::ostream &operator<<(std::ostream &oss, const util::hex64 &v) {
    oss << "0x" << std::setw(16) << std::setfill('0') << std::hex;
    return oss;
}

std::ostream &operator<<(std::ostream &oss, const util::hex32 &v) {
    oss << "0x" << std::setw(8) << std::setfill('0') << std::hex;
    return oss;
}

std::ostream &operator<<(std::ostream &oss, const util::hex16 &v) {
    oss << "0x" << std::setw(4) << std::setfill('0') << std::hex;
    return oss;
}

std::ostream &operator<<(std::ostream &oss, const util::hex8 &v) {
    oss << "0x" << std::setw(2) << std::setfill('0') << std::hex;
    return oss;
}

LuaDefine(unittests_util, "c") {
    // Test the unioning of ID vectors.
    util::IdVector idv1,idv2;
    idv1.push_back(1);
    idv1.push_back(6);
    idv1.push_back(4);
    idv2.push_back(5);
    idv2.push_back(1);
    idv2.push_back(6);
    util::IdVector joined = util::sort_union_id_vectors(idv1, idv2);
    LuaAssert(L, joined.size() == 4);
    LuaAssert(L, joined[0] == 1);
    LuaAssert(L, joined[1] == 4);
    LuaAssert(L, joined[2] == 5);
    LuaAssert(L, joined[3] == 6);

    // Test the string split routine.
    util::StringVec sv1 = util::split("foo,bar,baz", ',');
    LuaAssert(L, sv1.size() == 3);
    LuaAssert(L, sv1[0] == "foo");
    LuaAssert(L, sv1[1] == "bar");
    LuaAssert(L, sv1[2] == "baz");
    util::StringVec sv2 = util::split(",foo,,bar", ',');
    LuaAssert(L, sv2.size() == 4);
    LuaAssert(L, sv2[0]=="");
    LuaAssert(L, sv2[1]=="foo");
    LuaAssert(L, sv2[2]=="");
    LuaAssert(L, sv2[3]=="bar");

    // test toupper and tolower
    LuaAssert(L, util::toupper("fooBar") == "FOOBAR");
    LuaAssert(L, util::tolower("fooBar") == "foobar");

    // test validinteger, strtoint, strtodouble
    LuaAssert(L, util::validinteger("123") == true);
    LuaAssert(L, util::validinteger("123.4") == false);
    LuaAssert(L, util::validinteger("12ab") == false);
    LuaAssert(L, util::validinteger("") == false);
    LuaAssert(L, util::strtoint("123", -5) == 123);
    LuaAssert(L, util::strtoint("123.4", -5) == -5);
    LuaAssert(L, util::strtoint("12ab", -5) == -5);
    LuaAssert(L, util::strtoint("", -5) == -5);
    LuaAssert(L, util::strtodouble("123.5") == 123.5);
    LuaAssert(L, std::isnan(util::strtodouble("12ab")));
    LuaAssert(L, std::isnan(util::strtodouble("")));
    
    // Test trim, ltrim, rtrim
    LuaAssert(L, util::ltrim("  foo  ") == "foo  ");
    LuaAssert(L, util::rtrim("  foo  ") == "  foo");
    LuaAssert(L, util::trim("  foo  ") == "foo");
    LuaAssert(L, util::trim("foo") == "foo");
    LuaAssert(L, util::trim("") == "");
    
    // Test distance_squared
    LuaAssert(L, util::distance_squared(1, 1, 5, 4) == 25.0);
    LuaAssert(L, util::distance_squared(5, 4, 1, 1) == 25.0);

    // Test XYZ.
    util::XYZ xyza(3,4,5), xyzb(3,4,5), xyzc(3,4,6);
    LuaAssert(L, xyza.x == 3);
    LuaAssert(L, xyza.y == 4);
    LuaAssert(L, xyza.z == 5);
    LuaAssert(L, xyza == xyzb);
    LuaAssert(L, xyza != xyzc);
    LuaAssert(L, xyza.debug_string() == "(3,4,5)");
    
    return 0;
}