PWG2/RESONANCES/AliRsnExpression.h

   1 #ifndef ALIRSNEXPRESSION_H
   2 #define ALIRSNEXPRESSION_H
   3
   4 #include <TObject.h>
   5 #include <TString.h>
   6
   7 #include "AliRsnCutSet.h"
   8
   9 class TObjArray;
  10 class AliRsnCutSet;
  11 /**
  12   @author Martin Vala <Martin.Vala@cern.ch>
  13  */
  14
  15 // These are the valid operators types.
  16
  17 class AliRsnExpression : public TObject
  18 {
  19
  20   public:
  21
  22   // operators for complex cut expressions
  23     enum ECutOp
  24     {
  25         kOpAND=1,   // AND '&'
  26         kOpOR,      // OR '|'
  27         kOpNOT      // Unary negation '!'
  28     };
  29
  30     AliRsnExpression() : fVname ( 0 ), fArg1 ( 0 ), fArg2 ( 0 ), fOperator ( 0 )  {}
  31     AliRsnExpression ( TString exp );
  32     virtual    ~AliRsnExpression();
  33     AliRsnExpression ( const AliRsnExpression& exp );
  34     AliRsnExpression&    operator= ( const AliRsnExpression& exp );
  35
  36     virtual Bool_t     Value ( TObjArray & vars );
  37     virtual TString     Unparse() const;
  38
  39     void SetCutSet ( AliRsnCutSet* theValue ) { sCutSet = theValue; }
  40     AliRsnCutSet* GetCutSet() const { return sCutSet; }
  41
  42
  43     TString    fVname;   // Variable name
  44     static AliRsnCutSet        *sCutSet;
  45
  46   private:
  47     AliRsnExpression*   fArg1;         // left argument
  48     AliRsnExpression*   fArg2;         // right argument
  49     Int_t                 fOperator;     // operator
  50
  51     AliRsnExpression ( int op, AliRsnExpression* a );
  52     AliRsnExpression ( int op, AliRsnExpression* a, AliRsnExpression* b );
  53
  54     TObjArray*    Tokenize ( TString str ) const;
  55     static AliRsnExpression*    Element ( TObjArray &st, Int_t &i );
  56     static AliRsnExpression*    Primary ( TObjArray &st, Int_t &i );
  57     static AliRsnExpression*    Expression ( TObjArray &st, Int_t &i );
  58
  59     ClassDef ( AliRsnExpression, 1 ); // Class to evaluate an expression
  60 };
  61
  62
  63 ///////////////////////////////////////////////////////////////////////////
  64
  65 class AliRsnVariableExpression: public AliRsnExpression
  66 {
  67   public:
  68     AliRsnVariableExpression ( TString a) : AliRsnExpression() { fVname = a;  };
  69     ~AliRsnVariableExpression() {}
  70     virtual Bool_t    Value ( TObjArray& pgm );
  71     virtual TString    Unparse() const { return fVname; }
  72
  73     ClassDef ( AliRsnVariableExpression, 1 ); // Class to define a variable expression
  74 };
  75
  76 #endif