source: branches/2886_SymRegGrammarEnumeration/HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration/GrammarEnumeration/SearchDataStructure.cs @ 15800

using System;
using System.Collections;
using System.Collections.Generic;

namespace HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration.GrammarEnumeration {

  public struct StoredSymbolString {
    public readonly int Hash;
    public readonly SymbolString SymbolString;

    public StoredSymbolString(int hash, SymbolString symbolString) {
      Hash = hash;
      SymbolString = symbolString;
    }
  }

  public enum StorageType {
    Stack, Queue, RandomList
  }

  class SearchDataStore : IEnumerable<SymbolString> {

    private Dictionary<int, SymbolString> storedValues; // Store hash-references and associated, actual values 
    private Action<int> storeInternal; // Store hash-references 
    private Func<int> fetchInternal; // Fetch hash-reference

    public SearchDataStore(StorageType storageType) {
      storedValues = new Dictionary<int, SymbolString>();

      switch (storageType) {
        case StorageType.Stack:
          InitStack();
          break;
        case StorageType.Queue:
          InitQueue();
          break;
        case StorageType.RandomList:
          InitRandomList();
          break;
      }

    }

    #region SearchStrategies

    private void InitStack() {
      Stack<int> stack = new Stack<int>();

      storeInternal = hash => stack.Push(hash);
      fetchInternal = () => stack.Pop();
    }

    private void InitQueue() {
      Queue<int> queue = new Queue<int>();

      storeInternal = hash => queue.Enqueue(hash);
      fetchInternal = () => queue.Dequeue();
    }

    private void InitRandomList() {
      List<int> list = new List<int>();
      System.Random rand = new System.Random(999);

      storeInternal = hash => list.Add(hash);
      fetchInternal = () => {
        int indexOfHash = rand.Next(list.Count);
        int result = list[indexOfHash];
        list.RemoveAt(indexOfHash);
        return result;
      };
    }

    #endregion

    #region Interface

    public StoredSymbolString GetNext() {
      int hash = fetchInternal.Invoke();
      SymbolString result = storedValues[hash];
      storedValues.Remove(hash);
      return new StoredSymbolString(hash, result);
    }

    public void Store(int hash, SymbolString s) {
      storeInternal.Invoke(hash);
      storedValues[hash] = s;
    }

    public bool Contains(int hash) {
      return storedValues.ContainsKey(hash);
    }

    #endregion

    #region Collection-Interface

    public int Count {
      get { return storedValues.Count; }
    }

    public IEnumerator<SymbolString> GetEnumerator() {
      return storedValues.Values.GetEnumerator();
    }

    IEnumerator IEnumerable.GetEnumerator() {
      return GetEnumerator();
    }

    #endregion
  }
}