Pibd-23_Lisov_N.A._AirFighter._Hard/Classes/EntityWithEngines.java

package Classes;

import java.util.ArrayList;
import java.util.Random;

public class EntityWithEngines<T extends EntityAircraft,U extends IDrawingEngines>
{
    Random rnd = new Random();
    public ArrayList<T> entities;
    public ArrayList<U> engines;

    public EntityWithEngines()
    {
        entities = new ArrayList<T>();
        engines = new ArrayList<U>();
    }

    public void addEntity(T aircraft)
    {
        entities.add(aircraft);
    }

    public void addEngines(U engine)
    {
        engines.add(engine);
    }

    public DrawingObjectAircraft createObject()
    {
        if(entities.size() == 0 || engines.size() == 0)
        {
            return null;
        }
        EntityAircraft entity = (EntityAircraft) entities.get(rnd.nextInt(0, entities.size()));
        IDrawingEngines engine = (IDrawingEngines) engines.get(rnd.nextInt(0,engines.size()));

        if(entity instanceof EntityMilitaryAircraft modAircraft)
        {
            return new DrawingObjectAircraft(new DrawingMilitaryAircraft(modAircraft,engine));
        }
        return new DrawingObjectAircraft(new DrawingAircraft(entity,engine));

    }

}