forked from RyanAshbaugh/AnimatSimulationMSU
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Stimuli.py
81 lines (56 loc) · 1.86 KB
/
Stimuli.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# -*- coding: utf-8 -*-
"""
Created on Tue Apr 22 18:46:58 2014
World Stimulus Module
--food,prey etc
Contains base Stimulus class and all derived classes
"""
##################### Stimuli Base Class ######################################################
## Anything animat can interact with in world ##
class Stimulus():
#constructor
#**kwargs
# startPos : takes (x,y) coordinate for starting position
# potency : sets smell potency? not sure if this is needed yet, just putting here
def __init__(self, startPos = (0,0)):
self.pos = startPos
def getPos(self):
return [self.pos[0],self.pos[1]]
##################### Food CLass ######################################################
## Basic food object ##
class Food_A(Stimulus):
def __init__(self,loc,amount=1.0):
Stimulus.__init__(self,startPos = loc)
self.type = "A"
self.amt = amount
self.smellStr = 1
self.calories = 10
self.image = "apple_bunch.png"
def getType(self):
return self.type
def getAmount(self):
return self.amt
def getSmell(self):
return (self.amt * self.smellStr)
def getCalories(self):
return self.calories
def decrAmt(self):
self.amt -= 1.0
class Food_B(Stimulus):
def __init__(self,loc,amount=1.0):
Stimulus.__init__(self,startPos = loc)
self.type = "B"
self.amt = amount
self.smellStr = 1
self.calories = 5
self.image = "apple.png"
def getType(self):
return self.type
def getAmount(self):
return self.amt
def getSmell(self):
return (self.amt * self.smellStr)/10.0 #divide by 10 to scale down to appropriate levels for network
def getCalories(self):
return self.calories
def decrAmt(self):
self.amt -= 1.0