IBayesNetFactory.h

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (C) 2005 by Christophe GONZALES and Pierre-Henri WUILLEMIN  *
 *   {prenom.nom}_at_lip6.fr                                               *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#ifndef GUM_INTERFACE_BAYESNET_FACTORY_H
#define GUM_INTERFACE_BAYESNET_FACTORY_H

#include <string>
#include <vector>

#include <agrum/graphicalModels/variableNodeMap.h>
#include <agrum/multidim/implementations/multiDimAdressable.h>
#include <agrum/variables/labelizedVariable.h>

namespace gum {
  class IBayesNetFactory {
    public:
    enum class factory_state : char {
      NONE,
      NETWORK,
      VARIABLE,
      PARENTS,
      RAW_CPT,
      FACT_CPT,
      FACT_ENTRY
    };

    IBayesNetFactory() : __verbose(false){};
    // just to make some compilers happy
    virtual ~IBayesNetFactory(){};

    void setVerbose() { __verbose = true; };

    void resetVerbose() { __verbose = false; };

    bool isVerbose() { return __verbose; };

    virtual const DiscreteVariable& varInBN(NodeId id) = 0;
    virtual factory_state           state() const = 0;
    virtual NodeId                  variableId(const std::string& name) const = 0;
    virtual Size                    cptDomainSize(NodeId n) const = 0;

    virtual void   startNetworkDeclaration() = 0;
    virtual void   addNetworkProperty(const std::string& propName,
                                      const std::string& propValue) = 0;
    virtual void   endNetworkDeclaration() = 0;
    virtual void   startVariableDeclaration() = 0;
    virtual void   variableName(const std::string& name) = 0;
    virtual void   variableDescription(const std::string& desc) = 0;
    virtual void   addModality(const std::string& name) = 0;
    virtual void   setVariableCPTImplementation(MultiDimAdressable* impl) = 0;
    virtual NodeId endVariableDeclaration() = 0;
    virtual void   startParentsDeclaration(const std::string& var) = 0;
    virtual void   addParent(const std::string& var) = 0;
    virtual void   endParentsDeclaration() = 0;
    virtual void   startRawProbabilityDeclaration(const std::string& var) = 0;
    virtual void   rawConditionalTable(const std::vector< std::string >& variables,
                                       const std::vector< float >& rawTable) = 0;
    virtual void   rawConditionalTable(const std::vector< float >& rawTable) = 0;
    virtual void   endRawProbabilityDeclaration() = 0;
    virtual void startFactorizedProbabilityDeclaration(const std::string& var) = 0;
    virtual void startFactorizedEntry() = 0;
    virtual void endFactorizedEntry() = 0;
    virtual void setParentModality(const std::string& parent,
                                   const std::string& modality) = 0;
    virtual void
                 setVariableValuesUnchecked(const std::vector< float >& values) = 0;
    virtual void setVariableValues(const std::vector< float >& values) = 0;
    virtual void endFactorizedProbabilityDeclaration() = 0;
    virtual void setVariable(const DiscreteVariable& var) = 0;
    virtual void setVariableCPT(const std::string&  varName,
                                MultiDimAdressable* table,
                                bool                redefineParents) = 0;

    private:
    bool __verbose;
  };
} /* namespace gum */

#endif   // GUM_ASBTRACT_BAYESNET_FACTORY_H