X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/9fdd58e794e43fc2435265c4fb2f6933758f3c0b..9c340da9ffa670c20c3658a2be43d35868a9f5d9:/include/stg/user_traff.h diff --git a/include/stg/user_traff.h b/include/stg/user_traff.h index 1b9e2035..d820588e 100644 --- a/include/stg/user_traff.h +++ b/include/stg/user_traff.h @@ -45,12 +45,16 @@ public: typedef ContainerType::size_type IndexType; DIR_TRAFF() : traff(DIR_NUM) {} - DIR_TRAFF(const DIR_TRAFF & ts) : traff(ts.traff) {} - DIR_TRAFF & operator=(const DIR_TRAFF & ts) { traff = ts.traff; return *this; } const uint64_t & operator[](IndexType idx) const { return traff[idx]; } uint64_t & operator[](IndexType idx) { return traff[idx]; } IndexType size() const { return traff.size(); } + void Reset() + { + for (IndexType i = 0; i < traff.size(); ++i) + traff[i] = 0; + } + private: ContainerType traff; }; @@ -79,12 +83,18 @@ public: typedef ContainerType::size_type IndexType; DIR_TRAFF_RES() : traff(DIR_NUM) {} - DIR_TRAFF_RES(const DIR_TRAFF & ts) + explicit DIR_TRAFF_RES(const DIR_TRAFF & ts) : traff(ts.size()) { for (IndexType i = 0; i < ts.size(); ++i) traff[i] = ts[i]; } + DIR_TRAFF_RES & operator=(const DIR_TRAFF & ts) + { + for (IndexType i = 0; i < ts.size(); ++i) + traff[i] = ts[i]; + return *this; + } const ValueType & operator[](IndexType idx) const { return traff[idx]; } ValueType & operator[](IndexType idx) { return traff[idx]; } DIR_TRAFF GetData() const