public static void upload(String imei, File data) {
try {
File pbFile =
new File(
Play.configuration.getProperty("application.uploadDataDirectory"),
imei + "_" + new Date().getTime() + ".pb");
Logger.info(pbFile.toString());
data.renameTo(pbFile);
byte[] dataFrame = new byte[(int) pbFile.length()];
;
DataInputStream dataInputStream =
new DataInputStream(new BufferedInputStream(new FileInputStream(pbFile)));
dataInputStream.read(dataFrame);
Upload upload = Upload.parseFrom(dataFrame);
Phone phone = Phone.find("imei = ?", imei).first();
if (phone == null) badRequest();
for (Upload.Route r : upload.getRouteList()) {
if (r.getPointList().size() <= 1) continue;
Agency a = Agency.find("gtfsAgencyId = ?", "DEFAULT").first();
Route route = new Route("", r.getRouteName(), RouteType.BUS, r.getRouteDescription(), a);
route.phone = phone;
route.routeNotes = r.getRouteNotes();
route.vehicleCapacity = r.getVehicleCapacity();
route.vehicleType = r.getVehicleType();
route.captureTime = new Date(r.getStartTime());
route.save();
List<String> points = new ArrayList<String>();
Integer pointSequence = 1;
for (Upload.Route.Point p : r.getPointList()) {
points.add(new Double(p.getLon()).toString() + " " + new Double(p.getLat()).toString());
RoutePoint.addRoutePoint(p, route.id, pointSequence);
pointSequence++;
}
String linestring = "LINESTRING(" + StringUtils.join(points, ", ") + ")";
BigInteger tripShapeId = TripShape.nativeInsert(TripShape.em(), "", linestring, 0.0);
TripPattern tp = new TripPattern();
tp.route = route;
tp.headsign = r.getRouteName();
tp.shape = TripShape.findById(tripShapeId.longValue());
tp.save();
Integer sequenceId = 0;
for (Upload.Route.Stop s : r.getStopList()) {
BigInteger stopId = Stop.nativeInsert(Stop.em(), s);
TripPatternStop tps = new TripPatternStop();
tps.stop = Stop.findById(stopId.longValue());
tps.stopSequence = sequenceId;
tps.defaultTravelTime = s.getArrivalTimeoffset();
tps.defaultDwellTime = s.getDepartureTimeoffset() - s.getArrivalTimeoffset();
tps.pattern = tp;
tps.board = s.getBoard();
tps.alight = s.getAlight();
tps.save();
sequenceId++;
}
// ProcessGisExport gisExport = new ProcessGisExport(tp.id);
// gisExport.doJob();
}
Logger.info("Routes uploaded: " + upload.getRouteList().size());
dataInputStream.close();
ok();
} catch (Exception e) {
e.printStackTrace();
badRequest();
}
}
public static void exportCsv(String unitId) throws IOException {
Phone p = Phone.find("unitId = ?", unitId).first();
if (p == null) index(true);
List<Route> routes = Route.find("phone = ?", p).fetch();
File outputDirectory =
new File(Play.configuration.getProperty("application.exportDataDirectory"), unitId);
File outputZipFile =
new File(
Play.configuration.getProperty("application.exportDataDirectory"), unitId + ".zip");
// write routes
File routesFile = new File(outputDirectory, unitId + "_routes.csv");
File stopsFile = new File(outputDirectory, unitId + "_stops.csv");
if (!outputDirectory.exists()) {
outputDirectory.mkdir();
}
if (outputZipFile.exists()) {
outputZipFile.delete();
}
FileWriter routesCsv = new FileWriter(routesFile);
CSVWriter rotuesCsvWriter = new CSVWriter(routesCsv);
FileWriter stopsCsv = new FileWriter(stopsFile);
CSVWriter stopsCsvWriter = new CSVWriter(stopsCsv);
String[] routesHeader =
"unit_id, route_id, route_name, route_description, field_notes, vehicle_type, vehicle_capacity, start_capture"
.split(",");
String[] stopsHeader =
"route_id, stop_sequence, lat, lon, travel_time, dwell_time, board, alight".split(",");
rotuesCsvWriter.writeNext(routesHeader);
stopsCsvWriter.writeNext(stopsHeader);
for (Route r : routes) {
String[] routeData = new String[routesHeader.length];
routeData[0] = unitId;
routeData[1] = r.id.toString();
routeData[2] = r.routeLongName;
routeData[3] = r.routeDesc;
routeData[4] = r.routeNotes;
routeData[5] = r.vehicleType;
routeData[6] = r.vehicleCapacity;
routeData[7] = (r.captureTime != null) ? r.captureTime.toGMTString() : "";
rotuesCsvWriter.writeNext(routeData);
List<TripPatternStop> stops = TripPatternStop.find("pattern.route = ?", r).fetch();
for (TripPatternStop s : stops) {
String[] stopData = new String[stopsHeader.length];
stopData[0] = r.id.toString();
stopData[1] = s.stopSequence.toString();
stopData[2] = "" + s.stop.location.getCoordinate().y;
stopData[3] = "" + s.stop.location.getCoordinate().x;
stopData[4] = "" + s.defaultTravelTime;
stopData[5] = "" + s.defaultDwellTime;
stopData[6] = "" + s.board;
stopData[7] = "" + s.alight;
stopsCsvWriter.writeNext(stopData);
}
}
rotuesCsvWriter.flush();
rotuesCsvWriter.close();
stopsCsvWriter.flush();
stopsCsvWriter.close();
DirectoryZip.zip(outputDirectory, outputZipFile);
FileUtils.deleteDirectory(outputDirectory);
redirect("/public/data/exports/" + unitId + ".zip");
}