BipartitePartialMatch.h

// BipartitePartialMatch.h:
// ------------------------
//

#ifndef __IPSDKMATH_BIPARTITEPARTIALMATCH_H__
#define __IPSDKMATH_BIPARTITEPARTIALMATCH_H__

// suppression warnings
// warning C4251: 'ipsdk::math::BipartitePartialMatch::_inputWeights' : class 'std::vector<ipsdk::ipReal64,std::allocator<_Ty>>' needs to have dll-interface to be used by clients of class 'ipsdk::math::BipartitePartialMatch'
#pragma warning (push)
#pragma warning (disable : 4251)

#include <IPSDKMath/IPSDKMathExports.h>
#include <IPSDKUtil/BaseCollections.h>

namespace ipsdk {
namespace math {


class IPSDKMATH_API BipartitePartialMatch
{
public:
    BipartitePartialMatch();
    ~BipartitePartialMatch();

// methods
public:
    static ipUInt32     getUnAssignedValue();

    bool                isInit() const;

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

    ipUInt32            getNbRows() const;

    ipUInt32            getNbColumns() const;

    ipReal64            getThreshold() 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& getRowAssignments() const;

    const UInt32Vector& getColumnAssignments() const;

    void                clear();

// attributes
protected:
    ipUInt32        _nbRows;

    ipUInt32        _nbColumns;

    ipReal64        _threshold;

    Real64Vector    _inputWeights;

    UInt32Vector    _rowAssignments;

    UInt32Vector    _columnAssignments;
};


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


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

#pragma warning (pop)

#endif // __IPSDKMATH_BIPARTITEPARTIALMATCH_H__