Show root feature in treeview

se-sic · Nov 20, 2023 · 90ff5a8 · 90ff5a8
1 parent 1c016ad
commit 90ff5a8
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tools/fm-editor/tree/FeatureTreeViewModel.h b/tools/fm-editor/tree/FeatureTreeViewModel.h
@@ -11,11 +11,15 @@ class FeatureTreeViewModel : public QAbstractItemModel {
   FeatureTreeViewModel(vara::feature::FeatureModel *Model, QObject *Parent)
       : QAbstractItemModel(Parent) {
     auto UniqueRoot = FeatureTreeItem::createFeatureTreeItem(Model->getRoot());
-    RootItem = UniqueRoot.get();
+    RootItem = new FeatureTreeItemFeature(nullptr);
+    RootItem->addChild(UniqueRoot.get());
+    auto RawRoot = UniqueRoot.get();
     Items.push_back(std::move(UniqueRoot));
-    buildRecursive(RootItem);
+    buildRecursive(RawRoot);
   }
-  ~FeatureTreeViewModel() override { std::destroy(Items.begin(), Items.end()); }
+  ~FeatureTreeViewModel() override {
+    delete RootItem;
+    std::destroy(Items.begin(), Items.end()); }
 
   std::vector<std::unique_ptr<FeatureTreeItem>> *getItems();
   [[nodiscard]] QVariant data(const QModelIndex &Index,