[MSCTF] Sync with Wine Staging 1.7.55. CORE-10536
[reactos.git] / reactos / dll / win32 / msctf / threadmgr.c
index 9993c9c..1aceda3 100644 (file)
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  */
 
-#define WIN32_NO_STATUS
-#define _INC_WINDOWS
-#define COM_NO_WINDOWS_H
-
-#include <config.h>
-
-//#include <stdarg.h>
-
-#define COBJMACROS
-
-#include <wine/debug.h>
-//#include "windef.h"
-#include <winbase.h>
-//#include "winreg.h"
-//#include "winuser.h"
-//#include "shlwapi.h"
-//#include "winerror.h"
-#include <objbase.h>
-#include <olectl.h>
-
-//#include "wine/unicode.h"
-#include <wine/list.h>
-
-#include <msctf.h>
 #include "msctf_internal.h"
 
-WINE_DEFAULT_DEBUG_CHANNEL(msctf);
-
 typedef struct tagThreadMgrSink {
     struct list         entry;
     union {
@@ -83,22 +57,22 @@ typedef struct tagAssociatedWindow
 } AssociatedWindow;
 
 typedef struct tagACLMulti {
-    const ITfThreadMgrVtbl *ThreadMgrVtbl;
-    const ITfSourceVtbl *SourceVtbl;
-    const ITfKeystrokeMgrVtbl *KeystrokeMgrVtbl;
-    const ITfMessagePumpVtbl *MessagePumpVtbl;
-    const ITfClientIdVtbl *ClientIdVtbl;
+    ITfThreadMgrEx ITfThreadMgrEx_iface;
+    ITfSource ITfSource_iface;
+    ITfKeystrokeMgr ITfKeystrokeMgr_iface;
+    ITfMessagePump ITfMessagePump_iface;
+    ITfClientId ITfClientId_iface;
     /* const ITfThreadMgrExVtbl *ThreadMgrExVtbl; */
     /* const ITfConfigureSystemKeystrokeFeedVtbl *ConfigureSystemKeystrokeFeedVtbl; */
     /* const ITfLangBarItemMgrVtbl *LangBarItemMgrVtbl; */
-    /* const ITfUIElementMgrVtbl *UIElementMgrVtbl; */
-    const ITfSourceSingleVtbl *SourceSingleVtbl;
+    ITfUIElementMgr ITfUIElementMgr_iface;
+    ITfSourceSingle ITfSourceSingle_iface;
     LONG refCount;
 
     /* Aggregation */
     ITfCompartmentMgr  *CompartmentMgr;
 
-    const ITfThreadMgrEventSinkVtbl *ThreadMgrEventSinkVtbl; /* internal */
+    ITfThreadMgrEventSink ITfThreadMgrEventSink_iface; /* internal */
 
     ITfDocumentMgr *focus;
     LONG activationCount;
@@ -122,7 +96,7 @@ typedef struct tagACLMulti {
 } ThreadMgr;
 
 typedef struct tagEnumTfDocumentMgr {
-    const IEnumTfDocumentMgrsVtbl *Vtbl;
+    IEnumTfDocumentMgrs IEnumTfDocumentMgrs_iface;
     LONG refCount;
 
     struct list *index;
@@ -131,35 +105,49 @@ typedef struct tagEnumTfDocumentMgr {
 
 static HRESULT EnumTfDocumentMgr_Constructor(struct list* head, IEnumTfDocumentMgrs **ppOut);
 
-static inline ThreadMgr *impl_from_ITfSourceVtbl(ITfSource *iface)
+static inline ThreadMgr *impl_from_ITfThreadMgrEx(ITfThreadMgrEx *iface)
 {
-    return (ThreadMgr *)((char *)iface - FIELD_OFFSET(ThreadMgr,SourceVtbl));
+    return CONTAINING_RECORD(iface, ThreadMgr, ITfThreadMgrEx_iface);
 }
 
-static inline ThreadMgr *impl_from_ITfKeystrokeMgrVtbl(ITfKeystrokeMgr *iface)
+static inline ThreadMgr *impl_from_ITfSource(ITfSource *iface)
 {
-    return (ThreadMgr *)((char *)iface - FIELD_OFFSET(ThreadMgr,KeystrokeMgrVtbl));
+    return CONTAINING_RECORD(iface, ThreadMgr, ITfSource_iface);
 }
 
-static inline ThreadMgr *impl_from_ITfMessagePumpVtbl(ITfMessagePump *iface)
+static inline ThreadMgr *impl_from_ITfKeystrokeMgr(ITfKeystrokeMgr *iface)
 {
-    return (ThreadMgr *)((char *)iface - FIELD_OFFSET(ThreadMgr,MessagePumpVtbl));
+    return CONTAINING_RECORD(iface, ThreadMgr, ITfKeystrokeMgr_iface);
 }
 
-static inline ThreadMgr *impl_from_ITfClientIdVtbl(ITfClientId *iface)
+static inline ThreadMgr *impl_from_ITfMessagePump(ITfMessagePump *iface)
 {
-    return (ThreadMgr *)((char *)iface - FIELD_OFFSET(ThreadMgr,ClientIdVtbl));
+    return CONTAINING_RECORD(iface, ThreadMgr, ITfMessagePump_iface);
+}
+
+static inline ThreadMgr *impl_from_ITfClientId(ITfClientId *iface)
+{
+    return CONTAINING_RECORD(iface, ThreadMgr, ITfClientId_iface);
 }
 
 static inline ThreadMgr *impl_from_ITfThreadMgrEventSink(ITfThreadMgrEventSink *iface)
 {
-    return (ThreadMgr *)((char *)iface - FIELD_OFFSET(ThreadMgr,ThreadMgrEventSinkVtbl));
+    return CONTAINING_RECORD(iface, ThreadMgr, ITfThreadMgrEventSink_iface);
 }
 
-static inline ThreadMgr *impl_from_ITfSourceSingleVtbl(ITfSourceSingle* iface)
+static inline ThreadMgr *impl_from_ITfUIElementMgr(ITfUIElementMgr *iface)
+{
+    return CONTAINING_RECORD(iface, ThreadMgr, ITfUIElementMgr_iface);
+}
 
+static inline ThreadMgr *impl_from_ITfSourceSingle(ITfSourceSingle *iface)
 {
-    return (ThreadMgr *)((char *)iface - FIELD_OFFSET(ThreadMgr,SourceSingleVtbl));
+    return CONTAINING_RECORD(iface, ThreadMgr, ITfSourceSingle_iface);
+}
+
+static inline EnumTfDocumentMgr *impl_from_IEnumTfDocumentMgrs(IEnumTfDocumentMgrs *iface)
+{
+    return CONTAINING_RECORD(iface, EnumTfDocumentMgr, IEnumTfDocumentMgrs_iface);
 }
 
 static void free_sink(ThreadMgrSink *sink)
@@ -247,43 +235,48 @@ static void ThreadMgr_Destructor(ThreadMgr *This)
     HeapFree(GetProcessHeap(),0,This);
 }
 
-static HRESULT WINAPI ThreadMgr_QueryInterface(ITfThreadMgr *iface, REFIID iid, LPVOID *ppvOut)
+static HRESULT WINAPI ThreadMgr_QueryInterface(ITfThreadMgrEx *iface, REFIID iid, LPVOID *ppvOut)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     *ppvOut = NULL;
 
-    if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfThreadMgr))
+    if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfThreadMgr)
+        || IsEqualIID(iid, &IID_ITfThreadMgrEx))
     {
-        *ppvOut = This;
+        *ppvOut = &This->ITfThreadMgrEx_iface;
     }
     else if (IsEqualIID(iid, &IID_ITfSource))
     {
-        *ppvOut = &This->SourceVtbl;
+        *ppvOut = &This->ITfSource_iface;
     }
     else if (IsEqualIID(iid, &IID_ITfKeystrokeMgr))
     {
-        *ppvOut = &This->KeystrokeMgrVtbl;
+        *ppvOut = &This->ITfKeystrokeMgr_iface;
     }
     else if (IsEqualIID(iid, &IID_ITfMessagePump))
     {
-        *ppvOut = &This->MessagePumpVtbl;
+        *ppvOut = &This->ITfMessagePump_iface;
     }
     else if (IsEqualIID(iid, &IID_ITfClientId))
     {
-        *ppvOut = &This->ClientIdVtbl;
+        *ppvOut = &This->ITfClientId_iface;
     }
     else if (IsEqualIID(iid, &IID_ITfCompartmentMgr))
     {
         *ppvOut = This->CompartmentMgr;
     }
+    else if (IsEqualIID(iid, &IID_ITfUIElementMgr))
+    {
+        *ppvOut = &This->ITfUIElementMgr_iface;
+    }
     else if (IsEqualIID(iid, &IID_ITfSourceSingle))
     {
-        *ppvOut = &This->SourceSingleVtbl;
+        *ppvOut = &This->ITfSourceSingle_iface;
     }
 
     if (*ppvOut)
     {
-        ITfThreadMgr_AddRef(iface);
+        ITfThreadMgrEx_AddRef(iface);
         return S_OK;
     }
 
@@ -291,15 +284,15 @@ static HRESULT WINAPI ThreadMgr_QueryInterface(ITfThreadMgr *iface, REFIID iid,
     return E_NOINTERFACE;
 }
 
-static ULONG WINAPI ThreadMgr_AddRef(ITfThreadMgr *iface)
+static ULONG WINAPI ThreadMgr_AddRef(ITfThreadMgrEx *iface)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     return InterlockedIncrement(&This->refCount);
 }
 
-static ULONG WINAPI ThreadMgr_Release(ITfThreadMgr *iface)
+static ULONG WINAPI ThreadMgr_Release(ITfThreadMgrEx *iface)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     ULONG ret;
 
     ret = InterlockedDecrement(&This->refCount);
@@ -312,31 +305,17 @@ static ULONG WINAPI ThreadMgr_Release(ITfThreadMgr *iface)
  * ITfThreadMgr functions
  *****************************************************/
 
-static HRESULT WINAPI ThreadMgr_fnActivate( ITfThreadMgr* iface, TfClientId *ptid)
+static HRESULT WINAPI ThreadMgr_Activate(ITfThreadMgrEx *iface, TfClientId *id)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
-
-    TRACE("(%p) %p\n",This, ptid);
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
 
-    if (!ptid)
-        return E_INVALIDARG;
-
-    if (!processId)
-    {
-        GUID guid;
-        CoCreateGuid(&guid);
-        ITfClientId_GetClientId((ITfClientId*)&This->ClientIdVtbl,&guid,&processId);
-    }
-
-    activate_textservices(iface);
-    This->activationCount++;
-    *ptid = processId;
-    return S_OK;
+    TRACE("(%p) %p\n", This, id);
+    return ITfThreadMgrEx_ActivateEx(iface, id, 0);
 }
 
-static HRESULT WINAPI ThreadMgr_fnDeactivate( ITfThreadMgr* iface)
+static HRESULT WINAPI ThreadMgr_Deactivate(ITfThreadMgrEx *iface)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     TRACE("(%p)\n",This);
 
     if (This->activationCount == 0)
@@ -348,7 +327,7 @@ static HRESULT WINAPI ThreadMgr_fnDeactivate( ITfThreadMgr* iface)
     {
         if (This->focus)
         {
-            ITfThreadMgrEventSink_OnSetFocus((ITfThreadMgrEventSink*)&This->ThreadMgrEventSinkVtbl, 0, This->focus);
+            ITfThreadMgrEventSink_OnSetFocus(&This->ITfThreadMgrEventSink_iface, 0, This->focus);
             ITfDocumentMgr_Release(This->focus);
             This->focus = 0;
         }
@@ -359,10 +338,9 @@ static HRESULT WINAPI ThreadMgr_fnDeactivate( ITfThreadMgr* iface)
     return S_OK;
 }
 
-static HRESULT WINAPI ThreadMgr_CreateDocumentMgr( ITfThreadMgr* iface, ITfDocumentMgr
-**ppdim)
+static HRESULT WINAPI ThreadMgr_CreateDocumentMgr(ITfThreadMgrEx *iface, ITfDocumentMgr **ppdim)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     DocumentMgrEntry *mgrentry;
     HRESULT hr;
 
@@ -371,7 +349,7 @@ static HRESULT WINAPI ThreadMgr_CreateDocumentMgr( ITfThreadMgr* iface, ITfDocum
     if (mgrentry == NULL)
         return E_OUTOFMEMORY;
 
-    hr = DocumentMgr_Constructor((ITfThreadMgrEventSink*)&This->ThreadMgrEventSinkVtbl, ppdim);
+    hr = DocumentMgr_Constructor(&This->ITfThreadMgrEventSink_iface, ppdim);
 
     if (SUCCEEDED(hr))
     {
@@ -384,10 +362,9 @@ static HRESULT WINAPI ThreadMgr_CreateDocumentMgr( ITfThreadMgr* iface, ITfDocum
     return hr;
 }
 
-static HRESULT WINAPI ThreadMgr_EnumDocumentMgrs( ITfThreadMgr* iface, IEnumTfDocumentMgrs
-**ppEnum)
+static HRESULT WINAPI ThreadMgr_EnumDocumentMgrs(ITfThreadMgrEx *iface, IEnumTfDocumentMgrs **ppEnum)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     TRACE("(%p) %p\n",This,ppEnum);
 
     if (!ppEnum)
@@ -396,10 +373,9 @@ static HRESULT WINAPI ThreadMgr_EnumDocumentMgrs( ITfThreadMgr* iface, IEnumTfDo
     return EnumTfDocumentMgr_Constructor(&This->CreatedDocumentMgrs, ppEnum);
 }
 
-static HRESULT WINAPI ThreadMgr_GetFocus( ITfThreadMgr* iface, ITfDocumentMgr
-**ppdimFocus)
+static HRESULT WINAPI ThreadMgr_GetFocus(ITfThreadMgrEx *iface, ITfDocumentMgr **ppdimFocus)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     TRACE("(%p)\n",This);
 
     if (!ppdimFocus)
@@ -417,10 +393,10 @@ static HRESULT WINAPI ThreadMgr_GetFocus( ITfThreadMgr* iface, ITfDocumentMgr
     return S_OK;
 }
 
-static HRESULT WINAPI ThreadMgr_SetFocus( ITfThreadMgr* iface, ITfDocumentMgr *pdimFocus)
+static HRESULT WINAPI ThreadMgr_SetFocus(ITfThreadMgrEx *iface, ITfDocumentMgr *pdimFocus)
 {
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     ITfDocumentMgr *check;
-    ThreadMgr *This = (ThreadMgr *)iface;
 
     TRACE("(%p) %p\n",This,pdimFocus);
 
@@ -429,7 +405,7 @@ static HRESULT WINAPI ThreadMgr_SetFocus( ITfThreadMgr* iface, ITfDocumentMgr *p
     else if (FAILED(ITfDocumentMgr_QueryInterface(pdimFocus,&IID_ITfDocumentMgr,(LPVOID*) &check)))
         return E_INVALIDARG;
 
-    ITfThreadMgrEventSink_OnSetFocus((ITfThreadMgrEventSink*)&This->ThreadMgrEventSinkVtbl, check, This->focus);
+    ITfThreadMgrEventSink_OnSetFocus(&This->ITfThreadMgrEventSink_iface, check, This->focus);
 
     if (This->focus)
         ITfDocumentMgr_Release(This->focus);
@@ -465,7 +441,7 @@ static LRESULT CALLBACK ThreadFocusHookProc(int nCode, WPARAM wParam, LPARAM lPa
             {
                 TRACE("Triggering Associated window focus\n");
                 if (This->focus != wnd->docmgr)
-                    ThreadMgr_SetFocus((ITfThreadMgr*)This, wnd->docmgr);
+                    ThreadMgr_SetFocus(&This->ITfThreadMgrEx_iface, wnd->docmgr);
                 break;
             }
         }
@@ -490,11 +466,11 @@ static HRESULT SetupWindowsHook(ThreadMgr *This)
     return S_FALSE;
 }
 
-static HRESULT WINAPI ThreadMgr_AssociateFocus( ITfThreadMgr* iface, HWND hwnd,
+static HRESULT WINAPI ThreadMgr_AssociateFocus(ITfThreadMgrEx *iface, HWND hwnd,
 ITfDocumentMgr *pdimNew, ITfDocumentMgr **ppdimPrev)
 {
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     struct list *cursor, *cursor2;
-    ThreadMgr *This = (ThreadMgr *)iface;
     AssociatedWindow *wnd;
 
     TRACE("(%p) %p %p %p\n",This,hwnd,pdimNew,ppdimPrev);
@@ -532,36 +508,37 @@ ITfDocumentMgr *pdimNew, ITfDocumentMgr **ppdimPrev)
     return S_OK;
 }
 
-static HRESULT WINAPI ThreadMgr_IsThreadFocus( ITfThreadMgr* iface, BOOL *pfThreadFocus)
+static HRESULT WINAPI ThreadMgr_IsThreadFocus(ITfThreadMgrEx *iface, BOOL *pfThreadFocus)
 {
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     HWND focus;
-    ThreadMgr *This = (ThreadMgr *)iface;
+
     TRACE("(%p) %p\n",This,pfThreadFocus);
     focus = GetFocus();
     *pfThreadFocus = (focus == NULL);
     return S_OK;
 }
 
-static HRESULT WINAPI ThreadMgr_GetFunctionProvider( ITfThreadMgr* iface, REFCLSID clsid,
+static HRESULT WINAPI ThreadMgr_GetFunctionProvider(ITfThreadMgrEx *iface, REFCLSID clsid,
 ITfFunctionProvider **ppFuncProv)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
 
-static HRESULT WINAPI ThreadMgr_EnumFunctionProviders( ITfThreadMgr* iface,
+static HRESULT WINAPI ThreadMgr_EnumFunctionProviders(ITfThreadMgrEx *iface,
 IEnumTfFunctionProviders **ppEnum)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
 
-static HRESULT WINAPI ThreadMgr_GetGlobalCompartment( ITfThreadMgr* iface,
+static HRESULT WINAPI ThreadMgr_GetGlobalCompartment(ITfThreadMgrEx *iface,
 ITfCompartmentMgr **ppCompMgr)
 {
-    ThreadMgr *This = (ThreadMgr *)iface;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
     HRESULT hr;
     TRACE("(%p) %p\n",This, ppCompMgr);
 
@@ -580,14 +557,46 @@ ITfCompartmentMgr **ppCompMgr)
     return S_OK;
 }
 
-static const ITfThreadMgrVtbl ThreadMgr_ThreadMgrVtbl =
+static HRESULT WINAPI ThreadMgr_ActivateEx(ITfThreadMgrEx *iface, TfClientId *id, DWORD flags)
+{
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
+
+    TRACE("(%p) %p, %#x\n", This, id, flags);
+
+    if (!id)
+        return E_INVALIDARG;
+
+    if (flags)
+        FIXME("Unimplemented flags %#x\n", flags);
+
+    if (!processId)
+    {
+        GUID guid;
+        CoCreateGuid(&guid);
+        ITfClientId_GetClientId(&This->ITfClientId_iface, &guid, &processId);
+    }
+
+    activate_textservices((ITfThreadMgr *)iface);
+    This->activationCount++;
+    *id = processId;
+    return S_OK;
+}
+
+static HRESULT WINAPI ThreadMgr_GetActiveFlags(ITfThreadMgrEx *iface, DWORD *flags)
+{
+    ThreadMgr *This = impl_from_ITfThreadMgrEx(iface);
+
+    FIXME("STUB:(%p)\n", This);
+    return E_NOTIMPL;
+}
+
+static const ITfThreadMgrExVtbl ThreadMgrExVtbl =
 {
     ThreadMgr_QueryInterface,
     ThreadMgr_AddRef,
     ThreadMgr_Release,
-
-    ThreadMgr_fnActivate,
-    ThreadMgr_fnDeactivate,
+    ThreadMgr_Activate,
+    ThreadMgr_Deactivate,
     ThreadMgr_CreateDocumentMgr,
     ThreadMgr_EnumDocumentMgrs,
     ThreadMgr_GetFocus,
@@ -596,26 +605,28 @@ static const ITfThreadMgrVtbl ThreadMgr_ThreadMgrVtbl =
     ThreadMgr_IsThreadFocus,
     ThreadMgr_GetFunctionProvider,
     ThreadMgr_EnumFunctionProviders,
-    ThreadMgr_GetGlobalCompartment
-};
+    ThreadMgr_GetGlobalCompartment,
 
+    ThreadMgr_ActivateEx,
+    ThreadMgr_GetActiveFlags
+};
 
 static HRESULT WINAPI Source_QueryInterface(ITfSource *iface, REFIID iid, LPVOID *ppvOut)
 {
-    ThreadMgr *This = impl_from_ITfSourceVtbl(iface);
-    return ThreadMgr_QueryInterface((ITfThreadMgr *)This, iid, *ppvOut);
+    ThreadMgr *This = impl_from_ITfSource(iface);
+    return ITfThreadMgrEx_QueryInterface(&This->ITfThreadMgrEx_iface, iid, ppvOut);
 }
 
 static ULONG WINAPI Source_AddRef(ITfSource *iface)
 {
-    ThreadMgr *This = impl_from_ITfSourceVtbl(iface);
-    return ThreadMgr_AddRef((ITfThreadMgr*)This);
+    ThreadMgr *This = impl_from_ITfSource(iface);
+    return ITfThreadMgrEx_AddRef(&This->ITfThreadMgrEx_iface);
 }
 
 static ULONG WINAPI Source_Release(ITfSource *iface)
 {
-    ThreadMgr *This = impl_from_ITfSourceVtbl(iface);
-    return ThreadMgr_Release((ITfThreadMgr *)This);
+    ThreadMgr *This = impl_from_ITfSource(iface);
+    return ITfThreadMgrEx_Release(&This->ITfThreadMgrEx_iface);
 }
 
 /*****************************************************
@@ -624,8 +635,8 @@ static ULONG WINAPI Source_Release(ITfSource *iface)
 static HRESULT WINAPI ThreadMgrSource_AdviseSink(ITfSource *iface,
         REFIID riid, IUnknown *punk, DWORD *pdwCookie)
 {
+    ThreadMgr *This = impl_from_ITfSource(iface);
     ThreadMgrSink *tms;
-    ThreadMgr *This = impl_from_ITfSourceVtbl(iface);
 
     TRACE("(%p) %s %p %p\n",This,debugstr_guid(riid),punk,pdwCookie);
 
@@ -658,8 +669,8 @@ static HRESULT WINAPI ThreadMgrSource_AdviseSink(ITfSource *iface,
 
 static HRESULT WINAPI ThreadMgrSource_UnadviseSink(ITfSource *iface, DWORD pdwCookie)
 {
+    ThreadMgr *This = impl_from_ITfSource(iface);
     ThreadMgrSink *sink;
-    ThreadMgr *This = impl_from_ITfSourceVtbl(iface);
 
     TRACE("(%p) %x\n",This,pdwCookie);
 
@@ -676,12 +687,11 @@ static HRESULT WINAPI ThreadMgrSource_UnadviseSink(ITfSource *iface, DWORD pdwCo
     return S_OK;
 }
 
-static const ITfSourceVtbl ThreadMgr_SourceVtbl =
+static const ITfSourceVtbl ThreadMgrSourceVtbl =
 {
     Source_QueryInterface,
     Source_AddRef,
     Source_Release,
-
     ThreadMgrSource_AdviseSink,
     ThreadMgrSource_UnadviseSink,
 };
@@ -692,26 +702,26 @@ static const ITfSourceVtbl ThreadMgr_SourceVtbl =
 
 static HRESULT WINAPI KeystrokeMgr_QueryInterface(ITfKeystrokeMgr *iface, REFIID iid, LPVOID *ppvOut)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
-    return ThreadMgr_QueryInterface((ITfThreadMgr *)This, iid, *ppvOut);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
+    return ITfThreadMgrEx_QueryInterface(&This->ITfThreadMgrEx_iface, iid, ppvOut);
 }
 
 static ULONG WINAPI KeystrokeMgr_AddRef(ITfKeystrokeMgr *iface)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
-    return ThreadMgr_AddRef((ITfThreadMgr*)This);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
+    return ITfThreadMgrEx_AddRef(&This->ITfThreadMgrEx_iface);
 }
 
 static ULONG WINAPI KeystrokeMgr_Release(ITfKeystrokeMgr *iface)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
-    return ThreadMgr_Release((ITfThreadMgr *)This);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
+    return ITfThreadMgrEx_Release(&This->ITfThreadMgrEx_iface);
 }
 
 static HRESULT WINAPI KeystrokeMgr_AdviseKeyEventSink(ITfKeystrokeMgr *iface,
         TfClientId tid, ITfKeyEventSink *pSink, BOOL fForeground)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     CLSID textservice;
     ITfKeyEventSink *check = NULL;
 
@@ -751,7 +761,7 @@ static HRESULT WINAPI KeystrokeMgr_AdviseKeyEventSink(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_UnadviseKeyEventSink(ITfKeystrokeMgr *iface,
         TfClientId tid)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     CLSID textservice;
     ITfKeyEventSink *check = NULL;
     TRACE("(%p) %x\n",This,tid);
@@ -783,7 +793,7 @@ static HRESULT WINAPI KeystrokeMgr_UnadviseKeyEventSink(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_GetForeground(ITfKeystrokeMgr *iface,
         CLSID *pclsid)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     TRACE("(%p) %p\n",This,pclsid);
     if (!pclsid)
         return E_INVALIDARG;
@@ -798,7 +808,7 @@ static HRESULT WINAPI KeystrokeMgr_GetForeground(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_TestKeyDown(ITfKeystrokeMgr *iface,
         WPARAM wParam, LPARAM lParam, BOOL *pfEaten)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
@@ -806,7 +816,7 @@ static HRESULT WINAPI KeystrokeMgr_TestKeyDown(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_TestKeyUp(ITfKeystrokeMgr *iface,
         WPARAM wParam, LPARAM lParam, BOOL *pfEaten)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
@@ -814,7 +824,7 @@ static HRESULT WINAPI KeystrokeMgr_TestKeyUp(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_KeyDown(ITfKeystrokeMgr *iface,
         WPARAM wParam, LPARAM lParam, BOOL *pfEaten)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
@@ -822,7 +832,7 @@ static HRESULT WINAPI KeystrokeMgr_KeyDown(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_KeyUp(ITfKeystrokeMgr *iface,
         WPARAM wParam, LPARAM lParam, BOOL *pfEaten)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
@@ -830,7 +840,7 @@ static HRESULT WINAPI KeystrokeMgr_KeyUp(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_GetPreservedKey(ITfKeystrokeMgr *iface,
         ITfContext *pic, const TF_PRESERVEDKEY *pprekey, GUID *pguid)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
@@ -838,7 +848,7 @@ static HRESULT WINAPI KeystrokeMgr_GetPreservedKey(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_IsPreservedKey(ITfKeystrokeMgr *iface,
         REFGUID rguid, const TF_PRESERVEDKEY *pprekey, BOOL *pfRegistered)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     struct list *cursor;
 
     TRACE("(%p) %s (%x %x) %p\n",This,debugstr_guid(rguid), (pprekey)?pprekey->uVKey:0, (pprekey)?pprekey->uModifiers:0, pfRegistered);
@@ -864,7 +874,7 @@ static HRESULT WINAPI KeystrokeMgr_PreserveKey(ITfKeystrokeMgr *iface,
         TfClientId tid, REFGUID rguid, const TF_PRESERVEDKEY *prekey,
         const WCHAR *pchDesc, ULONG cchDesc)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     struct list *cursor;
     PreservedKey *newkey;
 
@@ -907,7 +917,7 @@ static HRESULT WINAPI KeystrokeMgr_PreserveKey(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_UnpreserveKey(ITfKeystrokeMgr *iface,
         REFGUID rguid, const TF_PRESERVEDKEY *pprekey)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     PreservedKey* key = NULL;
     struct list *cursor;
     TRACE("(%p) %s (%x %x)\n",This,debugstr_guid(rguid),(pprekey)?pprekey->uVKey:0, (pprekey)?pprekey->uModifiers:0);
@@ -936,7 +946,7 @@ static HRESULT WINAPI KeystrokeMgr_UnpreserveKey(ITfKeystrokeMgr *iface,
 static HRESULT WINAPI KeystrokeMgr_SetPreservedKeyDescription(ITfKeystrokeMgr *iface,
         REFGUID rguid, const WCHAR *pchDesc, ULONG cchDesc)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
@@ -944,7 +954,7 @@ static HRESULT WINAPI KeystrokeMgr_SetPreservedKeyDescription(ITfKeystrokeMgr *i
 static HRESULT WINAPI KeystrokeMgr_GetPreservedKeyDescription(ITfKeystrokeMgr *iface,
         REFGUID rguid, BSTR *pbstrDesc)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
@@ -952,17 +962,16 @@ static HRESULT WINAPI KeystrokeMgr_GetPreservedKeyDescription(ITfKeystrokeMgr *i
 static HRESULT WINAPI KeystrokeMgr_SimulatePreservedKey(ITfKeystrokeMgr *iface,
         ITfContext *pic, REFGUID rguid, BOOL *pfEaten)
 {
-    ThreadMgr *This = impl_from_ITfKeystrokeMgrVtbl(iface);
+    ThreadMgr *This = impl_from_ITfKeystrokeMgr(iface);
     FIXME("STUB:(%p)\n",This);
     return E_NOTIMPL;
 }
 
-static const ITfKeystrokeMgrVtbl ThreadMgr_KeystrokeMgrVtbl =
+static const ITfKeystrokeMgrVtbl KeystrokeMgrVtbl =
 {
     KeystrokeMgr_QueryInterface,
     KeystrokeMgr_AddRef,
     KeystrokeMgr_Release,
-
     KeystrokeMgr_AdviseKeyEventSink,
     KeystrokeMgr_UnadviseKeyEventSink,
     KeystrokeMgr_GetForeground,
@@ -985,20 +994,20 @@ static const ITfKeystrokeMgrVtbl ThreadMgr_KeystrokeMgrVtbl =
 
 static HRESULT WINAPI MessagePump_QueryInterface(ITfMessagePump *iface, REFIID iid, LPVOID *ppvOut)
 {
-    ThreadMgr *This = impl_from_ITfMessagePumpVtbl(iface);
-    return ThreadMgr_QueryInterface((ITfThreadMgr *)This, iid, *ppvOut);
+    ThreadMgr *This = impl_from_ITfMessagePump(iface);
+    return ITfThreadMgrEx_QueryInterface(&This->ITfThreadMgrEx_iface, iid, ppvOut);
 }
 
 static ULONG WINAPI MessagePump_AddRef(ITfMessagePump *iface)
 {
-    ThreadMgr *This = impl_from_ITfMessagePumpVtbl(iface);
-    return ThreadMgr_AddRef((ITfThreadMgr*)This);
+    ThreadMgr *This = impl_from_ITfMessagePump(iface);
+    return ITfThreadMgrEx_AddRef(&This->ITfThreadMgrEx_iface);
 }
 
 static ULONG WINAPI MessagePump_Release(ITfMessagePump *iface)
 {
-    ThreadMgr *This = impl_from_ITfMessagePumpVtbl(iface);
-    return ThreadMgr_Release((ITfThreadMgr *)This);
+    ThreadMgr *This = impl_from_ITfMessagePump(iface);
+    return ITfThreadMgrEx_Release(&This->ITfThreadMgrEx_iface);
 }
 
 static HRESULT WINAPI MessagePump_PeekMessageA(ITfMessagePump *iface,
@@ -1041,12 +1050,11 @@ static HRESULT WINAPI MessagePump_GetMessageW(ITfMessagePump *iface,
     return S_OK;
 }
 
-static const ITfMessagePumpVtbl ThreadMgr_MessagePumpVtbl =
+static const ITfMessagePumpVtbl MessagePumpVtbl =
 {
     MessagePump_QueryInterface,
     MessagePump_AddRef,
     MessagePump_Release,
-
     MessagePump_PeekMessageA,
     MessagePump_GetMessageA,
     MessagePump_PeekMessageW,
@@ -1059,29 +1067,29 @@ static const ITfMessagePumpVtbl ThreadMgr_MessagePumpVtbl =
 
 static HRESULT WINAPI ClientId_QueryInterface(ITfClientId *iface, REFIID iid, LPVOID *ppvOut)
 {
-    ThreadMgr *This = impl_from_ITfClientIdVtbl(iface);
-    return ThreadMgr_QueryInterface((ITfThreadMgr *)This, iid, *ppvOut);
+    ThreadMgr *This = impl_from_ITfClientId(iface);
+    return ITfThreadMgrEx_QueryInterface(&This->ITfThreadMgrEx_iface, iid, ppvOut);
 }
 
 static ULONG WINAPI ClientId_AddRef(ITfClientId *iface)
 {
-    ThreadMgr *This = impl_from_ITfClientIdVtbl(iface);
-    return ThreadMgr_AddRef((ITfThreadMgr*)This);
+    ThreadMgr *This = impl_from_ITfClientId(iface);
+    return ITfThreadMgrEx_AddRef(&This->ITfThreadMgrEx_iface);
 }
 
 static ULONG WINAPI ClientId_Release(ITfClientId *iface)
 {
-    ThreadMgr *This = impl_from_ITfClientIdVtbl(iface);
-    return ThreadMgr_Release((ITfThreadMgr *)This);
+    ThreadMgr *This = impl_from_ITfClientId(iface);
+    return ITfThreadMgrEx_Release(&This->ITfThreadMgrEx_iface);
 }
 
 static HRESULT WINAPI ClientId_GetClientId(ITfClientId *iface,
     REFCLSID rclsid, TfClientId *ptid)
 
 {
+    ThreadMgr *This = impl_from_ITfClientId(iface);
     HRESULT hr;
     ITfCategoryMgr *catmgr;
-    ThreadMgr *This = impl_from_ITfClientIdVtbl(iface);
 
     TRACE("(%p) %s\n",This,debugstr_guid(rclsid));
 
@@ -1092,12 +1100,11 @@ static HRESULT WINAPI ClientId_GetClientId(ITfClientId *iface,
     return hr;
 }
 
-static const ITfClientIdVtbl ThreadMgr_ClientIdVtbl =
+static const ITfClientIdVtbl ClientIdVtbl =
 {
     ClientId_QueryInterface,
     ClientId_AddRef,
     ClientId_Release,
-
     ClientId_GetClientId
 };
 
@@ -1107,19 +1114,19 @@ static const ITfClientIdVtbl ThreadMgr_ClientIdVtbl =
 static HRESULT WINAPI ThreadMgrEventSink_QueryInterface(ITfThreadMgrEventSink *iface, REFIID iid, LPVOID *ppvOut)
 {
     ThreadMgr *This = impl_from_ITfThreadMgrEventSink(iface);
-    return ThreadMgr_QueryInterface((ITfThreadMgr *)This, iid, *ppvOut);
+    return ITfThreadMgrEx_QueryInterface(&This->ITfThreadMgrEx_iface, iid, ppvOut);
 }
 
 static ULONG WINAPI ThreadMgrEventSink_AddRef(ITfThreadMgrEventSink *iface)
 {
     ThreadMgr *This = impl_from_ITfThreadMgrEventSink(iface);
-    return ThreadMgr_AddRef((ITfThreadMgr*)This);
+    return ITfThreadMgrEx_AddRef(&This->ITfThreadMgrEx_iface);
 }
 
 static ULONG WINAPI ThreadMgrEventSink_Release(ITfThreadMgrEventSink *iface)
 {
     ThreadMgr *This = impl_from_ITfThreadMgrEventSink(iface);
-    return ThreadMgr_Release((ITfThreadMgr *)This);
+    return ITfThreadMgrEx_Release(&This->ITfThreadMgrEx_iface);
 }
 
 
@@ -1209,12 +1216,11 @@ static HRESULT WINAPI ThreadMgrEventSink_OnPopContext(
     return S_OK;
 }
 
-static const ITfThreadMgrEventSinkVtbl ThreadMgr_ThreadMgrEventSinkVtbl =
+static const ITfThreadMgrEventSinkVtbl ThreadMgrEventSinkVtbl =
 {
     ThreadMgrEventSink_QueryInterface,
     ThreadMgrEventSink_AddRef,
     ThreadMgrEventSink_Release,
-
     ThreadMgrEventSink_OnInitDocumentMgr,
     ThreadMgrEventSink_OnUninitDocumentMgr,
     ThreadMgrEventSink_OnSetFocus,
@@ -1222,31 +1228,111 @@ static const ITfThreadMgrEventSinkVtbl ThreadMgr_ThreadMgrEventSinkVtbl =
     ThreadMgrEventSink_OnPopContext
 };
 
+/*****************************************************
+ * ITfUIElementMgr functions
+ *****************************************************/
+static HRESULT WINAPI UIElementMgr_QueryInterface(ITfUIElementMgr *iface, REFIID iid, void **ppvOut)
+{
+    ThreadMgr *This = impl_from_ITfUIElementMgr(iface);
+
+    return ITfThreadMgrEx_QueryInterface(&This->ITfThreadMgrEx_iface, iid, *ppvOut);
+}
+
+static ULONG WINAPI UIElementMgr_AddRef(ITfUIElementMgr *iface)
+{
+    ThreadMgr *This = impl_from_ITfUIElementMgr(iface);
+
+    return ITfThreadMgrEx_AddRef(&This->ITfThreadMgrEx_iface);
+}
+
+static ULONG WINAPI UIElementMgr_Release(ITfUIElementMgr *iface)
+{
+    ThreadMgr *This = impl_from_ITfUIElementMgr(iface);
+
+    return ITfThreadMgrEx_Release(&This->ITfThreadMgrEx_iface);
+}
+
+static HRESULT WINAPI UIElementMgr_BeginUIElement(ITfUIElementMgr *iface, ITfUIElement *element,
+                                                  BOOL *show, DWORD *id)
+{
+    ThreadMgr *This = impl_from_ITfUIElementMgr(iface);
+
+    FIXME("STUB:(%p)\n", This);
+    return E_NOTIMPL;
+}
+
+static HRESULT WINAPI UIElementMgr_UpdateUIElement(ITfUIElementMgr *iface, DWORD id)
+{
+    ThreadMgr *This = impl_from_ITfUIElementMgr(iface);
+
+    FIXME("STUB:(%p)\n", This);
+    return E_NOTIMPL;
+}
+
+static HRESULT WINAPI UIElementMgr_EndUIElement(ITfUIElementMgr *iface, DWORD id)
+{
+    ThreadMgr *This = impl_from_ITfUIElementMgr(iface);
+
+    FIXME("STUB:(%p)\n", This);
+    return E_NOTIMPL;
+}
+
+static HRESULT WINAPI UIElementMgr_GetUIElement(ITfUIElementMgr *iface, DWORD id,
+                                                ITfUIElement **element)
+{
+    ThreadMgr *This = impl_from_ITfUIElementMgr(iface);
+
+    FIXME("STUB:(%p)\n", This);
+    return E_NOTIMPL;
+}
+
+static HRESULT WINAPI UIElementMgr_EnumUIElements(ITfUIElementMgr *iface,
+                                                  IEnumTfUIElements **enum_elements)
+{
+    ThreadMgr *This = impl_from_ITfUIElementMgr(iface);
+
+    FIXME("STUB:(%p)\n", This);
+    return E_NOTIMPL;
+}
+
+static const ITfUIElementMgrVtbl ThreadMgrUIElementMgrVtbl =
+{
+    UIElementMgr_QueryInterface,
+    UIElementMgr_AddRef,
+    UIElementMgr_Release,
+
+    UIElementMgr_BeginUIElement,
+    UIElementMgr_UpdateUIElement,
+    UIElementMgr_EndUIElement,
+    UIElementMgr_GetUIElement,
+    UIElementMgr_EnumUIElements
+};
+
 /*****************************************************
  * ITfSourceSingle functions
  *****************************************************/
 static HRESULT WINAPI ThreadMgrSourceSingle_QueryInterface(ITfSourceSingle *iface, REFIID iid, LPVOID *ppvOut)
 {
-    ThreadMgr *This = impl_from_ITfSourceSingleVtbl(iface);
-    return ThreadMgr_QueryInterface((ITfThreadMgr *)This, iid, *ppvOut);
+    ThreadMgr *This = impl_from_ITfSourceSingle(iface);
+    return ITfThreadMgrEx_QueryInterface(&This->ITfThreadMgrEx_iface, iid, ppvOut);
 }
 
 static ULONG WINAPI ThreadMgrSourceSingle_AddRef(ITfSourceSingle *iface)
 {
-    ThreadMgr *This = impl_from_ITfSourceSingleVtbl(iface);
-    return ThreadMgr_AddRef((ITfThreadMgr *)This);
+    ThreadMgr *This = impl_from_ITfSourceSingle(iface);
+    return ITfThreadMgrEx_AddRef(&This->ITfThreadMgrEx_iface);
 }
 
 static ULONG WINAPI ThreadMgrSourceSingle_Release(ITfSourceSingle *iface)
 {
-    ThreadMgr *This = impl_from_ITfSourceSingleVtbl(iface);
-    return ThreadMgr_Release((ITfThreadMgr *)This);
+    ThreadMgr *This = impl_from_ITfSourceSingle(iface);
+    return ITfThreadMgrEx_Release(&This->ITfThreadMgrEx_iface);
 }
 
 static HRESULT WINAPI ThreadMgrSourceSingle_AdviseSingleSink( ITfSourceSingle *iface,
     TfClientId tid, REFIID riid, IUnknown *punk)
 {
-    ThreadMgr *This = impl_from_ITfSourceSingleVtbl(iface);
+    ThreadMgr *This = impl_from_ITfSourceSingle(iface);
     FIXME("STUB:(%p) %i %s %p\n",This, tid, debugstr_guid(riid),punk);
     return E_NOTIMPL;
 }
@@ -1254,19 +1340,18 @@ static HRESULT WINAPI ThreadMgrSourceSingle_AdviseSingleSink( ITfSourceSingle *i
 static HRESULT WINAPI ThreadMgrSourceSingle_UnadviseSingleSink( ITfSourceSingle *iface,
     TfClientId tid, REFIID riid)
 {
-    ThreadMgr *This = impl_from_ITfSourceSingleVtbl(iface);
+    ThreadMgr *This = impl_from_ITfSourceSingle(iface);
     FIXME("STUB:(%p) %i %s\n",This, tid, debugstr_guid(riid));
     return E_NOTIMPL;
 }
 
-static const ITfSourceSingleVtbl ThreadMgr_SourceSingleVtbl =
+static const ITfSourceSingleVtbl SourceSingleVtbl =
 {
     ThreadMgrSourceSingle_QueryInterface,
     ThreadMgrSourceSingle_AddRef,
     ThreadMgrSourceSingle_Release,
-
     ThreadMgrSourceSingle_AdviseSingleSink,
-    ThreadMgrSourceSingle_UnadviseSingleSink,
+    ThreadMgrSourceSingle_UnadviseSingleSink
 };
 
 HRESULT ThreadMgr_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
@@ -1279,8 +1364,8 @@ HRESULT ThreadMgr_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
     This = TlsGetValue(tlsIndex);
     if (This)
     {
-        ThreadMgr_AddRef((ITfThreadMgr*)This);
-        *ppOut = (IUnknown*)This;
+        ThreadMgr_AddRef(&This->ITfThreadMgrEx_iface);
+        *ppOut = (IUnknown*)&This->ITfThreadMgrEx_iface;
         return S_OK;
     }
 
@@ -1288,13 +1373,14 @@ HRESULT ThreadMgr_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
     if (This == NULL)
         return E_OUTOFMEMORY;
 
-    This->ThreadMgrVtbl= &ThreadMgr_ThreadMgrVtbl;
-    This->SourceVtbl = &ThreadMgr_SourceVtbl;
-    This->KeystrokeMgrVtbl= &ThreadMgr_KeystrokeMgrVtbl;
-    This->MessagePumpVtbl= &ThreadMgr_MessagePumpVtbl;
-    This->ClientIdVtbl = &ThreadMgr_ClientIdVtbl;
-    This->ThreadMgrEventSinkVtbl = &ThreadMgr_ThreadMgrEventSinkVtbl;
-    This->SourceSingleVtbl = &ThreadMgr_SourceSingleVtbl;
+    This->ITfThreadMgrEx_iface.lpVtbl = &ThreadMgrExVtbl;
+    This->ITfSource_iface.lpVtbl = &ThreadMgrSourceVtbl;
+    This->ITfKeystrokeMgr_iface.lpVtbl = &KeystrokeMgrVtbl;
+    This->ITfMessagePump_iface.lpVtbl = &MessagePumpVtbl;
+    This->ITfClientId_iface.lpVtbl = &ClientIdVtbl;
+    This->ITfThreadMgrEventSink_iface.lpVtbl = &ThreadMgrEventSinkVtbl;
+    This->ITfUIElementMgr_iface.lpVtbl = &ThreadMgrUIElementMgrVtbl;
+    This->ITfSourceSingle_iface.lpVtbl = &SourceSingleVtbl;
     This->refCount = 1;
     TlsSetValue(tlsIndex,This);
 
@@ -1312,7 +1398,7 @@ HRESULT ThreadMgr_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
     list_init(&This->ThreadMgrEventSink);
 
     TRACE("returning %p\n", This);
-    *ppOut = (IUnknown *)This;
+    *ppOut = (IUnknown *)&This->ITfThreadMgrEx_iface;
     return S_OK;
 }
 
@@ -1327,12 +1413,12 @@ static void EnumTfDocumentMgr_Destructor(EnumTfDocumentMgr *This)
 
 static HRESULT WINAPI EnumTfDocumentMgr_QueryInterface(IEnumTfDocumentMgrs *iface, REFIID iid, LPVOID *ppvOut)
 {
-    EnumTfDocumentMgr *This = (EnumTfDocumentMgr *)iface;
+    EnumTfDocumentMgr *This = impl_from_IEnumTfDocumentMgrs(iface);
     *ppvOut = NULL;
 
     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_IEnumTfDocumentMgrs))
     {
-        *ppvOut = This;
+        *ppvOut = &This->IEnumTfDocumentMgrs_iface;
     }
 
     if (*ppvOut)
@@ -1347,13 +1433,13 @@ static HRESULT WINAPI EnumTfDocumentMgr_QueryInterface(IEnumTfDocumentMgrs *ifac
 
 static ULONG WINAPI EnumTfDocumentMgr_AddRef(IEnumTfDocumentMgrs *iface)
 {
-    EnumTfDocumentMgr *This = (EnumTfDocumentMgr*)iface;
+    EnumTfDocumentMgr *This = impl_from_IEnumTfDocumentMgrs(iface);
     return InterlockedIncrement(&This->refCount);
 }
 
 static ULONG WINAPI EnumTfDocumentMgr_Release(IEnumTfDocumentMgrs *iface)
 {
-    EnumTfDocumentMgr *This = (EnumTfDocumentMgr *)iface;
+    EnumTfDocumentMgr *This = impl_from_IEnumTfDocumentMgrs(iface);
     ULONG ret;
 
     ret = InterlockedDecrement(&This->refCount);
@@ -1365,7 +1451,7 @@ static ULONG WINAPI EnumTfDocumentMgr_Release(IEnumTfDocumentMgrs *iface)
 static HRESULT WINAPI EnumTfDocumentMgr_Next(IEnumTfDocumentMgrs *iface,
     ULONG ulCount, ITfDocumentMgr **rgDocumentMgr, ULONG *pcFetched)
 {
-    EnumTfDocumentMgr *This = (EnumTfDocumentMgr *)iface;
+    EnumTfDocumentMgr *This = impl_from_IEnumTfDocumentMgrs(iface);
     ULONG fetched = 0;
 
     TRACE("(%p)\n",This);
@@ -1396,8 +1482,9 @@ static HRESULT WINAPI EnumTfDocumentMgr_Next(IEnumTfDocumentMgrs *iface,
 
 static HRESULT WINAPI EnumTfDocumentMgr_Skip( IEnumTfDocumentMgrs* iface, ULONG celt)
 {
+    EnumTfDocumentMgr *This = impl_from_IEnumTfDocumentMgrs(iface);
     ULONG i;
-    EnumTfDocumentMgr *This = (EnumTfDocumentMgr *)iface;
+
     TRACE("(%p)\n",This);
     for(i = 0; i < celt && This->index != NULL; i++)
         This->index = list_next(This->head, This->index);
@@ -1406,7 +1493,7 @@ static HRESULT WINAPI EnumTfDocumentMgr_Skip( IEnumTfDocumentMgrs* iface, ULONG
 
 static HRESULT WINAPI EnumTfDocumentMgr_Reset( IEnumTfDocumentMgrs* iface)
 {
-    EnumTfDocumentMgr *This = (EnumTfDocumentMgr *)iface;
+    EnumTfDocumentMgr *This = impl_from_IEnumTfDocumentMgrs(iface);
     TRACE("(%p)\n",This);
     This->index = list_head(This->head);
     return S_OK;
@@ -1415,7 +1502,7 @@ static HRESULT WINAPI EnumTfDocumentMgr_Reset( IEnumTfDocumentMgrs* iface)
 static HRESULT WINAPI EnumTfDocumentMgr_Clone( IEnumTfDocumentMgrs *iface,
     IEnumTfDocumentMgrs **ppenum)
 {
-    EnumTfDocumentMgr *This = (EnumTfDocumentMgr *)iface;
+    EnumTfDocumentMgr *This = impl_from_IEnumTfDocumentMgrs(iface);
     HRESULT res;
 
     TRACE("(%p)\n",This);
@@ -1425,17 +1512,17 @@ static HRESULT WINAPI EnumTfDocumentMgr_Clone( IEnumTfDocumentMgrs *iface,
     res = EnumTfDocumentMgr_Constructor(This->head, ppenum);
     if (SUCCEEDED(res))
     {
-        EnumTfDocumentMgr *new_This = (EnumTfDocumentMgr *)*ppenum;
+        EnumTfDocumentMgr *new_This = impl_from_IEnumTfDocumentMgrs(*ppenum);
         new_This->index = This->index;
     }
     return res;
 }
 
-static const IEnumTfDocumentMgrsVtbl IEnumTfDocumentMgrs_Vtbl ={
+static const IEnumTfDocumentMgrsVtbl EnumTfDocumentMgrsVtbl =
+{
     EnumTfDocumentMgr_QueryInterface,
     EnumTfDocumentMgr_AddRef,
     EnumTfDocumentMgr_Release,
-
     EnumTfDocumentMgr_Clone,
     EnumTfDocumentMgr_Next,
     EnumTfDocumentMgr_Reset,
@@ -1450,7 +1537,7 @@ static HRESULT EnumTfDocumentMgr_Constructor(struct list* head, IEnumTfDocumentM
     if (This == NULL)
         return E_OUTOFMEMORY;
 
-    This->Vtbl= &IEnumTfDocumentMgrs_Vtbl;
+    This->IEnumTfDocumentMgrs_iface.lpVtbl= &EnumTfDocumentMgrsVtbl;
     This->refCount = 1;
     This->head = head;
     This->index = list_head(This->head);
@@ -1460,9 +1547,9 @@ static HRESULT EnumTfDocumentMgr_Constructor(struct list* head, IEnumTfDocumentM
     return S_OK;
 }
 
-void ThreadMgr_OnDocumentMgrDestruction(ITfThreadMgr *tm, ITfDocumentMgr *mgr)
+void ThreadMgr_OnDocumentMgrDestruction(ITfThreadMgr *iface, ITfDocumentMgr *mgr)
 {
-    ThreadMgr *This = (ThreadMgr *)tm;
+    ThreadMgr *This = impl_from_ITfThreadMgrEx((ITfThreadMgrEx *)iface);
     struct list *cursor;
     LIST_FOR_EACH(cursor, &This->CreatedDocumentMgrs)
     {