Pileus Git - lackey/blob - src/date.c

   1 /*
   2  * Copyright (C) 2012 Andy Spencer <andy753421@gmail.com>
   3  *
   4  * This program is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 /* Time Keeping Bugs Abound! */
  19
  20 #include <stdio.h>
  21 #include <time.h>
  22
  23 #include "date.h"
  24
  25 /* Global data */
  26 date_t  NOW;
  27 date_t  SEL;
  28
  29 /* Initialize */
  30 void date_init(void)
  31 {
  32         time_t     sec = time(NULL);
  33         struct tm *tm  = localtime(&sec);
  34
  35         SEL.year  = tm->tm_year+1900;
  36         SEL.month = tm->tm_mon;
  37         SEL.day   = tm->tm_mday-1;
  38
  39         date_sync();
  40 }
  41
  42 void date_sync(void)
  43 {
  44         time_t     sec = time(NULL);
  45         struct tm *tm  = localtime(&sec);
  46
  47         NOW.year  = tm->tm_year+1900;
  48         NOW.month = tm->tm_mon;
  49         NOW.day   = tm->tm_mday-1;
  50         NOW.hour  = tm->tm_hour;
  51         NOW.min   = tm->tm_min;
  52         NOW.sec   = tm->tm_sec;
  53 }
  54
  55 /* Time functions */
  56 int is_leap_year(year_t year)
  57 {
  58         return (year % 400 == 0) ? 1 :
  59                (year % 100 == 0) ? 0 :
  60                (year % 4   == 0) ? 1 : 0;
  61 }
  62
  63 int days_in_year(year_t year)
  64 {
  65         return 365 + is_leap_year(year);
  66 }
  67
  68 int days_in_month(year_t year, month_t month)
  69 {
  70         static int mdays[] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
  71         int days = mdays[month];
  72         if (month == FEB)
  73                 days += is_leap_year(year);
  74         return days;
  75 }
  76
  77 int weeks_in_month(year_t year, month_t month)
  78 {
  79         int start = start_of_month(year, month);
  80         int days  = days_in_month(year, month);
  81         return ((start + days)-1) / 7 + 1;
  82 }
  83
  84 wday_t day_of_week(year_t year, month_t month, day_t day)
  85 {
  86         static int tmp[] = {0, 3, 2, 5, 0, 3,
  87                             5, 1, 4, 6, 2, 4};
  88         if (month < MAR)
  89                 year--;
  90         int start = year + year / 4
  91                          - year / 100
  92                          + year / 400
  93                          + tmp[month];
  94         return (start + day + 1) % 7;
  95 }
  96
  97 wday_t start_of_month(year_t year, month_t month)
  98 {
  99         return day_of_week(year, month, 0);
 100 }
 101
 102 void add_days(year_t *year, month_t *month, day_t *day, int days)
 103 {
 104         time_t time = mktime(&(struct tm){
 105                         .tm_year = *year-1900,
 106                         .tm_mon  = *month,
 107                         .tm_mday = *day+1,
 108                         .tm_hour = 12});
 109         time  += days*24*60*60;
 110         struct tm *tm = localtime(&time);
 111         *year  = tm->tm_year+1900;
 112         *month = tm->tm_mon;
 113         *day   = tm->tm_mday-1;
 114 }
 115
 116 void add_months(year_t *year, month_t *month, int months)
 117 {
 118         int total = *year*12 + *month + months;
 119         *year  = total / 12;
 120         *month = total % 12;
 121 }
 122
 123 stamp_t get_time(date_t *date)
 124 {
 125         return mktime(&(struct tm){
 126                 .tm_year = date->year-1900,
 127                 .tm_mon  = date->month,
 128                 .tm_mday = date->day+1,
 129                 .tm_hour = date->hour,
 130                 .tm_min  = date->min});
 131 }
 132
 133 int get_mins(date_t *start, date_t *end)
 134 {
 135         return (get_time(end)-get_time(start))/60;
 136 }
 137
 138 int compare(date_t *a, date_t *b)
 139 {
 140         int rval = a->year  < b->year  ? -1 : a->year  > b->year  ? 1 :
 141                    a->month < b->month ? -1 : a->month > b->month ? 1 :
 142                    a->day   < b->day   ? -1 : a->day   > b->day   ? 1 :
 143                    a->hour  < b->hour  ? -1 : a->hour  > b->hour  ? 1 :
 144                    a->min   < b->min   ? -1 : a->min   > b->min   ? 1 : 0;
 145         return rval;
 146 }
 147
 148 int same_day(date_t *a, date_t *b)
 149 {
 150         return a->year  == b->year  &&
 151                a->month == b->month &&
 152                a->day   == b->day;
 153 }
 154
 155 int before(date_t *start, int year, int month, int day, int hour, int min)
 156 {
 157         return compare(start, &(date_t){year, month, day, hour, min}) < 0;
 158 }
 159
 160 int all_day(date_t *start, date_t *end)
 161 {
 162         date_t test = *start;
 163         add_days(&test.year, &test.month, &test.day, 1);
 164         return compare(&test, end) <= 0;
 165 }
 166
 167 int no_date(date_t *date)
 168 {
 169         return date->year == 0;
 170 }
 171
 172 /* Debug functions */
 173 const char *month_to_str(month_t month)
 174 {
 175         static const char *map[] =
 176                 { "Jan", "Feb", "Mar", "Apr", "May", "Jun",
 177                   "Jul", "Aug", "Sep", "Oct", "Nov", "Dec", };
 178         return map[month%12];
 179 }
 180 const char *month_to_string(month_t month)
 181 {
 182         static const char *map[] =
 183                 { "January",   "February", "March",    "April",
 184                   "May",       "June",     "July",     "August",
 185                   "September", "October",  "November", "December" };
 186         return map[month%12];
 187 }
 188
 189 const char *day_to_st(wday_t day)
 190 {
 191         static const char *map[] =
 192                 { "Su","Mo", "Tu", "We", "Th", "Fr", "Sa" };
 193         return map[day%7];
 194 }
 195 const char *day_to_str(wday_t day)
 196 {
 197         static const char *map[] =
 198                 { "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat" };
 199         return map[day%7];
 200 }
 201 const char *day_to_string(wday_t day)
 202 {
 203         static const char *map[] =
 204                 { "Sunday",   "Monday", "Tuesday", "Wednesday",
 205                   "Thursday", "Friday", "Saturday" };
 206         return map[day%7];
 207 }
 208
 209 /* Test functions */
 210 void date_test(void)
 211 {
 212         printf("Info\n");
 213         printf("  Year   Month       Start   Weeks   Days\n");
 214         for (int y = 2012; y <= 2012; y++)
 215         for (int m = JAN;  m <= DEC;  m++) {
 216                 printf("  %-5d",  y);
 217                 printf("  %-10s", month_to_string(m));
 218                 printf("  %-6s",  day_to_str(start_of_month(y,m)));
 219                 printf("  %-6d",  weeks_in_month(y,m));
 220                 printf("  %-2d",  days_in_month(y,m));
 221                 printf("\n");
 222         }
 223 }