C++ ProjectTreeItem类代码示例

void ProjectModel::addProject(Project *project)
{
    //Add project and its name.
    ProjectTreeItem *childProject = new ProjectTreeItem(project, project->projectName(), rootItem);

    //Add each file in the project with its name
    for(int i = 0; i < project->filesCount(); i++)
    {
        StringAndAnnotations saa = project->file(i);
        ProjectTreeItem *childFile = new ProjectTreeItem(project, saa.first, childProject);

        //For each file add the annotations (one per row in the model)
        for(int j = 0; j < saa.second.size(); j++)
        {
            QVariant var;
            var.setValue(saa.second.at(j));

            ProjectTreeItem * childAnnotation = new ProjectTreeItem(project, var, childFile);
            childFile->appendChild(childAnnotation);
        }

        childProject->appendChild(childFile);
    }

    rootItem->appendChild(childProject);
}

QVariant ProjectModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid())
        return QVariant();

    if (role != Qt::DisplayRole && role != Qt::UserRole)
        return QVariant();

    ProjectTreeItem *item = static_cast<ProjectTreeItem*>(index.internalPointer());

    return item->data(index.column(), role);
}

  void BondDelegate::primitiveUpdated(Primitive *primitive)
  {
    if (primitive->type() == Primitive::MoleculeType) {
      initialize();
      return;
    }

    if (primitive->type() != Primitive::BondType)
      return;
    
    ProjectTreeItem *item = m_label->child(primitive->index()); 
    item->setData(0, tr("bond %1").arg(primitive->index()));
  }

/**
 * @brief ProjectManagerTreeModel::parent Get parent of item with index.
 * @param child Child where we want get parent item.
 * @return Parent of given child.
 */
QModelIndex ProjectManagerTreeModel::parent(const QModelIndex &child) const
{
    if(!child.isValid())
        return QModelIndex();

    ProjectTreeItem* cItem = static_cast<ProjectTreeItem*>(child.internalPointer());
    ProjectTreeItem* pItem = cItem->getParent();

    if(pItem == root)
        return QModelIndex();

    return createIndex(pItem->getRow(), 0, pItem);
}

int ProjectModel::rowCount(const QModelIndex &parent) const
{
    ProjectTreeItem *parentItem;
    if (parent.column() > 0)
        return 0;

    if (!parent.isValid())
        parentItem = rootItem;
    else
        parentItem = static_cast<ProjectTreeItem*>(parent.internalPointer());

    return parentItem->childCount();
}

QModelIndex ProjectModel::parent(const QModelIndex &index) const
{
    if (!index.isValid())
        return QModelIndex();

    ProjectTreeItem *childItem = static_cast<ProjectTreeItem*>(index.internalPointer());
    ProjectTreeItem *parentItem = childItem->parent();

    if (parentItem == rootItem)
        return QModelIndex();

    return createIndex(parentItem->row(), 0, parentItem);
}

  ProjectTreeItem* ProjectTreeModelDelegate::insertExpandableItem(ProjectTreeItem *parent)
  {
    // insert at the end
    int position = parent->childCount();
    d->model->insertRows(parent, position, 1);
    // retrieve the item
    ProjectTreeItem *item = parent->child(position);
    // set it to non-terminal
    item->setTerminal(false);
    // store it (for hasExpandableItem & fetchMore)
    d->expandableItems.append(item);

    return item;
  }

bool ProjectModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
    if(!index.isValid())
        return false;

    if(role == Qt::EditRole)
    {
        ProjectTreeItem *item = static_cast<ProjectTreeItem*>(index.internalPointer());
        item->setData(value);
        return true;
    }

    return false;
}

void ProjectTree::loadItem(QTreeWidgetItem *item)
{
    ProjectTreeItem *clickedItem = static_cast<ProjectTreeItem*>(item);
    if (!clickedItem->isLoaded()) {
        QDir* rootDir = new QDir(clickedItem->getFileInfo()->absoluteFilePath());

        clickedItem->setLoaded(true);

        for (QFileInfo fileInfo : rootDir->entryInfoList(QDir::Filter::AllEntries | QDir::Filter::NoDotAndDotDot, QDir::SortFlag::DirsFirst)) {
            ProjectTreeItem* child = new ProjectTreeItem(fileInfo);
            clickedItem->addChild(child);
        }
        resizeColumnToContents(0);
    }
}

void ProjectTree::on_itemClicked(QTreeWidgetItem *item, int column)
{
    ProjectTreeItem *clickedItem = static_cast<ProjectTreeItem*>(item);

    if (clickedItem->getFileInfo()->isDir()) {
        if (clickedItem->isExpanded())
            clickedItem->setExpanded(false);
        else
            clickedItem->setExpanded(true);
    }
    else
        emit fileClicked(clickedItem);

    (void)column;
}

 void BondDelegate::primitiveAdded(Primitive *primitive)
 {
   if (primitive->type() != Primitive::BondType) 
     return;
     
   // add the new primitive to the end
   model()->insertRows(m_label, m_label->childCount(), 1); 
   // get the item
   ProjectTreeItem *item = m_label->child(m_label->childCount() - 1);
   item->setData(0, tr("bond %1").arg(primitive->index()));
   // set the primitive
   PrimitiveList primitives;
   primitives.append(primitive);
   item->setPrimitives(primitives);
 }

/**
 * @brief ProjectManagerTreeModel::projectRemoved Call this method if project with given name was removed.
 * @param projName Name of the old project.
 */
void ProjectManagerTreeModel::projectRemoved(QString projName)
{
    int row = root->getRowCount();

    for(int i = 0; i < row; ++i)
    {
        ProjectTreeItem* item = root->getChild(i);
        if(item->getDataString() == projName)
        {
            insertData = projName;
            insertType = ProjectTreeItem::PROJECT;
            removeRows(i, 1);
            return;
        }
    }
}

  void SelectionDelegate::initialize()
  {
    // remove any existing rows
    if (m_label->childCount())
      model()->removeRows(m_label, 0, m_label->childCount());

    QList<QString> names = m_widget->namedSelections();
    // add the selectionss...
    model()->insertRows(m_label, 0, names.size());
    for (int i = 0; i < m_label->childCount(); ++i) {
      ProjectTreeItem *item = m_label->child(i);
      item->setData(0,   names.at(i));
      item->setData(1, QString("%1").arg(i));
      // set the primitive
      PrimitiveList primitives = m_widget->namedSelectionPrimitives(i);
      item->setPrimitives(primitives);
    }

  }

/**
 * @brief ProjectManagerTreeModel::projectActiveChanged Call this method if active project was changed.
 * @param newProject New active project.
 * @param oldProject Old active project.
 */
void ProjectManagerTreeModel::projectActiveChanged(QString newProject, QString oldProject)
{    
    // new project is set

    int rowNumber = root->getRowCount();

    ProjectTreeItem* oldP = NULL;
    ProjectTreeItem* newP = NULL;

    int oldRow = 0;
    int newRow = 0;

    for(int i = 0; i < rowNumber; ++i)
    {
        ProjectTreeItem* item = root->getChild(i);

        if(item->getDataString() == newProject)
        {
            newP = item;
            newRow = i;
        }
        else if(item->getDataString() == oldProject)
        {
            oldP = item;
            oldRow = i;
        }
    }

    if(oldP != NULL)
    {
        QModelIndex index = createIndex(oldRow, 0, oldP);

        dataChanged(index,index);
    }

    if(newP != NULL)
    {
        QModelIndex indexNew = createIndex(newRow, 0, newP);

        dataChanged(indexNew,indexNew);
    }
}

void ProjectModel::rollback(QModelIndex &index, int n)
{
    ProjectTreeItem *item = static_cast<ProjectTreeItem*>(index.internalPointer());
    QVariant var = item->data(0, Qt::UserRole);

    CompleteAnnotationItem *currentAnnotation = var.value<CompleteAnnotationItem*>();
    Revision<AnnotationItem*> revision = currentAnnotation->getRevision(n);

    RevisionAnnotationList revisions;

    for(int i = 0; i < n; i++){
        revisions << currentAnnotation->getRevision(i);
    }
    CompleteAnnotationItem *newAnnotation =
            new CompleteAnnotationItem(revision.getValue()->getId(),
                                       revision.getValue(), revisions);

    var.setValue<CompleteAnnotationItem*>(newAnnotation);
    item->setData(var);
}