Pileus Git - lackey/blob - view/year.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 #include <string.h>
  19 #include <ncurses.h>
  20
  21 #include "main.h"
  22 #include "util.h"
  23
  24 /* Constants */
  25 #define MW (2*7+6)
  26
  27 /* Static data */
  28 static WINDOW *win;
  29
  30 /* Helper functions */
  31 static void print_month(month_t month, int y, int x)
  32 {
  33         const char *name  = month_to_string(month);
  34         const int   start = start_of_month(YEAR, month);
  35         const char  days  = days_in_month(YEAR, month);
  36         mvwprintw(win, y, x+MW/2-strlen(name)/2, "%s", name);
  37         wmove(win, y+1, x);
  38         for (int d = 0; d < 7; d++)
  39                 wprintw(win, "%-3s", day_to_st(d));
  40         for (int d = 0; d < days; d++) {
  41                 int row = (start + d) / 7;
  42                 int col = (start + d) % 7;
  43                 if (month == MONTH && d == DAY) wattron(win, A_REVERSE);
  44                 mvwprintw(win, y+2+row, x+col*3, "%2d", d+1);
  45                 if (month == MONTH && d == DAY) wattroff(win, A_REVERSE);
  46         }
  47 }
  48
  49 /* Year init */
  50 void year_init(WINDOW *_win)
  51 {
  52         win = _win;
  53 }
  54
  55 /* Year draw */
  56 void year_draw(void)
  57 {
  58         int w = MW*3 + 2*3;
  59         int x = COLS/2 - w/2;
  60         int y = 0;
  61         int h[4] = {};
  62
  63         /* Clear */
  64         werase(win);
  65
  66         /* Determine heights */
  67         for (int m = 0; m < 12; m++) {
  68                 int weeks = weeks_in_month(YEAR, m);
  69                 h[m/3] = MAX(h[m/3], weeks+2);
  70         }
  71         int sum = h[0]+h[1]+h[2]+h[3];
  72
  73         /* Print Header */
  74         mvwprintw(win, y++, COLS/2-2, "%d", YEAR);
  75
  76         /* Print Months */
  77         for (int m = 0; m < 12; m++) {
  78                 print_month(m, y, x);
  79                 if (m % 3 == 2) {
  80                         x  = COLS/2 - w/2;
  81                         y += h[m/3]+1;
  82                 } else {
  83                         x += 3+MW;
  84                 }
  85         }
  86
  87         /* Print Lines */
  88         y = 1;
  89         mvwvline(win, y, x+(MW+3)*1-2, ACS_VLINE, sum+3);
  90         mvwvline(win, y, x+(MW+3)*2-2, ACS_VLINE, sum+3);
  91         for (int i = 0; i < 3; i++) {
  92                 y += h[i];
  93                 mvwhline(win, y, x,        ACS_HLINE, w);
  94                 mvwaddch(win, y, x+(MW+3)*1-2, ACS_PLUS);
  95                 mvwaddch(win, y, x+(MW+3)*2-2, ACS_PLUS);
  96                 y++;
  97         }
  98 }
  99
 100 /* Year run */
 101 int year_run(int key, mmask_t btn, int row, int col)
 102 {
 103         day_t d = DAY;
 104         month_t m = MONTH;
 105         year_t y = YEAR;
 106         wday_t day = day_of_week(YEAR, MONTH, DAY);
 107         int week = (start_of_month(y, m) + d) / 7;
 108         int dir = 0;
 109
 110         /* Step years */
 111         if (key == 'i')
 112                 YEAR--;
 113         if (key == 'o')
 114                 YEAR++;
 115
 116         /* Get direction */
 117         if (key == 'h' || key == 'k')
 118                 dir = -1;
 119         if (key == 'j' || key == 'l')
 120                 dir =  1;
 121
 122         /* Step up/down */
 123         if (key == 'j' || key == 'k') {
 124                 for (int i = 0; i < 90/7; i++) {
 125                         add_days(&y, &m, &d, dir*7);
 126                         if (day_of_week(y, m, d) == day &&
 127                             y == YEAR && m%3 == MONTH%3) {
 128                                 MONTH = m;
 129                                 DAY = d;
 130                                 break;
 131                         }
 132                 }
 133         }
 134
 135         /* Step left/right */
 136         if (key == 'h' || key == 'l') {
 137                 for (int i = 0; i < 90; i++) {
 138                         add_days(&y, &m, &d, dir);
 139                         if ((start_of_month(y, m) + d) / 7 == week &&
 140                             y == YEAR && m/3 == MONTH/3) {
 141                                 MONTH = m;
 142                                 DAY = d;
 143                                 break;
 144                         }
 145                 }
 146         }
 147
 148         /* Refresh */
 149         year_draw();
 150         wrefresh(win);
 151         return 0;
 152 }