Handle parent change in Partition copy constructor.

src/core/partition.cpp

@@ -90,7 +90,7 @@ Partition::~Partition()
 
 /** @param other Partition to copy
 */
-Partition::Partition(const Partition& other) :
+Partition::Partition(const Partition& other, PartitionNode* parent) :
     PartitionNode(),
     m_Children(),
     m_Parent(other.m_Parent),
@@ -106,10 +106,12 @@ Partition::Partition(const Partition& other) :
     m_SectorSize(other.m_SectorSize),
     m_State(other.m_State)
 {
+    if ( parent )
+        m_Parent = parent;
+
     setPartitionPath(other.m_PartitionPath);
     for (const auto &child : other.children()) {
-        Partition* p = new Partition(*child);
+        Partition* p = new Partition(*child, this);
-        p->setParent(this);
         m_Children.append(p);
     }
 }

src/core/partition.h

@@ -116,7 +116,7 @@ public:
     Partition(PartitionNode* parent, const Device& device, const PartitionRole& role, FileSystem* fs, qint64 sectorStart, qint64 sectorEnd, QString partitionPath, PartitionTable::Flags availableFlags = PartitionTable::FlagNone, const QString& mountPoint = QString(), bool mounted = false, PartitionTable::Flags activeFlags = PartitionTable::FlagNone, State state = StateNone);
     ~Partition();
 
-    Partition(const Partition&);
+    Partition(const Partition& other, PartitionNode* parent = nullptr);
     Partition& operator=(const Partition&);
 
     bool operator==(const Partition& other) const;

src/core/partitiontable.cpp

@@ -63,7 +63,7 @@ PartitionTable::PartitionTable(const PartitionTable& other)
     for (Partitions::const_iterator it = other.m_Children.constBegin();
          it != other.m_Children.constEnd(); ++it)
     {
-        m_Children.append(new Partition(**it));
+        m_Children.append(new Partition(**it, this));
     }
 }