From 1875a2d8ae387280bb5aeb83fa859df71798db5d Mon Sep 17 00:00:00 2001
From: David Peake <dave.peake@gmail.com>
Date: Sat, 29 Dec 2012 08:59:15 +0000
Subject: [PATCH] Added command line options

Conflicts:
	wmr100.c
---
 wmr100.c | 52 ++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 46 insertions(+), 6 deletions(-)

diff --git a/wmr100.c b/wmr100.c
index a7d129e..ce8933e 100644
--- a/wmr100.c
+++ b/wmr100.c
@@ -24,10 +24,18 @@
 #include <signal.h>
 #include <time.h>
 #include <unistd.h>
+#include <ctype.h>
 
 #define WMR100_VENDOR_ID  0x0fde
 #define WMR100_PRODUCT_ID 0xca01
 
+/* globals */
+#define OUTPUT_BOTH 3
+#define OUTPUT_FILE 2
+#define OUTPUT_STDOUT 1
+
+int gOutput = OUTPUT_BOTH;
+
 /* constants */
 int const RECV_PACKET_LEN   = 8;
 int const BUF_SIZE = 255;
@@ -240,9 +248,13 @@ void wmr_log_data(WMR *wmr, char *msg) {
                 }
         }
 
-    fprintf(out, "DATA[%s]:%s\n", outstr, msg);
-    fflush(out);
-    printf("DATA[%s]:%s\n", outstr, msg);
+    if (gOutput & OUTPUT_FILE)
+    {
+        fprintf(out, "DATA[%s]:%s\n", outstr, msg);
+        fflush(out);
+    }
+    if (gOutput & OUTPUT_STDOUT)
+        printf("DATA[%s]:%s\n", outstr, msg);
 }
 
 void wmr_handle_rain(WMR *wmr, unsigned char *data, int len)
@@ -389,7 +401,8 @@ void wmr_handle_clock(WMR *wmr, unsigned char *data, int len)
 
 void wmr_handle_packet(WMR *wmr, unsigned char *data, int len)
 {
-    dump_packet(data, len);
+    if (gOutput & OUTPUT_STDOUT)
+        dump_packet(data, len);
     
     switch(data[1]) {
     case 0x41:
@@ -504,13 +517,40 @@ void cleanup(int sig_num)
     exit(0);
 }
 
-int main(void)
+int main(int argc, char* argv[])
 {
     int ret;
-
+    int c;
+    
     signal(SIGINT, cleanup);
     signal(SIGTERM, cleanup);
 
+    /* Parse the command line parameters */
+    while ((c = getopt(argc, argv, "hsfb")) != -1)
+    {
+        switch (c)
+        {
+        case 'h':
+            fprintf(stderr, "Options:\n\t-s: output to sdtout only\n\t-f: output to file only\n\t-b: output to both [default]\n");
+            return 1;
+        case 's': 
+            gOutput = OUTPUT_STDOUT;
+            break;
+        case 'f':
+            gOutput = OUTPUT_FILE;
+            break;
+        case 'b':
+            gOutput = OUTPUT_BOTH;
+            break;
+        case '?':
+            if (isprint(optopt))
+                fprintf(stderr, "Unknown option `-%c'.\n", optopt);
+            return 1;
+        default:
+            abort();
+        }
+    }
+
     wmr = wmr_new();
     if (wmr == NULL) {
       fprintf(stderr, "wmr_new failed\n");