fix(candidate): fix COM pointer reference

rime · Aug 28, 2018 · 63d6d9a · 63d6d9a
1 parent 91cbd2c
commit 63d6d9a
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/WeaselTSF/CandidateList.cpp b/WeaselTSF/CandidateList.cpp
@@ -15,6 +15,7 @@ CandidateList::CandidateList(WeaselTSF * pTextService)
 	_cRef = 1;
 	_tsf = pTextService;
 	_tsf->AddRef();
+
 }
 
 CandidateList::~CandidateList()
@@ -303,8 +304,9 @@ HWND CandidateList::_GetActiveWnd()
 void CandidateList::_StartUI()
 {
 	BOOL pbShow = TRUE;
-	ComPtr<ITfUIElementMgr> emgr;
-	_tsf->_pThreadMgr->QueryInterface(emgr.GetAddressOf());
+	ComPtr<ITfThreadMgr> pThreadMgr;
+	pThreadMgr.Copy(_tsf->_pThreadMgr);
+	ComPtr<ITfUIElementMgr> emgr = pThreadMgr.As<ITfUIElementMgr>();
 
 	if (emgr) {
 		if (!_ui->IsShown())
@@ -317,8 +319,9 @@ void CandidateList::_StartUI()
 
 void CandidateList::_EndUI()
 {
-	ComPtr<ITfUIElementMgr> emgr;
-	_tsf->_pThreadMgr->QueryInterface(emgr.GetAddressOf());
+	ComPtr<ITfThreadMgr> pThreadMgr;
+	pThreadMgr.Copy(_tsf->_pThreadMgr);
+	ComPtr<ITfUIElementMgr> emgr = pThreadMgr.As<ITfUIElementMgr>();
 	if (emgr)
 		emgr->EndUIElement(uiid);
 	if (_ui->IsShown())