tbf/ResultsBrowser/Filters/WMPositionFilter.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Windows.Forms;
using System.Xml.Serialization;
using NHibernate;
using Results.Entities;
using ResultsBrowser.Resources;

namespace ResultsBrowser.Filters
{
    public class WMPositionFilter : IFilter
    {
        public static string GetFilterID() { return "WM_position"; }
        public static string GetFilterName() { return "WM Position"; }

        ///
        /// Obligatory wrappers
        ///
        public string GetThisFilterID() { return GetFilterID(); }
        public void Save(StringBuilder sb) { (new XmlSerializer(this.GetType())).Serialize(new StringWriter(sb), this); }
        public void Data2UI() { userIface.Data2UI(); }
        public void UI2Data() { userIface.UI2Data(); }
        public UserControl GetFilterUI() { return userIface as UserControl; }

        ///
        /// Serialized filter data
        ///
        public int WMPosition;


        private WMPositionFilterUI userIface;

        public WMPositionFilter()
        {
            userIface = new WMPositionFilterUI(this);
        }

        public IList<WaterMeter> ExecuteQuery(ISession[] sessions, IList<WaterMeter> waterMeters)
        {
            IList<WaterMeter> filteredWaterMeters = new List<WaterMeter>();

            if (waterMeters == null)
            {
                foreach (var se in sessions)
                {
                    IList<Batch> batches = se.QueryOver<Batch>()
                                             .List<Batch>();

                    foreach (var b in batches)
                    {
                        foreach (var wm in b.WaterMeters) if (wm.WMPosition == this.WMPosition) filteredWaterMeters.Add(wm);
                    }
                }
            }
            else
            {
                foreach (var wm in waterMeters)
                {
                    if (wm.WMPosition == this.WMPosition) filteredWaterMeters.Add(wm);
                }
            }

            return filteredWaterMeters;
        }
    }
}