run_index_list.cpp

#include "run_index_list.h"
#include "TMath.h"
 
ClassImp(run_index_list)
 
 
 
void run_index_list::Add(const run_index_list& list)
{
   // Add all in list to this list
   for (auto& r : list) Add(r);
}
 
 
 
 
void run_index_list::Remove(const run_index_list& list)
{
   // Remove all in list to this list
   for (auto& r : list) Remove(r);
}
 
 
 
 
TString run_index_list::GetList(bool no_commas) const
{
   // Return list of [run.index] as formatted string:
   //
   // no_commas=true (default):
   //  "1 1.1 1.2 1.3 2 2.1" etc.
   //
   // no_commas=false:
   //  "1, 1.1, 1.2, 1.3, 2, 2.1" etc.
   TString list;
   bool first = true;
   TString separator = (no_commas ? " " : ", ");
   for (auto& r : fRunList) {
      if (!first)
         list += separator;
      list += r.as_string();
      first = false;
   }
   return list;
}
 
 
 
 
void run_index_list::SetList(const TString& slist)
{
   // Set list from formatted string of [run.index]
   // with or without commas:
   //
   //  "1, 1.1, 1.2, 1.3, 2, 2.1"
   //  "1 1.1 1.2 1.3 2 2.1" etc.
 
   Clear();
   KVString b(slist);
   b.Begin(" ,");
   while (!b.End()) {
      auto n = b.Next(kTRUE);
      Add(run_index_t(n));
   }
}
 
 
 
 
void run_index_list::SetListSelection(const TString& slist, const run_index_list& full_list)
{
   // \param slist a list of runfile names and/or run number ranges. Runfile names must contain '.',
   //   for the first file of a run use 'r.0'. Just 'r' represents the entire run, i.e. 'r.0', 'r.1', 'r.2', etc.
   // \param full_list a list of all existing runfiles
   //
   // Examples of use:
   //
   // Suppose that full_list contains the runfile list {1, 1.1, 1.2, 2, 3, 3.1}
   //~~~
   // slist="1 2 3"
   // slist="1,2,3"
   // slist="1-3"
   // slist="1,2-3"
   //  => fill list with {1, 1.1, 1.2, 2, 3, 3.1}
   //
   // slist="1.0,3"
   // slist="1.0 3"
   //  => fill list with {1, 3, 3.1}
   //~~~
 
   Clear();
   KVNumberList runs_all_files;
   KVString _list(slist);
   _list.Begin(", ");
   while (!_list.End()) {
      KVString n = _list.Next();
      if (n.Contains("-")) {
         // KVNumberList format for run range
         KVNumberList nl(n);
         for (auto run : nl) {
            runs_all_files.Add(run);
         }
      }
      else if (!n.Contains(".")) {
         runs_all_files.Add(n.Atoi());
      }
      else {
         // individual runfile
         n.ReplaceAll(".0", ".-1");
         Add(run_index_t{n});
      }
   }
   if (!runs_all_files.IsEmpty()) {
      // now add all runfiles corresponding to runs in number lists
      for (auto& run : full_list) {
         if (runs_all_files.Contains(run.run()))
            Add(run);
      }
   }
}
 
 
 
 
TString run_index_list::AsString(Int_t maxlen) const
{
   //Get string containing list.
   //
   //If maxlen>0, and if length of resulting string is longer than maxlen,
   //we truncate the list to show only the beginning and the end of the list,
   //with "..." in between, i.e. "6000-...-8910". Note that the minimum size of
   //the resulting string is 5 i.e. "6...0".
   //
   //Returns empty string if list is empty.
 
   auto tmp = GetList();
   if (maxlen) {
      maxlen = TMath::Max(maxlen, 5);
      if (tmp.Length() > maxlen) {
         Int_t len_left = maxlen - 3; // 3 for "..."
         Int_t len_start = len_left / 2;
         Int_t len_end = len_left - len_start;
         TString tmp2 = tmp(0, len_start);
         tmp2 += "...";
         tmp2 += tmp2(tmp2.Length() - len_end, tmp2.Length() - 1);
         tmp = tmp2;
      }
   }
   return tmp;
}
 
 
 
 
void run_index_list::Inter(const run_index_list& list)
{
   //keep the AND logic operation result between 'list' and this list
   //i.e. keep only run_index_t which appear in both lists
 
   auto tmp = *this;
   Clear();
   for (auto& t : tmp) {
      if (list.Contains(t)) Add(t);
   }
}
 
 
 
 
KVNumberList run_index_list::GetRunNumberList() const
{
   // Returns just the list of run numbers
   KVNumberList l;
   for (auto& r : *this) l.Add(r.run());
   return l;
}