开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > Shell编程 > 查看源码
DataObject.h
资源名称：adfview_src.zip　[点击查看]
上传用户：hy_wanghao
上传日期：2007-01-08
资源大小：279k
文件大小：12k
源码类别：

Shell编程

开发平台：

Visual C++

DataObject.h：源码内容
  1. #if !defined(AFX_DATAOBJECT_H__20001229_075F_904E_7D1B_0080AD509054__INCLUDED_)
  2. #define AFX_DATAOBJECT_H__20001229_075F_904E_7D1B_0080AD509054__INCLUDED_
  4. #pragma once
  6. #include "AmigaFileStream.h"
  7. #include "Misc.h"
  8. #include "EnumFormatETC.h"
  11. /////////////////////////////////////////////////////////////////////////////
  12. // CDataObject
  14. class ATL_NO_VTABLE CDataObject : 
  15.    public CComObjectRootEx<CComSingleThreadModel>, 
  16.    public IDataObject
  17. {
  18. public:
  20. DECLARE_NOT_AGGREGATABLE(CDataObject)
  21. DECLARE_PROTECT_FINAL_CONSTRUCT()
  23. BEGIN_COM_MAP(CDataObject)
  24.    COM_INTERFACE_ENTRY(IDataObject)
  25. END_COM_MAP()
  27.    CFolder *m_pFolder;          // CFolder reference
  28.    CPidlList m_pidls;           // The list of files
  29.    // The source window's HWND
  30.    HWND m_hWnd;
  31.    // These flags are used in the communication going on when doing
  32.    // a copy/cut/paste operation.
  33.    DWORD m_dwPasteSucceeded;
  34.    DWORD m_dwPerformedDropEffect;
  35.    DWORD m_dwPreferredDropEffect;
  37.    HRESULT FinalConstruct()
  38.    {
  39.       m_pFolder = NULL;
  40.       m_hWnd = NULL;
  41.       m_dwPasteSucceeded =
  42.       m_dwPerformedDropEffect =
  43.       m_dwPreferredDropEffect = DROPEFFECT_NONE;
  44.       return S_OK;
  45.    }
  47.    void FinalRelease()
  48.    {
  49.       if( m_pFolder!=NULL ) m_pFolder->Release();
  50.    }
  52.    HRESULT _Init(CFolder *pFolder, HWND hWnd, LPCITEMIDLIST *pPidls, int nCount)
  53.    {
  54.       ATLTRACE(_T("CDataObject::_Initn"));
  55.       ATLASSERT(pFolder);
  56.       ATLASSERT(pPidls);
  57.       m_pFolder = pFolder;
  58.       m_pFolder->AddRef();
  59.       m_hWnd = hWnd;
  60.       m_pidls.Copy(pPidls, nCount, TRUE);
  61.       return S_OK;
  62.    }
  64. private:
  65.    HRESULT _GetAmigaFileDescriptor(CAdfVolume &vol, int i, FILEDESCRIPTOR *p)
  66.    {
  67.       ATLASSERT(vol.IsOpen());
  68.       ATLASSERT(p);
  69.       
  70.       ::ZeroMemory(p, sizeof(FILEDESCRIPTOR));
  72.       CPidl pidl;
  73.       pidl.Copy(m_pFolder->m_pidlPath);
  74.       pidl.Concatenate(m_pidls[i]);
  76.       TCHAR szPath[MAX_PATH];
  77.       PidlGetFilePath(szPath, pidl);
  78.       if( vol.ChangeDirectory(szPath)==FALSE ) return E_FAIL;
  80.       // Get information about the file
  81.       CAdfFile f;
  82.       TCHAR szName[MAXNAMELEN+1];
  83.       PidlGetName(szName, pidl);
  84.       if( vol.GetFile(szName, _T("r"), f)==FALSE ) return E_FAIL;
  86.       p->dwFlags = FD_FILESIZE | FD_CREATETIME | FD_WRITESTIME | FD_ATTRIBUTES;
  87.       f.GetName(p->cFileName, lengthof(p->cFileName));
  88.       p->nFileSizeHigh = 0;
  89.       p->nFileSizeLow = f.GetSize();
  90.       p->ftCreationTime = f.GetFileTime();
  91.       p->ftLastWriteTime = f.GetFileTime();
  92.       DWORD dwAttribs = 0;
  93.       if( f.IsDirectory() ) dwAttribs |= FILE_ATTRIBUTE_DIRECTORY;
  94.       DWORD dwAccess = f.GetAccess();
  95.       if( hasW(dwAccess) ) dwAttribs |= FILE_ATTRIBUTE_READONLY;
  96.       if( hasH(dwAccess) ) dwAttribs |= FILE_ATTRIBUTE_HIDDEN;
  97.       if( hasA(dwAccess) ) dwAttribs |= FILE_ATTRIBUTE_ARCHIVE;
  98.       if( dwAttribs==0 ) dwAttribs = FILE_ATTRIBUTE_NORMAL;
  99.       p->dwFileAttributes = dwAttribs;
  100.       // BUG: We loose several Amiga file attributes
  102.       return S_OK;
  103.    }
  105.    HRESULT _GetFileDescriptors(FILEDESCRIPTOR *p)
  106.    {
  107.       ATLASSERT(p);
  108.       CAdfDevice dev;
  109.       CAdfVolume vol;
  110.       HRESULT Hr;
  111.       HR( m_pFolder->_OpenAmigaDevice(m_pFolder->m_pidl, TRUE, NULL, dev, vol) );
  112.       for( UINT i=0; i<m_pidls.GetCount(); i++ ) {
  113.          _GetAmigaFileDescriptor(vol,i,p);
  114.          p++;
  115.       }
  116.       return S_OK;
  117.    }
  119. // IDataObject
  120. public:
  122.    STDMETHOD(GetData)(FORMATETC *pFormatetc, STGMEDIUM *pMedium)
  123.    {
  124.       ATLTRACE(_T("CDataObject::GetDatan"));
  125.       ATLASSERT(m_pFolder);
  126.       VALIDATE_POINTER(pFormatetc);
  127.       VALIDATE_POINTER(pMedium);
  128.       if( m_pidls.GetCount()==0 ) return E_UNEXPECTED;
  130.       ATLTRACE(_T("Format = 0x%04x, '%s'n"), pFormatetc->cfFormat, DbgGetCF(pFormatetc->cfFormat));
  131.       ATLTRACE(_T("TYMED = 0x%04xn"), pFormatetc->tymed);
  133.       HRESULT Hr;
  134.       ::ZeroMemory(pMedium, sizeof(STGMEDIUM));
  136.       if( pFormatetc->cfFormat == _Module.m_CFSTR_ADFID ) {
  137.          // Internal CF used to query which CFolder the IDataObject comes from
  138.          HR( _GetDWORD(pMedium, m_pFolder->m_dwFolderID) );
  139.          return S_OK;
  140.       }
  141.       else if( pFormatetc->cfFormat == _Module.m_CFSTR_FILEDESCRIPTOR ) {
  142.          // The CFSTR_FILEDESCRIPTOR format will be queried by the Shell before
  143.          // the CFSTR_FILECONTENTS.
  144.          // It returns descriptions of the files the IDataObject contains.
  145.          if( (pFormatetc->tymed & TYMED_HGLOBAL)==0 ) return DV_E_TYMED;
  146.      
  147.          FILEGROUPDESCRIPTOR fgd = { 0 };
  148.          DWORD fgd_size = sizeof(fgd)-sizeof(FILEDESCRIPTOR);
  149.          fgd.cItems = m_pidls.GetCount();
  151.          DWORD dwTotalSize = fgd_size + (sizeof(FILEDESCRIPTOR)*fgd.cItems);
  152.          HGLOBAL hMem = ::GlobalAlloc(GMEM_MOVEABLE | GMEM_SHARE | GMEM_ZEROINIT | GMEM_DISCARDABLE, dwTotalSize);
  153.          if( hMem==NULL ) return E_OUTOFMEMORY;
  154.          LPBYTE p = (LPBYTE)::GlobalLock(hMem);
  155.          ::CopyMemory(p, &fgd, fgd_size );
  156.          p += fgd_size;
  157.          HR( _GetFileDescriptors((FILEDESCRIPTOR *)p) );
  158.          ::GlobalUnlock(hMem);         
  160.          pMedium->tymed = TYMED_HGLOBAL;
  161.          pMedium->hGlobal = hMem;
  162.          pMedium->pUnkForRelease = NULL;
  163.          return S_OK;
  164.       }
  165.       else if( pFormatetc->cfFormat == _Module.m_CFSTR_FILECONTENTS ) {
  166.          // The CFSTR_FILECONTENTS gets called for each file.
  167.          // It must return the IStream for the file.
  168.          if( (pFormatetc->tymed & TYMED_ISTREAM)==0 ) return DV_E_TYMED;
  169.          // HACK: Explorer will call us with index -1 when clipdata goes
  170.          //       out of scope (view closes) ????
  171.          LONG iIndex = pFormatetc->lindex;
  172.          if( iIndex<0 || iIndex>=(LONG)(m_pidls.GetCount()) ) return DV_E_LINDEX;
  174.          // Need to initialize Stream attributes
  175.          CPidl pidl;
  176.          pidl.Copy(m_pFolder->m_pidlPath);
  177.          pidl.Concatenate(m_pidls[iIndex]);
  179.          // Extract the filepath and name contained in the
  180.          // full PIDL.
  181.          TCHAR szPath[MAX_PATH];
  182.          TCHAR szName[MAXNAMELEN+1];
  183.          PidlGetFilePath(szPath, pidl);
  184.          PidlGetName(szName, pidl);
  186.          CComObject<CAmigaFileStream> *pStream;
  187.          HR( CComObject<CAmigaFileStream>::CreateInstance(&pStream) );
  188.          HR( pStream->_Init(m_pFolder->m_pidl, szPath, szName) );
  190.          HR( pStream->QueryInterface(IID_IStream, (LPVOID *)&pMedium->pstm) );
  191.          pMedium->tymed = TYMED_ISTREAM;
  192.          pMedium->pUnkForRelease = NULL;
  193.          return S_OK;        
  194.       }
  195.       else if( pFormatetc->cfFormat == _Module.m_CFSTR_PERFORMEDDROPEFFECT ) {
  196.          HR( _GetDWORD(pMedium, m_dwPerformedDropEffect) );
  197.          return S_OK;
  198.       }
  199.       else if( pFormatetc->cfFormat == _Module.m_CFSTR_PREFERREDDROPEFFECT ) {        
  200.          // If we're not doing a "cut" operation (like in cut&paste)
  201.          // let the IDropTarget decide which effect we prefer.
  202.          if( m_dwPreferredDropEffect==DROPEFFECT_NONE ) return DV_E_FORMATETC;
  203.          HR( _GetDWORD(pMedium, m_dwPreferredDropEffect) );
  204.          return S_OK;
  205.       }
  206.       return DV_E_FORMATETC;
  207.    }
  209.    STDMETHOD(GetDataHere)(FORMATETC* /*pFormatetc*/, STGMEDIUM* /*pMedium*/)
  210.    {
  211.       ATLTRACENOTIMPL(_T("CDataObject::GetDataHere"));
  212.    }
  214.    STDMETHOD(QueryGetData)(FORMATETC* pFormatetc)
  215.    {
  216.       ATLASSERT(pFormatetc);
  217.       ATLTRACE(_T("CDataObject::QueryGetData = 0x%04x, '%s'n"), pFormatetc->cfFormat, DbgGetCF(pFormatetc->cfFormat));
  218.       if( pFormatetc->cfFormat!=_Module.m_CFSTR_FILEDESCRIPTOR &&
  219.           pFormatetc->cfFormat!=_Module.m_CFSTR_FILECONTENTS )
  220.       {
  221.           return DV_E_FORMATETC;
  222.       }
  223.       if( (pFormatetc->dwAspect & DVASPECT_CONTENT)==0 ) 
  224.       {
  225.          return DV_E_DVASPECT;
  226.       }
  227.       return S_OK;
  228.    }
  230.    STDMETHOD(EnumFormatEtc)(DWORD /*dwDirection*/, IEnumFORMATETC** ppenumFormatEtc)
  231.    {
  232.       ATLTRACE(_T("CDataObject::EnumFormatEtcn"));
  233.       VALIDATE_POINTER(ppenumFormatEtc);
  234.       HRESULT Hr;
  235.       CComObject<CEnumFORMATETC> *obj;
  236.       HR( CComObject<CEnumFORMATETC>::CreateInstance(&obj) );
  237.       return obj->QueryInterface(IID_IEnumFORMATETC, (LPVOID *)ppenumFormatEtc);
  238.    }
  240.    STDMETHOD(SetData)(FORMATETC* pFormatEtc, STGMEDIUM* pMedium, BOOL fRelease)
  241.    {
  242.       ATLTRACE(_T("CDataObject::SetDatan"));
  243.       ATLASSERT(m_pFolder);
  244.       VALIDATE_POINTER(pFormatEtc);
  245.       VALIDATE_POINTER(pMedium);
  246.       HRESULT Hr;
  248.       ATLTRACE(_T("Format = 0x%04x, '%s'n"), pFormatEtc->cfFormat, DbgGetCF(pFormatEtc->cfFormat));
  249.       ATLTRACE(_T("TYMED = 0x%04xn"), pFormatEtc->tymed);
  251.       if( pFormatEtc->cfFormat == _Module.m_CFSTR_PASTESUCCEEDED ) {
  252.          HR( _SetDWORD(pMedium, fRelease, m_dwPasteSucceeded) );
  253.          // To handle Delete-on-Paste, we check this guy...
  254.          if( m_dwPasteSucceeded==DROPEFFECT_MOVE ) {
  255.             // If we all agreed that it was a 'move', then we're
  256.             // responsible for deleting the files.
  257.             if( m_dwPerformedDropEffect==DROPEFFECT_MOVE ) {
  258.                m_pFolder->_DeleteFiles(m_pidls, m_pidls.GetCount());
  259.                ::OleSetClipboard(NULL);
  260.             }
  261.             // At least we should refresh the view
  262.             m_pFolder->_SendViewMessage(m_hWnd, WM_COMMAND, ID_VIEW_REFRESH);
  263.          }
  264.          return S_OK;
  265.       }
  266.       else if( pFormatEtc->cfFormat == _Module.m_CFSTR_PERFORMEDDROPEFFECT ) {
  267.          HR( _SetDWORD(pMedium, fRelease, m_dwPerformedDropEffect) );
  268.          return S_OK;
  269.       }
  270.       else if( pFormatEtc->cfFormat == _Module.m_CFSTR_PREFERREDDROPEFFECT ) {
  271.          HR( _SetDWORD(pMedium, fRelease, m_dwPreferredDropEffect) );
  272.          return S_OK;
  273.       }
  274.       return DV_E_FORMATETC;
  275.    }
  277.    STDMETHOD(GetCanonicalFormatEtc)(FORMATETC*, FORMATETC*)
  278.    {
  279.       ATLTRACENOTIMPL(_T("CDataObject::GetCanonicalFormatEtc"));
  280.    }
  282.    STDMETHOD(DAdvise)(FORMATETC *, DWORD, IAdviseSink *, DWORD *)
  283.    {
  284.       ATLTRACENOTIMPL(_T("CDataObject::DAdvise"));
  285.    }
  287.    STDMETHOD(DUnadvise)(DWORD)
  288.    {
  289.       ATLTRACENOTIMPL(_T("CDataObject::DUnadvise"));
  290.    }
  292.    STDMETHOD(EnumDAdvise)(IEnumSTATDATA **)
  293.    {
  294.       ATLTRACENOTIMPL(_T("CDataObject::EnumDAdvise"));
  295.    }
  297.    // Implementation
  299.    HRESULT _SetDWORD(STGMEDIUM *pMedium, BOOL fRelease, DWORD &dwData)
  300.    {
  301.       ATLASSERT(pMedium);
  302.       ATLASSERT(pMedium->hGlobal);
  303.       LPDWORD p = (LPDWORD)::GlobalLock(pMedium->hGlobal);
  304.       ATLASSERT(p);
  305.       if( p==NULL ) return E_OUTOFMEMORY;
  306.       dwData = *p;
  307.       ::GlobalUnlock(pMedium->hGlobal);
  308.       if( fRelease ) ::ReleaseStgMedium(pMedium);
  309.       return S_OK;
  310.    }
  312.    HRESULT _GetDWORD(STGMEDIUM *pMedium, DWORD dwData)
  313.    {
  314.       ATLASSERT(pMedium);
  315.       HGLOBAL hMem = ::GlobalAlloc(GMEM_MOVEABLE | GMEM_SHARE | GMEM_DISCARDABLE, sizeof(DWORD));
  316.       if( hMem==NULL ) return E_OUTOFMEMORY;
  317.       LPDWORD p = (LPDWORD)::GlobalLock(hMem);
  318.       ATLASSERT(p);
  319.       *p = dwData;
  320.       ::GlobalUnlock(hMem);         
  321.       pMedium->tymed = TYMED_HGLOBAL;
  322.       pMedium->hGlobal = hMem;
  323.       pMedium->pUnkForRelease = NULL;
  324.       return S_OK;
  325.    }
  326. };
  329. /////////////////////////////////////////////////////////////////////////////
  330. // IDataObject helper functions
  332. ATLINLINE HRESULT DataObj_SetDWORD(IDataObject *pDataObj, CLIPFORMAT cf, DWORD dw)
  333. {
  334.    ATLASSERT(pDataObj);
  335.    HRESULT Hr = E_OUTOFMEMORY;
  336.    LPDWORD pdw = (LPDWORD)::GlobalAlloc(GPTR, sizeof(DWORD));
  337.    if( pdw ) {
  338.       FORMATETC fmte = { cf, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
  339.       STGMEDIUM medium;
  340.       *pdw = dw;       
  341.       medium.tymed = TYMED_HGLOBAL;
  342.       medium.hGlobal = pdw;
  343.       medium.pUnkForRelease = NULL;
  344.       Hr = pDataObj->SetData(&fmte, &medium, TRUE); 
  345.       if( FAILED(Hr) ) ::GlobalFree((HGLOBAL)pdw);
  346.    }
  347.    return Hr;
  348. }
  350. ATLINLINE HRESULT DataObj_GetDWORD(IDataObject *pDataObj, CLIPFORMAT cf, LPDWORD pdwOut)
  351. {
  352.    ATLASSERT(pDataObj);
  353.    ATLASSERT(pdwOut);
  354.    FORMATETC fmte = { cf, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
  355.    STGMEDIUM medium;
  356.    HRESULT Hr = pDataObj->GetData(&fmte, &medium);
  357.    if( SUCCEEDED(Hr) ) {
  358.       LPDWORD pdw = (LPDWORD)::GlobalLock(medium.hGlobal);
  359.       if( pdw ) {
  360.          *pdwOut = *pdw;
  361.          ::GlobalUnlock(medium.hGlobal);
  362.       }   
  363.       else {
  364.          Hr = E_UNEXPECTED;
  365.       }
  366.       ::ReleaseStgMedium(&medium);
  367.    }
  368.    return Hr;
  369. }
  372. #endif // !defined(AFX_DATAOBJECT_H__20001229_075F_904E_7D1B_0080AD509054__INCLUDED_)