source: branches/LearningClassifierSystems/HeuristicLab.Encodings.DecisionList/3.3/DecisionList.cs @ 9334

#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.Collections.Generic;
using System.Linq;
using HeuristicLab.Common;
using HeuristicLab.Core;
using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;

namespace HeuristicLab.Encodings.DecisionList {
  [StorableClass]
  [Item("DecisionList", "")]
  public class DecisionList : Item {
    [Storable]
    private IList<Rule> rules;
    public IList<Rule> Rules {
      get { return rules; }
    }

    [Storable]
    private IAction defaultAction;
    public IAction DefaultAction {
      get { return defaultAction; }
    }

    [StorableConstructor]
    protected DecisionList(bool deserializing) : base(deserializing) { }
    protected DecisionList(DecisionList original, Cloner cloner)
      : base(original, cloner) {
    }
    public DecisionList()
      : base() {
      rules = new List<Rule>();
      defaultAction = null;
    }
    public DecisionList(List<Rule> rules)
      : base() {
      this.rules = rules;
      defaultAction = null;
    }
    public DecisionList(List<Rule> rules, IAction defaultAction)
      : base() {
      if (rules.Any(x => x.Action.Match(defaultAction))) { throw new ArgumentException("If a default action is used. The default action is not allowed be used in any rule."); }
      this.rules = rules;
      this.defaultAction = defaultAction;
    }
    public override IDeepCloneable Clone(Cloner cloner) {
      return new DecisionList(this, cloner);
    }

    // for convenience
    public IEnumerable<IAction> Evaluate(IEnumerable<DecisionListInput> input) {
      int numberOfAliveRules;
      return Evaluate(input, out numberOfAliveRules);
    }
    // for convenience
    public IEnumerable<IAction> Evaluate(IEnumerable<DecisionListInput> input, out int numberOfAliveRules) {
      double theoryLengtgh;
      return Evaluate(input, out numberOfAliveRules, out theoryLengtgh);
    }
    public IEnumerable<IAction> Evaluate(IEnumerable<DecisionListInput> input, out int numberOfAliveRules, out double theoryLengtgh) {
      var estimated = new List<IAction>();
      var activatedRules = new HashSet<Rule>();
      foreach (var dli in input) {
        foreach (var rule in rules) {
          if (rule.MatchInput(dli)) {
            estimated.Add(rule.Action);
            activatedRules.Add(rule);
            break;
          }
        }
      }
      numberOfAliveRules = activatedRules.Count;
      theoryLengtgh = activatedRules.Sum(x => x.ComputeTheoryLength());

      return estimated;
    }
  }
}