#include "_anim.h"
#include "ns3/netanim-module.h"
#include "ns3/core-module.h"

ns3::AnimationInterface setAnimation(ns3::NodeContainer nodes)
{	
	ns3::AnimationInterface anim ("anim_dos.xml");
	anim.SetConstantPosition(nodes.Get(0), 0.0, 3.0);	
	anim.SetConstantPosition(nodes.Get(1), 3.0, 0.0);	
	anim.SetConstantPosition(nodes.Get(2), 4.0, 3.0);	
	anim.SetConstantPosition(nodes.Get(3), 4.1, 4.0);	
	anim.SetConstantPosition(nodes.Get(4), 4.2, 5.0);
	
	anim.UpdateNodeDescription (nodes.Get(0)->GetId(), "Atacker n0");	
	anim.UpdateNodeDescription (nodes.Get(1)->GetId(), "Server");	
	anim.UpdateNodeDescription (nodes.Get(2)->GetId(), "Client n2");	
	anim.UpdateNodeDescription (nodes.Get(3)->GetId(), "Client n3");	
	anim.UpdateNodeDescription (nodes.Get(4)->GetId(), "Client n4");
	 
	anim.UpdateNodeSize(nodes.Get(0)->GetId(), 0.5, 0.5);
	anim.UpdateNodeSize(nodes.Get(1)->GetId(), 1.0, 1.0);
	anim.UpdateNodeSize(nodes.Get(2)->GetId(), 0.5, 0.5);
	anim.UpdateNodeSize(nodes.Get(3)->GetId(), 0.5, 0.5);
	anim.UpdateNodeSize(nodes.Get(4)->GetId(), 0.5, 0.5);

	anim.EnablePacketMetadata (true);
	//anim.SetMobilityPollInterval (ns3::Seconds (1));
	anim.SetStartTime (ns3::Seconds (0));
	anim.SetStopTime (ns3::Seconds (20));

	return anim;
}