From 73597dff9dd5119d2a176595679422b2c0f0d5cc Mon Sep 17 00:00:00 2001
From: Joaquin <67109235+Taiga74164@users.noreply.github.com>
Date: Fri, 9 Sep 2022 01:14:32 -0600
Subject: [PATCH] Added Oculies filter to auto pickup/pickup range

---
 cheat-library/src/user/cheat/world/AutoLoot.cpp | 11 ++++++++---
 cheat-library/src/user/cheat/world/AutoLoot.h   |  1 +
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/cheat-library/src/user/cheat/world/AutoLoot.cpp b/cheat-library/src/user/cheat/world/AutoLoot.cpp
index 602ad44..c1cd036 100644
--- a/cheat-library/src/user/cheat/world/AutoLoot.cpp
+++ b/cheat-library/src/user/cheat/world/AutoLoot.cpp
@@ -23,6 +23,7 @@ namespace cheat::feature
 		NF(f_PickupFilter_Animals,	 "Animals filter",			"AutoLoot", true),
 		NF(f_PickupFilter_DropItems, "Drop items filter",		"AutoLoot", true),
 		NF(f_PickupFilter_Resources, "Resources filter",		"AutoLoot", true),
+		NF(f_PickupFilter_Oculus, "Oculus filter",				"AutoLoot", true),
 		NF(f_Chest,			 "Chests",							"AutoLoot", false),
 		NF(f_Leyline,		 "Leylines",						"AutoLoot", false),
 		NF(f_Investigate,	 "Search points",					"AutoLoot", false),
@@ -123,7 +124,8 @@ namespace cheat::feature
 			ConfigWidget("Enabled", f_PickupFilter, "Enable pickup filter.\n");
 			ConfigWidget("Animals", f_PickupFilter_Animals, "Fish, Lizard, Frog, Flying animals."); ImGui::SameLine();
 			ConfigWidget("Drop Items", f_PickupFilter_DropItems, "Material, Mineral, Artifact."); ImGui::SameLine();
-			ConfigWidget("Resources", f_PickupFilter_Resources, "Everything beside Animals and Drop Items (Plants, Books, etc).");
+			ConfigWidget("Resources", f_PickupFilter_Resources, "Everything beside Animals and Drop Items (Plants, Books, etc)."); ImGui::SameLine();
+			ConfigWidget("Oculus", f_PickupFilter_Oculus, "Filter Oculus");
 	    }
     	ImGui::EndGroupPanel();
     }
@@ -248,20 +250,23 @@ namespace cheat::feature
 
 	void AutoLoot::OnCheckIsInPosition(bool& result, app::BaseEntity* entity)
 	{
+		// TODO: Maybe add a list of filter for all GatherObject instead of just using entityType in general.
+		auto& manager = game::EntityManager::instance();
+
 		if (f_AutoPickup || f_UseCustomRange) {
 			float pickupRange = f_UseCustomRange ? f_CustomRange : g_default_range;
 			if (f_PickupFilter)
 			{
 				if (!f_PickupFilter_Animals && entity->fields.entityType == app::EntityType__Enum_1::EnvAnimal ||
 					!f_PickupFilter_DropItems && entity->fields.entityType == app::EntityType__Enum_1::DropItem ||
-					!f_PickupFilter_Resources && entity->fields.entityType == app::EntityType__Enum_1::GatherObject)
+					!f_PickupFilter_Resources && entity->fields.entityType == app::EntityType__Enum_1::GatherObject ||
+					!f_PickupFilter_Oculus && game::filters::combined::Oculies.IsValid(manager.entity(entity->fields._runtimeID_k__BackingField)))
 				{
 					result = false;
 					return;
 				}
 			}
 			
-			auto& manager = game::EntityManager::instance();
 			result = manager.avatar()->distance(entity) < pickupRange;
 		}
 	}
diff --git a/cheat-library/src/user/cheat/world/AutoLoot.h b/cheat-library/src/user/cheat/world/AutoLoot.h
index 5b7503f..4b9d5ba 100644
--- a/cheat-library/src/user/cheat/world/AutoLoot.h
+++ b/cheat-library/src/user/cheat/world/AutoLoot.h
@@ -28,6 +28,7 @@ namespace cheat::feature
 		config::Field<bool> f_PickupFilter_Animals;
 		config::Field<bool> f_PickupFilter_DropItems;
 		config::Field<bool> f_PickupFilter_Resources;
+		config::Field<bool> f_PickupFilter_Oculus;
 
 		static AutoLoot& GetInstance();