include/stg/user_traff.h

   1 /*
   2  *    This program is free software; you can redistribute it and/or modify
   3  *    it under the terms of the GNU General Public License as published by
   4  *    the Free Software Foundation; either version 2 of the License, or
   5  *    (at your option) any later version.
   6  *
   7  *    This program is distributed in the hope that it will be useful,
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  *    GNU General Public License for more details.
  11  *
  12  *    You should have received a copy of the GNU General Public License
  13  *    along with this program; if not, write to the Free Software
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  15  */
  16
  17 /*
  18  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
  19  */
  20
  21 /*
  22  $Revision: 1.7 $
  23  $Date: 2010/10/07 19:48:52 $
  24  $Author: faust $
  25  */
  26
  27 #ifndef USER_TRAFF_H
  28 #define USER_TRAFF_H
  29
  30 #include <iostream>
  31 #include <vector>
  32
  33 #include "const.h"
  34 #include "os_int.h"
  35
  36 enum TRAFF_DIRECTION {TRAFF_UPLOAD, TRAFF_DOWNLOAD};
  37
  38 class DIR_TRAFF
  39 {
  40     friend std::ostream & operator<< (std::ostream & o, const DIR_TRAFF & traff);
  41
  42 public:
  43     typedef std::vector<uint64_t> ContainerType;
  44     typedef ContainerType::size_type IndexType;
  45
  46     //-------------------------------------------------------------------------
  47     DIR_TRAFF();
  48     DIR_TRAFF(const DIR_TRAFF & ts);
  49     DIR_TRAFF & operator=(const DIR_TRAFF & ts);
  50     ~DIR_TRAFF();
  51     uint64_t operator[](IndexType idx) const;
  52     uint64_t & operator[](IndexType idx);
  53     DIR_TRAFF operator+(const DIR_TRAFF & ts);
  54
  55 private:
  56     ContainerType traff;
  57 };
  58 //-----------------------------------------------------------------------------
  59
  60 //-----------------------------------------------------------------------------
  61 inline DIR_TRAFF::DIR_TRAFF()
  62     : traff(DIR_NUM, 0)
  63 {
  64 }
  65 //-----------------------------------------------------------------------------
  66 inline DIR_TRAFF::DIR_TRAFF(const DIR_TRAFF & ts)
  67     : traff(ts.traff)
  68 {
  69 }
  70 //-----------------------------------------------------------------------------
  71 inline DIR_TRAFF::~DIR_TRAFF()
  72 {
  73 }
  74 //-----------------------------------------------------------------------------
  75 inline DIR_TRAFF & DIR_TRAFF::operator=(const DIR_TRAFF & ts)
  76 {
  77 traff = ts.traff;
  78 return *this;
  79 }
  80 //-----------------------------------------------------------------------------
  81 inline uint64_t & DIR_TRAFF::operator[](IndexType idx)
  82 {
  83 return traff[idx];
  84 }
  85 //-----------------------------------------------------------------------------
  86 inline uint64_t DIR_TRAFF::operator[](IndexType idx) const
  87 {
  88 return traff[idx];
  89 }
  90 //-----------------------------------------------------------------------------
  91 inline DIR_TRAFF DIR_TRAFF::operator+(const DIR_TRAFF & ts)
  92 {
  93 for (IndexType i = 0; i < DIR_NUM; i++)
  94     {
  95     traff[i] = traff[i] + ts.traff[i];
  96     }
  97 return *this;
  98 }
  99 //-----------------------------------------------------------------------------
 100 inline std::ostream & operator<<(std::ostream & o, const DIR_TRAFF & traff)
 101 {
 102 bool first = true;
 103 for (DIR_TRAFF::IndexType i = 0; i < DIR_NUM; ++i)
 104     {
 105     if (first)
 106         first = false;
 107     else
 108         o << ",";
 109     o << traff[i];
 110     }
 111 return o;
 112 }
 113 //-----------------------------------------------------------------------------
 114 #endif