wmweather+/eta.c

   1 #include "config.h"
   2
   3 /*  Copyright (C) 2002  Brad Jorsch <anomie@users.sourceforge.net>
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program; if not, write to the Free Software
  17     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  18 */
  19
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <time.h>
  23 #include <unistd.h>
  24 #include <string.h>
  25 #include <ctype.h>
  26 #include <sys/stat.h>
  27
  28 #include "wmweather+.h"
  29 #include "forecast.h"
  30 #include "getLine.h"
  31 #include "convert.h"
  32 #include "download.h"
  33 #include "diff.h"
  34 #include "die.h"
  35 #include "sunzenith.h"
  36 #include "moon.h"
  37 #include "subst.h"
  38
  39 /* Important variables */
  40 #define ETA_MAX 21
  41 static time_t eta_time=0;
  42 static char *eta_file=NULL;
  43 static char *eta_newfile=NULL;
  44 static char *eta_req[2]={ NULL, NULL };
  45 static struct forecast forecasts[ETA_MAX];
  46
  47
  48 /********* init functions ************/
  49 static int parse_eta(char *file);
  50
  51 static void reset_eta(void){
  52     int i;
  53
  54     for(i=0; i<ETA_MAX; i++) reset_forecast(&forecasts[i]);
  55 }
  56
  57 void init_eta(void){
  58     char *e;
  59     int i;
  60     struct subst_val subs[]={
  61         { 's', STRING, &eta_station },
  62         { 0, 0, 0 }
  63     };
  64
  65     strncpy(bigbuf, eta_station, BIGBUF_LEN-14);
  66     bigbuf[BIGBUF_LEN-14]='\0';
  67     for(e=bigbuf; *e!='\0'; e++);
  68     strcpy(e, ".eta.txt");
  69     eta_file=get_pid_filename(bigbuf);
  70     strcpy(e, ".new-eta.txt");
  71     eta_newfile=get_pid_filename(bigbuf);
  72
  73     if((eta_req[0]=subst(eta_uri, subs))==NULL) die("init_eta");
  74     if(eta_post!=NULL && (eta_req[1]=subst(eta_post, subs))==NULL) die("init_eta");
  75     eta_time=0;
  76
  77     /* Remove stale file */
  78     unlink(eta_file);
  79     unlink(eta_newfile);
  80     reset_eta();
  81     for(i=0; i<ETA_MAX; i++) add_forecast(&forecasts[i], "ETA", eta_station);
  82 }
  83
  84
  85 /********* download functions ************/
  86
  87 static void eta_callback(char *filename, void *v){
  88     struct stat statbuf;
  89
  90     if(stat(eta_newfile, &statbuf)>=0){
  91         if(S_ISREG(statbuf.st_mode) && statbuf.st_size!=0
  92            && diff(eta_newfile, eta_file) && parse_eta(eta_newfile)){
  93             eta_time=find_next_time(eta_newfile, "MOS GUIDANCE", 720);
  94             rename(eta_newfile, eta_file);
  95         } else {
  96             unlink(eta_newfile);
  97             if(!parse_eta(eta_file)) reset_eta();
  98         }
  99     }
 100 }
 101
 102 void eta_cleanup(void){
 103     if(eta_file==NULL) return;
 104     unlink(eta_newfile);
 105     unlink(eta_file);
 106 }
 107
 108 void update_eta(int force){
 109     time_t t;
 110
 111     if(eta_file==NULL) return;
 112
 113     t=time(NULL)/60;
 114     if(!force && eta_time>t) return;
 115
 116     eta_time=find_next_time(eta_file, "MOS GUIDANCE", 15);
 117     download_file(eta_newfile, eta_req[0], eta_req[1], force?DOWNLOAD_KILL_OTHER_REQUESTS:0, eta_callback, NULL);
 118 }
 119
 120
 121 /********* parse functions ************/
 122
 123 #define NEXT(s) free(s); \
 124     len=getLine(&s, fp); \
 125     if(strstr(s, "</PRE>")!=NULL) len=0;
 126
 127 #define DIE() return (free(s), fclose(fp), 0)
 128 #define SPLIT(s) { \
 129     ID[0]=s[0]; \
 130     ID[1]=s[1]; \
 131     ID[2]=s[2]; \
 132     ID[3]='\0'; \
 133     memset(split,'\0',sizeof(split)); \
 134     for(n=0, c=s+4; c<s+len && n<ETA_MAX; n++, c+=3){ \
 135         split[n][0]=c[0]; \
 136         split[n][1]=c[1]; \
 137         split[n][2]=c[2]; \
 138         split[n][3]='\0'; \
 139     } \
 140 }
 141 #define ASSIGN(field) \
 142     for(n=0; n<ETA_MSG_MAX; n++) forecasts[n].field=atoi(split[n]);
 143
 144 #define ASSIGN2(field, inval) \
 145     for(n=0; n<ETA_MSG_MAX; n++){ \
 146         i=atoi(split[n]); \
 147         if(i!=inval) forecasts[n].field=i; \
 148     }
 149
 150 static int parse_eta(char *file){
 151     int ETA_MSG_MAX = ETA_MAX;
 152     FILE *fp;
 153     char *s, *c;
 154     int len;
 155     int mon, day;
 156     int h, i=0, j, k, m, n, x, y, z;
 157     char ID[4];
 158     char split[ETA_MAX][4];
 159
 160     reset_eta();
 161     if((fp=fopen(file, "r"))==NULL) return 0;
 162
 163     /* Look for something like an ETA coded forecast */
 164     c=NULL;
 165     while(!feof(fp)){
 166         len=getLine(&s, fp);
 167         if((c=strstr(s, "MOS GUIDANCE"))!=NULL) break;
 168         free(s);
 169     }
 170     if(c==NULL) return (fclose(fp), 0);
 171     c=strchr(c, '/');
 172     if(c==NULL || !isdigit(*(c-1)) || !isdigit(*(c+1))) DIE();
 173     m=atoi(c-2);
 174     c=strchr(c+1, '/');
 175     if(c==NULL || !isdigit(*(c-1)) || !isdigit(*(c+1))) DIE();
 176     y=atoi(c+1)-1900;
 177
 178     /* get first date */
 179     NEXT(s);
 180     if(len<10) DIE();
 181     if(strncmp(s, "DT ", 3)) DIE();
 182     mon=13;
 183     c=s;
 184     i=4;
 185     while(mon>12){
 186         c=strchr(c+1, '/');
 187         if(c==NULL) DIE();
 188         for(mon=1; mon<=12; mon++){
 189             if(!strncmp(c+1, monthnames[mon], 3) && isspace(*(c+4))) break;
 190             if(!strncmp(c+1, monthnames2[mon], 4) && isspace(*(c+5))){
 191                 i=5;
 192                 break;
 193             }
 194         }
 195     }
 196     day=atoi(c+i);
 197     if(day<1) DIE();
 198     if(c>s+4) day--;
 199     if(mon<m) y++;
 200
 201     NEXT(s);
 202     if(len<10) DIE();
 203     if(strncmp(s, "HR ", 3)) DIE();
 204     x=day;
 205     m=mon;
 206     SPLIT(s);
 207     for(n=0; n<ETA_MAX; n++){
 208         if(split[n][0]=='\0'){
 209             ETA_MSG_MAX = n;
 210             break;
 211         }
 212         i=atoi(split[n]);
 213         if(i==0){
 214             x++;
 215             fix_date(&mon, &x, &y, NULL);
 216         }
 217         m=mon;
 218         j=x;
 219         z=y;
 220         h=utc2local(i*100, &m, &j, &z, &k)/100;
 221         forecasts[n].month=m;
 222         forecasts[n].day=j;
 223         forecasts[n].year=z;
 224         forecasts[n].hour=h;
 225         forecasts[n].wday=k;
 226         if(latitude!=999 && calcSolarZenith(latitude, longitude, y, mon, x, i*60)>90)
 227             forecasts[n].moon=calc_moon(m, j, z, h*100);
 228     }
 229
 230     while(1){
 231         NEXT(s);
 232         if(len<=10) break;
 233         SPLIT(s);
 234
 235         if(!strcmp(ID, "X/N")) j=1;
 236         else if(!strcmp(ID, "N/X")) j=2;
 237         else j=0;
 238         if(j!=0){
 239             for(n=0; n<ETA_MSG_MAX; n++){
 240                 if(!isdigit(split[n][2])) continue;
 241                 i=atoi(split[n]);
 242                 k=day+(j>>1);
 243                 for(m=0; m<ETA_MSG_MAX; m++){
 244                     if((j&1)==1 &&
 245                        ((forecasts[m].day==k-1 && forecasts[m].hour>=19)
 246                         || (forecasts[m].day==k && forecasts[m].hour<19)))
 247                         forecasts[m].high=i;
 248                     if((j&1)==0 &&
 249                        ((forecasts[m].day==k-1 && forecasts[m].hour>=8)
 250                         || (forecasts[m].day==k && forecasts[m].hour<8)))
 251                         forecasts[m].low=i;
 252                 }
 253                 j++;
 254             }
 255             continue;
 256         }
 257         if(!strcmp(ID, "TMP")){
 258             ASSIGN(temp);
 259             continue;
 260         }
 261         if(!strcmp(ID, "DPT")){
 262             ASSIGN(dewpt);
 263             continue;
 264         }
 265         if(!strcmp(ID, "WDR")){
 266             for(n=0; n<ETA_MSG_MAX; n++){
 267                 i=atoi(split[n]);
 268                 if(i==99) forecasts[n].winddir=0;
 269                 else forecasts[n].winddir=((int)((i+1.125)/2.25))%16+1;
 270             }
 271             continue;
 272         }
 273         if(!strcmp(ID, "WSP")){
 274             ASSIGN2(windspeed, 99);
 275             continue;
 276         }
 277         if(!strcmp(ID, "P06")){
 278             for(m=0; m<ETA_MSG_MAX; m++){
 279                 if(!isdigit(split[m][2])) continue;
 280                 i=atoi(split[m]);
 281                 if(i!=999){
 282                     forecasts[m].pcp_total=i;
 283                     /* ETA_MSG_MAX-2 because the last 2
 284                      * are already 6-hour intervals */
 285                     if(m>0 && m<ETA_MSG_MAX-2) forecasts[m-1].pcp_total=i;
 286                 }
 287             }
 288             continue;
 289         }
 290         if(!strcmp(ID, "T06")){
 291             for(m=1; m<ETA_MSG_MAX; m+=2){
 292                 if(!isdigit(split[m][2])) continue;
 293                 i=atoi(split[m]); if(i==999) i=0;
 294                 j=atoi(split[m+1]+1); if(j==99) j=0;
 295                 j=i*j/100;
 296                 forecasts[m].tstorm=forecasts[m+1].tstorm=i;
 297                 forecasts[m].svtstorm=forecasts[m+1].svtstorm=j;
 298             }
 299             continue;
 300         }
 301         if(!strcmp(ID, "Q06")){
 302             for(m=0; m<ETA_MSG_MAX; m++){
 303                 if(!isdigit(split[m][2])) continue;
 304                 i=atoi(split[m]);
 305                 if(i!=999){
 306                     forecasts[m].precipamt=i;
 307                     /* ETA_MSG_MAX-2 because the last 2
 308                      * are already 6-hour intervals */
 309                     if(m>0 && m<ETA_MSG_MAX-2) forecasts[m-1].precipamt=i;
 310                 }
 311             }
 312             continue;
 313         }
 314         if(!strcmp(ID, "CLD")){
 315             for(m=0; m<ETA_MSG_MAX; m++){
 316                 if(split[m][1]=='C') forecasts[m].sky=0;
 317                 if(split[m][1]=='F') forecasts[m].sky=1;
 318                 if(split[m][1]=='S') forecasts[m].sky=2;
 319                 if(split[m][1]=='B') forecasts[m].sky=3;
 320                 if(split[m][1]=='O') forecasts[m].sky=4;
 321             }
 322             continue;
 323         }
 324         if(!strcmp(ID, "VIS")){
 325             ASSIGN2(vis, 9);
 326             continue;
 327         }
 328         if(!strcmp(ID, "OBV")){
 329             for(m=0; m<ETA_MSG_MAX; m++){
 330                 if(split[m][2]=='N') forecasts[m].obs=0;
 331                 if(split[m][2]=='R' || split[m][2]=='G') forecasts[m].obs=1;
 332                 if(split[m][2]=='Z') forecasts[m].obs=2;
 333                 if(split[m][2]=='L') forecasts[m].obs=3;
 334             }
 335             continue;
 336         }
 337         if(!strcmp(ID, "POZ")){
 338             ASSIGN2(frz, 999);
 339             continue;
 340         }
 341         if(!strcmp(ID, "POS")){
 342             ASSIGN2(snow, 999);
 343             continue;
 344         }
 345     }
 346     free(s);
 347     fclose(fp);
 348
 349     for(m=0; m<ETA_MSG_MAX; m++){
 350         forecasts[m].rh=rh_F(forecasts[m].temp, forecasts[m].dewpt);
 351         forecasts[m].heatindex=heatindex_F(forecasts[m].temp, forecasts[m].rh);
 352         forecasts[m].windchill=windchill_F(forecasts[m].temp, forecasts[m].windspeed);
 353         forecasts[m].rain=93-forecasts[m].frz-forecasts[m].snow;
 354         forecasts[m].rain=forecasts[m].rain*forecasts[m].pcp_total/93;
 355         forecasts[m].snow=forecasts[m].snow*forecasts[m].pcp_total/93;
 356         forecasts[m].frz=forecasts[m].frz*forecasts[m].pcp_total/93;
 357     }
 358
 359     return 1;
 360 }
 361 #undef NEXT
 362 #undef DIE
 363 #undef SPLIT
 364 #undef ASSIGN
 365 #undef ASSIGN2