335 lines
11 KiB
Java
335 lines
11 KiB
Java
package ru.bclib.util;
|
|
|
|
import com.google.common.collect.Lists;
|
|
import com.mojang.math.Vector3f;
|
|
import net.minecraft.core.BlockPos;
|
|
import net.minecraft.core.BlockPos.MutableBlockPos;
|
|
import net.minecraft.util.Mth;
|
|
import net.minecraft.world.level.WorldGenLevel;
|
|
import net.minecraft.world.level.block.state.BlockState;
|
|
import ru.bclib.sdf.SDF;
|
|
import ru.bclib.sdf.operator.SDFUnion;
|
|
import ru.bclib.sdf.primitive.SDFLine;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Random;
|
|
import java.util.function.Function;
|
|
|
|
public class SplineHelper {
|
|
public static List<Vector3f> makeSpline(float x1, float y1, float z1, float x2, float y2, float z2, int points) {
|
|
List<Vector3f> spline = Lists.newArrayList();
|
|
spline.add(new Vector3f(x1, y1, z1));
|
|
int count = points - 1;
|
|
for (int i = 1; i < count; i++) {
|
|
float delta = (float) i / (float) count;
|
|
float x = Mth.lerp(delta, x1, x2);
|
|
float y = Mth.lerp(delta, y1, y2);
|
|
float z = Mth.lerp(delta, z1, z2);
|
|
spline.add(new Vector3f(x, y, z));
|
|
}
|
|
spline.add(new Vector3f(x2, y2, z2));
|
|
return spline;
|
|
}
|
|
|
|
public static List<Vector3f> smoothSpline(List<Vector3f> spline, int segmentPoints) {
|
|
List<Vector3f> result = Lists.newArrayList();
|
|
Vector3f start = spline.get(0);
|
|
for (int i = 1; i < spline.size(); i++) {
|
|
Vector3f end = spline.get(i);
|
|
for (int j = 0; j < segmentPoints; j++) {
|
|
float delta = (float) j / segmentPoints;
|
|
delta = 0.5F - 0.5F * Mth.cos(delta * 3.14159F);
|
|
result.add(lerp(start, end, delta));
|
|
}
|
|
start = end;
|
|
}
|
|
result.add(start);
|
|
return result;
|
|
}
|
|
|
|
private static Vector3f lerp(Vector3f start, Vector3f end, float delta) {
|
|
float x = Mth.lerp(delta, start.x(), end.x());
|
|
float y = Mth.lerp(delta, start.y(), end.y());
|
|
float z = Mth.lerp(delta, start.z(), end.z());
|
|
return new Vector3f(x, y, z);
|
|
}
|
|
|
|
public static void offsetParts(List<Vector3f> spline, Random random, float dx, float dy, float dz) {
|
|
int count = spline.size();
|
|
for (int i = 1; i < count; i++) {
|
|
Vector3f pos = spline.get(i);
|
|
float x = pos.x() + (float) random.nextGaussian() * dx;
|
|
float y = pos.y() + (float) random.nextGaussian() * dy;
|
|
float z = pos.z() + (float) random.nextGaussian() * dz;
|
|
pos.set(x, y, z);
|
|
}
|
|
}
|
|
|
|
public static void powerOffset(List<Vector3f> spline, float distance, float power) {
|
|
int count = spline.size();
|
|
float max = count + 1;
|
|
for (int i = 1; i < count; i++) {
|
|
Vector3f pos = spline.get(i);
|
|
float x = (float) i / max;
|
|
float y = pos.y() + (float) Math.pow(x, power) * distance;
|
|
pos.set(pos.x(), y, pos.z());
|
|
}
|
|
}
|
|
|
|
public static SDF buildSDF(List<Vector3f> spline, float radius1, float radius2, Function<BlockPos, BlockState> placerFunction) {
|
|
int count = spline.size();
|
|
float max = count - 2;
|
|
SDF result = null;
|
|
Vector3f start = spline.get(0);
|
|
for (int i = 1; i < count; i++) {
|
|
Vector3f pos = spline.get(i);
|
|
float delta = (float) (i - 1) / max;
|
|
SDF line = new SDFLine().setRadius(Mth.lerp(delta, radius1, radius2)).setStart(start.x(), start.y(), start.z()).setEnd(pos.x(), pos.y(), pos.z()).setBlock(placerFunction);
|
|
result = result == null ? line : new SDFUnion().setSourceA(result).setSourceB(line);
|
|
start = pos;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
public static SDF buildSDF(List<Vector3f> spline, Function<Float, Float> radiusFunction, Function<BlockPos, BlockState> placerFunction) {
|
|
int count = spline.size();
|
|
float max = count - 2;
|
|
SDF result = null;
|
|
Vector3f start = spline.get(0);
|
|
for (int i = 1; i < count; i++) {
|
|
Vector3f pos = spline.get(i);
|
|
float delta = (float) (i - 1) / max;
|
|
SDF line = new SDFLine().setRadius(radiusFunction.apply(delta)).setStart(start.x(), start.y(), start.z()).setEnd(pos.x(), pos.y(), pos.z()).setBlock(placerFunction);
|
|
result = result == null ? line : new SDFUnion().setSourceA(result).setSourceB(line);
|
|
start = pos;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
public static boolean fillSpline(List<Vector3f> spline, WorldGenLevel world, BlockState state, BlockPos pos, Function<BlockState, Boolean> replace) {
|
|
Vector3f startPos = spline.get(0);
|
|
for (int i = 1; i < spline.size(); i++) {
|
|
Vector3f endPos = spline.get(i);
|
|
if (!(fillLine(startPos, endPos, world, state, pos, replace))) {
|
|
return false;
|
|
}
|
|
startPos = endPos;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public static void fillSplineForce(List<Vector3f> spline, WorldGenLevel world, BlockState state, BlockPos pos, Function<BlockState, Boolean> replace) {
|
|
Vector3f startPos = spline.get(0);
|
|
for (int i = 1; i < spline.size(); i++) {
|
|
Vector3f endPos = spline.get(i);
|
|
fillLineForce(startPos, endPos, world, state, pos, replace);
|
|
startPos = endPos;
|
|
}
|
|
}
|
|
|
|
public static boolean fillLine(Vector3f start, Vector3f end, WorldGenLevel world, BlockState state, BlockPos pos, Function<BlockState, Boolean> replace) {
|
|
float dx = end.x() - start.x();
|
|
float dy = end.y() - start.y();
|
|
float dz = end.z() - start.z();
|
|
float max = MHelper.max(Math.abs(dx), Math.abs(dy), Math.abs(dz));
|
|
int count = MHelper.floor(max + 1);
|
|
dx /= max;
|
|
dy /= max;
|
|
dz /= max;
|
|
float x = start.x();
|
|
float y = start.y();
|
|
float z = start.z();
|
|
boolean down = Math.abs(dy) > 0.2;
|
|
|
|
BlockState bState;
|
|
MutableBlockPos bPos = new MutableBlockPos();
|
|
for (int i = 0; i < count; i++) {
|
|
bPos.set(x + pos.getX(), y + pos.getY(), z + pos.getZ());
|
|
bState = world.getBlockState(bPos);
|
|
if (bState.equals(state) || replace.apply(bState)) {
|
|
BlocksHelper.setWithoutUpdate(world, bPos, state);
|
|
bPos.setY(bPos.getY() - 1);
|
|
bState = world.getBlockState(bPos);
|
|
if (down && bState.equals(state) || replace.apply(bState)) {
|
|
BlocksHelper.setWithoutUpdate(world, bPos, state);
|
|
}
|
|
}
|
|
else {
|
|
return false;
|
|
}
|
|
x += dx;
|
|
y += dy;
|
|
z += dz;
|
|
}
|
|
bPos.set(end.x() + pos.getX(), end.y() + pos.getY(), end.z() + pos.getZ());
|
|
bState = world.getBlockState(bPos);
|
|
if (bState.equals(state) || replace.apply(bState)) {
|
|
BlocksHelper.setWithoutUpdate(world, bPos, state);
|
|
bPos.setY(bPos.getY() - 1);
|
|
bState = world.getBlockState(bPos);
|
|
if (down && bState.equals(state) || replace.apply(bState)) {
|
|
BlocksHelper.setWithoutUpdate(world, bPos, state);
|
|
}
|
|
return true;
|
|
}
|
|
else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
public static void fillLineForce(Vector3f start, Vector3f end, WorldGenLevel world, BlockState state, BlockPos pos, Function<BlockState, Boolean> replace) {
|
|
float dx = end.x() - start.x();
|
|
float dy = end.y() - start.y();
|
|
float dz = end.z() - start.z();
|
|
float max = MHelper.max(Math.abs(dx), Math.abs(dy), Math.abs(dz));
|
|
int count = MHelper.floor(max + 1);
|
|
dx /= max;
|
|
dy /= max;
|
|
dz /= max;
|
|
float x = start.x();
|
|
float y = start.y();
|
|
float z = start.z();
|
|
boolean down = Math.abs(dy) > 0.2;
|
|
|
|
BlockState bState;
|
|
MutableBlockPos bPos = new MutableBlockPos();
|
|
for (int i = 0; i < count; i++) {
|
|
bPos.set(x + pos.getX(), y + pos.getY(), z + pos.getZ());
|
|
bState = world.getBlockState(bPos);
|
|
if (replace.apply(bState)) {
|
|
BlocksHelper.setWithoutUpdate(world, bPos, state);
|
|
bPos.setY(bPos.getY() - 1);
|
|
bState = world.getBlockState(bPos);
|
|
if (down && replace.apply(bState)) {
|
|
BlocksHelper.setWithoutUpdate(world, bPos, state);
|
|
}
|
|
}
|
|
x += dx;
|
|
y += dy;
|
|
z += dz;
|
|
}
|
|
bPos.set(end.x() + pos.getX(), end.y() + pos.getY(), end.z() + pos.getZ());
|
|
bState = world.getBlockState(bPos);
|
|
if (replace.apply(bState)) {
|
|
BlocksHelper.setWithoutUpdate(world, bPos, state);
|
|
bPos.setY(bPos.getY() - 1);
|
|
bState = world.getBlockState(bPos);
|
|
if (down && replace.apply(bState)) {
|
|
BlocksHelper.setWithoutUpdate(world, bPos, state);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static boolean canGenerate(List<Vector3f> spline, float scale, BlockPos start, WorldGenLevel world, Function<BlockState, Boolean> canReplace) {
|
|
int count = spline.size();
|
|
Vector3f vec = spline.get(0);
|
|
MutableBlockPos mut = new MutableBlockPos();
|
|
float x1 = start.getX() + vec.x() * scale;
|
|
float y1 = start.getY() + vec.y() * scale;
|
|
float z1 = start.getZ() + vec.z() * scale;
|
|
for (int i = 1; i < count; i++) {
|
|
vec = spline.get(i);
|
|
float x2 = start.getX() + vec.x() * scale;
|
|
float y2 = start.getY() + vec.y() * scale;
|
|
float z2 = start.getZ() + vec.z() * scale;
|
|
|
|
for (float py = y1; py < y2; py += 3) {
|
|
if (py - start.getY() < 10) continue;
|
|
float lerp = (py - y1) / (y2 - y1);
|
|
float x = Mth.lerp(lerp, x1, x2);
|
|
float z = Mth.lerp(lerp, z1, z2);
|
|
mut.set(x, py, z);
|
|
if (!canReplace.apply(world.getBlockState(mut))) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
x1 = x2;
|
|
y1 = y2;
|
|
z1 = z2;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public static boolean canGenerate(List<Vector3f> spline, BlockPos start, WorldGenLevel world, Function<BlockState, Boolean> canReplace) {
|
|
int count = spline.size();
|
|
Vector3f vec = spline.get(0);
|
|
MutableBlockPos mut = new MutableBlockPos();
|
|
float x1 = start.getX() + vec.x();
|
|
float y1 = start.getY() + vec.y();
|
|
float z1 = start.getZ() + vec.z();
|
|
for (int i = 1; i < count; i++) {
|
|
vec = spline.get(i);
|
|
float x2 = start.getX() + vec.x();
|
|
float y2 = start.getY() + vec.y();
|
|
float z2 = start.getZ() + vec.z();
|
|
|
|
for (float py = y1; py < y2; py += 3) {
|
|
if (py - start.getY() < 10) continue;
|
|
float lerp = (py - y1) / (y2 - y1);
|
|
float x = Mth.lerp(lerp, x1, x2);
|
|
float z = Mth.lerp(lerp, z1, z2);
|
|
mut.set(x, py, z);
|
|
if (!canReplace.apply(world.getBlockState(mut))) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
x1 = x2;
|
|
y1 = y2;
|
|
z1 = z2;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public static Vector3f getPos(List<Vector3f> spline, float index) {
|
|
int i = (int) index;
|
|
int last = spline.size() - 1;
|
|
if (i >= last) {
|
|
return spline.get(last);
|
|
}
|
|
float delta = index - i;
|
|
Vector3f p1 = spline.get(i);
|
|
Vector3f p2 = spline.get(i + 1);
|
|
float x = Mth.lerp(delta, p1.x(), p2.x());
|
|
float y = Mth.lerp(delta, p1.y(), p2.y());
|
|
float z = Mth.lerp(delta, p1.z(), p2.z());
|
|
return new Vector3f(x, y, z);
|
|
}
|
|
|
|
public static void rotateSpline(List<Vector3f> spline, float angle) {
|
|
for (Vector3f v : spline) {
|
|
float sin = (float) Math.sin(angle);
|
|
float cos = (float) Math.cos(angle);
|
|
float x = v.x() * cos + v.z() * sin;
|
|
float z = v.x() * sin + v.z() * cos;
|
|
v.set(x, v.y(), z);
|
|
}
|
|
}
|
|
|
|
public static List<Vector3f> copySpline(List<Vector3f> spline) {
|
|
List<Vector3f> result = new ArrayList<Vector3f>(spline.size());
|
|
for (Vector3f v : spline) {
|
|
result.add(new Vector3f(v.x(), v.y(), v.z()));
|
|
}
|
|
return result;
|
|
}
|
|
|
|
public static void scale(List<Vector3f> spline, float scale) {
|
|
scale(spline, scale, scale, scale);
|
|
}
|
|
|
|
public static void scale(List<Vector3f> spline, float x, float y, float z) {
|
|
for (Vector3f v : spline) {
|
|
v.set(v.x() * x, v.y() * y, v.z() * z);
|
|
}
|
|
}
|
|
|
|
public static void offset(List<Vector3f> spline, Vector3f offset) {
|
|
for (Vector3f v : spline) {
|
|
v.set(offset.x() + v.x(), offset.y() + v.y(), offset.z() + v.z());
|
|
}
|
|
}
|
|
}
|