Added algorithm to work out if cattle is infected

Author: benn-mt

src/models/Cattle.js

@@ -6,15 +6,19 @@ export class Cattle {
   #location;
   #status;
   #imageUrl;
+  fieldX;
+  fieldY;
 
-  constructor(id, sex, tag, breed, location, status = 'Healthy', imageUrl = null) {
+  constructor(id, sex, tag, breed, location, status = 'Healthy', imageUrl = null, fieldX = 0.0, fieldY = 0.0) {
     this.#id = id;
     this.#sex = sex;
     this.#tag = tag;
     this.#breed = breed;
     this.#location = location;
     this.#status = status;
     this.#imageUrl = imageUrl;
+    this.fieldX = fieldX;
+    this.fieldY = fieldY;
   }
 
   // Getters
@@ -53,6 +57,23 @@ export class Cattle {
     return this.#status.toLowerCase().replace(/\s+/g, '-');
   }
 
+  getPossiblyInfected(cattle){
+    for (let i = 0; i < cattle.length; i++){
+      let cattleDistance = Math.sqrt(((cattle[i].fieldX - this.fieldX) * (cattle[i].fieldX - this.fieldX)) + ((cattle[i].fieldY - this.fieldY) * (cattle[i].fieldY - this.fieldY)));
+      if (cattleDistance < 10){
+        if (cattle[i].#status === 'Infected'){
+          return true;
+        }
+      }
+    }
+
+    if (this.#status === 'Infected'){
+      return true;
+    } else {
+      return false;
+    }
+  }
+
   toJSON() {
     return {
       id: this.#id,