From 3fb30b108828e3ad6bf811c0f3372a79ec6e9595 Mon Sep 17 00:00:00 2001
From: Nathanael Sensfelder <SpamShield0@MultiAgentSystems.org>
Date: Mon, 18 Nov 2019 10:40:03 +0100
Subject: ...

---
 src/battle/struct/btl_condition_parameters.erl | 37 ++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

(limited to 'src/battle/struct/btl_condition_parameters.erl')

diff --git a/src/battle/struct/btl_condition_parameters.erl b/src/battle/struct/btl_condition_parameters.erl
index eb78245..e00dfeb 100644
--- a/src/battle/struct/btl_condition_parameters.erl
+++ b/src/battle/struct/btl_condition_parameters.erl
@@ -35,6 +35,11 @@
       ataxia_set_targets/2,
       ataxia_set_targets/3,
 
+      get_locations/1,
+      set_locations/2,
+      ataxia_set_locations/2,
+      ataxia_set_locations/3,
+
       get_uses/1,
       set_uses/2,
       ataxia_set_uses/2,
@@ -58,6 +63,38 @@
 %% EXPORTED FUNCTIONS %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 
+%%%%%%%%%%%%%%%%%
+%%%% Locations %%%%
+%%%%%%%%%%%%%%%%%
+-spec get_locations (type(_)) -> list(shr_location:type()).
+get_locations (Params) -> Params#btl_cond_params.locations.
+
+-spec set_locations (list(shr_location:type()), type(ODT)) -> type(ODT).
+set_locations (Locations, Params) ->
+   Params#btl_cond_params{ locations = Locations }.
+
+-spec ataxia_set_locations
+   (
+      list(shr_location:type()),
+      type(ODT)
+   )
+   -> {type(ODT), ataxic:basic()}.
+ataxia_set_locations (Locations, Params) ->
+   ataxia_set_locations(Locations, ataxic:constant(Locations), Params).
+
+-spec ataxia_set_locations
+   (
+      list(shr_location:type()),
+      ataxic:basic(),
+      type(ODT)
+   )
+   -> {type(ODT), ataxic:basic()}.
+ataxia_set_locations (Locations, LocationsAtaxicUpdate, Params) ->
+   {
+      set_locations(Locations, Params),
+      ataxic:update_field(#btl_cond_params.locations, LocationsAtaxicUpdate)
+   }.
+
 %%%%%%%%%%%%%%%%%
 %%%% Targets %%%%
 %%%%%%%%%%%%%%%%%
-- 
cgit v1.2.3-70-g09d2