X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/becf6dfe4fe2ecd43792aa53a302c5866483f306..a5ce0ca3674081d3b761eb1bc88654d6fd9d3755:/projects/stargazer/tariff_impl.h

diff --git a/projects/stargazer/tariff_impl.h b/projects/stargazer/tariff_impl.h
index 2086b488..cdfd7112 100644
--- a/projects/stargazer/tariff_impl.h
+++ b/projects/stargazer/tariff_impl.h
@@ -28,60 +28,64 @@
  $Author: faust $
  */
 
-#ifndef TARIFF_H
-#define TARIFF_H
+#ifndef TARIFF_IMPL_H
+#define TARIFF_IMPL_H
 
 #include <ctime>
 
 #include <string>
 #include <list>
 
-#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
 
-class TARIFF
-{
+class TARIFF_IMPL : public TARIFF {
 public:
-    TARIFF()
-        : tariffData()
+    TARIFF_IMPL()
+        : TARIFF(),
+          tariffData()
     {};
-    TARIFF(const std::string & name)
-        : tariffData(name)
+    TARIFF_IMPL(const std::string & name)
+        : TARIFF(),
+          tariffData(name)
     {};
-    TARIFF(const TARIFF_DATA & td)
-        : tariffData(td)
+    TARIFF_IMPL(const TARIFF_DATA & td)
+        : TARIFF(),
+          tariffData(td)
     {};
-    TARIFF(const TARIFF & t)
-        : tariffData(t.tariffData)
+    TARIFF_IMPL(const TARIFF_IMPL & t)
+        : TARIFF(),
+          tariffData(t.tariffData)
     {};
-    ~TARIFF() {};
+    virtual ~TARIFF_IMPL() {};
 
     double  GetPriceWithTraffType(uint64_t up,
                                   uint64_t down,
                                   int dir,
                                   time_t t) const;
-    double  GetFreeMb() const { return tariffData.tariffConf.free; };
-    double  GetPassiveCost() const { return tariffData.tariffConf.passiveCost; };
-    double  GetFee() const { return tariffData.tariffConf.fee; };
-    double  GetFree() const { return tariffData.tariffConf.free; };
+    double  GetFreeMb() const { return tariffData.tariffConf.free; }
+    double  GetPassiveCost() const { return tariffData.tariffConf.passiveCost; }
+    double  GetFee() const { return tariffData.tariffConf.fee; }
+    double  GetFree() const { return tariffData.tariffConf.free; }
 
-    void    PrintTariff() const;
+    void    Print() const;
 
-    const   std::string & GetName() const { return tariffData.tariffConf.name; };
-    void    SetName(const std::string & name) { tariffData.tariffConf.name = name; };
+    const   std::string & GetName() const { return tariffData.tariffConf.name; }
+    void    SetName(const std::string & name) { tariffData.tariffConf.name = name; }
 
-    int     GetTraffType() const { return tariffData.tariffConf.traffType; };
+    int     GetTraffType() const { return tariffData.tariffConf.traffType; }
     int64_t GetTraffByType(uint64_t up, uint64_t down) const;
     int     GetThreshold(int dir) const;
-    void    GetTariffData(TARIFF_DATA * td) 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(); };
+    bool     operator==(const TARIFF & rhs) const { return GetName() == rhs.GetName(); }
+    bool     operator!=(const TARIFF & rhs) const { return GetName() != rhs.GetName(); }
 
 private:
     TARIFF_DATA     tariffData;
@@ -89,6 +93,5 @@ private:
     double  GetPriceWithoutFreeMb(int dir, int mb, time_t t) const;
     int     Interval(int dir, time_t t) const;
 };
-//-----------------------------------------------------------------------------
 
 #endif