/****************************************************************************

**

** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the qt3to4 porting application of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include "codemodelattributes.h"

#include "tokenengine.h"

QT_BEGIN_NAMESPACE

using namespace CodeModel;

using namespace TokenEngine;

/*

     Walk the codemodel.

*/

void CodeModelAttributes::createAttributes(TranslationUnit translationUnit)

{

    m_translationUnit = translationUnit;

    parseScope(const_cast<CodeModel::NamespaceScope *>(translationUnit.codeModel()));

}

/*

    Create attributes for each name use and assign to the token.

*/

void CodeModelAttributes::parseNameUse(CodeModel::NameUse *nameUse)

{

    // Get the container for this token.

    TokenRef ref = nameUse->nameToken();

    const int containerIndex = ref.containerIndex();

    TokenAttributes *attributes = ref.tokenContainer().tokenAttributes();

    if (!areAttributesEnabled(attributes))

        return;

    // Test if the nameUse refers to a UnknownType. If so we add and

    // "unknown" attribute.

    if (TypeMember *typeMember =  nameUse->declaration()->toTypeMember()) {

        if (typeMember->type()->toUnknownType()) {

            attributes->addAttribute(containerIndex, "unknown", nameUse->name());

            return;

        }

    }

    // Add attributes this namnUse.

    attributes->addAttribute(containerIndex, "nameUse", nameUse->name());

    attributes->addAttribute(containerIndex, "parentScope",

                             nameUse->declaration()->parent()->name() );

    if (CodeModel::Scope * skop = nameUse->declaration()->parent()->parent()) {

        attributes->addAttribute(containerIndex, "grandParentScope", skop->name());

    }

    createNameTypeAttribute(nameUse);

}

/*

    Create attributes for members and assign to token.

*/

void CodeModelAttributes::parseMember(CodeModel::Member *member)

{

    if(!member || member->name() == QByteArray())

        return;

    //get the container for this token

    TokenRef ref = member->nameToken();

    const int containerIndex = ref.containerIndex();

    TokenAttributes *attributes = ref.tokenContainer().tokenAttributes();

    if (areAttributesEnabled(attributes)) {

        //add attributes for this declaration

        static const QByteArray textDeclaration = "declaration";

        attributes->addAttribute(containerIndex, textDeclaration, member->name());

        createNameTypeAttribute(member);

    }

    CodeModelWalker::parseMember(member);

}

void CodeModelAttributes::parseFunctionMember(CodeModel::FunctionMember *member)

{

    CodeModel::ArgumentCollection arguments = member->arguments();

    CodeModel::ArgumentCollection::ConstIterator it = arguments.constBegin();

    TokenRef ref = member->nameToken();

    TokenAttributes *attributes = ref.tokenContainer().tokenAttributes();

    if (areAttributesEnabled(attributes)) {

        while (it != arguments.constEnd()) {

            const int containerIndex = (*it)->nameToken().containerIndex();

            const QByteArray name = (*it)->name();

            attributes->addAttribute(containerIndex, "declaration", name);

            attributes->addAttribute(containerIndex, "nameType", "variable");

            ++it;

        }

    }

    CodeModelWalker::parseFunctionMember(member);

}

/*

    NameType attributes gives information on what kind of member this is.

*/

void CodeModelAttributes::createNameTypeAttribute(CodeModel::Member *member)

{

    if(!member)

        return;

    //get the container for the token accosiated with this member.

    TokenRef ref = member->nameToken();

    const int containerIndex = ref.containerIndex();

    TokenAttributes *attributes = ref.tokenContainer().tokenAttributes();

    createNameTypeAttributeAtIndex(attributes, containerIndex, member);

}

/*

    A NameUse has the same NameType as the declaration it is referring to.

*/

void CodeModelAttributes::createNameTypeAttribute(CodeModel::NameUse *nameUse)

{

    if(!nameUse)

        return;

    //get the container for the token accosiated with this NameUse.

    TokenRef ref = nameUse->nameToken();

    const int containerIndex = ref.containerIndex();

    TokenAttributes *attributes = ref.tokenContainer().tokenAttributes();

    createNameTypeAttributeAtIndex(attributes, containerIndex, nameUse->declaration());

}

void CodeModelAttributes::createNameTypeAttributeAtIndex(TokenEngine::TokenAttributes *attributes,

                                        int index, CodeModel::Member *member)

{

    QByteArray nameType = "unknown";

    if (member->toFunctionMember()) {

        nameType = "function";

    } else if (CodeModel::VariableMember *variableMember = member->toVariableMember()) {

        if (variableMember->type()->toEnumType())

            nameType = "enumerator";

        else

            nameType = "variable";

    } else if (CodeModel::TypeMember *typeMember =  member->toTypeMember()) {

        if (CodeModel::Type *type = typeMember->type()) {

            if (type->toClassType()) {

                nameType = "class";

            } else if (type->toEnumType()) {

                nameType = "enum";

            }

        }

    }

    attributes->addAttribute(index, "nameType", nameType);

}

bool CodeModelAttributes::areAttributesEnabled(const TokenAttributes *attributes) const

{

    static const QByteArray tstCreateAttributes("CreateAttributes");

    static const QByteArray tstTrue("True");

    return (attributes->attribute(tstCreateAttributes) == tstTrue);

}

QT_END_NAMESPACE