using System;

using System.Collections.Generic;

using System.ComponentModel;

using System.Data;

using System.Drawing;

using System.Linq;

using System.Text;

using System.Threading.Tasks;

using System.Windows.Forms;

using System.IO;

using CodeProject.Dialog;

using System.Drawing.Imaging;

using System.ServiceModel;

using System.Threading;

using System.Diagnostics;

using System.Deployment.Application;

using System.Reflection;

//

using SharpDisplayClient;

using SharpDisplay;

namespace SharpDisplayManager

{

    //Types declarations

    public delegate uint ColorProcessingDelegate(int aX, int aY, uint aPixel);

    public delegate int CoordinateTranslationDelegate(System.Drawing.Bitmap aBmp, int aInt);

    //Delegates are used for our thread safe method

    public delegate void AddClientDelegate(string aSessionId, ICallback aCallback);

    public delegate void RemoveClientDelegate(string aSessionId);

    public delegate void SetFieldDelegate(string SessionId, DataField aField);

    public delegate void SetFieldsDelegate(string SessionId, System.Collections.Generic.IList<DataField> aFields);

    public delegate void SetLayoutDelegate(string SessionId, TableLayout aLayout);

    public delegate void SetClientNameDelegate(string aSessionId, string aName);

    /// <summary>

    /// Our Display manager main form

    /// </summary>

    public partial class MainForm : Form

    {

        DateTime LastTickTime;

        Display iDisplay;

        System.Drawing.Bitmap iBmp;

        bool iCreateBitmap; //Workaround render to bitmap issues when minimized

        ServiceHost iServiceHost;

        // Our collection of clients sorted by session id.

        public Dictionary<string, ClientData> iClients;

        // The name of the client which informations are currently displayed.

        public string iCurrentClientSessionId;

        ClientData iCurrentClientData;

        //

        public bool iClosing;

        //Function pointer for pixel color filtering

        ColorProcessingDelegate iColorFx;

        //Function pointer for pixel X coordinate intercept

        CoordinateTranslationDelegate iScreenX;

        //Function pointer for pixel Y coordinate intercept

        CoordinateTranslationDelegate iScreenY;

		/// <summary>

		/// Manage run when Windows startup option

		/// </summary>

		private StartupManager iStartupManager;

		/// <summary>

		/// System tray icon.

		/// </summary>

		private NotifyIconAdv iNotifyIcon;

        public MainForm()

        {

            iCurrentClientSessionId = "";

            iCurrentClientData = null;

            LastTickTime = DateTime.Now;

            iDisplay = new Display();

            iClients = new Dictionary<string, ClientData>();

			iStartupManager = new StartupManager();

			iNotifyIcon = new NotifyIconAdv();

            InitializeComponent();

            UpdateStatus();

            //We have a bug when drawing minimized and reusing our bitmap

            iBmp = new System.Drawing.Bitmap(tableLayoutPanel.Width, tableLayoutPanel.Height, PixelFormat.Format32bppArgb);

            iCreateBitmap = false;

			if (Properties.Settings.Default.StartMinimized)

			{

				WindowState = FormWindowState.Minimized;

			}

        }

		/// <summary>

		/// 

		/// </summary>

		/// <param name="sender"></param>

		/// <param name="e"></param>

        private void MainForm_Load(object sender, EventArgs e)

        {

			if (ApplicationDeployment.IsNetworkDeployed)

			{

				this.Text += " - v" + ApplicationDeployment.CurrentDeployment.CurrentVersion;

			}

			else

			{

				this.Text += " - development";

			}

            StartServer();

            if (Properties.Settings.Default.DisplayConnectOnStartup)

            {

                OpenDisplayConnection();

            }

			//Check if "run on Windows startup" is enabled

			checkBoxAutoStart.Checked=iStartupManager.Startup;

			//Setup notification icon

			iNotifyIcon.Icon = GetIcon("vfd.ico");

			iNotifyIcon.Text = "Sharp Display Manager";

			iNotifyIcon.Visible = true;

			iNotifyIcon.DoubleClick += delegate(object obj, EventArgs args)

			{

				SysTrayHideShow();

			};

			// To make sure start up with minimize to tray works

			if (WindowState == FormWindowState.Minimized && Properties.Settings.Default.MinimizeToTray)

			{

				Visible = false;

			}

        }

		/// <summary>

		/// Access icons from embedded resources.

		/// </summary>

		/// <param name="name"></param>

		/// <returns></returns>

		public static Icon GetIcon(string name)

		{

			name = "SharpDisplayManager.Resources." + name;

			string[] names =

			  Assembly.GetExecutingAssembly().GetManifestResourceNames();

			for (int i = 0; i < names.Length; i++)

			{

				if (names[i].Replace('\\', '.') == name)

				{

					using (Stream stream = Assembly.GetExecutingAssembly().

					  GetManifestResourceStream(names[i]))

					{

						return new Icon(stream);

					}

				}

			}

			return null;

		}

        /// <summary>

        /// Set our current client.

        /// This will take care of applying our client layout and set data fields.

        /// </summary>

        /// <param name="aSessionId"></param>

        void SetCurrentClient(string aSessionId)

        {

            if (aSessionId == iCurrentClientSessionId)

            {

                //Given client is already the current one.

                //Don't bother changing anything then.

                return;

            }

            //Set current client ID.

            iCurrentClientSessionId = aSessionId;

            //Fetch and set current client data.

            iCurrentClientData = iClients[aSessionId];

            //Apply layout and set data fields.

            UpdateTableLayoutPanel(iCurrentClientData);

        }

        private void buttonFont_Click(object sender, EventArgs e)

        {

            //fontDialog.ShowColor = true;

            //fontDialog.ShowApply = true;

            fontDialog.ShowEffects = true;

            MarqueeLabel label = (MarqueeLabel)tableLayoutPanel.Controls[0];

            fontDialog.Font = label.Font;

            fontDialog.FixedPitchOnly = checkBoxFixedPitchFontOnly.Checked;

            //fontDialog.ShowHelp = true;

            //fontDlg.MaxSize = 40;

            //fontDlg.MinSize = 22;

            //fontDialog.Parent = this;

            //fontDialog.StartPosition = FormStartPosition.CenterParent;

            //DlgBox.ShowDialog(fontDialog);

            //if (fontDialog.ShowDialog(this) != DialogResult.Cancel)

            if (DlgBox.ShowDialog(fontDialog) != DialogResult.Cancel)

            {

                //MsgBox.Show("MessageBox MsgBox", "MsgBox caption");

                //MessageBox.Show("Ok");

                foreach (MarqueeLabel ctrl in tableLayoutPanel.Controls)

                {

                    ctrl.Font = fontDialog.Font;

                }

                cds.Font = fontDialog.Font;

                Properties.Settings.Default.Save();

                //

                CheckFontHeight();

            }

        }

        /// <summary>

        ///

        /// </summary>

        void CheckFontHeight()

        {

            //Show font height and width

            labelFontHeight.Text = "Font height: " + cds.Font.Height;

            float charWidth = IsFixedWidth(cds.Font);

            if (charWidth == 0.0f)

            {

                labelFontWidth.Visible = false;

            }

            else

            {

                labelFontWidth.Visible = true;

                labelFontWidth.Text = "Font width: " + charWidth;

            }

            MarqueeLabel label = null;

            //Get the first label control we can find

            foreach (Control ctrl in tableLayoutPanel.Controls)

            {

                if (ctrl is MarqueeLabel)

                {

                    label = (MarqueeLabel)ctrl;

                    break;

                }

            }

            //Now check font height and show a warning if needed.

            if (label != null && label.Font.Height > label.Height)

            {

                labelWarning.Text = "WARNING: Selected font is too height by " + (label.Font.Height - label.Height) + " pixels!";

                labelWarning.Visible = true;

            }

            else

            {

                labelWarning.Visible = false;

            }

        }

        private void buttonCapture_Click(object sender, EventArgs e)

        {

            System.Drawing.Bitmap bmp = new System.Drawing.Bitmap(tableLayoutPanel.Width, tableLayoutPanel.Height);

            tableLayoutPanel.DrawToBitmap(bmp, tableLayoutPanel.ClientRectangle);

            //Bitmap bmpToSave = new Bitmap(bmp);

            bmp.Save("D:\\capture.png");

            ((MarqueeLabel)tableLayoutPanel.Controls[0]).Text = "Captured";

            /*

            string outputFileName = "d:\\capture.png";

            using (MemoryStream memory = new MemoryStream())

            {

                using (FileStream fs = new FileStream(outputFileName, FileMode.OpenOrCreate, FileAccess.ReadWrite))

                {

                    bmp.Save(memory, System.Drawing.Imaging.ImageFormat.Png);

                    byte[] bytes = memory.ToArray();

                    fs.Write(bytes, 0, bytes.Length);

                }

            }

             */

        }

        private void CheckForRequestResults()

        {

            if (iDisplay.IsRequestPending())

            {

                switch (iDisplay.AttemptRequestCompletion())

                {

                    case Display.TMiniDisplayRequest.EMiniDisplayRequestFirmwareRevision:

                        toolStripStatusLabelConnect.Text += " v" + iDisplay.FirmwareRevision();

                        //Issue next request then

                        iDisplay.RequestPowerSupplyStatus();

                        break;

                    case Display.TMiniDisplayRequest.EMiniDisplayRequestPowerSupplyStatus:

                        if (iDisplay.PowerSupplyStatus())

                        {

                            toolStripStatusLabelPower.Text = "ON";

                        }

                        else

                        {

                            toolStripStatusLabelPower.Text = "OFF";

                        }

                        //Issue next request then

                        iDisplay.RequestDeviceId();

                        break;

                    case Display.TMiniDisplayRequest.EMiniDisplayRequestDeviceId:

                        toolStripStatusLabelConnect.Text += " - " + iDisplay.DeviceId();

                        //No more request to issue

                        break;

                }

            }

        }

        public static uint ColorWhiteIsOn(int aX, int aY, uint aPixel)

        {

            if ((aPixel & 0x00FFFFFF) == 0x00FFFFFF)

            {

                return 0xFFFFFFFF;

            }

            return 0x00000000;

        }

        public static uint ColorUntouched(int aX, int aY, uint aPixel)

        {

            return aPixel;

        }

        public static uint ColorInversed(int aX, int aY, uint aPixel)

        {

            return ~aPixel;

        }

        public static uint ColorChessboard(int aX, int aY, uint aPixel)

        {

            if ((aX % 2 == 0) && (aY % 2 == 0))

            {

                return ~aPixel;

            }

            else if ((aX % 2 != 0) && (aY % 2 != 0))

            {

                return ~aPixel;

            }

            return 0x00000000;

        }

        public static int ScreenReversedX(System.Drawing.Bitmap aBmp, int aX)

        {

            return aBmp.Width - aX - 1;

        }

        public int ScreenReversedY(System.Drawing.Bitmap aBmp, int aY)

        {

            return iBmp.Height - aY - 1;

        }

        public int ScreenX(System.Drawing.Bitmap aBmp, int aX)

        {

            return aX;

        }

        public int ScreenY(System.Drawing.Bitmap aBmp, int aY)

        {

            return aY;

        }

        /// <summary>

        /// Select proper pixel delegates according to our current settings.

        /// </summary>

        private void SetupPixelDelegates()

        {

            //Select our pixel processing routine

            if (cds.InverseColors)

            {

                //iColorFx = ColorChessboard;

                iColorFx = ColorInversed;

            }

            else

            {

                iColorFx = ColorWhiteIsOn;

            }

            //Select proper coordinate translation functions

            //We used delegate/function pointer to support reverse screen without doing an extra test on each pixels

            if (cds.ReverseScreen)

            {

                iScreenX = ScreenReversedX;

                iScreenY = ScreenReversedY;

            }

            else

            {

                iScreenX = ScreenX;

                iScreenY = ScreenY;

            }

        }

        //This is our timer tick responsible to perform our render

        private void timer_Tick(object sender, EventArgs e)

        {

            //Update our animations

            DateTime NewTickTime = DateTime.Now;

            //Update animation for all our marquees

            foreach (Control ctrl in tableLayoutPanel.Controls)

            {

                if (ctrl is MarqueeLabel)

                {

                    ((MarqueeLabel)ctrl).UpdateAnimation(LastTickTime, NewTickTime);

                }

            }

            //Update our display

            if (iDisplay.IsOpen())

            {

                CheckForRequestResults();

                //Draw to bitmap

                if (iCreateBitmap)

                {

                    iBmp = new System.Drawing.Bitmap(tableLayoutPanel.Width, tableLayoutPanel.Height, PixelFormat.Format32bppArgb);

                }

                tableLayoutPanel.DrawToBitmap(iBmp, tableLayoutPanel.ClientRectangle);

                //iBmp.Save("D:\\capture.png");

                //Send it to our display

                for (int i = 0; i < iBmp.Width; i++)

                {

                    for (int j = 0; j < iBmp.Height; j++)

                    {

                        unchecked

                        {

                            //Get our processed pixel coordinates

                            int x = iScreenX(iBmp, i);

                            int y = iScreenY(iBmp, j);

                            //Get pixel color

                            uint color = (uint)iBmp.GetPixel(i, j).ToArgb();

                            //Apply color effects

                            color = iColorFx(x,y,color);

                            //Now set our pixel

                            iDisplay.SetPixel(x, y, color);

                        }

                    }

                }

                iDisplay.SwapBuffers();

            }

            //Compute instant FPS

            toolStripStatusLabelFps.Text = (1.0/NewTickTime.Subtract(LastTickTime).TotalSeconds).ToString("F0") + " / " + (1000/timer.Interval).ToString() + " FPS";

            LastTickTime = NewTickTime;

        }

        private void OpenDisplayConnection()

        {

            CloseDisplayConnection();

            if (iDisplay.Open((Display.TMiniDisplayType)cds.DisplayType))

            {

                UpdateStatus();

                iDisplay.RequestFirmwareRevision();

            }

            else

            {

                UpdateStatus();

                toolStripStatusLabelConnect.Text = "Connection error";

            }

        }

        private void CloseDisplayConnection()

        {

            iDisplay.Close();

            UpdateStatus();

        }

        private void buttonOpen_Click(object sender, EventArgs e)

        {

            OpenDisplayConnection();

        }

        private void buttonClose_Click(object sender, EventArgs e)

        {

            CloseDisplayConnection();

        }

        private void buttonClear_Click(object sender, EventArgs e)

        {

            iDisplay.Clear();

            iDisplay.SwapBuffers();

        }

        private void buttonFill_Click(object sender, EventArgs e)

        {

            iDisplay.Fill();

            iDisplay.SwapBuffers();

        }

        private void trackBarBrightness_Scroll(object sender, EventArgs e)

        {

            cds.Brightness = trackBarBrightness.Value;

            Properties.Settings.Default.Save();

            iDisplay.SetBrightness(trackBarBrightness.Value);

        }

        /// <summary>

        /// CDS stands for Current Display Settings

        /// </summary>

        private DisplaySettings cds

        {

            get

            {

                DisplaysSettings settings = Properties.Settings.Default.DisplaysSettings;

                if (settings == null)

                {

                    settings = new DisplaysSettings();

                    settings.Init();

                    Properties.Settings.Default.DisplaysSettings = settings;

                }

                //Make sure all our settings have been created

                while (settings.Displays.Count <= Properties.Settings.Default.CurrentDisplayIndex)

                {

                    settings.Displays.Add(new DisplaySettings());

                }

                DisplaySettings displaySettings = settings.Displays[Properties.Settings.Default.CurrentDisplayIndex];

                return displaySettings;

            }

        }

        /// <summary>

        /// Check if the given font has a fixed character pitch.

        /// </summary>

        /// <param name="ft"></param>

        /// <returns>0.0f if this is not a monospace font, otherwise returns the character width.</returns>

        public float IsFixedWidth(Font ft)

        {

            Graphics g = CreateGraphics();

            char[] charSizes = new char[] { 'i', 'a', 'Z', '%', '#', 'a', 'B', 'l', 'm', ',', '.' };

            float charWidth = g.MeasureString("I", ft, Int32.MaxValue, StringFormat.GenericTypographic).Width;

            bool fixedWidth = true;

            foreach (char c in charSizes)

                if (g.MeasureString(c.ToString(), ft, Int32.MaxValue, StringFormat.GenericTypographic).Width != charWidth)

                    fixedWidth = false;

            if (fixedWidth)

            {

                return charWidth;

            }

            return 0.0f;

        }

        private void UpdateStatus()

        {

            //Synchronize UI with settings

            //Load settings

            checkBoxShowBorders.Checked = cds.ShowBorders;

            tableLayoutPanel.CellBorderStyle = (cds.ShowBorders ? TableLayoutPanelCellBorderStyle.Single : TableLayoutPanelCellBorderStyle.None);

            //Set the proper font to each of our labels

            foreach (MarqueeLabel ctrl in tableLayoutPanel.Controls)

            {

                ctrl.Font = cds.Font;

            }

            CheckFontHeight();

            checkBoxConnectOnStartup.Checked = Properties.Settings.Default.DisplayConnectOnStartup;

			checkBoxMinimizeToTray.Checked = Properties.Settings.Default.MinimizeToTray;

			checkBoxStartMinimized.Checked = Properties.Settings.Default.StartMinimized;

            checkBoxReverseScreen.Checked = cds.ReverseScreen;

            checkBoxInverseColors.Checked = cds.InverseColors;

            comboBoxDisplayType.SelectedIndex = cds.DisplayType;

            timer.Interval = cds.TimerInterval;

            maskedTextBoxTimerInterval.Text = cds.TimerInterval.ToString();

            //

            SetupPixelDelegates();

            if (iDisplay.IsOpen())

            {

                //Only setup brightness if display is open

                trackBarBrightness.Minimum = iDisplay.MinBrightness();

                trackBarBrightness.Maximum = iDisplay.MaxBrightness();

                trackBarBrightness.Value = cds.Brightness;

                trackBarBrightness.LargeChange = Math.Max(1, (iDisplay.MaxBrightness() - iDisplay.MinBrightness()) / 5);

                trackBarBrightness.SmallChange = 1;

                iDisplay.SetBrightness(cds.Brightness);

                //

                buttonFill.Enabled = true;

                buttonClear.Enabled = true;

                buttonOpen.Enabled = false;

                buttonClose.Enabled = true;

                trackBarBrightness.Enabled = true;

                toolStripStatusLabelConnect.Text = "Connected - " + iDisplay.Vendor() + " - " + iDisplay.Product();

                //+ " - " + iDisplay.SerialNumber();

                if (iDisplay.SupportPowerOnOff())

                {

                    buttonPowerOn.Enabled = true;

                    buttonPowerOff.Enabled = true;

                }

                else

                {

                    buttonPowerOn.Enabled = false;

                    buttonPowerOff.Enabled = false;

                }

                if (iDisplay.SupportClock())

                {

                    buttonShowClock.Enabled = true;

                    buttonHideClock.Enabled = true;

                }

                else

                {

                    buttonShowClock.Enabled = false;

                    buttonHideClock.Enabled = false;

                }

            }

            else

            {

                buttonFill.Enabled = false;

                buttonClear.Enabled = false;

                buttonOpen.Enabled = true;

                buttonClose.Enabled = false;

                trackBarBrightness.Enabled = false;

                buttonPowerOn.Enabled = false;

                buttonPowerOff.Enabled = false;

                buttonShowClock.Enabled = false;

                buttonHideClock.Enabled = false;

                toolStripStatusLabelConnect.Text = "Disconnected";

                toolStripStatusLabelPower.Text = "N/A";

            }

        }

        private void checkBoxShowBorders_CheckedChanged(object sender, EventArgs e)

        {

            //Save our show borders setting

            tableLayoutPanel.CellBorderStyle = (checkBoxShowBorders.Checked ? TableLayoutPanelCellBorderStyle.Single : TableLayoutPanelCellBorderStyle.None);

            cds.ShowBorders = checkBoxShowBorders.Checked;

            Properties.Settings.Default.Save();

            CheckFontHeight();

        }

        private void checkBoxConnectOnStartup_CheckedChanged(object sender, EventArgs e)

        {

            //Save our connect on startup setting

            Properties.Settings.Default.DisplayConnectOnStartup = checkBoxConnectOnStartup.Checked;

            Properties.Settings.Default.Save();

        }

		private void checkBoxMinimizeToTray_CheckedChanged(object sender, EventArgs e)

		{

			//Save our "Minimize to tray" setting

			Properties.Settings.Default.MinimizeToTray = checkBoxMinimizeToTray.Checked;

			Properties.Settings.Default.Save();

		}

		private void checkBoxStartMinimized_CheckedChanged(object sender, EventArgs e)

		{

			//Save our "Start minimized" setting

			Properties.Settings.Default.StartMinimized = checkBoxStartMinimized.Checked;

			Properties.Settings.Default.Save();

		}

		private void checkBoxAutoStart_CheckedChanged(object sender, EventArgs e)

		{

			iStartupManager.Startup = checkBoxAutoStart.Checked;

		}

        private void checkBoxReverseScreen_CheckedChanged(object sender, EventArgs e)

        {

            //Save our reverse screen setting

            cds.ReverseScreen = checkBoxReverseScreen.Checked;

            Properties.Settings.Default.Save();

            SetupPixelDelegates();

        }

        private void checkBoxInverseColors_CheckedChanged(object sender, EventArgs e)

        {

            //Save our inverse colors setting

            cds.InverseColors = checkBoxInverseColors.Checked;

            Properties.Settings.Default.Save();

            SetupPixelDelegates();

        }

        private void MainForm_Resize(object sender, EventArgs e)

        {

            if (WindowState == FormWindowState.Minimized)

            {

                // Do some stuff

                //iBmp = new System.Drawing.Bitmap(tableLayoutPanel.Width, tableLayoutPanel.Height, PixelFormat.Format32bppArgb);

                iCreateBitmap = true;

            }

        }

        private void MainForm_FormClosing(object sender, FormClosingEventArgs e)

        {

            StopServer();

            e.Cancel = iClosing;

        }

        public void StartServer()

        {

            iServiceHost = new ServiceHost

                (

                    typeof(Session),

                    new Uri[] { new Uri("net.tcp://localhost:8001/") }

                );

            iServiceHost.AddServiceEndpoint(typeof(IService), new NetTcpBinding(SecurityMode.None, true), "DisplayService");

            iServiceHost.Open();

        }

        public void StopServer()

        {

            if (iClients.Count > 0 && !iClosing)

            {

                //Tell our clients

                iClosing = true;

                BroadcastCloseEvent();

            }

            else if (iClosing)

            {

                if (MessageBox.Show("Force exit?", "Waiting for clients...", MessageBoxButtons.YesNo, MessageBoxIcon.Warning) == DialogResult.Yes)

                {

                    iClosing = false; //We make sure we force close if asked twice

                }

            }

            else

            {

                //We removed that as it often lags for some reason

                //iServiceHost.Close();

            }

        }

        public void BroadcastCloseEvent()

        {

            Trace.TraceInformation("BroadcastCloseEvent - start");

            var inactiveClients = new List<string>();

            foreach (var client in iClients)

            {

                //if (client.Key != eventData.ClientName)

                {

                    try

                    {

                        Trace.TraceInformation("BroadcastCloseEvent - " + client.Key);

                        client.Value.Callback.OnCloseOrder(/*eventData*/);

                    }

                    catch (Exception ex)

                    {

                        inactiveClients.Add(client.Key);

                    }

                }

            }

            if (inactiveClients.Count > 0)

            {

                foreach (var client in inactiveClients)

                {

                    iClients.Remove(client);

                    Program.iMainForm.treeViewClients.Nodes.Remove(Program.iMainForm.treeViewClients.Nodes.Find(client, false)[0]);

                }

            }

        }

        private void buttonStartClient_Click(object sender, EventArgs e)

        {

            Thread clientThread = new Thread(SharpDisplayClient.Program.Main);

            clientThread.Start();

            BringToFront();

        }

        private void buttonSuspend_Click(object sender, EventArgs e)

        {

            LastTickTime = DateTime.Now; //Reset timer to prevent jump

            timer.Enabled = !timer.Enabled;

            if (!timer.Enabled)

            {

                buttonSuspend.Text = "Run";

            }

            else

            {

                buttonSuspend.Text = "Pause";

            }

        }

        private void buttonCloseClients_Click(object sender, EventArgs e)

        {

            BroadcastCloseEvent();

        }

        private void treeViewClients_AfterSelect(object sender, TreeViewEventArgs e)

        {

        }

        /// <summary>

        ///

        /// </summary>

        /// <param name="aSessionId"></param>

        /// <param name="aCallback"></param>

        public void AddClientThreadSafe(string aSessionId, ICallback aCallback)

        {

            if (this.InvokeRequired)

            {

                //Not in the proper thread, invoke ourselves

                AddClientDelegate d = new AddClientDelegate(AddClientThreadSafe);

                this.Invoke(d, new object[] { aSessionId, aCallback });

            }

            else

            {

                //We are in the proper thread

                //Add this session to our collection of clients

                ClientData newClient = new ClientData(aSessionId, aCallback);

                Program.iMainForm.iClients.Add(aSessionId, newClient);

                //Add this session to our UI

                UpdateClientTreeViewNode(newClient);

            }

        }

        /// <summary>

        ///

        /// </summary>

        /// <param name="aSessionId"></param>

        public void RemoveClientThreadSafe(string aSessionId)

        {

            if (this.InvokeRequired)

            {

                //Not in the proper thread, invoke ourselves

                RemoveClientDelegate d = new RemoveClientDelegate(RemoveClientThreadSafe);

                this.Invoke(d, new object[] { aSessionId });

            }

            else

            {

                //We are in the proper thread

                //Remove this session from both client collection and UI tree view

                if (Program.iMainForm.iClients.Keys.Contains(aSessionId))

                {

                    Program.iMainForm.iClients.Remove(aSessionId);

                    Program.iMainForm.treeViewClients.Nodes.Remove(Program.iMainForm.treeViewClients.Nodes.Find(aSessionId, false)[0]);

                }

                if (iClosing && iClients.Count == 0)

                {

                    //We were closing our form

                    //All clients are now closed

                    //Just resume our close operation

                    iClosing = false;

                    Close();

                }

            }

        }

        /// <summary>

        ///

        /// </summary>

        /// <param name="aSessionId"></param>

        /// <param name="aLayout"></param>

        public void SetClientLayoutThreadSafe(string aSessionId, TableLayout aLayout)

        {

            if (this.InvokeRequired)

            {

                //Not in the proper thread, invoke ourselves

                SetLayoutDelegate d = new SetLayoutDelegate(SetClientLayoutThreadSafe);

                this.Invoke(d, new object[] { aSessionId, aLayout });

            }

            else

            {

                ClientData client = iClients[aSessionId];

                if (client != null)

                {

                    client.Layout = aLayout;

                    UpdateTableLayoutPanel(client);

                    //

                    UpdateClientTreeViewNode(client);

                }

            }

        }

        /// <summary>

        ///

        /// </summary>

        /// <param name="aSessionId"></param>

        /// <param name="aField"></param>

        public void SetClientFieldThreadSafe(string aSessionId, DataField aField)

        {

            if (this.InvokeRequired)

            {

                //Not in the proper thread, invoke ourselves

                SetFieldDelegate d = new SetFieldDelegate(SetClientFieldThreadSafe);

                this.Invoke(d, new object[] { aSessionId, aField });

            }

            else

            {

                //We are in the proper thread

                //Call the non-thread-safe variant

                SetClientField(aSessionId, aField);

            }

        }

        /// <summary>

        ///

        /// </summary>

        /// <param name="aSessionId"></param>

        /// <param name="aField"></param>

        private void SetClientField(string aSessionId, DataField aField)

        {

            SetCurrentClient(aSessionId);

            ClientData client = iClients[aSessionId];

            if (client != null)

            {

                bool somethingChanged = false;

                //Make sure all our fields are in place

                while (client.Fields.Count < (aField.Index + 1))

                {

                    //Add a text field with proper index

                    client.Fields.Add(new DataField(client.Fields.Count));

                    somethingChanged = true;

                }

                if (client.Fields[aField.Index].IsSameLayout(aField))

                {

                    //Same layout just update our field

                    client.Fields[aField.Index] = aField;

                    //

                    if (aField.IsText && tableLayoutPanel.Controls[aField.Index] is MarqueeLabel)

                    {

                        //Text field control already in place, just change the text

                        MarqueeLabel label = (MarqueeLabel)tableLayoutPanel.Controls[aField.Index];

                        somethingChanged = (label.Text != aField.Text || label.TextAlign != aField.Alignment);

                        label.Text = aField.Text;

                        label.TextAlign = aField.Alignment;

                    }

                    else if (aField.IsBitmap && tableLayoutPanel.Controls[aField.Index] is PictureBox)

                    {

                        somethingChanged = true; //TODO: Bitmap comp or should we leave that to clients?

                        //Bitmap field control already in place just change the bitmap

                        PictureBox pictureBox = (PictureBox)tableLayoutPanel.Controls[aField.Index];

                        pictureBox.Image = aField.Bitmap;

                    }

                    else

                    {

                        somethingChanged = true;

                        //The requested control in our layout it not of the correct type

                        //Wrong control type, re-create them all

                        UpdateTableLayoutPanel(iCurrentClientData);

                    }

                }

                else

                {

                    somethingChanged = true;

                    //Different layout, need to rebuild it

                    client.Fields[aField.Index] = aField;

                    UpdateTableLayoutPanel(iCurrentClientData);

                }

                //

                if (somethingChanged)

                {

                    UpdateClientTreeViewNode(client);

                }

            }

        }

        /// <summary>

        ///

        /// </summary>

        /// <param name="aTexts"></param>

        public void SetClientFieldsThreadSafe(string aSessionId, System.Collections.Generic.IList<DataField> aFields)