From 1b3f3cf3271197697557092e23f434dcccf668a1 Mon Sep 17 00:00:00 2001
From: stevereis <stevereis93@gmail.com>
Date: Tue, 26 Oct 2021 08:49:48 +0200
Subject: [PATCH] Add axis label and categories

---
 .../engine/models/result/common/chart-axis.model.ts    | 10 ++++++++++
 api/src/engine/models/result/heat-map-result.model.ts  |  9 +++++----
 2 files changed, 15 insertions(+), 4 deletions(-)
 create mode 100644 api/src/engine/models/result/common/chart-axis.model.ts

diff --git a/api/src/engine/models/result/common/chart-axis.model.ts b/api/src/engine/models/result/common/chart-axis.model.ts
new file mode 100644
index 0000000..f504588
--- /dev/null
+++ b/api/src/engine/models/result/common/chart-axis.model.ts
@@ -0,0 +1,10 @@
+import { ObjectType, Field } from '@nestjs/graphql';
+
+@ObjectType()
+export class ChartAxis {
+  @Field({ nullable: true, defaultValue: '' })
+  label?: string;
+
+  @Field(() => [String], { nullable: true, defaultValue: [] })
+  categories: string[];
+}
diff --git a/api/src/engine/models/result/heat-map-result.model.ts b/api/src/engine/models/result/heat-map-result.model.ts
index abfd2f4..ab4d2d8 100644
--- a/api/src/engine/models/result/heat-map-result.model.ts
+++ b/api/src/engine/models/result/heat-map-result.model.ts
@@ -1,4 +1,5 @@
 import { Field, ObjectType } from '@nestjs/graphql';
+import { ChartAxis } from './common/chart-axis.model';
 import { Result } from './common/result.model';
 
 @ObjectType()
@@ -9,9 +10,9 @@ export class HeatMapResult extends Result {
   @Field(() => [[Number]])
   matrix: number[][];
 
-  @Field(() => [String])
-  xLabels: string[];
+  @Field(() => [ChartAxis])
+  xAxis: ChartAxis;
 
-  @Field(() => [String])
-  yLabels: string[];
+  @Field(() => [ChartAxis])
+  yAxis: ChartAxis;
 }
-- 
GitLab