From 4b020802c4d5c3610c9fada91bcd6d63ee3cc257 Mon Sep 17 00:00:00 2001
From: Devin Matte <devinmatte@gmail.com>
Date: Mon, 24 Jun 2024 22:59:21 -0400
Subject: [PATCH] Use new categories

---
 common/types/reliability.ts                         |  3 ++-
 modules/reliability/charts/DelayBreakdownGraph.tsx  | 13 ++++++++++++-
 modules/reliability/charts/DelayByCategoryGraph.tsx |  5 ++++-
 3 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/common/types/reliability.ts b/common/types/reliability.ts
index fc4d7512..d17b06cc 100644
--- a/common/types/reliability.ts
+++ b/common/types/reliability.ts
@@ -2,7 +2,7 @@ import type { Line } from './lines';
 
 export interface LineDelays {
   date: string;
-  disabled_train: number;
+  disabled_vehicle: number;
   door_problem: number;
   flooding: number;
   fire: number;
@@ -12,6 +12,7 @@ export interface LineDelays {
   police_activity: number;
   power_problem: number;
   signal_problem: number;
+  mechanical_problem: number;
   brake_problem: number;
   switch_problem: number;
   total_delay_time: number;
diff --git a/modules/reliability/charts/DelayBreakdownGraph.tsx b/modules/reliability/charts/DelayBreakdownGraph.tsx
index aa49f2f6..60d227bf 100644
--- a/modules/reliability/charts/DelayBreakdownGraph.tsx
+++ b/modules/reliability/charts/DelayBreakdownGraph.tsx
@@ -56,7 +56,7 @@ export const DelayBreakdownGraph: React.FC<DelayBreakdownGraphProps> = ({
                 fill: true,
                 pointHoverRadius: 6,
                 pointHoverBackgroundColor: lineColor,
-                data: data.map((datapoint) => datapoint.disabled_train),
+                data: data.map((datapoint) => datapoint.disabled_vehicle),
               },
               {
                 label: `Door Problem`,
@@ -124,6 +124,17 @@ export const DelayBreakdownGraph: React.FC<DelayBreakdownGraphProps> = ({
                 pointHoverBackgroundColor: lineColor,
                 data: data.map((datapoint) => datapoint.track_issue),
               },
+              {
+                label: `Mechanical Problem`,
+                borderColor: '#451a03',
+                backgroundColor: hexWithAlpha('#451a03', 0.8),
+                pointRadius: 0,
+                pointBorderWidth: 0,
+                fill: true,
+                pointHoverRadius: 6,
+                pointHoverBackgroundColor: lineColor,
+                data: data.map((datapoint) => datapoint.mechanical_problem),
+              },
               {
                 label: `Flooding`,
                 borderColor: '#0ea5e9',
diff --git a/modules/reliability/charts/DelayByCategoryGraph.tsx b/modules/reliability/charts/DelayByCategoryGraph.tsx
index 2d9f0c0a..98e99e60 100644
--- a/modules/reliability/charts/DelayByCategoryGraph.tsx
+++ b/modules/reliability/charts/DelayByCategoryGraph.tsx
@@ -48,6 +48,7 @@ export const DelayByCategoryGraph: React.FC<DelayByCategoryGraphProps> = ({
     '🎚️ Switch Problem',
     '🛑 Brake Issue',
     '🛤️ Track Issue',
+    '🔧 Mechanical Problem',
     '🌊 Flooding',
     '🚓 Police Activity',
     '🚑 Medical Emergency',
@@ -62,6 +63,7 @@ export const DelayByCategoryGraph: React.FC<DelayByCategoryGraphProps> = ({
     '#10b981',
     '#4c1d95',
     '#8b5cf6',
+    '#451a03',
     '#0ea5e9',
     '#1d4ed8',
     '#be123c',
@@ -71,13 +73,14 @@ export const DelayByCategoryGraph: React.FC<DelayByCategoryGraphProps> = ({
   const delayTotals: number[] = sumArray(
     data.map((datapoint) => {
       return [
-        datapoint.disabled_train,
+        datapoint.disabled_vehicle,
         datapoint.door_problem,
         datapoint.power_problem,
         datapoint.signal_problem,
         datapoint.switch_problem,
         datapoint.brake_problem,
         datapoint.track_issue,
+        datapoint.mechanical_problem,
         datapoint.flooding,
         datapoint.police_activity,
         datapoint.medical_emergency,