SegeX SgxVariant：VC封装支持多维数组的变体类型（VRIANT 、SafeArray）（免费免积分）

#include "stdafx.h" #include <comutil.h> #include "SgxVariant.h" CSgxVariant::CSgxVariant(void) { } CSgxVariant::CSgxVariant(VARTYPE vtNew, ULONG nCount, void* pData) { Redim(nCount, vtNew); int nElementSize = GetElementSizeFromVt(vtNew); if(nElementSize >= 0 && pData != NULL) SetData(pData, nCount, nElementSize); } CSgxVariant::CSgxVariant(VARIANT v) : _variant_t(v) {} CSgxVariant::~CSgxVariant(void) { } VARTYPE CSgxVariant::GetDataType() { VARTYPE vta; if (IsArray()) { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if (*ppArr == NULL) //空数组 return VT_ERROR; HRESULT hr = SafeArrayGetVartype(*ppArr, &vta); if (FAILED(hr)) { ASSERT(FALSE); return VT_ERROR; } return vta; } else { return vt; } } int CSgxVariant::GetArrayType() { //支持四种形式， 1 传值模式的数组，vt ： VT_ARRAY parray保存 // 2 引用形式的数组，vt: VT_ARRAY | VT_BYREFT pparray保存 // 3 引用传递VARIANT，vt：VT_BYREF | VT_VARIANT , pvarVal->vt: VT_ARRAY pvarVal->parray 或 4 pvalVal->pparray保存 if(vt & VT_ARRAY ) { if(vt & VT_BYREF) return VARIANT_ARRAY_TYPE_PP; else return VARIANT_ARRAY_TYPE_P; } else if(vt & VT_BYREF && vt & VT_VARIANT && pvarVal->vt & VT_ARRAY ) { if(pvarVal->vt & VT_BYREF) return VARIANT_ARRAY_TYPE_V_PP; else return VARIANT_ARRAY_TYPE_V_P; } return 0; } VARTYPE CSgxVariant::GetArrayVT() { if(vt & VT_ARRAY ) { return vt; } else if(vt & VT_BYREF && vt & VT_VARIANT && pvarVal->vt & VT_ARRAY ) { return pvarVal->vt; } return 0; } BOOL CSgxVariant::IsArray() { return GetArrayType() > 0; } SAFEARRAY** CSgxVariant::GetSafeArrayPtrPtr() { int nType = GetArrayType(); ASSERT(nType > 0); switch (nType) { case VARIANT_ARRAY_TYPE_P: return &parray; break; case VARIANT_ARRAY_TYPE_PP: return pparray; break; case VARIANT_ARRAY_TYPE_V_P: return &pvarVal->parray; break; case VARIANT_ARRAY_TYPE_V_PP: return pvarVal->pparray; break; } return NULL; } void* CSgxVariant::GetArrayData() { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if(ppArr == NULL) return NULL; else return (*ppArr)->pvData; } long CSgxVariant::GetDimensions() { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if(ppArr == NULL) return -1; return (*ppArr)->cDims; } long CSgxVariant::GetSize() { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if(ppArr == NULL) return -1; ASSERT((*ppArr)->cDims == 1); return (*ppArr)->rgsabound[0].cElements; } long CSgxVariant::GetSize(long nDim) { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if(ppArr == NULL) return -1; long Lbound, Ubound; SafeArrayGetLBound(*ppArr, nDim + 1, &Lbound);//nDim base 1 SafeArrayGetUBound(*ppArr, nDim + 1, &Ubound);//nDim base 1 return Ubound - Lbound + 1; //return (*ppArr)->rgsabound[nDim].cElements; } BOOL CSgxVariant::Redim(long n1, VARTYPE vtNew) { ASSERT(n1 >=0); SAFEARRAYBOUND MyBound[1]; MyBound[0].cElements = n1; MyBound[0].lLbound = 0; if(!(vt & VT_ARRAY) || GetDataType() != vtNew) { VariantInit(this); vt = VT_ARRAY | vtNew; this->parray=SafeArrayCreateVector(vtNew, 0, n1); if (parray == NULL) return FALSE; } else { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); HRESULT hr = SafeArrayRedim(*ppArr, MyBound); if (FAILED(hr)) return FALSE; } if (vtNew == VT_BSTR) _InitBStrArray(); return TRUE; } BOOL CSgxVariant::Redim(long n1, long n2, VARTYPE vtNew) { ASSERT(n1 >=0 && n2 >=0); SAFEARRAYBOUND MyBound[2]; MyBound[0].cElements = n1; MyBound[0].lLbound = 0; MyBound[1].cElements = n2; MyBound[1].lLbound = 0; if (!(vt & VT_ARRAY) || GetDataType() != vtNew) { VariantInit(this); vt = VT_ARRAY | vtNew; this->parray=SafeArrayCreate(vtNew, 2, MyBound); if (parray == NULL) return FALSE; } else { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if (*ppArr != NULL) SafeArrayDestroy(*ppArr); *ppArr = SafeArrayCreate(vtNew, 2, MyBound); if (*ppArr == NULL) return FALSE; //return SafeArrayRedim(*ppArr, MyBound) == S_OK; } return TRUE; } BOOL CSgxVariant::Redim(long n1, long n2, long n3, VARTYPE vtNew) { ASSERT(n1 >=0 && n2 >=0 && n3 >= 0); SAFEARRAYBOUND MyBound[3]; MyBound[0].cElements = n1; MyBound[0].lLbound = 0; MyBound[1].cElements = n2; MyBound[1].lLbound = 0; MyBound[2].cElements = n3; MyBound[2].lLbound = 0; if (!(vt & VT_ARRAY) || GetDataType() != vtNew) { VariantInit(this); vt = VT_ARRAY | vtNew; this->parray=SafeArrayCreate(vtNew, 3, MyBound); if (parray == NULL) return FALSE; } else { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if (*ppArr != NULL) SafeArrayDestroy(*ppArr); *ppArr = SafeArrayCreate(vtNew, 3, MyBound); if (*ppArr == NULL) return FALSE; //return SafeArrayRedim(*ppArr, MyBound) == S_OK; } return TRUE; } BOOL CSgxVariant::Redim(long n1, long n2, long n3, long n4, VARTYPE vtNew) { ASSERT(n1 >= 0 && n2 >= 0 && n3 >= 0 && n4 >= 0); SAFEARRAYBOUND MyBound[4]; MyBound[0].cElements = n1; MyBound[0].lLbound = 0; MyBound[1].cElements = n2; MyBound[1].lLbound = 0; MyBound[2].cElements = n3; MyBound[2].lLbound = 0; MyBound[3].cElements = n4; MyBound[3].lLbound = 0; if (!(vt & VT_ARRAY) || GetDataType() != vtNew) { VariantInit(this); vt = VT_ARRAY | vtNew; this->parray=SafeArrayCreate(vtNew, 3, MyBound); if (parray == NULL) return FALSE; } else { SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if (*ppArr != NULL) SafeArrayDestroy(*ppArr); *ppArr = SafeArrayCreate(vtNew, 4, MyBound); if (*ppArr == NULL) return FALSE; //return SafeArrayRedim(*ppArr, MyBound) == S_OK; } return TRUE; } BOOL CSgxVariant::RedimPreserve(long nNewCount) { ASSERT(nNewCount >= 0); SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if(ppArr == NULL) return FALSE; int nElementSize = GetElementSizeFromVt(vt); if(nElementSize <= 0) return FALSE; //保留原始数据 BYTE* pNewData=NULL, *pOldData=(BYTE*)(*ppArr)->pvData; try{ pNewData= new BYTE[nNewCount * nElementSize]; } catch(...) { return FALSE; } int nCopyCount; if(nNewCount < GetSize())//缩小数组 nCopyCount=nNewCount; else //增大数组 { nCopyCount=GetSize(); ZeroMemory(pNewData + nCopyCount*nElementSize,(nNewCount-GetSize())* nElementSize); } memcpy(pNewData, pOldData, nElementSize*nCopyCount); //重新设置大小 SAFEARRAYBOUND BoundNew[1]; BoundNew[0].cElements = nNewCount; BoundNew[0].lLbound = 0; SafeArrayRedim(*ppArr,BoundNew); //赋值 //ULONG i; //long ix[1]; //BYTE* p = pNewData; //for(i=0; i<nNewCount; ++i) //{ // ix[0]=i; // SafeArrayPutElement(parray, ix, (void *)p); // p += nElementSize; //} BOOL bRet = SetData((void*)pNewData, nNewCount, nElementSize); //清理 delete [] pNewData; return bRet; } BOOL CSgxVariant::SetData(void* pData, long nCount, int nElemenstSize) { long nOldCount=GetSize(); ASSERT(!(nOldCount < 1 || nCount > nOldCount || nElemenstSize <= 0)); if(nOldCount < 1 || nCount > nOldCount || nElemenstSize <= 0) return FALSE; if(pData == NULL || nCount == 0) return TRUE; SAFEARRAY** ppArr = GetSafeArrayPtrPtr(); if(ppArr == NULL) return FALSE; ASSERT((*ppArr)->cbElements == nElemenstSize); BYTE* p = (BYTE*)pData; long ix[1]; //指定的数据个数不够 if(!::AfxIsValidAddress(p, nElemenstSize * nCount, TRUE)) return FALSE; for(long i = 0; i < nCount; i++) { ix[0]=i; SafeArrayPutElement(*ppArr, ix, (void *)p); p += nElemenstSize; } return TRUE; } //static functions long CSgxVariant::_GetSize(VARIANT& v) { CSgxVariant sv; sv.Attach(v); long nSize = sv.GetSize(); v = sv.Detach(); return nSize; } BOOL CSgxVariant::EqualVt(VARTYPE vt1, VARTYPE vt2) { NormalizeVt(vt1); Normaliz