drwnSparseFactor.h

/*****************************************************************************
** DARWIN: A FRAMEWORK FOR MACHINE LEARNING RESEARCH AND DEVELOPMENT
** Distributed under the terms of the BSD license (see the LICENSE file)
** Copyright (c) 2007-2017, Stephen Gould
** All rights reserved.
**
******************************************************************************
** FILENAME:    drwnSparseFactor.h
** AUTHOR(S):   Stephen Gould <stephen.gould@anu.edu.au>
**              Albert Chen <chenay@student.unimelb.edu.au>
**
*****************************************************************************/

#pragma once

#include "drwnBase.h"
#include "drwnIO.h"

#include "drwnVarUniverse.h"
#include "drwnVarAssignment.h"
#include "drwnTableFactor.h"
#include "drwnTableFactorMapping.h"

using namespace std;

// drwnSparseFactor --------------------------------------------------------

class drwnSparseFactor : public drwnFactor
{
 protected:
    std::map<drwnLocalAssignment, double> _assignments;  
    
 public:
    drwnSparseFactor(const drwnVarUniversePtr& ptr) : drwnFactor(ptr) { }
    ~drwnSparseFactor() { }

    // access functions
    const char *type() const { return "drwnSparseFactor"; }
    drwnSparseFactor* clone(void) const { DRWN_TODO; return NULL; }

    using drwnFactor::addVariable;

    int entries() const { return _assignments.size(); }
    const std::map<drwnLocalAssignment, double>& getAssignments() const { return _assignments; }

    double getValueOf(const drwnFullAssignment& y) const;
    double getValueOf(const drwnPartialAssignment& y) const;
    void setValueOf(const drwnFullAssignment& y, double val);
    void setValueOf(const drwnPartialAssignment& y, double val);

 private:
    void dfaToVals(const drwnFullAssignment& y, drwnLocalAssignment& vals) const;
};