aboutsummaryrefslogtreecommitdiff
path: root/extension/src/openvic2/Date.cpp
blob: 3c5b8d86645a57ac6f47431e234a8914696f653b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#include <sstream>
#include "Date.hpp"

namespace OpenVic2 {
   bool Timespan::operator<  (Timespan const& other) const { return days <  other.days; }
   bool Timespan::operator>  (Timespan const& other) const { return days >  other.days; }
   bool Timespan::operator<= (Timespan const& other) const { return days <= other.days; }
   bool Timespan::operator>= (Timespan const& other) const { return days >= other.days; }
   bool Timespan::operator== (Timespan const& other) const { return days == other.days; }
   bool Timespan::operator!= (Timespan const& other) const { return days != other.days; }

   Timespan Timespan::operator+ (Timespan const& other) const { return Timespan(days + other.days); }
   Timespan Timespan::operator- (Timespan const& other) const { return Timespan(days - other.days); }
   Timespan Timespan::operator* (int64_t const& factor) const { return Timespan(days * factor); }
   Timespan Timespan::operator/ (int64_t const& factor) const { return Timespan(days / factor); }

   Timespan& Timespan::operator+= (Timespan const& other) {
      days += other.days;
      return *this;
   }
   Timespan& Timespan::operator-= (Timespan const& other) {
      days -= other.days;
      return *this;
   }

   Timespan fromYearZero(year_t year, month_t month, date_t day) {
      int64_t daysElapsed = year * DAYS_IN_YEAR;
      size_t daysSinceMonthStart = (day == 0) ? 0 : day - 1; //Underflow protection
      for (size_t x = 0; x < month && x < MONTHS_IN_YEAR; x++) {
         daysElapsed += DAYS_IN_MONTH[x];
      }
      daysElapsed += daysSinceMonthStart;
      return Timespan(daysElapsed);
   }

   //This function is not set up to handle dates before Year 0
   YearMonthDayBundle toGregorianDate(Timespan const& timespan) {
      year_t year = 0;
      month_t month = 0;
      date_t day = 0;

      if (timespan >= 0) {
         year = timespan.days / DAYS_IN_YEAR;
         int64_t remainingDays = timespan.days % DAYS_IN_YEAR;

         for (size_t x = 0; x < MONTHS_IN_YEAR && remainingDays >= DAYS_IN_MONTH[x]; x++) {
            remainingDays -= DAYS_IN_MONTH[x];
            month++;
         }

         //Corrects month and day to be 1-indexed
         month++;
         day++;
      }
      return std::make_tuple(year, month, day);
   }


   Date::Date(Timespan const& timespan) : ts(timespan) { updateDate(ts); }

   Date::Date(year_t year, month_t month, date_t day) {
      ts = fromYearZero(year, month, day);
      updateDate(ts);
   }

   void Date::updateDate(Timespan const& timespan) {
      gregorianDate = toGregorianDate(timespan);
   }

   size_t Date::getDay()   const { return std::get<2>(gregorianDate); }
   size_t Date::getMonth() const { return std::get<1>(gregorianDate); }
   size_t Date::getYear()  const { return std::get<0>(gregorianDate); }

   bool Date::operator<  (Date const& other) const { return ts <  other.ts; }
   bool Date::operator>  (Date const& other) const { return ts >  other.ts; }
   bool Date::operator<= (Date const& other) const { return ts <= other.ts; }
   bool Date::operator>= (Date const& other) const { return ts >= other.ts; }
   bool Date::operator== (Date const& other) const { return ts == other.ts; }
   bool Date::operator!= (Date const& other) const { return ts != other.ts; }

   Date Date::operator+ (Timespan timespan) const { return Date(ts + timespan); }
   Timespan Date::operator- (Date const& other) const { return ts - other.ts; }

   Date& Date::operator+= (Timespan const& timespan) {
      ts += timespan;
      updateDate(ts);
      return *this;
   }
   Date& Date::operator-= (Timespan const& timespan) {
      ts -= timespan;
      updateDate(ts);
      return *this;
   }
   Date Date::operator++ (int) {
      Date oldCopy = *this;
      (*this) += 1;
      return oldCopy;
   }

   Date::operator std::string() const {
      std::stringstream ss;
      ss << getYear() << '.' << getMonth() << '.' << getDay();
      return ss.str();
   }
   std::ostream &operator<<(std::ostream &out, Date const& date) {
      return out << static_cast<std::string>(date);
   }
}