AbstractNodes.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
#include <variant>
#include <iostream>
 
namespace Belle2 {
  template<class AVariableManager>
  class AbstractBooleanNode {
  public:
    typedef typename AVariableManager::Object Object;
    typedef typename AVariableManager::Var Var;
 
    virtual bool check(const Object* p) const = 0;
    virtual void print() const = 0;
    virtual std::string decompile() const = 0;
    virtual ~AbstractBooleanNode() {};
  };
 
  template<class AVariableManager>
  class AbstractExpressionNode {
  public:
    typedef typename AVariableManager::Object Object;
    typedef typename AVariableManager::Var Var;
 
    virtual typename AVariableManager::VarVariant evaluate(const Object* p) const = 0;
    virtual void print() const = 0;
    virtual std::string decompile() const = 0;
    virtual ~AbstractExpressionNode() {}
 
  };
 
  enum class NodeType : int {
    UnaryBooleanNode,
    BinaryBooleanNode,
    UnaryRelationalNode,
    BinaryRelationalNode,
    TernaryRelationalNode,
    UnaryExpressionNode,
    BinaryExpressionNode,
    FunctionNode,
    IdentifierNode,
    DoubleNode,
    IntegerNode,
    BooleanNode
  };
 
  enum class BooleanOperator : int {
    AND,
    OR
  };
 
  enum class ComparisonOperator : int {
    EQUALEQUAL,
    GREATEREQUAL,
    LESSEQUAL,
    GREATER,
    LESS,
    NOTEQUAL
  };
 
  enum class ArithmeticOperation : int {
    PLUS,
    MINUS,
    PRODUCT,
    DIVISION,
    POWER
  };
 
  void injectBooleanOperatorToStream(std::ostream& stream, const BooleanOperator& boperator);
 
  void injectComparisonOperatorToStream(std::ostream& stream, const ComparisonOperator& coperator);
 
  void injectArithmeticOperatorToStream(std::ostream& stream, const ArithmeticOperation& aoperation);
 
}