lootrandomizer.h

Go to the documentation of this file.

/*
 * lootrandomizer.h by Stefano Angeleri
 *
 * Copyright (C) 2010 Atomic Blue ([email protected], http://www.atomicblue.org)
 *
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation (version 2 of the License)
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#ifndef __LOOTRANDOMIZER_H__
#define __LOOTRANDOMIZER_H__

class RandomizedOverlay;

struct LootModifier
{
    uint32_t id;                       
    uint32_t mod_id;                   
    csString modifier_type;            
    csString name;                     
    csString effect;                   
    csString equip_script;             
    float    probability;              
    float    probabilityRange;         
    csString stat_req_modifier;        
    float    cost_modifier;            
    csString mesh;                     
    csString icon;                     
    csString not_usable_with;          
    csHash<bool, uint32_t> itemRestrain; 

    // Return true if this modifier is allowed with the given item stats ID.
    bool IsAllowed(uint32_t itemID);
};

struct ValueModifier
{
    csString name;  
    csString op;    
    float value;    
};

class MathScript;
class LootRandomizer
{
protected:
    csArray<LootModifier*> prefixes;   
    csArray<LootModifier*> suffixes;   
    csArray<LootModifier*> adjectives; 
    csHash<LootModifier*, uint32_t> LootModifiersById;  

    // precalculated max values for probability. min is always 0
    float prefix_max;
    float adjective_max;
    float suffix_max;

public:
    LootRandomizer(CacheManager* cachemanager);
    ~LootRandomizer();

    void AddLootModifier(LootModifier* entry);

    LootModifier* GetModifier(uint32_t id);
 
    bool GetModifiers(uint32_t itemID, csArray<psGMSpawnMods::ItemModifier>& mods);

    psItem* SetModifiers(psItem* item, csArray<uint32_t>& mods);

    psItem* RandomizeItem(psItem* item,
                          float cost,
                          bool lootTesting = false,
                          size_t numModifiers = 0);

    float CalcModifierCostCap(psCharacter* chr);

    void ApplyModifier(psItemStats* baseItem, RandomizedOverlay* overlay, csArray<uint32_t> &modifiersIds);

    float GetModifierPercentProbability(int modifierID, int modifierType);

protected:
    MathScript* modifierCostCalc; 
    CacheManager* cacheManager; 

private:
    void AddModifier(LootModifier* oper1, LootModifier* oper2);

    bool SetAttribute(const csString &op, const csString &attrName, float modifier, RandomizedOverlay* overlay, psItemStats* baseItem, csArray<ValueModifier> &values);

    void SetAttributeApplyOP(float* value[], float modifier, size_t amount, const csString &op);

    csString GenerateScriptXML(csString &name, csString &equip_script, csArray<ValueModifier> &values);
};

#endif