OTS/OTSIncAMeasureApp/ServiceCenter/BrukerCSharp/EDSController.cs




using System;
using System.Collections.Generic;

namespace OTSModelSharp.ServiceCenter
{
    using  OTSCLRINTERFACE;
    using OTSMeasureApp.ServiceCenter;
    using System.Drawing;
    

    public class BrukerEDSController : IEDSController
    {
        private COTSControlFunExport eds;
        static IEDSController edsctrl = null;
        private List<String> keyElenamelist = new List<string>();
        public static IEDSController GetEDSController(int imgwidth,int imgheight,int expectCount)
        {
            string deviceType = FileHelper.GetXMLInformations("EDSName");
            if (edsctrl == null)
            {
                if (deviceType == "FEI")
                {
                    edsctrl = new FEIEDSController(expectCount);

                }
                else if (deviceType == "Oxford")
                {
                    edsctrl = new OxfordEDSController(expectCount);

                }
                else if (deviceType == "Bruker")
                {
                    edsctrl = new BrukerEDSController("Bruker",expectCount);


                }
                else if (deviceType == "OffLine")
                {
                    edsctrl = new EDSController("OffLine",expectCount);

                }
                edsctrl.SetResolution(imgwidth, imgheight);


            }
            return edsctrl;
        }

        private BrukerEDSController(string deviceType,int expectcount)
        {
         
                eds = COTSControlFunExport.GetControllerInstance(deviceType);
            eds.SetExpectCount(expectcount);
           
        }
        private void ProcessXrayInfo(COTSParticleClr partWithXrayInfo)//sometime the result will contain repeat percentage data for one element.It must be processed.
        {
         var eleChemistry =  partWithXrayInfo.GetXray().GetElementQuantifyData();
            Dictionary<string, List<double>> eleInfoDic = new Dictionary<string, List<double>>();
            bool hasRepeatEle = false;
            foreach (var ele in eleChemistry)
            {
                if (eleInfoDic.ContainsKey(ele.GetName()))//contain repeat data;
                {
                    eleInfoDic[ele.GetName()].Add(ele.GetPercentage());
                    hasRepeatEle = true;
                }
                else
                {

                    eleInfoDic.Add(ele.GetName(), new List<double>() { ele.GetPercentage() });
                }



            }
            if (hasRepeatEle)
            {
                Dictionary<string, double> resultInfo = new Dictionary<string, double>();
                foreach (var eleInfo in eleInfoDic)
                {
                    double newPercentData=0;
                    foreach (var p  in eleInfo.Value)
                    {
                        newPercentData += p;
                    }
                    newPercentData = newPercentData / eleInfo.Value.Count;
                    resultInfo.Add(eleInfo.Key, newPercentData);              
                }
                foreach (var e in eleChemistry)
                {
                    e.SetPercentage(resultInfo[e.GetName()]);
                
                }
                partWithXrayInfo.GetXray().SetElementQuantifyData(eleChemistry);


            }


        }
        public bool GetXRayByFeatures(List<COTSParticleClr> a_listParticles,  double a_nXRayAQTime, bool a_bElementInfo)
        {
            bool result = false;
           
            if (!eds.IsConnected())
            {
                return false;
            }

            if (keyElenamelist.Count > 0)
            { 
                if (this.GetEDSType() == EDSTYPE.BRUKER)
                {
                    List<CElementChemistryClr> elementChemistryClrs = new List<CElementChemistryClr>();
                    for (int j = 0; j < keyElenamelist.Count; j++)
                    {
                        CElementChemistryClr chemistryClr = new CElementChemistryClr();
                        chemistryClr.SetName(keyElenamelist[j]);
                        elementChemistryClrs.Add(chemistryClr);
                    }
                    for (int i = 0; i < a_listParticles.Count; i++)
                    {

                        a_listParticles[i].GetXray().SetElementQuantifyData(elementChemistryClrs);
                    }
                }
            }


            COTSParticleClr[] parts = a_listParticles.ToArray();


            result= eds.GetXRayByFeatures((uint)a_nXRayAQTime, parts, a_bElementInfo);


            if (result == true)
            {
                foreach (var p in a_listParticles)
                {
                    ProcessXrayInfo(p);
                }


            }
            return result;
        }

        public bool GetXRayByParts(List<COTSParticleClr> a_listParticles, uint a_nXRayAQTime, bool a_bElementInfo)
        {
            bool result = false;
           
                if (!eds.IsConnected())
                {
                    return false;
                }

                int xrayNum = a_listParticles.Count;
                Point[] Ps = new Point[xrayNum];

                for (int i = 0; i < xrayNum; i++)
                {
                    Point p = (Point)a_listParticles[i].GetXRayPos();
                    Ps[i].X = p.X;
                    Ps[i].Y = p.Y;
                }
            if (keyElenamelist.Count > 0)
            { 
                if (this.GetEDSType() == EDSTYPE.BRUKER)
                {
                    List<CElementChemistryClr> elementChemistryClrs = new List<CElementChemistryClr>();
                
                    for (int j = 0; j < keyElenamelist.Count; j++)
                    {
                        CElementChemistryClr chemistryClr = new CElementChemistryClr();
                        chemistryClr.SetName(keyElenamelist[j]);
                        elementChemistryClrs.Add(chemistryClr);
                    }
                    for (int i = 0; i < a_listParticles.Count; i++)
                    {
                   
                        a_listParticles[i].GetXray().SetElementQuantifyData(elementChemistryClrs);
                    }
                }
            }
           
            int nSize = a_listParticles.Count;
            if (nSize > 1024)
            {
               
                COTSParticleClr[] partsTemp = new COTSParticleClr[1024];
                Point[] PsTemp = new Point[1024];

                int nTimes = nSize / 1024;
                for (int i = 0; i < nTimes; i++)
                {
                    NLog.LogManager.GetCurrentClassLogger().Warn("begin 1024 batch");
                    for (int m = 0; m < 1024; m++)
                    {
                        partsTemp[m]=a_listParticles[i * 1024 + m];
                        PsTemp[m] = Ps[i * 1024 + m];
                    }

                    if (!eds.GetXRayByPoints(a_nXRayAQTime, PsTemp, partsTemp, a_bElementInfo))
                    {
                        NLog.LogManager.GetCurrentClassLogger().Error("GetXRayByPoints: failed to get element.");
                        return false;
                    }
                   

                }

                int nLast = nSize % 1024;
                if (nLast != 0)
                {
                   
                    COTSParticleClr[] lastParts = new COTSParticleClr[nLast];
                    Point[] lastPs = new Point[nLast];
                   
                    for (int m = 0; m < nLast; m++)
                    {
                        
                        lastParts[m] = a_listParticles[nTimes * 1024 + m];
                        lastPs[m] = Ps[nTimes * 1024 + m];
                    }

                    if (!eds.GetXRayByPoints(a_nXRayAQTime, lastPs, lastParts, a_bElementInfo))
                    {
                        NLog.LogManager.GetCurrentClassLogger().Error("GetXRayByPoints: failed to get element.");
                        return false;
                    }                 

                   

                }


            }
            else 
            {
                COTSParticleClr[] parts = a_listParticles.ToArray();



                result = eds.GetXRayByPoints(a_nXRayAQTime, Ps, parts, a_bElementInfo);



            }
           




            //if (result == true)
            //{
            //    foreach (var p in a_listParticles)
            //    {
            //        ProcessXrayInfo(p);
            //    }


            //}
            return result;
        }
      

        public bool CollectSpectrum(uint a_nXRayAQTime, ref uint[] a_XrayData)
        {

                if (!eds.IsConnected())
                {
                    return false;
                }

                return eds.CollectSpectrum(a_nXRayAQTime, ref a_XrayData);
            
        }


        public bool Init()
        {
            if (!eds.IsConnected())
            {
                eds.ConncetSem();
            }
                bool m_init = eds.EDSInit();
                return m_init;
            
        }

      

        public bool Connect()
        {
           
                if (!eds.Init())
                {
                    return false;
                }
                if (eds.IsConnected())
                {
                    return true;
                }

                return eds.ConncetSem();
           
        }

        public EDSTYPE GetEDSType()
        {
            EDSTYPE t;
            switch (eds.EDSGetType())
            {
                case 1:
                    t = EDSTYPE.OFFLINE;
                    break;
                case 3:
                    t = EDSTYPE.BRUKER;
                    break;
                case 4:
                    t = EDSTYPE.OXFORD;
                    break;
                default:
                    t = EDSTYPE.OFFLINE;
                    break;
                
            }
            return t;
        }

    

        public void SetFilterKeyEleNames(List<string> KeyNameList)
        {
            this.keyElenamelist = KeyNameList;
        }

       

        public void SetResolution(int resolutionWidth, int resolutionHeight)
        {
            eds.SetImageSize(resolutionWidth, resolutionHeight);
            return ;
        }

        public bool QuantifyXrayByPart(COTSParticleClr part)
        {
           return eds.QuantifyXrayByPart(part);
        }

        public int GetExpectCount()
        {
            return eds.GetExpectCount();
        }
    }
}