diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTFood.py b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTFood.py new file mode 100644 index 0000000..3619f20 --- /dev/null +++ b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTFood.py @@ -0,0 +1,9 @@ +from src.AI.DecisionTrees.projectSpecificClasses.DTEntities.DTSurvivalEntity import DTSurvivalEntity +from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject +from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification + + +class DTFood(DTSurvivalEntity): + def __init__(self, distanceFromPlayer: DistFromObject): + super().__init__(distanceFromPlayer) + self.classification = SurvivalClassification.FOOD diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTRestPlace.py b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTRestPlace.py new file mode 100644 index 0000000..bb2543a --- /dev/null +++ b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTRestPlace.py @@ -0,0 +1,9 @@ +from src.AI.DecisionTrees.projectSpecificClasses.DTEntities.DTSurvivalEntity import DTSurvivalEntity +from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject +from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification + + +class DTRestPlace(DTSurvivalEntity): + def __init__(self, distanceFromPlayer: DistFromObject): + super().__init__(distanceFromPlayer) + self.classification = SurvivalClassification.REST diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTSurvivalEntity.py b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTSurvivalEntity.py new file mode 100644 index 0000000..b8fbab8 --- /dev/null +++ b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTSurvivalEntity.py @@ -0,0 +1,8 @@ +from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject + + +class DTSurvivalEntity: + DistFromObjectFromPlayer: DistFromObject + + def __init__(self, distanceFromPlayer: DistFromObject): + self.distanceFromPlayer = distanceFromPlayer diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTWater.py b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTWater.py new file mode 100644 index 0000000..4aaf1b3 --- /dev/null +++ b/src/AI/DecisionTrees/projectSpecificClasses/DTEntities/DTWater.py @@ -0,0 +1,9 @@ +from src.AI.DecisionTrees.projectSpecificClasses.DTEntities.DTSurvivalEntity import DTSurvivalEntity +from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject +from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification + + +class DTWater(DTSurvivalEntity): + def __init__(self, distanceFromPlayer: DistFromObject): + super().__init__(distanceFromPlayer) + self.classification = SurvivalClassification.WATER diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DTPlayerStats.py b/src/AI/DecisionTrees/projectSpecificClasses/DTPlayerStats.py new file mode 100644 index 0000000..e596bcb --- /dev/null +++ b/src/AI/DecisionTrees/projectSpecificClasses/DTPlayerStats.py @@ -0,0 +1,12 @@ +from src.AI.DecisionTrees.projectSpecificClasses.PlayerStatsValue import PlayerStatsValue + + +class DTPlayerStats: + hungerAmount: PlayerStatsValue + thirstAmount: PlayerStatsValue + staminaAmount: PlayerStatsValue + + def __init__(self, hungerAmount: PlayerStatsValue, thirstAmount: PlayerStatsValue, staminaAmount: PlayerStatsValue): + self.staminaAmount = staminaAmount + self.thirstAmount = thirstAmount + self.hungerAmount = hungerAmount