def test_PerParticleEmitter_emit_empty_source(self): from lepton import ParticleGroup from lepton.emitter import PerParticleEmitter emitter = PerParticleEmitter(ParticleGroup()) group = ParticleGroup() self.assertEqual(len(group), 0) emitter.emit(10, group) group.update(0) self.assertEqual(len(group), 0)
def test_PerParticleEmitter_emit_empty_source(self):
from lepton import ParticleGroup
from lepton.emitter import PerParticleEmitter
emitter = PerParticleEmitter(ParticleGroup())
group = ParticleGroup()
self.assertEqual(len(group), 0)
emitter.emit(10, group)
group.update(0)
self.assertEqual(len(group), 0)
def __init__(self, position):
x, y = position
spark_emitter = StaticEmitter(
template=Particle(position=(uniform(x - 5,
x + 5), uniform(y - 5,
y + 5), 0),
size=(10, ) * 3,
color=self.color),
deviation=Particle(velocity=(gauss(0, 5), gauss(0, 5), 0),
age=1.5),
velocity=domain.Sphere((0, gauss(40, 20), 0), 60, 60))
spark_emitter.emit(int(gauss(60, 40)) + 50, self.sparks)
spread = abs(gauss(0.4, 1.0))
self.trail_emitter = PerParticleEmitter(
self.sparks,
rate=uniform(5, 30),
template=Particle(size=(6, ) * 3, color=self.color),
deviation=Particle(velocity=(spread, spread, spread),
age=self.lifetime * 0.75))
self.trails.bind_controller(self.trail_emitter)
self.splosions.add(self)
pyglet.clock.schedule_once(self.die, self.lifetime)
def test_PerParticleEmitter_template(self):
from lepton import Particle, ParticleGroup
from lepton.emitter import PerParticleEmitter
source_group = ParticleGroup()
source_group.new(Particle(position=(1, 1, 1)))
source_group.new(Particle(position=(2, 2, 2)))
source_group.update(0)
emitter = PerParticleEmitter(source_group,
rate=1,
template=Particle(position=(1.0, 1.0,
1.0),
velocity=(0, 5, 2),
color=(0.5, 0.5, 0.5,
1.0)))
group = ParticleGroup()
count = emitter(1, group)
group.update(0)
self.assertEqual(count, len(source_group))
self.assertEqual(len(group), len(source_group))
particle = list(group)[0]
self.assertVector(particle.position, (1, 1, 1))
self.assertVector(particle.velocity, (0, 5, 2))
self.assertColor(particle.color, (0.5, 0.5, 0.5, 1.0))
particle = list(group)[1]
self.assertVector(particle.position, (2, 2, 2))
self.assertVector(particle.velocity, (0, 5, 2))
self.assertColor(particle.color, (0.5, 0.5, 0.5, 1.0))
def test_PerParticleEmitter_emit(self): from lepton import ParticleGroup from lepton.emitter import PerParticleEmitter source_group = ParticleGroup() source_group.new(object()) source_group.new(object()) source_group.new(object()) source_group.update(0) emitter = PerParticleEmitter(source_group) group = ParticleGroup() self.assertEqual(len(group), 0) emitter.emit(10, group) group.update(0) expected = 10 * len(source_group) self.assertEqual(len(group), expected) # Negative emit value is equivilant to zero emitter.emit(-10, group) group.update(0) self.assertEqual(len(group), expected)
def __init__(self):
color = (uniform(0, 1), uniform(0, 1), uniform(0, 1), 1)
while max(color[:3]) < 0.9:
color = (uniform(0, 1), uniform(0, 1), uniform(0, 1), 1)
spark_emitter = StaticEmitter(
template=Particle(position=(uniform(-50, 50), uniform(-30, 30),
uniform(-30, 30)),
color=color),
deviation=Particle(velocity=(gauss(0, 5), gauss(0, 5), gauss(0,
5)),
age=1.5),
velocity=domain.Sphere((0, gauss(40, 20), 0), 60, 60))
self.sparks = ParticleGroup(controllers=[
Lifetime(self.lifetime * 0.75),
Movement(damping=0.93),
ColorBlender([(0, (1, 1, 1, 1)), (2, color),
(self.lifetime, color)]),
Fader(fade_out_start=1.0, fade_out_end=self.lifetime * 0.5),
],
renderer=PointRenderer(
abs(gauss(10, 3)), spark_texturizer))
spark_emitter.emit(int(gauss(60, 40)) + 50, self.sparks)
spread = abs(gauss(0.4, 1.0))
self.trail_emitter = PerParticleEmitter(
self.sparks,
rate=uniform(5, 30),
template=Particle(color=color),
deviation=Particle(velocity=(spread, spread, spread),
age=self.lifetime * 0.75))
self.trails = ParticleGroup(controllers=[
Lifetime(self.lifetime * 1.5),
Movement(damping=0.83),
ColorBlender([(0, (1, 1, 1, 1)), (1, color),
(self.lifetime, color)]),
Fader(max_alpha=0.75,
fade_out_start=0,
fade_out_end=gauss(self.lifetime, self.lifetime * 0.3)),
self.trail_emitter
],
renderer=PointRenderer(
10, trail_texturizer))
pyglet.clock.schedule_once(self.die, self.lifetime * 2)
def test_PerParticleEmitter_partial(self):
from lepton import Particle, ParticleGroup
from lepton.emitter import PerParticleEmitter
source_group = ParticleGroup()
source_group.new(object())
source_group.new(object())
source_group.new(object())
source_group.update(0)
emitter = PerParticleEmitter(source_group, rate=1)
group = ParticleGroup()
# It should take four quarter second updates to emit any
self.assertEqual(emitter(0.25, group), 0)
self.assertEqual(emitter(0.25, group), 0)
self.assertEqual(emitter(0.25, group), 0)
self.assertEqual(emitter(0.25, group), len(source_group))
group.update(0)
self.assertEqual(len(group), len(source_group))
def test_PerParticleEmitter_time_to_live(self):
from lepton import Particle, ParticleGroup
from lepton.emitter import PerParticleEmitter
source_group = ParticleGroup()
source_group.new(object())
source_group.new(object())
source_group.new(object())
source_group.new(object())
source_group.update(0)
emitter = PerParticleEmitter(source_group, rate=1, time_to_live=3.0)
group = ParticleGroup(controllers=[emitter])
count = emitter(2, group)
self.assertEqual(count, 2 * len(source_group))
self.assertEqual(emitter.time_to_live, 1)
self.failUnless(emitter in group.controllers)
count = emitter(2, group)
# Since only one second remained before expiring
# only one particle per should be emitted
self.assertEqual(count, len(source_group))
self.assertEqual(emitter.time_to_live, 0)
self.failUnless(emitter not in group.controllers)
def test_PerParticleEmitter_emit(self):
from lepton import ParticleGroup
from lepton.emitter import PerParticleEmitter
source_group = ParticleGroup()
source_group.new(object())
source_group.new(object())
source_group.new(object())
source_group.update(0)
emitter = PerParticleEmitter(source_group)
group = ParticleGroup()
self.assertEqual(len(group), 0)
emitter.emit(10, group)
group.update(0)
expected = 10 * len(source_group)
self.assertEqual(len(group), expected)
# Negative emit value is equivilant to zero
emitter.emit(-10, group)
group.update(0)
self.assertEqual(len(group), expected)
def __init__(self,
x,
y,
z,
r,
side=-1,
hp=50,
controlable=False,
weapon_range=5,
dispersion=5,
agility=50,
weapon_base_damage=2,
guidance=100,
shortguide=0,
partColor=(0.6, 0.5, 0.2, 1),
firerate=10,
shots=1,
vo=30,
maxvel=10,
ammoMaxvel=20,
combatDistance=50,
behavior=0,
commander=None,
multipleTargets=False,
name="",
ammoDamp=0.98):
self.name = name
self.domain = Sphere(
(x, y, z), r
) # a.center -> vector del centro , a.outer_radius -> radio externo , a.inner_radius -> radio interno
self.size = r
self.controller = Collector(self.domain, callback=self.contact)
self.magnet = Magnet(self.domain, charge=guidance, exponent=shortguide)
self.commander = commander
self.mission = self.domain.center
self.target = None
self.alive = True
self.targetMode = ['standard', 1]
self.behavior = behavior # 0:free 1: escort 2: slave
self.hp = hp
self.agility = agility
self.maxvel = maxvel
self.timer = {0: 0, 1: 2, 2: 0, 3: 0, 4: 0} # timers placeholder
self.counter = {0: 0, 1: 0} # counters placeholder
self.side = side
self.combatDistance = combatDistance
self.velocity = Vec3(0, 0, 0)
self.multipleTargets = multipleTargets
self.firerate = firerate
self.weapon_base_damage = weapon_base_damage
wbd = self.weapon_base_damage
rr = r * 2
self.dispersion = dispersion
self.vo = vo
self.ammoDamp = ammoDamp
self.ammoMaxvel = ammoMaxvel
self.shots = shots
self.weapon_range = weapon_range
self.xx = self.yy = self.zz = 0
self.Objective = Sphere((0, 0, 0), 1)
self.color = partColor # (0.4,0.5,0.4,0.5)
#self.prevController = Collector(self.domain)#DUMMY CONTROLLER
self.controlable = controlable
self.impacto = ParticleGroup(renderer=BillboardRenderer(texturizer),
controllers=[
Lifetime(1),
Fader(fade_out_start=0,
fade_out_end=1),
])
self.deathplosion = ParticleGroup(
renderer=BillboardRenderer(texturizer),
controllers=[
Lifetime(self.size / 5 + 1),
Fader(fade_out_start=0, fade_out_end=self.size / 5 + 1),
])
self.selector_emitter = StaticEmitter(template=Particle(
position=(0, 0, 0),
color=self.color,
))
self.impacto_emitter = StaticEmitter(
template=Particle(
position=(0, 0, 0),
color=(0.9, 0.8, 0.8),
),
position=self.domain,
#size=[(5, 5, 5), (10, 10, 10), (15, 15, 15)],
)
self.hull = ParticleGroup(renderer=BillboardRenderer(texturizer2),
controllers=[
Lifetime(100000),
Movement(max_velocity=self.maxvel,
damping=0.98),
Magnet(self.Objective,
charge=self.agility,
exponent=0),
])
emiter = StaticEmitter(position=self.domain,
template=Particle(
color=self.color,
size=(rr, rr, rr),
))
emiter.emit(1, self.hull)
if trails:
if maxvel / r >= 20:
self.trail = ParticleGroup(
renderer=BillboardRenderer(texturizer2),
controllers=[
Lifetime(trailSize[0]),
Fader(fade_in_start=0,
fade_in_end=0.1,
fade_out_start=0,
fade_out_end=trailSize[0]),
Growth(-1 * r),
PerParticleEmitter(self.hull,
rate=trailSize[1],
template=Particle(
color=self.color,
size=(rr, rr, rr),
)),
])
self.ammo = ParticleGroup(renderer=BillboardRenderer(texturizer),
controllers=[
self.magnet,
Movement(min_velocity=0,
max_velocity=self.ammoMaxvel,
damping=self.ammoDamp),
Lifetime(self.weapon_range),
Fader(fade_out_start=self.weapon_range -
1,
fade_out_end=self.weapon_range),
])
self.weapon = PerParticleEmitter(
self.hull, # rate=self.firerate,
template=Particle(
velocity=self.velocity, # fixed value
position=(self.getPosition()),
color=partColor,
),
position=self.domain,
size=[(wbd * 0.5, wbd * 0.5, wbd * 0.5), (wbd, wbd, wbd),
(wbd * 1.5, wbd * 1.5, wbd * 1.5)],
deviation=Particle(
velocity=(self.dispersion, self.dispersion,
self.dispersion * d3),
rotation=(0, 0, math.pi / 6),
#color=(0.05,0.05,0.05,0),
))
class ship(object):
def __init__(self,
x,
y,
z,
r,
side=-1,
hp=50,
controlable=False,
weapon_range=5,
dispersion=5,
agility=50,
weapon_base_damage=2,
guidance=100,
shortguide=0,
partColor=(0.6, 0.5, 0.2, 1),
firerate=10,
shots=1,
vo=30,
maxvel=10,
ammoMaxvel=20,
combatDistance=50,
behavior=0,
commander=None,
multipleTargets=False,
name="",
ammoDamp=0.98):
self.name = name
self.domain = Sphere(
(x, y, z), r
) # a.center -> vector del centro , a.outer_radius -> radio externo , a.inner_radius -> radio interno
self.size = r
self.controller = Collector(self.domain, callback=self.contact)
self.magnet = Magnet(self.domain, charge=guidance, exponent=shortguide)
self.commander = commander
self.mission = self.domain.center
self.target = None
self.alive = True
self.targetMode = ['standard', 1]
self.behavior = behavior # 0:free 1: escort 2: slave
self.hp = hp
self.agility = agility
self.maxvel = maxvel
self.timer = {0: 0, 1: 2, 2: 0, 3: 0, 4: 0} # timers placeholder
self.counter = {0: 0, 1: 0} # counters placeholder
self.side = side
self.combatDistance = combatDistance
self.velocity = Vec3(0, 0, 0)
self.multipleTargets = multipleTargets
self.firerate = firerate
self.weapon_base_damage = weapon_base_damage
wbd = self.weapon_base_damage
rr = r * 2
self.dispersion = dispersion
self.vo = vo
self.ammoDamp = ammoDamp
self.ammoMaxvel = ammoMaxvel
self.shots = shots
self.weapon_range = weapon_range
self.xx = self.yy = self.zz = 0
self.Objective = Sphere((0, 0, 0), 1)
self.color = partColor # (0.4,0.5,0.4,0.5)
#self.prevController = Collector(self.domain)#DUMMY CONTROLLER
self.controlable = controlable
self.impacto = ParticleGroup(renderer=BillboardRenderer(texturizer),
controllers=[
Lifetime(1),
Fader(fade_out_start=0,
fade_out_end=1),
])
self.deathplosion = ParticleGroup(
renderer=BillboardRenderer(texturizer),
controllers=[
Lifetime(self.size / 5 + 1),
Fader(fade_out_start=0, fade_out_end=self.size / 5 + 1),
])
self.selector_emitter = StaticEmitter(template=Particle(
position=(0, 0, 0),
color=self.color,
))
self.impacto_emitter = StaticEmitter(
template=Particle(
position=(0, 0, 0),
color=(0.9, 0.8, 0.8),
),
position=self.domain,
#size=[(5, 5, 5), (10, 10, 10), (15, 15, 15)],
)
self.hull = ParticleGroup(renderer=BillboardRenderer(texturizer2),
controllers=[
Lifetime(100000),
Movement(max_velocity=self.maxvel,
damping=0.98),
Magnet(self.Objective,
charge=self.agility,
exponent=0),
])
emiter = StaticEmitter(position=self.domain,
template=Particle(
color=self.color,
size=(rr, rr, rr),
))
emiter.emit(1, self.hull)
if trails:
if maxvel / r >= 20:
self.trail = ParticleGroup(
renderer=BillboardRenderer(texturizer2),
controllers=[
Lifetime(trailSize[0]),
Fader(fade_in_start=0,
fade_in_end=0.1,
fade_out_start=0,
fade_out_end=trailSize[0]),
Growth(-1 * r),
PerParticleEmitter(self.hull,
rate=trailSize[1],
template=Particle(
color=self.color,
size=(rr, rr, rr),
)),
])
self.ammo = ParticleGroup(renderer=BillboardRenderer(texturizer),
controllers=[
self.magnet,
Movement(min_velocity=0,
max_velocity=self.ammoMaxvel,
damping=self.ammoDamp),
Lifetime(self.weapon_range),
Fader(fade_out_start=self.weapon_range -
1,
fade_out_end=self.weapon_range),
])
self.weapon = PerParticleEmitter(
self.hull, # rate=self.firerate,
template=Particle(
velocity=self.velocity, # fixed value
position=(self.getPosition()),
color=partColor,
),
position=self.domain,
size=[(wbd * 0.5, wbd * 0.5, wbd * 0.5), (wbd, wbd, wbd),
(wbd * 1.5, wbd * 1.5, wbd * 1.5)],
deviation=Particle(
velocity=(self.dispersion, self.dispersion,
self.dispersion * d3),
rotation=(0, 0, math.pi / 6),
#color=(0.05,0.05,0.05,0),
))
def nearest_enemy(self,
ships,
min_dist=99999,
side=-1,
addCollectors=False,
limit=10,
lazy=lazysearch):
colla = 0
target = None
min_dist **= 2
colld = min_dist
if self.targetMode[0] == 'custom':
selfRadius = self.targetMode[1]
else:
selfRadius = self.size
selfAmmo = self.ammo
if lazy != -1:
c = lazysearch
for ship in ships:
if colla < limit and c > 0:
if ship != self and ship.side != side:
#var = math.fabs(ship.size - selfRadius) + 0.1
#var = 1 if math.fabs(ship.domain.radius - selfRadius) < selfRadius else 5
var = 0.1 if math.fabs(ship.size -
selfRadius) < selfRadius else 2
dist = distance_to_ship(self, ship, squared=True) * var
if addCollectors:
if dist < colld:
colla += 1
selfAmmo.bind_controller(*[ship.controller])
if dist < min_dist and c > 0:
c -= 1
target = ship
min_dist = dist
else:
break
else:
for ship in ships:
if colla < limit:
if ship != self and ship.side != side:
#var = math.fabs(ship.size - selfRadius) + 0.1
#var = 1 if math.fabs(ship.domain.radius - selfRadius) < selfRadius else 5
var = 0.1 if math.fabs(ship.size -
selfRadius) < selfRadius else 2
dist = distance_to_ship(self, ship, squared=True) * var
if addCollectors:
if dist < colld:
colla += 1
selfAmmo.bind_controller(*[ship.controller])
if dist < min_dist:
target = ship
min_dist = dist
else:
break
return target
def getPosition(self):
return (self.domain.center[0], self.domain.center[1],
self.domain.center[2])
def getPositionVec3(self):
return Vec3(self.domain.center[0], self.domain.center[1],
self.domain.center[2])
def contact(self, particle, group, bola):
if self.hp > 0:
#dam = math.ceil(particle.size[0] * (particle.color[0] + particle.color[1] + particle.color[2])) # color-based damage
dam = math.ceil(particle.size[0])
self.hp -= dam
self.impacto_emitter.template.position = self.getPosition()
self.impacto_emitter.template.size = (dam, dam, dam)
self.impacto_emitter.emit(1, self.impacto)
if self.hp <= 0:
self.color = (0.5, 0.1, 0.1)
self.controlable = False
self.destroy()
def destroy(self,
flag=True): # flag = True if it has been destroyed in combat
global trails
self.alive = False
for part in self.hull: # remove hull particle
self.hull.kill(part)
if flag: # if destroyed in combat, explode
s = self.size
ss = s * 10
self.impacto_emitter.template.size = (ss, ss, ss)
self.impacto_emitter.emit(int(2 * s + 1), self.deathplosion)
ships.remove(self)
battle.removeShip(self.side)
def DirToTarget(self, target):
direction = target.getPositionVec3() - self.getPositionVec3()
direction = direction.normalize()
return direction
def targetManagement(self, ships):
stimer = self.timer
if stimer[2] <= 0:
stimer[2] = random.randint(100, 500)
if self.target == None or distance_to_ship(
self, self.target, squared=True) > (
self.weapon_range * self.ammoMaxvel * 1.5)**2:
[
self.ammo.unbind_controller(i)
for i in self.ammo.controllers
if i.__class__.__name__ == "Collector"
]
self.target = self.nearest_enemy(
ships,
side=self.side,
min_dist=self.weapon_range * self.ammoMaxvel * 1.5,
addCollectors=self.multipleTargets,
lazy=False)
if self.target and self.target.alive:
self.magnet.domain = self.target.domain
if not self.multipleTargets:
self.ammo.bind_controller(*[self.target.controller
]) # NO COLLECTOR TEST
if self.commander and self.target == None:
self.mission = self.commander.getOrders(self).domain.center
if stimer[3] <= 0:
stimer[3] = 30
if self.target and not self.target.alive: # targeting a dead ship
self.target = None
stimer[2] = 10
if stimer[1] <= 0 and self.target:
for part in self.hull:
h = part
d = self.DirToTarget(self.target)
self.weapon.template.velocity = d * self.vo + h.velocity
self.weapon.emit(self.shots, self.ammo)
stimer[2] -= 1
stimer[3] -= 1
def moveAround(self, ships):
timer = self.timer
for part in self.hull:
self.domain.center = part.position
timer[1] -= 1
if timer[1] < 0:
timer[1] = self.firerate
if not self.controlable:
self.targetManagement(ships)
if timer[0] <= 0:
timer[0] = random.randint(100, 300)
agg = int(self.combatDistance)
self.counter[0] += 1
if self.counter[0] == 3:
agg += 100
self.counter[0] = 0
self.xx = random.randint(-agg, agg)
self.yy = random.randint(-agg, agg)
self.zz = random.randint(-agg, agg)
if timer[4] <= 0:
timer[4] = 2
if self.target and self.target.alive:
stdc = self.target.domain.center
self.Objective.center[0] = stdc[0] + self.xx
self.Objective.center[1] = stdc[1] + self.yy
if d3:
self.Objective.center[2] = stdc[2] + self.zz
else:
if self.commander:
pos = self.mission
else:
pos = self.domain.center
self.Objective.center[0] = pos[0] + self.xx
self.Objective.center[1] = pos[1] + self.yy
if d3:
self.Objective.center[2] = pos[2] + self.zz
timer[0] -= 1
timer[4] -= 1
electron_emitter = StaticEmitter(
template=Particle(
position=(-20, 0, 0),
size=(25, 25, 25),
color=(0.1, 0.1, 1.0),
),
velocity=Disc((0, 0, 0), (-1, 0, 0), 36, 36),
)
# Trails for electrons
trail_emitter = PerParticleEmitter(
electrons,
rate=80,
template=Particle(color=(1, 0, 0, 1), size=(4.25, 4.25, 0)),
deviation=Particle(up=(0, 0, math.pi),
rotation=(0, 0, math.pi),
size=(0.5, 0.5, 0),
velocity=(1, 1, 1),
color=(0, 1, 0),
age=trail_lifetime / 2.0),
)
trails = ParticleGroup(controllers=[
Lifetime(trail_lifetime * 1.5),
Movement(damping=0.7, max_velocity=60),
Magnet(nucleus, charge=17000.0),
Magnet(nucleus, charge=-17000.0, exponent=2.5),
Collector(Sphere((0, 0, 0), 1)),
Fader(fade_in_end=0.75,
max_alpha=0.3,
fade_out_start=0,
def test_PerParticleEmitter_source_group(self):
from lepton.emitter import PerParticleEmitter
from lepton import ParticleGroup
source_group = ParticleGroup()
emitter = PerParticleEmitter(source_group)
self.failUnless(emitter.source_group is source_group)
dust = ParticleGroup(
controllers=[
dust_emitter,
Lifetime(8),
Gravity((0, -20, 0)),
Drag(0.0, 0.10, fluid_velocity=(80, 0, 0), domain=front),
Drag(0.0, 0.10, fluid_velocity=(-80, 0, 0), domain=back),
Magnet(charge=500, domain=vortex, exponent=0.75, epsilon=0.5),
Movement(),
],
renderer=PointRenderer(16, texturizer),
)
trail_emitter = PerParticleEmitter(
dust,
rate=30,
template=Particle(color=(1, 1, 1), ),
)
trail = ParticleGroup(
controllers=[
trail_emitter,
Lifetime(.5),
Fader(max_alpha=0.09, fade_out_start=0, fade_out_end=0.5)
],
renderer=PointRenderer(14, texturizer),
)
win.set_visible(True)
win.on_resize(win.width, win.height)
pyglet.clock.schedule_interval(default_system.update, (1.0 / 60.0))
