items.py

import global_values as g
import graphics as gfx
import entities
import particles
import creatures
import levels
import actions
import sounds

import pygame as p
import random as r
import math as m

class Item:
    def __init__(self, name):
        self.name = name

        try:
            self.icon = gfx.load_image(name+"_icon")
        except:
            self.icon = None

        try:
            self.surface = gfx.load_image(name)
        except:
            self.surface = None

        self.amount = 1

    def pickup(self):
        pass

    def __getstate__(self):
        return gfx.pickle_state(self.__dict__)

    def __setstate__(self, state):
        self.__dict__ = gfx.unpickle_state(state)

class Consumable(Item):
    def __init__(self, name):
        super().__init__(name)
        
    def consume(self):
        pass

class Medkit(Consumable):
    def __init__(self):
        super().__init__("Medkit")

    def consume(self):
        super().consume()
        g.player.change_health(3)

class HealthDrink(Consumable):
    def __init__(self):
        super().__init__("Vit Drink")

    def consume(self):
        super().consume()
        actions.VarChangeAction( g.player.pipe, 6, g.player, "health", g.player.health+3, blocking=False, blockable=False, force=False, max_val=g.player.max_health)


class Ammunition(Item):
    def __init__(self, name, gun_name, ammunition_amount):
        self.gun_name = gun_name
        self.ammunition_amount = ammunition_amount
        super().__init__(name)

    def pickup(self):
        super().pickup()
        gun = g.player.inventory.check_for_named_item(self.gun_name)
        if gun:
            gun.change_ammunition(self.ammunition_amount)
        else:
            g.player.inventory.ammunition_reserves[self.gun_name] = g.player.inventory.ammunition_reserves.get(self.gun_name, 0) + self.ammunition_amount

        g.player.inventory.remove(self)

class HandgunAmmunition(Ammunition):
    def __init__(self):
        super().__init__("H. ammo", "Handgun", 10)
class ShotgunAmmunition(Ammunition):
    def __init__(self):
        super().__init__("S. ammo", "Shotgun", 5)
class RevolverAmmunition(Ammunition):
    def __init__(self):
        super().__init__("R. ammo", "Revolver", 3)
         
        

class Gun(Item):
    def __init__(self, name, damage, cooldown, fire_range, projectiles=1, spread=0, max_ammunition=50, recharge=0, stun=0, fire_effect=1):
        super().__init__(name)
        
        self.damage = damage
        self.max_cooldown = cooldown
        self.range = fire_range

        self.last_fire_time = None
        self.max_ammunition = max_ammunition
        self.ammunition = int(self.max_ammunition*0.2)
        self.recharge = recharge
        self.stun = stun
        self.fire_effect = fire_effect

        self.projectiles = projectiles
        self.spread = spread

        self.holder = None

    def attempt_fire(self):
        if self.ammunition >= 1:

            if not self.last_fire_time or (p.time.get_ticks()-self.last_fire_time)/1000 >= self.max_cooldown:
                self.fire()
                self.change_ammunition(-1)
                self.ammunition = int(self.ammunition)
                self.last_fire_time = p.time.get_ticks()
                return True
        return False

    def change_ammunition(self, amount):
        self.ammunition += amount
        if self.ammunition < 0:
            self.ammunition = 0

        if self.ammunition >= self.max_ammunition:
            self.ammunition = self.max_ammunition

    def fire(self):
        """
        Fire weapon
        """
        #I am lazy
        if self.holder is None:
            self.holder = g.player

        sounds.play_sound(self.name+"_fire", volume=0.5)
        for _ in range(self.projectiles):
            angle = self.holder.angle + ((r.random()-0.5)*self.spread)

            bullet_rect = p.Rect(self.holder.rect.centerx, self.holder.rect.centery, 1, 1)
            bullet_mask = p.mask.Mask((bullet_rect.w, bullet_rect.h), fill=True)
            bullet_mask.fill()

            bullet = entities.Entity(bullet_rect,
            self.holder.level, collision_exceptions=[self.holder], solid=False, mask=bullet_mask)
            target_x = self.holder.rect.centerx + m.cos(angle)*self.range
            target_y = self.holder.rect.centery + m.sin(angle)*self.range

            result = bullet.move_towards(target_x, target_y, self.range, detail=True)

            if result:
                if isinstance(result, creatures.Creature):
                    if bullet.y < result.rect.bottom-(result.rect.h*0.70):
                        #boom, headshot
                        damage = self.damage*1.5
                        headshot = True
                    else:
                        damage = self.damage
                        headshot = False

                    if self.damage > 0:
                        particles.create_blood(bullet.level, (bullet.x, bullet.y), self.holder.angle + m.pi ,headshot=headshot)

                    result.take_damage(damage, source=self.holder)
                    if self.stun:
                        result.stun(self.stun)

                if isinstance(result, levels.Level) and self.damage:
                                                                #level, position, angle, colour, spread, amount, power, timer
                    particles.Particles(bullet.level, (bullet.x, bullet.y), self.holder.angle + m.pi, g.colour_remaps["beige"], 1, 5, 5, 10)

            else:
                pass
            bullet.delete()

#guns
class Handgun(Gun):
    def __init__(self):
        super().__init__("Handgun", 1, 0.5, 200)

class Shotgun(Gun):
    def __init__(self):
        super().__init__("Shotgun", 0.8, 1, 100, projectiles=5, spread=0.5, max_ammunition=25, recharge=0)

class Stungun(Gun):
    def __init__(self):
        super().__init__("Stungun", 0.0, 1.5, 20, projectiles=1, spread=0, max_ammunition=3, recharge=0.4, stun=5, fire_effect=2)
        self.ammunition = self.max_ammunition

class Revolver(Gun):
    def __init__(self):
        super().__init__("Revolver", 5, 1.5, 300, projectiles=1, spread=0, max_ammunition=6)

#keys
class BathroomKey(Item):
    def __init__(self):
     super().__init__("K. Bath")
class ReactorKey(Item):
    def __init__(self):
        super().__init__("K. Reactor")
class StorageKey(Item):
    def __init__(self):
        super().__init__("K. Store")

def create_item(name):
    """
    Create item from name
    """
    # XXX be careful what you name your items (._.)
    new_item = eval(f"{name}()")
    return new_item