/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the QtGui module of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include "qprintpreviewwidget.h"

#include "private/qwidget_p.h"

#include <private/qprinter_p.h>

#include <QtCore/qmath.h>

#include <QtGui/qboxlayout.h>

#include <QtGui/qgraphicsitem.h>

#include <QtGui/qgraphicsview.h>

#include <QtGui/qscrollbar.h>

#include <QtGui/qstyleoption.h>

#ifndef QT_NO_PRINTPREVIEWWIDGET

QT_BEGIN_NAMESPACE

namespace {

class PageItem : public QGraphicsItem

{

public:

    PageItem(int _pageNum, const QPicture* _pagePicture, QSize _paperSize, QRect _pageRect)

        : pageNum(_pageNum), pagePicture(_pagePicture),

          paperSize(_paperSize), pageRect(_pageRect)

    {

        qreal border = qMax(paperSize.height(), paperSize.width()) / 25;

        brect = QRectF(QPointF(-border, -border),

                       QSizeF(paperSize)+QSizeF(2*border, 2*border));

        setCacheMode(DeviceCoordinateCache);

    }

    inline QRectF boundingRect() const

    { return brect; }

    inline int pageNumber() const

    { return pageNum; }

    void paint(QPainter *painter, const QStyleOptionGraphicsItem *item, QWidget *widget);

private:

    int pageNum;

    const QPicture* pagePicture;

    QSize paperSize;

    QRect pageRect;

    QRectF brect;

};

void PageItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget)

{

    Q_UNUSED(widget);

#if 0

    // Draw item bounding rect, for debugging

    painter->save();

    painter->setPen(QPen(Qt::red, 0));

    painter->setBrush(Qt::NoBrush);

    painter->drawRect(QRectF(-border()+1.0, -border()+1.0, boundingRect().width()-2, boundingRect().height()-2));

    painter->restore();

#endif

    QRectF paperRect(0,0, paperSize.width(), paperSize.height());

    // Draw shadow

    painter->setClipRect(option->exposedRect);

    qreal shWidth = paperRect.width()/100;

    QRectF rshadow(paperRect.topRight() + QPointF(0, shWidth),

                   paperRect.bottomRight() + QPointF(shWidth, 0));

    QLinearGradient rgrad(rshadow.topLeft(), rshadow.topRight());

    rgrad.setColorAt(0.0, QColor(0,0,0,255));

    rgrad.setColorAt(1.0, QColor(0,0,0,0));

    painter->fillRect(rshadow, QBrush(rgrad));

    QRectF bshadow(paperRect.bottomLeft() + QPointF(shWidth, 0),

                   paperRect.bottomRight() + QPointF(0, shWidth));

    QLinearGradient bgrad(bshadow.topLeft(), bshadow.bottomLeft());

    bgrad.setColorAt(0.0, QColor(0,0,0,255));

    bgrad.setColorAt(1.0, QColor(0,0,0,0));

    painter->fillRect(bshadow, QBrush(bgrad));

    QRectF cshadow(paperRect.bottomRight(),

                   paperRect.bottomRight() + QPointF(shWidth, shWidth));

    QRadialGradient cgrad(cshadow.topLeft(), shWidth, cshadow.topLeft());

    cgrad.setColorAt(0.0, QColor(0,0,0,255));

    cgrad.setColorAt(1.0, QColor(0,0,0,0));

    painter->fillRect(cshadow, QBrush(cgrad));

    painter->setClipRect(paperRect & option->exposedRect);

    painter->fillRect(paperRect, Qt::white);

    if (!pagePicture)

        return;

    painter->drawPicture(pageRect.topLeft(), *pagePicture);

    // Effect: make anything drawn in the margins look washed out.

    QPainterPath path;

    path.addRect(paperRect);

    path.addRect(pageRect);

    painter->setPen(QPen(Qt::NoPen));

    painter->setBrush(QColor(255, 255, 255, 180));

    painter->drawPath(path);

#if 0

    // Draw frame around paper.

    painter->setPen(QPen(Qt::black, 0));

    painter->setBrush(Qt::NoBrush);

    painter->drawRect(paperRect);

#endif

    // todo: drawtext "Page N" below paper

}

class GraphicsView : public QGraphicsView

{

    Q_OBJECT

public:

    GraphicsView(QWidget* parent = 0)

        : QGraphicsView(parent)

    {

#ifdef Q_WS_MAC

        setFrameStyle(QFrame::NoFrame);

#endif

    }

signals:

    void resized();

protected:

    void resizeEvent(QResizeEvent* e)

    {

        QGraphicsView::resizeEvent(e);

        emit resized();

    }

    void showEvent(QShowEvent* e)

    {

        QGraphicsView::showEvent(e);

        emit resized();

    }

};

} // anonymous namespace

class QPrintPreviewWidgetPrivate : public QWidgetPrivate

{

    Q_DECLARE_PUBLIC(QPrintPreviewWidget)

public:

    QPrintPreviewWidgetPrivate()

        : scene(0), curPage(1),

          viewMode(QPrintPreviewWidget::SinglePageView),

          zoomMode(QPrintPreviewWidget::FitInView),

          zoomFactor(1), initialized(false), fitting(true)

    {}

    // private slots

    void _q_fit(bool doFitting = false);

    void _q_updateCurrentPage();

    void init();

    void populateScene();

    void layoutPages();

    void generatePreview();

    void setCurrentPage(int pageNumber);

    void zoom(qreal zoom);

    void setZoomFactor(qreal zoomFactor);

    int calcCurrentPage();

    GraphicsView *graphicsView;

    QGraphicsScene *scene;

    int curPage;

    QList<const QPicture *> pictures;

    QList<QGraphicsItem *> pages;

    QPrintPreviewWidget::ViewMode viewMode;

    QPrintPreviewWidget::ZoomMode zoomMode;

    qreal zoomFactor;

    bool ownPrinter;

    QPrinter* printer;

    bool initialized;

    bool fitting;

};

void QPrintPreviewWidgetPrivate::_q_fit(bool doFitting)

{

    Q_Q(QPrintPreviewWidget);

    if (curPage < 1 || curPage > pages.count())

        return;

    if (!doFitting && !fitting)

        return;

    if (doFitting && fitting) {

        QRect viewRect = graphicsView->viewport()->rect();

        if (zoomMode == QPrintPreviewWidget::FitInView) {

            QList<QGraphicsItem*> containedItems = graphicsView->items(viewRect, Qt::ContainsItemBoundingRect);

            foreach(QGraphicsItem* item, containedItems) {

                PageItem* pg = static_cast<PageItem*>(item);

                if (pg->pageNumber() == curPage)

                    return;

            }

        }

        int newPage = calcCurrentPage();

        if (newPage != curPage)

            curPage = newPage;

    }

    QRectF target = pages.at(curPage-1)->sceneBoundingRect();

    if (viewMode == QPrintPreviewWidget::FacingPagesView) {

        // fit two pages

        if (curPage % 2)

            target.setLeft(target.left() - target.width());

        else

            target.setRight(target.right() + target.width());

    } else if (viewMode == QPrintPreviewWidget::AllPagesView) {

        target = scene->itemsBoundingRect();

    }

    if (zoomMode == QPrintPreviewWidget::FitToWidth) {

        QTransform t;

        qreal scale = graphicsView->viewport()->width() / target.width();

        t.scale(scale, scale);

        graphicsView->setTransform(t);

        if (doFitting && fitting) {

            QRectF viewSceneRect = graphicsView->viewportTransform().mapRect(graphicsView->viewport()->rect());

            viewSceneRect.moveTop(target.top());

            graphicsView->ensureVisible(viewSceneRect); // Nah...

        }

    } else {

        graphicsView->fitInView(target, Qt::KeepAspectRatio);

        if (zoomMode == QPrintPreviewWidget::FitInView) {

            int step = qRound(graphicsView->matrix().mapRect(target).height());

            graphicsView->verticalScrollBar()->setSingleStep(step);

            graphicsView->verticalScrollBar()->setPageStep(step);

        }

    }

    zoomFactor = graphicsView->transform().m11() * (float(printer->logicalDpiY()) / q->logicalDpiY());

    emit q->previewChanged();

}

void QPrintPreviewWidgetPrivate::_q_updateCurrentPage()

{

    Q_Q(QPrintPreviewWidget);

    if (viewMode == QPrintPreviewWidget::AllPagesView)

        return;

    int newPage = calcCurrentPage();

    if (newPage != curPage) {

        curPage = newPage;

        emit q->previewChanged();

    }

}

int QPrintPreviewWidgetPrivate::calcCurrentPage()

{

    int maxArea = 0;

    int newPage = curPage;

    QRect viewRect = graphicsView->viewport()->rect();

    QList<QGraphicsItem*> items = graphicsView->items(viewRect);

    for (int i=0; i<items.size(); ++i) {

        PageItem* pg = static_cast<PageItem*>(items.at(i));

        QRect overlap = graphicsView->mapFromScene(pg->sceneBoundingRect()).boundingRect() & viewRect;

        int area = overlap.width() * overlap.height();

        if (area > maxArea) {

            maxArea = area;

            newPage = pg->pageNumber();

        } else if (area == maxArea && pg->pageNumber() < newPage) {

            newPage = pg->pageNumber();

        }

    }

    return newPage;

}

void QPrintPreviewWidgetPrivate::init()

{

    Q_Q(QPrintPreviewWidget);

    graphicsView = new GraphicsView;

    graphicsView->setInteractive(false);

    graphicsView->setDragMode(QGraphicsView::ScrollHandDrag);

    graphicsView->setViewportUpdateMode(QGraphicsView::SmartViewportUpdate);

    QObject::connect(graphicsView->verticalScrollBar(), SIGNAL(valueChanged(int)),

                     q, SLOT(_q_updateCurrentPage()));

    QObject::connect(graphicsView, SIGNAL(resized()), q, SLOT(_q_fit()));

    scene = new QGraphicsScene(graphicsView);

    scene->setBackgroundBrush(Qt::gray);

    graphicsView->setScene(scene);

    QVBoxLayout *layout = new QVBoxLayout;

    q->setLayout(layout);

    layout->setContentsMargins(0, 0, 0, 0);

    layout->addWidget(graphicsView);

}

void QPrintPreviewWidgetPrivate::populateScene()

{

    // remove old pages

    for (int i = 0; i < pages.size(); i++)

        scene->removeItem(pages.at(i));

    qDeleteAll(pages);

    pages.clear();

    int numPages = pictures.count();

    QSize paperSize = printer->paperRect().size();

    QRect pageRect = printer->pageRect();

    for (int i = 0; i < numPages; i++) {

        PageItem* item = new PageItem(i+1, pictures.at(i), paperSize, pageRect);

        scene->addItem(item);

        pages.append(item);

    }

}

void QPrintPreviewWidgetPrivate::layoutPages()

{

    int numPages = pages.count();

    if (numPages < 1)

        return;

    int numPagePlaces = numPages;

    int cols = 1; // singleMode and default

    if (viewMode == QPrintPreviewWidget::AllPagesView) {

        if (printer->orientation() == QPrinter::Portrait)

            cols = qCeil(qSqrt((float) numPages));

        else

            cols = qFloor(qSqrt((float) numPages));

        cols += cols % 2;  // Nicer with an even number of cols

    }

    else if (viewMode == QPrintPreviewWidget::FacingPagesView) {

        cols = 2;

        numPagePlaces += 1;

    }

    int rows = qCeil(qreal(numPagePlaces) / cols);

    qreal itemWidth = pages.at(0)->boundingRect().width();

    qreal itemHeight = pages.at(0)->boundingRect().height();

    int pageNum = 1;

    for (int i = 0; i < rows && pageNum <= numPages; i++) {

        for (int j = 0; j < cols && pageNum <= numPages; j++) {

            if (!i && !j && viewMode == QPrintPreviewWidget::FacingPagesView) {

                // Front page doesn't have a facing page

                continue;

            } else {

                pages.at(pageNum-1)->setPos(QPointF(j*itemWidth, i*itemHeight));

                pageNum++;

            }

        }

    }

    scene->setSceneRect(scene->itemsBoundingRect());

}

void QPrintPreviewWidgetPrivate::generatePreview()

{

    //### If QPrinter::setPreviewMode() becomes public, handle the

    //### case that we have been constructed with a printer that

    //### _already_ has been preview-painted to, so we should

    //### initially just show the pages it already contains, and not

    //### emit paintRequested() until the user changes some parameter

    Q_Q(QPrintPreviewWidget);

    printer->d_func()->setPreviewMode(true);

    emit q->paintRequested(printer);

    printer->d_func()->setPreviewMode(false);

    pictures = printer->d_func()->previewPages();

    populateScene(); // i.e. setPreviewPrintedPictures() e.l.

    layoutPages();

    curPage = qBound(1, curPage, pages.count());

    if (fitting)

        _q_fit();

    emit q->previewChanged();

}

void QPrintPreviewWidgetPrivate::setCurrentPage(int pageNumber)

{

    if (pageNumber < 1 || pageNumber > pages.count())

        return;

    int lastPage = curPage;

    curPage = pageNumber;

    if (lastPage != curPage && lastPage > 0 && lastPage <= pages.count()) {

        if (zoomMode != QPrintPreviewWidget::FitInView) {

            QScrollBar *hsc = graphicsView->horizontalScrollBar();

            QScrollBar *vsc = graphicsView->verticalScrollBar();

            QPointF pt = graphicsView->transform().map(pages.at(curPage-1)->pos());

            vsc->setValue(int(pt.y()) - 10);

            hsc->setValue(int(pt.x()) - 10);

        } else {

            graphicsView->centerOn(pages.at(curPage-1));

        }

    }

}

void QPrintPreviewWidgetPrivate::zoom(qreal zoom)

{

    zoomFactor *= zoom;

    graphicsView->scale(zoom, zoom);

}

void QPrintPreviewWidgetPrivate::setZoomFactor(qreal _zoomFactor)

{

    Q_Q(QPrintPreviewWidget);

    zoomFactor = _zoomFactor;

    graphicsView->resetTransform();

    int dpi_y = q->logicalDpiY();

    int printer_dpi_y = printer->logicalDpiY();

    graphicsView->scale(zoomFactor*(dpi_y/float(printer_dpi_y)),

                        zoomFactor*(dpi_y/float(printer_dpi_y)));

}

///////////////////////////////////////

/*!

    \class QPrintPreviewWidget

    \since 4.4

    \brief The QPrintPreviewWidget class provides a widget for

    previewing page layouts for printer output.

    \ingroup printing

    QPrintPreviewDialog uses a QPrintPreviewWidget internally, and the

    purpose of QPrintPreviewWidget is to make it possible to embed the

    preview into other widgets. It also makes it possible to build a different

    user interface around it than the default one provided with QPrintPreviewDialog.

    Using QPrintPreviewWidget is straightforward:

    \list 1

    \o Create the QPrintPreviewWidget

    Construct the QPrintPreviewWidget either by passing in an

    exisiting QPrinter object, or have QPrintPreviewWidget create a

    default constructed QPrinter object for you.

    \o Connect the paintRequested() signal to a slot.

    When the widget needs to generate a set of preview pages, a

    paintRequested() signal will be emitted from the widget. Connect a

    slot to this signal, and draw onto the QPrinter passed in as a

    signal parameter. Call QPrinter::newPage(), to start a new

    page in the preview.

    \endlist

    \sa QPrinter, QPrintDialog, QPageSetupDialog, QPrintPreviewDialog

*/

/*!

    \enum QPrintPreviewWidget::ViewMode

    This enum is used to describe the view mode of the preview widget.

    \value SinglePageView   A mode where single pages in the preview

                            is viewed.

    \value FacingPagesView  A mode where the facing pages in the preview

                            is viewed.

    \value AllPagesView     A view mode where all the pages in the preview

                            is viewed.

*/

/*!

    \enum QPrintPreviewWidget::ZoomMode

    This enum is used to describe zoom mode of the preview widget.

    \value CustomZoom  The zoom is set to a custom zoom value.

    \value FitToWidth  This mode fits the current page to the width of the view.

    \value FitInView   This mode fits the current page inside the view.

*/

/*!

    Constructs a QPrintPreviewWidget based on \a printer and with \a

    parent as the parent widget. The widget flags \a flags are passed on

    to the QWidget constructor.

    \sa QWidget::setWindowFlags()

*/

QPrintPreviewWidget::QPrintPreviewWidget(QPrinter *printer, QWidget *parent, Qt::WindowFlags flags)

    : QWidget(*new QPrintPreviewWidgetPrivate, parent, flags)

{

    Q_D(QPrintPreviewWidget);

    d->printer = printer;