[Forge 1.10.2] How to get the weight modifier of the enchantment?

[lethinh]

Could you tell me how to get the weight modifier of the enchantment? I  have tried this code but this only for 1.7.10. And I don't how it should work in 1.10.2:

 

Spoiler

public int getWeightModifier(Enchantment ench) {
        switch (ench.getRarity()) {
        case 1:
            return 8;
        case 2:
            return 4;
        case 3:
        case 4:
        case 5:
            return 2;
        }
        return 1;
    }