mirror of
https://github.com/Winds-Studio/Leaf.git
synced 2025-12-19 15:09:25 +00:00
Fix radix sort leak
sync changes from 1.21.8
This commit is contained in:
@@ -2,45 +2,32 @@ package org.dreeam.leaf.util;
|
||||
|
||||
import net.minecraft.world.entity.Entity;
|
||||
|
||||
import java.lang.reflect.Array; // Required for Array.newInstance
|
||||
import java.util.List;
|
||||
public final class FastBitRadixSort {
|
||||
|
||||
public class FastBitRadixSort {
|
||||
private static final int SMALL_ARRAY_THRESHOLD = 6;
|
||||
private static final long[] LONGS = new long[0];
|
||||
private long[] bitsBuffer = LONGS;
|
||||
|
||||
private static final int SMALL_ARRAY_THRESHOLD = 2;
|
||||
private Entity[] entityBuffer = new Entity[0];
|
||||
private long[] bitsBuffer = new long[0];
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
public <T extends Entity, T_REF extends Entity> T[] sort(List<T> entities, T_REF referenceEntity, Class<T> entityClass) {
|
||||
int size = entities.size();
|
||||
public void sort(Object[] entities, int size, net.minecraft.core.Position target) {
|
||||
if (size <= 1) {
|
||||
T[] resultArray = (T[]) Array.newInstance(entityClass, size);
|
||||
return entities.toArray(resultArray);
|
||||
return;
|
||||
}
|
||||
|
||||
if (this.entityBuffer.length < size) {
|
||||
this.entityBuffer = new Entity[size];
|
||||
if (this.bitsBuffer.length < size) {
|
||||
this.bitsBuffer = new long[size];
|
||||
}
|
||||
double tx = target.x();
|
||||
double ty = target.y();
|
||||
double tz = target.z();
|
||||
for (int i = 0; i < size; i++) {
|
||||
this.entityBuffer[i] = entities.get(i);
|
||||
this.bitsBuffer[i] = Double.doubleToRawLongBits(
|
||||
referenceEntity.distanceToSqr(entities.get(i))
|
||||
);
|
||||
this.bitsBuffer[i] = Double.doubleToRawLongBits(((Entity) entities[i]).distanceToSqr(tx, ty, tz));
|
||||
}
|
||||
|
||||
fastRadixSort(this.entityBuffer, this.bitsBuffer, 0, size - 1, 62);
|
||||
|
||||
T[] resultArray = (T[]) Array.newInstance(entityClass, size);
|
||||
for (int i = 0; i < size; i++) {
|
||||
resultArray[i] = entityClass.cast(this.entityBuffer[i]);
|
||||
}
|
||||
return resultArray;
|
||||
fastRadixSort(entities, this.bitsBuffer, 0, size - 1, 62);
|
||||
}
|
||||
|
||||
private void fastRadixSort(
|
||||
Entity[] ents,
|
||||
private static void fastRadixSort(
|
||||
Object[] ents,
|
||||
long[] bits,
|
||||
int low,
|
||||
int high,
|
||||
@@ -79,15 +66,15 @@ public class FastBitRadixSort {
|
||||
}
|
||||
}
|
||||
|
||||
private void insertionSort(
|
||||
Entity[] ents,
|
||||
private static void insertionSort(
|
||||
Object[] ents,
|
||||
long[] bits,
|
||||
int low,
|
||||
int high
|
||||
) {
|
||||
for (int i = low + 1; i <= high; i++) {
|
||||
int j = i;
|
||||
Entity currentEntity = ents[j];
|
||||
Object currentEntity = ents[j];
|
||||
long currentBits = bits[j];
|
||||
|
||||
while (j > low && bits[j - 1] > currentBits) {
|
||||
@@ -100,8 +87,8 @@ public class FastBitRadixSort {
|
||||
}
|
||||
}
|
||||
|
||||
private void swap(Entity[] ents, long[] bits, int a, int b) {
|
||||
Entity tempEntity = ents[a];
|
||||
private static void swap(Object[] ents, long[] bits, int a, int b) {
|
||||
Object tempEntity = ents[a];
|
||||
ents[a] = ents[b];
|
||||
ents[b] = tempEntity;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user