document.cpp

上传用户:binglan008
上传日期:2007-01-28
资源大小:53508k
文件大小:11k
源码类别:Linux/Unix编程
开发平台:MultiPlatform
  1. /****************************************************************************
  2. **
  3. ** Copyright (C) 2005-2007 Trolltech ASA. All rights reserved.
  4. **
  5. ** This file is part of the demonstration applications of the Qt Toolkit.
  6. **
  7. ** This file may be used under the terms of the GNU General Public
  8. ** License version 2.0 as published by the Free Software Foundation
  9. ** and appearing in the file LICENSE.GPL included in the packaging of
  10. ** this file.  Please review the following information to ensure GNU
  11. ** General Public Licensing requirements will be met:
  12. ** http://trolltech.com/products/qt/licenses/licensing/opensource/
  13. **
  14. ** If you are unsure which license is appropriate for your use, please
  15. ** review the following information:
  16. ** http://trolltech.com/products/qt/licenses/licensing/licensingoverview
  17. ** or contact the sales department at sales@trolltech.com.
  18. **
  19. ** In addition, as a special exception, Trolltech gives you certain
  20. ** additional rights. These rights are described in the Trolltech GPL
  21. ** Exception version 1.0, which can be found at
  22. ** http://www.trolltech.com/products/qt/gplexception/ and in the file
  23. ** GPL_EXCEPTION.txt in this package.
  24. **
  25. ** In addition, as a special exception, Trolltech, as the sole copyright
  26. ** holder for Qt Designer, grants users of the Qt/Eclipse Integration
  27. ** plug-in the right for the Qt/Eclipse Integration to link to
  28. ** functionality provided by Qt Designer and its related libraries.
  29. **
  30. ** Trolltech reserves all rights not expressly granted herein.
  31. **
  32. ** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
  33. ** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
  34. **
  35. ****************************************************************************/
  36. #include <math.h>
  37. #include <qevent.h>
  38. #include <QPainter>
  39. #include <QTextStream>
  40. #include <QUndoStack>
  41. #include "document.h"
  42. #include "commands.h"
  43. static const int resizeHandleWidth = 6;
  44. /******************************************************************************
  45. ** Shape
  46. */
  47. const QSize Shape::minSize(80, 50);
  48. Shape::Shape(Type type, const QColor &color, const QRect &rect)
  49.     : m_type(type), m_rect(rect), m_color(color)
  50. {
  51. }
  52. Shape::Type Shape::type() const
  53. {
  54.     return m_type;
  55. }
  56. QRect Shape::rect() const
  57. {
  58.     return m_rect;
  59. }
  60. QColor Shape::color() const
  61. {
  62.     return m_color;
  63. }
  64. QString Shape::name() const
  65. {
  66.     return m_name;
  67. }
  68. QRect Shape::resizeHandle() const
  69. {
  70.     QPoint br = m_rect.bottomRight();
  71.     return QRect(br - QPoint(resizeHandleWidth, resizeHandleWidth), br);
  72. }
  73. QString Shape::typeToString(Type type)
  74. {
  75.     QString result;
  76.     switch (type) {
  77.         case Rectangle:
  78.             result = QLatin1String("Rectangle");
  79.             break;
  80.         case Circle:
  81.             result = QLatin1String("Circle");
  82.             break;
  83.         case Triangle:
  84.             result = QLatin1String("Triangle");
  85.             break;
  86.     }
  87.     return result;
  88. }
  89. Shape::Type Shape::stringToType(const QString &s, bool *ok)
  90. {
  91.     if (ok != 0)
  92.         *ok = true;
  93.     if (s == QLatin1String("Rectangle"))
  94.         return Rectangle;
  95.     if (s == QLatin1String("Circle"))
  96.         return Circle;
  97.     if (s == QLatin1String("Triangle"))
  98.         return Triangle;
  99.     if (ok != 0)
  100.         *ok = false;
  101.     return Rectangle;
  102. }
  103. /******************************************************************************
  104. ** Document
  105. */
  106. Document::Document(QWidget *parent)
  107.     : QWidget(parent), m_currentIndex(-1), m_mousePressIndex(-1), m_resizeHandlePressed(false)
  108. {
  109.     m_undoStack = new QUndoStack(this);
  110.     setAutoFillBackground(true);
  111.     setBackgroundRole(QPalette::Base);
  112.     QPalette pal = palette();
  113.     pal.setBrush(QPalette::Base, QPixmap(":/icons/background.png"));
  114.     pal.setColor(QPalette::HighlightedText, Qt::red);
  115.     setPalette(pal);
  116. }
  117. QString Document::addShape(const Shape &shape)
  118. {
  119.     QString name = Shape::typeToString(shape.type());
  120.     name = uniqueName(name);
  121.     m_shapeList.append(shape);
  122.     m_shapeList[m_shapeList.count() - 1].m_name = name;
  123.     setCurrentShape(m_shapeList.count() - 1);
  124.     return name;
  125. }
  126. void Document::deleteShape(const QString &shapeName)
  127. {
  128.     int index = indexOf(shapeName);
  129.     if (index == -1)
  130.         return;
  131.     update(m_shapeList.at(index).rect());
  132.     m_shapeList.removeAt(index);
  133.     if (index <= m_currentIndex) {
  134.         m_currentIndex = -1;
  135.         if (index == m_shapeList.count())
  136.             --index;
  137.         setCurrentShape(index);
  138.     }
  139. }
  140. Shape Document::shape(const QString &shapeName) const
  141. {
  142.     int index = indexOf(shapeName);
  143.     if (index == -1)
  144.         return Shape();
  145.     return m_shapeList.at(index);
  146. }
  147. void Document::setShapeRect(const QString &shapeName, const QRect &rect)
  148. {
  149.     int index = indexOf(shapeName);
  150.     if (index == -1)
  151.         return;
  152.     Shape &shape = m_shapeList[index];
  153.     update(shape.rect());
  154.     update(rect);
  155.     shape.m_rect = rect;
  156. }
  157. void Document::setShapeColor(const QString &shapeName, const QColor &color)
  158. {
  159.     int index = indexOf(shapeName);
  160.     if (index == -1)
  161.         return;
  162.     Shape &shape = m_shapeList[index];
  163.     shape.m_color = color;
  164.     update(shape.rect());
  165. }
  166. QUndoStack *Document::undoStack() const
  167. {
  168.     return m_undoStack;
  169. }
  170. bool Document::load(QTextStream &stream)
  171. {
  172.     m_shapeList.clear();
  173.     while (!stream.atEnd()) {
  174.         QString shapeType, shapeName, colorName;
  175.         int left, top, width, height;
  176.         stream >> shapeType >> shapeName >> colorName >> left >> top >> width >> height;
  177.         if (stream.status() != QTextStream::Ok)
  178.             return false;
  179.         bool ok;
  180.         Shape::Type type = Shape::stringToType(shapeType, &ok);
  181.         if (!ok)
  182.             return false;
  183.         QColor color(colorName);
  184.         if (!color.isValid())
  185.             return false;
  186.         Shape shape(type);
  187.         shape.m_name = shapeName;
  188.         shape.m_color = color;
  189.         shape.m_rect = QRect(left, top, width, height);
  190.         m_shapeList.append(shape);
  191.     }
  192.     m_currentIndex = m_shapeList.isEmpty() ? -1 : 0;
  193.     return true;
  194. }
  195. void Document::save(QTextStream &stream)
  196. {
  197.     for (int i = 0; i < m_shapeList.count(); ++i) {
  198.         const Shape &shape = m_shapeList.at(i);
  199.         QRect r = shape.rect();
  200.         stream << Shape::typeToString(shape.type()) << QLatin1Char(' ')
  201.                 << shape.name() << QLatin1Char(' ')
  202.                 << shape.color().name() << QLatin1Char(' ')
  203.                 << r.left() << QLatin1Char(' ')
  204.                 << r.top() << QLatin1Char(' ')
  205.                 << r.width() << QLatin1Char(' ')
  206.                 << r.height();
  207.         if (i != m_shapeList.count() - 1)
  208.             stream << QLatin1Char('n');
  209.     }
  210.     m_undoStack->setClean();
  211. }
  212. QString Document::fileName() const
  213. {
  214.     return m_fileName;
  215. }
  216. void Document::setFileName(const QString &fileName)
  217. {
  218.     m_fileName = fileName;
  219. }
  220. int Document::indexAt(const QPoint &pos) const
  221. {
  222.     for (int i = m_shapeList.count() - 1; i >= 0; --i) {
  223.         if (m_shapeList.at(i).rect().contains(pos))
  224.             return i;
  225.     }
  226.     return -1;
  227. }
  228. void Document::mousePressEvent(QMouseEvent *event)
  229. {
  230.     event->accept();
  231.     int index = indexAt(event->pos());;
  232.     if (index != -1) {
  233.         setCurrentShape(index);
  234.         const Shape &shape = m_shapeList.at(index);
  235.         m_resizeHandlePressed = shape.resizeHandle().contains(event->pos());
  236.         if (m_resizeHandlePressed)
  237.             m_mousePressOffset = shape.rect().bottomRight() - event->pos();
  238.         else
  239.             m_mousePressOffset = event->pos() - shape.rect().topLeft();
  240.     }
  241.     m_mousePressIndex = index;
  242. }
  243. void Document::mouseReleaseEvent(QMouseEvent *event)
  244. {
  245.     event->accept();
  246.     m_mousePressIndex = -1;
  247. }
  248. void Document::mouseMoveEvent(QMouseEvent *event)
  249. {
  250.     event->accept();
  251.     if (m_mousePressIndex == -1)
  252.         return;
  253.     const Shape &shape = m_shapeList.at(m_mousePressIndex);
  254.     QRect rect;
  255.     if (m_resizeHandlePressed) {
  256.         rect = QRect(shape.rect().topLeft(), event->pos() + m_mousePressOffset);
  257.     } else {
  258.         rect = shape.rect();
  259.         rect.moveTopLeft(event->pos() - m_mousePressOffset);
  260.     }
  261.     QSize size = rect.size().expandedTo(Shape::minSize);
  262.     rect.setSize(size);
  263.     m_undoStack->push(new SetShapeRectCommand(this, shape.name(), rect));
  264. }
  265. static QGradient gradient(const QColor &color, const QRect &rect)
  266. {
  267.     QColor c = color;
  268.     c.setAlpha(160);
  269.     QLinearGradient result(rect.topLeft(), rect.bottomRight());
  270.     result.setColorAt(0, c.dark(150));
  271.     result.setColorAt(0.5, c.light(200));
  272.     result.setColorAt(1, c.dark(150));
  273.     return result;
  274. }
  275. static QPolygon triangle(const QRect &rect)
  276. {
  277.     QPolygon result(3);
  278.     result.setPoint(0, rect.center().x(), rect.top());
  279.     result.setPoint(1, rect.right(), rect.bottom());
  280.     result.setPoint(2, rect.left(), rect.bottom());
  281.     return result;
  282. }
  283. void Document::paintEvent(QPaintEvent *event)
  284. {
  285.     QRegion paintRegion = event->region();
  286.     QPainter painter(this);
  287.     QPalette pal = palette();
  288.     for (int i = 0; i < m_shapeList.count(); ++i) {
  289.         const Shape &shape = m_shapeList.at(i);
  290.         if (!paintRegion.contains(shape.rect()))
  291.             continue;
  292.         QPen pen = pal.text().color();
  293.         pen.setWidth(i == m_currentIndex ? 2 : 1);
  294.         painter.setPen(pen);
  295.         painter.setBrush(gradient(shape.color(), shape.rect()));
  296.         QRect rect = shape.rect();
  297.         rect.adjust(1, 1, -resizeHandleWidth/2, -resizeHandleWidth/2);
  298.         // paint the shape
  299.         switch (shape.type()) {
  300.             case Shape::Rectangle:
  301.                 painter.drawRect(rect);
  302.                 break;
  303.             case Shape::Circle:
  304.                 painter.setRenderHint(QPainter::Antialiasing);
  305.                 painter.drawEllipse(rect);
  306.                 painter.setRenderHint(QPainter::Antialiasing, false);
  307.                 break;
  308.             case Shape::Triangle:
  309.                 painter.setRenderHint(QPainter::Antialiasing);
  310.                 painter.drawPolygon(triangle(rect));
  311.                 painter.setRenderHint(QPainter::Antialiasing, false);
  312.                 break;
  313.         }
  314.         // paint the resize handle
  315.         painter.setPen(pal.text().color());
  316.         painter.setBrush(Qt::white);
  317.         painter.drawRect(shape.resizeHandle().adjusted(0, 0, -1, -1));
  318.         // paint the shape name
  319.         painter.setBrush(pal.text());
  320.         if (shape.type() == Shape::Triangle)
  321.             rect.adjust(0, rect.height()/2, 0, 0);
  322.         painter.drawText(rect, Qt::AlignCenter, shape.name());
  323.     }
  324. }
  325. void Document::setCurrentShape(int index)
  326. {
  327.     QString currentName;
  328.     if (m_currentIndex != -1)
  329.         update(m_shapeList.at(m_currentIndex).rect());
  330.     m_currentIndex = index;
  331.     if (m_currentIndex != -1) {
  332.         const Shape &current = m_shapeList.at(m_currentIndex);
  333.         update(current.rect());
  334.         currentName = current.name();
  335.     }
  336.     emit currentShapeChanged(currentName);
  337. }
  338. int Document::indexOf(const QString &shapeName) const
  339. {
  340.     for (int i = 0; i < m_shapeList.count(); ++i) {
  341.         if (m_shapeList.at(i).name() == shapeName)
  342.             return i;
  343.     }
  344.     return -1;
  345. }
  346. QString Document::uniqueName(const QString &name) const
  347. {
  348.     QString unique;
  349.     for (int i = 0; ; ++i) {
  350.         unique = name;
  351.         if (i > 0)
  352.             unique += QString::number(i);
  353.         if (indexOf(unique) == -1)
  354.             break;
  355.     }
  356.     return unique;
  357. }
  358. QString Document::currentShapeName() const
  359. {
  360.     if (m_currentIndex == -1)
  361.         return QString();
  362.     return m_shapeList.at(m_currentIndex).name();
  363. }