///////////////////////////////////////////////////////////////////////////////

// File name: PlotGraph.h

// Description: This file defines the methods of plotting the Hungarian steps.

// You may need to install gnuplot first.

// A plotting lib is included in ./libCgnuplot (Author: N. Devillard)

// The plotting funtion is stripped from Dr. Dylan Shell's code

// (see ./libCgnuplot/graph_hungarian.c: void draw_graph(int new_member_of_T) )

// Lantao Liu, Nov 3, 2009

///////////////////////////////////////////////////////////////////////////////

#ifndef PLOT_GRAPH_H

#define PLOT_GRAPH_H

#include "Matrix.h"

#include "BipartiteGraph.h"

#include "plot/Cgnuplot.h"

namespace hungarian

{

//#define PERIOD 4 //uniformly defined in Matrix.h

extern int _Period;

class PlotGraph{

public:

PlotGraph():g(NULL),p(NULL){ period = _Period != 0 ? _Period : PERIOD; }

~PlotGraph(){}

void SetPeriod(unsigned int _period){ period = _period; }

unsigned int GetPeriod(void){ return period; }

//gnuplot_ctrl* InitPlot(void){ return gnuplot_init(); }

void InitPlot(void){ g = gnuplot_init(); }

void ClosePlot(gnuplot_ctrl* _handle){ gnuplot_close(_handle); }

void ClosePlot(void){ ClosePlot(g); }

void PlotBipartiteGraph(BipartiteGraph& _bg,

vector<VID>& _S,

vector<VID>& _T,

vector<VID>& _N,

vector<EID>& _EG,

vector<EID>& _M,

int target_task = -1);

void PlotAugmentingPath(BipartiteGraph& _bg, vector<EID>& _path);

void DisplayData(const vector<VID>& vs);

void DisplayData(const vector<EID>& es);

//public:

private:

gnuplot_ctrl *g;

gnuplot_ctrl *p;

unsigned int period; //plot refreshing period

};

} //namespace

#endif