#include "../Platform/stdafx.h"
#include "../Headers/net.minecraft.world.entity.item.h"
#include "../Headers/net.minecraft.world.level.h"
#include "../Headers/net.minecraft.world.level.redstone.h"
#include "../Headers/net.minecraft.world.item.h"
#include "../Entities/Entity.h"
#include "WeightedPressurePlateTile.h"

WeightedPressurePlateTile::WeightedPressurePlateTile(int id,
                                                     const std::wstring& tex,
                                                     Material* material,
                                                     int maxWeight)
    : BasePressurePlateTile(id, tex, material) {
    this->maxWeight = maxWeight;

    // 4J Stu - Move this from base class to use virtual function
    updateShape(getDataForSignal(Redstone::SIGNAL_MAX));
}

int WeightedPressurePlateTile::getSignalStrength(Level* level, int x, int y,
                                                 int z) {
    int weightOfEntities =
        level->getEntitiesOfClass(typeid(Entity), getSensitiveAABB(x, y, z))
            ->size();
    int count = std::min(weightOfEntities, maxWeight);

    if (count <= 0) {
        return 0;
    } else {
        float pct = std::min(maxWeight, count) / (float)maxWeight;
        return Mth::ceil(pct * Redstone::SIGNAL_MAX);
    }
}

int WeightedPressurePlateTile::getSignalForData(int data) { return data; }

int WeightedPressurePlateTile::getDataForSignal(int signal) { return signal; }

int WeightedPressurePlateTile::getTickDelay(Level* level) {
    return SharedConstants::TICKS_PER_SECOND / 2;
}