Mindustry/core/src/mindustry/entities/def/UnitComp.java

package mindustry.entities.def;

import arc.*;
import arc.math.*;
import arc.math.geom.*;
import arc.util.*;
import mindustry.annotations.Annotations.*;
import mindustry.content.*;
import mindustry.entities.*;
import mindustry.entities.units.*;
import mindustry.game.EventType.*;
import mindustry.gen.*;
import mindustry.graphics.*;
import mindustry.type.*;
import mindustry.world.*;
import mindustry.world.blocks.*;

import static mindustry.Vars.*;

@Component
abstract class UnitComp implements Healthc, Velc, Statusc, Teamc, Itemsc, Hitboxc, Rotc, Massc, Unitc, Weaponsc, Drawc, Boundedc,
        DrawLayerGroundc, DrawLayerFlyingc, DrawLayerGroundShadowsc, DrawLayerFlyingShadowsc, Syncc{
    @Import float x, y, rotation, elevation;

    private UnitController controller;
    private UnitType type;

    @Override
    public float clipSize(){
        return type.region.getWidth() * 2f;
    }

    @Override
    public int itemCapacity(){
        return type.itemCapacity;
    }

    @Override
    public float bounds(){
        return hitSize() *  2f;
    }

    @Override
    public void controller(UnitController controller){
        this.controller = controller;
        if(controller.unit() != this) controller.unit(this);
    }

    @Override
    public UnitController controller(){
        return controller;
    }

    @Override
    public void set(UnitType def, UnitController controller){
        type(type);
        controller(controller);
    }

    @Override
    public void collision(Hitboxc other, float x, float y){
        if(!(other instanceof Unitc)) return;
        Unitc unit = (Unitc)other;

        if(isGrounded() != unit.isGrounded()) return;

        float scale = 2f;
        hitbox(Tmp.r1);
        other.hitbox(Tmp.r2);
        Vec2 v = Geometry.overlap(Tmp.r1, Tmp.r2, true);
        float tm = mass() + unit.mass();
        float s1 = mass() / tm, s2 = unit.mass() / tm;
        move(v.x*s2/scale, v.y*s2/scale);
        unit.move(-v.x*s1/scale, -v.y*s1/scale);
    }

    @Override
    public void type(UnitType type){
        this.type = type;
        maxHealth(type.health);
        heal();
        drag(type.drag);
        hitSize(type.hitsize);
        controller(type.createController());
        setupWeapons(type);

        elevation = type.flying ? 1f : 0f;
    }

    @Override
    public UnitType type(){
        return type;
    }

    public void lookAt(float angle){
        rotation = Angles.moveToward(rotation, angle, type.rotateSpeed);
    }

    public void lookAt(Position pos){
        lookAt(angleTo(pos));
    }

    @Override
    public void afterRead(){
        //set up type info after reading
        type(this.type);
    }

    @Override
    public void update(){
        drag(type.drag * (isGrounded() ? (floorOn().dragMultiplier) : 1f));

        //apply knockback based on spawns
        //TODO move elsewhere
        if(team() != state.rules.waveTeam){
            float relativeSize = state.rules.dropZoneRadius + bounds()/2f + 1f;
            for(Tile spawn : spawner.getGroundSpawns()){
                if(withinDst(spawn.worldx(), spawn.worldy(), relativeSize)){
                    vel().add(Tmp.v1.set(this).sub(spawn.worldx(), spawn.worldy()).setLength(0.1f + 1f - dst(spawn) / relativeSize).scl(0.45f * Time.delta()));
                }
            }
        }

        Tile tile = tileOn();
        Floor floor = floorOn();

        if(tile != null){
            //unit block update
            tile.block().unitOn(tile, this);

            //apply damage
            if(floor.damageTaken > 0f){
                damageContinuous(floor.damageTaken);
            }
        }

        controller.update();
    }

    @Override
    public boolean isImmune(StatusEffect effect){
        return type.immunities.contains(effect);
    }

    @Override
    public void draw(){
        type.drawBody(this);
        type.drawWeapons(this);
        if(type.drawCell) type.drawCell(this);
        if(type.drawItems) type.drawItems(this);
        type.drawLight(this);
    }

    @Override
    public void drawFlyingShadows(){
        if(isFlying()) type.drawShadow(this);
    }

    @Override
    public void drawGroundShadows(){
        type.drawOcclusion(this);
    }

    @Override
    public void drawFlying(){
        if(isFlying()) draw();
    }

    @Override
    public void drawGround(){
        if(isGrounded()) draw();
    }

    @Override
    public boolean isPlayer(){
        return controller instanceof Playerc;
    }

    @Override
    public void killed(){
        float explosiveness = 2f + item().explosiveness * stack().amount;
        float flammability = item().flammability * stack().amount;
        Damage.dynamicExplosion(x, y, flammability, explosiveness, 0f, bounds() / 2f, Pal.darkFlame);

        Effects.scorch(x, y, (int)(hitSize() / 5));
        Fx.explosion.at(this);
        Effects.shake(2f, 2f, this);
        type.deathSound.at(this);

        Events.fire(new UnitDestroyEvent(this));

        if(explosiveness > 7f && isLocal()){
            Events.fire(Trigger.suicideBomb);
        }
    }

    @Override
    public boolean canMine(Item item){
        return type.drillTier >= item.hardness;
    }

    @Override
    public float miningSpeed(){
        return type.mineSpeed;
    }

    @Override
    public boolean offloadImmediately(){
        return false;
    }
}