Astronand/Mindustry
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Select command from reconstructor / Units save command when controlled

Browse Source
This commit is contained in:
Anuken
2023-05-30 20:08:48 -04:00
parent 7911a22cc9
commit b6f5d424da
8 changed files with 103 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
core/src/mindustry/ai/UnitCommand.java
View File

@@ -2,6 +2,7 @@ package mindustry.ai;
import arc.*;
import arc.func.*;
import arc.scene.style.*;
import arc.struct.*;
import mindustry.ai.types.*;
import mindustry.entities.units.*;
@@ -60,6 +61,10 @@ public class UnitCommand{
return Core.bundle.get("command." + name);
}
public TextureRegionDrawable getIcon(){
return Icon.icons.get(icon, Icon.cancel);
}
@Override
public String toString(){
return "UnitCommand:" + name;

13
core/src/mindustry/entities/comp/PlayerComp.java
View File

@@ -8,6 +8,8 @@ import arc.scene.ui.layout.*;
import arc.util.*;
import arc.util.pooling.*;
import mindustry.*;
import mindustry.ai.*;
import mindustry.ai.types.*;
import mindustry.annotations.Annotations.*;
import mindustry.content.*;
import mindustry.entities.units.*;
@@ -36,6 +38,8 @@ abstract class PlayerComp implements UnitController, Entityc, Syncc, Timerc, Dra
@ReadOnly Team team = Team.sharded;
@SyncLocal boolean typing, shooting, boosting;
@SyncLocal float mouseX, mouseY;
/** command the unit had before it was controlled. */
@Nullable @NoSync UnitCommand lastCommand;
boolean admin;
String name = "frog";
Color color = new Color();
@@ -203,9 +207,18 @@ abstract class PlayerComp implements UnitController, Entityc, Syncc, Timerc, Dra
if(unit == null) throw new IllegalArgumentException("Unit cannot be null. Use clearUnit() instead.");
if(this.unit == unit) return;
//save last command this unit had
if(unit.controller() instanceof CommandAI ai){
lastCommand = ai.command;
}
if(this.unit != Nulls.unit){
//un-control the old unit
this.unit.resetController();
//restore last command issued before it was controlled
if(lastCommand != null && this.unit.controller() instanceof CommandAI ai){
ai.command(lastCommand);
}
}
this.unit = unit;
if(unit != Nulls.unit){

9
core/src/mindustry/io/TypeIO.java
View File

@@ -301,12 +301,13 @@ public class TypeIO{
return Nulls.unit;
}
public static void writeCommand(Writes write, UnitCommand command){
write.b(command.id);
public static void writeCommand(Writes write, @Nullable UnitCommand command){
write.b(command == null ? 255 : command.id);
}
public static UnitCommand readCommand(Reads read){
return UnitCommand.all.get(read.ub());
public static @Nullable UnitCommand readCommand(Reads read){
int val = read.ub();
return val == 255 ? null : UnitCommand.all.get(val);
}
public static void writeEntity(Writes write, Entityc entity){

71
core/src/mindustry/world/blocks/units/Reconstructor.java
View File

@@ -1,13 +1,18 @@
package mindustry.world.blocks.units;
import arc.*;
import arc.Graphics.*;
import arc.Graphics.Cursor.*;
import arc.graphics.g2d.*;
import arc.math.*;
import arc.math.geom.*;
import arc.scene.ui.*;
import arc.scene.ui.layout.*;
import arc.struct.*;
import arc.util.*;
import arc.util.io.*;
import mindustry.*;
import mindustry.ai.*;
import mindustry.content.*;
import mindustry.entities.*;
import mindustry.entities.units.*;
@@ -35,6 +40,7 @@ public class Reconstructor extends UnitBlock{
regionRotated2 = 2;
commandable = true;
ambientSound = Sounds.respawning;
configurable = true;
}
@Override
@@ -132,6 +138,7 @@ public class Reconstructor extends UnitBlock{
public class ReconstructorBuild extends UnitBuild{
public @Nullable Vec2 commandPos;
public @Nullable UnitCommand command;
public float fraction(){
return progress / constructTime;
@@ -157,6 +164,51 @@ public class Reconstructor extends UnitBlock{
return hasUpgrade(unit.type) && !upgrade(unit.type).isBanned();
}
public boolean canSetCommand(){
var output = unit();
return output != null && output.commands.length > 1;
}
@Override
public Cursor getCursor(){
return canSetCommand() ? super.getCursor() : SystemCursor.arrow;
}
@Override
public boolean shouldShowConfigure(Player player){
return canSetCommand();
}
@Override
public void buildConfiguration(Table table){
var unit = unit();
if(unit == null){
deselect();
return;
}
var group = new ButtonGroup<ImageButton>();
group.setMinCheckCount(0);
int i = 0, columns = 4;
table.background(Styles.black6);
var list = unit().commands;
for(var item : list){
ImageButton button = table.button(item.getIcon(), Styles.clearNoneTogglei, 40f, () -> {
command = (command == item ? null : item);
deselect();
}).tooltip(item.localized()).group(group).get();
button.update(() -> button.setChecked(command == item));
if(++i % columns == 0){
table.row();
}
}
}
@Override
public boolean acceptPayload(Building source, Payload payload){
if(!(this.payload == null
@@ -252,9 +304,17 @@ public class Reconstructor extends UnitBlock{
//upgrade the unit
if(progress >= constructTime){
payload.unit = upgrade(payload.unit.type).create(payload.unit.team());
if(commandPos != null && payload.unit.isCommandable()){
payload.unit.command().commandPosition(commandPos);
if(payload.unit.isCommandable()){
if(commandPos != null){
payload.unit.command().commandPosition(commandPos);
}
if(command != null){
//this already checks if it is a valid command for the unit type
payload.unit.command().command(command);
}
}
progress %= 1f;
Effect.shake(2f, 3f, this);
Fx.producesmoke.at(this);
@@ -303,7 +363,7 @@ public class Reconstructor extends UnitBlock{
@Override
public byte version(){
return 2;
return 3;
}
@Override
@@ -312,6 +372,7 @@ public class Reconstructor extends UnitBlock{
write.f(progress);
TypeIO.writeVecNullable(write, commandPos);
TypeIO.writeCommand(write, command);
}
@Override
@@ -325,6 +386,10 @@ public class Reconstructor extends UnitBlock{
if(revision >= 2){
commandPos = TypeIO.readVecNullable(read);
}
if(revision >= 3){
command = TypeIO.readCommand(read);
}
}
}