X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/6f7160dd027c8ce7e2760aa369768ba39ab80181..638e02a86c3c80557ef75c38f9e6f3f56628ede1:/projects/stargazer/tariff_impl.h?ds=sidebyside

diff --git a/projects/stargazer/tariff_impl.h b/projects/stargazer/tariff_impl.h
index aaa6a3b9..8fc32fcf 100644
--- a/projects/stargazer/tariff_impl.h
+++ b/projects/stargazer/tariff_impl.h
@@ -36,9 +36,9 @@
 #include <string>
 #include <list>
 
-#include "tariff.h"
-#include "os_int.h"
-#include "tariff_conf.h"
+#include "stg/tariff.h"
+#include "stg/os_int.h"
+#include "stg/tariff_conf.h"
 
 #define TARIFF_DAY     0
 #define TARIFF_NIGHT   1
@@ -46,18 +46,22 @@
 class TARIFF_IMPL : public TARIFF {
 public:
     TARIFF_IMPL()
-        : tariffData()
-    {};
+        : TARIFF(),
+          tariffData()
+    {}
     TARIFF_IMPL(const std::string & name)
-        : tariffData(name)
-    {};
+        : TARIFF(),
+          tariffData(name)
+    {}
     TARIFF_IMPL(const TARIFF_DATA & td)
-        : tariffData(td)
-    {};
+        : TARIFF(),
+          tariffData(td)
+    {}
     TARIFF_IMPL(const TARIFF_IMPL & t)
-        : tariffData(t.tariffData)
-    {};
-    virtual ~TARIFF_IMPL() {};
+        : TARIFF(),
+          tariffData(t.tariffData)
+    {}
+    virtual ~TARIFF_IMPL() {}
 
     double  GetPriceWithTraffType(uint64_t up,
                                   uint64_t down,
@@ -67,6 +71,8 @@ public:
     double  GetPassiveCost() const { return tariffData.tariffConf.passiveCost; }
     double  GetFee() const { return tariffData.tariffConf.fee; }
     double  GetFree() const { return tariffData.tariffConf.free; }
+    PERIOD  GetPeriod() const { return tariffData.tariffConf.period; }
+    CHANGE_POLICY  GetChangePolicy() const { return tariffData.tariffConf.change_policy; }
 
     void    Print() const;
 
@@ -78,15 +84,15 @@ public:
     int     GetThreshold(int dir) const;
     const TARIFF_DATA & GetTariffData() const { return tariffData; }
 
-    TARIFF & operator=(const TARIFF_DATA & td);
-    TARIFF & operator=(const TARIFF & t);
-    bool     operator==(const TARIFF & rhs) const { return GetName() == rhs.GetName(); }
-    bool     operator!=(const TARIFF & rhs) const { return GetName() != rhs.GetName(); }
+    TARIFF_IMPL & operator=(const TARIFF_DATA & td);
+    TARIFF_IMPL & operator=(const TARIFF_IMPL & t);
+    bool     operator==(const TARIFF_IMPL & rhs) const { return GetName() == rhs.GetName(); }
+    bool     operator!=(const TARIFF_IMPL & rhs) const { return GetName() != rhs.GetName(); }
 
 private:
     TARIFF_DATA     tariffData;
 
-    double  GetPriceWithoutFreeMb(int dir, int mb, time_t t) const;
+    double  GetPriceWithoutFreeMb(int dir, int64_t mb, time_t t) const;
     int     Interval(int dir, time_t t) const;
 };