root / otg / trunk / src / cpp / auto / cbr_generator_helper.cpp

/*
 * This will be automatically generated in the future
 */

#include <iostream>
#include <sstream>
using namespace std;

#include "popt_helper.h"
#include "cbr_generator_helper.h"

  
static void 
callback(
  poptContext con,
  enum poptCallbackReason reason,
  const struct poptOption * opt,
  char * arg, 
  void * data
) {
  // cout << "CBR Callback: " << opt->longName << " arg: " << arg <<endl;
   
  CBR_Generator* gen = (CBR_Generator*)data;

  if (strcmp(opt->longName, "size") == 0) {
       gen->setPktSize(atoi(arg));
       cout << "packet size parameter is set as "<< atoi(arg) << endl;
  } else  if (strcmp(opt->longName, "interval") == 0) {
    double interv = double(atof(arg))/1000.0;
    cout << "Packet inteval is set as " << interv << "second " <<endl;
    gen->setPktInterval(interv);
  } else if (strcmp(opt->longName, "rate") == 0) {
    double ra = double(atof(arg))*1e3;
    cout << "Rate is set as " << ra  << "bps" << endl;
    gen->setRate(ra);
  }
  // else if (strcmp(opt->longName, "pause") == 0){
  //      gen->pauseGenerator();
  //} else if (strcmp(opt->longName, "resume") == 0){
  //      gen->resumeGenerator();
  //      } 
}

const struct poptOption* 
cbr_generator_get_options(
  const CBR_Generator* gen,
  const char* prefix
) {
  struct poptOption* opts = (struct poptOption*)calloc(5, sizeof(struct poptOption));
  struct poptOption* p = opts;
  popt_set(p++, NULL, NULL, '\0', POPT_ARG_CALLBACK, (void*)callback, 0, (char*)gen);
  popt_set(p++, prefix, "size", '\0', POPT_ARG_INT, NULL, 0, "Size of packet", "bytes");
  popt_set(p++, prefix, "interval", '\0', POPT_ARG_FLOAT, NULL, 0, 
                          "Internval between consecutive packets", "msec");
  popt_set(p++, prefix, "rate", '\0', POPT_ARG_FLOAT, NULL, 0, 
                          "Data rate of the flow", "kbps");
  // popt_set(p++, prefix, "pause",'\0',POPT_ARG_NONE, NULL, 0, "Pause the traffic generation");
  // popt_set(p++, prefix, "resume",'\0',POPT_ARG_NONE, NULL,0,"Resume the traffic generation");
     
  popt_set(p);

  return opts;
}