embedding/qa/testembed/PrintProgressDialog.cpp

   1 // PrintProgressDialog.cpp : implementation file
   2 //
   3
   4 #include "stdafx.h"
   5 #include "testembed.h"
   6 #include "PrintProgressDialog.h"
   7 #include "BrowserView.h"
   8 #include "nsIWebBrowser.h"
   9 #include "nsIWebBrowserPrint.h"
  10
  11 #ifdef _DEBUG
  12 #define new DEBUG_NEW
  13 #undef THIS_FILE
  14 static char THIS_FILE[] = __FILE__;
  15 #endif
  16
  17 /////////////////////////////////////////////////////////////////////////////
  18 // CPrintProgressDialog dialog
  19
  20 class CDlgPrintListener : public nsIWebProgressListener
  21 {
  22 // Construction
  23 public:
  24         CDlgPrintListener(CPrintProgressDialog* aDlg);
  25
  26   NS_DECL_ISUPPORTS
  27   NS_DECL_NSIWEBPROGRESSLISTENER
  28
  29   void ClearDlg() { m_PrintDlg = NULL; } // weak reference
  30
  31 // Implementation
  32 protected:
  33         CPrintProgressDialog* m_PrintDlg;
  34 };
  35
  36 NS_IMPL_ADDREF(CDlgPrintListener)
  37 NS_IMPL_RELEASE(CDlgPrintListener)
  38
  39 NS_INTERFACE_MAP_BEGIN(CDlgPrintListener)
  40     NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIWebProgressListener)
  41     NS_INTERFACE_MAP_ENTRY(nsIWebProgressListener)
  42 NS_INTERFACE_MAP_END
  43
  44
  45 CDlgPrintListener::CDlgPrintListener(CPrintProgressDialog* aDlg) :
  46   m_PrintDlg(aDlg)
  47 {
  48   //NS_ADDREF_THIS();
  49 }
  50
  51 /* void onStateChange (in nsIWebProgress aWebProgress, in nsIRequest aRequest, in unsigned long aStateFlags, in nsresult aStatus); */
  52 NS_IMETHODIMP
  53 CDlgPrintListener::OnStateChange(nsIWebProgress *aWebProgress, nsIRequest *aRequest, PRUint32 aStateFlags, nsresult aStatus)
  54 {
  55   if (m_PrintDlg) {
  56     if (aStateFlags == (nsIWebProgressListener::STATE_START|nsIWebProgressListener::STATE_IS_DOCUMENT)) {
  57       return m_PrintDlg->OnStartPrinting();
  58
  59     } else if (aStateFlags == (nsIWebProgressListener::STATE_STOP|nsIWebProgressListener::STATE_IS_DOCUMENT)) {
  60       return m_PrintDlg->OnEndPrinting(aStatus);
  61     }
  62   }
  63   return NS_OK;
  64 }
  65
  66 /* void onProgressChange (in nsIWebProgress aWebProgress, in nsIRequest aRequest, in long aCurSelfProgress, in long aMaxSelfProgress, in long aCurTotalProgress, in long aMaxTotalProgress); */
  67 NS_IMETHODIMP
  68 CDlgPrintListener::OnProgressChange(nsIWebProgress *aWebProgress, nsIRequest *aRequest, PRInt32 aCurSelfProgress, PRInt32 aMaxSelfProgress, PRInt32 aCurTotalProgress, PRInt32 aMaxTotalProgress)
  69 {
  70   if (m_PrintDlg) {
  71     return m_PrintDlg->OnProgressPrinting(aCurSelfProgress, aMaxSelfProgress);
  72   }
  73   return NS_OK;
  74 }
  75
  76 /* void onLocationChange (in nsIWebProgress aWebProgress, in nsIRequest aRequest, in nsIURI location); */
  77 NS_IMETHODIMP
  78 CDlgPrintListener::OnLocationChange(nsIWebProgress *aWebProgress, nsIRequest *aRequest, nsIURI *location)
  79 {
  80     return NS_OK;
  81 }
  82
  83 /* void onStatusChange (in nsIWebProgress aWebProgress, in nsIRequest aRequest, in nsresult aStatus, in wstring aMessage); */
  84 NS_IMETHODIMP
  85 CDlgPrintListener::OnStatusChange(nsIWebProgress *aWebProgress, nsIRequest *aRequest, nsresult aStatus, const PRUnichar *aMessage)
  86 {
  87     return NS_OK;
  88 }
  89
  90 /* void onSecurityChange (in nsIWebProgress aWebProgress, in nsIRequest aRequest, in unsigned long state); */
  91 NS_IMETHODIMP
  92 CDlgPrintListener::OnSecurityChange(nsIWebProgress *aWebProgress, nsIRequest *aRequest, PRUint32 state)
  93 {
  94     return NS_OK;
  95 }
  96
  97
  98 /////////////////////////////////////////////////////////////////////////////
  99 // CPrintProgressDialog dialog
 100
 101
 102 CPrintProgressDialog::CPrintProgressDialog(nsIWebBrowser* aWebBrowser,
 103                                            nsIPrintSettings* aPrintSettings,
 104                                            CWnd* pParent /*=NULL*/)
 105         : CDialog(CPrintProgressDialog::IDD, pParent),
 106   m_WebBrowser(aWebBrowser),
 107   m_PrintListener(nsnull),
 108   m_PrintSettings(aPrintSettings),
 109   m_InModalMode(PR_FALSE)
 110 {
 111         //{{AFX_DATA_INIT(CPrintProgressDialog)
 112                 // NOTE: the ClassWizard will add member initialization here
 113         //}}AFX_DATA_INIT
 114 }
 115
 116 CPrintProgressDialog::~CPrintProgressDialog()
 117 {
 118   CDlgPrintListener * pl = (CDlgPrintListener*)m_PrintListener.get();
 119   if (pl) {
 120     pl->ClearDlg();
 121   }
 122 }
 123
 124
 125 void CPrintProgressDialog::DoDataExchange(CDataExchange* pDX)
 126 {
 127         CDialog::DoDataExchange(pDX);
 128         //{{AFX_DATA_MAP(CPrintProgressDialog)
 129                 // NOTE: the ClassWizard will add DDX and DDV calls here
 130         //}}AFX_DATA_MAP
 131 }
 132
 133
 134 BEGIN_MESSAGE_MAP(CPrintProgressDialog, CDialog)
 135         //{{AFX_MSG_MAP(CPrintProgressDialog)
 136         //}}AFX_MSG_MAP
 137 END_MESSAGE_MAP()
 138
 139 /////////////////////////////////////////////////////////////////////////////
 140 // CPrintProgressDialog message handlers
 141 static void GetLocalRect(CWnd * aWnd, CRect& aRect, CWnd * aParent)
 142 {
 143   CRect wr;
 144   aParent->GetWindowRect(wr);
 145
 146   CRect cr;
 147   aParent->GetClientRect(cr);
 148
 149   aWnd->GetWindowRect(aRect);
 150
 151   int borderH = wr.Height() - cr.Height();
 152   int borderW = (wr.Width() - cr.Width())/2;
 153   aRect.top    -= wr.top+borderH-borderW;
 154   aRect.left   -= wr.left+borderW;
 155   aRect.right  -= wr.left+borderW;
 156   aRect.bottom -= wr.top+borderH-borderW;
 157
 158 }
 159
 160 BOOL CPrintProgressDialog::OnInitDialog()
 161 {
 162         CDialog::OnInitDialog();
 163
 164         CRect clientRect;
 165         GetClientRect(&clientRect);
 166
 167         CRect titleRect;
 168   GetLocalRect(GetDlgItem(IDC_PPD_DOC_TITLE_STATIC), titleRect, this);
 169
 170         CRect itemRect;
 171   GetLocalRect(GetDlgItem(IDC_PPD_DOC_TXT), itemRect, this);
 172
 173   CRect progRect;
 174   progRect.left   = titleRect.left;
 175   progRect.top    = itemRect.top+itemRect.Height()+5;
 176   progRect.right  = clientRect.Width()-(2*titleRect.left);
 177   progRect.bottom = progRect.top+titleRect.Height();
 178
 179
 180         m_wndProgress.Create (WS_CHILD | WS_VISIBLE, progRect, this, -1);
 181         m_wndProgress.SetPos (0);
 182
 183         return TRUE;  // return TRUE unless you set the focus to a control
 184                       // EXCEPTION: OCX Property Pages should return FALSE
 185 }
 186
 187
 188 int CPrintProgressDialog::DoModal( )
 189 {
 190   PRBool doModal = PR_FALSE;
 191   nsCOMPtr<nsIWebBrowserPrint> print(do_GetInterface(m_WebBrowser));
 192   if(print)
 193   {
 194     m_PrintListener = new CDlgPrintListener(this); // constructor addrefs
 195     if (m_PrintListener) {
 196       // doModal will be set to false if the print job was cancelled
 197       nsIWebProgressListener * wpl = static_cast<nsIWebProgressListener*>(m_PrintListener);
 198       doModal = NS_SUCCEEDED(print->Print(m_PrintSettings, wpl)) == PR_TRUE;
 199     }
 200   }
 201
 202   if (doModal) {
 203     m_InModalMode = PR_TRUE;
 204     return CDialog::DoModal();
 205   }
 206   return 0;
 207 }
 208
 209
 210 /* void OnStartPrinting (); */
 211 NS_IMETHODIMP
 212 CPrintProgressDialog::OnStartPrinting()
 213 {
 214   return NS_OK;
 215 }
 216
 217 /* void OnProgressPrinting (in PRUint32 aProgress, in PRUint32 aProgressMax); */
 218 NS_IMETHODIMP
 219 CPrintProgressDialog::OnProgressPrinting(PRUint32 aProgress, PRUint32 aProgressMax)
 220 {
 221   if (m_wndProgress.m_hWnd == NULL) return NS_OK;
 222
 223   // Initialize the progress meter we we get the "zero" progress
 224   // which also tells us the max progress
 225   if (aProgress == 0) {
 226           CWnd *pWnd = GetDlgItem(IDC_PPD_DOC_TXT);
 227           if(pWnd)
 228                   pWnd->SetWindowText(m_URL);
 229
 230           m_wndProgress.SetRange(0, aProgressMax);
 231     m_wndProgress.SetPos(0);
 232   }
 233         m_wndProgress.SetPos(aProgress);
 234   RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
 235
 236   return NS_OK;
 237 }
 238
 239 /* void OnEndPrinting (in PRUint32 aStatus); */
 240 NS_IMETHODIMP
 241 CPrintProgressDialog::OnEndPrinting(PRUint32 aStatus)
 242 {
 243   // Here we need to know whether we have gone "modal"
 244   // because we could get notified here if the user cancels
 245   // before we ever get a chance to go into the modal loop
 246   if (m_InModalMode) {
 247     EndDialog(1);
 248   }
 249   return NS_OK;
 250 }
 251
 252 void CPrintProgressDialog::OnCancel()
 253 {
 254   nsCOMPtr<nsIWebBrowserPrint> print(do_GetInterface(m_WebBrowser));
 255   if (print) {
 256     print->Cancel();
 257   }
 258
 259         CDialog::OnCancel();
 260 }
 261
 262 void CPrintProgressDialog::SetURI(const char* aTitle)
 263 {
 264   m_URL = _T(aTitle);
 265 }