#include maps\mp\zombies\_zm_riotshield;
#include maps\mp\_utility;
#include maps\mp\zombies\_zm_utility;
#include common_scripts\utility;
#include maps\mp\zombies\_zm_weap_riotshield;
#include maps\mp\zombies\_zm_buildables;
#include maps\mp\zombies\_zm_equipment;
#include maps\mp\zombies\_zm_weapons;

doriotshielddeploy( origin, angles )
{
    self endon( "death" );
    self endon( "disconnect" );
    self endon( "start_riotshield_deploy" );
    self notify( "deployed_riotshield" );
    self maps\mp\zombies\_zm_buildables::track_placed_buildables( level.riotshield_name );

    if ( isdefined( self.current_equipment ) && self.current_equipment == level.riotshield_name )
        self maps\mp\zombies\_zm_equipment::equipment_to_deployed( level.riotshield_name );

    zoffset = level.riotshield_placement_zoffset;
    shield_ent = self spawnriotshieldcover( origin + ( 0, 0, zoffset ), angles );
    item_ent = deployriotshield( self, shield_ent );
    primaries = self getweaponslistprimaries();

    self maps\mp\zombies\_zm_weapons::switch_back_primary_weapon(self.riotshield_prev_wep);

    if ( isdefined( level.equipment_planted ) )
        self [[ level.equipment_planted ]]( shield_ent, level.riotshield_name, self );

    if ( isdefined( level.equipment_safe_to_drop ) )
    {
        if ( !self [[ level.equipment_safe_to_drop ]]( shield_ent ) )
        {
            self notify( "destroy_riotshield" );
            shield_ent delete();
            item_ent delete();
            return;
        }
    }

    self.riotshieldretrievetrigger = item_ent;
    self.riotshieldentity = shield_ent;
    self thread watchdeployedriotshieldents();
    self thread deleteshieldondamage( self.riotshieldentity );
    self thread deleteshieldmodelonweaponpickup( self.riotshieldretrievetrigger );
    self thread deleteriotshieldonplayerdeath();
    self thread watchshieldtriggervisibility( self.riotshieldretrievetrigger );
    self.riotshieldentity thread watchdeployedriotshielddamage();
    return shield_ent;
}

trackriotshield()
{
    self endon( "death" );
    self endon( "disconnect" );
    self.hasriotshield = self hasweapon( level.riotshield_name );
    self.hasriotshieldequipped = self getcurrentweapon() == level.riotshield_name;
    self.shield_placement = 0;

    if ( self.hasriotshield )
    {
        if ( self.hasriotshieldequipped )
        {
            self.shield_placement = 1;
            self updateriotshieldmodel();
        }
        else
        {
            self.shield_placement = 2;
            self updateriotshieldmodel();
        }
    }

    for (;;)
    {
        self waittill( "weapon_change", newweapon );

        foreach (wep in self getWeaponsListPrimaries())
        {
            if (wep == newweapon)
            {
                self.riotshield_prev_wep = newweapon;
                break;
            }
        }

        if ( newweapon == level.riotshield_name )
        {
            if ( self.hasriotshieldequipped )
                continue;

            if ( isdefined( self.riotshieldentity ) )
                self notify( "destroy_riotshield" );

            self.shield_placement = 1;
            self updateriotshieldmodel();

            self.hasriotshield = 1;
            self.hasriotshieldequipped = 1;
            continue;
        }

        if ( self ismantling() && newweapon == "none" )
            continue;

        if ( self.hasriotshieldequipped )
        {
            assert( self.hasriotshield );
            self.hasriotshield = self hasweapon( level.riotshield_name );

            if ( isdefined( self.riotshield_hidden ) && self.riotshield_hidden )
            {

            }
            else if ( self.hasriotshield )
                self.shield_placement = 2;
            else if ( isdefined( self.shield_ent ) )
                assert( self.shield_placement == 3 );
            else
                self.shield_placement = 0;

            self updateriotshieldmodel();
            self.hasriotshieldequipped = 0;
            continue;
        }

        if ( self.hasriotshield )
        {
            if ( !self hasweapon( level.riotshield_name ) )
            {
                self.shield_placement = 0;
                self updateriotshieldmodel();
                self.hasriotshield = 0;
            }

            continue;
        }

        if ( self hasweapon( level.riotshield_name ) )
        {
            self.shield_placement = 2;
            self updateriotshieldmodel();
            self.hasriotshield = 1;
        }
    }
}