run_index.h

#pragma once
 
#include "Rtypes.h"
#include <optional>
#include <iostream>
#include "KVString.h"
 
class run_index_t : public std::pair<int, std::optional<int>> {
public:
   run_index_t(int run, std::optional<int> index)
      : std::pair<int, std::optional<int>>(run, index)
   {
      if (index.has_value() && index.value() == -1)
         second = std::nullopt;
   }
   run_index_t() = default;
   run_index_t(const run_index_t&) = default;
   run_index_t(run_index_t&&) = default;
   run_index_t& operator=(const run_index_t&) = default;
   run_index_t& operator=(run_index_t&&) = default;
   run_index_t(const TString &rs)
   {
      set(rs);
   }
 
   int run() const
   {
      return first;
   }
   int index(int no_index = -1) const
   {
      return second.value_or(no_index);
   }
   bool has_index() const
   {
      return second.has_value();
   }
   void set_run(int r)
   {
      first = r;
   }
   void set_index(int i)
   {
      second = i;
   }
 
   void clear()
   {
      first = 0;
      second = std::nullopt;
   }
   void set(const TString& rs)
   {
      clear();
      KVString n(rs);
      n.Begin(".");
      while (!n.End()) {
         if (!run())
            set_run(n.Next().Atoi());
         else
            set_index(n.Next().Atoi());
      }
   }
   TString as_string() const
   {
      TString s = Form("%d", run());
      if (has_index()) s += "." + TString(Form("%d", index()));
      return s;
   }
   TString index_string() const
   {
      if (has_index()) return Form("%d", index());
      return "-";
   }
   friend std::ostream& operator<<(std::ostream& out, const run_index_t& rndx)
   {
      return out << rndx.as_string();
   }
 
   friend bool operator<(const run_index_t& A, const run_index_t& B)
   {
      if (A.run() < B.run()) return true;
      if (A.run() > B.run()) return false;
      return A.index() < B.index();
   }
   friend bool operator==(const run_index_t& A, const run_index_t& B)
   {
      return (A.run() == B.run()) && (A.index() == B.index());
   }
 
   ClassDef(run_index_t, 0)
};