tools.cpp

Go to the documentation of this file.

//
// Copyright (c) 2018, University of Edinburgh
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
//  * Redistributions of source code must retain the above copyright notice,
//    this list of conditions and the following disclaimer.
//  * Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
//  * Neither the name of  nor the names of its contributors may be used to
//    endorse or promote products derived from this software without specific
//    prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
//

#include <cxxabi.h>  // The demangler for gcc... this makes this system dependent!
#include <fstream>
#include <iostream>
#include <random>
#include <regex>
#include <typeinfo>  // The run-time type information (RTTI) Functionality of C++

#include <geometric_shapes/shapes.h>
#include <octomap/OcTree.h>
#include "exotica_core/tools.h"

#include <ros/package.h>

namespace exotica
{
std_msgs::ColorRGBA RandomColor()
{
    std_msgs::ColorRGBA ret;
    ret.a = 1.0;
    std::random_device rd_;
    std::mt19937 gen(rd_());
    std::uniform_real_distribution<> dis(0.0, 1.0);
    ret.r = static_cast<float>(dis(gen));
    ret.g = static_cast<float>(dis(gen));
    ret.b = static_cast<float>(dis(gen));
    return ret;
}

void SaveMatrix(std::string file_name,
                const Eigen::Ref<const Eigen::MatrixXd> mat)
{
    std::ofstream myfile;
    myfile.open(file_name);
    if (myfile.good())
    {
        myfile << mat;
        myfile.close();
    }
    else
    {
        myfile.close();
        ThrowPretty("Can't open file!");
    }
}

void LoadOBJ(const std::string& data, Eigen::VectorXi& tri,
             Eigen::VectorXd& vert)
{
    std::stringstream ss(data);
    std::string line;
    tri.resize(0, 1);
    vert.resize(0, 1);
    int vn = 0, tn = 0;
    double v[3];
    int vv[9];
    while (std::getline(ss, line))
    {
        if (line.compare(0, 2, "v ") == 0)
        {
            vert.conservativeResize((vn + 1) * 3);
            std::stringstream sss(line.substr(2));
            sss >> v[0] >> v[1] >> v[2];
            vert(vn * 3) = v[0];
            vert(vn * 3 + 1) = v[1];
            vert(vn * 3 + 2) = v[2];
            ++vn;
        }
        else if (line.compare(0, 2, "f ") == 0)
        {
            std::stringstream sss(line.substr(2));
            int i;
            for (i = 0; i < 9 && sss >> vv[i]; ++i)
            {
                while (sss.peek() == '/' || sss.peek() == ' ')
                    sss.ignore();
            }
            if (i < 8)
            {
                ThrowPretty("Invalid format!");
            }
            tri.conservativeResize((tn + 1) * 3);
            tri(tn * 3) = vv[0] - 1;
            tri(tn * 3 + 1) = vv[3] - 1;
            tri(tn * 3 + 2) = vv[6] - 1;
            ++tn;
        }
    }
}

std::shared_ptr<octomap::OcTree> LoadOctree(const std::string& file_path)
{
    std::shared_ptr<octomap::OcTree> octree(new octomap::OcTree(file_path));
    if (!octree) ThrowPretty("Could not load OcTree!");
    return octree;
}

std::shared_ptr<shapes::Shape> LoadOctreeAsShape(const std::string& file_path)
{
    std::shared_ptr<octomap::OcTree> octree = LoadOctree(file_path);
    std::shared_ptr<shapes::Shape> shape(new shapes::OcTree(octree));
    return shape;
}

std::string GetTypeName(const std::type_info& type)
{
    int status;
    std::string name;
    char* temp;  

    temp = abi::__cxa_demangle(type.name(), 0, 0, &status);
    name = std::string(temp);
    free(temp);
    return name;
}

std::string ParsePath(const std::string& path)
{
    std::string ret = path;
    std::smatch matches;
    std::regex_search(ret, matches, std::regex("\\{([^\\}]+){1,}\\}"));
    for (auto& match : matches)
    {
        std::string package = match.str();
        if (package[0] == '{' || package == "") continue;
        std::string package_path = ros::package::getPath(package);
        if (package_path == "") ThrowPretty("Unknown package '" << package << "'");
        try
        {
            ret = std::regex_replace(ret, std::regex("\\{" + package + "\\}"), package_path, std::regex_constants::match_any);
        }
        catch (const std::regex_error& e)
        {
            ThrowPretty("Package name resolution failed (regex error " << e.code() << ")");
        }
    }
    std::regex_search(ret, matches, std::regex("package://([^\\/]+){1,}"));
    for (auto& match : matches)
    {
        std::string package = match.str();
        if (package.substr(0, 10) == "package://" || package == "") continue;
        std::string package_path = ros::package::getPath(package);
        if (package_path == "") ThrowPretty("Unknown package '" << package << "'");
        try
        {
            ret = std::regex_replace(ret, std::regex("package://" + package + "/"), package_path + "/", std::regex_constants::match_any);
        }
        catch (const std::regex_error& e)
        {
            ThrowPretty("Package name resolution failed (regex error " << e.code() << ")");
        }
    }
    return ret;
}

std::string LoadFile(const std::string& path)
{
    std::string file_name = ParsePath(path);
    std::ifstream fstream(file_name);
    if (!fstream) ThrowPretty("File does not exist '" << file_name << "'");
    try
    {
        return std::string((std::istreambuf_iterator<char>(fstream)), std::istreambuf_iterator<char>());
    }
    catch (const std::ifstream::failure& e)
    {
        ThrowPretty("Can't read file '" << file_name << "'");
    }
}

bool PathExists(const std::string& path)
{
    std::ifstream file(ParsePath(path));
    return (bool)file;
}
}  // namespace exotica