Grasscutter/src/main/java/emu/grasscutter/data/excels/ItemData.java

package emu.grasscutter.data.excels;

import emu.grasscutter.data.GameResource;
import emu.grasscutter.data.ResourceType;
import emu.grasscutter.game.props.FightProperty;
import it.unimi.dsi.fastutil.ints.IntOpenHashSet;
import it.unimi.dsi.fastutil.ints.IntSet;

@ResourceType(name = {"MaterialExcelConfigData.json", "WeaponExcelConfigData.json", "ReliquaryExcelConfigData.json"})
public class ItemData extends GameResource {
	
	private int id;
    private int stackLimit = 1;
    private int maxUseCount;
    private int rankLevel;
    private String effectName;
    private int[] satiationParams;
    private int rank;
    private int weight;
    private int gadgetId;
    
    private int[] destroyReturnMaterial;
    private int[] destroyReturnMaterialCount;
    
    // Food
    private String foodQuality;
    private String useTarget;
    private String[] iseParam;
    
    // String enums
    private String itemType;
    private String materialType;
    private String equipType;
    private String effectType;
    private String destroyRule;
    
    // Relic
    private int mainPropDepotId;
    private int appendPropDepotId;
    private int appendPropNum;
    private int setId;
    private int[] addPropLevels;
    private int baseConvExp;
    private int maxLevel;
    
    // Weapon
    private int weaponPromoteId;
    private int weaponBaseExp;
    private int storyId;
    private int avatarPromoteId;
    private int awakenMaterial;
    private int[] awakenCosts;
    private int[] skillAffix;
    private WeaponProperty[] weaponProp;
    
    // Hash
    private String icon;
    private long nameTextMapHash;
    
    // Post load
    private transient emu.grasscutter.game.inventory.MaterialType materialEnumType;
    private transient emu.grasscutter.game.inventory.ItemType itemEnumType;
    private transient emu.grasscutter.game.inventory.EquipType equipEnumType;
    
    private IntSet addPropLevelSet;
    
    @Override
	public int getId(){
        return this.id;
    }
    
    public String getMaterialTypeString(){
        return this.materialType;
    }
    
    public int getStackLimit(){
        return this.stackLimit;
    }
    
    public int getMaxUseCount(){
        return this.maxUseCount;
    }
    
    public String getUseTarget(){
        return this.useTarget;
    }
    
    public String[] getUseParam(){
        return this.iseParam;
    }
    
    public int getRankLevel(){
        return this.rankLevel;
    }
    
    public String getFoodQuality(){
        return this.foodQuality;
    }
    
    public String getEffectName(){
        return this.effectName;
    }
    
    public int[] getSatiationParams(){
        return this.satiationParams;
    }
    
    public int[] getDestroyReturnMaterial(){
        return this.destroyReturnMaterial;
    }
    
    public int[] getDestroyReturnMaterialCount(){
        return this.destroyReturnMaterialCount;
    }
    
    public long getNameTextMapHash(){
        return this.nameTextMapHash;
    }
    
    public String getIcon(){
        return this.icon;
    }
    
    public String getItemTypeString(){
        return this.itemType;
    }
    
    public int getRank(){
        return this.rank;
    }
    
    public int getGadgetId() {
		return gadgetId;
    }
    
	public int getBaseConvExp() {
		return baseConvExp;
	}
	
	public int getMainPropDepotId() {
		return mainPropDepotId;
	}
	
	public int getAppendPropDepotId() {
		return appendPropDepotId;
	}
	
	public int getAppendPropNum() {
		return appendPropNum;
	}
	
	public int getSetId() {
		return setId;
	}
	
	public int getWeaponPromoteId() {
		return weaponPromoteId;
	}
	
	public int getWeaponBaseExp() {
		return weaponBaseExp;
	}
	
	public int getAwakenMaterial() {
        	return awakenMaterial;
    	}
	
	public int[] getAwakenCosts() {
		return awakenCosts;
	}
	
	public IntSet getAddPropLevelSet() {
		return addPropLevelSet;
	}
	
	public int[] getSkillAffix() {
		return skillAffix;
	}
	
	public WeaponProperty[] getWeaponProperties() {
		return weaponProp;
	}
	
	public int getMaxLevel() {
		return maxLevel;
	}
	
	public emu.grasscutter.game.inventory.ItemType getItemType() {
    	return this.itemEnumType;
    }
    
    public emu.grasscutter.game.inventory.MaterialType getMaterialType() {
    	return this.materialEnumType;
    }
    
    public emu.grasscutter.game.inventory.EquipType getEquipType() {
    	return this.equipEnumType;
    }
    
    public boolean canAddRelicProp(int level) {
    	return this.addPropLevelSet != null & this.addPropLevelSet.contains(level);
    }
    
	public boolean isEquip() {
		return this.itemEnumType == emu.grasscutter.game.inventory.ItemType.ITEM_RELIQUARY || this.itemEnumType == emu.grasscutter.game.inventory.ItemType.ITEM_WEAPON;
	}
    
    @Override
	public void onLoad() {
    	this.itemEnumType = emu.grasscutter.game.inventory.ItemType.getTypeByName(getItemTypeString());
    	this.materialEnumType = emu.grasscutter.game.inventory.MaterialType.getTypeByName(getMaterialTypeString());

		if (this.itemEnumType == emu.grasscutter.game.inventory.ItemType.ITEM_RELIQUARY) {
			this.equipEnumType = emu.grasscutter.game.inventory.EquipType.getTypeByName(this.equipType);
			if (this.addPropLevels != null && this.addPropLevels.length > 0) {
				this.addPropLevelSet = new IntOpenHashSet(this.addPropLevels);
			}
		} else if (this.itemEnumType == emu.grasscutter.game.inventory.ItemType.ITEM_WEAPON) {
			this.equipEnumType = emu.grasscutter.game.inventory.EquipType.EQUIP_WEAPON;
		} else {
			this.equipEnumType = emu.grasscutter.game.inventory.EquipType.EQUIP_NONE;
		}
		
		if (this.getWeaponProperties() != null) {
			for (WeaponProperty weaponProperty : this.getWeaponProperties()) {
				weaponProperty.onLoad();
			}
		}
    }
    
    public static class WeaponProperty {
    	private FightProperty fightProp;
        private String propType;
        private float initValue;
        private String type;

        public String getPropType(){
            return this.propType;
        }
        
        public float getInitValue(){
            return this.initValue;
        }
        
        public String getType(){
            return this.type;
        }

		public FightProperty getFightProp() {
			return fightProp;
		}

		public void onLoad() {
			this.fightProp = FightProperty.getPropByName(propType);
		}
        
    }
}