]> git.stg.codes - stg.git/blob - include/stg/user_traff.h
1b9e2035ce9e8466457bf4c2857ff61cf1451367
[stg.git] / 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 "resetable.h"
31 #include "const.h"
32 #include "os_int.h"
33
34 #include <iostream>
35 #include <vector>
36
37 enum TRAFF_DIRECTION {TRAFF_UPLOAD, TRAFF_DOWNLOAD};
38
39 class DIR_TRAFF
40 {
41     friend std::ostream & operator<< (std::ostream & o, const DIR_TRAFF & traff);
42
43 public:
44     typedef std::vector<uint64_t> ContainerType;
45     typedef ContainerType::size_type IndexType;
46
47     DIR_TRAFF() : traff(DIR_NUM) {}
48     DIR_TRAFF(const DIR_TRAFF & ts) : traff(ts.traff) {}
49     DIR_TRAFF & operator=(const DIR_TRAFF & ts) { traff = ts.traff; return *this; }
50     const uint64_t & operator[](IndexType idx) const { return traff[idx]; }
51     uint64_t & operator[](IndexType idx) { return traff[idx]; }
52     IndexType size() const { return traff.size(); }
53
54 private:
55     ContainerType traff;
56 };
57
58 //-----------------------------------------------------------------------------
59 inline
60 std::ostream & operator<<(std::ostream & o, const DIR_TRAFF & traff)
61 {
62 bool first = true;
63 for (DIR_TRAFF::IndexType i = 0; i < traff.size(); ++i)
64     {
65     if (first)
66         first = false;
67     else
68         o << ",";
69     o << traff[i];
70     }
71 return o;
72 }
73
74 class DIR_TRAFF_RES
75 {
76 public:
77     typedef RESETABLE<uint64_t> ValueType;
78     typedef std::vector<ValueType> ContainerType;
79     typedef ContainerType::size_type IndexType;
80
81     DIR_TRAFF_RES() : traff(DIR_NUM) {}
82     DIR_TRAFF_RES(const DIR_TRAFF & ts)
83         : traff(ts.size())
84     {
85     for (IndexType i = 0; i < ts.size(); ++i)
86         traff[i] = ts[i];
87     }
88     const ValueType & operator[](IndexType idx) const { return traff[idx]; }
89     ValueType & operator[](IndexType idx) { return traff[idx]; }
90     DIR_TRAFF GetData() const
91     {
92     DIR_TRAFF res;
93     for (IndexType i = 0; i < traff.size(); ++i)
94         if (!traff[i].empty())
95             res[i] = traff[i].data();
96     return res;
97     }
98
99 private:
100     ContainerType traff;
101 };
102
103 #endif