s444517/AI2020_Project
1
2
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updating for individual projecet

Browse Source
This commit is contained in:
Kamila Bobkowska 2020-05-02 15:10:49 +00:00
parent 95bcef8753
commit d98a738d48
1 changed files with 86 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

170
models/dumpster.py
View File

@ -1,84 +1,86 @@
import numpy as np
import pygame
class trash():
def __init__(self):
self.id = 0
self.plastic = 0
self.paper = 0
self.mixed = 0
self.organic = 0
self.glass = 0
self.color = 0
self.xy = 0
self.value = 1
def giveID(self,id):
self.id = id
def throwAway(self):
trash = np.random.randint(1,6)
if trash == 1:
self.plastic += 1
return True
elif trash == 2:
self.paper += 1
return True
elif trash == 3:
self.glass += 1
return True
elif trash == 4:
self.organic += 1
return True
elif trash == 5:
self.mixed += 1
return True
else:
return False
def empty(self):
if self.plastic > 0:
self.plastic -= 1
print("taking plastic")
return True
elif self.paper > 0:
self.paper -= 1
print("taking paper")
return True
elif self.glass > 0:
self.glass -= 1
print("taking glass")
return True
elif self.organic > 0:
self.organic -= 1
print("taking organic")
return True
elif self.mixed > 0:
self.mixed -= 1
print("taking mixed")
return True
else:
print("can't take trash from this dumpster")
return False
def colour(self):
if self.plastic > 0:
self.color = 1
return True
elif self.paper > 0:
self.color = 2
return True
elif self.glass > 0:
self.color = 3
return True
elif self.organic > 0:
self.color = 4
return True
elif self.mixed > 0:
self.color = 5
return True
else:
return False
import numpy as np
import pygame
class trash():
def __init__(self):
self.id = 0
self.plastic = 0
self.paper = 0
self.mixed = 0
self.organic = 0
self.glass = 0
self.color = 0
self.xy = 0
self.value = 1
self.numberx = 0
self.numbery = 0
def giveID(self,id):
self.id = id
def throwAway(self):
trash = np.random.randint(1,6)
if trash == 1:
self.plastic += 1
return True
elif trash == 2:
self.paper += 1
return True
elif trash == 3:
self.glass += 1
return True
elif trash == 4:
self.organic += 1
return True
elif trash == 5:
self.mixed += 1
return True
else:
return False
def empty(self):
if self.plastic > 0:
self.plastic -= 1
print("taking plastic")
return True
elif self.paper > 0:
self.paper -= 1
print("taking paper")
return True
elif self.glass > 0:
self.glass -= 1
print("taking glass")
return True
elif self.organic > 0:
self.organic -= 1
print("taking organic")
return True
elif self.mixed > 0:
self.mixed -= 1
print("taking mixed")
return True
else:
print("can't take trash from this dumpster")
return False
def colour(self):
if self.plastic > 0:
self.color = 1
return True
elif self.paper > 0:
self.color = 2
return True
elif self.glass > 0:
self.color = 3
return True
elif self.organic > 0:
self.color = 4
return True
elif self.mixed > 0:
self.color = 5
return True
else:
return False