source: XIOS/dev/XIOS_DEV_CMIP6/src/calendar_util.cpp @ 2328

#include "calendar_util.hpp"

namespace xios
{
    /// ////////////////////// Définitions ////////////////////// ///

    CDuration operator*(const double& scal, const CDuration& ddr)
    { return (ddr * scal); }

    CDuration operator-(const CDuration& ddr, const CDuration& dr)
    {
      CDuration dur(ddr);
      dur.year -= dr.year; dur.month  -= dr.month;  dur.day    -= dr.day;
      dur.hour -= dr.hour; dur.minute -= dr.minute; dur.second -= dr.second; dur.timestep -= dr.timestep;
      return dur;
    }

    CDuration operator+(const CDuration& ddr, const CDuration& dr)
    {
      CDuration dur(ddr);
      dur.year += dr.year; dur.month  += dr.month;  dur.day    += dr.day;
      dur.hour += dr.hour; dur.minute += dr.minute; dur.second += dr.second; dur.timestep += dr.timestep;
      return dur;
    }

    CDuration operator*(const CDuration& ddr, const double& scal)
    {
      CDuration dur(ddr);
      dur.year *= scal; dur.month  *= scal; dur.day    *= scal;
      dur.hour *= scal; dur.minute *= scal; dur.second *= scal; dur.timestep *= scal;
      return dur;
    }

    CDuration operator-(const CDuration& ddr)
    {
      CDuration dur(ddr);
      dur.year     = -dur.year;
      dur.month    = -dur.month;
      dur.day      = -dur.day;
      dur.hour     = -dur.hour;
      dur.minute   = -dur.minute;
      dur.second   = -dur.second;
      dur.timestep = -dur.timestep;
      return dur;
    }

    //-----------------------------------------------------------------

    CDate operator+(const CDate& dt, const CDuration& dr)
    {
      CDuration drr(dr);
      int year = 0, month = 0, day = 0, hour = 0, minute = 0, second = 0;
      const CCalendar& c = dt.getRelCalendar();
      const bool calendarHasMonths = (c.getYearLength() > 0);

      drr.timestep = 0;
      if (dr.timestep)
      {
        if (c.getTimeStep() == NoneDu)
          ERROR("operator+(const CDate& dt, const CDuration& dr)",
                << "Impossible to use the timestep before it is set.");
        drr = drr + dr.timestep * c.getTimeStep();
      }

      // Handle the time part of the date
      drr.second += dt.getSecond();
      drr.minute += dt.getMinute();
      drr.hour   += dt.getHour();

      if (!calendarHasMonths) // Handle the day and year immediately if there is no months
      {
        drr.day  += dt.getDay() - 1;
        drr.year += dt.getYear();
      }

      drr.resolve(c, true); // Force the time to be positive

      second = drr.second;
      minute = drr.minute;
      hour   = drr.hour;

      if (calendarHasMonths)
      {
        // Ajustement des mois en fonction des jours.
        CDate dtt(dt);
        drr.day += dtt.getDay() - 1;
        dtt.setDay(1);

        if (drr.day >= 0)
        {
          for(; c.getMonthLength(dtt) <= drr.day; dtt.addMonth(1))
          { drr.day -= c.getMonthLength(dtt); drr.month += 1; }

          day = drr.day + 1;
        }
        else
        {
          dtt.addMonth(-1);
          drr.month -= 1;
          for(; c.getMonthLength(dtt) < -drr.day; dtt.addMonth(-1))
          { drr.day += c.getMonthLength(dtt); drr.month -= 1; }
          day = c.getMonthLength(dtt) + drr.day + 1;
        }

        drr.resolve(c);

        // Ajustement des années en fonction des mois.
        month += dt.getMonth() + drr.month;
        if (month < 0) { drr.year--; month += c.getYearLength(); }
        if (month > c.getYearLength()) { drr.year++; month -= c.getYearLength(); }
        if (month == 0){ month = c.getYearLength(); drr.year--; }

        year += dt.getYear() + drr.year;
      }
      else // if (!calendarHasMonths)
      {
        day   = drr.day + 1;
        month = 1;
        year  = drr.year;
      }

      return (CDate(c, year, month, day, hour, minute, second));
    }

    CDate operator-(const CDate& dt, const CDuration& dr) { return (dt + (-dr)); }

    //-----------------------------------------------------------------

    CDuration operator-(const CDate& dt0, const CDate& dt1)
    {
      // TODO :: Vérifier que les deux dates (dt0 et dt1) ont une référence vers le même calendrier.
      CDuration dur( dt0.getYear() - dt1.getYear(), dt0.getMonth()  - dt1.getMonth() , dt0.getDay()   - dt1.getDay(),
      dt0.getHour() - dt1.getHour(), dt0.getMinute() - dt1.getMinute(), dt0.getSecond() - dt1.getSecond() );
      return (dur.resolve(dt0.getRelCalendar()));
    }

    //-----------------------------------------------------------------

    /// Les opérateurs de comparaison. (Non testés pour le moment)

    bool operator==(const CDuration& ddr, const CDuration& dr)
    {
      return ((ddr.year == dr.year) && (ddr.month  == dr.month)  && (dr.day    == ddr.day) &&
              (ddr.hour == dr.hour) && (ddr.minute == dr.minute) && (dr.second == ddr.second) &&
              (ddr.timestep == dr.timestep));
    }

    bool operator!=(const CDuration& ddr, const CDuration& dr)
    {
      return !(ddr == dr);
    }

    bool operator==(const CDate& dt0, const CDate& dt1)
    {
      // TODO :: Vérifier que les deux dates (dt0 et dt1) ont une référence vers le même calendrier.
      return ((dt0.getYear() == dt1.getYear()) && (dt0.getMonth()  == dt1.getMonth())  && (dt1.getDay()    == dt0.getDay()) &&
              (dt0.getHour() == dt1.getHour()) && (dt0.getMinute() == dt1.getMinute()) && (dt1.getSecond() == dt0.getSecond()));
    }

    bool operator<(const CDate& dt0, const CDate& dt1)
    {
      // TODO :: Vérifier que les deux dates (dt0 et dt1) ont une référence vers le même calendrier.
      if (dt0.getYear() < dt1.getYear())
      {
        return true;
      }
      else if (dt0.getYear() == dt1.getYear())
      {
        if (dt0.getMonth() < dt1.getMonth())
        {
          return true;
        }
        else if (dt0.getMonth() == dt1.getMonth())
        {
          if (dt0.getDay() < dt1.getDay())
          {
             return true;
          }
          else if (dt0.getDay() == dt1.getDay())
          {
            if (dt0.getHour() < dt1.getHour())
            {
              return true;
            }
            else if (dt0.getHour() == dt1.getHour())
            {
              if (dt0.getMinute() < dt1.getMinute())
              {
                return true;
              }
              else if (dt0.getMinute() == dt1.getMinute())
              {
                if (dt0.getSecond() < dt1.getSecond())
                  return true;
              }
            }
          }
        }
      }
      return false;
    }

    //-----------------------------------------------------------------

    bool operator!=(const CDate& dt0, const CDate& dt1) { return !(dt1 == dt0); }
    bool operator> (const CDate& dt0, const CDate& dt1) { return (dt1 < dt0); }
    bool operator>=(const CDate& dt0, const CDate& dt1) { return (dt0 > dt1 || dt1 == dt0); }
    bool operator<=(const CDate& dt0, const CDate& dt1) { return (dt0 < dt1 || dt1 == dt0); }

    ///----------------------------------------------------------------

    bool DurationFakeLessComparator::operator()(const CDuration& dur1, const CDuration& dur2) const
    {
      if (dur1.year < dur2.year)
        return true;
      else if (dur1.year == dur2.year)
      {
        if (dur1.month < dur2.month)
          return true;
        else if (dur1.month == dur2.month)
        {
          if (dur1.day < dur2.day)
            return true;
          else if (dur1.day == dur2.day)
          {
            if (dur1.hour < dur2.hour)
              return true;
            else if (dur1.hour == dur2.hour)
            {
              if (dur1.minute < dur2.minute)
                return true;
              else if (dur1.minute == dur2.minute)
              {
                if (dur1.second < dur2.second)
                  return true;
                else if (dur1.second == dur2.second)
                  return (dur1.timestep < dur2.timestep);
              }
            }
          }
        }
      }
      return false;
    }

    ///----------------------------------------------------------------

} // namespace xios