1 files changed, 194 insertions, 0 deletions
diff --git a/bigtop-data-generators/bigtop-weatherman/src/main/java/org/apache/bigtop/datagenerators/weatherman/internal/Driver.java b/bigtop-data-generators/bigtop-weatherman/src/main/java/org/apache/bigtop/datagenerators/weatherman/internal/Driver.java
new file mode 100644
index 00000000..0bd8c41c
--- /dev/null
+++ b/bigtop-data-generators/bigtop-weatherman/src/main/java/org/apache/bigtop/datagenerators/weatherman/internal/Driver.java
@@ -0,0 +1,194 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.bigtop.datagenerators.weatherman.internal;
+
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.FileWriter;
+import java.io.Writer;
+import java.util.List;
+import java.util.Random;
+
+import org.apache.bigtop.datagenerators.locations.Location;
+import org.apache.bigtop.datagenerators.locations.LocationReader;
+import org.apache.bigtop.datagenerators.samplers.SeedFactory;
+import org.apache.bigtop.datagenerators.weatherman.WeatherGenerator;
+import org.apache.bigtop.datagenerators.weatherman.WeatherRecord;
+import org.joda.time.DateTime;
+import org.joda.time.LocalDate;
+import org.joda.time.format.DateTimeFormat;
+import org.joda.time.format.DateTimeFormatter;
+
+import com.google.common.collect.Lists;
+
+public class Driver {
+  List<Location> locations;
+  Location location;
+  int simulationLength;
+  LocalDate startDate;
+  SeedFactory seedFactory;
+  File outputDir;
+
+  static final int NPARAMS = 5;
+
+  private void printUsage() {
+    String usage = "BigPetStore Data Generator\n" + "\n"
+            + "Usage: java -jar bigpetstore-weather-generator.java outputDir zipcode simulationLength startDate [seed]\n"
+            + "\n" + "outputDir - (string) directory to write files\n"
+            + "zipcode - (string) location zipcode\n"
+            + "simulationLength - (int) number of days to simulate\n"
+            + "startDate - (string) simulation start date in YYYY-MM-DD format\n"
+            + "seed - (long) seed for RNG. If not given, one is randomly generated.\n";
+
+    System.out.println(usage);
+  }
+
+  private Location findLocation(String zipcode) {
+    for (Location location : locations) {
+      if (location.getZipcode().equals(zipcode))
+        return location;
+    }
+
+    return null;
+  }
+
+  /**
+   * Returns local date from string in YYYY-MM-DD format
+   *
+   * @param text
+   * @return
+   */
+  private LocalDate parseDate(String text) {
+    DateTimeFormatter dtFormatter = DateTimeFormat.forPattern("yyyy-MM-dd");
+    DateTime dt = dtFormatter.parseDateTime(text);
+
+    return dt.toLocalDate();
+  }
+
+  public void parseArgs(String[] args) {
+    if (args.length != NPARAMS && args.length != (NPARAMS - 1)) {
+      printUsage();
+      System.exit(1);
+    }
+
+    int i = -1;
+
+    outputDir = new File(args[++i]);
+    if (!outputDir.exists()) {
+      System.err.println("Given path (" + args[i] + ") does not exist.\n");
+      printUsage();
+      System.exit(1);
+    }
+
+    if (!outputDir.isDirectory()) {
+      System.err.println("Given path (" + args[i] + ") is not a directory.\n");
+      printUsage();
+      System.exit(1);
+    }
+
+    location = findLocation(args[++i]);
+    if (location == null) {
+      System.err.println(
+              "No location found for given zipcode \"" + args[i] + "\"");
+      printUsage();
+      System.exit(1);
+    }
+
+    try {
+      simulationLength = Integer.parseInt(args[++i]);
+    } catch (Exception e) {
+      System.err.println("Unable to parse '" + args[i]
+              + "' as an int for simulationLength.\n");
+      printUsage();
+      System.exit(1);
+    }
+
+    try {
+      startDate = parseDate(args[++i]);
+    } catch (Exception e) {
+      System.err.println("Unable to parse '" + args[i]
+              + "'. Expected string in 'YYYY-MM-DD' format.\n");
+      printUsage();
+      System.exit(1);
+    }
+
+    long seed = (new Random()).nextLong();
+    if (args.length == NPARAMS) {
+      try {
+        seed = Long.parseLong(args[++i]);
+      } catch (Exception e) {
+        System.err.println(
+                "Unable to parse '" + args[i] + "' as a long for the seed.\n");
+        printUsage();
+        System.exit(1);
+      }
+    }
+
+    seedFactory = new SeedFactory(seed);
+  }
+
+  private void writeWeather(Location location, List<WeatherRecord> trajectory)
+          throws Exception {
+    File outputFile = new File(outputDir.toString() + File.separator
+            + location.getZipcode() + ".txt");
+    Writer output = new BufferedWriter(new FileWriter(outputFile));
+
+    output.write(
+            "date,city,state,zipcode,temperature,windchill,windspeed,precipitation,rainfall,snowfall\n");
+    for (WeatherRecord weather : trajectory) {
+      String record = weather.getDate() + ",";
+      record += location.getCity() + ",";
+      record += location.getState() + ",";
+      record += location.getZipcode() + ",";
+      record += weather.getTemperature() + ",";
+      record += weather.getWindChill() + ",";
+      record += weather.getWindSpeed() + ",";
+      record += weather.getPrecipitation() + ",";
+      record += weather.getRainFall() + ",";
+      record += weather.getSnowFall() + "\n";
+
+      output.write(record);
+    }
+
+    output.close();
+  }
+
+  public void run(String[] args) throws Exception {
+    locations = new LocationReader().readData();
+
+    parseArgs(args);
+
+    WeatherGenerator generator = new WeatherGenerator(startDate, location,
+            seedFactory);
+
+    LocalDate date = startDate;
+    LocalDate endDate = startDate.plusDays(simulationLength);
+    List<WeatherRecord> trajectory = Lists.newArrayList();
+
+    while (date.isBefore(endDate)) {
+      WeatherRecord weather = generator.sample();
+      trajectory.add(weather);
+      date = weather.getDate();
+    }
+
+    writeWeather(location, trajectory);
+  }
+
+  public static void main(String[] args) throws Exception {
+    Driver driver = new Driver();
+    driver.run(args);
+  }
+}