codeparameter.cpp

Go to the documentation of this file.

/***************************************************************************
 *   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      Brian Thomas <[email protected]>   *
 *   copyright (C) 2004-2013                                               *
 *   Umbrello UML Modeller Authors <[email protected]>                *
 ***************************************************************************/

// own header
#include "codeparameter.h"

// local includes
#include "association.h"
#include "attribute.h"
#include "classifiercodedocument.h"
#include "debug_utils.h"
#include "umldoc.h"
#include "umlobject.h"
#include "umlrole.h"
#include "uml.h"
#include "codegenfactory.h"

CodeParameter::CodeParameter(ClassifierCodeDocument * parentDoc, UMLObject * parentObject)
        : QObject(parentObject)
{
    setObjectName("ACodeParam");
    initFields(parentDoc, parentObject);
}

CodeParameter::~CodeParameter()
{
}

bool CodeParameter::getAbstract()
{
    return m_parentObject->isAbstract();
}

bool CodeParameter::getStatic()
{
    return m_parentObject->isStatic();
}

QString CodeParameter::getName() const
{
    return m_parentObject->name();
}

QString CodeParameter::getTypeName()
{
    UMLAttribute * at = (UMLAttribute*) m_parentObject;
    return at->getTypeName();
}

Uml::Visibility::Enum CodeParameter::getVisibility() const
{
    return m_parentObject->visibility();
}

void CodeParameter::setInitialValue(const QString &new_var)
{
    m_initialValue = new_var;
}

QString CodeParameter::getInitialValue()
{
    return m_initialValue;
}

void CodeParameter::setComment(CodeComment * object)
{
    m_comment = object;
}

CodeComment * CodeParameter::getComment()
{
    return m_comment;
}

ClassifierCodeDocument * CodeParameter::getParentDocument()
{
    return m_parentDocument;
}

UMLObject * CodeParameter::getParentObject()
{
    return m_parentObject;
}

// need to get the ID of the parent object
// this is kind of broken for UMLRoles.
QString CodeParameter::ID()
{
    UMLRole * role = dynamic_cast<UMLRole*>(m_parentObject);
    if(role)
    {
        // cant use Role "ID" as that is used to distinquish if its
        // role "A" or "B"
        UMLAssociation *assoc = role->parentAssociation();
        return Uml::ID::toString(assoc->id());
    } else
        return Uml::ID::toString(m_parentObject->id());

}

void CodeParameter::setAttributesOnNode(QDomDocument & doc, QDomElement & blockElement)
{
    // set local attributes
    blockElement.setAttribute("parent_id", ID());

    // setting ID's takes special treatment
    // as UMLRoles arent properly stored in the XMI right now.
    // (change would break the XMI format..save for big version change)
    UMLRole * role = dynamic_cast<UMLRole*>(m_parentObject);
    if(role)
        blockElement.setAttribute("role_id", role->role());
    else
        blockElement.setAttribute("role_id","-1");

    blockElement.setAttribute("initialValue", getInitialValue());

    // a comment which we will store in its own separate child node block
    QDomElement commElement = doc.createElement("header");
    getComment()->saveToXMI(doc, commElement); // comment
    blockElement.appendChild(commElement);
}

void CodeParameter::setAttributesFromNode(QDomElement & root)
{
    // set local attributes, parent object first
    QString idStr = root.attribute("parent_id","-1");
    Uml::ID::Type id = Uml::ID::fromString(idStr);

    // always disconnect
    m_parentObject->disconnect(this);

    // now, what is the new object we want to set?
    UMLObject * obj = UMLApp::app()->document()->findObjectById(id);
    if(obj)
    {
        // FIX..one day.
        // Ugh. This is UGLY, but we have to do it this way because UMLRoles
        // don't go into the document list of UMLobjects, and have the same
        // ID as their parent UMLAssociations. So..the drill is then special
        // for Associations..in that case we need to find out which role will
        // serve as the parameter here. The REAL fix, of course, would be to
        // treat UMLRoles on a more even footing, but im not sure how that change
        // might ripple throughout the code and cause problems. Thus, since the
        // change appears to be needed for only this part, I'll do this crappy
        // change instead. -b.t.
        UMLAssociation * assoc = dynamic_cast<UMLAssociation*>(obj);
        if(assoc) {
            // In this case we init with indicated role child obj.
            UMLRole * role = 0;
            int role_id = root.attribute("role_id","-1").toInt();
            if(role_id == 1)
                role = assoc->getUMLRole(Uml::RoleType::A);
            else if(role_id == 0)
                role = assoc->getUMLRole(Uml::RoleType::B);
            else
                uError() << "corrupt save file? "
                    << "cant get proper UMLRole for codeparameter uml id:"
                    << Uml::ID::toString(id) << " w/role_id:" << role_id;

            // init using UMLRole obj
            initFields (m_parentDocument, role);

        } else
            initFields (m_parentDocument, obj); // just the regular approach

    } else
        uError() << "Cant load CodeParam: parentUMLObject w/id:"
            << Uml::ID::toString(id) << " not found, corrupt save file?";

    // other attribs now
    setInitialValue(root.attribute("initialValue",""));

    // load comment now
    // by looking for our particular child element
    QDomNode node = root.firstChild();
    QDomElement element = node.toElement();
    bool gotComment = false;
    while(!element.isNull()) {
        QString tag = element.tagName();
        if(tag == "header") {
            QDomNode cnode = element.firstChild();
            QDomElement celem = cnode.toElement();
            getComment()->loadFromXMI(celem);
            gotComment = true;
            break;
        }
        node = element.nextSibling();
        element = node.toElement();
    }

    if(!gotComment)
        uWarning()<<" loadFromXMI : Warning: unable to initialize CodeComment in codeparam:"<<this;
}

void CodeParameter::syncToParent()
{
    getComment()->setText(getParentObject()->doc());

    updateContent();
}

void CodeParameter::initFields(ClassifierCodeDocument * doc, UMLObject * obj)
{
    m_parentObject = obj;

    m_parentDocument = doc;
    m_initialValue.clear();

    m_comment = CodeGenFactory::newCodeComment(m_parentDocument);
    m_comment->setText(getParentObject()->doc());

    connect(m_parentObject, SIGNAL(modified()), this, SLOT(syncToParent()));
}

#include "codeparameter.moc"