1*b1cdbd2cSJim Jagielski /************************************************************** 2*b1cdbd2cSJim Jagielski * 3*b1cdbd2cSJim Jagielski * Licensed to the Apache Software Foundation (ASF) under one 4*b1cdbd2cSJim Jagielski * or more contributor license agreements. See the NOTICE file 5*b1cdbd2cSJim Jagielski * distributed with this work for additional information 6*b1cdbd2cSJim Jagielski * regarding copyright ownership. The ASF licenses this file 7*b1cdbd2cSJim Jagielski * to you under the Apache License, Version 2.0 (the 8*b1cdbd2cSJim Jagielski * "License"); you may not use this file except in compliance 9*b1cdbd2cSJim Jagielski * with the License. You may obtain a copy of the License at 10*b1cdbd2cSJim Jagielski * 11*b1cdbd2cSJim Jagielski * http://www.apache.org/licenses/LICENSE-2.0 12*b1cdbd2cSJim Jagielski * 13*b1cdbd2cSJim Jagielski * Unless required by applicable law or agreed to in writing, 14*b1cdbd2cSJim Jagielski * software distributed under the License is distributed on an 15*b1cdbd2cSJim Jagielski * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 16*b1cdbd2cSJim Jagielski * KIND, either express or implied. See the License for the 17*b1cdbd2cSJim Jagielski * specific language governing permissions and limitations 18*b1cdbd2cSJim Jagielski * under the License. 19*b1cdbd2cSJim Jagielski * 20*b1cdbd2cSJim Jagielski *************************************************************/ 21*b1cdbd2cSJim Jagielski 22*b1cdbd2cSJim Jagielski 23*b1cdbd2cSJim Jagielski 24*b1cdbd2cSJim Jagielski #ifndef COMPTR_HXX 25*b1cdbd2cSJim Jagielski #define COMPTR_HXX 26*b1cdbd2cSJim Jagielski 27*b1cdbd2cSJim Jagielski #include <sal/types.h> 28*b1cdbd2cSJim Jagielski #include <osl/diagnose.h> 29*b1cdbd2cSJim Jagielski #include <shobjidl.h> 30*b1cdbd2cSJim Jagielski 31*b1cdbd2cSJim Jagielski template< class T_INTERFACE , 32*b1cdbd2cSJim Jagielski REFIID P_IID = IID_NULL , 33*b1cdbd2cSJim Jagielski REFCLSID P_CLSID = CLSID_NULL > 34*b1cdbd2cSJim Jagielski class ComPtr 35*b1cdbd2cSJim Jagielski { 36*b1cdbd2cSJim Jagielski public: 37*b1cdbd2cSJim Jagielski 38*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- 39*b1cdbd2cSJim Jagielski /** initialize com ptr with null. 40*b1cdbd2cSJim Jagielski */ ComPtr()41*b1cdbd2cSJim Jagielski ComPtr() 42*b1cdbd2cSJim Jagielski { 43*b1cdbd2cSJim Jagielski m_pInterface = NULL; 44*b1cdbd2cSJim Jagielski } 45*b1cdbd2cSJim Jagielski 46*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- 47*b1cdbd2cSJim Jagielski /** initialize com ptr with given interface. 48*b1cdbd2cSJim Jagielski */ ComPtr(T_INTERFACE * pInterface)49*b1cdbd2cSJim Jagielski ComPtr(T_INTERFACE* pInterface) 50*b1cdbd2cSJim Jagielski { 51*b1cdbd2cSJim Jagielski m_pInterface = pInterface; 52*b1cdbd2cSJim Jagielski if (m_pInterface) 53*b1cdbd2cSJim Jagielski m_pInterface->AddRef(); 54*b1cdbd2cSJim Jagielski } 55*b1cdbd2cSJim Jagielski 56*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- 57*b1cdbd2cSJim Jagielski /** copy ctor. 58*b1cdbd2cSJim Jagielski */ ComPtr(const ComPtr<T_INTERFACE,P_IID,P_CLSID> & aCopy)59*b1cdbd2cSJim Jagielski ComPtr(const ComPtr< T_INTERFACE, P_IID, P_CLSID >& aCopy) 60*b1cdbd2cSJim Jagielski { 61*b1cdbd2cSJim Jagielski m_pInterface = aCopy.m_pInterface; 62*b1cdbd2cSJim Jagielski if (m_pInterface) 63*b1cdbd2cSJim Jagielski m_pInterface->AddRef(); 64*b1cdbd2cSJim Jagielski } 65*b1cdbd2cSJim Jagielski 66*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- 67*b1cdbd2cSJim Jagielski /** initialize object by quering external object for the right interface. 68*b1cdbd2cSJim Jagielski */ ComPtr(IUnknown * pIUnknown)69*b1cdbd2cSJim Jagielski ComPtr(IUnknown* pIUnknown) 70*b1cdbd2cSJim Jagielski { 71*b1cdbd2cSJim Jagielski if (pIUnknown) 72*b1cdbd2cSJim Jagielski pIUnknown->QueryInterface(P_IID, (void**)&m_pInterface); 73*b1cdbd2cSJim Jagielski } 74*b1cdbd2cSJim Jagielski 75*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- 76*b1cdbd2cSJim Jagielski /** deinitialize com object right. 77*b1cdbd2cSJim Jagielski */ ~ComPtr()78*b1cdbd2cSJim Jagielski ~ComPtr() 79*b1cdbd2cSJim Jagielski { 80*b1cdbd2cSJim Jagielski release(); 81*b1cdbd2cSJim Jagielski } 82*b1cdbd2cSJim Jagielski 83*b1cdbd2cSJim Jagielski public: 84*b1cdbd2cSJim Jagielski 85*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- create()86*b1cdbd2cSJim Jagielski HRESULT create() 87*b1cdbd2cSJim Jagielski { 88*b1cdbd2cSJim Jagielski return CoCreateInstance(P_CLSID, NULL, CLSCTX_ALL, P_IID, (void**)&m_pInterface); 89*b1cdbd2cSJim Jagielski } 90*b1cdbd2cSJim Jagielski 91*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- operator T_INTERFACE*() const92*b1cdbd2cSJim Jagielski operator T_INTERFACE*() const 93*b1cdbd2cSJim Jagielski { 94*b1cdbd2cSJim Jagielski return m_pInterface; 95*b1cdbd2cSJim Jagielski } 96*b1cdbd2cSJim Jagielski 97*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- operator *() const98*b1cdbd2cSJim Jagielski T_INTERFACE& operator*() const 99*b1cdbd2cSJim Jagielski { 100*b1cdbd2cSJim Jagielski return *m_pInterface; 101*b1cdbd2cSJim Jagielski } 102*b1cdbd2cSJim Jagielski 103*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- operator &()104*b1cdbd2cSJim Jagielski T_INTERFACE** operator&() 105*b1cdbd2cSJim Jagielski { 106*b1cdbd2cSJim Jagielski return &m_pInterface; 107*b1cdbd2cSJim Jagielski } 108*b1cdbd2cSJim Jagielski 109*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- operator ->() const110*b1cdbd2cSJim Jagielski T_INTERFACE* operator->() const 111*b1cdbd2cSJim Jagielski { 112*b1cdbd2cSJim Jagielski return m_pInterface; 113*b1cdbd2cSJim Jagielski } 114*b1cdbd2cSJim Jagielski 115*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- operator =(T_INTERFACE * pInterface)116*b1cdbd2cSJim Jagielski T_INTERFACE* operator=(T_INTERFACE* pInterface) 117*b1cdbd2cSJim Jagielski { 118*b1cdbd2cSJim Jagielski if ( equals(pInterface) ) 119*b1cdbd2cSJim Jagielski return m_pInterface; 120*b1cdbd2cSJim Jagielski 121*b1cdbd2cSJim Jagielski m_pInterface->Release(); 122*b1cdbd2cSJim Jagielski m_pInterface = pInterface; 123*b1cdbd2cSJim Jagielski if (m_pInterface) 124*b1cdbd2cSJim Jagielski m_pInterface->AddRef(); 125*b1cdbd2cSJim Jagielski 126*b1cdbd2cSJim Jagielski return m_pInterface; 127*b1cdbd2cSJim Jagielski } 128*b1cdbd2cSJim Jagielski 129*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- operator =(IUnknown * pIUnknown)130*b1cdbd2cSJim Jagielski T_INTERFACE* operator=(IUnknown* pIUnknown) 131*b1cdbd2cSJim Jagielski { 132*b1cdbd2cSJim Jagielski if (pIUnknown) 133*b1cdbd2cSJim Jagielski pIUnknown->QueryInterface(P_IID, (void**)&m_pInterface); 134*b1cdbd2cSJim Jagielski return m_pInterface; 135*b1cdbd2cSJim Jagielski } 136*b1cdbd2cSJim Jagielski 137*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- operator =(const ComPtr<T_INTERFACE,P_IID,P_CLSID> & aCopy)138*b1cdbd2cSJim Jagielski T_INTERFACE* operator=(const ComPtr< T_INTERFACE, P_IID, P_CLSID >& aCopy) 139*b1cdbd2cSJim Jagielski { 140*b1cdbd2cSJim Jagielski m_pInterface = aCopy.m_pInterface; 141*b1cdbd2cSJim Jagielski if (m_pInterface) 142*b1cdbd2cSJim Jagielski m_pInterface->AddRef(); 143*b1cdbd2cSJim Jagielski 144*b1cdbd2cSJim Jagielski return m_pInterface; 145*b1cdbd2cSJim Jagielski } 146*b1cdbd2cSJim Jagielski 147*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- get() const148*b1cdbd2cSJim Jagielski T_INTERFACE* get() const 149*b1cdbd2cSJim Jagielski { 150*b1cdbd2cSJim Jagielski return m_pInterface; 151*b1cdbd2cSJim Jagielski } 152*b1cdbd2cSJim Jagielski 153*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- attach(T_INTERFACE * pInterface)154*b1cdbd2cSJim Jagielski void attach(T_INTERFACE* pInterface) 155*b1cdbd2cSJim Jagielski { 156*b1cdbd2cSJim Jagielski if (pInterface) 157*b1cdbd2cSJim Jagielski { 158*b1cdbd2cSJim Jagielski m_pInterface->Release(); 159*b1cdbd2cSJim Jagielski m_pInterface = pInterface; 160*b1cdbd2cSJim Jagielski } 161*b1cdbd2cSJim Jagielski } 162*b1cdbd2cSJim Jagielski 163*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- detach()164*b1cdbd2cSJim Jagielski T_INTERFACE* detach() 165*b1cdbd2cSJim Jagielski { 166*b1cdbd2cSJim Jagielski T_INTERFACE* pInterface = m_pInterface; 167*b1cdbd2cSJim Jagielski m_pInterface = NULL; 168*b1cdbd2cSJim Jagielski return pInterface; 169*b1cdbd2cSJim Jagielski } 170*b1cdbd2cSJim Jagielski 171*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- release()172*b1cdbd2cSJim Jagielski void release() 173*b1cdbd2cSJim Jagielski { 174*b1cdbd2cSJim Jagielski if (m_pInterface) 175*b1cdbd2cSJim Jagielski { 176*b1cdbd2cSJim Jagielski m_pInterface->Release(); 177*b1cdbd2cSJim Jagielski m_pInterface = NULL; 178*b1cdbd2cSJim Jagielski } 179*b1cdbd2cSJim Jagielski } 180*b1cdbd2cSJim Jagielski 181*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- 182*b1cdbd2cSJim Jagielski template< class T_QUERYINTERFACE > query(T_QUERYINTERFACE ** pQuery)183*b1cdbd2cSJim Jagielski HRESULT query(T_QUERYINTERFACE** pQuery) 184*b1cdbd2cSJim Jagielski { 185*b1cdbd2cSJim Jagielski return m_pInterface->QueryInterface(__uuidof(T_QUERYINTERFACE), (void**)pQuery); 186*b1cdbd2cSJim Jagielski } 187*b1cdbd2cSJim Jagielski 188*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- query(REFIID rIID,void ** pQuery)189*b1cdbd2cSJim Jagielski HRESULT query(REFIID rIID , 190*b1cdbd2cSJim Jagielski void** pQuery) 191*b1cdbd2cSJim Jagielski { 192*b1cdbd2cSJim Jagielski return m_pInterface->QueryInterface(rIID, pQuery); 193*b1cdbd2cSJim Jagielski } 194*b1cdbd2cSJim Jagielski 195*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- unknown(IUnknown ** pQuery)196*b1cdbd2cSJim Jagielski HRESULT unknown(IUnknown** pQuery) 197*b1cdbd2cSJim Jagielski { 198*b1cdbd2cSJim Jagielski return m_pInterface->QueryInterface(IID_IUnknown, (void**)pQuery); 199*b1cdbd2cSJim Jagielski } 200*b1cdbd2cSJim Jagielski 201*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- equals(IUnknown * pCheck)202*b1cdbd2cSJim Jagielski ::sal_Bool equals(IUnknown* pCheck) 203*b1cdbd2cSJim Jagielski { 204*b1cdbd2cSJim Jagielski if ( 205*b1cdbd2cSJim Jagielski ( ! m_pInterface ) && 206*b1cdbd2cSJim Jagielski ( ! pCheck ) 207*b1cdbd2cSJim Jagielski ) 208*b1cdbd2cSJim Jagielski return sal_True; 209*b1cdbd2cSJim Jagielski 210*b1cdbd2cSJim Jagielski IUnknown* pCurrent = NULL; 211*b1cdbd2cSJim Jagielski m_pInterface->QueryInterface(IID_IUnknown, (void**)&pCurrent); 212*b1cdbd2cSJim Jagielski 213*b1cdbd2cSJim Jagielski ::sal_Bool bEquals = (pCheck == pCurrent); 214*b1cdbd2cSJim Jagielski pCurrent->Release(); 215*b1cdbd2cSJim Jagielski 216*b1cdbd2cSJim Jagielski return bEquals; 217*b1cdbd2cSJim Jagielski } 218*b1cdbd2cSJim Jagielski 219*b1cdbd2cSJim Jagielski //--------------------------------------------------------------------- is()220*b1cdbd2cSJim Jagielski ::sal_Bool is() 221*b1cdbd2cSJim Jagielski { 222*b1cdbd2cSJim Jagielski return (m_pInterface != 0); 223*b1cdbd2cSJim Jagielski } 224*b1cdbd2cSJim Jagielski 225*b1cdbd2cSJim Jagielski private: 226*b1cdbd2cSJim Jagielski T_INTERFACE* m_pInterface; 227*b1cdbd2cSJim Jagielski }; 228*b1cdbd2cSJim Jagielski 229*b1cdbd2cSJim Jagielski #endif 230