qgslayoutmultiframe.cpp

Go to the documentation of this file.

/***************************************************************************
                              qgslayoutmultiframe.cpp
                              -----------------------
    begin                : October 2017
    copyright            : (C) 2017 by Nyall Dawson
    email                : nyall dot dawson at gmail dot com
 ***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
#include "qgslayoutmultiframe.h"
 
#include "qgsexpressioncontextutils.h"
#include "qgslayout.h"
#include "qgslayoutframe.h"
#include "qgslayoutmultiframeundocommand.h"
#include "qgslayoutpagecollection.h"
#include "qgslayoutundostack.h"
 
#include <QString>
#include <QUuid>
 
#include "moc_qgslayoutmultiframe.cpp"
 
using namespace Qt::StringLiterals;
 
QgsLayoutMultiFrame::QgsLayoutMultiFrame( QgsLayout *layout )
  : QgsLayoutObject( layout )
  , mUuid( QUuid::createUuid().toString() )
{
  mLayout->addMultiFrame( this );
 
  connect( mLayout->pageCollection(), &QgsLayoutPageCollection::changed, this, &QgsLayoutMultiFrame::handlePageChange );
}
 
QgsLayoutMultiFrame::~QgsLayoutMultiFrame()
{
  deleteFrames();
}
 
QSizeF QgsLayoutMultiFrame::fixedFrameSize( const int frameIndex ) const
{
  Q_UNUSED( frameIndex )
  return QSizeF( 0, 0 );
}
 
QSizeF QgsLayoutMultiFrame::minFrameSize( const int frameIndex ) const
{
  Q_UNUSED( frameIndex )
  return QSizeF( 0, 0 );
}
 
double QgsLayoutMultiFrame::findNearbyPageBreak( double yPos )
{
  return yPos;
}
 
void QgsLayoutMultiFrame::addFrame( QgsLayoutFrame *frame, bool recalcFrameSizes )
{
  if ( !frame || mFrameItems.contains( frame ) )
    return;
 
  mFrameItems.push_back( frame );
  frame->mMultiFrame = this;
  connect( frame, &QgsLayoutItem::sizePositionChanged, this, &QgsLayoutMultiFrame::recalculateFrameSizes );
  connect( frame, &QgsLayoutFrame::destroyed, this, [this, frame] { handleFrameRemoval( frame ); } );
  if ( mLayout && !frame->scene() )
  {
    mLayout->addLayoutItem( frame );
  }
 
  if ( recalcFrameSizes )
  {
    recalculateFrameSizes();
  }
}
 
void QgsLayoutMultiFrame::setResizeMode( ResizeMode mode )
{
  if ( mode != mResizeMode )
  {
    mLayout->undoStack()->beginMacro( tr( "Change Resize Mode" ) );
    mResizeMode = mode;
    recalculateFrameSizes();
    mLayout->undoStack()->endMacro();
    emit changed();
  }
}
 
QList<QgsLayoutFrame *> QgsLayoutMultiFrame::frames() const
{
  return mFrameItems;
}
 
void QgsLayoutMultiFrame::recalculateFrameSizes()
{
  if ( mFrameItems.empty() )
  {
    return;
  }
 
  QSizeF size = totalSize();
  double totalHeight = size.height();
 
  if ( totalHeight < 1 )
  {
    return;
  }
 
  if ( mBlockUndoCommands )
    mLayout->undoStack()->blockCommands( true );
 
  double currentY = 0;
  double currentHeight = 0;
  QgsLayoutFrame *currentItem = nullptr;
 
  for ( int i = 0; i < mFrameItems.size(); ++i )
  {
    if ( mResizeMode != RepeatOnEveryPage && currentY >= totalHeight )
    {
      if ( mResizeMode == RepeatUntilFinished || mResizeMode == ExtendToNextPage ) //remove unneeded frames in extent mode
      {
        bool removingPages = true;
        for ( int j = mFrameItems.size(); j > i; --j )
        {
          int numPagesBefore = mLayout->pageCollection()->pageCount();
          removeFrame( j - 1, removingPages );
          //if removing the frame didn't also remove the page, then stop removing pages
          removingPages = removingPages && ( mLayout->pageCollection()->pageCount() < numPagesBefore );
        }
        return;
      }
    }
 
    currentItem = mFrameItems.value( i );
    currentHeight = currentItem->rect().height();
    if ( mResizeMode == RepeatOnEveryPage )
    {
      currentItem->setContentSection( QRectF( 0, 0, currentItem->rect().width(), currentHeight ) );
    }
    else
    {
      currentHeight = findNearbyPageBreak( currentY + currentHeight ) - currentY;
      currentItem->setContentSection( QRectF( 0, currentY, currentItem->rect().width(), currentHeight ) );
    }
    currentItem->update();
    currentY += currentHeight;
  }
 
  //at end of frames but there is  still content left. Add other pages if ResizeMode ==
  if ( mLayout->pageCollection()->pageCount() > 0 && currentItem && mResizeMode != UseExistingFrames )
  {
    while ( ( mResizeMode == RepeatOnEveryPage ) || currentY < totalHeight )
    {
      //find out on which page the lower left point of the last frame is
      int page = mLayout->pageCollection()->predictPageNumberForPoint( QPointF( 0, currentItem->pos().y() + currentItem->rect().height() ) ) + 1;
 
      if ( mResizeMode == RepeatOnEveryPage )
      {
        if ( page >= mLayout->pageCollection()->pageCount() )
        {
          break;
        }
      }
      else
      {
        //add new pages if required
        for ( int p = mLayout->pageCollection()->pageCount() - 1; p < page; ++p )
        {
          mLayout->pageCollection()->extendByNewPage();
        }
      }
 
      double currentPageHeight = mLayout->pageCollection()->page( page )->rect().height();
 
      double frameHeight = 0;
      switch ( mResizeMode )
      {
        case RepeatUntilFinished:
        case RepeatOnEveryPage:
        {
          frameHeight = currentItem->rect().height();
          break;
        }
        case ExtendToNextPage:
        {
          frameHeight = ( currentY + currentPageHeight ) > totalHeight ? totalHeight - currentY : currentPageHeight;
          break;
        }
 
        case UseExistingFrames:
          break;
      }
 
      double newFrameY = mLayout->pageCollection()->page( page )->pos().y();
      if ( mResizeMode == RepeatUntilFinished || mResizeMode == RepeatOnEveryPage )
      {
        newFrameY += currentItem->pagePos().y();
      }
 
      //create new frame
      QgsLayoutFrame *newFrame = createNewFrame( currentItem, QPointF( currentItem->pos().x(), newFrameY ), QSizeF( currentItem->rect().width(), frameHeight ) );
 
      if ( mResizeMode == RepeatOnEveryPage )
      {
        newFrame->setContentSection( QRectF( 0, 0, newFrame->rect().width(), newFrame->rect().height() ) );
        currentY += frameHeight;
      }
      else
      {
        double contentHeight = findNearbyPageBreak( currentY + newFrame->rect().height() ) - currentY;
        newFrame->setContentSection( QRectF( 0, currentY, newFrame->rect().width(), contentHeight ) );
        currentY += contentHeight;
      }
 
      currentItem = newFrame;
    }
  }
 
  if ( mBlockUndoCommands )
    mLayout->undoStack()->blockCommands( false );
}
 
void QgsLayoutMultiFrame::recalculateFrameRects()
{
  if ( mFrameItems.empty() )
  {
    //no frames, nothing to do
    return;
  }
 
  const QList< QgsLayoutFrame * > frames = mFrameItems;
  for ( QgsLayoutFrame *frame : frames )
  {
    frame->refreshItemSize();
  }
}
 
void QgsLayoutMultiFrame::refreshDataDefinedProperty( const QgsLayoutObject::DataDefinedProperty )
{}
 
QgsLayoutFrame *QgsLayoutMultiFrame::createNewFrame( QgsLayoutFrame *currentFrame, QPointF pos, QSizeF size )
{
  if ( !currentFrame )
  {
    return nullptr;
  }
 
  QgsLayoutFrame *newFrame = new QgsLayoutFrame( mLayout, this );
  newFrame->attemptSetSceneRect( QRectF( pos.x(), pos.y(), size.width(), size.height() ) );
 
  //copy some settings from the parent frame
  newFrame->setBackgroundColor( currentFrame->backgroundColor() );
  newFrame->setBackgroundEnabled( currentFrame->hasBackground() );
  newFrame->setBlendMode( currentFrame->blendMode() );
  newFrame->setFrameEnabled( currentFrame->frameEnabled() );
  newFrame->setFrameStrokeColor( currentFrame->frameStrokeColor() );
  newFrame->setFrameJoinStyle( currentFrame->frameJoinStyle() );
  newFrame->setFrameStrokeWidth( currentFrame->frameStrokeWidth() );
  newFrame->setItemOpacity( currentFrame->itemOpacity() );
  newFrame->setHideBackgroundIfEmpty( currentFrame->hideBackgroundIfEmpty() );
 
  addFrame( newFrame, false );
 
  return newFrame;
}
 
QString QgsLayoutMultiFrame::displayName() const
{
  return tr( "<Multiframe>" );
}
 
QgsAbstractLayoutUndoCommand *QgsLayoutMultiFrame::createCommand( const QString &text, int id, QUndoCommand *parent )
{
  return new QgsLayoutMultiFrameUndoCommand( this, text, id, parent );
}
 
QgsExpressionContext QgsLayoutMultiFrame::createExpressionContext() const
{
  QgsExpressionContext context = QgsLayoutObject::createExpressionContext();
  context.appendScope( QgsExpressionContextUtils::multiFrameScope( this ) );
  return context;
}
 
void QgsLayoutMultiFrame::beginCommand( const QString &commandText, QgsLayoutMultiFrame::UndoCommand command )
{
  if ( !mLayout )
    return;
 
  mLayout->undoStack()->beginCommand( this, commandText, command );
}
 
void QgsLayoutMultiFrame::endCommand()
{
  if ( mLayout )
    mLayout->undoStack()->endCommand();
}
 
void QgsLayoutMultiFrame::cancelCommand()
{
  if ( mLayout )
    mLayout->undoStack()->cancelCommand();
}
 
void QgsLayoutMultiFrame::finalizeRestoreFromXml()
{
  for ( int i = 0; i < mFrameUuids.count(); ++i )
  {
    QgsLayoutFrame *frame = nullptr;
    const QString uuid = mFrameUuids.at( i );
    if ( !uuid.isEmpty() )
    {
      QgsLayoutItem *item = mLayout->itemByUuid( uuid, true );
      frame = qobject_cast< QgsLayoutFrame * >( item );
    }
    if ( !frame )
    {
      const QString templateUuid = mFrameTemplateUuids.at( i );
      if ( !templateUuid.isEmpty() )
      {
        QgsLayoutItem *item = mLayout->itemByTemplateUuid( templateUuid );
        frame = qobject_cast< QgsLayoutFrame * >( item );
      }
    }
 
    if ( frame )
    {
      addFrame( frame );
    }
  }
}
 
void QgsLayoutMultiFrame::refresh()
{
  QgsLayoutObject::refresh();
  refreshDataDefinedProperty();
}
 
void QgsLayoutMultiFrame::handleFrameRemoval( QgsLayoutFrame *frame )
{
  if ( mBlockUpdates )
    return;
 
  if ( !frame )
  {
    return;
  }
  int index = mFrameItems.indexOf( frame );
  if ( index == -1 )
  {
    return;
  }
 
  mFrameItems.removeAt( index );
  if ( !mFrameItems.isEmpty() )
  {
    if ( resizeMode() != QgsLayoutMultiFrame::RepeatOnEveryPage && !mIsRecalculatingSize )
    {
      //removing a frame forces the multi frame to UseExistingFrames resize mode
      //otherwise the frame may not actually be removed, leading to confusing ui behavior
      mResizeMode = QgsLayoutMultiFrame::UseExistingFrames;
      emit changed();
      recalculateFrameSizes();
    }
  }
}
 
void QgsLayoutMultiFrame::handlePageChange()
{
  if ( mLayout->pageCollection()->pageCount() < 1 )
  {
    return;
  }
 
  if ( mResizeMode != RepeatOnEveryPage )
  {
    return;
  }
 
  //remove items beginning on non-existing pages
  for ( int i = mFrameItems.size() - 1; i >= 0; --i )
  {
    QgsLayoutFrame *frame = mFrameItems.at( i );
    int page = mLayout->pageCollection()->predictPageNumberForPoint( frame->pos() );
    if ( page >= mLayout->pageCollection()->pageCount() )
    {
      removeFrame( i );
    }
  }
 
  if ( !mFrameItems.empty() )
  {
    //page number of the last item
    QgsLayoutFrame *lastFrame = mFrameItems.last();
    int lastItemPage = mLayout->pageCollection()->predictPageNumberForPoint( lastFrame->pos() );
 
    for ( int i = lastItemPage + 1; i < mLayout->pageCollection()->pageCount(); ++i )
    {
      //copy last frame to current page
      auto newFrame = std::make_unique< QgsLayoutFrame >( mLayout, this );
 
      newFrame->attemptSetSceneRect( QRectF( lastFrame->pos().x(), mLayout->pageCollection()->page( i )->pos().y() + lastFrame->pagePos().y(), lastFrame->rect().width(), lastFrame->rect().height() ) );
      lastFrame = newFrame.get();
      addFrame( newFrame.release(), false );
    }
  }
 
  recalculateFrameSizes();
  update();
}
 
void QgsLayoutMultiFrame::removeFrame( int i, const bool removeEmptyPages )
{
  if ( i >= mFrameItems.count() )
  {
    return;
  }
 
  QgsLayoutFrame *frameItem = mFrameItems.at( i );
  if ( mLayout )
  {
    mIsRecalculatingSize = true;
    int pageNumber = frameItem->page();
    //remove item, but don't create undo command
    mLayout->undoStack()->blockCommands( true );
    mLayout->removeLayoutItem( frameItem );
    //if frame was the only item on the page, remove the page
    if ( removeEmptyPages && mLayout->pageCollection()->pageIsEmpty( pageNumber ) )
    {
      mLayout->pageCollection()->deletePage( pageNumber );
    }
    mLayout->undoStack()->blockCommands( false );
    mIsRecalculatingSize = false;
  }
 
  if ( i >= mFrameItems.count() )
  {
    return;
  }
 
  mFrameItems.removeAt( i );
}
 
void QgsLayoutMultiFrame::update()
{
  for ( QgsLayoutFrame *frame : std::as_const( mFrameItems ) )
  {
    frame->update();
  }
}
 
void QgsLayoutMultiFrame::deleteFrames()
{
  mBlockUpdates = true;
  ResizeMode bkResizeMode = mResizeMode;
  mResizeMode = UseExistingFrames;
  mLayout->undoStack()->blockCommands( true );
  for ( QgsLayoutFrame *frame : std::as_const( mFrameItems ) )
  {
    mLayout->removeLayoutItem( frame );
  }
  mLayout->undoStack()->blockCommands( false );
  mFrameItems.clear();
  mResizeMode = bkResizeMode;
  mBlockUpdates = false;
}
 
QgsLayoutFrame *QgsLayoutMultiFrame::frame( int i ) const
{
  if ( i < 0 || i >= mFrameItems.size() )
  {
    return nullptr;
  }
  return mFrameItems.at( i );
}
 
int QgsLayoutMultiFrame::frameIndex( QgsLayoutFrame *frame ) const
{
  return mFrameItems.indexOf( frame );
}
 
bool QgsLayoutMultiFrame::writeXml( QDomElement &parentElement, QDomDocument &doc, const QgsReadWriteContext &context, bool includeFrames ) const
{
  QDomElement element = doc.createElement( u"LayoutMultiFrame"_s );
  element.setAttribute( u"resizeMode"_s, mResizeMode );
  element.setAttribute( u"uuid"_s, mUuid );
  element.setAttribute( u"templateUuid"_s, mUuid );
  element.setAttribute( u"type"_s, type() );
 
  for ( QgsLayoutFrame *frame : mFrameItems )
  {
    if ( !frame )
      continue;
 
    QDomElement childItem = doc.createElement( u"childFrame"_s );
    childItem.setAttribute( u"uuid"_s, frame->uuid() );
    childItem.setAttribute( u"templateUuid"_s, frame->uuid() );
 
    if ( includeFrames )
    {
      frame->writeXml( childItem, doc, context );
    }
 
    element.appendChild( childItem );
  }
 
  writeObjectPropertiesToElement( element, doc, context );
  writePropertiesToElement( element, doc, context );
  parentElement.appendChild( element );
  return true;
}
 
bool QgsLayoutMultiFrame::readXml( const QDomElement &element, const QDomDocument &doc, const QgsReadWriteContext &context, bool includeFrames )
{
  if ( element.nodeName() != "LayoutMultiFrame"_L1 )
  {
    return false;
  }
 
  mBlockUndoCommands = true;
  mLayout->undoStack()->blockCommands( true );
 
  readObjectPropertiesFromElement( element, doc, context );
 
  mUuid = element.attribute( u"uuid"_s, QUuid::createUuid().toString() );
  mTemplateUuid = element.attribute( u"templateUuid"_s, QUuid::createUuid().toString() );
  mResizeMode = static_cast< ResizeMode >( element.attribute( u"resizeMode"_s, u"0"_s ).toInt() );
 
  deleteFrames();
  mFrameUuids.clear();
  mFrameTemplateUuids.clear();
  QDomNodeList elementNodes = element.elementsByTagName( u"childFrame"_s );
  for ( int i = 0; i < elementNodes.count(); ++i )
  {
    QDomNode elementNode = elementNodes.at( i );
    if ( !elementNode.isElement() )
      continue;
 
    QDomElement frameElement = elementNode.toElement();
 
    QString uuid = frameElement.attribute( u"uuid"_s );
    mFrameUuids << uuid;
    QString templateUuid = frameElement.attribute( u"templateUuid"_s );
    mFrameTemplateUuids << templateUuid;
 
    if ( includeFrames )
    {
      QDomNodeList frameNodes = frameElement.elementsByTagName( u"LayoutItem"_s );
      if ( !frameNodes.isEmpty() )
      {
        QDomElement frameItemElement = frameNodes.at( 0 ).toElement();
        auto newFrame = std::make_unique< QgsLayoutFrame >( mLayout, this );
        newFrame->readXml( frameItemElement, doc, context );
        addFrame( newFrame.release(), false );
      }
    }
  }
 
  bool result = readPropertiesFromElement( element, doc, context );
 
  mBlockUndoCommands = false;
  mLayout->undoStack()->blockCommands( false );
  return result;
}
 
bool QgsLayoutMultiFrame::writePropertiesToElement( QDomElement &, QDomDocument &, const QgsReadWriteContext & ) const
{
  return true;
}
 
bool QgsLayoutMultiFrame::readPropertiesFromElement( const QDomElement &, const QDomDocument &, const QgsReadWriteContext & )
{
  return true;
}