#include <vector>
#include<iostream>


#include "Arc.h"
#include "node.h"

#include "Network.h"

    

using namespace std;



//adds an input node to input node vector
void Network::addInputNode(node * newnode)
{
   iNodes.push_back(newnode);

}

//adds a lumped node to lumped node vector
void Network::addLumpedNode(node * newnode)
{
	lNodes.push_back(newnode);
}

//adds a predictive node to predictive node vector
void Network::addPredictiveNode(node * newnode)
{
	pNodes.push_back(newnode);
}


//adds a output node to output node vector
void Network::addOutputNode(node * newnode)
{
	oNodes.push_back(newnode);
}

void Network::calcLumpedNodes()
{
		int iSize;
		int i;
					
		iSize = lNodes.size();
		for(i=0; i<iSize; i++)
		  (lNodes.at(i))->calcCurProb();
	   

}
void Network::calcPredictiveNodes()
{
		int iSize;
		int i;

	    iSize = pNodes.size();
		for(i=0; i<iSize; i++)
			(pNodes.at(i))->calcCurProb();

}

void Network::calcOutputNodes()
{
		int iSize;
		int i;

	    iSize = oNodes.size();
		for(i=0; i<iSize; i++)
			(oNodes.at(i))->calcCurProb();

}


void Network::printNodes()
{
	vector<node*>::iterator iter;


	iter = iNodes.begin();
	cout<<"*******************INPUT NODES************************"<<endl;
	while(iter!= iNodes.end())
	{
		
		cout<<"Name: "<<((node*)*iter)->name<<endl;
		cout<<"Current Probability: "<<((node*)*iter)->current_probability<<endl;
		cout<<"Prior Probability: "<<((node*)*iter)->getPriorProb()<<endl;
		cout<<"Odds All Events: "<<((node*)*iter)->getOddsAllEvents()<<endl;
		cout<<"Overall Lambda: "<<((node*)*iter)->getOverallLambda()<<endl;
		cout<<endl;
		iter++;
	}

	iter = lNodes.begin();
	cout<<"*******************LUMPED NODES************************"<<endl;
	while(iter!= lNodes.end())
	{
		cout<<"Name: "<<((node*)*iter)->name<<endl;
		cout<<"Current Probability: "<<((node*)*iter)->getCurProb()<<endl;
		cout<<"Prior Probability: "<<((node*)*iter)->getPriorProb()<<endl;
		cout<<"Odds All Events: "<<((node*)*iter)->getOddsAllEvents()<<endl;

		cout<<"Overall Lambda: "<<((node*)*iter)->getOverallLambda()<<endl;
		cout<<endl;
		iter++;
	}

	iter = pNodes.begin();
	cout<<"*******************PREDICTIVE NODES************************"<<endl;
	while(iter!= pNodes.end())
	{
		cout<<"Name: "<<((node*)*iter)->name<<endl;
		cout<<"Current Probability: "<<((node*)*iter)->current_probability<<endl;
		cout<<"Prior Probability: "<<((node*)*iter)->getPriorProb()<<endl;
		cout<<"Odds All Events: "<<((node*)*iter)->getOddsAllEvents()<<endl;

		cout<<"Overall Lambda: "<<((node*)*iter)->getOverallLambda()<<endl;
		cout<<endl;
		iter++;
	}

	iter = oNodes.begin();
	cout<<"*******************OUTPUT NODES************************"<<endl;
	while(iter!= oNodes.end())
	{
		cout<<"Name: "<<((node*)*iter)->name<<endl;
		cout<<"Current Probability: "<<((node*)*iter)->current_probability<<endl;
		cout<<"Prior Probability: "<<((node*)*iter)->getPriorProb()<<endl;
		cout<<"Odds All Events: "<<((node*)*iter)->getOddsAllEvents()<<endl;

		cout<<"Overall Lambda: "<<((node*)*iter)->getOverallLambda()<<endl;
		cout<<endl;
		iter++;
	}


}