From 587b63139fb88395bdaead57b5db703f5c1572a6 Mon Sep 17 00:00:00 2001
From: Alexander Navratil <navratil.alex@gmail.com>
Date: Wed, 1 Aug 2018 07:09:24 +0200
Subject: [PATCH] add Reihenfolge and Richtung to Steig fix listSteige being
 empty

---
 build.gradle                                     |  2 +-
 .../echtzeitdaten/EchtzeitdatenApi.java          | 12 ++++++++----
 .../echtzeitdaten/model/Richtung.java            |  6 ++++++
 .../alexnavratil/echtzeitdaten/model/Steig.java  | 16 +++++++++++++++-
 4 files changed, 30 insertions(+), 6 deletions(-)
 create mode 100644 src/main/java/at/alexnavratil/echtzeitdaten/model/Richtung.java

diff --git a/build.gradle b/build.gradle
index 71579b7..7cd300d 100644
--- a/build.gradle
+++ b/build.gradle
@@ -4,7 +4,7 @@ plugins {
 }
 
 group 'at.alexnavratil'
-version '1.0-SNAPSHOT'
+version '1.1-SNAPSHOT'
 
 sourceCompatibility = 1.8
 
diff --git a/src/main/java/at/alexnavratil/echtzeitdaten/EchtzeitdatenApi.java b/src/main/java/at/alexnavratil/echtzeitdaten/EchtzeitdatenApi.java
index f6fbebf..79ddd7e 100644
--- a/src/main/java/at/alexnavratil/echtzeitdaten/EchtzeitdatenApi.java
+++ b/src/main/java/at/alexnavratil/echtzeitdaten/EchtzeitdatenApi.java
@@ -10,10 +10,8 @@
 import okhttp3.Request;
 import okhttp3.Response;
 
-import java.io.File;
 import java.io.IOException;
 import java.io.InputStreamReader;
-import java.nio.charset.StandardCharsets;
 import java.util.*;
 
 public class EchtzeitdatenApi {
@@ -68,8 +66,10 @@ public void readStaticData() throws IOException {
                         Integer.parseInt(row.getField("STEIG_ID")),
                         Integer.parseInt(row.getField("FK_HALTESTELLEN_ID")),
                         linieIndexMap.get(Integer.parseInt(row.getField("FK_LINIEN_ID"))),
-                        Integer.parseInt(row.getField("RBL_NUMMER"))
-                );
+                        Integer.parseInt(row.getField("RBL_NUMMER")),
+                        Integer.parseInt(row.getField("REIHENFOLGE")),
+                        parseRichtung(row.getField("RICHTUNG")));
+                this.steigList.add(s);
                 if (steigHaltestellenIndexMap.containsKey(s.getHaltestellenId())) {
                     steigHaltestellenIndexMap.get(s.getHaltestellenId()).add(s);
                 } else {
@@ -124,6 +124,10 @@ private TransferType parseType(String type){
         }
     }
 
+    private Richtung parseRichtung(String richtung) {
+        return richtung.equals("H") ? Richtung.H : Richtung.R;
+    }
+
     public List<Haltestelle> listHaltestellen(){
         if(!readStaticData) {
             throw new IllegalStateException("Please call readStaticData before calling any static data method");
diff --git a/src/main/java/at/alexnavratil/echtzeitdaten/model/Richtung.java b/src/main/java/at/alexnavratil/echtzeitdaten/model/Richtung.java
new file mode 100644
index 0000000..b96fd96
--- /dev/null
+++ b/src/main/java/at/alexnavratil/echtzeitdaten/model/Richtung.java
@@ -0,0 +1,6 @@
+package at.alexnavratil.echtzeitdaten.model;
+
+public enum Richtung {
+    H,
+    R
+}
diff --git a/src/main/java/at/alexnavratil/echtzeitdaten/model/Steig.java b/src/main/java/at/alexnavratil/echtzeitdaten/model/Steig.java
index a9f8b12..c2e53ca 100644
--- a/src/main/java/at/alexnavratil/echtzeitdaten/model/Steig.java
+++ b/src/main/java/at/alexnavratil/echtzeitdaten/model/Steig.java
@@ -7,12 +7,16 @@ public class Steig {
     private final int haltestellenId;
     private final Linie linie;
     private final int rbl;
+    private final int reihenfolge;
+    private final Richtung richtung;
 
-    public Steig(int id, int haltestellenId, Linie linie, int rbl) {
+    public Steig(int id, int haltestellenId, Linie linie, int rbl, int reihenfolge, Richtung richtung) {
         this.id = id;
         this.haltestellenId = haltestellenId;
         this.linie = linie;
         this.rbl = rbl;
+        this.reihenfolge = reihenfolge;
+        this.richtung = richtung;
     }
 
     public int getId() {
@@ -31,6 +35,14 @@ public int getRbl() {
         return rbl;
     }
 
+    public int getReihenfolge() {
+        return reihenfolge;
+    }
+
+    public Richtung getRichtung() {
+        return richtung;
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
@@ -51,6 +63,8 @@ public String toString() {
                 ", haltestellenId=" + haltestellenId +
                 ", linie=" + linie +
                 ", rbl=" + rbl +
+                ", reihenfolge=" + reihenfolge +
+                ", richtung=" + richtung +
                 '}';
     }
 }