iterator pattern changes

Author: anupavanm

IteratorPattern/Program.cs

@@ -5,103 +5,83 @@
 
 namespace IteratorPattern
 {
-class RadioStation
-{
-  private float mFrequency;
-
-  public RadioStation(float frequency)
-  {
-    mFrequency = frequency;
-  }
-
-  public float GetFrequecy()
+  class RadioStation
   {
-    return mFrequency;
-  }
+    private float mFrequency;
 
-}
-
-class StationList : IEnumerable
-{
-  private List<RadioStation> mStations;
-
-  public StationList()
-  {
-    mStations = new List<RadioStation>();
-  }
+    public RadioStation(float frequency)
+    {
+      mFrequency = frequency;
+    }
 
-  public void AddStation(RadioStation station)
-  {
-    mStations.Add(station);
-  }
+    public float GetFrequecy()
+    {
+      return mFrequency;
+    }
 
-  public void RemoveStation(RadioStation station)
-  {
-    mStations.Remove(station);
   }
 
-  public IEnumerator GetEnumerator()
+  class StationList : IEnumerable<RadioStation>
   {
-    return new StationIterator(mStations);
-  }
-}
+    List<RadioStation> mStations = new List<RadioStation>();
 
-class StationIterator : IEnumerator
-{
-  private List<RadioStation> mStations;
-  private int currentPosition = -1;
+    public RadioStation this[int index]
+    {
+      get { return mStations[index]; }
+      set { mStations.Insert(index, value); }
+    }
 
-  public object Current
-  {
-    get
+    public void Add(RadioStation station)
     {
-      return mStations[currentPosition];
+      mStations.Add(station);
     }
-  }
 
-  public StationIterator(List<RadioStation> stations)
-  {
-    mStations = stations;
-  }
-  public bool MoveNext()
-  {
-    if(currentPosition < mStations.Count - 1)
+    public void Remove(RadioStation station)
     {
-      currentPosition = currentPosition + 1;
-      return true;
+      mStations.Remove(station);
     }
 
-    return false;
-  }
+    public IEnumerator<RadioStation> GetEnumerator()
+    {
+      return mStations.GetEnumerator();
+    }
 
-  public void Reset()
-  {
-    currentPosition = -1;
+    IEnumerator IEnumerable.GetEnumerator()
+    {
+      //Use can switch to this internal collection if you do not want to transform
+      //return this.GetEnumerator();
+
+      //use this if you want to transform the object before rendering
+      foreach (var x in mStations)
+      {
+        yield return x;
+      }
+    }
   }
-}
 
-class Program
-{
-  static void Main(string[] args)
+  class Program
   {
-    var stations = new StationList();
-    var station1 = new RadioStation(89);
-    stations.AddStation(station1);
+    static void Main(string[] args)
+    {
+      var stations = new StationList();
+      var station1 = new RadioStation(89);
+      stations.Add(station1);
 
-    var station2 = new RadioStation(101);
-    stations.AddStation(station2);
+      var station2 = new RadioStation(101);
+      stations.Add(station2);
 
-    var station3 = new RadioStation(102);
-    stations.AddStation(station3);
+      var station3 = new RadioStation(102);
+      stations.Add(station3);
 
-    foreach(RadioStation station in stations)
-    {
-      Console.WriteLine(station.GetFrequecy());
-    }
+      foreach (var x in stations)
+      {
+        Console.Write(x.GetFrequecy());
+      }
 
-    stations.RemoveStation(station2); // Will Remove station 101
+      var q = stations.Where(x => x.GetFrequecy() == 89).FirstOrDefault();
+      Console.WriteLine(q.GetFrequecy());
 
-    Console.ReadLine();
+      Console.ReadLine();
+    }
   }
 }
-}