bleedingsourcesmanagerbase.c

Go to the documentation of this file.

class BleedingSourcesManagerBase
{
    ref map<int, ref BleedingSource> m_BleedingSources = new map<int, ref BleedingSource>;
    ref map<string, ref BleedingSourceZone> m_BleedingSourceZone = new map<string, ref BleedingSourceZone>;
    ref map<int, int> m_BleedingSourcesByLocation = new map<int, int>;
    static ref map<int, int> m_BleedingSourcesZonesMaskByLocation = new map<int, int>;//for each inventory location, keep a bitmask where a bit is set to 1 for each bleeding source zone that belongs to that location
    static ref set<int> m_BleedingSourcesLocationsList = new set<int>;
    ItemBase m_Item;//item used to remove the bleeding source
    PlayerBase m_Player;
    //ref map<string, int> m_FireGeomToBit = new map<string, int>;
    ref map<int, string> m_BitToFireGeom = new map<int, string>;
    protected int m_Bit;
    int m_BitOffset = 0;
    
    void BleedingSourcesManagerBase( PlayerBase player )
    {
        m_Player = player;
        Init();
    }
    
    protected void Init()
    {
        //dmgZone_head
        RegisterBleedingZoneEx("Head",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 0 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_HIGH, "BleedingSourceEffect",InventorySlots.MASK);
        //dmgZone_torso
        RegisterBleedingZoneEx("Neck", PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL,"", "-180 0 0" , "0.02 -0.05 0.05", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_HIGH, "BleedingSourceEffect",InventorySlots.HEADGEAR);
        RegisterBleedingZoneEx("Pelvis",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -135 0" , "0 0.12 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_HIGH, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("Spine",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 55 0" , "0 -0.1 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_HIGH, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("Spine1",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 155 0" , "0 -0.1 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_HIGH, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("Spine2",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 55 0" , "0 -0.07 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_HIGH, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("Spine3",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 155 0" , "0 -0.05 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_HIGH, "BleedingSourceEffect",InventorySlots.BODY);
        //dmgZone_leftArm
        RegisterBleedingZoneEx("LeftShoulder",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -90 0" , "0 0.07 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect", InventorySlots.BODY);
        RegisterBleedingZoneEx("LeftArm", PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "","0 90 90" , "0 -0.05 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("LeftArmRoll",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 -0.05 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("LeftForeArm",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 -0.05 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.BODY);
        //dmgZone_rightArm
        RegisterBleedingZoneEx("RightShoulder",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 -0.07 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("RightArm", PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL,"", "0 -90 0" , "0 0.05 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("RightArmRoll",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -90 0" , "0 0.05 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.BODY);
        RegisterBleedingZoneEx("RightForeArm",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -90 0" , "0 0.05 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.BODY);
        //dmgZone_leftHand
        RegisterBleedingZoneEx("LeftForeArmRoll",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0.1 0 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_LOW, "BleedingSourceEffectLight", InventorySlots.GLOVES);
        //dmgZone_rightHand
        RegisterBleedingZoneEx("RightForeArmRoll",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "-0.1 0 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_LOW, "BleedingSourceEffectLight", InventorySlots.GLOVES);
        //dmgZone_leftLeg
        RegisterBleedingZoneEx("LeftLeg",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 -0.07 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.LEGS);
        RegisterBleedingZoneEx("LeftLegRoll",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 -0.07 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.LEGS);
        RegisterBleedingZoneEx("LeftUpLeg",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 -0.12 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.LEGS);
        RegisterBleedingZoneEx("LeftUpLegRoll",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 -0.07 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM , "BleedingSourceEffect",InventorySlots.LEGS);
        //dmgZone_rightLeg
        RegisterBleedingZoneEx("RightLeg",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -90 0" , "0 0.06 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.LEGS);
        RegisterBleedingZoneEx("RightLegRoll",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -90 0" , "0 0 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.LEGS);
        RegisterBleedingZoneEx("RightUpLeg",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -90 0" , "0 0.12 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.LEGS);
        RegisterBleedingZoneEx("RightUpLegRoll",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -90 0" , "0 0.06 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_MEDIUM, "BleedingSourceEffect",InventorySlots.LEGS);
        //dmgZone_leftFoot
        RegisterBleedingZoneEx("LeftFoot",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 180 0" , "0 0 0.035", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_LOW, "BleedingSourceEffectLight", InventorySlots.LEGS);
        RegisterBleedingZoneEx("LeftToeBase",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 -90 0" , "0 0.07 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_LOW, "BleedingSourceEffect",InventorySlots.FEET);
        //dmgZone_rightFoot
        RegisterBleedingZoneEx("RightFoot",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 0 0" , "0 0 -0.035", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_LOW, "BleedingSourceEffectLight", InventorySlots.LEGS);
        RegisterBleedingZoneEx("RightToeBase",PlayerConstants.BLEEDING_SOURCE_DURATION_NORMAL, "", "0 90 0" , "0 -0.07 0", PlayerConstants.BLEEDING_SOURCE_FLOW_MODIFIER_LOW, "BleedingSourceEffect", InventorySlots.FEET);
    }
    
    protected void SetItem(ItemBase item)
    {
        m_Item = item;
    }
    
    
    set<int> GetBleedingSourcesLocations()
    {
        return m_BleedingSourcesLocationsList;
    }
    
    
    //for inventory location, get the active bits
    int GetBleedingSourceBitsByInvLocation(int inv_location)
    {
        if (m_BleedingSourcesByLocation.Contains(inv_location))
        {
            return m_BleedingSourcesByLocation.Get(inv_location);
        }
 
        return 0;
    }
 
    //for inventory location, get the active bits
    int GetFreeBleedingSourceBitsByInvLocation(int inv_location)
    {
        if (m_BleedingSourcesZonesMaskByLocation.Contains(inv_location))
        {
            return (~m_BleedingSourcesZonesMaskByLocation.Get(inv_location) & GetBleedingSourceBitsByInvLocation(inv_location)) | (m_BleedingSourcesZonesMaskByLocation.Get(inv_location) & ~GetBleedingSourceBitsByInvLocation(inv_location));//xor
        }
 
        return 0;
    }
    
    bool IsBleedingSourceActive(string name)
    {
        name.ToLower();
        int bit = GetBitFromSelectionName(name);
 
        return m_Player.GetBleedingBits() & bit;
    }
    
    
    protected int GetBitFromSelectionID(int id)
    {
        CachedObjectsArrays.ARRAY_STRING.Clear();
        m_Player.GetActionComponentNameList(id, CachedObjectsArrays.ARRAY_STRING, "fire");
        
        for (int i = 0; i < CachedObjectsArrays.ARRAY_STRING.Count(); i++)
        {
            /*
            string name = CachedObjectsArrays.ARRAY_STRING.Get(i);
            PrintString(name);
            */
            int bit = GetBitFromSelectionName(CachedObjectsArrays.ARRAY_STRING.Get(i));
            if (  bit !=0 )
            {
                return bit;
            }
        }
        return 0;
    }
    
    protected BleedingSourceZone GetBleedingSourceMeta(int bit)
    {
        return m_BleedingSourceZone.Get(GetSelectionNameFromBit(bit));
    }
    
    int GetRegisteredSourcesCount()
    {
        return m_BitOffset;
    }
    
    protected void RegisterBleedingZoneEx(string name, int max_time, string bone = "", vector orientation = "0 0 0", vector offset = "0 0 0", float flow_modifier = 1, string particle_name = "BleedingSourceEffect", int inv_location = 0)
    {
        if ( m_BitOffset == BIT_INT_SIZE)
        {
            Error("Too many bleeding sources, max is "+BIT_INT_SIZE.ToString());
        }
        else
        {
            name.ToLower();
            //PrintString(name);
            //int bit = Math.Pow(2, m_BitOffset);
            m_Bit = 1 << m_BitOffset;
            //PrintString(bit.ToString());
            string bone_name = bone;
            
            if (bone_name == "")
            {
                bone_name = name;
            }
            
            m_BleedingSourceZone.Insert(name, new BleedingSourceZone(name, m_Bit, offset, orientation, bone_name, max_time, flow_modifier, particle_name));
            m_BleedingSourceZone.Get(name).SetInvLocation(inv_location);
            m_BitToFireGeom.Insert(m_Bit, name);
            m_BleedingSourcesZonesMaskByLocation.Set( inv_location, m_BleedingSourcesZonesMaskByLocation.Get(inv_location) | m_Bit);//set a bit to 1 to already existing bitmask for that location
            m_BleedingSourcesLocationsList.Insert(inv_location);
            m_BitOffset++;
        }
    }
    protected void RegisterBleedingZone(string name, int max_time, string bone = "", vector orientation = "0 0 0", vector offset = "0 0 0", float flow_modifier = 1, string particle_name = "BleedingSourceEffect")
    {
        RegisterBleedingZoneEx(name, max_time, bone, orientation, offset, flow_modifier, particle_name);
    }
    
    void RemoveAllSources()
    {
        //int count = m_BleedingSources.Count();
        while (m_BleedingSources.Count() > 0 )
        {
            int bit = m_BleedingSources.GetKey(0);
            RemoveBleedingSource(bit);
        }
    }
    
    protected int GetBitFromSelectionName(string name)
    {
        if (!m_BleedingSourceZone.Get(name))
        {
            return 0;
        }
        return m_BleedingSourceZone.Get(name).GetBit();
    }   
    
    protected string GetSelectionNameFromBit(int bit)
    {
        return m_BitToFireGeom.Get(bit);
    }
    
    bool AttemptAddBleedingSource(int component)
    {
        int bit = GetBitFromSelectionID(component);
        
        if ( bit == 0 )
        {
            return false;
        }
        
        if ( CanAddBleedingSource(bit) )
        {
            AddBleedingSource(bit);
            return true;
        }
        return false;
    }
    
    bool AttemptAddBleedingSourceBySelection(string selection_name)
    {
        selection_name.ToLower();
        int bit = GetBitFromSelectionName(selection_name);
        
        if ( bit == 0 )
        {
            return false;
        }
        
        if ( CanAddBleedingSource(bit) )
        {
            AddBleedingSource(bit);
            return true;
        }
        return false;
    }
    
    
    bool AttemptAddBleedingSourceDirectly(int bit, eBleedingSourceType type = eBleedingSourceType.NORMAL, int context = 0)
    {
        if(CanAddBleedingSource(bit))
        {
            AddBleedingSourceEx(bit, type, context);
            return true;
        }  
        return false;
    }
    
    
    protected bool CanAddBleedingSource(int bit)
    {
        if (!GetBleedingSourceMeta(bit)) 
        {
            return false;
        }
        if ((m_Player.GetBleedingBits() & bit) == 0)
        {
            return true;
        }
        return false;
    }
    
    protected void AddBleedingSourceEx(int bit, eBleedingSourceType type = eBleedingSourceType.NORMAL, int context = 0)
    {
        AddBleedingSource(bit);
        m_BleedingSources.Get(bit).SetType(type);
    }
    
    protected void AddBleedingSource(int bit)
    {
        BleedingSourceZone bsz = GetBleedingSourceMeta(bit);
        vector orientation = bsz.GetOrientation();
        vector offset = bsz.GetOffset();
        string bone_name =  bsz.GetBoneName();
        float flow_modifier = bsz.GetFlowModifier();
        int max_time = bsz.GetMaxTime();
        string particle_name = bsz.GetParticleName();
        int inventory_loc = bsz.GetInvLocation();
        m_BleedingSourcesByLocation.Set(inventory_loc, (m_BleedingSourcesByLocation.Get(inventory_loc) | bit));
        m_BleedingSources.Insert(bit, new BleedingSource(m_Player, bit,bone_name, orientation, offset, max_time, flow_modifier, particle_name) );
    }
    
    int GetBleedingSourceActiveTime(int bit)
    {
        int time = -1;
        if (m_BleedingSources.Contains(bit))
        {
            time = m_BleedingSources.Get(bit).GetActiveTime();
        }
        return time;
    }
    
    void SetBleedingSourceActiveTime(int bit, int time)
    {
        if (m_BleedingSources.Contains(bit))
        {
            m_BleedingSources.Get(bit).SetActiveTime(time);
        }
    }
    
    void SetBleedingSourceType(int bit, eBleedingSourceType type)
    {
        if (m_BleedingSources.Contains(bit))
        {
            m_BleedingSources.Get(bit).SetType(type);
        }
    }   
    eBleedingSourceType GetBleedingSourceType(int bit)
    {
        if (m_BleedingSources.Contains(bit))
        {
            return m_BleedingSources.Get(bit).GetType();
        }
        return -1;
    }
    
    protected bool RemoveBleedingSource(int bit)
    {
        if (m_BleedingSources.Contains(bit))
        {
            BleedingSourceZone bsz = GetBleedingSourceMeta(bit);
            int inventory_loc = bsz.GetInvLocation();
            m_BleedingSourcesByLocation.Set(inventory_loc, (m_BleedingSourcesByLocation.Get(inventory_loc) & ~bit));//deactivate the bit
            m_BleedingSources.Remove(bit);
            return true;
        }
        return false;
    }
 
    int GetBleedingSourcesCount()
    {
        return m_BleedingSources.Count();
    }
    
    /*void ChangeBleedingIndicatorVisibility(bool visible)
    {
        int count = m_BleedingSources.Count();
        for (int i = 0; i < count; i++)
        {
            m_BleedingSources.GetElement(i).ToggleSourceBleedingIndication(visible);
        }
    }*/
}