#include "toonzqt/schematicviewer.h"

// TnzQt includes

#include "toonzqt/fxtypes.h"

#include "toonzqt/schematicnode.h"

#include "toonzqt/fxschematicnode.h"

#include "toonzqt/schematicgroupeditor.h"

#include "toonzqt/stageschematicscene.h"

#include "toonzqt/fxschematicscene.h"

#include "toonzqt/menubarcommand.h"

#include "toonzqt/tselectionhandle.h"

#include "toonzqt/gutil.h"

#include "toonzqt/imageutils.h"

#include "toonzqt/dvscrollwidget.h"

// TnzLib includes

#include "toonz/txsheethandle.h"

#include "toonz/tcolumnhandle.h"

#include "toonz/tobjecthandle.h"

#include "toonz/tfxhandle.h"

#include "toonz/txsheet.h"

#include "toonz/txshlevelcolumn.h"

#include "toonz/tcolumnfx.h"

#include "toonz/txshzeraryfxcolumn.h"

#include "toonz/preferences.h"

#include "toonz/fxdag.h"

#include "toonz/tapplication.h"

#include "toonz/tscenehandle.h"

#include "toonz/txshleveltypes.h"

#include "../toonz/menubarcommandids.h"

#include "tools/cursormanager.h"

#include "tools/cursors.h"

// Qt includes

#include <qgraphicsscenemouseevent></qgraphicsscenemouseevent>

#include <qmouseevent></qmouseevent>

#include <qgraphicsitem></qgraphicsitem>

#include <qtoolbar></qtoolbar>

#include <qtoolbutton></qtoolbutton>

#include <qmenu></qmenu>

#include <qicon></qicon>

#include <qaction></qaction>

#include <qmainwindow></qmainwindow>

#include <qvboxlayout></qvboxlayout>

#include <qgestureevent></qgestureevent>

#include <qdebug></qdebug>

// STD includes

#include "assert.h"

#include "math.h"

namespace {

class SchematicZoomer final : public ImageUtils::ShortcutZoomer {

public:

  SchematicZoomer(QWidget *parent) : ShortcutZoomer(parent) {}

  bool zoom(bool zoomin, bool resetZoom) override {

    static_cast<schematicsceneviewer *="">(getWidget())->zoomQt(zoomin, resetZoom);</schematicsceneviewer>

    return true;

  }

};

}  // namespace

//==================================================================

//

// SchematicScene

//

//==================================================================

SchematicScene::SchematicScene(QWidget *parent) : QGraphicsScene(parent) {

  setSceneRect(0, 0, 50000, 50000);

  setItemIndexMethod(NoIndex);

}

//------------------------------------------------------------------

SchematicScene::~SchematicScene() { clearAllItems(); }

//------------------------------------------------------------------

void SchematicScene::showEvent(QShowEvent *se) {

  TSelectionHandle *selHandle = TSelectionHandle::getCurrent();

  connect(selHandle, SIGNAL(selectionSwitched(TSelection *, TSelection *)),

          this, SLOT(onSelectionSwitched(TSelection *, TSelection *)));

  clearSelection();

}

//------------------------------------------------------------------

void SchematicScene::hideEvent(QHideEvent *se) {

  TSelectionHandle *selHandle = TSelectionHandle::getCurrent();

  disconnect(selHandle, SIGNAL(selectionSwitched(TSelection *, TSelection *)),

             this, SLOT(onSelectionSwitched(TSelection *, TSelection *)));

}

//------------------------------------------------------------------

/*! Removes and then deletes all item in the scene.

*/

void SchematicScene::clearAllItems() {

  clearSelection();

  m_highlightedLinks.clear();

  QList<schematicwindoweditor *=""> editors;</schematicwindoweditor>

  QList<schematicnode *=""> nodes;</schematicnode>

  QList<schematiclink *=""> links;</schematiclink>

  int i;

  QList<qgraphicsitem *=""> sceneItems = items();</qgraphicsitem>

  int size                          = sceneItems.size();

  // create nodes and links list

  for (i = 0; i < size; i++) {

    QGraphicsItem *item           = sceneItems.at(i);

    SchematicWindowEditor *editor = dynamic_cast<schematicwindoweditor *="">(item);</schematicwindoweditor>

    SchematicNode *node           = dynamic_cast<schematicnode *="">(item);</schematicnode>

    SchematicLink *link           = dynamic_cast<schematiclink *="">(item);</schematiclink>

    if (editor) editors.append(editor);

    if (node) nodes.append(node);

    if (link) links.append(link);

  }

  while (links.size() > 0) {

    SchematicLink *link = links.back();

    removeItem(link);

    links.removeLast();

    SchematicPort *startPort = link->getStartPort();

    SchematicPort *endPort   = link->getEndPort();

    if (startPort) startPort->removeLink(link);

    if (endPort) endPort->removeLink(link);

    delete link;

  }

  while (editors.size() > 0) {

    SchematicWindowEditor *editor = editors.back();

    removeItem(editor);

    editors.removeLast();

    delete editor;

  }

  while (nodes.size() > 0) {

    SchematicNode *node = nodes.back();

    removeItem(node);

    nodes.removeLast();

    delete node;

  }

  assert(items().size() == 0);

}

//------------------------------------------------------------------

/*! check if any item exists in the rect

*/

bool SchematicScene::isAnEmptyZone(const QRectF &rect) {

  QList<qgraphicsitem *=""> allItems = items();</qgraphicsitem>

  for (auto const level : allItems) {

    SchematicNode *node = dynamic_cast<schematicnode *="">(level);</schematicnode>

    if (!node) continue;

    FxSchematicNode *fxNode = dynamic_cast<fxschematicnode *="">(node);</fxschematicnode>

    if (fxNode && fxNode->isA(eXSheetFx)) continue;

    if (node->boundingRect().translated(node->scenePos()).intersects(rect))

      return false;

  }

  return true;

}

//------------------------------------------------------------------

QVector<schematicnode *=""> SchematicScene::getPlacedNode(SchematicNode *node) {</schematicnode>

  QRectF rect = node->boundingRect().translated(node->scenePos());

  QList<qgraphicsitem *=""> allItems = items();</qgraphicsitem>

  QVector<schematicnode *=""> nodes;</schematicnode>

  for (auto const item : allItems) {

    SchematicNode *placedNode = dynamic_cast<schematicnode *="">(item);</schematicnode>

    if (!placedNode || placedNode == node) continue;

    QRectF nodeRect =

        placedNode->boundingRect().translated(placedNode->scenePos());

    QRectF enlargedRect = rect.adjusted(-10, -10, 10, 10);

    if (enlargedRect.contains(nodeRect)) nodes.push_back(placedNode);

  }

  return nodes;

}

//==================================================================

//

// SchematicSceneViewer

//

//==================================================================

SchematicSceneViewer::SchematicSceneViewer(QWidget *parent)

    : QGraphicsView(parent)

    , m_buttonState(Qt::NoButton)

    , m_oldWinPos()

    , m_oldScenePos()

    , m_firstShowing(true) {

  setObjectName("SchematicSceneViewer");

  setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);

  setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);

  setDragMode(QGraphicsView::NoDrag);

  setTransformationAnchor(QGraphicsView::NoAnchor);

  setRenderHint(QPainter::SmoothPixmapTransform);

  setRenderHint(QPainter::TextAntialiasing);

  setRenderHint(QPainter::Antialiasing);

  setInteractive(true);

  setViewportUpdateMode(QGraphicsView::SmartViewportUpdate);

  show();

  setAttribute(Qt::WA_AcceptTouchEvents);

  grabGesture(Qt::SwipeGesture);

  grabGesture(Qt::PanGesture);

  grabGesture(Qt::PinchGesture);

}

//------------------------------------------------------------------

SchematicSceneViewer::~SchematicSceneViewer() {}

//------------------------------------------------------------------

/*! Reimplemets the QGraphicsView::mousePressEvent()

*/

void SchematicSceneViewer::mousePressEvent(QMouseEvent *me) {

  if (m_gestureActive && m_touchDevice == QTouchDevice::TouchScreen &&

      !m_stylusUsed) {

    return;

  }

  m_buttonState = me->button();

  m_oldWinPos   = me->pos();

  m_oldScenePos = mapToScene(m_oldWinPos);

  if (m_buttonState == Qt::LeftButton) {

    if (m_cursorMode == CursorMode::Zoom) {

      m_zoomPoint = me->pos();

      m_zooming   = true;

      return;

    } else if (m_cursorMode == CursorMode::Hand) {

      m_firstPanPoint = mapToScene(me->pos());

      m_panning       = true;

      return;

    }

  } else if (m_buttonState == Qt::MidButton) {

    m_firstPanPoint = mapToScene(me->pos());

  }

  bool drawRect                       = true;

  QList<qgraphicsitem *=""> pointedItems = items(me->pos());</qgraphicsitem>

  int i;

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

    SchematicWindowEditor *editor =

        dynamic_cast<schematicwindoweditor *="">(pointedItems[i]);</schematicwindoweditor>

    if (!editor) {

      drawRect = false;

      break;

    }

  }

  if (m_buttonState == Qt::LeftButton && drawRect)

    setDragMode(QGraphicsView::RubberBandDrag);

  QGraphicsView::mousePressEvent(me);

}

//------------------------------------------------------------------

/*! Reimplemets the QGraphicsView::mouseMoveEvent()

*/

void SchematicSceneViewer::mouseMoveEvent(QMouseEvent *me) {

  if (m_gestureActive && m_touchDevice == QTouchDevice::TouchScreen &&

      !m_stylusUsed) {

    return;

  }

  QPoint currWinPos    = me->pos();

  QPointF currScenePos = mapToScene(currWinPos);

  if ((m_cursorMode == CursorMode::Hand && m_panning) ||

      m_buttonState == Qt::MidButton) {

    QPointF deltaPoint = ((mapToScene(me->pos()) * getDevPixRatio()) -

                          (m_firstPanPoint * getDevPixRatio()));

    panQt(deltaPoint);

  } else {

    if (m_cursorMode == CursorMode::Zoom && m_zooming) {

      //	  int deltaY = (m_prevWinPos.y() - me->pos().y()) * 10;

      int deltaY     = (m_oldWinPos.y() - me->pos().y()) * 10;

      double factorY = exp(deltaY * 0.001);

      changeScale(m_zoomPoint, factorY);

    }

    m_oldWinPos   = currWinPos;

    m_oldScenePos = currScenePos;

  }

  QGraphicsView::mouseMoveEvent(me);

}

//------------------------------------------------------------------

/*! Reimplemets the QGraphicsView::mouseReleaseEvent()

*/

void SchematicSceneViewer::mouseReleaseEvent(QMouseEvent *me) {

  // for touchscreens but not touchpads...

  if (m_gestureActive && m_touchDevice == QTouchDevice::TouchScreen &&

      !m_stylusUsed) {

    m_gestureActive = false;

    m_zooming       = false;

    m_panning       = false;

    m_stylusUsed    = false;

    m_scaleFactor   = 0.0;

    return;

  }

  m_zooming    = false;

  m_panning    = false;

  m_stylusUsed = false;

  m_buttonState = Qt::NoButton;

  QGraphicsView::mouseReleaseEvent(me);

  setDragMode(QGraphicsView::NoDrag);

  // update();

}

//------------------------------------------------------------------

void SchematicSceneViewer::mouseDoubleClickEvent(QMouseEvent *event) {

  if (m_gestureActive && !m_stylusUsed) {

    fitScene();

    m_gestureActive = false;

    return;

  }

  QGraphicsView::mouseDoubleClickEvent(event);

}

//------------------------------------------------------------------

void SchematicSceneViewer::keyPressEvent(QKeyEvent *ke) {

  ke->ignore();

  QGraphicsView::keyPressEvent(ke);

  if (!ke->isAccepted()) SchematicZoomer(this).exec(ke);

}

//------------------------------------------------------------------

/*! Reimplemets the QGraphicsView::wheelEvent()

*/

void SchematicSceneViewer::wheelEvent(QWheelEvent *me) {

  if ((m_gestureActive == true && m_touchDevice == QTouchDevice::TouchScreen) ||

      m_gestureActive == false) {

    me->accept();

    double factor = exp(me->delta() * 0.001);

    changeScale(me->pos(), factor);

  }

}

//------------------------------------------------------------------

void SchematicSceneViewer::zoomQt(bool zoomin, bool resetZoom) {

#if QT_VERSION >= 0x050000

  double scale2 = matrix().determinant();

#else

  double scale2 = matrix().det();

#endif

  // if the zoom scale is changed over 100% in FxSchematic, call updateScene

  bool beforeIsLarge = (scale2 >= 1.0);

  if (resetZoom ||

      ((scale2 < 100000 || !zoomin) && (scale2 > 0.001 * 0.05 || zoomin))) {

    double oldZoomScale = sqrt(scale2);

    double zoomScale    = resetZoom ? 1 : ImageUtils::getQuantizedZoomFactor(

                                           oldZoomScale, zoomin);

    QMatrix scale =

        QMatrix().scale(zoomScale / oldZoomScale, zoomScale / oldZoomScale);

    // See QGraphicsView::mapToScene()'s doc for details

    QRect rect(0, 0, width(), height());

    QRectF sceneCenterRect(

        mapToScene(QRect(rect.center(), QSize(2, 2))).boundingRect());

    setMatrix(scale, true);

    centerOn(sceneCenterRect.center());

#if QT_VERSION >= 0x050000

    bool afterIsLarge = (matrix().determinant() >= 1.0);

#else

    bool afterIsLarge  = (matrix().det() >= 1.0);

#endif

    if (beforeIsLarge != afterIsLarge) {

      FxSchematicScene *fxScene = qobject_cast<fxschematicscene *="">(scene());</fxschematicscene>

      if (fxScene) fxScene->updateScene();

    }

  }

}

//------------------------------------------------------------------

/*! The view is scaled around the point \b winPos by \b scaleFactor;

*/

void SchematicSceneViewer::changeScale(const QPoint &winPos,

                                       qreal scaleFactor) {

#if QT_VERSION >= 0x050000

  bool beforeIsLarge = (matrix().determinant() >= 1.0);

#else

  bool beforeIsLarge   = (matrix().det() >= 1.0);

#endif

  QPointF startScenePos = mapToScene(winPos);

  QMatrix scale         = QMatrix().scale(scaleFactor, scaleFactor);

  setMatrix(scale, true);

  QPointF endScenePos = mapToScene(winPos);

  QPointF delta       = endScenePos - startScenePos;

  translate(delta.x(), delta.y());

#if QT_VERSION >= 0x050000

  bool afterIsLarge = (matrix().determinant() >= 1.0);

#else

  bool afterIsLarge    = (matrix().det() >= 1.0);

#endif

  if (beforeIsLarge != afterIsLarge) {

    FxSchematicScene *fxScene = qobject_cast<fxschematicscene *="">(scene());</fxschematicscene>

    if (fxScene) fxScene->updateScene();

  }

}

//------------------------------------------------------------------

void SchematicSceneViewer::fitScene() {

  if (scene()) {

#if QT_VERSION >= 0x050000

    bool beforeIsLarge = (matrix().determinant() >= 1.0);

#else

    bool beforeIsLarge = (matrix().det() >= 1.0);

#endif

    QRectF rect = scene()->itemsBoundingRect();

    fitInView(rect, Qt::KeepAspectRatio);

#if QT_VERSION >= 0x050000

    bool afterIsLarge = (matrix().determinant() >= 1.0);

#else

    bool afterIsLarge  = (matrix().det() >= 1.0);

#endif

    if (beforeIsLarge != afterIsLarge) {

      FxSchematicScene *fxScene = qobject_cast<fxschematicscene *="">(scene());</fxschematicscene>

      if (fxScene) fxScene->updateScene();

    }

  }

}

//------------------------------------------------------------------

void SchematicSceneViewer::centerOnCurrent() {

  SchematicScene *schematicScene = dynamic_cast<schematicscene *="">(scene());</schematicscene>

  QGraphicsItem *node            = schematicScene->getCurrentNode();

  if (node) centerOn(node);

}

//------------------------------------------------------------------

void SchematicSceneViewer::reorderScene() {

  SchematicScene *schematicScene = dynamic_cast<schematicscene *="">(scene());</schematicscene>

  schematicScene->reorderScene();

}

//------------------------------------------------------------------

void SchematicSceneViewer::normalizeScene() {

#if QT_VERSION >= 0x050000

  bool beforeIsLarge = (matrix().determinant() >= 1.0);

#else

  bool beforeIsLarge   = (matrix().det() >= 1.0);

#endif

  // See QGraphicsView::mapToScene()'s doc for details

  QRect rect(0, 0, width(), height());

  QRectF sceneCenterRect(

      mapToScene(QRect(rect.center(), QSize(2, 2))).boundingRect());

  resetMatrix();

#if defined(MACOSX)

  scale(1.32, 1.32);

#endif

  centerOn(sceneCenterRect.center());

#if QT_VERSION >= 0x050000

  bool afterIsLarge = (matrix().determinant() >= 1.0);

#else

  bool afterIsLarge    = (matrix().det() >= 1.0);

#endif

  if (beforeIsLarge != afterIsLarge) {

    FxSchematicScene *fxScene = qobject_cast<fxschematicscene *="">(scene());</fxschematicscene>

    if (fxScene) fxScene->updateScene();

  }

}

//------------------------------------------------------------------

void SchematicSceneViewer::panQt(const QPointF &delta) {

  setInteractive(false);

  // I need to disable QGraphicsView event handling to avoid the generation of

  // 'virtual' mouseMoveEvent

  translate(delta.x(), delta.y());

  // translate has changed the matrix affecting the mapToScene() method. I

  // have to recompute currScenePos

  setInteractive(true);

}

void SchematicSceneViewer::showEvent(QShowEvent *se) {

  QGraphicsView::showEvent(se);

  if (m_firstShowing) {

    m_firstShowing = false;

    QRectF rect    = scene()->itemsBoundingRect();

    resetMatrix();

    centerOn(rect.center());

  }

}

//------------------------------------------------------------------

void SchematicSceneViewer::enterEvent(QEvent *e) {

  switch (m_cursorMode) {

  case CursorMode::Hand:

    setToolCursor(this, ToolCursor::PanCursor);

    break;

  case CursorMode::Zoom:

    setToolCursor(this, ToolCursor::ZoomCursor);

    break;

  default:

    setToolCursor(this, ToolCursor::StrokeSelectCursor);

    break;

  }

}

//------------------------------------------------------------------

void SchematicSceneViewer::leaveEvent(QEvent *e) { setCursor(Qt::ArrowCursor); }

//------------------------------------------------------------------

void SchematicSceneViewer::tabletEvent(QTabletEvent *e) {

  if (e->type() == QTabletEvent::TabletPress) {

    m_stylusUsed = e->pointerType() ? true : false;

  } else if (e->type() == QTabletEvent::TabletRelease) {

    m_stylusUsed = false;

  }

  e->accept();

}

//------------------------------------------------------------------

void SchematicSceneViewer::gestureEvent(QGestureEvent *e) {

  m_gestureActive = false;

  if (QGesture *swipe = e->gesture(Qt::SwipeGesture)) {

    m_gestureActive = true;

  } else if (QGesture *pan = e->gesture(Qt::PanGesture)) {

    m_gestureActive = true;

  }

  if (QGesture *pinch = e->gesture(Qt::PinchGesture)) {

    QPinchGesture *gesture = static_cast<qpinchgesture *="">(pinch);</qpinchgesture>

    QPinchGesture::ChangeFlags changeFlags = gesture->changeFlags();

    QPoint firstCenter                     = gesture->centerPoint().toPoint();

    if (gesture->state() == Qt::GestureStarted) {

      m_gestureActive = true;

    } else if (gesture->state() == Qt::GestureFinished) {

      m_gestureActive = false;

      m_zooming       = false;

      m_scaleFactor   = 0.0;

    } else {

      if (changeFlags & QPinchGesture::ScaleFactorChanged) {

        double scaleFactor = gesture->scaleFactor();

        // the scale factor makes for too sensitive scaling

        // divide the change in half

        if (scaleFactor > 1) {

          double decimalValue = scaleFactor - 1;

          decimalValue /= 1.5;

          scaleFactor = 1 + decimalValue;

        } else if (scaleFactor < 1) {

          double decimalValue = 1 - scaleFactor;

          decimalValue /= 1.5;

          scaleFactor = 1 - decimalValue;

        }

        if (!m_zooming) {

          double delta = scaleFactor - 1;

          m_scaleFactor += delta;

          if (m_scaleFactor > .2 || m_scaleFactor < -.2) {

            m_zooming = true;

          }

        }

        if (m_zooming) {

          changeScale(firstCenter * getDevPixRatio(), scaleFactor);

        }

        m_gestureActive = true;

      }

      if (changeFlags & QPinchGesture::CenterPointChanged) {

        QPointF centerDelta = (gesture->centerPoint() * getDevPixRatio()) -

                              (gesture->lastCenterPoint() * getDevPixRatio());

        if (centerDelta.manhattanLength() > 1) {

          // panQt(centerDelta.toPoint());

        }

        m_gestureActive = true;

      }

    }

  }

  e->accept();

}

void SchematicSceneViewer::touchEvent(QTouchEvent *e, int type) {

  if (type == QEvent::TouchBegin) {

    m_touchActive   = true;

    m_firstPanPoint = e->touchPoints().at(0).pos();

    // obtain device type

    m_touchDevice = e->device()->type();

  } else if (m_touchActive) {

    // touchpads must have 2 finger panning for tools and navigation to be

    // functional on other devices, 1 finger panning is preferred

    if ((e->touchPoints().count() == 2 &&

         m_touchDevice == QTouchDevice::TouchPad) ||

        (e->touchPoints().count() == 1 &&

         m_touchDevice == QTouchDevice::TouchScreen)) {

      QTouchEvent::TouchPoint panPoint = e->touchPoints().at(0);

      if (!m_panning) {

        QPointF deltaPoint = panPoint.pos() - m_firstPanPoint;

        // minimize accidental and jerky zooming/rotating during 2 finger

        // panning

        if ((deltaPoint.manhattanLength() > 100) && !m_zooming) {

          m_panning = true;

        }

      }

      if (m_panning) {

        QPointF centerDelta = (panPoint.pos() * getDevPixRatio()) -

                              (panPoint.lastPos() * getDevPixRatio());

        panQt(centerDelta.toPoint());

      }

    }

  }

  if (type == QEvent::TouchEnd || type == QEvent::TouchCancel) {

    m_touchActive = false;

    m_panning     = false;

  }

  e->accept();

}

bool SchematicSceneViewer::event(QEvent *e) {

  /*

    switch (e->type()) {

    case QEvent::TabletPress:

      QTabletEvent *te = static_cast<qtabletevent *="">(e);</qtabletevent>

      qDebug() << "[event] TabletPress: pointerType(" << te->pointerType()

               << ") device(" << te->device() << ")";

      break;

    case QEvent::TabletRelease:

      qDebug() << "[event] TabletRelease";

      break;

    case QEvent::TouchBegin:

      qDebug() << "[event] TouchBegin";

      break;

    case QEvent::TouchEnd:

      qDebug() << "[event] TouchEnd";

      break;

    case QEvent::TouchCancel:

      qDebug() << "[event] TouchCancel";

      break;

    case QEvent::MouseButtonPress:

      qDebug() << "[event] MouseButtonPress";

      break;

    case QEvent::MouseButtonDblClick:

      qDebug() << "[event] MouseButtonDblClick";

      break;

    case QEvent::MouseButtonRelease:

      qDebug() << "[event] MouseButtonRelease";

      break;

    case QEvent::Gesture:

      qDebug() << "[event] Gesture";

      break;

    default:

      break;

    }

  */

  if (CommandManager::instance()

          ->getAction(MI_TouchGestureControl)

          ->isChecked()) {

    if (e->type() == QEvent::Gesture) {

      gestureEvent(static_cast<qgestureevent *="">(e));</qgestureevent>

      return true;

    }

    if (e->type() == QEvent::TouchBegin || e->type() == QEvent::TouchEnd ||

        e->type() == QEvent::TouchCancel || e->type() == QEvent::TouchUpdate) {

      touchEvent(static_cast<qtouchevent *="">(e), e->type());</qtouchevent>

      m_gestureActive = true;

      return true;

    }

  }

  return QGraphicsView::event(e);

}

//------------------------------------------------------------------

void SchematicSceneViewer::setCursorMode(CursorMode cursorMode) {

  m_cursorMode = cursorMode;

}

//==================================================================

//

// SchematicViewer

//

//==================================================================

SchematicViewer::SchematicViewer(QWidget *parent)

    : QWidget(parent)

    , m_fullSchematic(true)

    , m_maximizedNode(false)

    , m_sceneHandle(0)

    , m_cursorMode(CursorMode::Select) {

  m_viewer     = new SchematicSceneViewer(this);

  m_stageScene = new StageSchematicScene(this);

  m_fxScene    = new FxSchematicScene(this);

  m_commonToolbar = new QToolBar(m_viewer);

  m_stageToolbar  = new QToolBar(m_viewer);

  m_fxToolbar     = new QToolBar(m_viewer);

  m_swapToolbar   = new QToolBar(m_viewer);

  m_commonToolbar->setObjectName("MediumPaddingToolBar");

  m_stageToolbar->setObjectName("MediumPaddingToolBar");

  m_fxToolbar->setObjectName("MediumPaddingToolBar");

  m_swapToolbar->setObjectName("MediumPaddingToolBar");

  createToolbars();

  createActions();

  // layout

  QVBoxLayout *mainLayout = new QVBoxLayout();

  mainLayout->setMargin(0);

  mainLayout->setSpacing(0);

  {

    mainLayout->addWidget(m_viewer, 1);

    QFrame *bottomFrame = new QFrame(this);

    bottomFrame->setObjectName("SchematicBottomFrame");

    QHBoxLayout *horizontalLayout = new QHBoxLayout();

    horizontalLayout->setMargin(0);

    horizontalLayout->setSpacing(0);

    {

      horizontalLayout->addWidget(m_commonToolbar);

      horizontalLayout->addStretch();

      horizontalLayout->addWidget(m_fxToolbar);

      horizontalLayout->addWidget(m_stageToolbar);

      horizontalLayout->addWidget(m_swapToolbar);

    }

    bottomFrame->setLayout(horizontalLayout);

    mainLayout->addWidget(bottomFrame, 0);

  }

  setLayout(mainLayout);

  connect(m_fxScene, SIGNAL(showPreview(TFxP)), this,

          SIGNAL(showPreview(TFxP)));

  connect(m_fxScene, SIGNAL(doCollapse(const QList<tfxp> &)), this,</tfxp>

          SIGNAL(doCollapse(const QList<tfxp> &)));</tfxp>

  connect(m_stageScene, SIGNAL(doCollapse(QList<tstageobjectid>)), this,</tstageobjectid>

          SIGNAL(doCollapse(QList<tstageobjectid>)));</tstageobjectid>

  connect(m_fxScene, SIGNAL(doExplodeChild(const QList<tfxp> &)), this,</tfxp>

          SIGNAL(doExplodeChild(const QList<tfxp> &)));</tfxp>

  connect(m_stageScene, SIGNAL(doExplodeChild(QList<tstageobjectid>)), this,</tstageobjectid>

          SIGNAL(doExplodeChild(QList<tstageobjectid>)));</tstageobjectid>

  connect(m_stageScene, SIGNAL(editObject()), this, SIGNAL(editObject()));

  connect(m_fxScene, SIGNAL(editObject()), this, SIGNAL(editObject()));

  m_viewer->setScene(m_stageScene);

  m_fxToolbar->hide();

  setFocusProxy(m_viewer);

}

//------------------------------------------------------------------

SchematicViewer::~SchematicViewer() {}

//------------------------------------------------------------------

void SchematicViewer::getNodeColor(int ltype, QColor &nodeColor) {

  switch (ltype) {

  case TZI_XSHLEVEL:

  case OVL_XSHLEVEL:

    nodeColor = getFullcolorColumnColor();

    break;

  case PLI_XSHLEVEL:

    nodeColor = getVectorColumnColor();

    break;

  case TZP_XSHLEVEL:

    nodeColor = getLevelColumnColor();

    break;

  case ZERARYFX_XSHLEVEL:

    nodeColor = getFxColumnColor();

    break;

  case CHILD_XSHLEVEL:

    nodeColor = getChildColumnColor();

    break;

  case MESH_XSHLEVEL:

    nodeColor = getMeshColumnColor();

    break;

  case PLT_XSHLEVEL:

    nodeColor = getPaletteColumnColor();

    break;

  case eNormalFx:

    nodeColor = getNormalFxColor();

    break;

  case eZeraryFx:

    nodeColor = getFxColumnColor();

    break;

  case eMacroFx:

    nodeColor = getMacroFxColor();

    break;

  case eGroupedFx:

    nodeColor = getGroupColor();

    break;

  case eNormalImageAdjustFx:

    nodeColor = getImageAdjustFxColor();

    break;

  case eNormalLayerBlendingFx:

    nodeColor = getLayerBlendingFxColor();

    break;

  case eNormalMatteFx:

    nodeColor = getMatteFxColor();

    break;

  default:

    nodeColor = grey210;

    break;

  }

}

//------------------------------------------------------------------

void SchematicViewer::setApplication(TApplication *app) {

  m_stageScene->setXsheetHandle(app->getCurrentXsheet());

  m_stageScene->setObjectHandle(app->getCurrentObject());

  m_stageScene->setFxHandle(app->getCurrentFx());

  m_stageScene->setColumnHandle(app->getCurrentColumn());

  m_stageScene->setSceneHandle(app->getCurrentScene());

  m_stageScene->setFrameHandle(app->getCurrentFrame());

  m_fxScene->setApplication(app);

}

//------------------------------------------------------------------

void SchematicViewer::updateSchematic() {

  m_stageScene->updateScene();

  m_fxScene->updateScene();

}

//------------------------------------------------------------------

void SchematicViewer::setSchematicScene(SchematicScene *scene) {

  if (scene) {

    m_viewer->setScene(scene);

    m_viewer->centerOn(scene->sceneRect().center());

  }

}

//------------------------------------------------------------------

void SchematicViewer::createToolbars() {

  // Initialize them

  m_stageToolbar->setMovable(false);

  m_stageToolbar->setIconSize(QSize(17, 17));

  m_stageToolbar->setLayoutDirection(Qt::RightToLeft);

  m_stageToolbar->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

  m_commonToolbar->setMovable(false);

  m_commonToolbar->setIconSize(QSize(17, 17));

  m_commonToolbar->setLayoutDirection(Qt::RightToLeft);

  m_commonToolbar->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

  m_fxToolbar->setMovable(false);

  m_fxToolbar->setIconSize(QSize(17, 17));

  m_fxToolbar->setLayoutDirection(Qt::RightToLeft);

  m_fxToolbar->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);

  m_swapToolbar->setMovable(false);

  m_swapToolbar->setIconSize(QSize(17, 17));

  m_swapToolbar->setLayoutDirection(Qt::RightToLeft);

  m_swapToolbar->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);