The leather armor pieces are now correctly colored

This commit is contained in:
R0bbyYT 2020-12-05 20:45:54 +01:00
parent e3f6b73241
commit a1ef2b21c3

View File

@ -4,122 +4,151 @@ import net.minestom.server.chat.ChatColor;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jglrxavpok.hephaistos.nbt.NBTCompound; import org.jglrxavpok.hephaistos.nbt.NBTCompound;
/** Represents the item meat for leather armor parts. */
public class LeatherArmorMeta implements ItemMeta { public class LeatherArmorMeta implements ItemMeta {
private boolean modified; private static final int BIT_MASK = 0xFF;
private byte red, green, blue;
/** private boolean modified;
* Sets the color of the leather armor piece. private byte red;
* private byte green;
* @param color the color of the leather armor private byte blue;
*/
public void setColor(ChatColor color) { /**
// TODO using "CHAT color" is pretty weird, maybe that the class should be renamed to "Color" * Sets the color of the leather armor piece.
this.red = color.getRed(); *
this.green = color.getGreen(); * @param color the color of the leather armor
this.blue = color.getBlue(); */
this.modified = true; public void setColor(ChatColor color) {
// TODO using "CHAT color" is pretty weird, maybe that the class should be renamed to "Color"
this.red = color.getRed();
this.green = color.getGreen();
this.blue = color.getBlue();
this.modified = true;
}
/**
* Changes the color of the leather armor piece.
*
* @param red The red color of the leather armor piece.
* @param green The green color of the leather armor piece.
* @param blue The blue color of the leather armor piece.
*/
public void setColor(byte red, byte green, byte blue) {
this.red = red;
this.green = green;
this.blue = blue;
this.modified = true;
}
/** Resets the color to the default leather one. */
public void reset() {
this.red = 0;
this.green = 0;
this.blue = 0;
this.modified = false;
}
/**
* Gets the red component.
*
* @return the red component
*/
public int getRed() {
return BIT_MASK & red;
}
/**
* Gets the green component.
*
* @return the green component
*/
public int getGreen() {
return BIT_MASK & green;
}
/**
* Gets the blue component.
*
* @return the blue component
*/
public int getBlue() {
return BIT_MASK & blue;
}
/**
* Gets if the color of this armor piece have been changed.
*
* @return true if the color has been changed, false otherwise
*/
public boolean isModified() {
return modified;
}
/** {@inheritDoc} */
@Override
public boolean hasNbt() {
return modified;
}
/** {@inheritDoc} */
@Override
public boolean isSimilar(@NotNull ItemMeta itemMeta) {
if (!(itemMeta instanceof LeatherArmorMeta)) return false;
final LeatherArmorMeta leatherArmorMeta = (LeatherArmorMeta) itemMeta;
return leatherArmorMeta.isModified() == isModified()
&& leatherArmorMeta.getRed() == getRed()
&& leatherArmorMeta.getGreen() == getGreen()
&& leatherArmorMeta.getBlue() == getBlue();
}
/** {@inheritDoc} */
@Override
public void read(@NotNull NBTCompound compound) {
if (compound.containsKey("display")) {
final NBTCompound nbtCompound = compound.getCompound("display");
if (nbtCompound.containsKey("color")) {
final int color = nbtCompound.getInt("color");
// Sets the color of the leather armor piece
// This also fixes that the armor pieces do not decolorize again when you are in creative
// mode.
this.setColor(
(byte) ((color >> 16) & BIT_MASK),
(byte) ((color >> 8) & BIT_MASK),
(byte) ((color) & BIT_MASK));
}
} }
}
/** /** {@inheritDoc} */
* Resets the color to the default leather one. @Override
*/ public void write(@NotNull NBTCompound compound) {
public void reset() { if (modified) {
this.red = 0; NBTCompound displayCompound;
this.green = 0; if (!compound.containsKey("display")) {
this.blue = 0; displayCompound = new NBTCompound();
this.modified = false; } else {
displayCompound = compound.getCompound("display");
}
final int color = this.getRed() << 16 | this.getGreen() << 8 | this.getBlue();
displayCompound.setInt("color", color);
// Adds the color compound to the display compound
compound.set("display", displayCompound);
} }
}
/** /** {@inheritDoc} */
* Gets the red component. @NotNull
* @Override
* @return the red component public ItemMeta copy() {
*/ LeatherArmorMeta leatherArmorMeta = new LeatherArmorMeta();
public int getRed() { leatherArmorMeta.modified = this.isModified();
return red; leatherArmorMeta.red = (byte) this.getRed();
} leatherArmorMeta.green = (byte) this.getGreen();
leatherArmorMeta.blue = (byte) this.getBlue();
/** return leatherArmorMeta;
* Gets the green component. }
*
* @return the green component
*/
public int getGreen() {
return green;
}
/**
* Gets the blue component.
*
* @return the blue component
*/
public int getBlue() {
return blue;
}
/**
* Gets if the color of this armor piece have been changed.
*
* @return true if the color has been changed, false otherwise
*/
public boolean isModified() {
return modified;
}
@Override
public boolean hasNbt() {
return modified;
}
@Override
public boolean isSimilar(@NotNull ItemMeta itemMeta) {
if (!(itemMeta instanceof LeatherArmorMeta))
return false;
final LeatherArmorMeta leatherArmorMeta = (LeatherArmorMeta) itemMeta;
return leatherArmorMeta.modified == modified &&
leatherArmorMeta.red == red &&
leatherArmorMeta.green == green &&
leatherArmorMeta.blue == blue;
}
@Override
public void read(@NotNull NBTCompound compound) {
if (compound.containsKey("display")) {
final NBTCompound nbtCompound = compound.getCompound("display");
if (nbtCompound.containsKey("color")) {
final int color = nbtCompound.getInt("color");
this.red = (byte) ((color >> 16) & 0x000000FF);
this.green = (byte) ((color >> 8) & 0x000000FF);
this.blue = (byte) ((color) & 0x000000FF);
}
}
}
@Override
public void write(@NotNull NBTCompound compound) {
if (modified) {
NBTCompound displayCompound;
if (!compound.containsKey("display")) {
displayCompound = new NBTCompound();
} else {
displayCompound = compound.getCompound("display");
}
final int color = red << 16 + green << 8 + blue;
displayCompound.setInt("color", color);
}
}
@NotNull
@Override
public ItemMeta copy() {
LeatherArmorMeta leatherArmorMeta = new LeatherArmorMeta();
leatherArmorMeta.modified = modified;
leatherArmorMeta.red = red;
leatherArmorMeta.green = green;
leatherArmorMeta.blue = blue;
return leatherArmorMeta;
}
} }