FactoredMatrix.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_FACTORED_UTILS_FACTORED_MATRIX_HEADER_FILE
#define AI_TOOLBOX_FACTORED_UTILS_FACTORED_MATRIX_HEADER_FILE
 
#include <AIToolbox/Factored/Types.hpp>
 
namespace AIToolbox::Factored {
    struct BasisFunction {
        PartialKeys tag;
        Vector values;
    };
 
    struct FactoredVector {
        double getValue(const Factors & space, const Factors & value) const;
 
        double getValue(const Factors & space, const Factors & value, const Vector & weights) const;
 
        FactoredVector & operator*=(const Vector & w);
 
        FactoredVector & operator*=(const double v);
 
        std::vector<BasisFunction> bases;
    };
 
    struct BasisMatrix {
        PartialKeys tag;
        PartialKeys actionTag;
        Matrix2D values;
    };
 
    struct FactoredMatrix2D {
        double getValue(const Factors & space, const Factors & actions, const Factors & value, const Factors & action) const;
 
        double getValue(const Factors & space, const Factors & actions, const Factors & value, const Factors & action, const Vector & weights) const;
 
        FactoredMatrix2D & operator*=(const Vector & w);
 
        FactoredMatrix2D & operator*=(const double v);
 
        std::vector<BasisMatrix> bases;
    };
 
    // @}
 
    // This stuff is in FactoredVectorOps.cpp
    // BasisFunction - BasisFunction
 
    BasisFunction dot(const Factors & space, const BasisFunction & lhs, const BasisFunction & rhs);
    BasisFunction plus(const Factors & space, const BasisFunction & lhs, const BasisFunction & rhs);
    BasisFunction minus(const Factors & space, const BasisFunction & lhs, const BasisFunction & rhs);
 
    BasisFunction plusSubset(const Factors & space, BasisFunction retval, const BasisFunction & rhs);
    BasisFunction minusSubset(const Factors & space, BasisFunction retval, const BasisFunction & rhs);
 
    BasisFunction & plusEqualSubset(const Factors & space, BasisFunction & retval, const BasisFunction & rhs);
    BasisFunction & minusEqualSubset(const Factors & space, BasisFunction & retval, const BasisFunction & rhs);
 
    // FactoredVector - BasisFunction
 
    FactoredVector plus(const Factors & space, FactoredVector retval, const BasisFunction & rhs);
    FactoredVector & plusEqual(const Factors & space, FactoredVector & retval, const BasisFunction & basis);
    FactoredVector & plusEqual(const Factors & space, FactoredVector & retval, BasisFunction && basis);
 
    FactoredVector minus(const Factors & space, FactoredVector retval, const BasisFunction & rhs, bool clearZero = false);
    FactoredVector & minusEqual(const Factors & space, FactoredVector & retval, const BasisFunction & basis, bool clearZero = false);
 
    // FactoredVector - FactoredVector
 
    FactoredVector plus(const Factors & space, FactoredVector retval, const FactoredVector & rhs);
    FactoredVector & plusEqual(const Factors & space, FactoredVector & retval, const FactoredVector & rhs);
    FactoredVector & plusEqual(const Factors & space, FactoredVector & retval, FactoredVector && rhs);
 
    FactoredVector minus(const Factors & space, FactoredVector retval, const FactoredVector & rhs, bool clearZero = false);
    FactoredVector & minusEqual(const Factors & space, FactoredVector & retval, const FactoredVector & rhs, bool clearZero = false);
 
    // This stuff is in FactoredMatrix2DOps.cpp
    // BasisMatrix - BasisMatrix
 
    BasisMatrix plus(const Factors & space, const Factors & actions, const BasisMatrix & lhs, const BasisMatrix & rhs);
    BasisMatrix plusSubset(const Factors & space, const Factors & actions, BasisMatrix retval, const BasisMatrix & rhs);
    BasisMatrix & plusEqualSubset(const Factors & space, const Factors & actions, BasisMatrix & retval, const BasisMatrix & rhs);
 
    // FactoredMatrix2D - BasisMatrix
 
    FactoredMatrix2D & plusEqual(const Factors & space, const Factors & actions, FactoredMatrix2D & retval, const BasisMatrix & basis);
    FactoredMatrix2D & plusEqual(const Factors & space, const Factors & actions, FactoredMatrix2D & retval, BasisMatrix && basis);
 
    // FactoredMatrix2D - FactoredMatrix2D
 
    FactoredMatrix2D & plusEqual(const Factors & space, const Factors & actions, FactoredMatrix2D & retval, const FactoredMatrix2D & rhs);
    FactoredMatrix2D & plusEqual(const Factors & space, const Factors & actions, FactoredMatrix2D & retval, FactoredMatrix2D && rhs);
 
    // These are in FactoredMatrix.cpp
    // Scalar ops
 
    FactoredVector operator*(FactoredVector lhs, const Vector & w);
    FactoredVector operator*(const Vector & w, FactoredVector rhs);
    FactoredVector operator*(FactoredVector lhs, const double v);
    FactoredVector operator*(const double v, FactoredVector rhs);
 
    FactoredMatrix2D operator*(FactoredMatrix2D lhs, const Vector & w);
    FactoredMatrix2D operator*(const Vector & w, FactoredMatrix2D rhs);
    FactoredMatrix2D operator*(FactoredMatrix2D lhs, const double v);
    FactoredMatrix2D operator*(const double v, FactoredMatrix2D rhs);
}
 
#endif