mirror of https://github.com/GeyserMC/Geyser.git
443 lines
19 KiB
Java
443 lines
19 KiB
Java
/*
|
|
* Copyright (c) 2019-2021 GeyserMC. http://geysermc.org
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*
|
|
* @author GeyserMC
|
|
* @link https://github.com/GeyserMC/Geyser
|
|
*/
|
|
|
|
package org.geysermc.connector.entity.living;
|
|
|
|
import com.github.steveice10.mc.protocol.data.game.entity.metadata.EntityMetadata;
|
|
import com.github.steveice10.mc.protocol.data.game.entity.metadata.MetadataType;
|
|
import com.github.steveice10.mc.protocol.data.game.entity.metadata.Rotation;
|
|
import com.nukkitx.math.vector.Vector3f;
|
|
import com.nukkitx.protocol.bedrock.data.entity.EntityData;
|
|
import com.nukkitx.protocol.bedrock.data.entity.EntityFlag;
|
|
import com.nukkitx.protocol.bedrock.data.inventory.ItemData;
|
|
import com.nukkitx.protocol.bedrock.packet.MoveEntityAbsolutePacket;
|
|
import lombok.Getter;
|
|
import org.geysermc.connector.entity.LivingEntity;
|
|
import org.geysermc.connector.entity.type.EntityType;
|
|
import org.geysermc.connector.network.session.GeyserSession;
|
|
|
|
public class ArmorStandEntity extends LivingEntity {
|
|
|
|
// These are used to store the state of the armour stand for use when handling invisibility
|
|
@Getter
|
|
private boolean isMarker = false;
|
|
private boolean isInvisible = false;
|
|
private boolean isSmall = false;
|
|
|
|
/**
|
|
* On Java Edition, armor stands always show their name. Invisibility hides the name on Bedrock.
|
|
* By having a second entity, we can allow an invisible entity with the name tag.
|
|
* (This lets armor on armor stands still show)
|
|
*/
|
|
private ArmorStandEntity secondEntity = null;
|
|
/**
|
|
* Whether this is the primary armor stand that holds the armor and not the name tag.
|
|
*/
|
|
private boolean primaryEntity = true;
|
|
/**
|
|
* Whether the entity's position must be updated to included the offset.
|
|
*
|
|
* This should be true when the Java server marks the armor stand as invisible, but we shrink the entity
|
|
* to allow the nametag to appear. Basically:
|
|
* - Is visible: this is irrelevant (false)
|
|
* - Has armor, no name: false
|
|
* - Has armor, has name: false, with a second entity
|
|
* - No armor, no name: false
|
|
* - No armor, yes name: true
|
|
*/
|
|
private boolean positionRequiresOffset = false;
|
|
/**
|
|
* Whether we should update the position of this armor stand after metadata updates.
|
|
*/
|
|
private boolean positionUpdateRequired = false;
|
|
private GeyserSession session;
|
|
|
|
public ArmorStandEntity(long entityId, long geyserId, EntityType entityType, Vector3f position, Vector3f motion, Vector3f rotation) {
|
|
super(entityId, geyserId, entityType, position, motion, rotation);
|
|
}
|
|
|
|
@Override
|
|
public void spawnEntity(GeyserSession session) {
|
|
this.session = session;
|
|
this.rotation = Vector3f.from(rotation.getX(), rotation.getX(), rotation.getX());
|
|
super.spawnEntity(session);
|
|
}
|
|
|
|
@Override
|
|
public boolean despawnEntity(GeyserSession session) {
|
|
if (secondEntity != null) {
|
|
secondEntity.despawnEntity(session);
|
|
}
|
|
return super.despawnEntity(session);
|
|
}
|
|
|
|
@Override
|
|
public void moveRelative(GeyserSession session, double relX, double relY, double relZ, Vector3f rotation, boolean isOnGround) {
|
|
if (secondEntity != null) {
|
|
secondEntity.moveRelative(session, relX, relY, relZ, rotation, isOnGround);
|
|
}
|
|
super.moveRelative(session, relX, relY, relZ, rotation, isOnGround);
|
|
}
|
|
|
|
@Override
|
|
public void moveAbsolute(GeyserSession session, Vector3f position, Vector3f rotation, boolean isOnGround, boolean teleported) {
|
|
if (secondEntity != null) {
|
|
secondEntity.moveAbsolute(session, applyOffsetToPosition(position), rotation, isOnGround, teleported);
|
|
} else if (positionRequiresOffset) {
|
|
// Fake the height to be above where it is so the nametag appears in the right location for invisible non-marker armour stands
|
|
position = applyOffsetToPosition(position);
|
|
}
|
|
|
|
super.moveAbsolute(session, position, Vector3f.from(rotation.getX(), rotation.getX(), rotation.getX()), isOnGround, teleported);
|
|
}
|
|
|
|
@Override
|
|
public void updateBedrockMetadata(EntityMetadata entityMetadata, GeyserSession session) {
|
|
super.updateBedrockMetadata(entityMetadata, session);
|
|
if (entityMetadata.getId() == 2) {
|
|
updateSecondEntityStatus(false);
|
|
} else if (entityMetadata.getId() == 15 && entityMetadata.getType() == MetadataType.BYTE) {
|
|
byte xd = (byte) entityMetadata.getValue();
|
|
|
|
// isSmall
|
|
boolean newIsSmall = (xd & 0x01) == 0x01;
|
|
if (newIsSmall != isSmall) {
|
|
if (positionRequiresOffset) {
|
|
// Fix new inconsistency with offset
|
|
this.position = fixOffsetForSize(position, newIsSmall);
|
|
positionUpdateRequired = true;
|
|
}
|
|
|
|
isSmall = newIsSmall;
|
|
if (!isMarker) {
|
|
toggleSmallStatus();
|
|
}
|
|
}
|
|
|
|
// setMarker
|
|
boolean oldIsMarker = isMarker;
|
|
isMarker = (xd & 0x10) == 0x10;
|
|
if (oldIsMarker != isMarker) {
|
|
if (isMarker) {
|
|
metadata.put(EntityData.BOUNDING_BOX_WIDTH, 0.0f);
|
|
metadata.put(EntityData.BOUNDING_BOX_HEIGHT, 0.0f);
|
|
metadata.put(EntityData.SCALE, 0f);
|
|
} else {
|
|
toggleSmallStatus();
|
|
}
|
|
|
|
updateSecondEntityStatus(false);
|
|
}
|
|
|
|
// The following values don't do anything on normal Bedrock.
|
|
// But if given a resource pack, then we can use these values to control armor stand visual properties
|
|
metadata.getFlags().setFlag(EntityFlag.ANGRY, (xd & 0x04) != 0x04); // Has arms
|
|
metadata.getFlags().setFlag(EntityFlag.ADMIRING, (xd & 0x08) == 0x08); // Has no baseplate
|
|
} else {
|
|
EntityData dataLeech = null;
|
|
EntityFlag negativeXToggle = null;
|
|
EntityFlag negativeYToggle = null;
|
|
EntityFlag negativeZToggle = null;
|
|
switch (entityMetadata.getId()) {
|
|
case 16 -> { // Head
|
|
dataLeech = EntityData.MARK_VARIANT;
|
|
negativeXToggle = EntityFlag.INTERESTED;
|
|
negativeYToggle = EntityFlag.CHARGED;
|
|
negativeZToggle = EntityFlag.POWERED;
|
|
}
|
|
case 17 -> { // Body
|
|
dataLeech = EntityData.VARIANT;
|
|
negativeXToggle = EntityFlag.IN_LOVE;
|
|
negativeYToggle = EntityFlag.CELEBRATING;
|
|
negativeZToggle = EntityFlag.CELEBRATING_SPECIAL;
|
|
}
|
|
case 18 -> { // Left arm
|
|
dataLeech = EntityData.TRADE_TIER;
|
|
negativeXToggle = EntityFlag.CHARGING;
|
|
negativeYToggle = EntityFlag.CRITICAL;
|
|
negativeZToggle = EntityFlag.DANCING;
|
|
}
|
|
case 19 -> { // Right arm
|
|
dataLeech = EntityData.MAX_TRADE_TIER;
|
|
negativeXToggle = EntityFlag.ELDER;
|
|
negativeYToggle = EntityFlag.EMOTING;
|
|
negativeZToggle = EntityFlag.IDLING;
|
|
}
|
|
case 20 -> { // Left leg
|
|
dataLeech = EntityData.SKIN_ID;
|
|
negativeXToggle = EntityFlag.IS_ILLAGER_CAPTAIN;
|
|
negativeYToggle = EntityFlag.IS_IN_UI;
|
|
negativeZToggle = EntityFlag.LINGERING;
|
|
}
|
|
case 21 -> { // Right leg
|
|
dataLeech = EntityData.HURT_DIRECTION;
|
|
negativeXToggle = EntityFlag.IS_PREGNANT;
|
|
negativeYToggle = EntityFlag.SHEARED;
|
|
negativeZToggle = EntityFlag.STALKING;
|
|
}
|
|
}
|
|
if (dataLeech != null) {
|
|
// Indicate that rotation should be checked
|
|
metadata.getFlags().setFlag(EntityFlag.BRIBED, true);
|
|
|
|
Rotation rotation = (Rotation) entityMetadata.getValue();
|
|
int rotationX = getRotation(rotation.getPitch());
|
|
int rotationY = getRotation(rotation.getYaw());
|
|
int rotationZ = getRotation(rotation.getRoll());
|
|
// The top bit acts like binary and determines if each rotation goes above 100
|
|
// We don't do this for the negative values out of concerns of the number being too big
|
|
int topBit = (Math.abs(rotationX) >= 100 ? 4 : 0) + (Math.abs(rotationY) >= 100 ? 2 : 0) + (Math.abs(rotationZ) >= 100 ? 1 : 0);
|
|
int value = (topBit * 1000000) + ((Math.abs(rotationX) % 100) * 10000) + ((Math.abs(rotationY) % 100) * 100) + (Math.abs(rotationZ) % 100);
|
|
metadata.put(dataLeech, value);
|
|
// Set the entity flags if a value is negative
|
|
metadata.getFlags().setFlag(negativeXToggle, rotationX < 0);
|
|
metadata.getFlags().setFlag(negativeYToggle, rotationY < 0);
|
|
metadata.getFlags().setFlag(negativeZToggle, rotationZ < 0);
|
|
}
|
|
}
|
|
if (secondEntity != null) {
|
|
secondEntity.updateBedrockMetadata(entityMetadata, session);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void updateBedrockMetadata(GeyserSession session) {
|
|
if (secondEntity != null) {
|
|
secondEntity.updateBedrockMetadata(session);
|
|
}
|
|
super.updateBedrockMetadata(session);
|
|
if (positionUpdateRequired) {
|
|
positionUpdateRequired = false;
|
|
updatePosition();
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void setInvisible(GeyserSession session, boolean value) {
|
|
// Check if the armour stand is invisible and store accordingly
|
|
if (primaryEntity) {
|
|
isInvisible = value;
|
|
updateSecondEntityStatus(false);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void setHelmet(ItemData helmet) {
|
|
super.setHelmet(helmet);
|
|
updateSecondEntityStatus(true);
|
|
}
|
|
|
|
@Override
|
|
public void setChestplate(ItemData chestplate) {
|
|
super.setChestplate(chestplate);
|
|
updateSecondEntityStatus(true);
|
|
}
|
|
|
|
@Override
|
|
public void setLeggings(ItemData leggings) {
|
|
super.setLeggings(leggings);
|
|
updateSecondEntityStatus(true);
|
|
}
|
|
|
|
@Override
|
|
public void setBoots(ItemData boots) {
|
|
super.setBoots(boots);
|
|
updateSecondEntityStatus(true);
|
|
}
|
|
|
|
@Override
|
|
public void setHand(ItemData hand) {
|
|
super.setHand(hand);
|
|
updateSecondEntityStatus(true);
|
|
}
|
|
|
|
@Override
|
|
public void setOffHand(ItemData offHand) {
|
|
super.setOffHand(offHand);
|
|
updateSecondEntityStatus(true);
|
|
}
|
|
|
|
/**
|
|
* Determine if we need to load or unload the second entity.
|
|
*
|
|
* @param sendMetadata whether to send a metadata update after a change.
|
|
*/
|
|
private void updateSecondEntityStatus(boolean sendMetadata) {
|
|
// A secondary entity always has to have the offset applied, so it remains invisible and the nametag shows.
|
|
if (!primaryEntity) return;
|
|
if (!isInvisible || isMarker) {
|
|
// It is either impossible to show armor, or the armor stand isn't invisible. We good.
|
|
metadata.getFlags().setFlag(EntityFlag.INVISIBLE, false);
|
|
updateOffsetRequirement(false);
|
|
if (positionUpdateRequired) {
|
|
positionUpdateRequired = false;
|
|
updatePosition();
|
|
}
|
|
|
|
if (secondEntity != null) {
|
|
secondEntity.despawnEntity(session);
|
|
secondEntity = null;
|
|
}
|
|
return;
|
|
}
|
|
//boolean isNametagEmpty = metadata.getString(EntityData.NAMETAG).isEmpty() || metadata.getByte(EntityData.NAMETAG_ALWAYS_SHOW, (byte) -1) == (byte) 0; - may not be necessary?
|
|
boolean isNametagEmpty = metadata.getString(EntityData.NAMETAG).isEmpty();
|
|
if (!isNametagEmpty && (!helmet.equals(ItemData.AIR) || !chestplate.equals(ItemData.AIR) || !leggings.equals(ItemData.AIR)
|
|
|| !boots.equals(ItemData.AIR) || !hand.equals(ItemData.AIR) || !offHand.equals(ItemData.AIR))) {
|
|
// If the second entity exists, no need to recreate it.
|
|
// We can't stuff this check above or else it'll fall into another else case and delete the second entity
|
|
if (secondEntity != null) return;
|
|
|
|
// Create the second entity. It doesn't need to worry about the items, but it does need to worry about
|
|
// the metadata as it will hold the name tag.
|
|
secondEntity = new ArmorStandEntity(0, session.getEntityCache().getNextEntityId().incrementAndGet(),
|
|
EntityType.ARMOR_STAND, position, motion, rotation);
|
|
secondEntity.primaryEntity = false;
|
|
if (!this.positionRequiresOffset) {
|
|
// Ensure the offset is applied for the 0 scale
|
|
secondEntity.position = secondEntity.applyOffsetToPosition(secondEntity.position);
|
|
}
|
|
// Copy metadata
|
|
secondEntity.isSmall = isSmall;
|
|
secondEntity.getMetadata().putAll(metadata);
|
|
// Copy the flags so they aren't the same object in memory
|
|
secondEntity.getMetadata().putFlags(metadata.getFlags().copy());
|
|
// Guarantee this copy is NOT invisible
|
|
secondEntity.getMetadata().getFlags().setFlag(EntityFlag.INVISIBLE, false);
|
|
// Scale to 0 to show nametag
|
|
secondEntity.getMetadata().put(EntityData.SCALE, 0.0f);
|
|
// No bounding box as we don't want to interact with this entity
|
|
secondEntity.getMetadata().put(EntityData.BOUNDING_BOX_WIDTH, 0.0f);
|
|
secondEntity.getMetadata().put(EntityData.BOUNDING_BOX_HEIGHT, 0.0f);
|
|
secondEntity.spawnEntity(session);
|
|
|
|
// Reset scale of the proper armor stand
|
|
this.metadata.put(EntityData.SCALE, isSmall ? 0.55f : 1f);
|
|
// Set the proper armor stand to invisible to show armor
|
|
this.metadata.getFlags().setFlag(EntityFlag.INVISIBLE, true);
|
|
// Update the position of the armor stand
|
|
updateOffsetRequirement(false);
|
|
} else if (isNametagEmpty) {
|
|
// We can just make an invisible entity
|
|
// Reset scale of the proper armor stand
|
|
metadata.put(EntityData.SCALE, isSmall ? 0.55f : 1f);
|
|
// Set the proper armor stand to invisible to show armor
|
|
metadata.getFlags().setFlag(EntityFlag.INVISIBLE, true);
|
|
// Update offset
|
|
updateOffsetRequirement(false);
|
|
|
|
if (secondEntity != null) {
|
|
secondEntity.despawnEntity(session);
|
|
secondEntity = null;
|
|
}
|
|
} else {
|
|
// Nametag is not empty and there is no armor
|
|
// We don't need to make a new entity
|
|
metadata.getFlags().setFlag(EntityFlag.INVISIBLE, false);
|
|
metadata.put(EntityData.SCALE, 0.0f);
|
|
// As the above is applied, we need an offset
|
|
updateOffsetRequirement(true);
|
|
|
|
if (secondEntity != null) {
|
|
secondEntity.despawnEntity(session);
|
|
secondEntity = null;
|
|
}
|
|
}
|
|
if (sendMetadata) {
|
|
this.updateBedrockMetadata(session);
|
|
}
|
|
}
|
|
|
|
private int getRotation(float rotation) {
|
|
rotation = rotation % 360f;
|
|
if (rotation < -180f) {
|
|
rotation += 360f;
|
|
} else if (rotation >= 180f) {
|
|
// 181 -> -179
|
|
rotation = -(180 - (rotation - 180));
|
|
}
|
|
return (int) rotation;
|
|
}
|
|
|
|
/**
|
|
* If this armor stand is not a marker, set its bounding box size and scale.
|
|
*/
|
|
private void toggleSmallStatus() {
|
|
metadata.put(EntityData.BOUNDING_BOX_WIDTH, isSmall ? 0.25f : entityType.getWidth());
|
|
metadata.put(EntityData.BOUNDING_BOX_HEIGHT, isSmall ? 0.9875f : entityType.getHeight());
|
|
metadata.put(EntityData.SCALE, isSmall ? 0.55f : 1f);
|
|
}
|
|
|
|
/**
|
|
* @return the selected position with the position offset applied.
|
|
*/
|
|
private Vector3f applyOffsetToPosition(Vector3f position) {
|
|
return position.add(0d, entityType.getHeight() * (isSmall ? 0.55d : 1d), 0d);
|
|
}
|
|
|
|
/**
|
|
* @return an adjusted offset for the new small status.
|
|
*/
|
|
private Vector3f fixOffsetForSize(Vector3f position, boolean isNowSmall) {
|
|
position = removeOffsetFromPosition(position);
|
|
return position.add(0d, entityType.getHeight() * (isNowSmall ? 0.55d : 1d), 0d);
|
|
}
|
|
|
|
/**
|
|
* @return the selected position with the position offset removed.
|
|
*/
|
|
private Vector3f removeOffsetFromPosition(Vector3f position) {
|
|
return position.sub(0d, entityType.getHeight() * (isSmall ? 0.55d : 1d), 0d);
|
|
}
|
|
|
|
/**
|
|
* Set the offset to a new value; if it changed, update the position, too.
|
|
*/
|
|
private void updateOffsetRequirement(boolean newValue) {
|
|
if (newValue != positionRequiresOffset) {
|
|
this.positionRequiresOffset = newValue;
|
|
if (positionRequiresOffset) {
|
|
this.position = applyOffsetToPosition(position);
|
|
} else {
|
|
this.position = removeOffsetFromPosition(position);
|
|
}
|
|
positionUpdateRequired = true;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Updates position without calling movement code.
|
|
*/
|
|
private void updatePosition() {
|
|
MoveEntityAbsolutePacket moveEntityPacket = new MoveEntityAbsolutePacket();
|
|
moveEntityPacket.setRuntimeEntityId(geyserId);
|
|
moveEntityPacket.setPosition(position);
|
|
moveEntityPacket.setRotation(Vector3f.from(rotation.getX(), rotation.getX(), rotation.getX()));
|
|
moveEntityPacket.setOnGround(onGround);
|
|
moveEntityPacket.setTeleported(false);
|
|
session.sendUpstreamPacket(moveEntityPacket);
|
|
}
|
|
}
|