Dispatch files in core/ gui/ and jobs/ dirs

src/modules/partition/core/PartitionModel.h

@@ -0,0 +1,105 @@
+/* === This file is part of Calamares - <http://github.com/calamares> ===
+ *
+ *   Copyright 2014, Aurélien Gâteau <agateau@kde.org>
+ *
+ *   Calamares 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 3 of the License, or
+ *   (at your option) any later version.
+ *
+ *   Calamares is distributed in the hope that it will be useful,
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with Calamares. If not, see <http://www.gnu.org/licenses/>.
+ */
+#ifndef PARTITIONMODEL_H
+#define PARTITIONMODEL_H
+
+// Qt
+#include <QAbstractItemModel>
+
+class Device;
+class Partition;
+class PartitionNode;
+
+/**
+ * A Qt tree model which exposes the partitions of a device.
+ *
+ * Its depth is only more than 1 if the device has extended partitions.
+ *
+ * Note on updating:
+ *
+ * The Device class does not notify the outside world of changes on the
+ * Partition objects it owns. Since a Qt model must notify its views *before*
+ * and *after* making changes, it is important to make use of
+ * the PartitionModel::ResetHelper class to wrap changes.
+ *
+ * This is what PartitionCoreModule does when it create jobs.
+ */
+class PartitionModel : public QAbstractItemModel
+{
+public:
+    /**
+     * This helper class must be instantiated on the stack *before* making
+     * changes to the device represented by this model. It will cause the model
+     * to emit modelAboutToBeReset() when instantiated and modelReset() when
+     * destructed.
+     */
+    class ResetHelper
+    {
+    public:
+        ResetHelper( PartitionModel* model );
+        ~ResetHelper();
+
+        ResetHelper( const ResetHelper& ) = delete;
+        ResetHelper& operator=( const ResetHelper& ) = delete;
+    private:
+        PartitionModel* m_model;
+    };
+
+    enum
+    {
+        // The raw size, as a qlonglong. This is different from the DisplayRole of
+        // SizeColumn, which is a human-readable string.
+        SizeRole = Qt::UserRole + 1,
+        IsFreeSpaceRole
+    };
+
+    enum Column
+    {
+        NameColumn,
+        FileSystemColumn,
+        MountPointColumn,
+        SizeColumn,
+        ColumnCount // Must remain last
+    };
+
+    PartitionModel( QObject* parent = 0 );
+    /**
+     * device must remain alive for the life of PartitionModel
+     */
+    void init( Device* device );
+
+    // QAbstractItemModel API
+    QModelIndex index( int row, int column, const QModelIndex& parent = QModelIndex() ) const override;
+    QModelIndex parent( const QModelIndex& child ) const override;
+    int columnCount( const QModelIndex& parent = QModelIndex() ) const override;
+    int rowCount( const QModelIndex& parent = QModelIndex() ) const override;
+    QVariant data( const QModelIndex& index, int role = Qt::DisplayRole ) const override;
+    QVariant headerData( int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const override;
+
+    Partition* partitionForIndex( const QModelIndex& index ) const;
+
+    Device* device() const
+    {
+        return m_device;
+    }
+
+private:
+    Device* m_device;
+};
+
+#endif /* PARTITIONMODEL_H */