FileNameComponents.cc

Go to the documentation of this file.

#include "art/Framework/IO/detail/FileNameComponents.h"
#include "boost/regex.hpp"

#include <cassert>
#include <iomanip>
#include <sstream>

namespace art::detail {

  void
  FileNameComponents::add(std::string const& prefix,
                          std::string const& digitFormat)
  {
    components_.emplace_back(prefix, digitFormat);
  }

  void
  FileNameComponents::setSuffix(std::string suffix)
  {
    suffix_ = move(suffix);
  }

  bool
  FileNameComponents::operator<(FileNameComponents const& fnc) const
  {
    if (components_.size() < fnc.components_.size()) {
      return true;
    }
    if (components_.size() > fnc.components_.size()) {
      return false;
    }

    assert(components_.size() == fnc.components_.size());
    auto b1 = cbegin(components_), e1 = cend(components_);
    auto b2 = cbegin(fnc.components_);
    for (; b1 != e1; ++b1, ++b2) {
      if (b1->first < b2->first)
        return true;
    }
    assert(b2 == cend(fnc.components_));
    return false;
  }

  std::string
  FileNameComponents::fileNameWithIndex(std::size_t const index) const
  {
    std::ostringstream num_str;
    for (auto const& pr : components_) {
      auto const& prefix = pr.first;
      auto const& specifier = pr.second;
      num_str << prefix;
      if (!specifier.empty()) { // Zero-filling.
        num_str << std::setfill('0') << std::setw(std::stoul(specifier));
      }
      num_str << index;
    }
    num_str << suffix_;
    return num_str.str();
  }

  FileNameComponents
  componentsFromPattern(std::string const& pattern)
  {
    FileNameComponents result;

    boost::smatch match;
    auto sid = cbegin(pattern), se = cend(pattern);
    while (boost::regex_search(
      sid,
      se,
      match,
      boost::regex{"%(\\d+)?#", boost::regex_constants::ECMAScript})) {
      assert(match.size() == 2);
      // Subexpressions:
      //   0. Entire matched expression
      //   1. Possible fill format digits for numeric substitution.
      result.add(match.prefix(), match[1].str());
      sid = match[0].second;
    }
    // Get remaining characters of filename
    result.setSuffix(std::string(sid, se));
    return result;
  }
}