tbf/ResultsBrowser/Forms/StatisticsDlg.cs

using System;
using System.Collections.Generic;
using System.Windows.Forms;
using Results.Entities;
using ResultsBrowser.Resources;
using ResultsBrowser.Output.Printers.Statistics;
using System.IO;

namespace ResultsBrowser.Forms
{
    public partial class StatisticsDlg : Form
    {
        IList<WaterMeter> waterMeters;

        bool twoDim;

        Results.WMeterRsltItemSpec rowItem;
		Results.WMeterRsltItemSpec rowItem2;
		Results.WMeterRsltItemSpec rowItem3;
		Results.WMeterRsltItemSpec columnItem;
		Results.WMeterRsltItemSpec columnItem2;
		Results.WMeterRsltItemSpec columnItem3;

		string rowTestSpec;
		string rowTestSpec2;
		string rowTestSpec3;
		string columnTestSpec;
		string columnTestSpec2;
		string columnTestSpec3;

        IList<string> rowValues;
        IList<string> columnValues;

		int[,] occurances;

		string title;


		public StatisticsDlg(IList<WaterMeter> waterMeters, bool twoDim,
							 Results.WMeterRsltItemSpec rowItem,
							 Results.WMeterRsltItemSpec rowItem2,
							 Results.WMeterRsltItemSpec rowItem3,
							 Results.WMeterRsltItemSpec columnItem,
							 Results.WMeterRsltItemSpec columnItem2,
							 Results.WMeterRsltItemSpec columnItem3,
							 string rowTestSpec,
							 string rowTestSpec2,
							 string rowTestSpec3,
							 string columnTestSpec,
							 string columnTestSpec2,
							 string columnTestSpec3,
							 DateTime startDate, DateTime endDate)
        {
            InitializeComponent();

			if (!string.IsNullOrEmpty(Program.LocalSettings.TimePeriodFormat) && (startDate != new DateTime(0)) && (endDate != new DateTime(0)))
			{
				title = string.Format(Program.LocalSettings.TimePeriodFormat, startDate, endDate);
			}
			else
			{
				title = Strings.Statistics;
			}

			Text = title;

            closeButton.Text = Strings.OkBtnText;
			printButton.Text = Strings.Print;

            this.waterMeters = waterMeters;

            this.twoDim = twoDim;

			this.rowItem = rowItem;
			this.rowItem2 = rowItem2;
			this.rowItem3 = rowItem3;

			this.columnItem = columnItem;
			this.columnItem2 = columnItem2;
			this.columnItem3 = columnItem3;

            this.rowTestSpec = rowTestSpec;
			this.rowTestSpec2 = rowTestSpec2;
			this.rowTestSpec3 = rowTestSpec3;

			this.columnTestSpec = columnTestSpec;
			this.columnTestSpec2 = columnTestSpec2;
			this.columnTestSpec3 = columnTestSpec3;
        }

        private void StatisticsDlg_Load(object sender, EventArgs e)
        {
            Calculate();
        }

        bool Calculate()
        {
            try
            {
                rowValues = new List<string>();
                columnValues = new List<string>();

                ///
                /// Find discrete values of rows and columns
                ///
                foreach (var wm in waterMeters)
                {
                    string value = rowItem.Print(wm);
					if (rowItem2 != null) value = value + " | " + rowItem2.Print(wm);
					if (rowItem3 != null) value = value + " | " + rowItem3.Print(wm);
					if (!rowValues.Contains(value)) rowValues.Add(value);

                    if (twoDim)
                    {
                        value = columnItem.Print(wm);
						if (columnItem2 != null) value = value + " | " + columnItem2.Print(wm);
						if (columnItem3 != null) value = value + " | " + columnItem3.Print(wm);
						if (!columnValues.Contains(value)) columnValues.Add(value);
                    }
                }

                ///
                /// Calculate occurances
                ///
                int columnsCount = Math.Max(1, columnValues.Count);
                occurances = new int[rowValues.Count, columnsCount];

                foreach (var wm in waterMeters)
                {
                    for (int row = 0; row < rowValues.Count; row++)
                    {
						string rowValue = rowItem.Print(wm);
						if (rowItem2 != null) rowValue = rowValue + " | " + rowItem2.Print(wm);
						if (rowItem3 != null) rowValue = rowValue + " | " + rowItem3.Print(wm);

                        if (twoDim)
                        {
                            for (int column = 0; column < columnsCount; column++)
                            {
								string columnValue = columnItem.Print(wm);
								if (columnItem2 != null) columnValue = columnValue + " | " + columnItem2.Print(wm);
								if (columnItem3 != null) columnValue = columnValue + " | " + columnItem3.Print(wm);

								if (rowValue == rowValues[row] && columnValue == columnValues[column])
								{
									occurances[row, column]++;
								}
                            }
                        }
                        else
                        {
                            if (rowValue == rowValues[row]) occurances[row, 0]++;
                        }
                    }                    
                }

                ///
                /// Update ListBox
                ///
                statisticsListView.View = View.Details;
                statisticsListView.GridLines = true;
                statisticsListView.FullRowSelect = true;

                statisticsListView.Columns.Add(" ", 200);
                for (int column = 0; column < columnsCount; column++)
                {
                    statisticsListView.Columns.Add((twoDim ? columnValues[column] : Strings.Count), 70);
                }

                for (int row = 0; row < rowValues.Count; row++)
                {
                    ListViewItem lvi = new ListViewItem(rowValues[row]);
                    for (int column = 0; column < columnsCount; column++) lvi.SubItems.Add(occurances[row, column].ToString());
                    statisticsListView.Items.Add(lvi);
                }

                return true;
            }
            catch (Exception exc)
            {
                MessageBox.Show(exc.Message, Strings.Error, MessageBoxButtons.OK, MessageBoxIcon.Exclamation);
                return false;
            }
        }

        private void closeButton_Click(object sender, EventArgs e)
        {
            DialogResult = DialogResult.OK;
            Close();
        }

		private void printButton_Click(object sender, EventArgs e)
		{
            int columnsCount = Math.Max(1, columnValues.Count);
			string[,] table = new string[rowValues.Count + 1, columnsCount + 1];

			table[0, 0] = string.Empty;
			for (int column = 0; column < columnsCount; column++)
			{
				table[0, column + 1] = (twoDim ? columnValues[column] : Strings.Count);
			}

			for (int row = 0; row < rowValues.Count; row++)
			{
				table[row + 1, 0] = rowValues[row];
				for (int column = 0; column < columnsCount; column++) table[row + 1, column + 1] = occurances[row, column].ToString();
			}

			switch (MessageBox.Show(Strings.Yes_Portrait_No_Landscape, Strings.Page_orientation, MessageBoxButtons.YesNoCancel, MessageBoxIcon.Question))
			{
				case DialogResult.Yes:
					new StatisticsPrintDocument(title, table, Config.Entities.PageOrientation.Portrait).Print();
					break;
				case DialogResult.No:
					new StatisticsPrintDocument(title, table, Config.Entities.PageOrientation.Landscape).Print();
					break;
				default:
					break;
			}
		}

        private void exportButton_Click(object sender, EventArgs e)
        {
            if (rowValues == null || columnValues == null || occurances == null) return;

            SaveFileDialog dlg = new SaveFileDialog();
            dlg.Filter = "CSV files (*.csv)|*.csv|All files (*.*)|*.*";

            if (dlg.ShowDialog() == DialogResult.OK)
            {
                //IList<MonitoringDB.ItemSpec> items = currentQR.ProjectedItems;

                try
                {
                    using (TextWriter writer = new StreamWriter(dlg.FileName, false, System.Text.Encoding.UTF8))
                    {
                        if (twoDim)
                        {
                            /// Write column headers
                            for (int i = 0; i < columnValues.Count; i++)
                            {
                                writer.Write(string.Format(";\"{0}\"", columnValues[i]));
                            }
                            writer.WriteLine();
                        }
                        else
                        {
                            writer.WriteLine(string.Format("\"{0}\";\"{1}\"", rowItem.Caption, Strings.Count));
                        }

                        int nrColumns = Math.Max(1, columnValues.Count);

                        /// Write rows, row header first
                        for (int j = 0; j < rowValues.Count; j++)
                        {
                            writer.Write(string.Format("\"{0}\"", rowValues[j]));
                            for (int i = 0; i < nrColumns; i++)
                            {
                                writer.Write(string.Format(";{0}", occurances[j,i]));
                            }
                            writer.WriteLine();
                        }
                    }
                }
                catch (Exception exc)
                {
                    MessageBox.Show(exc.Message, Strings.Error_exporting_results, MessageBoxButtons.OK, MessageBoxIcon.Exclamation);
                }
            }
        }
    }
}