add [Expand] / [Collapse] text for expandable structs and expand/…

…collapse on click
spelunky-fyi · May 13, 2024 · 7fa6828 · 7fa6828
1 parent b284603
commit 7fa6828
Showing 1 changed file with 28 additions and 9 deletions.
diff --git a/src/QtHelpers/TreeViewMemoryFields.cpp b/src/QtHelpers/TreeViewMemoryFields.cpp
@@ -1861,21 +1861,28 @@ void S2Plugin::TreeViewMemoryFields::updateRow(int row, std::optional<uintptr_t>
         }
         case MemoryFieldType::EntitySubclass:
         {
-            // can't be a pointer, nothing to do here
+            // can't be a pointer
+            if (isExpanded(itemField->index()))
+                itemValue->setData("<font color='darkMagenta'><u>[Collapse]</u></font>", Qt::DisplayRole);
+            else
+                itemValue->setData("<font color='#aaa'><u>[Expand]</u></font>", Qt::DisplayRole);
+
             if (shouldUpdateChildren)
             {
                 for (int x = 0; x < itemField->rowCount(); ++x)
-                {
                     updateRow(x, newAddr, newAddrComparison, itemField);
-                }
             }
             break;
         }
         case MemoryFieldType::DefaultStructType:
         {
+            if (isExpanded(itemField->index()))
+                itemValue->setData("<font color='darkMagenta'><u>[Collapse]</u></font>", Qt::DisplayRole);
+            else
+                itemValue->setData("<font color='#aaa'><u>[Expand]</u></font>", Qt::DisplayRole);
+
             if (isPointer)
             {
-                itemValue->setData(itemValueHex->data(Qt::DisplayRole), Qt::DisplayRole);
                 if (comparisonActive)
                 {
                     itemComparisonValue->setData(itemComparisonValueHex->data(Qt::DisplayRole), Qt::DisplayRole);
@@ -1893,6 +1900,11 @@ void S2Plugin::TreeViewMemoryFields::updateRow(int row, std::optional<uintptr_t>
         }
         case MemoryFieldType::Dummy:
         {
+            if (isExpanded(itemField->index()))
+                itemValue->setData("<font color='darkMagenta'><u>[Collapse]</u></font>", Qt::DisplayRole);
+            else
+                itemValue->setData("<font color='#aaa'><u>[Expand]</u></font>", Qt::DisplayRole);
+
             if (shouldUpdateChildren)
             {
                 for (int x = 0; x < itemField->rowCount(); ++x)
@@ -2214,12 +2226,19 @@ void S2Plugin::TreeViewMemoryFields::cellClicked(const QModelIndex& index)
                     break;
                 }
                 case MemoryFieldType::DefaultStructType:
+                case MemoryFieldType::EntitySubclass:
+                case MemoryFieldType::Dummy:
                 {
-                    bool isPointer = getDataFrom(index, gsColField, gsRoleIsPointer).toBool();
-                    if (isPointer)
+                    auto fieldIndex = index.sibling(index.row(), gsColField);
+                    if (isExpanded(fieldIndex))
+                        collapse(fieldIndex);
+                    else
+                        expand(index.sibling(index.row(), gsColField));
+
+                    break;
                 }
                 case MemoryFieldType::UTF16Char:
-                    {
+                {
                     auto offset = clickedItem->data(gsRoleMemoryAddress).toULongLong();
                     if (offset != 0)
                     {
@@ -2249,15 +2268,15 @@ void S2Plugin::TreeViewMemoryFields::cellClicked(const QModelIndex& index)
                 {
                     auto offset = clickedItem->data(gsRoleMemoryAddress).toULongLong();
                     if (offset != 0)
-                        {
+                    {
                         auto fieldName = getDataFrom(index, gsColField, gsRoleUID).toString();
                         auto s = QString::fromStdString(ReadConstString(offset));
                         // [Known Issue]: Now way to safely determinate allowed lenght, so we just allow as much characters as there is already
                         auto dialog = new DialogEditString(fieldName, s, offset, s.length(), dataType, this);
                         dialog->exec();
                     }
                     break;
-                        }
+                }
                 case MemoryFieldType::UTF8StringFixedSize:
                 {
                     auto offset = clickedItem->data(gsRoleMemoryAddress).toULongLong();