PyDictConversion.cpp

Go to the documentation of this file.

#include "MUQ/Utilities/PyDictConversion.h"
 
using namespace muq::Utilities;
 
void muq::Utilities::AddDictToPtree(pybind11::dict dict, std::string basePath, boost::property_tree::ptree &pt) {
  pybind11::object keys = pybind11::list(dict.attr("keys")());
  std::vector<std::string> keysCpp = keys.cast<std::vector<std::string>>();
 
  for(auto& key : keysCpp){
 
    // Recursively add dictionaries
    if(pybind11::isinstance<pybind11::dict>(dict.attr("get")(key))){
      AddDictToPtree(dict.attr("get")(key), basePath + key + ".", pt);
 
    // Convert lists in the comma-separated strings
    }else if(pybind11::isinstance<pybind11::list>(dict.attr("get")(key))){
      std::string val = "";
      for(auto comp : pybind11::list(dict.attr("get")(key)))
        val += "," + std::string(pybind11::str(comp));
      pt.put(basePath + key, val.substr(1));
 
    // Add all the other objects through their "str" interpretation
    }else{
      pt.put(basePath + key, pybind11::str(dict.attr("get")(key)));
    }
  }
}
 
boost::property_tree::ptree muq::Utilities::ConvertDictToPtree(pybind11::dict dict) {
 
  boost::property_tree::ptree pt;
 
  AddDictToPtree(dict, "", pt);
 
  return pt;
}