OTS/OTSCPP/OTSControl/Bruker/OTSBrukerImpl.h

#pragma once

#include "OTSSingleton.h"

// Load ClientDll 
#include "OTSBrukerImpl_const.h"
#include "OTSBrukerClientDll.h"
#include "otsdataconst.h"
#include "Element.h"
#include "BSEImg.h"
#include "PosXray.h"
#include "OTSFeature.h"

using namespace OTSDATA;

namespace OTSController {

	typedef enum class __declspec(dllexport) CONTROL_TYPE
	{
		BRUKER_SEM = 0,					//든쓸�구
		BRUKER_SCAN = 1,				//暠獗�췄친욥
		BRUKER_XRAY = 2					//X-RAY친욥
	} CONTROL_TYPE;

	
	typedef enum class __declspec(dllexport) BRUKER_PARAM
	{
		BUF_LENGTH = 1024,
		MAX_SEGMENTS = 8192,
		MAX_PIXEL_MICROSECONDS = 32000,
		RT_SPECTRUM_BUFFER_SIZE = 64000,	    // SpectrumBufferSize
		RT_SPECTRUM_CHANNELS = 2000,				// XANA_CHANNELS = 2000,
		RT_SPECTRUM_CHANNELS_LONG = 3000			// RT_SPECTRUM_CHANNELS_LONG = 3000,
	} BRUKER_PARAM;

	class COTSBrukerImpl : public Singleton<COTSBrukerImpl>, public boost::noncopyable
	{

	public:

		// constructor
		COTSBrukerImpl();
		// destructor
		~COTSBrukerImpl();
		
		// initialization
		BOOL Init(CONTROL_TYPE a_nControlType, BOOL a_b32Bite = TRUE);
	
		BOOL DisConnect();
		// check connection
		BOOL CheckConnection(BOOL& a_bConnected);

		// SEM functions

		// SEM Data (mag, KV and working distance)
		BOOL GetSEMData(double& a_dMagnification, double& a_dHighVoltage, double& a_dWorkingDistance);
		BOOL SetSEMData(double a_dMagnification, double a_dHighVoltage, double a_dWorkingDistance);

		// SEM BC Data (brightness and contrast)
		BOOL GetSEMBCData(double& a_dBrightness, double& a_dContrast);
		BOOL SetSEMBCData(double a_dBrightness, double a_dContrast);

		// Probe Current
		BOOL GetProbeCurrent(double&);
		BOOL SetProbeCurrent(double);

		// spot size
		BOOL GetSEMSpotSize(double& a_dSpotSize);
		BOOL SetSEMSpotSize(double a_dSpotSize);

		// stage data
		BOOL GetSEMStageData(double& a_nPositionX, double& a_nPositionY, double& a_nPositionZ, double& a_dTilt, double& a_dRotation);
		BOOL SetSEMStageData(double a_nPositionX, double a_nPositionY, double a_nPositionZ, double a_dTilt, double a_dRotation);

		// external on/off
		BOOL SetSEMExternalOn(void);
		BOOL SetSEMExternalOff(void);

		// get scan Mode
		BOOL GetScanMode(long& a_nScanMode);
		// set scan mode
		BOOL SetScanMode(long a_nScanMode);

		// Switch SEM Off
		BOOL SwitchSEMOff(BOOL a_bHTValue, BOOL a_bBeamCurrent, BOOL a_bBeamBlank);

		// imaging functions
		
		// Get and set image config - dimensions, dwell time and enabled channels
		BOOL ImageGetConfiguration(DWORD& a_nWidth, DWORD& a_nHeight, DWORD& a_nAverage, BYTE& a_bCh1, BYTE& a_bCh2);
		BOOL ImageSetConfiguration(DWORD a_nWidth, DWORD a_nHeight, DWORD a_nAverage, BYTE a_bCh1, BYTE a_bCh2);
		
		// image collection, using the size and dwell time that was last set
		CBSEImgPtr AcquireImage();
		
		// set point
		BOOL ImageSetPoint(const CPoint& a_oPoint);
		
		// x-ray functions
		
		// live time
		float GetLiveTime(void);
		BOOL GetLiveRealTime(float& a_dLiveTime, float& a_dRealTime);
		
		// collect spectrum data by timeB
		BOOL CollectSpectrum(DWORD a_nMilliSeconds, long* a_pCounts, DWORD a_nBufferSize);

		// collect spectrum data by timeB and quantify
		BOOL QuantifySpectrumOut(DWORD a_nMilliSeconds, long* a_pCounts, DWORD a_nBufferSize, CElementChemistriesList& a_listElementChemistries);

		// quantify spectrum
		BOOL QuantifySpectrum(char* a_pcMethod, char* a_pcParams, char* a_pcResultBuffer, long a_nBufferSize);

		// collect x-ray
		BOOL CollectXRayPoints(std::vector<CPosXray*>& a_vXPoints, DWORD a_nACTimeMS);
		BOOL CollectXRayPoints(CPosXraysList& a_listXrayPois, DWORD a_nACTimeMS);
		BOOL CollectXRayPointsByFeatures(std::vector<CPosXray*>& a_vXPoints, std::vector<std::vector<BrukerSegment>>& a_vFeatures, DWORD a_nACTimeMS);
	
		BOOL ReadXRayPoints(std::vector<CPosXray*>& a_vXPoints, const DWORD a_nACTimeMS);
		BOOL ReadXRayPoints(CPosXraysList& a_listXrayPois, const DWORD a_nACTimeMS);
		BOOL ReadXRayPointsByFeature(std::vector<CPosXray*>& a_vXPoints, const DWORD a_nACTimeMS);
		BOOL ReadXRayPointsByFeature(CPosXraysList& a_vXPoints, const DWORD a_nACTimeMS);

		BOOL CollectOneXRayPoint(const CPoint& a_oPoi, DWORD a_nLifeTimeMilliSeconds, long* a_pnCounts, DWORD a_nBufferSize, bool a_bSetHeadStruc = FALSE);

		// spectrum measurement 
		BOOL StartSpectrumMeasurement();
		BOOL IsSpectrumMeterRunning(BOOL& a_bRunning);
		BOOL StopSpectrumMeasure(void);
		BOOL ReadSpectrum(long* a_pnCounts, DWORD a_nBufferSize);

		// get quantification method name
		BOOL GetQuantificationMethods(std::vector<CString>& a_vMethods);
		BOOL QuantifyXrayPoint(CPosXray* a_pXRayPoint, CString a_strMethodName);
		BOOL QuantifySpectrumFile(LPCTSTR a_sFilePathName, CString a_strMethodName, CElementChemistriesList& a_vElementChemistry);

		RTSpectrumHeaderRec* GetSectrumHeader();

		// if do x-ray quantification
		BOOL GetQuantificationFlag() { return m_bDoQuantification; }
		void SetQuantificationFlag(BOOL a_bDoQuantification) { m_bDoQuantification = a_bDoQuantification; }

		// m_bDoQuantification should be set first. 
		BOOL GetXRayByPoints(CPosXraysList& a_listXrayPois, DWORD a_nACTimeMS);
		BOOL SetXRayPoints(CPosXraysList& a_listXrayPois, const DWORD a_nACTimeMS);
		BOOL GetXRayByFeatures(CPosXraysList& a_vXPoints, std::vector<BrukerFeature> a_vFeatures, DWORD a_nACTimeMS);
		BOOL SetXRayPointsByFeature(CPosXraysList& a_vXPoints, const DWORD a_nACTimeMS);
	protected:
		// close client
		BOOL CloseClient(void);//there's no need to call from out ,the single instance destruction method will call it automatically.

		BOOL GetSpectrum(long a_nBufferIndex, RTSpectrumHeaderRec* a_poSpcHR, long a_nBufferSize);

		BOOL LoadSpectrum(char* a_sFilePathName);
		BOOL ReadSpectrum(void);

		BOOL PutSpectrum(char* m_pBuffer, long a_nBufferSize);
		BOOL CreateSpectrum(char* m_pParamBuffer, RTSpectrumHeaderRec* a_poSpcHR, char* m_pResultBuffer, long& a_nBufferSize);

		BOOL ShowSpectrum(long a_nBufferIndex, char* a_pSpectumName);

		BOOL GetSpectrometerParams(long a_nBufferIndex, char* a_pcBuffer, long& a_nBufferSize);

		BOOL GetElementChemistryData(CElementChemistry& a_oElementChemistryData, CString a_strElementChemistry);

		BOOL CopySpectrum(long* a_pnCounts, DWORD a_nBufferSize);

		BOOL StartSpectrumLifeTimeMeasurement(DWORD a_nLifeTime);
		BOOL SpectrumLifeTimeMeasurement(DWORD a_nLifeTime);
		BOOL StartSpectrumRealTimeMeasurement(DWORD a_nRealTime);
		BOOL SpectrumRealTimeMeasurement(DWORD a_nRealTime);
		BOOL SpectrumCountsMeasurement(DWORD a_nCounts, int a_nTimeLimit = -1);

		BOOL ConvertSpectrum(RTSpectrumHeaderRec* a_poSpcHR, long* a_pnCounts, DWORD a_nBufferSize);

		BOOL GetPointListSpectrum(long a_nIndex, long* a_pnSpec);

		BOOL GetFeatureListSpectrum(long a_nIndex, long* a_pnSpec);

		BOOL StartPointListMeasurement(DWORD a_nSegmentCount, BrukerSegment* a_poSegment, DWORD m_nMilliseconds);

		BOOL StartFeatureListMeasurement(DWORD a_nFeatureCount, BrukerFeature* a_poFeature, WORD* a_pwdPixelTimes);

		BOOL CheckDetectorState(long a_nDetectorId, BOOL& a_bState, long& a_nMaxEnergy);

		/// set SPU (used for send command to multi-detector system) value
		BOOL SetSPU(void);

		// open client
		BOOL OpenClient(void);
		BOOL OpenClient(char*, char*, char*, BYTE, BYTE, DWORD*);

	

		// query server
		BOOL QueryServers(void);

		//// instance termination
		//void FinishedInstance(void);					// this is to delete it

		CString GetQuantificationSpectrumPathName();


		//Get Load Dll Name
		BOOL GetLoadDllName(CString& csDllName);

		// spectrum header
		RTSpectrumHeaderRec* m_pSpectrumHead;

		// 
		std::shared_ptr<OTSBrukerClientDll> m_pBrukerDllHandle;

		std::shared_ptr<char> m_psServerName;				// server name
		BOOL m_bInit;										// initialize flag
		BOOL m_bSEMInst;									// SEM instance
		BOOL m_bScannerInst;								// scanner instance
		BOOL m_bXRAyControllerInst;							// x ray instance
		std::shared_ptr<char> m_psRTSpectrumBuffer;			// this is used to get spectrum from detector
		DWORD m_nClientID;
		BOOL m_bSEMExternal;
		int m_nSPU;
		BOOL m_bConnected;

		BOOL m_bDoQuantification;                           // if do x-ray quantification

		BOOL m_bShowQuantificationSpectrum;		

	};

}