scheduleProject.h

Go to the documentation of this file.

#ifndef GUM_SCHEDULE_PROJECT_H
#define GUM_SCHEDULE_PROJECT_H

#include <string>

#include <agrum/agrum.h>

#include <agrum/core/sequence.h>
#include <agrum/core/set.h>
#include <agrum/graphicalModels/inference/scheduleMultiDim.h>
#include <agrum/graphicalModels/inference/scheduleOperation.h>
#include <agrum/multidim/implementations/multiDimImplementation.h>
#include <agrum/variables/discreteVariable.h>

namespace gum {

  template < typename GUM_SCALAR >

  class ScheduleProject : public ScheduleOperation< GUM_SCALAR > {
    public:
    // ############################################################################
    // ############################################################################


    ScheduleProject(const ScheduleMultiDim< GUM_SCALAR >& table,
                    const Set< const DiscreteVariable* >& del_vars,
                    MultiDimImplementation< GUM_SCALAR >* (*project)(
                       const MultiDimImplementation< GUM_SCALAR >&,
                       const Set< const DiscreteVariable* >&));

    ScheduleProject(const ScheduleProject< GUM_SCALAR >&);

    virtual ScheduleProject< GUM_SCALAR >* newFactory() const;

    virtual ~ScheduleProject();


    // ############################################################################
    // ############################################################################

    ScheduleProject< GUM_SCALAR >& operator=(const ScheduleProject< GUM_SCALAR >&);


    bool operator==(const ScheduleOperation< GUM_SCALAR >&) const;


    bool operator!=(const ScheduleOperation< GUM_SCALAR >&) const;


    // ############################################################################
    // ############################################################################

    void execute();

    float nbOperations() const;


    std::pair< long, long > memoryUsage() const;

    const ScheduleMultiDim< GUM_SCALAR >& result() const;

    const Sequence< const ScheduleMultiDim< GUM_SCALAR >* >& multiDimArgs() const;

    const Sequence< const ScheduleMultiDim< GUM_SCALAR >* >&
       multiDimResults() const;

    std::string toString() const;


    private:
    // the table to project
    ScheduleMultiDim< GUM_SCALAR > __table;

    // the set of variables that should be removed from the table
    Set< const DiscreteVariable* > __del_vars;

    ScheduleMultiDim< GUM_SCALAR >* __result;

    mutable Sequence< const ScheduleMultiDim< GUM_SCALAR >* >* __args;

    mutable Sequence< const ScheduleMultiDim< GUM_SCALAR >* >* __results;

    MultiDimImplementation< GUM_SCALAR >* (*__project)(
       const MultiDimImplementation< GUM_SCALAR >&,
       const Set< const DiscreteVariable* >&);
  };

} /* namespace gum */

// always include the template implementation
#include <agrum/graphicalModels/inference/scheduleProject_tpl.h>

#endif /* GUM_SCHEDULE_PROJECTION_H */