source: branches/HiveHiveEngine/HeuristicLab.Operators/3.3/IntDataReducer.cs @ 7325

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2012 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System;
using System.Linq;
using HeuristicLab.Common;
using HeuristicLab.Core;
using HeuristicLab.Data;
using HeuristicLab.Parameters;
using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;

namespace HeuristicLab.Operators {
  [Item("IntDataReducer", "An operator to reduce int values of sub scopes")]
  [StorableClass]
  public sealed class IntDataReducer : SingleSuccessorOperator {
    #region Parameter Properties
    public ScopeTreeLookupParameter<IntValue> ParameterToReduce {
      get { return (ScopeTreeLookupParameter<IntValue>)Parameters["ParameterToReduce"]; }
    }
    public ValueLookupParameter<StringValue> TargetParameterName {
      get { return (ValueLookupParameter<StringValue>)Parameters["TargetParameterName"]; }
    }
    private LookupParameter<IntValue> TargetParameter {
      get { return (LookupParameter<IntValue>)Parameters["TargetParameter"]; }
    }
    public ValueParameter<ReductionType> ReductionOperation {
      get { return (ValueParameter<ReductionType>)Parameters["ReductionOperation"]; }
    }
    public ValueParameter<TargetOperationType> TargetOperation {
      get { return (ValueParameter<TargetOperationType>)Parameters["TargetOperation"]; }
    }
    #endregion

    [StorableConstructor]
    private IntDataReducer(bool deserializing) : base(deserializing) { }
    private IntDataReducer(IntDataReducer original, Cloner cloner)
      : base(original, cloner) {
    }
    public IntDataReducer()
      : base() {
      #region Create parameters
      Parameters.Add(new ScopeTreeLookupParameter<IntValue>("ParameterToReduce", "The parameter on which the reduction operation should be applied."));
      Parameters.Add(new ValueLookupParameter<StringValue>("TargetParameterName", "Parameter to store the result in.", new StringValue(string.Empty)));
      Parameters.Add(new LookupParameter<IntValue>("TargetParameter", "The actual parameter defined by TargetParameterName."));
      Parameters.Add(new ValueParameter<ReductionType>("ReductionOperation", "The reduction operation to perform."));
      Parameters.Add(new ValueParameter<TargetOperationType>("TargetOperation", "The target operation to perform."));
      #endregion
    }

    public override IDeepCloneable Clone(Cloner cloner) {
      return new IntDataReducer(this, cloner);
    }

    public override IOperation Apply() {
      var intValues = ParameterToReduce.ActualValue;

      if (intValues.Count() > 0) {
        TargetParameter.ActualName = TargetParameterName.ActualValue.Value;
        if (TargetParameter.ActualValue == null) TargetParameter.ActualValue = new IntValue();

        int result = 1;
        switch (ReductionOperation.Value.Value) {
          case ReductionTypes.Sum:
            result = intValues.Sum(x => x.Value);
            break;
          case ReductionTypes.Prod:
            intValues.ForEach(x => result *= x.Value);
            break;
          case ReductionTypes.Avg:
            result = (int)Math.Round(intValues.Average(x => x.Value));
            break;
          case ReductionTypes.Min:
            result = intValues.Min(x => x.Value);
            break;
          case ReductionTypes.Max:
            result = intValues.Max(x => x.Value);
            break;
        }

        switch (TargetOperation.Value.Value) {
          case TargetOperationTypes.Set:
            TargetParameter.ActualValue.Value = result;
            break;
          case TargetOperationTypes.Sum:
            TargetParameter.ActualValue.Value += result;
            break;
          case TargetOperationTypes.Prod:
            if (TargetParameter.ActualValue.Value == 0) TargetParameter.ActualValue.Value = 1;
            TargetParameter.ActualValue.Value *= result;
            break;
        }
      }
      return base.Apply();
    }
  }
}