From 9453307fd4547031e207ff05c533b5905e2749d6 Mon Sep 17 00:00:00 2001
From: Michael Rausch <mra106@uclive.ac.nz>
Date: Mon, 6 Mar 2017 17:56:07 +1300
Subject: [PATCH] Implemented leg class and added legs to the race

- #implement
---
 src/main/java/seng302/Leg.java  | 47 +++++++++++++++++++++++++++++++++
 src/main/java/seng302/Race.java | 13 ++++++++-
 2 files changed, 59 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/seng302/Leg.java

diff --git a/src/main/java/seng302/Leg.java b/src/main/java/seng302/Leg.java
new file mode 100644
index 00000000..9db54109
--- /dev/null
+++ b/src/main/java/seng302/Leg.java
@@ -0,0 +1,47 @@
+package seng302;
+
+public class Leg {
+	private int heading;
+	private int distance;
+	private String startLabel;
+	private boolean isFinishingLeg;
+
+	public Leg(int heading, int distance, String label){
+		this.heading = heading;
+		this.distance = distance;
+		this.startLabel = label;
+		this.isFinishingLeg = false;
+	}
+
+	public void setHeading(int heading){
+		this.heading = heading;
+	}
+
+	public int getHeading(){
+		return this.heading;
+	}
+
+	public void setDistance(int distance){
+		this.distance = distance;
+	}
+
+	public int getDistance(){
+		return this.distance;
+	}
+
+	public void setLabel(String label){
+		this.startLabel = label;
+	}
+
+	public String getLabel(){
+		return this.startLabel;
+	}
+
+	public void setFinishingLeg(boolean isFinishingLeg){
+		this.isFinishingLeg = isFinishingLeg;
+	}
+
+	public boolean getIsFinishingLeg(){
+		return this.isFinishingLeg;
+	}
+}
\ No newline at end of file
diff --git a/src/main/java/seng302/Race.java b/src/main/java/seng302/Race.java
index 9edc840d..8616dd93 100644
--- a/src/main/java/seng302/Race.java
+++ b/src/main/java/seng302/Race.java
@@ -8,10 +8,12 @@ import java.util.List;
 
 public class Race {
 	private ArrayList<Boat> boats;
+	private ArrayList<Leg> legs;
 	private int numberOfBoats = 0;
 
 	public Race(){
 		boats = new ArrayList<Boat>();
+		legs = new ArrayList<Leg>();
 	}
 
 	/*
@@ -71,7 +73,6 @@ public class Race {
 
 	/*
 		Prints the list of boats competing in the race
-
 	*/ 
 	public void displayStartingBoats(){
 		int numberOfBoats = this.getNumberOfBoats();
@@ -83,4 +84,14 @@ public class Race {
 			System.out.println(boats[i].getTeamName());
 		}
 	}
+
+	public void addLeg(Leg leg){
+		this.legs.add(leg);
+	}
+
+	public void printLegs(){
+		for (Leg leg : this.legs.toArray(new Leg[legs.size()])){
+			System.out.println(leg.getLabel());
+		}
+	}
 }
\ No newline at end of file