scheduleCombine.h

Go to the documentation of this file.

#ifndef GUM_SCHEDULE_COMBINE_H
#define GUM_SCHEDULE_COMBINE_H

#include <string>

#include <agrum/agrum.h>

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

namespace gum {

  template < typename GUM_SCALAR >

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


    ScheduleCombine(const ScheduleMultiDim< GUM_SCALAR >& table1,
                    const ScheduleMultiDim< GUM_SCALAR >& table2,
                    MultiDimImplementation< GUM_SCALAR >* (*combine)(
                       const MultiDimImplementation< GUM_SCALAR >&,
                       const MultiDimImplementation< GUM_SCALAR >&));

    ScheduleCombine(const ScheduleCombine< GUM_SCALAR >&);

    virtual ScheduleCombine< GUM_SCALAR >* newFactory() const;

    ~ScheduleCombine();


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

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


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


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


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

    void execute();

    float nbOperations() const;


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

    INLINE 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:
    ScheduleMultiDim< GUM_SCALAR > __table1;

    ScheduleMultiDim< GUM_SCALAR > __table2;


    ScheduleMultiDim< GUM_SCALAR >* __result;

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

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

    MultiDimImplementation< GUM_SCALAR >* (*__combine)(
       const MultiDimImplementation< GUM_SCALAR >&,
       const MultiDimImplementation< GUM_SCALAR >&);
  };

} /* namespace gum */

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

#endif /* GUM_SCHEDULE_COMBINE_H */