source: calamares/trunk/fuentes/src/modules/netinstall/PackageModel.cpp @ 7538

Last change on this file since 7538 was 7538, checked in by kbut, 13 months ago

sync with github

File size: 8.1 KB
Line 
1/* === This file is part of Calamares - <https://github.com/calamares> ===
2 *
3 *   Copyright (c) 2017, Kyle Robbertze <kyle@aims.ac.za>
4 *   Copyright 2017, Adriaan de Groot <groot@kde.org>
5 *
6 *   Calamares is free software: you can redistribute it and/or modify
7 *   it under the terms of the GNU General Public License as published by
8 *   the Free Software Foundation, either version 3 of the License, or
9 *   (at your option) any later version.
10 *
11 *   Calamares is distributed in the hope that it will be useful,
12 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
13 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 *   GNU General Public License for more details.
15 *
16 *   You should have received a copy of the GNU General Public License
17 *   along with Calamares. If not, see <http://www.gnu.org/licenses/>.
18 */
19
20#include "PackageModel.h"
21
22#include "utils/YamlUtils.h"
23
24PackageModel::PackageModel( const YAML::Node& data, QObject* parent ) :
25    QAbstractItemModel( parent ),
26    m_columnHeadings()
27{
28    m_rootItem = new PackageTreeItem();
29    setupModelData( data, m_rootItem );
30}
31
32PackageModel::~PackageModel()
33{
34    delete m_rootItem;
35}
36
37QModelIndex
38PackageModel::index( int row, int column, const QModelIndex& parent ) const
39{
40    if ( !hasIndex( row, column, parent ) )
41        return QModelIndex();
42
43    PackageTreeItem* parentItem;
44
45    if ( !parent.isValid() )
46        parentItem = m_rootItem;
47    else
48        parentItem = static_cast<PackageTreeItem*>( parent.internalPointer() );
49
50    PackageTreeItem* childItem = parentItem->child( row );
51    if ( childItem )
52        return createIndex( row, column, childItem );
53    else
54        return QModelIndex();
55}
56
57QModelIndex
58PackageModel::parent( const QModelIndex& index ) const
59{
60    if ( !index.isValid() )
61        return QModelIndex();
62
63    PackageTreeItem* child = static_cast<PackageTreeItem*>( index.internalPointer() );
64    PackageTreeItem* parent = child->parentItem();
65
66    if ( parent == m_rootItem )
67        return QModelIndex();
68    return createIndex( parent->row(), 0, parent );
69}
70
71int
72PackageModel::rowCount( const QModelIndex& parent ) const
73{
74    if ( parent.column() > 0 )
75        return 0;
76
77    PackageTreeItem* parentItem;
78    if ( !parent.isValid() )
79        parentItem = m_rootItem;
80    else
81        parentItem = static_cast<PackageTreeItem*>( parent.internalPointer() );
82
83    return parentItem->childCount();
84}
85
86int
87PackageModel::columnCount( const QModelIndex& parent ) const
88{
89    if ( parent.isValid() )
90        return static_cast<PackageTreeItem*>( parent.internalPointer() )->columnCount();
91    return m_rootItem->columnCount();
92}
93
94QVariant
95PackageModel::data( const QModelIndex& index, int role ) const
96{
97    if ( !index.isValid() )
98        return QVariant();
99
100    PackageTreeItem* item = static_cast<PackageTreeItem*>( index.internalPointer() );
101    if ( index.column() == 0 && role == Qt::CheckStateRole )
102        return item->isSelected();
103
104    if ( item->isHidden() && role == Qt::DisplayRole ) // Hidden group
105        return QVariant();
106
107    if ( role == Qt::DisplayRole )
108        return item->data( index.column() );
109    return QVariant();
110}
111
112bool
113PackageModel::setData( const QModelIndex& index, const QVariant& value, int role )
114{
115    if ( role == Qt::CheckStateRole && index.isValid() )
116    {
117        PackageTreeItem* item = static_cast<PackageTreeItem*>( index.internalPointer() );
118        item->setSelected( static_cast<Qt::CheckState>( value.toInt() ) );
119
120        emit dataChanged( this->index( 0, 0 ), index.sibling( index.column(), index.row() + 1 ),
121                          QVector<int>( Qt::CheckStateRole ) );
122    }
123    return true;
124}
125
126bool
127PackageModel::setHeaderData( int section, Qt::Orientation orientation,
128                             const QVariant& value, int role )
129{
130    Q_UNUSED( role );
131
132    if ( orientation == Qt::Horizontal )
133    {
134        if ( m_columnHeadings.value( section ) != QVariant() )
135            m_columnHeadings.replace( section, value );
136        else
137            m_columnHeadings.insert( section, value );
138        emit headerDataChanged( orientation, section, section );
139    }
140    return true;
141}
142
143Qt::ItemFlags
144PackageModel::flags( const QModelIndex& index ) const
145{
146    if ( !index.isValid() )
147        return Qt::ItemFlags();
148    if ( index.column() == 0 )
149        return Qt::ItemIsUserCheckable | QAbstractItemModel::flags( index );
150    return QAbstractItemModel::flags( index );
151}
152
153QVariant
154PackageModel::headerData( int section, Qt::Orientation orientation, int role ) const
155{
156    if ( orientation == Qt::Horizontal && role == Qt::DisplayRole )
157        return m_columnHeadings.value( section );
158    return QVariant();
159}
160
161QList<PackageTreeItem::ItemData>
162PackageModel::getPackages() const
163{
164    QList<PackageTreeItem*> items = getItemPackages( m_rootItem );
165    for ( auto package : m_hiddenItems )
166        if ( package->hiddenSelected() )
167            items.append( getItemPackages( package ) );
168    QList<PackageTreeItem::ItemData> packages;
169    for ( auto item : items )
170    {
171        PackageTreeItem::ItemData itemData;
172        itemData.preScript = item->parentItem()->preScript(); // Only groups have hooks
173        itemData.packageName = item->packageName(); // this seg faults
174        itemData.postScript = item->parentItem()->postScript(); // Only groups have hooks
175        itemData.isCritical = item->parentItem()->isCritical(); // Only groups are critical
176        packages.append( itemData );
177    }
178    return packages;
179}
180
181QList<PackageTreeItem*>
182PackageModel::getItemPackages( PackageTreeItem* item ) const
183{
184    QList<PackageTreeItem*> selectedPackages;
185    for ( int i = 0; i < item->childCount(); i++ )
186    {
187        if ( item->child( i )->isSelected() == Qt::Unchecked )
188            continue;
189
190        if ( !item->child( i )->childCount() ) // package
191            selectedPackages.append( item->child( i ) );
192        else
193            selectedPackages.append( getItemPackages( item->child( i ) ) );
194    }
195    return selectedPackages;
196
197}
198
199void
200PackageModel::setupModelData( const YAML::Node& data, PackageTreeItem* parent )
201{
202    for ( YAML::const_iterator it = data.begin(); it != data.end(); ++it )
203    {
204        const YAML::Node itemDefinition = *it;
205
206        QString name(
207            tr( CalamaresUtils::yamlToVariant( itemDefinition["name"] ).toByteArray() ) );
208        QString description(
209            tr( CalamaresUtils::yamlToVariant( itemDefinition["description"] ).toByteArray() ) );
210
211        PackageTreeItem::ItemData itemData;
212        itemData.name = name;
213        itemData.description = description;
214
215        if ( itemDefinition["pre-install"] )
216            itemData.preScript =
217                CalamaresUtils::yamlToVariant( itemDefinition["pre-install"] ).toString();
218        if ( itemDefinition["post-install"] )
219            itemData.postScript =
220                CalamaresUtils::yamlToVariant( itemDefinition["post-install"] ).toString();
221        PackageTreeItem* item = new PackageTreeItem( itemData, parent );
222
223        if ( itemDefinition["selected"] )
224            item->setSelected(
225                CalamaresUtils::yamlToVariant( itemDefinition["selected"] ).toBool() ?
226                Qt::Checked : Qt::Unchecked );
227        else
228            item->setSelected( parent->isSelected() ); // Inherit from it's parent
229
230        if ( itemDefinition["hidden"] )
231            item->setHidden(
232                CalamaresUtils::yamlToVariant( itemDefinition["hidden"] ).toBool() );
233
234        if ( itemDefinition["critical"] )
235            item->setCritical(
236                CalamaresUtils::yamlToVariant( itemDefinition["critical"] ).toBool() );
237
238        if ( itemDefinition["packages"] )
239            for ( YAML::const_iterator packageIt = itemDefinition["packages"].begin();
240                    packageIt != itemDefinition["packages"].end(); ++packageIt )
241                item->appendChild(
242                    new PackageTreeItem( CalamaresUtils::yamlToVariant( *packageIt ).toString(), item ) );
243
244        if ( itemDefinition["subgroups"] )
245            setupModelData( itemDefinition["subgroups"], item );
246
247        if ( item->isHidden() )
248            m_hiddenItems.append( item );
249        else
250        {
251            item->setCheckable( true );
252            parent->appendChild( item );
253        }
254    }
255}
Note: See TracBrowser for help on using the repository browser.