qkd-total-graph.cc

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2020 DOTFEESA www.tk.etf.unsa.ba
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation;
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * Author: Miralem Mehic <miralem.mehic@ieee.org>
 */
 
#include "qkd-total-graph.h"  
 
namespace ns3 {
 
NS_LOG_COMPONENT_DEFINE ("QKDTotalGraph");
 
NS_OBJECT_ENSURE_REGISTERED (QKDTotalGraph);
 
TypeId QKDTotalGraph::GetTypeId (void) 
{
  static TypeId tid = TypeId ("ns3::QKDTotalGraph")
    .SetParent<Object> () 
    ; 
  return tid;
}
 
QKDTotalGraph::~QKDTotalGraph(){}
 
QKDTotalGraph::QKDTotalGraph (){
    Init("","");
}
 
QKDTotalGraph::QKDTotalGraph (
    std::string graphName,
    std::string graphType
){
    Init(graphName, graphType);
}
 
void
QKDTotalGraph::Init (
    std::string graphName,
    std::string graphType
)
{   
    m_keymCurrent = 0;
    m_keymThreshold = 0;
 
    NS_LOG_FUNCTION (this <<  graphName);
   
    m_plotFileName = (graphName.empty()) ? "QKD Total Graph" : graphName;
    m_plotFileType = (graphType.empty()) ? "png" : graphType;; 
  
    m_gnuplot.SetOutputFilename (m_plotFileName + "." + m_plotFileType);
 
    m_dataset.SetTitle ("Amount of key material");
    m_dataset.SetStyle (Gnuplot2dDataset::LINES_POINTS);
    m_dataset.SetExtra(" ls 1"); 
    //m_dataset.Add( 0, m_keymCurrent); 
 
    m_datasetThreshold.SetTitle ("Threshold");
    m_datasetThreshold.SetStyle (Gnuplot2dDataset::LINES_POINTS);
    m_datasetThreshold.SetExtra(" ls 4"); 
    //m_datasetThreshold.Add( 0, m_keymThreshold); 
     
    std::string outputTerminalCommand;
 
    if (m_plotFileType=="png") 
        outputTerminalCommand = "pngcair";
    else if (m_plotFileType=="tex") 
        outputTerminalCommand = "epslatex";
    else
        outputTerminalCommand = m_plotFileType;
 
    m_gnuplot.AppendExtra ("set autoscale y");  
    m_gnuplot.AppendExtra ("set border linewidth 2"); 
 
    if (m_plotFileType=="tex") 
        m_gnuplot.AppendExtra ("set terminal " + outputTerminalCommand); 
    else
        m_gnuplot.AppendExtra ("set terminal " + outputTerminalCommand + " size 1524,768 enhanced font 'Helvetica,18'"); 
 
    m_gnuplot.AppendExtra ("set style line 1 linecolor rgb 'red' linetype 1 linewidth 1");   
    m_gnuplot.AppendExtra ("set style line 4 linecolor rgb \"#8A2BE2\" linetype 1 linewidth 1"); 
 
    m_gnuplot.AppendExtra ("set style line 7 linecolor rgb 'gray' linetype 0 linewidth 1"); 
    m_gnuplot.AppendExtra ("set style line 8 linecolor rgb 'gray' linetype 0 linewidth 1 "); 
 
    m_gnuplot.AppendExtra ("set grid ytics lt 0 lw 1 lc rgb \"#ccc\""); 
    m_gnuplot.AppendExtra ("set grid xtics lt 0 lw 1 lc rgb \"#ccc\""); 
    
    m_gnuplot.AppendExtra ("set mxtics 5"); 
    m_gnuplot.AppendExtra ("set grid mxtics xtics ls 7, ls 8"); 
 
    m_gnuplot.AppendExtra ("set arrow from graph 1,0 to graph 1.03,0 size screen 0.025,15,60 filled ls 3"); 
    m_gnuplot.AppendExtra ("set arrow from graph 0,1 to graph 0,1.03 size screen 0.025,15,60 filled ls 3");
 
    m_gnuplot.AppendExtra ("set style fill transparent solid 0.4 noborder");  
    //m_gnuplot.AppendExtra ("set logscale y 2");  
 
    if (m_plotFileType=="tex") {
        m_gnuplot.AppendExtra ("set key inside");  
        m_gnuplot.AppendExtra ("set key right top");   
    }else{
        m_gnuplot.AppendExtra ("set key outside");  
        m_gnuplot.AppendExtra ("set key right bottom"); 
    }
 
    std::string plotTitle;
 
    if(graphName.empty()){
        plotTitle = "QKD Key Material in Whole Network"; 
    }else{
         plotTitle = graphName; 
    }
 
    m_gnuplot.SetTitle (plotTitle); 
    //m_gnuplot.SetTerminal ("png");
    // Set the labels for each axis.
    m_gnuplot.SetLegend ("Time (seconds)", "Key material (bits)");
 
}
 
 
void
QKDTotalGraph::PrintGraph(){
 
    NS_LOG_FUNCTION (this << m_plotFileName);
  
    std::string tempPlotFileName1 = m_plotFileName;
    tempPlotFileName1 += "_data.dat"; 
    std::ofstream tempPlotFile1 (tempPlotFileName1.c_str());
    
    m_gnuplot.AddDataset (m_dataset);
    m_gnuplot.AddDataset (m_datasetThreshold); 
 
    m_plotFileName += ".plt"; 
    // Open the plot file.
    std::ofstream plotFile (m_plotFileName.c_str());
 
    // Write the plot file.
    m_gnuplot.GenerateOutput (plotFile, tempPlotFile1, tempPlotFileName1);
 
    // Close the plot file.
    plotFile.close ();  
}
 
void
QKDTotalGraph::ProcessMCurrent(uint32_t newValue, char sign){ 
    
    NS_LOG_FUNCTION (this << newValue << sign);
    m_simulationTime = Simulator::Now().GetSeconds();   
 
    if(sign == '+')
        m_keymCurrent += newValue;
    else if(sign == '-')
        m_keymCurrent -= newValue; 
 
    m_dataset.Add( m_simulationTime, m_keymCurrent);  
}
 
void
QKDTotalGraph::ProcessMThr(uint32_t newValue, char sign){ 
    
    NS_LOG_FUNCTION (this << newValue << sign);
    m_simulationTime = Simulator::Now().GetSeconds();   
 
    if(sign == '+')
        m_keymThreshold += newValue;
    else if(sign == '-')
        m_keymThreshold -= newValue; 
 
    m_datasetThreshold.Add( m_simulationTime, m_keymThreshold);  
} 
 
}