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