From 8bae723018193cd662663413060200efeba45f86 Mon Sep 17 00:00:00 2001
From: William JCM <w.jcm59@gmail.com>
Date: Mon, 11 Oct 2021 17:17:14 +0200
Subject: [PATCH] Mass: finish implementing armour part support.

---
 src/Mass/Mass.cpp | 97 +++++++++++++++++++++++++++++++++++++++++++++++
 src/Mass/Mass.h   |  2 +-
 2 files changed, 98 insertions(+), 1 deletion(-)

diff --git a/src/Mass/Mass.cpp b/src/Mass/Mass.cpp
index 4f5b1d1..35e52e3 100644
--- a/src/Mass/Mass.cpp
+++ b/src/Mass/Mass.cpp
@@ -653,6 +653,103 @@ void Mass::getArmourParts() {
     }
 }
 
+auto Mass::writeArmourPart(UnsignedLong index) -> bool {
+    if(index > _armour.parts.size()) {
+        return false;
+    }
+
+    auto unit_data = _mass->at<GenericStructProperty>("UnitData");
+
+    auto armour_array = unit_data->at<ArrayProperty>("Armor_10_12E266C44116DDAF57E99ABB575A4B3C");
+
+    auto part_prop = armour_array->at<GenericStructProperty>(index);
+
+    auto& part = _armour.parts[index];
+
+    if(part_prop->at<ByteProperty>("Slot_3_408BA56F4C9605C7E805CF91B642249C")->enumValue != part.slot) {
+        return false;
+    }
+
+    part_prop->at<IntProperty>("ID_5_ACD101864D3481DE96EDACACC09BDD25")->value = part.id;
+
+    auto part_styles = part_prop->at<ArrayProperty>("Styles_47_3E31870441DFD7DB8BEE5C85C26B365B");
+    for(UnsignedInt i = 0; i < part.styles.size(); i++) {
+        part_styles->at<IntProperty>(i)->value = part.styles[i];
+    }
+
+    auto decals_array = part_prop->at<ArrayProperty>("Decals_42_F358794A4F18497970F56BA9627D3603");
+    for(UnsignedInt i = 0; i < decals_array->items.size(); i++) {
+        auto decal_prop = decals_array->at<GenericStructProperty>(i);
+
+        auto& decal = part.decals[i];
+
+        decal_prop->at<IntProperty>("ID_3_694C0B35404D8A3168AEC89026BC8CF9")->value = decal.id;
+        auto colour_prop = decal_prop->at<ColourStructProperty>("Color_8_1B0B9D2B43DA6AAB9FA549B374D3E606");
+        colour_prop->r = decal.colour.r();
+        colour_prop->g = decal.colour.g();
+        colour_prop->b = decal.colour.b();
+        colour_prop->a = decal.colour.a();
+        auto pos_prop = decal_prop->at<VectorStructProperty>("Position_41_022C8FE84E1AAFE587261E88F2C72250");
+        pos_prop->x = decal.position.x();
+        pos_prop->y = decal.position.y();
+        pos_prop->z = decal.position.z();
+        auto u_prop = decal_prop->at<VectorStructProperty>("UAxis_37_EBEB715F45491AECACCC07A1AE4646D1");
+        u_prop->x = decal.uAxis.x();
+        u_prop->y = decal.uAxis.y();
+        u_prop->z = decal.uAxis.z();
+        auto v_prop = decal_prop->at<VectorStructProperty>("VAxis_39_C31EB2664EE202CAECFBBB84100B5E35");
+        v_prop->x = decal.vAxis.x();
+        v_prop->y = decal.vAxis.y();
+        v_prop->z = decal.vAxis.z();
+        auto offset_prop = decal_prop->at<Vector2DStructProperty>("Offset_29_B02BBBB74FC60F5EDBEBAB8020738020");
+        offset_prop->x = decal.offset.x();
+        offset_prop->y = decal.offset.y();
+        decal_prop->at<FloatProperty>("Scale_32_959D1C2747AFD8D62808468235CBBA40")->value = decal.scale;
+        decal_prop->at<FloatProperty>("Rotation_27_12D7C314493D203D5C2326A03C5F910F")->value = decal.rotation;
+        decal_prop->at<BoolProperty>("Flip_35_CECCFB184CCD9412BD93FE9A8B656BE1")->value = decal.flip;
+        decal_prop->at<BoolProperty>("Wrap_43_A7C68CDF4A92AF2ECDA53F953EE7CA62")->value = decal.wrap;
+    }
+
+    if(!_demo) {
+        auto accs_array = part_prop->at<ArrayProperty>("Accessories_52_D902DD4241FA0050C2529596255153F3");
+
+        for(UnsignedInt i = 0; i < accs_array->items.size(); i++) {
+            auto acc_prop = accs_array->at<GenericStructProperty>(i);
+
+            auto& accessory = part.accessories[i];
+
+            acc_prop->at<IntProperty>("AttachIndex_2_4AFCF6024E4BA7426C6B9F80B8179D20")->value = accessory.attachIndex;
+            acc_prop->at<IntProperty>("ID_4_5757B32647BAE263266259B8A7DFFFC1")->value = accessory.id;
+            auto acc_styles = acc_prop->at<ArrayProperty>("Styles_7_91DEB0F24E24D13FC9472882C11D0DFD");
+            for(UnsignedInt j = 0; j < acc_styles->items.size(); j++) {
+                acc_styles->at<IntProperty>(j)->value = accessory.styles[j];
+            }
+            auto rel_pos_prop = acc_prop->at<VectorStructProperty>("RelativePosition_14_BE8FB2A94074F34B3EDA6683B227D3A1");
+            rel_pos_prop->x = accessory.relativePosition.x();
+            rel_pos_prop->y = accessory.relativePosition.y();
+            rel_pos_prop->z = accessory.relativePosition.z();
+            auto rel_pos_offset_prop = acc_prop->at<VectorStructProperty>("RelativePositionOffset_15_98FD0CE74E44BBAFC2D46FB4CA4E0ED6");
+            rel_pos_offset_prop->x = accessory.relativePositionOffset.x();
+            rel_pos_offset_prop->y = accessory.relativePositionOffset.y();
+            rel_pos_offset_prop->z = accessory.relativePositionOffset.z();
+            auto rel_rot_prop = acc_prop->at<RotatorStructProperty>("RelativeRotation_20_C78C73274E6E78E7878F8C98ECA342C0");
+            rel_rot_prop->x = accessory.relativeRotation.x();
+            rel_rot_prop->y = accessory.relativeRotation.y();
+            rel_rot_prop->z = accessory.relativeRotation.z();
+            auto rel_rot_offset_prop = acc_prop->at<RotatorStructProperty>("RelativeRotationOffset_21_E07FA0EC46728B7BA763C6861249ABAA");
+            rel_rot_offset_prop->x = accessory.relativeRotationOffset.x();
+            rel_rot_offset_prop->y = accessory.relativeRotationOffset.y();
+            rel_rot_offset_prop->z = accessory.relativeRotationOffset.z();
+            auto local_scale_prop = acc_prop->at<VectorStructProperty>("LocalScale_24_DC2D93A742A41A46E7E61D988F15ED53");
+            local_scale_prop->x = accessory.localScale.x();
+            local_scale_prop->y = accessory.localScale.y();
+            local_scale_prop->z = accessory.localScale.z();
+        }
+    }
+
+    return _mass->saveToFile();
+}
+
 auto Mass::armourCustomStyles() -> Containers::StaticArrayView<16, CustomStyle> {
     return _armour.customStyles;
 }
diff --git a/src/Mass/Mass.h b/src/Mass/Mass.h
index 7a0f315..f557522 100644
--- a/src/Mass/Mass.h
+++ b/src/Mass/Mass.h
@@ -158,7 +158,7 @@ class Mass {
 
         auto armourParts() -> Containers::StaticArrayView<38, ArmourPart>;
         void getArmourParts();
-        auto setArmourPart(UnsignedLong index) -> bool;
+        auto writeArmourPart(UnsignedLong index) -> bool;
 
         auto armourCustomStyles() -> Containers::StaticArrayView<16, CustomStyle>;
         void getArmourCustomStyles();