Munkres.h

// Munkres.h:
// ----------
//

#ifndef __IPSDKMATH_MUNKRES_H__
#define __IPSDKMATH_MUNKRES_H__

// suppression warnings
// warning C4251: 'ipsdk::math::Munkres::_inputWeights' : class 'std::vector<ipsdk::ipReal64,std::allocator<_Ty>>' needs to have dll-interface to be used by clients of class 'ipsdk::math::Munkres'
// warning C4275: interface non dll class 'boost::noncopyable_::noncopyable' utilisée comme base d'une interface dll class 'ipsdk::math::Munkres'
#pragma warning (push)
#pragma warning (disable : 4275 4251)

#include <IPSDKMath/IPSDKMathExports.h>
#include <IPSDKUtil/BaseCollections.h>
#include <boost/noncopyable.hpp>

namespace ipsdk {
namespace math {


class IPSDKMATH_API Munkres : public boost::noncopyable
{
public:
    Munkres();
    ~Munkres();

// methods
public:
    bool                isInit() const;

    void                init(const ipUInt32 nbRows, const ipUInt32 nbColumns);

    ipUInt32            getNbRows() const;

    ipUInt32            getNbColumns() const;

    ipReal64*           getWeights(const ipUInt32 rowIdx);

    const ipReal64*     getWeights(const ipUInt32 rowIdx) const;

    ipReal64            getWeight(const ipUInt32 rowIdx,
                                  const ipUInt32 columnIdx) const;

    bool                isSolved() const;
    
    void                solve();

    const UInt32Vector& getAssignments() const;

    ipReal64            getTotalWeigth();

    void                clear();

// attributes
protected:
    ipUInt32        _nbRows;

    ipUInt32        _nbColumns;

    Real64Vector    _inputWeights;

    UInt32Vector    _assignments;
};


inline bool
Munkres::isInit() const
{
    return _inputWeights.empty() == false;
}


} // end of namespace math
} // end of namespace ipsdk

#pragma warning (pop)

#endif // __IPSDKMATH_MUNKRES_H__