cn.nukkit.level.generator.task.CallbackableChunkGenerationTask Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of powernukkitx Show documentation

Make Nukkit great again!

There is a newer version: 1.20.40-r1

package cn.nukkit.level.generator.task;

import cn.nukkit.api.PowerNukkitXOnly;
import cn.nukkit.api.Since;
import cn.nukkit.level.Level;
import cn.nukkit.level.format.generic.BaseFullChunk;
import cn.nukkit.level.generator.Generator;
import cn.nukkit.level.generator.SimpleChunkManager;
import cn.nukkit.scheduler.AsyncTask;

import java.util.function.Consumer;

@PowerNukkitXOnly
@Since("1.19.21-r2")
public class CallbackableChunkGenerationTask extends AsyncTask {

    private final Level level;
    private final T structure;
    private final Consumer callback;
    public boolean state = true;
    private BaseFullChunk chunk;

    public CallbackableChunkGenerationTask(Level level, BaseFullChunk chunk, T structure, Consumer callback) {
        this.chunk = chunk;
        this.level = level;
        this.structure = structure;
        this.callback = callback;
    }

    @Override
    public void onRun() {
        this.state = false;

        Generator generator = this.level.getGenerator();
        if (generator != null) {
            SimpleChunkManager manager = (SimpleChunkManager) generator.getChunkManager();
            if (manager != null) {
                manager.cleanChunks(this.level.getSeed());
                synchronized (manager) {
                    try {
                        BaseFullChunk chunk = this.chunk;
                        if (chunk != null) {
                            synchronized (chunk) {
                                if (!chunk.isGenerated()) {
                                    manager.setChunk(chunk.getX(), chunk.getZ(), chunk);
                                    generator.generateChunk(chunk.getX(), chunk.getZ());
                                    chunk = manager.getChunk(chunk.getX(), chunk.getZ());
                                    chunk.setGenerated();
                                }
                            }
                            this.chunk = chunk;
                            this.state = true;
                        }
                    } finally {
                        manager.cleanChunks(this.level.getSeed());
                    }
                }
            }
        }

        if (this.state && this.chunk != null) {
            this.callback.accept(this.structure);
        }
    }
}