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 __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))