Copy the KDE 3.5 branch to branches/trinity for new KDE 3.5 features.

BUG:215923


git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/kdesdk@1054174 283d02a7-25f6-0310-bc7c-ecb5cbfe19da

1 files changed, 218 insertions, 0 deletions

diff --git a/umbrello/umbrello/enumwidget.cpp b/umbrello/umbrello/enumwidget.cpp
new file mode 100644
index 00000000..2d642b31
--- /dev/null
+++ b/umbrello/umbrello/enumwidget.cpp
@@ -0,0 +1,218 @@
+/***************************************************************************
+ *                                                                         *
+ *   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.                                   *
+ *                                                                         *
+ *   copyright (C) 2003-2007                                               *
+ *   Umbrello UML Modeller Authors <uml-devel@uml.sf.net>                  *
+ ***************************************************************************/
+
+// own header
+#include "enumwidget.h"
+
+// qt/kde includes
+#include <qpainter.h>
+#include <kdebug.h>
+
+// app includes
+#include "enum.h"
+#include "enumliteral.h"
+#include "classifier.h"
+#include "umlclassifierlistitemlist.h"
+#include "classifierlistitem.h"
+#include "umlview.h"
+#include "umldoc.h"
+#include "uml.h"
+#include "listpopupmenu.h"
+#include "object_factory.h"
+
+
+EnumWidget::EnumWidget(UMLView* view, UMLObject* o) : UMLWidget(view, o) {
+    init();
+}
+
+void EnumWidget::init() {
+    UMLWidget::setBaseType(Uml::wt_Enum);
+    setSize(100, 30);
+    m_pMenu = 0;
+    //set defaults from m_pView
+    if (m_pView) {
+        //check to see if correct
+        const Settings::OptionState& ops = m_pView->getOptionState();
+        m_bShowPackage = ops.classState.showPackage;
+    } else {
+        // For completeness only. Not supposed to happen.
+        m_bShowPackage = false;
+    }
+    if (! UMLApp::app()->getDocument()->loading())
+        updateComponentSize();
+}
+
+EnumWidget::~EnumWidget() {}
+
+void EnumWidget::draw(QPainter& p, int offsetX, int offsetY) {
+    UMLWidget::setPen(p);
+    if(UMLWidget::getUseFillColour())
+        p.setBrush(UMLWidget::getFillColour());
+    else
+        p.setBrush(m_pView -> viewport() -> backgroundColor());
+
+    const int w = width();
+    const int h = height();
+
+    const QFontMetrics &fm = getFontMetrics(FT_NORMAL);
+    const int fontHeight  = fm.lineSpacing();
+    QString name;
+    if ( m_bShowPackage ) {
+        name = m_pObject->getFullyQualifiedName();
+    } else {
+        name = this -> getName();
+    }
+
+    p.drawRect(offsetX, offsetY, w, h);
+    p.setPen(QPen(Qt::black));
+
+    QFont font = UMLWidget::getFont();
+    font.setBold(true);
+    p.setFont(font);
+    p.drawText(offsetX + ENUM_MARGIN, offsetY,
+               w - ENUM_MARGIN * 2,fontHeight,
+               Qt::AlignCenter, m_pObject->getStereotype(true));
+
+    font.setItalic( m_pObject -> getAbstract() );
+    p.setFont(font);
+    p.drawText(offsetX + ENUM_MARGIN, offsetY + fontHeight,
+               w - ENUM_MARGIN * 2, fontHeight, Qt::AlignCenter, name);
+    font.setBold(false);
+    font.setItalic(false);
+    p.setFont(font);
+
+    int y = fontHeight * 2;
+
+    UMLWidget::setPen(p);
+
+    p.drawLine(offsetX, offsetY + y, offsetX + w - 1, offsetY + y);
+
+    QFontMetrics fontMetrics(font);
+    UMLClassifier *classifier = (UMLClassifier*)m_pObject;
+    UMLClassifierListItem* enumLiteral = 0;
+    UMLClassifierListItemList list = classifier->getFilteredList(Uml::ot_EnumLiteral);
+    for (enumLiteral = list.first(); enumLiteral; enumLiteral = list.next()) {
+        QString text = enumLiteral->getName();
+        p.setPen( QPen(Qt::black) );
+        p.drawText(offsetX + ENUM_MARGIN, offsetY + y,
+                   fontMetrics.width(text), fontHeight, Qt::AlignVCenter, text);
+        y+=fontHeight;
+    }
+
+    if (m_bSelected) {
+        drawSelected(&p, offsetX, offsetY);
+    }
+}
+
+QSize EnumWidget::calculateSize() {
+    if (!m_pObject) {
+        return UMLWidget::calculateSize();
+    }
+
+    int width, height;
+    QFont font = UMLWidget::getFont();
+    font.setItalic(false);
+    font.setUnderline(false);
+    font.setBold(false);
+    const QFontMetrics fm(font);
+
+    const int fontHeight = fm.lineSpacing();
+
+    int lines = 1;//always have one line - for name
+    lines++; //for the stereotype
+
+    const int numberOfEnumLiterals = ((UMLEnum*)m_pObject)->enumLiterals();
+
+    height = width = 0;
+    //set the height of the enum
+
+    lines += numberOfEnumLiterals;
+    if (numberOfEnumLiterals == 0) {
+        height += fontHeight / 2; //no enum literals, so just add a bit of space
+    }
+
+    height += lines * fontHeight;
+
+    //now set the width of the concept
+    //set width to name to start with
+    if (m_bShowPackage)  {
+        width = getFontMetrics(FT_BOLD_ITALIC).boundingRect(m_pObject->getFullyQualifiedName()).width();
+    } else {
+        width = getFontMetrics(FT_BOLD_ITALIC).boundingRect(getName()).width();
+    }
+    int w = getFontMetrics(FT_BOLD).boundingRect(m_pObject->getStereotype(true)).width();
+
+
+    width = w > width?w:width;
+
+    UMLClassifier *classifier = (UMLClassifier*)m_pObject;
+    UMLClassifierListItemList list = classifier->getFilteredList(Uml::ot_EnumLiteral);
+    UMLClassifierListItem* listItem = 0;
+    for (listItem = list.first(); listItem; listItem = list.next()) {
+        int w = fm.width( listItem->getName() );
+        width = w > width?w:width;
+    }
+
+    //allow for width margin
+    width += ENUM_MARGIN * 2;
+
+    return QSize(width, height);
+}
+
+void EnumWidget::slotMenuSelection(int sel) {
+    switch(sel) {
+    case ListPopupMenu::mt_EnumLiteral:
+        if (Object_Factory::createChildObject(static_cast<UMLClassifier*>(m_pObject),
+                                              Uml::ot_EnumLiteral) )  {
+            UMLApp::app()->getDocument()->setModified();
+        }
+        break;
+    }
+    UMLWidget::slotMenuSelection(sel);
+}
+
+void EnumWidget::setShowPackage(bool _status) {
+    m_bShowPackage = _status;
+    updateComponentSize();
+    update();
+}
+
+bool EnumWidget::getShowPackage() const {
+    return m_bShowPackage;
+}
+
+void EnumWidget::saveToXMI( QDomDocument& qDoc, QDomElement& qElement ) {
+    QDomElement conceptElement = qDoc.createElement("enumwidget");
+    UMLWidget::saveToXMI(qDoc, conceptElement);
+
+    conceptElement.setAttribute("showpackage", m_bShowPackage);
+    qElement.appendChild(conceptElement);
+}
+
+bool EnumWidget::loadFromXMI( QDomElement & qElement ) {
+    if ( !UMLWidget::loadFromXMI(qElement) ) {
+        return false;
+    }
+    QString showpackage = qElement.attribute("showpackage", "0");
+
+    m_bShowPackage = (bool)showpackage.toInt();
+
+    return true;
+}
+
+void EnumWidget::toggleShowPackage() {
+    m_bShowPackage = !m_bShowPackage;
+    updateComponentSize();
+    update();
+
+    return;
+}
+