4jcraft/targets/minecraft/world/item/enchantment/ProtectionEnchantment.h

#pragma once

#include <memory>

#include "Enchantment.h"

class Entity;

class ProtectionEnchantment : public Enchantment {
public:
    static const int ALL = 0;
    static const int FIRE = 1;
    static const int FALL = 2;
    static const int EXPLOSION = 3;
    static const int PROJECTILE = 4;

private:
    static const int names[];
    static const int minCost[];
    static const int levelCost[];
    static const int levelCostSpan[];

public:
    const int type;

    ProtectionEnchantment(int id, int frequency, int type);

    virtual int getMinCost(int level);
    virtual int getMaxCost(int level);
    virtual int getMaxLevel();
    virtual int getDamageProtection(int level, DamageSource* source);
    virtual int getDescriptionId();
    virtual bool isCompatibleWith(Enchantment* other) const;
    static int getFireAfterDampener(std::shared_ptr<Entity> entity, int time);
    static double getExplosionKnockbackAfterDampener(
        std::shared_ptr<Entity> entity, double power);
};