KVParticleCondition.h

Go to the documentation of this file.

/*
$Id: KVParticleCondition.h,v 1.3 2007/03/26 10:14:56 franklan Exp $
$Revision: 1.3 $
$Date: 2007/03/26 10:14:56 $
*/
 
 
#ifndef __KVPARTICLECONDITION_H
#define __KVPARTICLECONDITION_H
 
#include "KVBase.h"
#include "KVString.h"
#include "KVHashList.h"
#include "KVNucleus.h"
 
#ifdef USING_ROOT6
#include <vector>
#include <functional>
#endif
 
class KVClassFactory;
 
class KVParticleCondition : public KVBase {
    static KVHashList fgOptimized;
    mutable Int_t fNUsing;
#ifdef USING_ROOT6
   using LambdaFunc = std::function<bool(const KVNucleus*)>;
   mutable LambdaFunc fLambdaCondition;
    LambdaFunc fSavedLambda1, fSavedLambda2;
   enum class LogOp { AND, OR } fOpType;
 
   void logical_operator_lambda_condition_test() const
   {
      if (IsLambda() && !IsSet()) {
         switch (fOpType) {
            case LogOp::AND:
               fLambdaCondition = [this](const KVNucleus * nuc) {
                  return (fSavedLambda1(nuc) && fSavedLambda2(nuc));
               };
               break;
            case LogOp::OR:
               fLambdaCondition = [this](const KVNucleus * nuc) {
                  return (fSavedLambda1(nuc) || fSavedLambda2(nuc));
               };
               break;
         }
      }
   }
#endif
 
protected:
 
    KVString fCondition;
    KVString fCondition_raw;
    KVString fCondition_brackets;
 
    mutable const KVParticleCondition* fOptimal;
    KVString fClassName;
    mutable KVClassFactory* cf;
    KVString fOptimizedClassName;
    mutable Bool_t fOptOK;
 
   void Optimize() const;
   virtual bool optimized_test(const KVNucleus*) const
   {
      return kFALSE;
   }
   void CreateClassFactory() const;
   void SetClassFactory(KVClassFactory* CF);
 
public:
 
   KVParticleCondition();
   KVParticleCondition(const KVString& cond);
   KVParticleCondition(const Char_t* cond);
   KVParticleCondition(const KVParticleCondition&);
#ifdef USING_ROOT6
   KVParticleCondition(const KVString& name, const LambdaFunc& F)
      : KVBase(name, "KVParticleCondition"), fLambdaCondition(F)
   {
      fOptimal = nullptr;
      cf = nullptr;
      fOptOK = kFALSE;
      fNUsing = 0;
   }
   bool IsLambda() const
   {
      return (bool)fLambdaCondition || ((bool)fSavedLambda1 && (bool)fSavedLambda2);
   }
#endif
   virtual ~KVParticleCondition();
#ifdef USING_ROOT6
   void Set(const KVString& name, const LambdaFunc& F)
   {
      fLambdaCondition = F;
      SetName(name);
   }
#endif
   void Set(const KVString&);
 
   Bool_t Test(const KVNucleus* nuc) const
   {
 
#ifdef USING_ROOT6
      logical_operator_lambda_condition_test();
#endif
      if (!IsSet()) return kTRUE;
#ifdef USING_ROOT6
      if (IsLambda()) return fLambdaCondition(nuc);
#endif
      if (!fOptimal) Optimize();
 
      return (fOptOK ? fOptimal->optimized_test(nuc) : kFALSE);
   }
 
   Bool_t Test(const KVNucleus& nuc) const
   {
 
      return Test(&nuc);
   }
 
   void SetParticleClassName(const Char_t* cl)
   {
      fClassName = cl;
   }
   void AddExtraInclude(const Char_t* inc_file);
 
   void Copy(TObject&) const;
 
   KVParticleCondition& operator=(const KVParticleCondition&);
   KVParticleCondition& operator=(const KVString&);
#ifdef USING_ROOT6
   KVParticleCondition& operator=(const LambdaFunc&);
#endif
   friend KVParticleCondition operator&&(const KVParticleCondition&, const KVParticleCondition&);
   friend KVParticleCondition operator||(const KVParticleCondition&, const KVParticleCondition&);
   KVParticleCondition& operator|=(const KVParticleCondition&);
   KVParticleCondition& operator&=(const KVParticleCondition&);
 
   void Print(Option_t* opt = "") const;
   static void PrintOptimizedList()
   {
      fgOptimized.Print();
   }
   Bool_t IsSet() const
   {
#ifdef USING_ROOT6
      return (fLambdaCondition || fCondition != "");
#else
      return fCondition != "";
#endif
   }
 
   ClassDef(KVParticleCondition, 2) //Implements parser of particle selection criteria
};
 
#endif