mirror of
https://github.com/plexusorg/Plex-FAWE.git
synced 2024-06-13 06:11:46 +00:00
443 lines
17 KiB
Java
443 lines
17 KiB
Java
/*
|
|
* WorldEdit, a Minecraft world manipulation toolkit
|
|
* Copyright (C) sk89q <http://www.sk89q.com>
|
|
* Copyright (C) WorldEdit team and contributors
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify it
|
|
* under the terms of the GNU Lesser General Public License as published by the
|
|
* Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
|
|
* for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package com.sk89q.worldedit.forge;
|
|
|
|
import com.sk89q.jnbt.CompoundTag;
|
|
import com.sk89q.worldedit.EditSession;
|
|
import com.sk89q.worldedit.MaxChangedBlocksException;
|
|
import com.sk89q.worldedit.Vector;
|
|
import com.sk89q.worldedit.Vector2D;
|
|
import com.sk89q.worldedit.WorldEditException;
|
|
import com.sk89q.worldedit.blocks.BaseBlock;
|
|
import com.sk89q.worldedit.blocks.BaseItemStack;
|
|
import com.sk89q.worldedit.blocks.LazyBlock;
|
|
import com.sk89q.worldedit.entity.BaseEntity;
|
|
import com.sk89q.worldedit.entity.Entity;
|
|
import com.sk89q.worldedit.internal.Constants;
|
|
import com.sk89q.worldedit.regions.Region;
|
|
import com.sk89q.worldedit.util.Location;
|
|
import com.sk89q.worldedit.util.TreeGenerator.TreeType;
|
|
import com.sk89q.worldedit.world.AbstractWorld;
|
|
import com.sk89q.worldedit.world.biome.BaseBiome;
|
|
import com.sk89q.worldedit.world.registry.WorldData;
|
|
|
|
import java.lang.ref.WeakReference;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Random;
|
|
import java.util.Set;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
import javax.annotation.Nullable;
|
|
import net.minecraft.block.Block;
|
|
import net.minecraft.block.BlockLeaves;
|
|
import net.minecraft.block.BlockOldLeaf;
|
|
import net.minecraft.block.BlockOldLog;
|
|
import net.minecraft.block.BlockPlanks;
|
|
import net.minecraft.block.state.IBlockState;
|
|
import net.minecraft.entity.EntityList;
|
|
import net.minecraft.entity.item.EntityItem;
|
|
import net.minecraft.init.Blocks;
|
|
import net.minecraft.inventory.IInventory;
|
|
import net.minecraft.nbt.NBTTagCompound;
|
|
import net.minecraft.tileentity.TileEntity;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.world.ChunkCoordIntPair;
|
|
import net.minecraft.world.World;
|
|
import net.minecraft.world.biome.BiomeGenBase;
|
|
import net.minecraft.world.chunk.Chunk;
|
|
import net.minecraft.world.chunk.IChunkProvider;
|
|
import net.minecraft.world.gen.ChunkProviderServer;
|
|
import net.minecraft.world.gen.feature.WorldGenBigMushroom;
|
|
import net.minecraft.world.gen.feature.WorldGenBigTree;
|
|
import net.minecraft.world.gen.feature.WorldGenBirchTree;
|
|
import net.minecraft.world.gen.feature.WorldGenCanopyTree;
|
|
import net.minecraft.world.gen.feature.WorldGenMegaJungle;
|
|
import net.minecraft.world.gen.feature.WorldGenMegaPineTree;
|
|
import net.minecraft.world.gen.feature.WorldGenSavannaTree;
|
|
import net.minecraft.world.gen.feature.WorldGenShrub;
|
|
import net.minecraft.world.gen.feature.WorldGenSwamp;
|
|
import net.minecraft.world.gen.feature.WorldGenTaiga1;
|
|
import net.minecraft.world.gen.feature.WorldGenTaiga2;
|
|
import net.minecraft.world.gen.feature.WorldGenTrees;
|
|
import net.minecraft.world.gen.feature.WorldGenerator;
|
|
|
|
import static com.google.common.base.Preconditions.checkNotNull;
|
|
|
|
/**
|
|
* An adapter to Minecraft worlds for WorldEdit.
|
|
*/
|
|
public class ForgeWorld extends AbstractWorld {
|
|
|
|
private static final Random random = new Random();
|
|
private static final int UPDATE = 1, NOTIFY = 2, NOTIFY_CLIENT = 4;
|
|
private static final Logger logger = Logger.getLogger(ForgeWorld.class.getCanonicalName());
|
|
|
|
private static final IBlockState JUNGLE_LOG = Blocks.LOG.getDefaultState().withProperty(BlockOldLog.VARIANT, BlockPlanks.EnumType.JUNGLE);
|
|
private static final IBlockState JUNGLE_LEAF = Blocks.LEAVES.getDefaultState().withProperty(BlockOldLeaf.VARIANT, BlockPlanks.EnumType.JUNGLE).withProperty(BlockLeaves.CHECK_DECAY, Boolean.valueOf(false));
|
|
private static final IBlockState JUNGLE_SHRUB = Blocks.LEAVES.getDefaultState().withProperty(BlockOldLeaf.VARIANT, BlockPlanks.EnumType.OAK).withProperty(BlockLeaves.CHECK_DECAY, Boolean.valueOf(false));
|
|
|
|
private final WeakReference<World> worldRef;
|
|
|
|
/**
|
|
* Construct a new world.
|
|
*
|
|
* @param world the world
|
|
*/
|
|
ForgeWorld(World world) {
|
|
checkNotNull(world);
|
|
this.worldRef = new WeakReference<World>(world);
|
|
}
|
|
|
|
/**
|
|
* Get the underlying handle to the world.
|
|
*
|
|
* @return the world
|
|
* @throws WorldEditException thrown if a reference to the world was lost (i.e. world was unloaded)
|
|
*/
|
|
public World getWorldChecked() throws WorldEditException {
|
|
World world = worldRef.get();
|
|
if (world != null) {
|
|
return world;
|
|
} else {
|
|
throw new WorldReferenceLostException("The reference to the world was lost (i.e. the world may have been unloaded)");
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the underlying handle to the world.
|
|
*
|
|
* @return the world
|
|
* @throws RuntimeException thrown if a reference to the world was lost (i.e. world was unloaded)
|
|
*/
|
|
public World getWorld() {
|
|
World world = worldRef.get();
|
|
if (world != null) {
|
|
return world;
|
|
} else {
|
|
throw new RuntimeException("The reference to the world was lost (i.e. the world may have been unloaded)");
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public String getName() {
|
|
return getWorld().getWorldInfo().getWorldName();
|
|
}
|
|
|
|
@Override
|
|
public boolean setBlock(Vector position, BaseBlock block, boolean notifyAndLight) throws WorldEditException {
|
|
checkNotNull(position);
|
|
checkNotNull(block);
|
|
|
|
World world = getWorldChecked();
|
|
int x = position.getBlockX();
|
|
int y = position.getBlockY();
|
|
int z = position.getBlockZ();
|
|
|
|
// First set the block
|
|
Chunk chunk = world.getChunkFromChunkCoords(x >> 4, z >> 4);
|
|
BlockPos pos = new BlockPos(x, y, z);
|
|
IBlockState old = chunk.getBlockState(pos);
|
|
IBlockState newState = Block.getBlockById(block.getId()).getStateFromMeta(block.getData());
|
|
IBlockState successState = chunk.setBlockState(pos, newState);
|
|
boolean successful = successState != null;
|
|
|
|
// Create the TileEntity
|
|
if (successful) {
|
|
if (block.hasNbtData()) {
|
|
// Kill the old TileEntity
|
|
world.removeTileEntity(pos);
|
|
NBTTagCompound nativeTag = NBTConverter.toNative(block.getNbtData());
|
|
nativeTag.setString("id", block.getNbtId());
|
|
TileEntityUtils.setTileEntity(world, position, nativeTag);
|
|
}
|
|
}
|
|
|
|
if (notifyAndLight) {
|
|
if (!successful) {
|
|
newState = old;
|
|
}
|
|
world.checkLight(pos);
|
|
world.markAndNotifyBlock(pos, chunk, old, newState, UPDATE | NOTIFY);
|
|
}
|
|
|
|
return successful;
|
|
}
|
|
|
|
@Override
|
|
public int getBlockLightLevel(Vector position) {
|
|
checkNotNull(position);
|
|
return getWorld().getLight(new BlockPos(position.getBlockX(), position.getBlockY(), position.getBlockZ()));
|
|
}
|
|
|
|
@Override
|
|
public boolean clearContainerBlockContents(Vector position) {
|
|
checkNotNull(position);
|
|
TileEntity tile = getWorld().getTileEntity(new BlockPos(position.getBlockX(), position.getBlockY(), position.getBlockZ()));
|
|
if ((tile instanceof IInventory)) {
|
|
IInventory inv = (IInventory) tile;
|
|
int size = inv.getSizeInventory();
|
|
for (int i = 0; i < size; i++) {
|
|
inv.setInventorySlotContents(i, null);
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public BaseBiome getBiome(Vector2D position) {
|
|
checkNotNull(position);
|
|
return new BaseBiome(BiomeGenBase.getIdForBiome(getWorld().getBiomeGenForCoords(new BlockPos(position.getBlockX(), 0, position.getBlockZ()))));
|
|
}
|
|
|
|
@Override
|
|
public boolean setBiome(Vector2D position, BaseBiome biome) {
|
|
checkNotNull(position);
|
|
checkNotNull(biome);
|
|
|
|
Chunk chunk = getWorld().getChunkFromBlockCoords(new BlockPos(position.getBlockX(), 0, position.getBlockZ()));
|
|
if ((chunk != null) && (chunk.isLoaded())) {
|
|
chunk.getBiomeArray()[((position.getBlockZ() & 0xF) << 4 | position.getBlockX() & 0xF)] = (byte) biome.getId();
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public void dropItem(Vector position, BaseItemStack item) {
|
|
checkNotNull(position);
|
|
checkNotNull(item);
|
|
|
|
if (item.getType() == 0) {
|
|
return;
|
|
}
|
|
|
|
EntityItem entity = new EntityItem(getWorld(), position.getX(), position.getY(), position.getZ(), ForgeWorldEdit.toForgeItemStack(item));
|
|
entity.setPickupDelay(10);
|
|
getWorld().spawnEntityInWorld(entity);
|
|
}
|
|
|
|
@Override
|
|
public boolean regenerate(Region region, EditSession editSession) {
|
|
BaseBlock[] history = new BaseBlock[256 * (getMaxY() + 1)];
|
|
|
|
for (Vector2D chunk : region.getChunks()) {
|
|
Vector min = new Vector(chunk.getBlockX() * 16, 0, chunk.getBlockZ() * 16);
|
|
|
|
for (int x = 0; x < 16; x++) {
|
|
for (int y = 0; y < getMaxY() + 1; y++) {
|
|
for (int z = 0; z < 16; z++) {
|
|
Vector pt = min.add(x, y, z);
|
|
int index = y * 16 * 16 + z * 16 + x;
|
|
history[index] = editSession.getBlock(pt);
|
|
}
|
|
}
|
|
}
|
|
try {
|
|
Set<Vector2D> chunks = region.getChunks();
|
|
IChunkProvider provider = getWorld().getChunkProvider();
|
|
if (!(provider instanceof ChunkProviderServer)) {
|
|
return false;
|
|
}
|
|
ChunkProviderServer chunkServer = (ChunkProviderServer) provider;
|
|
for (Vector2D coord : chunks) {
|
|
long pos = ChunkCoordIntPair.chunkXZ2Int(coord.getBlockX(), coord.getBlockZ());
|
|
Chunk mcChunk;
|
|
if (chunkServer.chunkExists(coord.getBlockX(), coord.getBlockZ())) {
|
|
mcChunk = chunkServer.loadChunk(coord.getBlockX(), coord.getBlockZ());
|
|
mcChunk.onChunkUnload();
|
|
}
|
|
chunkServer.droppedChunksSet.remove(pos);
|
|
chunkServer.id2ChunkMap.remove(pos);
|
|
mcChunk = chunkServer.provideChunk(coord.getBlockX(), coord.getBlockZ());
|
|
chunkServer.id2ChunkMap.add(pos, mcChunk);
|
|
chunkServer.loadedChunks.add(mcChunk);
|
|
if (mcChunk != null) {
|
|
mcChunk.onChunkLoad();
|
|
mcChunk.populateChunk(chunkServer, chunkServer.chunkGenerator);
|
|
}
|
|
}
|
|
} catch (Throwable t) {
|
|
logger.log(Level.WARNING, "Failed to generate chunk", t);
|
|
return false;
|
|
}
|
|
|
|
for (int x = 0; x < 16; x++) {
|
|
for (int y = 0; y < getMaxY() + 1; y++) {
|
|
for (int z = 0; z < 16; z++) {
|
|
Vector pt = min.add(x, y, z);
|
|
int index = y * 16 * 16 + z * 16 + x;
|
|
|
|
if (!region.contains(pt))
|
|
editSession.smartSetBlock(pt, history[index]);
|
|
else {
|
|
editSession.rememberChange(pt, history[index], editSession.rawGetBlock(pt));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
@Nullable
|
|
private static WorldGenerator createWorldGenerator(TreeType type) {
|
|
switch (type) {
|
|
case TREE: return new WorldGenTrees(true);
|
|
case BIG_TREE: return new WorldGenBigTree(true);
|
|
case REDWOOD: return new WorldGenTaiga2(true);
|
|
case TALL_REDWOOD: return new WorldGenTaiga1();
|
|
case BIRCH: return new WorldGenBirchTree(true, false);
|
|
case JUNGLE: return new WorldGenMegaJungle(true, 10, 20, JUNGLE_LOG, JUNGLE_LEAF);
|
|
case SMALL_JUNGLE: return new WorldGenTrees(true, 4 + random.nextInt(7), JUNGLE_LOG, JUNGLE_LEAF, false);
|
|
case SHORT_JUNGLE: return new WorldGenTrees(true, 4 + random.nextInt(7), JUNGLE_LOG, JUNGLE_LEAF, true);
|
|
case JUNGLE_BUSH: return new WorldGenShrub(JUNGLE_LOG, JUNGLE_SHRUB);
|
|
case RED_MUSHROOM: return new WorldGenBigMushroom(Blocks.BROWN_MUSHROOM_BLOCK);
|
|
case BROWN_MUSHROOM: return new WorldGenBigMushroom(Blocks.RED_MUSHROOM_BLOCK);
|
|
case SWAMP: return new WorldGenSwamp();
|
|
case ACACIA: return new WorldGenSavannaTree(true);
|
|
case DARK_OAK: return new WorldGenCanopyTree(true);
|
|
case MEGA_REDWOOD: return new WorldGenMegaPineTree(false, random.nextBoolean());
|
|
case TALL_BIRCH: return new WorldGenBirchTree(true, true);
|
|
case RANDOM:
|
|
case PINE:
|
|
case RANDOM_REDWOOD:
|
|
default:
|
|
return null;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public boolean generateTree(TreeType type, EditSession editSession, Vector position) throws MaxChangedBlocksException {
|
|
WorldGenerator generator = createWorldGenerator(type);
|
|
return generator != null ? generator.generate(getWorld(), random, ForgeAdapter.toBlockPos(position)) : false;
|
|
}
|
|
|
|
@Override
|
|
public WorldData getWorldData() {
|
|
return ForgeWorldData.getInstance();
|
|
}
|
|
|
|
@Override
|
|
public boolean isValidBlockType(int id) {
|
|
return (id == 0) || (net.minecraft.block.Block.getBlockById(id) != null);
|
|
}
|
|
|
|
@Override
|
|
public BaseBlock getBlock(Vector position) {
|
|
World world = getWorld();
|
|
BlockPos pos = new BlockPos(position.getBlockX(), position.getBlockY(), position.getBlockZ());
|
|
IBlockState state = world.getBlockState(pos);
|
|
TileEntity tile = getWorld().getTileEntity(pos);
|
|
|
|
if (tile != null) {
|
|
return new TileEntityBaseBlock(Block.getIdFromBlock(state.getBlock()), state.getBlock().getMetaFromState(state), tile);
|
|
} else {
|
|
return new BaseBlock(Block.getIdFromBlock(state.getBlock()), state.getBlock().getMetaFromState(state));
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public BaseBlock getLazyBlock(Vector position) {
|
|
World world = getWorld();
|
|
BlockPos pos = new BlockPos(position.getBlockX(), position.getBlockY(), position.getBlockZ());
|
|
IBlockState state = world.getBlockState(pos);
|
|
return new LazyBlock(Block.getIdFromBlock(state.getBlock()), state.getBlock().getMetaFromState(state), this, position);
|
|
}
|
|
|
|
@Override
|
|
public int hashCode() {
|
|
return getWorld().hashCode();
|
|
}
|
|
|
|
@Override
|
|
public boolean equals(Object o) {
|
|
if (o == null) {
|
|
return false;
|
|
} else if ((o instanceof ForgeWorld)) {
|
|
ForgeWorld other = ((ForgeWorld) o);
|
|
World otherWorld = other.worldRef.get();
|
|
World thisWorld = worldRef.get();
|
|
return otherWorld != null && thisWorld != null && otherWorld.equals(thisWorld);
|
|
} else if (o instanceof com.sk89q.worldedit.world.World) {
|
|
return ((com.sk89q.worldedit.world.World) o).getName().equals(getName());
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public List<? extends Entity> getEntities(Region region) {
|
|
List<Entity> entities = new ArrayList<Entity>();
|
|
for (net.minecraft.entity.Entity entity : getWorld().loadedEntityList) {
|
|
if (region.contains(new Vector(entity.posX, entity.posY, entity.posZ))) {
|
|
entities.add(new ForgeEntity(entity));
|
|
}
|
|
}
|
|
return entities;
|
|
}
|
|
|
|
@Override
|
|
public List<? extends Entity> getEntities() {
|
|
List<Entity> entities = new ArrayList<Entity>();
|
|
for (net.minecraft.entity.Entity entity : getWorld().loadedEntityList) {
|
|
entities.add(new ForgeEntity(entity));
|
|
}
|
|
return entities;
|
|
}
|
|
|
|
@Nullable
|
|
@Override
|
|
public Entity createEntity(Location location, BaseEntity entity) {
|
|
World world = getWorld();
|
|
net.minecraft.entity.Entity createdEntity = EntityList.createEntityByName(entity.getTypeId(), world);
|
|
if (createdEntity != null) {
|
|
CompoundTag nativeTag = entity.getNbtData();
|
|
if (nativeTag != null) {
|
|
NBTTagCompound tag = NBTConverter.toNative(entity.getNbtData());
|
|
for (String name : Constants.NO_COPY_ENTITY_NBT_FIELDS) {
|
|
tag.removeTag(name);
|
|
}
|
|
createdEntity.readFromNBT(tag);
|
|
}
|
|
|
|
createdEntity.setLocationAndAngles(location.getX(), location.getY(), location.getZ(), location.getYaw(), location.getPitch());
|
|
|
|
world.spawnEntityInWorld(createdEntity);
|
|
return new ForgeEntity(createdEntity);
|
|
} else {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Thrown when the reference to the world is lost.
|
|
*/
|
|
private static class WorldReferenceLostException extends WorldEditException {
|
|
private WorldReferenceLostException(String message) {
|
|
super(message);
|
|
}
|
|
}
|
|
|
|
}
|