1 | using System;
|
---|
2 | using System.Collections.Generic;
|
---|
3 | using System.Linq;
|
---|
4 | using System.Text;
|
---|
5 | using System.Xml;
|
---|
6 | using HeuristicLab.DataImporter.Data;
|
---|
7 | using HeuristicLab.DataImporter.Data.CommandBase;
|
---|
8 | using HeuristicLab.DataImporter.Data.Model;
|
---|
9 | using HeuristicLab.DataImporter.Command.View;
|
---|
10 | using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
|
---|
11 |
|
---|
12 | namespace HeuristicLab.DataImporter.Command {
|
---|
13 | [StorableClass]
|
---|
14 | [ViewableCommandInfoAttribute("Scale to normal distribution", 1, ColumnGroupState.DoubleColumnSelected, "Change Values",
|
---|
15 | Position = 5, OptionsView = typeof(NormalDistributionCommandView))]
|
---|
16 | public class NormalDistributionScalingCommand : ColumnGroupCommandWithAffectedColumnsBase {
|
---|
17 | private List<KeyValuePair<double, double>> oldNormalDistribution;
|
---|
18 |
|
---|
19 | private NormalDistributionScalingCommand()
|
---|
20 | : base(null, string.Empty, null) {
|
---|
21 | oldNormalDistribution = new List<KeyValuePair<double, double>>();
|
---|
22 | }
|
---|
23 |
|
---|
24 | public NormalDistributionScalingCommand(DataSet dataSet, string columnGroupName, int[] affectedColumns)
|
---|
25 | : base(dataSet, columnGroupName, affectedColumns) {
|
---|
26 | oldNormalDistribution = new List<KeyValuePair<double, double>>();
|
---|
27 | }
|
---|
28 |
|
---|
29 | public override string Description {
|
---|
30 | get { return "Scale to normal distribution"; }
|
---|
31 | }
|
---|
32 |
|
---|
33 | [Storable]
|
---|
34 | private double newMean;
|
---|
35 | public double Mean {
|
---|
36 | get { return this.newMean; }
|
---|
37 | set { this.newMean = value; }
|
---|
38 | }
|
---|
39 |
|
---|
40 | [Storable]
|
---|
41 | private double newStddev;
|
---|
42 | public double StandardDeviation {
|
---|
43 | get { return this.newStddev; }
|
---|
44 | set { this.newStddev = value; }
|
---|
45 | }
|
---|
46 |
|
---|
47 | public override void Execute() {
|
---|
48 | base.Execute();
|
---|
49 | DoubleColumn column;
|
---|
50 | double oldMean;
|
---|
51 | double oldStddev;
|
---|
52 | for (int i = 0; i < AffectedColumns.Length; i++) {
|
---|
53 | if (ColumnGroup.GetColumn(AffectedColumns[i]) is DoubleColumn) {
|
---|
54 | column = (DoubleColumn)ColumnGroup.GetColumn(AffectedColumns[i]);
|
---|
55 | if (column.NonNullValuesCount == 0)
|
---|
56 | continue;
|
---|
57 | oldMean = (double)column.Mean;
|
---|
58 | oldStddev = (double)column.StandardDeviation;
|
---|
59 | CalculateNewColumn(column, newMean, newStddev, oldMean, oldStddev);
|
---|
60 | oldNormalDistribution.Add(new KeyValuePair<double, double>(oldMean, oldStddev));
|
---|
61 | }
|
---|
62 | }
|
---|
63 | ColumnGroup.FireChanged();
|
---|
64 | ColumnGroup = null;
|
---|
65 | }
|
---|
66 |
|
---|
67 | public override void UndoExecute() {
|
---|
68 | base.UndoExecute();
|
---|
69 | DoubleColumn column;
|
---|
70 | double oldMean;
|
---|
71 | double oldStddev;
|
---|
72 | int j = 0;
|
---|
73 | for (int i = 0; i < AffectedColumns.Length; i++) {
|
---|
74 | if (ColumnGroup.GetColumn(AffectedColumns[i]) is DoubleColumn) {
|
---|
75 | column = (DoubleColumn)ColumnGroup.GetColumn(AffectedColumns[i]);
|
---|
76 | if (column.NonNullValuesCount == 0)
|
---|
77 | continue;
|
---|
78 | oldMean = oldNormalDistribution[j].Key;
|
---|
79 | oldStddev = oldNormalDistribution[j].Value;
|
---|
80 | CalculateNewColumn(column, oldMean, oldStddev, newMean, newStddev);
|
---|
81 | j++;
|
---|
82 | }
|
---|
83 | }
|
---|
84 | oldNormalDistribution.Clear();
|
---|
85 | ColumnGroup.FireChanged();
|
---|
86 | ColumnGroup = null;
|
---|
87 |
|
---|
88 | }
|
---|
89 |
|
---|
90 | private void CalculateNewColumn(DoubleColumn column, double newMean, double newStddev, double oldMean, double oldStddev) {
|
---|
91 | double newValue;
|
---|
92 | double oldValue;
|
---|
93 | for (int val = 0; val < column.TotalValuesCount; val++) {
|
---|
94 | if (column.GetValue(val) == null)
|
---|
95 | continue;
|
---|
96 | oldValue = (double)column.GetValue(val);
|
---|
97 | if (oldStddev == 0)
|
---|
98 | newValue = oldValue - oldMean + newMean;
|
---|
99 | else
|
---|
100 | newValue = ((oldValue - oldMean) / oldStddev) * newStddev + newMean;
|
---|
101 | column.ChangeValue(val, newValue);
|
---|
102 | }
|
---|
103 | }
|
---|
104 | }
|
---|
105 | }
|
---|