Ticket 37. The CHANGE_POLICY GetChangePolicy() function definition

[stg.git] / include / stg / tariff.h

diff --git a/include/stg/tariff.h b/include/stg/tariff.h
index 189dde5916c6dfe96a2817be093b5c53a9dd5097..3e150e32b050fd165e8f4b5534d9ac7fd63155b2 100644 (file)
--- a/include/stg/tariff.h
+++ b/include/stg/tariff.h
@@ -21,15 +21,34 @@
 #ifndef TARIFF_H
 #define TARIFF_H
 
 #ifndef TARIFF_H
 #define TARIFF_H
 

-#include <ctime>
+#include "os_int.h"

 
 #include <string>
 
 #include <string>

+#include <cstring>
+#include <ctime>
+#include <istream>

 
 

-#include "os_int.h"
-#include "tariff_conf.h"
+struct TARIFF_DATA;

 
 class TARIFF {
 public:
 
 class TARIFF {
 public:

+    enum CHANGE_POLICY { ALLOW = 0, TO_CHEAP, TO_EXPENSIVE, DENY };
+
+    enum PERIOD { DAY = 0, MONTH };
+
+    enum TRAFF_TYPE { TRAFF_UP = 0, TRAFF_DOWN, TRAFF_UP_DOWN, TRAFF_MAX };
+
+    static std::string ChangePolicyToString(CHANGE_POLICY change_policy);
+    static CHANGE_POLICY StringToChangePolicy(const std::string& value);
+
+    static std::string PeriodToString(PERIOD period);
+    static PERIOD StringToPeriod(const std::string& value);
+
+    static std::string TraffTypeToString(TRAFF_TYPE type);
+    static TRAFF_TYPE StringToTraffType(const std::string& value);
+    static TRAFF_TYPE IntToTraffType(int value);
+
+    virtual ~TARIFF() {}

     virtual double  GetPriceWithTraffType(uint64_t up,
                                           uint64_t down,
                                           int dir,
     virtual double  GetPriceWithTraffType(uint64_t up,
                                           uint64_t down,
                                           int dir,


@@ -38,6 +57,8 @@ public:
     virtual double  GetPassiveCost() const = 0;
     virtual double  GetFee() const = 0;
     virtual double  GetFree() const = 0;
     virtual double  GetPassiveCost() const = 0;
     virtual double  GetFee() const = 0;
     virtual double  GetFree() const = 0;

+    virtual PERIOD  GetPeriod() const = 0;
+    virtual CHANGE_POLICY  GetChangePolicy() const = 0;

 
     virtual const   std::string & GetName() const = 0;
     virtual void    SetName(const std::string & name) = 0;
 
     virtual const   std::string & GetName() const = 0;
     virtual void    SetName(const std::string & name) = 0;


@@ -48,4 +69,92 @@ public:
     virtual const TARIFF_DATA & GetTariffData() const = 0;
 };
 
     virtual const TARIFF_DATA & GetTariffData() const = 0;
 };
 

+inline
+std::string TARIFF::ChangePolicyToString(TARIFF::CHANGE_POLICY change_policy)
+{
+switch (change_policy)
+    {
+    case ALLOW: return "allow";
+    case TO_CHEAP: return "to_cheap";
+    case TO_EXPENSIVE: return "to_expensive";
+    case DENY: return "deny";
+    }
+return "allow"; // Classic behaviour.
+}
+
+inline
+TARIFF::CHANGE_POLICY TARIFF::StringToChangePolicy(const std::string& value)
+{
+if (strcasecmp(value.c_str(), "to_cheap") == 0)
+    return TO_CHEAP;
+if (strcasecmp(value.c_str(), "to_expensive") == 0)
+    return TO_EXPENSIVE;
+if (strcasecmp(value.c_str(), "deny") == 0)
+    return DENY;
+return ALLOW; // Classic behaviour.
+}
+
+inline
+std::string TARIFF::PeriodToString(TARIFF::PERIOD period)
+{
+switch (period)
+    {
+    case DAY: return "day";
+    case MONTH: return "month";
+    }
+return "month"; // Classic behaviour.
+}
+
+inline
+TARIFF::PERIOD TARIFF::StringToPeriod(const std::string& value)
+{
+if (strcasecmp(value.c_str(), "day") == 0)
+    return DAY;
+return MONTH; // Classic behaviour.
+}
+
+inline
+std::string TARIFF::TraffTypeToString(TARIFF::TRAFF_TYPE type)
+{
+switch (type)
+    {
+    case TRAFF_UP: return "up";
+    case TRAFF_DOWN: return "down";
+    case TRAFF_UP_DOWN: return "up+down";
+    case TRAFF_MAX: return "max";
+    }
+return "up+down";
+}
+
+inline
+TARIFF::TRAFF_TYPE TARIFF::StringToTraffType(const std::string& value)
+{
+if (strcasecmp(value.c_str(), "up") == 0)
+    return TRAFF_UP;
+if (strcasecmp(value.c_str(), "down") == 0)
+    return TRAFF_DOWN;
+if (strcasecmp(value.c_str(), "up+down") == 0)
+    return TRAFF_UP_DOWN;
+if (strcasecmp(value.c_str(), "max") == 0)
+    return TRAFF_MAX;
+return TRAFF_UP_DOWN;
+}
+
+inline
+std::istream & operator>>(std::istream & stream, TARIFF::TRAFF_TYPE & traffType)
+{
+    unsigned val;
+    stream >> val;
+    traffType = static_cast<TARIFF::TRAFF_TYPE>(val);
+    return stream;
+}
+
+inline
+TARIFF::TRAFF_TYPE TARIFF::IntToTraffType(int value)
+{
+    if (value < 0 || value > TRAFF_MAX)
+        return TRAFF_UP_DOWN;
+    return static_cast<TRAFF_TYPE>(value);
+}
+

 #endif
 #endif