src/richtext/richtextprint.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/richtext/richtextprint.cpp
   3 // Purpose:     Rich text printing classes
   4 // Author:      Julian Smart
   5 // Created:     2006-10-24
   6 // RCS-ID:      $Id$
   7 // Copyright:   (c) Julian Smart
   8 // Licence:     wxWindows licence
   9 /////////////////////////////////////////////////////////////////////////////
  10
  11 // For compilers that support precompilation, includes "wx/wx.h".
  12 #include "wx/wxprec.h"
  13
  14 #ifdef __BORLANDC__
  15     #pragma hdrstop
  16 #endif
  17
  18 #if wxUSE_RICHTEXT && wxUSE_PRINTING_ARCHITECTURE && wxUSE_STREAMS
  19
  20 #ifndef WX_PRECOMP
  21     #include "wx/log.h"
  22     #include "wx/intl.h"
  23     #include "wx/dc.h"
  24     #include "wx/settings.h"
  25     #include "wx/msgdlg.h"
  26 #endif
  27
  28 #include "wx/datetime.h"
  29 #include "wx/print.h"
  30 #include "wx/printdlg.h"
  31 #include "wx/richtext/richtextprint.h"
  32 #include "wx/wfstream.h"
  33
  34 /*!
  35  * wxRichTextPrintout
  36  */
  37
  38 wxRichTextPrintout::wxRichTextPrintout(const wxString& title) : wxPrintout(title)
  39 {
  40     m_numPages = wxRICHTEXT_PRINT_MAX_PAGES;
  41
  42     SetMargins(); // to default values
  43 }
  44
  45 wxRichTextPrintout::~wxRichTextPrintout()
  46 {
  47 }
  48
  49 void wxRichTextPrintout::OnPreparePrinting()
  50 {
  51     wxBusyCursor wait;
  52
  53     m_numPages = 1;
  54
  55     m_pageBreaksStart.Clear();
  56     m_pageBreaksEnd.Clear();
  57     m_pageYOffsets.Clear();
  58
  59     int lastStartPos = 0;
  60
  61     wxRect rect, headerRect, footerRect;
  62
  63     /// Sets the DC scaling and returns important page rectangles
  64     CalculateScaling(GetDC(), rect, headerRect, footerRect);
  65
  66     if (GetRichTextBuffer())
  67     {
  68         GetRichTextBuffer()->Invalidate(wxRICHTEXT_ALL);
  69
  70         wxRichTextDrawingContext context(GetRichTextBuffer());
  71         GetRichTextBuffer()->Layout(*GetDC(), context, rect, rect, wxRICHTEXT_FIXED_WIDTH|wxRICHTEXT_VARIABLE_HEIGHT);
  72
  73         // Now calculate the page breaks
  74
  75         int yOffset = 0;
  76
  77         wxRichTextLine* lastLine = NULL;
  78
  79         wxRichTextObjectList::compatibility_iterator node = GetRichTextBuffer()->GetChildren().GetFirst();
  80         while (node)
  81         {
  82             // child is a paragraph
  83             wxRichTextParagraph* child = wxDynamicCast(node->GetData(), wxRichTextParagraph);
  84             wxASSERT (child != NULL);
  85             if (child)
  86             {
  87                 wxRichTextLineList::compatibility_iterator node2 = child->GetLines().GetFirst();
  88                 while (node2)
  89                 {
  90                     wxRichTextLine* line = node2->GetData();
  91
  92                     int lineY = child->GetPosition().y + line->GetPosition().y - yOffset;
  93                     bool hasHardPageBreak = ((node2 == child->GetLines().GetFirst()) && child->GetAttributes().HasPageBreak());
  94
  95                     // Break the page if either we're going off the bottom, or this paragraph specifies
  96                     // an explicit page break
  97
  98                     if (((lineY + line->GetSize().y) > rect.GetBottom()) || hasHardPageBreak)
  99                     {
 100                         // Only if we're not at the start of the document, and
 101                         // even then, only if either it's a hard break or if the object
 102                         // can fit in a whole page (otherwise there's no point in making
 103                         // the rest of this page blank).
 104                         if (lastLine && (hasHardPageBreak || (line->GetSize().y <= rect.GetHeight())))
 105                         {
 106                             // New page starting at this line
 107                             int newY = rect.y;
 108
 109                             // We increase the offset by the difference between new and old positions
 110
 111                             int increaseOffsetBy = lineY - newY;
 112                             yOffset += increaseOffsetBy;
 113
 114                             m_pageBreaksStart.Add(lastStartPos);
 115                             m_pageBreaksEnd.Add(lastLine->GetAbsoluteRange().GetEnd());
 116                             m_pageYOffsets.Add(yOffset);
 117
 118                             lastStartPos = line->GetAbsoluteRange().GetStart();
 119                             m_numPages ++;
 120                         }
 121
 122                         lastLine = line;
 123
 124                         // Now create page breaks for the rest of the line, if it's larger than the page height
 125                         int contentLeft = line->GetSize().y - rect.GetHeight();
 126                         while (contentLeft >= 0)
 127                         {
 128                             yOffset += rect.GetHeight();
 129                             contentLeft -= rect.GetHeight();
 130
 131                             m_pageBreaksStart.Add(lastStartPos);
 132                             m_pageBreaksEnd.Add(lastLine->GetAbsoluteRange().GetEnd());
 133                             m_pageYOffsets.Add(yOffset);
 134
 135                             m_numPages ++;
 136                         }
 137                     }
 138
 139                     lastLine = line;
 140
 141                     node2 = node2->GetNext();
 142                 }
 143             }
 144
 145             node = node->GetNext();
 146         }
 147
 148         // Closing page break
 149         m_pageBreaksStart.Add(lastStartPos);
 150         m_pageBreaksEnd.Add(GetRichTextBuffer()->GetOwnRange().GetEnd());
 151         m_pageYOffsets.Add(yOffset);
 152     }
 153 }
 154
 155 bool wxRichTextPrintout::OnBeginDocument(int startPage, int endPage)
 156 {
 157     if (!wxPrintout::OnBeginDocument(startPage, endPage)) return false;
 158
 159     return true;
 160 }
 161
 162 bool wxRichTextPrintout::OnPrintPage(int page)
 163 {
 164     wxDC *dc = GetDC();
 165     if (dc)
 166     {
 167         if (HasPage(page))
 168             RenderPage(dc, page);
 169         return true;
 170     }
 171     else return false;
 172 }
 173
 174 void wxRichTextPrintout::GetPageInfo(int *minPage, int *maxPage, int *selPageFrom, int *selPageTo)
 175 {
 176     *minPage = 1;
 177     *maxPage = m_numPages;
 178     *selPageFrom = 1;
 179     *selPageTo = m_numPages;
 180 }
 181
 182 bool wxRichTextPrintout::HasPage(int pageNum)
 183 {
 184     return pageNum > 0 && pageNum <= m_numPages;
 185 }
 186
 187 void wxRichTextPrintout::RenderPage(wxDC *dc, int page)
 188 {
 189     if (!GetRichTextBuffer())
 190         return;
 191
 192     wxBusyCursor wait;
 193
 194     wxRect textRect, headerRect, footerRect;
 195
 196     /// Sets the DC scaling and returns important page rectangles
 197     CalculateScaling(dc, textRect, headerRect, footerRect);
 198
 199     if (page > 1 || m_headerFooterData.GetShowOnFirstPage())
 200     {
 201         if (m_headerFooterData.GetFont().IsOk())
 202             dc->SetFont(m_headerFooterData.GetFont());
 203         else
 204             dc->SetFont(*wxNORMAL_FONT);
 205         if (m_headerFooterData.GetTextColour().IsOk())
 206             dc->SetTextForeground(m_headerFooterData.GetTextColour());
 207         else
 208             dc->SetTextForeground(*wxBLACK);
 209         dc->SetBackgroundMode(wxBRUSHSTYLE_TRANSPARENT);
 210
 211         // Draw header, if any
 212         wxRichTextOddEvenPage oddEven = ((page % 2) == 1) ? wxRICHTEXT_PAGE_ODD : wxRICHTEXT_PAGE_EVEN;
 213
 214         wxString headerTextCentre = m_headerFooterData.GetHeaderText(oddEven, wxRICHTEXT_PAGE_CENTRE);
 215         wxString headerTextLeft = m_headerFooterData.GetHeaderText(oddEven, wxRICHTEXT_PAGE_LEFT);
 216         wxString headerTextRight = m_headerFooterData.GetHeaderText(oddEven, wxRICHTEXT_PAGE_RIGHT);
 217
 218         if (!headerTextLeft.IsEmpty())
 219         {
 220             SubstituteKeywords(headerTextLeft, GetTitle(), page, m_numPages);
 221
 222             //int tx, ty;
 223             //dc->GetTextExtent(headerTextLeft, & tx, & ty);
 224
 225             int x = headerRect.GetLeft();
 226             int y = headerRect.GetX();
 227             dc->DrawText(headerTextLeft, x, y);
 228         }
 229         if (!headerTextCentre.IsEmpty())
 230         {
 231             SubstituteKeywords(headerTextCentre, GetTitle(), page, m_numPages);
 232
 233             int tx, ty;
 234             dc->GetTextExtent(headerTextCentre, & tx, & ty);
 235
 236             int x = headerRect.GetWidth()/2 - tx/2 + headerRect.GetLeft();
 237             int y = headerRect.GetY();
 238             dc->DrawText(headerTextCentre, x, y);
 239         }
 240         if (!headerTextRight.IsEmpty())
 241         {
 242             SubstituteKeywords(headerTextRight, GetTitle(), page, m_numPages);
 243
 244             int tx, ty;
 245             dc->GetTextExtent(headerTextRight, & tx, & ty);
 246
 247             int x = headerRect.GetRight() - tx;
 248             int y = headerRect.GetY();
 249             dc->DrawText(headerTextRight, x, y);
 250         }
 251
 252         // Draw footer, if any
 253         wxString footerTextCentre = m_headerFooterData.GetFooterText(oddEven, wxRICHTEXT_PAGE_CENTRE);
 254         wxString footerTextLeft = m_headerFooterData.GetFooterText(oddEven, wxRICHTEXT_PAGE_LEFT);
 255         wxString footerTextRight = m_headerFooterData.GetFooterText(oddEven, wxRICHTEXT_PAGE_RIGHT);
 256
 257         if (!footerTextLeft.IsEmpty())
 258         {
 259             SubstituteKeywords(footerTextLeft, GetTitle(), page, m_numPages);
 260
 261             int tx, ty;
 262             dc->GetTextExtent(footerTextLeft, & tx, & ty);
 263
 264             int x = footerRect.GetLeft();
 265             int y = footerRect.GetBottom() - ty;
 266             dc->DrawText(footerTextLeft, x, y);
 267         }
 268         if (!footerTextCentre.IsEmpty())
 269         {
 270             SubstituteKeywords(footerTextCentre, GetTitle(), page, m_numPages);
 271
 272             int tx, ty;
 273             dc->GetTextExtent(footerTextCentre, & tx, & ty);
 274
 275             int x = footerRect.GetWidth()/2 - tx/2 + footerRect.GetLeft();
 276             int y = footerRect.GetBottom() - ty;
 277             dc->DrawText(footerTextCentre, x, y);
 278         }
 279         if (!footerTextRight.IsEmpty())
 280         {
 281             SubstituteKeywords(footerTextRight, GetTitle(), page, m_numPages);
 282
 283             int tx, ty;
 284             dc->GetTextExtent(footerTextRight, & tx, & ty);
 285
 286             int x = footerRect.GetRight() - tx;
 287             int y = footerRect.GetBottom() - ty;
 288             dc->DrawText(footerTextRight, x, y);
 289         }
 290     }
 291
 292     wxRichTextRange rangeToDraw(m_pageBreaksStart[page-1], m_pageBreaksEnd[page-1]);
 293
 294     wxPoint oldOrigin = dc->GetLogicalOrigin();
 295     double scaleX, scaleY;
 296     dc->GetUserScale(& scaleX, & scaleY);
 297
 298     int yOffset = 0;
 299     if (page > 1)
 300         yOffset = m_pageYOffsets[page-2];
 301
 302     if (yOffset != oldOrigin.y)
 303         dc->SetLogicalOrigin(oldOrigin.x, oldOrigin.y + yOffset);
 304
 305     dc->SetClippingRegion(wxRect(textRect.x, textRect.y + yOffset, textRect.width, textRect.height));
 306
 307     wxRichTextDrawingContext context(GetRichTextBuffer());
 308     GetRichTextBuffer()->Draw(*dc, context, rangeToDraw, wxRichTextSelection(), textRect, 0 /* descent */, wxRICHTEXT_DRAW_IGNORE_CACHE|wxRICHTEXT_DRAW_PRINT /* flags */);
 309
 310     dc->DestroyClippingRegion();
 311
 312     if (yOffset != oldOrigin.y)
 313         dc->SetLogicalOrigin(oldOrigin.x, oldOrigin.y);
 314 }
 315
 316 void wxRichTextPrintout::SetMargins(int top, int bottom, int left, int right)
 317 {
 318     m_marginTop = top;
 319     m_marginBottom = bottom;
 320     m_marginLeft = left;
 321     m_marginRight = right;
 322 }
 323
 324 /// Calculate scaling and rectangles, setting the device context scaling
 325 void wxRichTextPrintout::CalculateScaling(wxDC* dc, wxRect& textRect, wxRect& headerRect, wxRect& footerRect)
 326 {
 327     // Get the logical pixels per inch of screen and printer
 328     int ppiScreenX, ppiScreenY;
 329     GetPPIScreen(&ppiScreenX, &ppiScreenY);
 330     int ppiPrinterX, ppiPrinterY;
 331     GetPPIPrinter(&ppiPrinterX, &ppiPrinterY);
 332
 333     // This scales the DC so that the printout roughly represents the
 334     // the screen scaling.
 335     float scale = (float)((float)ppiPrinterX/(float)ppiScreenX);
 336
 337     // Now we have to check in case our real page size is reduced
 338     // (e.g. because we're drawing to a print preview memory DC)
 339     int pageWidth, pageHeight;
 340     int w, h;
 341     dc->GetSize(&w, &h);
 342     GetPageSizePixels(&pageWidth, &pageHeight);
 343
 344     // If printer pageWidth == current DC width, then this doesn't
 345     // change. But w might be the preview bitmap width, so scale down.
 346     float previewScale = (float)(w/(float)pageWidth);
 347     float overallScale = scale * previewScale;
 348
 349     // The dimensions used for indentation etc. have to be unscaled
 350     // during printing to be correct when scaling is applied.
 351     // Also, correct the conversions in wxRTC using DC instead of print DC.
 352     m_richTextBuffer->SetScale(scale * float(dc->GetPPI().x)/float(ppiPrinterX));
 353
 354     // Calculate margins
 355     int marginLeft = wxRichTextObject::ConvertTenthsMMToPixels(ppiPrinterX, m_marginLeft);
 356     int marginTop = wxRichTextObject::ConvertTenthsMMToPixels(ppiPrinterX, m_marginTop);
 357     int marginRight = wxRichTextObject::ConvertTenthsMMToPixels(ppiPrinterX, m_marginRight);
 358     int marginBottom = wxRichTextObject::ConvertTenthsMMToPixels(ppiPrinterX, m_marginBottom);
 359
 360     // Header and footer margins
 361     int headerMargin = wxRichTextObject::ConvertTenthsMMToPixels(ppiPrinterX, m_headerFooterData.GetHeaderMargin());
 362     int footerMargin = wxRichTextObject::ConvertTenthsMMToPixels(ppiPrinterX, m_headerFooterData.GetFooterMargin());
 363
 364     dc->SetUserScale(overallScale, overallScale);
 365
 366     wxRect rect((int) (marginLeft/scale), (int) (marginTop/scale),
 367                 (int) ((pageWidth - marginLeft - marginRight)/scale), (int)((pageHeight - marginTop - marginBottom)/scale));
 368
 369     headerRect = wxRect(0, 0, 0, 0);
 370
 371     if (!m_headerFooterData.GetHeaderText(wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_LEFT).IsEmpty() ||
 372         !m_headerFooterData.GetHeaderText(wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_CENTRE).IsEmpty() ||
 373         !m_headerFooterData.GetHeaderText(wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_RIGHT).IsEmpty() ||
 374
 375         !m_headerFooterData.GetHeaderText(wxRICHTEXT_PAGE_EVEN, wxRICHTEXT_PAGE_LEFT).IsEmpty() ||
 376         !m_headerFooterData.GetHeaderText(wxRICHTEXT_PAGE_EVEN, wxRICHTEXT_PAGE_CENTRE).IsEmpty() ||
 377         !m_headerFooterData.GetHeaderText(wxRICHTEXT_PAGE_EVEN, wxRICHTEXT_PAGE_RIGHT).IsEmpty())
 378     {
 379         if (m_headerFooterData.GetFont().IsOk())
 380             dc->SetFont(m_headerFooterData.GetFont());
 381         else
 382             dc->SetFont(*wxNORMAL_FONT);
 383
 384         int charHeight = dc->GetCharHeight();
 385
 386         int headerHeight = (int) (charHeight + headerMargin/scale);
 387
 388         headerRect = wxRect(rect.x, rect.y, rect.width, headerHeight);
 389
 390         rect.y += headerHeight;
 391         rect.height -= headerHeight;
 392     }
 393
 394     footerRect = wxRect(0, 0, 0, 0);
 395
 396     if (!m_headerFooterData.GetFooterText(wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_LEFT).IsEmpty() ||
 397         !m_headerFooterData.GetFooterText(wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_CENTRE).IsEmpty() ||
 398         !m_headerFooterData.GetFooterText(wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_RIGHT).IsEmpty() ||
 399
 400         !m_headerFooterData.GetFooterText(wxRICHTEXT_PAGE_EVEN, wxRICHTEXT_PAGE_LEFT).IsEmpty() ||
 401         !m_headerFooterData.GetFooterText(wxRICHTEXT_PAGE_EVEN, wxRICHTEXT_PAGE_CENTRE).IsEmpty() ||
 402         !m_headerFooterData.GetFooterText(wxRICHTEXT_PAGE_EVEN, wxRICHTEXT_PAGE_RIGHT).IsEmpty())
 403     {
 404         if (m_headerFooterData.GetFont().IsOk())
 405             dc->SetFont(m_headerFooterData.GetFont());
 406         else
 407             dc->SetFont(*wxNORMAL_FONT);
 408
 409         int charHeight = dc->GetCharHeight();
 410
 411         int footerHeight = (int) (charHeight + footerMargin/scale);
 412
 413         footerRect = wxRect(rect.x, rect.y + rect.height, rect.width, footerHeight);
 414
 415         rect.height -= footerHeight;
 416     }
 417
 418     textRect = rect;
 419 }
 420
 421 bool wxRichTextPrintout::SubstituteKeywords(wxString& str, const wxString& title, int pageNum, int pageCount)
 422 {
 423     wxString num;
 424
 425     num.Printf(wxT("%i"), pageNum);
 426     str.Replace(wxT("@PAGENUM@"), num);
 427
 428     num.Printf(wxT("%lu"), (unsigned long) pageCount);
 429     str.Replace(wxT("@PAGESCNT@"), num);
 430
 431     wxDateTime now = wxDateTime::Now();
 432
 433     str.Replace(wxT("@DATE@"), now.FormatDate());
 434     str.Replace(wxT("@TIME@"), now.FormatTime());
 435
 436     str.Replace(wxT("@TITLE@"), title);
 437
 438     return true;
 439 }
 440
 441 /*!
 442  * wxRichTextPrinting
 443  */
 444
 445 wxRichTextPrinting::wxRichTextPrinting(const wxString& name, wxWindow *parentWindow)
 446 {
 447     m_richTextBufferPrinting = NULL;
 448     m_richTextBufferPreview = NULL;
 449
 450     m_parentWindow = parentWindow;
 451     m_title = name;
 452     m_printData = NULL;
 453
 454     m_previewRect = wxRect(wxPoint(100, 100), wxSize(800, 800));
 455
 456     m_pageSetupData = new wxPageSetupDialogData;
 457     m_pageSetupData->EnableMargins(true);
 458     m_pageSetupData->SetMarginTopLeft(wxPoint(25, 25));
 459     m_pageSetupData->SetMarginBottomRight(wxPoint(25, 25));
 460 }
 461
 462 wxRichTextPrinting::~wxRichTextPrinting()
 463 {
 464     delete m_printData;
 465     delete m_pageSetupData;
 466     delete m_richTextBufferPrinting;
 467     delete m_richTextBufferPreview;
 468 }
 469
 470 wxPrintData *wxRichTextPrinting::GetPrintData()
 471 {
 472     if (m_printData == NULL)
 473         m_printData = new wxPrintData();
 474     return m_printData;
 475 }
 476
 477 /// Set print and page setup data
 478 void wxRichTextPrinting::SetPrintData(const wxPrintData& printData)
 479 {
 480     (*GetPrintData()) = printData;
 481 }
 482
 483 void wxRichTextPrinting::SetPageSetupData(const wxPageSetupDialogData& pageSetupData)
 484 {
 485     (*GetPageSetupData()) = pageSetupData;
 486 }
 487
 488 /// Set the rich text buffer pointer, deleting the existing object if present
 489 void wxRichTextPrinting::SetRichTextBufferPrinting(wxRichTextBuffer* buf)
 490 {
 491     if (m_richTextBufferPrinting)
 492     {
 493         delete m_richTextBufferPrinting;
 494         m_richTextBufferPrinting = NULL;
 495     }
 496     m_richTextBufferPrinting = buf;
 497 }
 498
 499 void wxRichTextPrinting::SetRichTextBufferPreview(wxRichTextBuffer* buf)
 500 {
 501     if (m_richTextBufferPreview)
 502     {
 503         delete m_richTextBufferPreview;
 504         m_richTextBufferPreview = NULL;
 505     }
 506     m_richTextBufferPreview = buf;
 507 }
 508
 509 #if wxUSE_FFILE && wxUSE_STREAMS
 510 bool wxRichTextPrinting::PreviewFile(const wxString& richTextFile)
 511 {
 512     SetRichTextBufferPreview(new wxRichTextBuffer);
 513
 514     if (!m_richTextBufferPreview->LoadFile(richTextFile))
 515     {
 516         SetRichTextBufferPreview(NULL);
 517         return false;
 518     }
 519     else
 520         SetRichTextBufferPrinting(new wxRichTextBuffer(*m_richTextBufferPreview));
 521
 522     wxRichTextPrintout *p1 = CreatePrintout();
 523     p1->SetRichTextBuffer(m_richTextBufferPreview);
 524
 525     wxRichTextPrintout *p2 = CreatePrintout();
 526     p2->SetRichTextBuffer(m_richTextBufferPrinting);
 527     return DoPreview(p1, p2);
 528 }
 529 #endif // wxUSE_FFILE && wxUSE_STREAMS
 530
 531 bool wxRichTextPrinting::PreviewBuffer(const wxRichTextBuffer& buffer)
 532 {
 533     SetRichTextBufferPreview(new wxRichTextBuffer(buffer));
 534     SetRichTextBufferPrinting(new wxRichTextBuffer(buffer));
 535
 536     wxRichTextPrintout *p1 = CreatePrintout();
 537     p1->SetRichTextBuffer(m_richTextBufferPreview);
 538
 539     wxRichTextPrintout *p2 = CreatePrintout();
 540     p2->SetRichTextBuffer(m_richTextBufferPrinting);
 541
 542     return DoPreview(p1, p2);
 543 }
 544
 545 #if wxUSE_FFILE && wxUSE_STREAMS
 546 bool wxRichTextPrinting::PrintFile(const wxString& richTextFile, bool showPrintDialog)
 547 {
 548     SetRichTextBufferPrinting(new wxRichTextBuffer);
 549
 550     if (!m_richTextBufferPrinting->LoadFile(richTextFile))
 551     {
 552         SetRichTextBufferPrinting(NULL);
 553         return false;
 554     }
 555
 556     wxRichTextPrintout *p = CreatePrintout();
 557     p->SetRichTextBuffer(m_richTextBufferPrinting);
 558
 559     bool ret = DoPrint(p, showPrintDialog);
 560     delete p;
 561     return ret;
 562 }
 563 #endif // wxUSE_FFILE && wxUSE_STREAMS
 564
 565 bool wxRichTextPrinting::PrintBuffer(const wxRichTextBuffer& buffer, bool showPrintDialog)
 566 {
 567     SetRichTextBufferPrinting(new wxRichTextBuffer(buffer));
 568
 569     wxRichTextPrintout *p = CreatePrintout();
 570     p->SetRichTextBuffer(m_richTextBufferPrinting);
 571
 572     bool ret = DoPrint(p, showPrintDialog);
 573     delete p;
 574     return ret;
 575 }
 576
 577 bool wxRichTextPrinting::DoPreview(wxRichTextPrintout *printout1, wxRichTextPrintout *printout2)
 578 {
 579     // Pass two printout objects: for preview, and possible printing.
 580     wxPrintDialogData printDialogData(*GetPrintData());
 581     wxPrintPreview *preview = new wxPrintPreview(printout1, printout2, &printDialogData);
 582     if (!preview->IsOk())
 583     {
 584         delete preview;
 585         return false;
 586     }
 587
 588     wxPreviewFrame *frame = new wxPreviewFrame(preview, m_parentWindow,
 589                                                m_title + _(" Preview"),
 590                                                m_previewRect.GetPosition(), m_previewRect.GetSize());
 591     frame->Centre(wxBOTH);
 592     frame->Initialize();
 593     frame->Show(true);
 594     return true;
 595 }
 596
 597 bool wxRichTextPrinting::DoPrint(wxRichTextPrintout *printout, bool showPrintDialog)
 598 {
 599     wxPrintDialogData printDialogData(*GetPrintData());
 600     wxPrinter printer(&printDialogData);
 601
 602     if (!printer.Print(m_parentWindow, printout, showPrintDialog))
 603     {
 604         return false;
 605     }
 606
 607     (*GetPrintData()) = printer.GetPrintDialogData().GetPrintData();
 608     return true;
 609 }
 610
 611 void wxRichTextPrinting::PageSetup()
 612 {
 613     if (!GetPrintData()->IsOk())
 614     {
 615         wxLogError(_("There was a problem during page setup: you may need to set a default printer."));
 616         return;
 617     }
 618
 619     m_pageSetupData->SetPrintData(*GetPrintData());
 620     wxPageSetupDialog pageSetupDialog(m_parentWindow, m_pageSetupData);
 621
 622     if (pageSetupDialog.ShowModal() == wxID_OK)
 623     {
 624         (*GetPrintData()) = pageSetupDialog.GetPageSetupData().GetPrintData();
 625         (*m_pageSetupData) = pageSetupDialog.GetPageSetupData();
 626     }
 627 }
 628
 629 wxRichTextPrintout *wxRichTextPrinting::CreatePrintout()
 630 {
 631     wxRichTextPrintout *p = new wxRichTextPrintout(m_title);
 632
 633     p->SetHeaderFooterData(GetHeaderFooterData());
 634     p->SetMargins(10*m_pageSetupData->GetMarginTopLeft().y,
 635                     10*m_pageSetupData->GetMarginBottomRight().y,
 636                     10*m_pageSetupData->GetMarginTopLeft().x,
 637                     10*m_pageSetupData->GetMarginBottomRight().x);
 638
 639     return p;
 640 }
 641
 642 /// Set/get header text, e.g. wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_LEFT
 643 void wxRichTextPrinting::SetHeaderText(const wxString& text, wxRichTextOddEvenPage page, wxRichTextPageLocation location)
 644 {
 645     m_headerFooterData.SetHeaderText(text, page, location);
 646 }
 647
 648 wxString wxRichTextPrinting::GetHeaderText(wxRichTextOddEvenPage page, wxRichTextPageLocation location) const
 649 {
 650     return m_headerFooterData.GetHeaderText(page, location);
 651 }
 652
 653 /// Set/get footer text, e.g. wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_LEFT
 654 void wxRichTextPrinting::SetFooterText(const wxString& text, wxRichTextOddEvenPage page, wxRichTextPageLocation location)
 655 {
 656     m_headerFooterData.SetFooterText(text, page, location);
 657 }
 658
 659 wxString wxRichTextPrinting::GetFooterText(wxRichTextOddEvenPage page, wxRichTextPageLocation location) const
 660 {
 661     return m_headerFooterData.GetFooterText(page, location);
 662 }
 663
 664 /*!
 665  * Header/footer data
 666  */
 667
 668 IMPLEMENT_CLASS(wxRichTextHeaderFooterData, wxObject)
 669
 670 /// Copy
 671 void wxRichTextHeaderFooterData::Copy(const wxRichTextHeaderFooterData& data)
 672 {
 673     int i;
 674     for (i = 0; i < 12; i++)
 675         m_text[i] = data.m_text[i];
 676     m_font = data.m_font;
 677     m_colour = data.m_colour;
 678     m_headerMargin = data.m_headerMargin;
 679     m_footerMargin = data.m_footerMargin;
 680     m_showOnFirstPage = data.m_showOnFirstPage;
 681 }
 682
 683 /// Set/get text
 684 void wxRichTextHeaderFooterData::SetText(const wxString& text, int headerFooter, wxRichTextOddEvenPage page, wxRichTextPageLocation location)
 685 {
 686     int idx = headerFooter + (2 * (int) page) + (4 * (int) location);
 687     wxASSERT( idx >= 0 && idx < 12 );
 688
 689     if (idx >= 0 && idx < 12)
 690         m_text[idx] = text;
 691 }
 692
 693 wxString wxRichTextHeaderFooterData::GetText(int headerFooter, wxRichTextOddEvenPage page, wxRichTextPageLocation location) const
 694 {
 695     int idx = headerFooter + (2 * (int) page) + (4 * (int) location);
 696     wxASSERT( idx >= 0 && idx < 12 );
 697
 698     if (idx >= 0 && idx < 12)
 699         return m_text[idx];
 700     else
 701         return wxEmptyString;
 702 }
 703
 704 /// Set/get header text, e.g. wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_LEFT
 705 void wxRichTextHeaderFooterData::SetHeaderText(const wxString& text, wxRichTextOddEvenPage page, wxRichTextPageLocation location)
 706 {
 707     if (page == wxRICHTEXT_PAGE_ALL)
 708     {
 709         SetText(text, 0, wxRICHTEXT_PAGE_ODD, location);
 710         SetText(text, 0, wxRICHTEXT_PAGE_EVEN, location);
 711     }
 712     else
 713         SetText(text, 0, page, location);
 714 }
 715
 716 wxString wxRichTextHeaderFooterData::GetHeaderText(wxRichTextOddEvenPage page, wxRichTextPageLocation location) const
 717 {
 718     return GetText(0, page, location);
 719 }
 720
 721 /// Set/get footer text, e.g. wxRICHTEXT_PAGE_ODD, wxRICHTEXT_PAGE_LEFT
 722 void wxRichTextHeaderFooterData::SetFooterText(const wxString& text, wxRichTextOddEvenPage page, wxRichTextPageLocation location)
 723 {
 724     if (page == wxRICHTEXT_PAGE_ALL)
 725     {
 726         SetText(text, 1, wxRICHTEXT_PAGE_ODD, location);
 727         SetText(text, 1, wxRICHTEXT_PAGE_EVEN, location);
 728     }
 729     else
 730         SetText(text, 1, page, location);
 731 }
 732
 733 wxString wxRichTextHeaderFooterData::GetFooterText(wxRichTextOddEvenPage page, wxRichTextPageLocation location) const
 734 {
 735     return GetText(1, page, location);
 736 }
 737
 738 /// Clear all text
 739 void wxRichTextHeaderFooterData::Clear()
 740 {
 741     int i;
 742     for (i = 0; i < 12; i++)
 743         m_text[i] = wxEmptyString;
 744 }
 745
 746 #endif // wxUSE_RICHTEXT & wxUSE_PRINTING_ARCHITECTURE
 747