Remove unnecessary DTEntities and add classification in general class

This commit is contained in:
Michał Czekański 2020-05-23 22:10:05 +02:00
parent f0af76520b
commit 94ec3c273e
4 changed files with 3 additions and 28 deletions

View File

@ -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

View File

@ -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

View File

@ -1,8 +1,10 @@
from src.AI.DecisionTrees.projectSpecificClasses.DistFromObject import DistFromObject
from src.AI.DecisionTrees.projectSpecificClasses.SurvivalClassification import SurvivalClassification
class DTSurvivalEntity:
DistFromObjectFromPlayer: DistFromObject
def __init__(self, distanceFromPlayer: DistFromObject):
def __init__(self, distanceFromPlayer: DistFromObject, classification: SurvivalClassification):
self.classification = classification
self.distanceFromPlayer = distanceFromPlayer

View File

@ -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