EnumeratorHandler_impl.hh

Go to the documentation of this file.

#include "casm/app/EnumeratorHandler.hh"
#include "casm/app/ProjectSettings.hh"

#include "casm/container/Enumerator.hh"
#include "casm/clex/PrimClex.hh"
#include "casm/app/casm_functions.hh"
#include "casm/completer/Handlers.hh"

namespace CASM {

  template<typename EnumeratorMapInserter, typename RuntimeLibInserter>
  std::pair<EnumeratorMapInserter, RuntimeLibInserter>
  load_enumerator_plugins(
    const ProjectSettings &set,
    EnumeratorMapInserter enum_it,
    RuntimeLibInserter lib_it) {

    const DirectoryStructure &dir = set.dir();

    if(dir.root_dir().empty()) {
      return std::make_pair(enum_it, lib_it);
    }

    if(fs::is_directory(dir.enumerator_plugins())) {

      // loop over custom enumerator files *.cc
      for(auto &entry : boost::make_iterator_range(fs::directory_iterator(dir.enumerator_plugins()), {})) {

        fs::path p = entry.path();
        std::string p_s = p.string();
        auto p_size = p_s.size();

        if(fs::is_regular_file(p) && p_s.compare(p_size - 3, p_size, ".cc") == 0) {

          fs::path f = p.filename();
          std::string f_s = f.string();
          auto f_size = f_s.size();

          std::string msg = "compiling new custom enumerator: " + f_s.substr(0, f_size - 3);

          // '-L$CASM_PREFIX/.libs' is a hack so 'make check' works
          auto lib_ptr = std::make_shared<RuntimeLibrary>(
                           p_s.substr(0, p_size - 3),
                           set.compile_options() + " " + include_path(dir.enumerator_plugins()),
                           set.so_options() + " -lcasm ",
                           msg,
                           set);

          auto make_interface = lib_ptr->get_function<EnumInterfaceBase* ()>(
                                  "make_" + f_s.substr(0, f_size - 3) + "_interface");

          std::unique_ptr<EnumInterfaceBase> ptr(make_interface());

          // will clone on insert
          *enum_it++ = *ptr;
          *lib_it++ = std::make_pair(ptr->name(), lib_ptr);
        }
      }
    }

    return std::make_pair(enum_it, lib_it);
  }

}