scheduleCliqueStoreMultiDim.h

Go to the documentation of this file.

#ifndef GUM_SCHEDULE_CLIQUE_STORE_MULTI_DIM_H
#define GUM_SCHEDULE_CLIQUE_STORE_MULTI_DIM_H

#include <string>

#include <agrum/agrum.h>

#include <agrum/core/sequence.h>
#include <agrum/graphicalModels/inference/scheduleMultiDim.h>
#include <agrum/graphicalModels/inference/scheduleOperation.h>
#include <agrum/graphs/graphElements.h>
#include <agrum/multidim/implementations/multiDimImplementation.h>

namespace gum {

  template < typename GUM_SCALAR >

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

    ScheduleCliqueStoreMultiDim(
       const ScheduleMultiDim< GUM_SCALAR >& table,
       NodeProperty< Set< const MultiDimImplementation< GUM_SCALAR >* > >&
              clique_tables,
       NodeId clique);

    ScheduleCliqueStoreMultiDim(const ScheduleCliqueStoreMultiDim< GUM_SCALAR >&);

    virtual ScheduleCliqueStoreMultiDim< GUM_SCALAR >* newFactory() const;

    virtual ~ScheduleCliqueStoreMultiDim();


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

    ScheduleCliqueStoreMultiDim< GUM_SCALAR >&
       operator=(const ScheduleCliqueStoreMultiDim< 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 Sequence< const ScheduleMultiDim< GUM_SCALAR >* >& multiDimArgs() const;

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

    std::string toString() const;


    private:
    // the table to store into the clique
    ScheduleMultiDim< GUM_SCALAR > __table;

    // a mapping assigning to each clique a set of tables
    NodeProperty< Set< const MultiDimImplementation< GUM_SCALAR >* > >* __tableSet;

    // the clique into which the table will be stored
    NodeId __clique;

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

} /* namespace gum */

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

#endif /* GUM_SCHEDULE_CLIQUE_STORE_MULTI_DIM_H */