DOCUMENT.CPP

/* 
* DOCUMENT.CPP
* Component Cosmo Chapter 10
*
* Implementation of the CCosmoDoc derivation of CDocument
*
* Copyright (c)1993-1995 Microsoft Corporation, All Rights Reserved
*
* Kraig Brockschmidt, Microsoft
* Internet : kraigb@microsoft.com
* Compuserve: >INTERNET:kraigb@microsoft.com
*/


#include "cocosmo.h"


/*
* CCosmoDoc::CCosmoDoc
* CCosmoDoc::~CCosmoDoc
*
* Constructor Parameters:
* hInst HINSTANCE of the application.
* pFR PCFrame of the frame object.
* pAdv PCDocumentAdviseSink to notify on events
*/

CCosmoDoc::CCosmoDoc(HINSTANCE hInst, PCFrame pFR
, PCDocumentAdviseSink pAdv)
: CDocument(hInst, pFR, pAdv)
{
m_pPL=NULL;
m_pPLAdv=NULL;
m_uPrevSize=SIZE_RESTORED;
m_pIConnectPt=NULL;
m_dwCookie=0;

m_pIStorage=NULL;
m_pp.psModel=PERSIST_UNKNOWN;
m_pp.pIP.pIPersistStorage=NULL; //Affects all pointers

//CHAPTER10MOD
m_pImpIAdviseSink=NULL;
m_dwConn=0;
//End CHAPTER10MOD
return;
}


CCosmoDoc::~CCosmoDoc(void)
{
//CHAPTER10MOD
LPDATAOBJECT pIDataObject;
HRESULT hr;

//Turn off the advise.
if (NULL!=m_pPL && 0!=m_dwConn)
{
hr=m_pPL->QueryInterface(IID_IDataObject
, (PPVOID)&pIDataObject);

if (SUCCEEDED(hr))
{
pIDataObject->DUnadvise(m_dwConn);
pIDataObject->Release();
}
}

DeleteInterfaceImp(m_pImpIAdviseSink);
//End CHAPTER10MOD

ReleaseInterface(m_pp.pIP.pIPersistStorage);
ReleaseInterface(m_pIStorage);

if (NULL!=m_pIConnectPt)
{
m_pIConnectPt->Unadvise(m_dwCookie);
ReleaseInterface(m_pIConnectPt);
}

ReleaseInterface(m_pPL);
ReleaseInterface(m_pPLAdv);

CoFreeUnusedLibraries();
return;
}






/*
* CCosmoDoc::Init
*
* Purpose:
* Initializes an already created document window. The client
* actually creates the window for us, then passes that here for
* further initialization.
*
* Parameters:
* pDI PDOCUMENTINIT containing initialization
* parameters.
*
* Return Value:
* BOOL TRUE if the function succeeded, FALSE otherwise.
*/

BOOL CCosmoDoc::Init(PDOCUMENTINIT pDI)
{
RECT rc;
HRESULT hr;
IConnectionPointContainer *pCPC;
//CHAPTER10MOD
FORMATETC fe;
LPDATAOBJECT pIDataObject;
//End CHAPTER10MOD

//Change the stringtable range to our customization.
pDI->idsMin=IDS_DOCUMENTMIN;
pDI->idsMax=IDS_DOCUMENTMAX;

//Do default initialization
if (!CDocument::Init(pDI))
return FALSE;

//CHAPTER10MOD
//Create the Polyline Component
hr=CoCreateInstance(CLSID_Polyline10, NULL, CLSCTX_INPROC_SERVER
, IID_IPolyline10, (PPVOID)&m_pPL);
//End CHAPTER10MOD

if (FAILED(hr))
{
//Warn that we could not load the Polyline
MessageBox(pDI->hWndDoc, PSZ(IDS_NOPOLYLINE)
, PSZ(IDS_CAPTION), MB_OK);
return FALSE;
}

//Initialize the contained Polyline which creates a window.
GetClientRect(m_hWnd, &rc);
InflateRect(&rc, -8, -8);

if (FAILED(m_pPL->Init(m_hWnd, &rc, WS_CHILD | WS_VISIBLE
, ID_POLYLINE)))
return FALSE;

//Set up an advise on the Polyline.
m_pPLAdv=new CPolylineAdviseSink(this);
m_pPLAdv->AddRef();

if (SUCCEEDED(m_pPL->QueryInterface(IID_IConnectionPointContainer
, (PPVOID)&pCPC)))
{
//CHAPTER10MOD
if (SUCCEEDED(pCPC->FindConnectionPoint
(IID_IPolylineAdviseSink10, &m_pIConnectPt)))
//End CHAPTER10MOD
{
m_pIConnectPt->Advise((LPUNKNOWN)m_pPLAdv, &m_dwCookie);
}

pCPC->Release();
}

/*
* Check for the storage model being used in Polyline.
* Any InitNew member will be called from CCosmoDoc::Load.
*/
hr=m_pPL->QueryInterface(IID_IPersistStorage
, (PPVOID)&m_pp.pIP.pIPersistStorage);

if (SUCCEEDED(hr))
m_pp.psModel=PERSIST_STORAGE;
else
{
hr=m_pPL->QueryInterface(IID_IPersistStreamInit
, (PPVOID)&m_pp.pIP.pIPersistStreamInit);

if (SUCCEEDED(hr))
m_pp.psModel=PERSIST_STREAMINIT;
else
{
hr=m_pPL->QueryInterface(IID_IPersistStream
, (PPVOID)&m_pp.pIP.pIPersistStream);

if (SUCCEEDED(hr))
m_pp.psModel=PERSIST_STREAM;
else
return FALSE;
}
}


//CHAPTER10MOD
/*
* Create an IAdviseSink and send it to the Polyline's
* IDataObject with the clipboard format for the Polyline
* (as in IPOLY10.H).
*/

//This is a private macro.
SETDefFormatEtc(fe, m_cf, TYMED_HGLOBAL);

m_pImpIAdviseSink=new CImpIAdviseSink(this, this);

if (NULL==m_pImpIAdviseSink)
return FALSE;

//Set up an advise for the Polyline format
hr=m_pPL->QueryInterface(IID_IDataObject, (PPVOID)&pIDataObject);

if (FAILED(hr))
return FALSE;

pIDataObject->DAdvise(&fe, ADVF_NODATA, m_pImpIAdviseSink
, &m_dwConn);
pIDataObject->Release();
//End CHAPTER10MOD

return TRUE;
}




//CHAPTER10MOD
//IUnknown interface for all others we implement in the document

/*
* CCosmoDoc::QueryInterface
* CCosmoDoc::AddRef
* CCosmoDoc::Release
*
* Purpose:
* IUnknown members for the CCosmoDoc implementation.
*/

STDMETHODIMP CCosmoDoc::QueryInterface(REFIID riid, PPVOID ppv)
{
*ppv=NULL;

//The document is the unknown
if (IID_IUnknown==riid)
*ppv=(LPUNKNOWN)this;

//CHAPTER10MOD
if (IID_IPolylineAdviseSink10==riid)
//End CHAPTER10MOD
*ppv=m_pPLAdv;

if (IID_IAdviseSink==riid)
*ppv=m_pImpIAdviseSink;

if (NULL!=*ppv)
{
((LPUNKNOWN)*ppv)->AddRef();
return NOERROR;
}

return ResultFromScode(S_FALSE);
}


STDMETHODIMP_(ULONG) CCosmoDoc::AddRef(void)
{
return ++m_cRef;
}


STDMETHODIMP_(ULONG) CCosmoDoc::Release(void)
{
/*
* The client destroys documents when CoCosmo terminates,
* so we don't need to destroy anything here.
*/
return --m_cRef;
}

//End CHAPTER10MOD






/*
* CCosmoDoc::FMessageHook
*
* Purpose:
* Processes WM_SIZE for the document so we can resize
* the Polyline.
*
* Parameters:
* <WndProc Parameters>
* pLRes LRESULT * in which to store the return
* value for the message.
*
* Return Value:
* BOOL TRUE to prevent further processing,
* FALSE otherwise.
*/

BOOL CCosmoDoc::FMessageHook(HWND hWnd, UINT iMsg, WPARAM wParam
, LPARAM lParam, LRESULT *pLRes)
{
UINT dx, dy;
RECT rc;

*pLRes=0;

if (WM_SIZE==iMsg)
{
//Don't effect the Polyline size to or from minimized state.
if (SIZE_MINIMIZED!=wParam && SIZE_MINIMIZED !=m_uPrevSize)
{
//When we change size, resize any Polyline we hold.
dx=LOWORD(lParam);
dy=HIWORD(lParam);

/*
* If we are getting WM_SIZE in response to a Polyline
* notification, then don't resize the Polyline window
* again.
*/
if (!m_fNoSize && NULL!=m_pPL)
{
//Resize the polyline to fit the new client
SetRect(&rc, 8, 8, dx-8, dy-8);
m_pPL->RectSet(&rc, FALSE);

/*
* We consider sizing something that makes the file
* dirty, but not until we've finished the create
* process, which is why we set fNoDirty to FALSE
* in WM_CREATE since we get a WM_SIZE on the first
* creation.
*/
if (!m_fNoDirty)
FDirtySet(TRUE);

SetRect(&rc, 0, 0, dx, dy);

if (NULL!=m_pAdv)
m_pAdv->OnSizeChange(this, &rc);

m_fNoDirty=FALSE;
}
}

m_uPrevSize=wParam;
}

/*
* We return FALSE even on WM_SIZE so we can let the default
* procedure handle maximized MDI child windows appropriately.
*/
return FALSE;
}








/*
* CCosmoDoc::Clear
*
* Purpose:
* Sets all contents in the document back to defaults with
* no filename.
*
* Paramters:
* None
*
* Return Value:
* None
*/

void CCosmoDoc::Clear(void)
{
//Completely reset the polyline
m_pPL->New();

CDocument::Clear();
return;
}






/*
* CCosmoDoc::Load
*
* Purpose:
* Loads a given document without any user interface overwriting
* the previous contents of the Polyline window. We do this by
* opening the file and telling the Polyline to load itself from
* that file.
*
* Parameters:
* fChangeFile BOOL indicating if we're to update the window
* title and the filename from using this file.
* pszFile LPTSTR to the filename to load, NULL if the file
* is new and untitled.
*
* Return Value:
* UINT An error value from DOCERR_*
*/

UINT CCosmoDoc::Load(BOOL fChangeFile, LPTSTR pszFile)
{
HRESULT hr;
LPSTORAGE pIStorage;

if (NULL==pszFile)
{
/*
* As a user of an IPersistStorage we have to provide all
* objects with an IStorage they can use for incremental
* access passing that storage to InitNew. Here we create
* a temporary file that we don't bother holding on to.
* If the object doesn't use it, then our Release destroys
* it immediately.
*/

hr=StgCreateDocfile(NULL, STGM_DIRECT | STGM_READWRITE
| STGM_CREATE | STGM_DELETEONRELEASE
| STGM_SHARE_EXCLUSIVE, 0, &pIStorage);

if (FAILED(hr))
return DOCERR_COULDNOTOPEN;

if (PERSIST_STORAGE==m_pp.psModel)
m_pp.pIP.pIPersistStorage->InitNew(pIStorage);
else
{
if (PERSIST_STREAMINIT==m_pp.psModel)
m_pp.pIP.pIPersistStreamInit->InitNew();
}

m_pIStorage=pIStorage;

Rename(NULL);
return DOCERR_NONE;
}

/*
* Open a storage and have the Polyline read its data
* using whatever persistence model it employs.
*/
hr=StgOpenStorage(pszFile, NULL, STGM_DIRECT | STGM_READWRITE
| STGM_SHARE_EXCLUSIVE, NULL, 0, &pIStorage);

if (FAILED(hr))
return DOCERR_COULDNOTOPEN;

if (PERSIST_STORAGE==m_pp.psModel)
hr=m_pp.pIP.pIPersistStorage->Load(pIStorage);
else
{
LPSTREAM pIStream;

hr=pIStorage->OpenStream(SZSTREAM, 0, STGM_DIRECT
| STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &pIStream);

if (SUCCEEDED(hr))
{
//This also works for PERSIST_STREAMINIT
hr=m_pp.pIP.pIPersistStream->Load(pIStream);
pIStream->Release();
}
}

m_pIStorage=pIStorage;

if (FAILED(hr))
return DOCERR_READFAILURE;

if (fChangeFile)
Rename(pszFile);

//Importing a file makes things dirty
FDirtySet(!fChangeFile);

return DOCERR_NONE;
}







/*
* CCosmoDoc::Save
*
* Purpose:
* Writes the file to a known filename, requiring that the user
* has previously used FileOpen or FileSaveAs in order to have
* a filename.
*
* Parameters:
* uType UINT indicating the type of file the user
* requested to save in the File Save As dialog.
* pszFile LPTSTR under which to save. If NULL, use the
* current name.
*
* Return Value:
* UINT An error value from DOCERR_*
*/

UINT CCosmoDoc::Save(UINT uType, LPTSTR pszFile)
{
BOOL fRename=TRUE;
HRESULT hr;
LPSTORAGE pIStorage;
BOOL fSameAsLoad;

//If Save or Save As under the same name, do Save.
if (NULL==pszFile || 0==lstrcmpi(pszFile, m_szFile))
{
fRename=FALSE;
pszFile=m_szFile;

/*
* If we're saving to an existing storage, just pass
* the IStorage we have from Load along with TRUE
* in fSameAsLoad.
*/

fSameAsLoad=TRUE;
}
else
{
/*
* In Component Cosmo, we only deal with one version of
* data; all the code in Chapter 1 Cosmo that dealt with
* 1.0 and 2.0 files has been removed.
*/

hr=StgCreateDocfile(pszFile, STGM_DIRECT | STGM_READWRITE
| STGM_CREATE | STGM_SHARE_EXCLUSIVE, 0, &pIStorage);

if (FAILED(hr))
return DOCERR_COULDNOTOPEN;

//Tell the object to save into this new storage
fSameAsLoad=FALSE;

//Update our variable
m_pIStorage->Release();
m_pIStorage=pIStorage;
}

if (PERSIST_STORAGE==m_pp.psModel)
{
hr=m_pp.pIP.pIPersistStorage->Save(m_pIStorage, fSameAsLoad);

if (SUCCEEDED(hr))
{
hr=m_pp.pIP.pIPersistStorage->SaveCompleted(fSameAsLoad
? NULL : m_pIStorage);
}
}
else
{
LPSTREAM pIStream;

hr=m_pIStorage->CreateStream(SZSTREAM, STGM_DIRECT
| STGM_CREATE | STGM_WRITE | STGM_SHARE_EXCLUSIVE
, 0, 0, &pIStream);

if (SUCCEEDED(hr))
{
//This also works for PERSIST_STREAMINIT
hr=m_pp.pIP.pIPersistStream->Save(pIStream, TRUE);
pIStream->Release();
}
}


if (FAILED(hr))
return DOCERR_WRITEFAILURE;

//Saving makes us clean
FDirtySet(FALSE);

if (fRename)
Rename(pszFile);

return DOCERR_NONE;
}






/*
* CCosmoDoc::Undo
*
* Purpose:
* Reverses a previous action.
*
* Parameters:
* None
*
* Return Value:
* None
*/

void CCosmoDoc::Undo(void)
{
m_pPL->Undo();
return;
}






/*
* CCosmoDoc::Clip
*
* Purpose:
* Places a private format, a metafile, and a bitmap of the display
* on the clipboard, optionally implementing Cut by deleting the
* data in the current window after rendering.
*
* Parameters:
* hWndFrame HWND of the main window.
* fCut BOOL indicating cut (TRUE) or copy (FALSE).
*
* Return Value:
* BOOL TRUE if successful, FALSE otherwise.
*/

BOOL CCosmoDoc::Clip(HWND hWndFrame, BOOL fCut)
{
BOOL fRet=TRUE;
HGLOBAL hMem;
UINT i;

//This array is so we can loop over the formats we provide.
static UINT rgcf[3]={0, CF_METAFILEPICT, CF_BITMAP};
const UINT cFormats=3;

if (!OpenClipboard(hWndFrame))
return FALSE;

//Clean out whatever junk is in the clipboard.
EmptyClipboard();

rgcf[0]=m_cf;

for (i=0; i < cFormats; i++)
{
//Copy private data first.
hMem=RenderFormat(rgcf[i]);

if (NULL!=hMem)
SetClipboardData(rgcf[i], hMem);
else
fRet &=FALSE;
}

//Free clipboard ownership.
CloseClipboard();

//Delete our current data if copying succeeded.
if (fRet && fCut)
{
m_pPL->New();
FDirtySet(TRUE);
}

return fRet;
}





/*
* CCosmoDoc::RenderFormat
*
* Purpose:
* Renders a specific clipboard format into global memory.
*
* Parameters:
* cf UINT format to render.
*
* Return Value:
* HGLOBAL Global memory handle containing the data.
*/

HGLOBAL CCosmoDoc::RenderFormat(UINT cf)
{
//CHAPTER10MOD
LPDATAOBJECT pIDataObject;
FORMATETC fe;
STGMEDIUM stm;
HRESULT hr;

hr=m_pPL->QueryInterface(IID_IDataObject, (PPVOID)&pIDataObject);

if (FAILED(hr))
return NULL;

//Private macro
SETDefFormatEtc(fe, cf
, TYMED_HGLOBAL | TYMED_GDI | TYMED_MFPICT);

pIDataObject->GetData(&fe, &stm);
pIDataObject->Release();

return stm.hGlobal;
//End CHAPTER10MOD
}







/*
* CCosmoDoc::FQueryPaste
*
* Purpose:
* Determines if we can paste data from the clipboard.
*
* Parameters:
* None
*
* Return Value:
* BOOL TRUE if data is available, FALSE otherwise.
*/

BOOL CCosmoDoc::FQueryPaste(void)
{
return IsClipboardFormatAvailable(m_cf);
}





/*
* CCosmoDoc::Paste
*
* Purpose:
* Retrieves the private data format from the clipboard and sets it
* to the current figure in the editor window.
*
* Note that if this function is called, then the clipboard format
* is available because the Paste menu item is only enabled if the
* format is present.
*
* Parameters:
* hWndFrame HWND of the main window.
*
* Return Value:
* BOOL TRUE if successful, FALSE otherwise.
*/

BOOL CCosmoDoc::Paste(HWND hWndFrame)
{
HGLOBAL hMem;
BOOL fRet=FALSE;
//CHAPTER10MOD
LPDATAOBJECT pIDataObject;
HRESULT hr;
FORMATETC fe;
STGMEDIUM stm;
//End CHAPTER10MOD

if (!OpenClipboard(hWndFrame))
return FALSE;

hMem=GetClipboardData(m_cf);
CloseClipboard();

if (NULL!=hMem)
{
//CHAPTER10MOD
hr=m_pPL->QueryInterface(IID_IDataObject
, (PPVOID)&pIDataObject);

if (FAILED(hr))
{
GlobalFree(hMem);
return FALSE;
}

SETDefFormatEtc(fe, m_cf, TYMED_HGLOBAL);

stm.tymed=TYMED_HGLOBAL;
stm.hGlobal=hMem;
stm.pUnkForRelease=NULL;

pIDataObject->SetData(&fe, &stm, FALSE);
pIDataObject->Release();

FDirtySet(TRUE);
//End CHAPTER10MOD
}

return TRUE;
}








/*
* CCosmoDoc::ColorSet
*
* Purpose:
* Changes a color used in our contained Polyline.
*
* Parameters:
* iColor UINT index of the color to change.
* cr COLORREF new color.
*
* Return Value:
* COLORREF Previous color for the given index.
*/

COLORREF CCosmoDoc::ColorSet(UINT iColor, COLORREF cr)
{
COLORREF crRet;

m_pPL->ColorSet(iColor, cr, &crRet);
return crRet;
}





/*
* CCosmoDoc::ColorGet
*
* Purpose:
* Retrieves a color currently in use in the Polyline.
*
* Parameters:
* iColor UINT index of the color to retrieve.
*
* Return Value:
* COLORREF Current color for the given index.
*/

COLORREF CCosmoDoc::ColorGet(UINT iColor)
{
COLORREF crRet;

m_pPL->ColorGet(iColor, &crRet);
return crRet;
}






/*
* CCosmoDoc::LineStyleSet
*
* Purpose:
* Changes the line style currently used in the Polyline
*
* Parameters:
* iStyle UINT index of the new line style to use.
*
* Return Value:
* UINT Previous line style.
*/

UINT CCosmoDoc::LineStyleSet(UINT iStyle)
{
UINT i;

m_pPL->LineStyleSet(iStyle, &i);
return i;
}







/*
* CCosmoDoc::LineStyleGet
*
* Purpose:
* Retrieves the line style currently used in the Polyline
*
* Parameters:
* None
*
* Return Value:
* UINT Current line style.
*/


UINT CCosmoDoc::LineStyleGet(void)
{
UINT i=0;

//m_pPL might not be valid yet.
if (NULL!=m_pPL)
m_pPL->LineStyleGet(&i);

return i;
}



//CHAPTER10MOD
//CPolylineAdviseSink moved to IADVSINK.CPP
//End CHAPTER10MOD