Pileus Git - lackey/blob - views/year.c

   1 /*
   2  * Copyright (C) 2012-2013 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 "util.h"
  22 #include "date.h"
  23 #include "cal.h"
  24 #include "view.h"
  25
  26 /* Constants */
  27 #define MW (2*7+6)
  28
  29 /* Static data */
  30 static WINDOW *win;
  31
  32 /* Helper functions */
  33 static void print_month(month_t month, int y, int x)
  34 {
  35         event_t    *event = EVENTS;
  36         const char *name  = month_to_string(month);
  37         const int   start = start_of_month(SEL.year, month);
  38         const char  days  = days_in_month(SEL.year, month);
  39         mvwprintw(win, y, x+MW/2-strlen(name)/2, "%s", name);
  40         wmove(win, y+1, x);
  41         for (int d = 0; d < 7; d++)
  42                 wprintw(win, "%-3s", day_to_st(d));
  43         for (int d = 0; d < days; d++) {
  44                 int row = (start + d) / 7;
  45                 int col = (start + d) % 7;
  46
  47                 int busy = 0;
  48                 while (event && before(&event->start, SEL.year, month, d, 24, 0)) {
  49                         if (!before(&event->start, SEL.year, month, d, 0, 0))
  50                                 busy = 1;
  51                         event = event->next;
  52                 }
  53
  54                 int today = month == SEL.month && d == SEL.day;
  55
  56                 int attr  = (busy  ? A_BOLD|A_UNDERLINE : A_DIM)
  57                           | (today ? A_REVERSE          : 0    );
  58
  59                 wattron(win, attr);
  60                 mvwprintw(win, y+2+row, x+col*3, "%2d", d+1);
  61                 wattroff(win, attr);
  62         }
  63 }
  64
  65 /* Year init */
  66 void year_init(WINDOW *_win)
  67 {
  68         win = _win;
  69 }
  70
  71 /* Year size */
  72 void year_size(int rows, int cols)
  73 {
  74 }
  75
  76 /* Year draw */
  77 void year_draw(void)
  78 {
  79         int w = MW*3 + 2*3;
  80         int x = COLS/2 - w/2;
  81         int y = 0;
  82         int h[4] = {};
  83
  84         /* Load cal data */
  85         cal_load(SEL.year, 0, 0, 366);
  86
  87         /* Determine heights */
  88         for (int m = 0; m < 12; m++) {
  89                 int weeks = weeks_in_month(SEL.year, m);
  90                 h[m/3] = MAX(h[m/3], weeks+2);
  91         }
  92         int sum = h[0]+h[1]+h[2]+h[3];
  93
  94         /* Print Header */
  95         if (COMPACT) wattron(win, A_REVERSE | A_BOLD);
  96         if (COMPACT) mvwhline(win, y, 0, A_REVERSE | A_BOLD, COLS);
  97         mvwprintw(win, y++, COLS/2-2, "%d", SEL.year);
  98         if (COMPACT) wattroff(win, A_REVERSE | A_BOLD);
  99
 100         /* Print Months */
 101         for (int m = 0; m < 12; m++) {
 102                 print_month(m, y, x);
 103                 if (m % 3 == 2) {
 104                         x  = COLS/2 - w/2;
 105                         y += h[m/3]+1;
 106                 } else {
 107                         x += 3+MW;
 108                 }
 109         }
 110
 111         /* Print Lines */
 112         y = 1;
 113         mvwvline(win, y, x+(MW+3)*1-2, ACS_VLINE, sum+3);
 114         mvwvline(win, y, x+(MW+3)*2-2, ACS_VLINE, sum+3);
 115         for (int i = 0; i < 3; i++) {
 116                 y += h[i];
 117                 mvwhline(win, y, x,        ACS_HLINE, w);
 118                 mvwaddch(win, y, x+(MW+3)*1-2, ACS_PLUS);
 119                 mvwaddch(win, y, x+(MW+3)*2-2, ACS_PLUS);
 120                 y++;
 121         }
 122 }
 123
 124 /* Year run */
 125 int year_run(int key, mmask_t btn, int row, int col)
 126 {
 127         day_t d = SEL.day;
 128         month_t m = SEL.month;
 129         year_t y = SEL.year;
 130         wday_t day = day_of_week(SEL.year, SEL.month, SEL.day);
 131         int week = (start_of_month(y, m) + d) / 7;
 132         int dir = 0;
 133
 134         /* Step years */
 135         if (key == 'i')
 136                 SEL.year--;
 137         if (key == 'o')
 138                 SEL.year++;
 139
 140         /* Get direction */
 141         if (key == 'h' || key == 'k')
 142                 dir = -1;
 143         if (key == 'j' || key == 'l')
 144                 dir =  1;
 145
 146         /* Step up/down */
 147         if (key == 'j' || key == 'k') {
 148                 for (int i = 0; i < 90/7; i++) {
 149                         add_days(&y, &m, &d, dir*7);
 150                         if (day_of_week(y, m, d) == day &&
 151                             y == SEL.year && m%3 == SEL.month%3) {
 152                                 SEL.month = m;
 153                                 SEL.day = d;
 154                                 break;
 155                         }
 156                 }
 157         }
 158
 159         /* Step left/right */
 160         if (key == 'h' || key == 'l') {
 161                 for (int i = 0; i < 90; i++) {
 162                         add_days(&y, &m, &d, dir);
 163                         if ((start_of_month(y, m) + d) / 7 == week &&
 164                             y == SEL.year && m/3 == SEL.month/3) {
 165                                 SEL.month = m;
 166                                 SEL.day = d;
 167                                 break;
 168                         }
 169                 }
 170         }
 171
 172         /* Refresh */
 173         if (dir || y != SEL.year) {
 174                 werase(win);
 175                 year_draw();
 176                 wrefresh(win);
 177         }
 178         return dir || y != SEL.year;
 179 }