ZombieSunday/weapon.py at master · flags/ZombieSunday · GitHub

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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
# Handles weapons (duh)
# Author list:
# 	Luke Martin <ltmartin@bsu.edu>
# 	Michael Milkovic <mlmilkovic@bsu.edu>
# 	Derek Onay <dsonay@bsu.edu>
# 	Ryan Wiesjahn <rwiesjahn@bsu.edu>
# LobForce

import somber as somber_engine
from bullet import *
import config

class Weapon:
	def __init__(self, somber, character, attachments=[None, None]):
		self.somber = somber
		self.attachments = attachments
		self.type = WeaponType.Default
		self.character = character
		self.firefire_check = False
		self.ammo = [0, config.PLAYER_AMMO]
		self.ammo[0] = self.ammo[1]
		self.ammo_consumption = config.DEFAULT_AMMO_CONSUMPTION
		self.rate = config.DEFAULT_RATE
		self.timer = self.rate

		self.set_weapon_type()

	def set_weapon_type(self):
		if self.attachments.count(Attachment.Speed) == 1:
			if self.attachments.count(Attachment.Fire) == 1:
				self.type = WeaponType.SpeedFire
				self.rate = config.SPEEDFIRE_RATE
				self.ammo_consumption = config.SPEEDFIRE_AMMO_CONSUMPTION
			elif self.attachments.count(Attachment.Lob) == 1:
				self.type = WeaponType.SpeedLob
				self.rate = config.SPEEDLOB_RATE
				self.ammo_consumption = config.SPEEDLOB_AMMO_CONSUMPTION
			elif self.attachments.count(Attachment.Force) == 1:
				self.type = WeaponType.SpeedForce
				self.rate = config.SPEEDFORCE_RATE
				self.ammo_consumption = config.SPEEDFORCE_AMMO_CONSUMPTION
			else:
				self.type = WeaponType.Speed
				self.rate = config.SPEED_RATE
				self.ammo_consumption = config.SPEED_AMMO_CONSUMPTION

		elif self.attachments.count(Attachment.Fire) == 1:
			if self.attachments.count(Attachment.Lob) == 1:
				self.type = WeaponType.FireLob
				self.rate = config.FIRELOB_RATE
				self.ammo_consumption = config.FIRELOB_AMMO_CONSUMPTION
			elif self.attachments.count(Attachment.Force) == 1:
				self.type = WeaponType.FireForce
				self.rate = config.FIREFORCE_RATE
				self.ammo_consumption = config.FIREFORCE_AMMO_CONSUMPTION
			else:
				self.type = WeaponType.Fire
				self.rate = config.FIRE_RATE
				self.ammo_consumption = config.FIRE_AMMO_CONSUMPTION

		elif self.attachments.count(Attachment.Lob) == 1:
			if self.attachments.count(Attachment.Force) == 1:
				self.type = WeaponType.LobForce
				self.rate = config.LOBFORCE_RATE
				self.ammo_consumption = config.LOBFORCE_AMMO_CONSUMPTION
			else:
				self.type = WeaponType.Lob
				self.rate = config.LOB_RATE
				self.ammo_consumption = config.LOB_AMMO_CONSUMPTION

		elif self.attachments.count(Attachment.Force) == 1:
				self.type = WeaponType.Force
				self.rate = config.FORCE_RATE
				self.ammo_consumption = config.FORCE_AMMO_CONSUMPTION

		elif self.attachments.count(Attachment.Speed) == 2:
			self.type = WeaponType.SpeedSpeed
			self.rate = config.SPEEDSPEED_RATE
			self.ammo_consumption = config.SPEEDSPEED_AMMO_CONSUMPTION

		elif self.attachments.count(Attachment.Fire) == 2:
			self.type = WeaponType.FireFire
			self.rate = config.FIREFIRE_RATE
			self.ammo_consumption = config.FIREFIRE_AMMO_CONSUMPTION

		elif self.attachments.count(Attachment.Lob) == 2:
			self.type = WeaponType.LobLob
			self.rate = config.LOBLOB_RATE
			self.ammo_consumption = config.LOBLOB_AMMO_CONSUMPTION

		elif self.attachments.count(Attachment.Force) == 2:
			self.type = WeaponType.ForceForce
			self.rate = config.FORCEFORCE_RATE
			self.ammo_consumption = config.FORCEFORCE_AMMO_CONSUMPTION

		elif self.attachments.count(None) == 2:
			self.type = WeaponType.Default
			self.rate = config.DEFAULT_RATE
			self.ammo_consumption = config.DEFAULT_AMMO_CONSUMPTION

		self.timer = self.rate

	def update(self, delta):
		self.weapon_timer(delta)

	def weapon_timer(self, delta):
		if self.timer < self.rate:
			self.timer += delta

	def fire(self):
		if self.ammo[0] - self.ammo_consumption >= 0:
			if self.timer >= self.rate:
				self.create_bullet()
				self.ammo[0] -= self.ammo_consumption
				self.timer = 0

	def create_bullet(self):
		self.somber.play_sound(config.SOUND_GUN_GENERIC)

		if self.type == WeaponType.Speed:
			SpeedBullet(self.somber)
		elif self.type == WeaponType.Fire:
			FireBullet(self.somber)
		elif self.type == WeaponType.Lob:
			LobBullet(self.somber)
		elif self.type == WeaponType.Force:
			ForceBullet(self.somber)
		elif self.type == WeaponType.SpeedSpeed:
			SpeedSpeedBullet(self.somber)
		elif self.type == WeaponType.SpeedFire:
			SpeedFireBullet(self.somber)
		elif self.type == WeaponType.SpeedLob:
			SpeedLobBullet(self.somber)
		elif self.type == WeaponType.SpeedForce:
			SpeedForceBullet(self.somber)
		elif self.type == WeaponType.FireFire:
			if not self.firefire_check:
				FireFireBullet(self.somber)
				self.firefire_check = True
		elif self.type == WeaponType.FireLob:
			FireLobBullet(self.somber)
		elif self.type == WeaponType.FireForce:
			FireForceBullet(self.somber)
		elif self.type == WeaponType.LobLob:
			LobLobBullet(self.somber)
		elif self.type == WeaponType.LobForce:
			LobForceBullet(self.somber)
		elif self.type == WeaponType.ForceForce:
			ForceForceBullet(self.somber)
		else:
			DefaultBullet(self.somber)


class WeaponType:
	Default, Speed, SpeedSpeed, SpeedFire, SpeedLob, SpeedForce, Fire, FireFire, FireLob, FireForce, Lob, LobLob, LobForce, Force, ForceForce = range(15)

class Attachment:
	Speed, Fire, Lob, Force = range(4)