From 06e649671fc4e5ec2f2036176f4c60219453f2ac Mon Sep 17 00:00:00 2001
From: "Engelbart Kai Sven (PEA4-Fe)" <KaiSven.Engelbart@de.bosch.com>
Date: Wed, 24 Feb 2021 13:58:14 +0100
Subject: [PATCH] Allow viscosity number predictions to be saved

---
 src/app/prediction/prediction.component.html | 16 ++++++++++++++-
 src/app/prediction/prediction.component.ts   | 21 +++++++++++++++-----
 2 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/src/app/prediction/prediction.component.html b/src/app/prediction/prediction.component.html
index db1dbdb..1bbe7ba 100644
--- a/src/app/prediction/prediction.component.html
+++ b/src/app/prediction/prediction.component.html
@@ -40,7 +40,7 @@
 </div>
 
 <div class="file-input space-below">
-	<rb-form-file name="spectrum-upload" label="spectrum file" maxSize="10000000" class="space-below" multiple
+	<rb-form-file name="spectrum-upload" label="Spectrum File" maxSize="10000000" class="space-below" multiple
 		(ngModelChange)="fileToArray($event)" placeholder="Select file or drag and drop" dragDrop ngModel>
 	</rb-form-file>
 
@@ -72,6 +72,20 @@
 	Export to PDF
 </rb-icon-button>
 
+<!-- Save Prediction -->
+<!-- Only available for single sample viscosity number predictions -->
+<div *ngIf="spectrumNames.length && !multipleSamples && activeGroup.group === 'Viscosity Number'">
+<rb-form-input name="prediction-name" label="Prediction Name" [(ngModel)]="predictionName"></rb-form-input>
+<rb-icon-button icon="forward-right" mode="secondary" (click)="savePrediction()" [rbModal]="modalSuccess">
+	Save Prediction
+</rb-icon-button>
+<ng-template #modalSuccess>
+	<rb-alert alertTitle="Success" type="success" okBtnLabel="Got it">
+		Prediction saved!
+	</rb-alert>
+</ng-template>
+</div>
+
 <div class="dpt-chart space-below">
 	<canvas baseChart
 		class="dpt-chart"
diff --git a/src/app/prediction/prediction.component.ts b/src/app/prediction/prediction.component.ts
index 42a9602..4106774 100644
--- a/src/app/prediction/prediction.component.ts
+++ b/src/app/prediction/prediction.component.ts
@@ -39,8 +39,11 @@ export class PredictionComponent implements OnInit {
 	loading = false;
 	activeGroup: ModelItemModel = new ModelItemModel();
 	activeModelIndex = 0;
+	predictionName: String;
+
 	// If true, spectra belong to different samples, otherwise multiple spectra from the same sample are given
 	multipleSamples = false;
+
 	spectrumNames: string[] = [];
 	spectrum: string[][] = [[]];
 	flattenedSpectra = [];
@@ -122,20 +125,28 @@ export class PredictionComponent implements OnInit {
 		});
 	}
 
-	groupChange(index) {  // Group was changed
+	// Group was changed
+	groupChange(index) {
 		this.activeGroup = this.d.arr.modelGroups[index];
 		this.activeModelIndex = 0;
 		this.result = undefined;
 	}
 
+	// Save prediction to database
+	// This is not suited for multiple samples
+	savePrediction() {
+		let prediction = { name: this.predictionName, value: this.result.mean[0].value };
+		this.api.post<any>('/prediction/new', prediction);
+	}
+
 	// Aggregates spectrum names and prediction values into an associative array
 	prepareExport() {
 		const zip = (a, b) => a.map((k, i) => [k, b[i]]);
 		const values = this.result.predictions
-		.map(prediction => prediction
-			 .filter(field => field.category === 'Prediction')
-			 .map(field => field.value));
-			 return zip(this.spectrumNames, values);
+			.map(prediction => prediction
+				.filter(field => field.category === 'Prediction')
+				.map(field => field.value));
+		return zip(this.spectrumNames, values);
 	}
 
 	// Generates a timestamp suitable for file naming