Remove unnecessary DTEntities and add classification in general class
This commit is contained in:
parent
f0af76520b
commit
94ec3c273e
@ -1,9 +0,0 @@
|
|||||||
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
|
|
@ -1,9 +0,0 @@
|
|||||||
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
|
|
@ -1,8 +1,10 @@
|
|||||||
from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject
|
from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject
|
||||||
|
from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
|
||||||
|
|
||||||
|
|
||||||
class DTSurvivalEntity:
|
class DTSurvivalEntity:
|
||||||
DistFromObjectFromPlayer: DistFromObject
|
DistFromObjectFromPlayer: DistFromObject
|
||||||
|
|
||||||
def __init__(self, distanceFromPlayer: DistFromObject):
|
def __init__(self, distanceFromPlayer: DistFromObject, classification: SurvivalClassification):
|
||||||
|
self.classification = classification
|
||||||
self.distanceFromPlayer = distanceFromPlayer
|
self.distanceFromPlayer = distanceFromPlayer
|
||||||
|
@ -1,9 +0,0 @@
|
|||||||
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
|
|
Loading…
Reference in New Issue
Block a user