m2dev-client-src/src/EterLib/IME.h

#pragma once

#include <imm.h>

#pragma comment(lib, "imm32.lib")

#include "DIMM.h"

#include <vector>
#include <string>

class IIMEEventSink
{
	public:
		virtual bool OnWM_CHAR( WPARAM wParam, LPARAM lParam ) = 0;
		virtual void OnUpdate() = 0;

		virtual void OnOpenCandidateList() = 0;
		virtual void OnCloseCandidateList() = 0;

		virtual void OnOpenReadingWnd() = 0;
		virtual void OnCloseReadingWnd() = 0;
};

class CIME
{
	public:
		enum
		{
			IMEREADING_MAXLEN = 128,
			IMESTR_MAXLEN = 1024,
			IMECANDIDATE_MAXLEN = 32768,
			MAX_CANDLIST = 10,
			MAX_CANDIDATE_LENGTH = 256
		};

	public:
		CIME();
		virtual ~CIME();

		bool Initialize(HWND hWnd);
		void Uninitialize(void);

		static void Clear();

		void SetMax(int iMax);
		void SetUserMax(int iMax);
		static void SetText(const char* c_szText, int len);
		int GetText(std::string & rstrText);

		// Candidate List
		int  GetCandidateCount();
		int  GetCandidatePageCount();
		int  GetCandidate(DWORD index, std::string & rstrText);
		int  GetCandidateSelection();

		// Reading Information
		int GetReading(std::string & rstrText);
		int GetReadingError();

		void SetInputMode(DWORD dwMode);
		DWORD GetInputMode();

		bool IsIMEEnabled();
		void EnableIME(bool bEnable=true);
		void DisableIME();

		void EnableCaptureInput();
		void DisableCaptureInput();
		bool IsCaptureEnabled();

		void SetNumberMode();
		void SetStringMode();
		bool __IsWritable(wchar_t key);
		void AddExceptKey(wchar_t key);
		void ClearExceptKey();

		void PasteTextFromClipBoard();
		void EnablePaste(bool bFlag);
		void PasteString(const char * str);
		static void FinalizeString(bool bSend = false);

		void SelectAll();
		void DeleteSelection();
		void CopySelectionToClipboard(HWND hWnd);
		void CutSelectionToClipboard(HWND hWnd);

		void UseDefaultIME();

		static int GetCurPos();
		static int GetCompLen();
		static int GetULBegin();
		static int GetULEnd();

		static int GetSelBegin();
		static int GetSelEnd();
		static void ClearSelection();

		static void CloseCandidateList();
		static void CloseReadingInformation();
		static void ChangeInputLanguage();
		static void ChangeInputLanguageWorker();

		LRESULT WMInputLanguage(HWND hWnd, UINT uiMsg, WPARAM wParam, LPARAM lParam);
		LRESULT WMStartComposition(HWND hWnd, UINT uiMsg, WPARAM wParam, LPARAM lParam);
		LRESULT WMComposition(HWND hWnd, UINT uiMsg, WPARAM wParam, LPARAM lParam);
		LRESULT WMEndComposition(HWND hWnd, UINT uiMsg, WPARAM wParam, LPARAM lParam);
		LRESULT WMNotify(HWND hWnd, UINT uiMsg, WPARAM wParam, LPARAM lParam);
		LRESULT WMChar(HWND hWnd, UINT uiMsg, WPARAM wParam, LPARAM lParam);

	public:
		struct SUndoState
		{
			std::wstring text;
			int curpos = 0;
			int lastpos = 0;
			int selbegin = 0;
			int selend = 0;
		};

		static void PushUndoState();
		static void Undo();
		static void Redo();
		static void ClearUndoRedo();

		static bool CanUndo();
		static bool CanRedo();

	private:
		static void RestoreState(const SUndoState& st);

	private:
		static std::vector<SUndoState> ms_undo;
		static std::vector<SUndoState> ms_redo;
		static const size_t MAX_UNDO = 64;

	protected:
		void IncCurPos();
		void DecCurPos();
		void SetCurPos(int offset);
		void DelCurPos();

	protected:
		static void CheckInputLocale();
		static void CheckToggleState();
		static void SetSupportLevel( DWORD dwImeLevel );

		void InsertString(wchar_t* szString, int iSize);

		void OnChar(wchar_t c);

		void ResultProcess(HIMC hImc);
		void CompositionProcessBuilding(HIMC hImc);
		void CompositionProcess(HIMC hImc);
		void AttributeProcess(HIMC hImc);
		void CandidateProcess(HIMC hImc);
		void ReadingProcess(HIMC hImc);

		bool IsMax(const wchar_t* wInput, int len);

		DWORD GetImeId(UINT uIndex = 0);
		bool GetReadingWindowOrientation();
		static void SetupImeApi();

		static INPUTCONTEXT* (WINAPI * _ImmLockIMC)( HIMC );
		static BOOL (WINAPI * _ImmUnlockIMC)( HIMC );
		static LPVOID (WINAPI * _ImmLockIMCC)( HIMCC );
		static BOOL (WINAPI * _ImmUnlockIMCC)( HIMCC );

		static UINT (WINAPI * _GetReadingString)( HIMC, UINT, LPWSTR, PINT, BOOL*, PUINT );
		static BOOL (WINAPI * _ShowReadingWindow)( HIMC, BOOL );

	protected:
		HIMC m_hOrgIMC;
		int m_max;
		int m_userMax;

		BOOL m_bOnlyNumberMode;

		std::vector<wchar_t> m_exceptKey;

		bool m_bEnablePaste;
		bool m_bUseDefaultIME;


	public:
		static bool ms_bInitialized;
		static bool ms_bDisableIMECompletely;
		static bool ms_bUILessMode;
		static bool ms_bImeEnabled;
		static bool ms_bCaptureInput;
		static bool ms_bChineseIME;
		static bool ms_bUseIMMCandidate;

		static HWND ms_hWnd;
		static HKL ms_hklCurrent;
		static wchar_t ms_szKeyboardLayout[KL_NAMELENGTH+1];
		static OSVERSIONINFOW ms_stOSVI;

		static HINSTANCE ms_hImm32Dll;
		static HINSTANCE ms_hCurrentImeDll;
		static DWORD ms_dwImeState;

		static DWORD ms_adwId[2];

		// IME Level
		static DWORD ms_dwIMELevel;
		static DWORD ms_dwIMELevelSaved;

		// Candidate List
		static bool ms_bCandidateList;
		static DWORD ms_dwCandidateCount;
		static bool ms_bVerticalCandidate;
		static int ms_iCandListIndexBase;
		static WCHAR ms_wszCandidate[CIME::MAX_CANDLIST][MAX_CANDIDATE_LENGTH];
		static DWORD ms_dwCandidateSelection;
		static DWORD ms_dwCandidatePageSize;

		// Reading Information
		static bool ms_bReadingInformation;
		static int ms_iReadingError;
		static bool ms_bHorizontalReading;
		static std::vector<wchar_t> ms_wstrReading;

		// Indicator
		static 	wchar_t* ms_wszCurrentIndicator;

		static IIMEEventSink* ms_pEvent;

		wchar_t m_wszComposition[IMESTR_MAXLEN];
		static wchar_t m_wText[IMESTR_MAXLEN];

		static int ms_compLen;
		static int ms_curpos;
		static int ms_lastpos;
		static int ms_ulbegin;
		static int ms_ulend;

		static int ms_selbegin;
		static int ms_selend;
		static int ms_compCaret;
};