KVNumberList.h

/*******************************************************************************
$Id: KVNumberList.h,v 1.20 2009/03/03 14:27:15 franklan Exp $
$Revision: 1.20 $
$Date: 2009/03/03 14:27:15 $
$Author: franklan $
*******************************************************************************/
 
#ifndef __KVNUMBERLIST_H
#define __KVNUMBERLIST_H
 
#include <vector>
#include <TString.h>
#include <TArrayI.h>
#include <TObject.h>
#include <TList.h>
#include "KVConfig.h"
 
typedef std::vector<Int_t> IntArray;
typedef std::vector<Int_t>::iterator IntArrayIter;
typedef std::vector<Int_t>::const_iterator IntArrayCIter;
 
class KVNumberList : public TObject {
 
   mutable TString fString;
    TArrayI fLowerBounds;
    TArrayI fUpperBounds;
    Int_t fNLimits;
    Int_t fMaxNLimits;
    Int_t fFirstValue;
    Int_t fLastValue;
    mutable Int_t fNValues;
 
    mutable TString fTMPSTR;
 
    mutable IntArrayIter fIterIndex;
    mutable IntArrayIter fEndList;
    mutable IntArray fValues;
    TString fName;
 
    mutable Bool_t fIsParsed;
    mutable IntArray fRandomFastArray;
 
   void init_numberlist();
   void clear();
   void ParseList();
   void AddLimits(Int_t min, Int_t max);
   void AddLimits(TString& string);
   void ParseAndFindLimits(const TString& string, const Char_t delim);
 
public:
 
   KVNumberList();
   KVNumberList(const KVNumberList&);
   KVNumberList(const Char_t*);
   KVNumberList(Int_t);
   KVNumberList(Int_t deb, Int_t fin, Int_t pas);
#ifdef WITH_CPP11
   KVNumberList(std::initializer_list<int> L);
#endif
   virtual ~KVNumberList()
   {
   }
 
   virtual void     SetName(const char* name);
   virtual const char* GetName() const
   {
      return fName.Data();
   }
 
   /* LIST MODIFIERS */
   /* Set the number list */
   void Set(const TString& l)
   {
      SetList(l);
   }
   void Set(Int_t min, Int_t max, Int_t pas = 1)
   {
      SetMinMax(min, max, pas);
   }
   void SetList(const TString&);
   void SetMinMax(Int_t min, Int_t max, Int_t pas = 1);
   ROOT_COPY_ASSIGN_OP(KVNumberList)
   void Copy(TObject&) const;
 
   /* Add numbers/lists to the list */
   void Add(Int_t);
   void Add(const KVNumberList&);
   void Add(Int_t, Int_t*);
   void Add(const IntArray&);
 
   /* Remove numbers/lists from the list */
   void Remove(Int_t);
   void Remove(const Char_t*);
   void Remove(const KVNumberList&);
   void Remove(Int_t n, Int_t* arr);
 
   /* Clear the list */
   void Clear(Option_t* = "");
 
   /* Intersection of two sets of numbers i.e. those which are in both lists */
   void Inter(const KVNumberList& list);
 
   /* LIST PROPERTIES */
   Bool_t Contains(Int_t val) const;
   Int_t First() const;
   Int_t Last() const;
   Int_t GetNValues() const;
   Int_t GetEntries() const
   {
      return GetNValues();
   }
   Bool_t IsEmpty() const
   {
      if (!fIsParsed) const_cast<KVNumberList*>(this)->ParseList();
      return (fNValues == 0);
   }
   Bool_t IsFull(Int_t vinf = -1, Int_t vsup = -1) const;
 
   /* LIST MEMBER ACCESS */
   Int_t At(Int_t index) const;
   Int_t operator[](Int_t index) const;
   IntArray GetArray() const;
   const Char_t* GetList() const;
   const Char_t* GetExpandedList() const;
   const Char_t* AsString(Int_t maxchars = 0) const;
   const Char_t* AsQuotedString() const;
   const Char_t* AsHumanReadableString() const;
 
   Int_t GetRandom() const;
   Int_t GetRandomFast() const;
   Bool_t PrepareRandomFast() const;
 
   /* LIST ITERATORS */
   Int_t Next(void) const;
   void Begin(void) const;
   Bool_t End(void) const
   {
      return (fIterIndex == fEndList);
   }
   IntArrayIter begin() const;
   IntArrayIter end() const;
 
   /* LIST ARITHMETIC OPERATIONS */
   KVNumberList operator+(const KVNumberList&) const;
   KVNumberList operator-(const KVNumberList&) const;
   KVNumberList GetComplementaryList() const;
   KVNumberList GetSubList(Int_t vinf, Int_t vsup) const;
   TList* CutInSubList(Int_t number);
   /* MISCELLANEOUS */
   /* Generate TTree::Draw selection string */
   TString GetLogical(const Char_t* observable) const;
   TString GetSQL(const Char_t* column) const;
   /* Convert to c-string */
   operator const char* () const
   {
      return GetList();
   }
   /* Print list properties */
   void Print(Option_t* = "") const;
   void PrintLimits() const;
   /* tests for equality */
   bool operator==(const KVNumberList&) const;
   bool operator!=(const KVNumberList&) const;
 
   ClassDef(KVNumberList, 4)    //Strings used to represent a set of ranges of values
};
 
#endif