1 files changed, 108 insertions, 2 deletions
diff --git a/frontend/src/app/_elements/column-table/column-table.component.ts b/frontend/src/app/_elements/column-table/column-table.component.ts
index f8c5f5f3..18e38203 100644
--- a/frontend/src/app/_elements/column-table/column-table.component.ts
+++ b/frontend/src/app/_elements/column-table/column-table.component.ts
@@ -1,4 +1,10 @@
-import { Component, OnInit } from '@angular/core';
+import { Component, Input, OnInit } from '@angular/core';
+import Dataset from 'src/app/_data/Dataset';
+import Experiment, { ColumnEncoding, Encoding, NullValReplacer, NullValueOptions } from 'src/app/_data/Experiment';
+import { DatasetsService } from 'src/app/_services/datasets.service';
+import { EncodingDialogComponent } from 'src/app/_modals/encoding-dialog/encoding-dialog.component';
+import { MatDialog } from '@angular/material/dialog';
+import { MissingvaluesDialogComponent } from 'src/app/_modals/missingvalues-dialog/missingvalues-dialog.component';
 
 @Component({
   selector: 'app-column-table',
@@ -7,9 +13,109 @@ import { Component, OnInit } from '@angular/core';
 })
 export class ColumnTableComponent implements OnInit {
 
-  constructor() { }
+  @Input() dataset?: Dataset;
+  @Input() experiment?: Experiment;
+
+  Object = Object;
+  Encoding = Encoding;
+  NullValueOptions = NullValueOptions;
+
+  constructor(private datasetService: DatasetsService, public dialog: MatDialog) { 
+    //ovo mi nece trebati jer primam dataset iz druge komponente
+    this.datasetService.getMyDatasets().subscribe((datasets) => {
+      this.dataset = datasets[0];
+      //console.log(this.dataset);
+      this.experiment = new Experiment();
+      for (let i = 0; i < this.dataset?.columnInfo.length; i++) {
+        this.experiment?.inputColumns.push(this.dataset.columnInfo[i].columnName);
+      }
+      this.resetColumnEncodings(Encoding.Label);
+    });
+  }
 
   ngOnInit(): void {
   }
 
+  changeInputColumns(target: any, columnName: string) {
+    if (this.experiment != undefined) {
+      if (target.currentTarget.checked) {
+        if (this.experiment.inputColumns.filter(x => x == columnName)[0] == undefined) {
+          this.experiment.inputColumns.push(columnName);
+        }
+      }
+      else {
+        this.experiment.inputColumns = this.experiment.inputColumns.filter(x => x != columnName);
+        //console.log("Input columns: ", this.experiment.inputColumns);
+        //TODO: da se zatamni kolona koja je unchecked
+      }
+    }
+  }
+
+  changeColumnType(target: any, indexOfCol: number) {
+    if (this.dataset != undefined) {
+      if (target.currentTarget.value == "Numerički") {
+        this.dataset.columnInfo[indexOfCol].isNumber = true;
+      }
+      else {
+        this.dataset.columnInfo[indexOfCol].isNumber = false;
+      }
+    }
+  }
+
+  resetColumnEncodings(encodingType: Encoding) {
+    if (this.experiment != undefined && this.dataset != undefined) {
+      this.experiment.encodings = [];
+      for (let i = 0; i < this.dataset?.columnInfo.length; i++) {
+        this.experiment.encodings.push(new ColumnEncoding(this.dataset?.columnInfo[i].columnName, encodingType));
+        //console.log(this.experiment.encodings);
+      }
+    }
+  }
+  openEncodingDialog() {
+      const dialogRef = this.dialog.open(EncodingDialogComponent, {
+        width: '300px'
+      });
+      dialogRef.afterClosed().subscribe(selectedEncoding => {
+        if (selectedEncoding != undefined)
+          this.resetColumnEncodings(selectedEncoding);
+      });
+  }
+
+  resetMissingValuesTreatment(selectedMissingValuesOption: NullValueOptions) {
+    if (this.experiment != undefined && this.dataset != undefined) {
+
+      if (selectedMissingValuesOption == NullValueOptions.DeleteColumns) {
+        this.experiment.nullValues = NullValueOptions.DeleteColumns;
+        this.experiment.nullValuesReplacers = [];
+        for (let i = 0; i < this.experiment.inputColumns.length; i++) {
+          this.experiment.nullValuesReplacers.push({
+            column: this.experiment.inputColumns[i],
+            option: NullValueOptions.DeleteColumns,
+            value: ""
+          });
+        }
+      } 
+      else if (selectedMissingValuesOption == NullValueOptions.DeleteRows) {
+        this.experiment.nullValues = NullValueOptions.DeleteRows;
+        this.experiment.nullValuesReplacers = [];
+        for (let i = 0; i < this.experiment.inputColumns.length; i++) {
+          this.experiment.nullValuesReplacers.push({
+            column: this.experiment.inputColumns[i],
+            option: NullValueOptions.DeleteRows,
+            value: ""
+          });
+        }
+      }
+
+    }
+  }
+  openMissingValuesDialog() {
+    const dialogRef = this.dialog.open(MissingvaluesDialogComponent, {
+      width: '400px'
+    });
+    dialogRef.afterClosed().subscribe(selectedMissingValuesOption => {
+      if (selectedMissingValuesOption != undefined) 
+        this.resetMissingValuesTreatment(selectedMissingValuesOption);
+    });
+  }
 }