299 lines
9.4 KiB
C++
299 lines
9.4 KiB
C++
|
#include "precomp.h"
|
||
|
|
#pragma hdrstop
|
||
|
|
#include <wiadebug.h>
|
||
|
|
#include <wiaffmt.h>
|
||
|
|
#include "wiadefui.h"
|
||
|
|
#include "wiauiext.h"
|
||
|
|
|
||
|
|
extern HINSTANCE g_hInstance;
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::Initialize( IWiaItem *pWiaItem, LONG nMediaType )
|
||
|
|
{
|
||
|
|
WIA_PUSH_FUNCTION((TEXT("CWiaDefaultUI::Initialize")));
|
||
|
|
m_WiaFormatPairs.Destroy();
|
||
|
|
m_nDefaultFormat = 0;
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
if (!pWiaItem)
|
||
|
|
return(E_POINTER);
|
||
|
|
GUID guidDefaultClipFormat = GUID_NULL;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Don't worry about failure, we will come up with our own default if the driver doesn't report on
|
||
|
|
//
|
||
|
|
PropStorageHelpers::GetProperty( pWiaItem, WIA_IPA_PREFERRED_FORMAT, guidDefaultClipFormat );
|
||
|
|
WIA_PRINTGUID((guidDefaultClipFormat,TEXT("guidDefaultClipFormat")));
|
||
|
|
if (guidDefaultClipFormat == GUID_NULL)
|
||
|
|
{
|
||
|
|
if (TYMED_FILE == nMediaType || TYMED_MULTIPAGE_FILE == nMediaType)
|
||
|
|
{
|
||
|
|
guidDefaultClipFormat = WiaImgFmt_BMP;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
guidDefaultClipFormat = WiaImgFmt_MEMORYBMP;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// if the transfer mechanism is known to be incompatible
|
||
|
|
// with the current tymed, change it
|
||
|
|
//
|
||
|
|
if (guidDefaultClipFormat == WiaImgFmt_BMP && nMediaType == TYMED_CALLBACK)
|
||
|
|
{
|
||
|
|
guidDefaultClipFormat = WiaImgFmt_MEMORYBMP;
|
||
|
|
}
|
||
|
|
else if ((guidDefaultClipFormat == WiaImgFmt_MEMORYBMP) && ((nMediaType == TYMED_FILE) || (TYMED_MULTIPAGE_FILE == nMediaType)))
|
||
|
|
{
|
||
|
|
guidDefaultClipFormat = WiaImgFmt_BMP;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get the data transfer interface
|
||
|
|
//
|
||
|
|
CComPtr<IWiaDataTransfer> pWiaDataTransfer;
|
||
|
|
hr = pWiaItem->QueryInterface( IID_IWiaDataTransfer, (void**)&pWiaDataTransfer );
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Get the format info enumerator
|
||
|
|
//
|
||
|
|
CComPtr<IEnumWIA_FORMAT_INFO> pEnumWIA_FORMAT_INFO;
|
||
|
|
hr = pWiaDataTransfer->idtEnumWIA_FORMAT_INFO(&pEnumWIA_FORMAT_INFO);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Enumerate the formats
|
||
|
|
//
|
||
|
|
ULONG ulFetched = 0;
|
||
|
|
WIA_FORMAT_INFO WiaFormatInfo;
|
||
|
|
while (pEnumWIA_FORMAT_INFO->Next(1,&WiaFormatInfo,&ulFetched) == S_OK)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// If this is the media type we are interested in...
|
||
|
|
//
|
||
|
|
if (static_cast<LONG>(WiaFormatInfo.lTymed) == nMediaType)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// The friendly description for this file type
|
||
|
|
//
|
||
|
|
CSimpleString strDescription;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get the file extension for this type
|
||
|
|
//
|
||
|
|
CSimpleString strExtension = CWiaFileFormat::GetExtension( WiaFormatInfo.guidFormatID, nMediaType, pWiaItem );
|
||
|
|
if (strExtension.Length())
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Save the extension
|
||
|
|
//
|
||
|
|
CSimpleString strExtensionPlusDot = TEXT(".");
|
||
|
|
strExtensionPlusDot += strExtension;
|
||
|
|
if (strExtensionPlusDot.Length())
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Get the description
|
||
|
|
//
|
||
|
|
SHFILEINFO SHFileInfo;
|
||
|
|
if (SHGetFileInfo(strExtensionPlusDot.String(), FILE_ATTRIBUTE_NORMAL, &SHFileInfo, sizeof(SHFILEINFO), SHGFI_USEFILEATTRIBUTES|SHGFI_TYPENAME ))
|
||
|
|
{
|
||
|
|
strDescription = SHFileInfo.szTypeName;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// SUCCESS! Save the extension and description
|
||
|
|
//
|
||
|
|
int nIndex = m_WiaFormatPairs.Append( CWiaFormatPair( static_cast<GUID>(WiaFormatInfo.guidFormatID), strExtension, CSimpleStringConvert::WideString(strDescription) ) );
|
||
|
|
|
||
|
|
//
|
||
|
|
// Save the default format index, if this is the default format
|
||
|
|
//
|
||
|
|
if (guidDefaultClipFormat == WiaFormatInfo.guidFormatID)
|
||
|
|
{
|
||
|
|
m_nDefaultFormat = nIndex;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::GetFormatCount( LONG *pnCount )
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
if (!pnCount)
|
||
|
|
return(E_POINTER);
|
||
|
|
*pnCount = m_WiaFormatPairs.Size();
|
||
|
|
return(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::GetFormatType( LONG nFormat, GUID *pcfClipFormat )
|
||
|
|
{
|
||
|
|
WIA_PUSH_FUNCTION((TEXT("CWiaDefaultUI::GetFormatType")));
|
||
|
|
WIA_PRINTGUID((*pcfClipFormat,TEXT("nFormat: %d, pcfClipFormat:"),nFormat));
|
||
|
|
WIA_TRACE((TEXT("m_WiaFormatPairs.Size(): %d"),m_WiaFormatPairs.Size()));
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
// Out of range
|
||
|
|
if (nFormat >= m_WiaFormatPairs.Size() || nFormat < 0)
|
||
|
|
return(E_FAIL);
|
||
|
|
*pcfClipFormat = m_WiaFormatPairs[nFormat].Type();
|
||
|
|
return(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::GetFormatExtension( LONG nFormat, LPWSTR pszExtension, int nMaxLen )
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
// Out of range
|
||
|
|
if (nFormat >= m_WiaFormatPairs.Size() || nFormat < 0)
|
||
|
|
return(E_FAIL);
|
||
|
|
CSimpleStringWide str = m_WiaFormatPairs[nFormat].Extension();
|
||
|
|
if (static_cast<int>(str.Length()) >= nMaxLen)
|
||
|
|
return(E_FAIL);
|
||
|
|
lstrcpyW( pszExtension, str.String() );
|
||
|
|
return(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::GetFormatDescription( LONG nFormat, LPWSTR pszDescription, int nMaxLen )
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
// Out of range
|
||
|
|
if (nFormat >= m_WiaFormatPairs.Size() || nFormat < 0)
|
||
|
|
return(E_FAIL);
|
||
|
|
CSimpleStringWide str = m_WiaFormatPairs[nFormat].Description();
|
||
|
|
if (static_cast<int>(str.Length()) >= nMaxLen)
|
||
|
|
return(E_FAIL);
|
||
|
|
lstrcpyW( pszDescription, str.String() );
|
||
|
|
return(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::GetDefaultClipboardFileFormat( GUID *pguidFormat )
|
||
|
|
{
|
||
|
|
if (!pguidFormat)
|
||
|
|
return E_POINTER;
|
||
|
|
return GetFormatType(m_nDefaultFormat,pguidFormat);
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::GetDefaultClipboardFileFormatIndex( LONG *pnIndex )
|
||
|
|
{
|
||
|
|
if (!pnIndex)
|
||
|
|
return E_POINTER;
|
||
|
|
if (m_nDefaultFormat >= m_WiaFormatPairs.Size() || m_nDefaultFormat < 0)
|
||
|
|
return(E_FAIL);
|
||
|
|
*pnIndex = m_nDefaultFormat;
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::GetClipboardFileExtension( GUID guidFormat, LPWSTR pszExt, DWORD nMaxLen )
|
||
|
|
{
|
||
|
|
if (!pszExt)
|
||
|
|
{
|
||
|
|
return E_POINTER;
|
||
|
|
}
|
||
|
|
|
||
|
|
CSimpleString strExtension = CWiaFileFormat::GetExtension( guidFormat );
|
||
|
|
if (strExtension.Length())
|
||
|
|
{
|
||
|
|
CSimpleStringWide strwExtension = CSimpleStringConvert::WideString(strExtension);
|
||
|
|
if (strwExtension.Length() < nMaxLen)
|
||
|
|
{
|
||
|
|
lstrcpyW( pszExt, strwExtension.String() );
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return S_FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::ChangeClipboardFileExtension( GUID guidFormat, LPWSTR pszFilename, DWORD nMaxLen )
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
if (!pszFilename || guidFormat==GUID_NULL) // We don't accept a default type here
|
||
|
|
{
|
||
|
|
return E_INVALIDARG;
|
||
|
|
}
|
||
|
|
|
||
|
|
WCHAR szExtension[MAX_PATH]=L"";
|
||
|
|
GetClipboardFileExtension( guidFormat, szExtension, ARRAYSIZE(szExtension) );
|
||
|
|
if (!lstrlenW(szExtension))
|
||
|
|
{
|
||
|
|
return S_FALSE; // Not really an error, just an unknown file type.
|
||
|
|
}
|
||
|
|
|
||
|
|
CSimpleStringWide strName(pszFilename);
|
||
|
|
// Make sure the string is valid
|
||
|
|
if (strName.Length())
|
||
|
|
{
|
||
|
|
int nPeriodFind = strName.ReverseFind( L'.' );
|
||
|
|
int nBSlashFind = strName.ReverseFind( L'\\' );
|
||
|
|
if (nPeriodFind < 0) // No extension found
|
||
|
|
{
|
||
|
|
strName += L'.';
|
||
|
|
strName += szExtension;
|
||
|
|
}
|
||
|
|
else if (nPeriodFind > nBSlashFind) // Assume this is an extension, because it is following a back slash
|
||
|
|
{
|
||
|
|
strName = strName.Left(nPeriodFind);
|
||
|
|
strName += L'.';
|
||
|
|
strName += szExtension;
|
||
|
|
}
|
||
|
|
else // It must not be an extension
|
||
|
|
{
|
||
|
|
strName += L'.';
|
||
|
|
strName += szExtension;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make sure this string can handle the addition of the extension
|
||
|
|
if ((strName.Length()+1) <= nMaxLen)
|
||
|
|
{
|
||
|
|
lstrcpyW( pszFilename, strName.String() );
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hr = E_INVALIDARG;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHODIMP CWiaDefaultUI::ConstructFileOpenDialogStyleString( BSTR *pbstrString )
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
int nLength = 0;
|
||
|
|
|
||
|
|
// For each ext: "Foo Files (*.foo)|*.foo|"
|
||
|
|
for (int i=0;i<m_WiaFormatPairs.Size();i++)
|
||
|
|
{
|
||
|
|
nLength += m_WiaFormatPairs[i].Description().Length() +
|
||
|
|
lstrlenW(L" (*.") +
|
||
|
|
m_WiaFormatPairs[i].Extension().Length() +
|
||
|
|
lstrlenW(L")|*.") +
|
||
|
|
m_WiaFormatPairs[i].Extension().Length() +
|
||
|
|
lstrlenW(L"|");
|
||
|
|
}
|
||
|
|
*pbstrString = SysAllocStringLen( NULL, nLength );
|
||
|
|
if (*pbstrString)
|
||
|
|
{
|
||
|
|
CSimpleStringWide strTmp;
|
||
|
|
for (i=0;i<m_WiaFormatPairs.Size();i++)
|
||
|
|
{
|
||
|
|
strTmp += m_WiaFormatPairs[i].Description() +
|
||
|
|
L" (*." +
|
||
|
|
m_WiaFormatPairs[i].Extension() +
|
||
|
|
L")|*." +
|
||
|
|
m_WiaFormatPairs[i].Extension() +
|
||
|
|
L"|";
|
||
|
|
CopyMemory( *pbstrString, strTmp.String(), nLength * sizeof(WCHAR) );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else hr = E_OUTOFMEMORY;
|
||
|
|
return(hr);
|
||
|
|
}
|
||
|
|
|