Plex-FAWE/worldedit-core/src/main/java/com/boydti/fawe/object/pattern/DataPattern.java

package com.boydti.fawe.object.pattern;

import com.boydti.fawe.FaweCache;

import com.sk89q.worldedit.world.block.BlockState;
import com.sk89q.worldedit.extent.Extent;
import com.sk89q.worldedit.function.pattern.Pattern;
import com.sk89q.worldedit.math.BlockVector3;
import com.sk89q.worldedit.world.block.BlockStateHolder;
import com.sk89q.worldedit.world.block.BlockTypes;


import static com.google.common.base.Preconditions.checkNotNull;

public class DataPattern extends AbstractExtentPattern {
    private final Pattern pattern;

    public DataPattern(Extent extent, Pattern parent) {
        super(extent);
        checkNotNull(parent);
        this.pattern = parent;
    }

    @Override
    public BlockStateHolder apply(BlockVector3 position) {
        BlockStateHolder oldBlock = getExtent().getBlock(position);
        BlockStateHolder newBlock = pattern.apply(position);
        return oldBlock.withPropertyId(newBlock.getInternalPropertiesId());
    }
}