/**************************************************************
 * 
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 * 
 *************************************************************/


#ifndef _SMARTARRAY_H
#define _SMARTARRAY_H


template< class sourceType>
class SmartArray
{
	SAFEARRAY *m_array;
public:

	SmartArray( sourceType * parParams, int count, VARTYPE destVartype): m_array(NULL)
	{
		HRESULT hr= S_OK;
		SAFEARRAYBOUND rgsabound[1];
		rgsabound[0].cElements= count;
		rgsabound[0].lLbound= 0;
		m_array= SafeArrayCreate( destVartype, 1, rgsabound);
		SafeArrayLock( m_array);	
		
		void* pData;
		if( m_array && (SUCCEEDED( SafeArrayAccessData( m_array, (void**)&pData)) ) )
		{

			for( int i=0; i< count; i++)
			{	
				CComVariant varSource( parParams[i]);
				switch (destVartype)
				{
				case VT_I1:
					{
						char* p= (char*) pData;
						if( SUCCEEDED( hr= varSource.ChangeType( destVartype)))
							p[i]= V_I1( &varSource);
						break;
					}
				case VT_I2:
					{
						short* p= (short*) pData;
						if( SUCCEEDED( hr=varSource.ChangeType( destVartype)))
							p[i]= V_I2( &varSource);
						break;
					}
				case VT_UI2:
					{
						unsigned short* p= (unsigned short*) pData;
						if( SUCCEEDED( hr=varSource.ChangeType( destVartype)))
							p[i]= V_UI2( &varSource);
						break;
					}
				case VT_I4:
					{
						long* p= (long*)pData;
					if( SUCCEEDED( hr=varSource.ChangeType( destVartype)))
						p[i]= V_I4( &varSource);
					break;
					}
				case VT_UI4:
					{
						unsigned long* p= (unsigned long*)pData;
						if( SUCCEEDED( hr=varSource.ChangeType( destVartype)))
						p[i]= V_UI4( &varSource);
						break;
					}
				case VT_R4:
					{
						float* p= (float*)pData;
					if( SUCCEEDED( hr=varSource.ChangeType( destVartype)))
						p[i]= V_R4( &varSource);
						break;
					}
				case VT_R8:
					{
						double* p= (double*)pData;
						if( SUCCEEDED( hr=varSource.ChangeType( destVartype)))
						p[i]= V_R8( &varSource);
						break;
					}
				case VT_BOOL:
					{
						VARIANT_BOOL* p= (VARIANT_BOOL*)pData;
						if( SUCCEEDED( hr=varSource.ChangeType( destVartype)))
						p[i]= V_BOOL( &varSource);
						break;
					}
				case VT_BSTR:
					{
					BSTR* pBstr= ( BSTR*)pData;
					if( SUCCEEDED( hr=varSource.ChangeType( destVartype)))
						pBstr[i]= SysAllocString(V_BSTR( &varSource));
					break;
					}
				case VT_VARIANT:
					{
						VARIANT *pVariant= (VARIANT*)pData;
						hr= VariantCopy( &pVariant[i], &varSource); break;
					}
//				case VT_UNKNOWN:
//					{
//						long* pUnk= (long*)pData;
//						pUnk[i]= reinterpret_cast<long>(parParams[i]);
//						((IUnknown*)pUnk[i])->AddRef(); break;
//					}
//				case VT_DISPATCH:
//					{
//						long* pDisp= (long*)pData;
//						pDisp[i]= (long)parParams[i];
//						((IDispatch*)pDisp[i])->AddRef(); break;
//					}
				default:
					hr= E_FAIL;
				}
			}
			if( FAILED( hr))
			{
				SafeArrayDestroy( m_array);
				m_array= NULL;
			}
		}
		SafeArrayUnaccessData( m_array);
	}
	~SmartArray(){ 
		SafeArrayUnlock( m_array);
		SafeArrayDestroy( m_array );
	}

	operator bool (){ return m_array == NULL ?  false : true; }

	operator SAFEARRAY* (){ return m_array;}

};

template<>
class SmartArray<IUnknown*>
{
	SAFEARRAY *m_array;
public:

	SmartArray( sourceType * parParams, int count, VARTYPE destVartype); 
//	{
//		ATLTRACE("SmartArray<IUnknown>");
//		HRESULT hr= S_OK;
//		SAFEARRAYBOUND rgsabound[1];
//		rgsabound[0].cElements= count;
//		rgsabound[0].lLbound= 0;
//		m_array= SafeArrayCreateVector( VT_UNKNOWN, 0, count);
//		SafeArrayLock( m_array);	
//		
//		IUnknown* *pData;
//		if( m_array && (SUCCEEDED( SafeArrayAccessData( m_array, (void**)&pData)) ) )
//		{
//
//			for( int i=0; i< count; i++)
//			{	
//				CComVariant varSource( parParams[i]);
//				switch (destVartype)
//				{
//
//				case VT_UNKNOWN:
//					{
//						pData[i]= parParams[i];
//						pData[i]->AddRef();
//					}
//				default:
//					hr= E_FAIL;
//				}
//			}
//			if( FAILED( hr))
//			{
//				SafeArrayDestroy( m_array);
//				m_array= NULL;
//			}
//		}
//		SafeArrayUnaccessData( m_array);
//	}
	~SmartArray(){ 
		SafeArrayUnlock( m_array);
		SafeArrayDestroy( m_array );
	}

	operator bool (){ return m_array == NULL ?  false : true; }

	operator SAFEARRAY* (){ return m_array;}

};

template <> SmartArray <IUnknown*>::SmartArray(sourceType * parParams, int count, VARTYPE destVartype):m_array(NULL)
{
	ATLTRACE("SmartArray<IUnknown>");
	HRESULT hr= S_OK;
	SAFEARRAYBOUND rgsabound[1];
	rgsabound[0].cElements= count;
	rgsabound[0].lLbound= 0;
	m_array= SafeArrayCreateVector( VT_UNKNOWN, 0, count);
	SafeArrayLock( m_array);	
	
	IUnknown* *pData;
	if( m_array && (SUCCEEDED( SafeArrayAccessData( m_array, (void**)&pData)) ) )
	{
		for( int i=0; i< count; i++)
		{	
			pData[i]= parParams[i];
			pData[i]->AddRef();
		}
	}
	SafeArrayUnaccessData( m_array);
};
#endif