EnergyFunction.h

Go to the documentation of this file.

 
#pragma once
 
#include <ogdf/basic/Graph.h>
#include <ogdf/basic/GraphAttributes.h>
#include <ogdf/basic/basic.h>
#include <ogdf/basic/geometry.h>
 
#include <string>
 
namespace ogdf {
namespace davidson_harel {
 
 
class EnergyFunction {
public:
    EnergyFunction(const string& funcname, GraphAttributes& AG);
 
    virtual ~EnergyFunction() { }
 
    virtual void computeEnergy() = 0;
 
    double computeCandidateEnergy(const node v, const DPoint& newPos);
 
    string getName() const { return m_name; }
 
    void candidateTaken();
 
#ifdef OGDF_DEBUG
    void printStatus() const;
#endif
 
    double energy() const { return m_energy; }
 
protected:
    const Graph& m_G; 
    const string m_name; 
    double m_candidateEnergy; 
    double m_energy; 
 
    DPoint testPos() { return m_testPos; }
 
    DPoint currentPos(const node v) const { return m_AG.point(v); }
 
    node testNode() const { return m_testNode; }
 
    virtual void internalCandidateTaken() = 0;
 
    virtual void compCandEnergy() = 0;
 
#ifdef OGDF_DEBUG
    virtual void printInternalData() const = 0;
#endif
 
private:
    EnergyFunction(const EnergyFunction& e) : m_G(e.m_G), m_name(e.m_name), m_AG(e.m_AG) { }
 
    EnergyFunction& operator=(const EnergyFunction& e);
 
    GraphAttributes& m_AG; 
    node m_testNode; 
    DPoint m_testPos; 
};
 
}
}