DataParameter value set based on position [1.12.2]

[Alexiy]

I have a little mod which adds one creature (dragon). It has four variations. The variation is registered in data manager. When the creature is spawned, it's variation is set based on the biome where it spawns.

 

Parameters of the dragon:

  public static final DataParameter<Integer> dragonType = EntityDataManager.createKey(EntityDragon.class, DataSerializers.VARINT);
    private static final DataParameter<Optional<UUID>> ownerID = EntityDataManager.createKey(EntityDragon.class,DataSerializers.OPTIONAL_UNIQUE_ID);
    private Entity owner;
    public static final DataParameter<Boolean> sitting = EntityDataManager.createKey(EntityDragon.class,DataSerializers.BOOLEAN);

@Override
    protected void entityInit() {
        super.entityInit();
        setSize(0.7f,0.85f);
        dataManager.register(dragonType,0);
        dataManager.register(ownerID, Optional.absent());
        dataManager.register(sitting,false);
    }

 

Setting the dragon type:

   @SubscribeEvent
    public void setDragonType(LivingSpawnEvent.SpecialSpawn livingSpawnEvent)
    {
        EntityLivingBase entityLivingBase=livingSpawnEvent.getEntityLiving();
        if(entityLivingBase instanceof EntityDragon)
        {
            BlockPos blockPos=new BlockPos(livingSpawnEvent.getX(),livingSpawnEvent.getY(),livingSpawnEvent.getZ());
            Biome biome=livingSpawnEvent.getWorld().getBiome(blockPos);
            int type=0;
            if(biome instanceof BiomeJungle) type=EntityDragon.GREEN;
            else if(biome instanceof BiomeDesert) type=EntityDragon.GOLD;
            else if(biome instanceof BiomeForest) type=EntityDragon.EBONY;
            else if(biome instanceof BiomeHills) type=EntityDragon.SILVER;
            entityLivingBase.getDataManager().set(EntityDragon.dragonType,type);
        }
    }

 

The question is, is it possible to set the type without using the event? I couldn't find a suitable method to override that would have the dragon's position set.