package emu.grasscutter.game.player; import java.util.*; import dev.morphia.annotations.Entity; import dev.morphia.annotations.Transient; import emu.grasscutter.GameConstants; import emu.grasscutter.Grasscutter; import emu.grasscutter.data.def.AvatarSkillDepotData; import emu.grasscutter.game.avatar.Avatar; import emu.grasscutter.game.entity.EntityAvatar; import emu.grasscutter.game.entity.EntityBaseGadget; import emu.grasscutter.game.props.ElementType; import emu.grasscutter.game.props.EnterReason; import emu.grasscutter.game.props.FightProperty; import emu.grasscutter.game.world.World; import emu.grasscutter.net.packet.BasePacket; import emu.grasscutter.net.packet.PacketOpcodes; import emu.grasscutter.net.proto.EnterTypeOuterClass.EnterType; import emu.grasscutter.net.proto.MotionStateOuterClass.MotionState; import emu.grasscutter.net.proto.PlayerDieTypeOuterClass.PlayerDieType; import emu.grasscutter.server.packet.send.PacketAvatarDieAnimationEndRsp; import emu.grasscutter.server.packet.send.PacketAvatarFightPropUpdateNotify; import emu.grasscutter.server.packet.send.PacketAvatarLifeStateChangeNotify; import emu.grasscutter.server.packet.send.PacketAvatarSkillInfoNotify; import emu.grasscutter.server.packet.send.PacketAvatarTeamUpdateNotify; import emu.grasscutter.server.packet.send.PacketChangeAvatarRsp; import emu.grasscutter.server.packet.send.PacketChangeMpTeamAvatarRsp; import emu.grasscutter.server.packet.send.PacketChangeTeamNameRsp; import emu.grasscutter.server.packet.send.PacketChooseCurAvatarTeamRsp; import emu.grasscutter.server.packet.send.PacketPlayerEnterSceneNotify; import emu.grasscutter.server.packet.send.PacketSceneTeamUpdateNotify; import emu.grasscutter.server.packet.send.PacketSetUpAvatarTeamRsp; import emu.grasscutter.server.packet.send.PacketWorldPlayerDieNotify; import it.unimi.dsi.fastutil.ints.Int2IntMap; import it.unimi.dsi.fastutil.ints.Int2IntOpenHashMap; import it.unimi.dsi.fastutil.ints.Int2ObjectMap; import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap; import it.unimi.dsi.fastutil.ints.IntOpenHashSet; import it.unimi.dsi.fastutil.ints.IntSet; @Entity public class TeamManager { @Transient private Player player; private Map teams; private int currentTeamIndex; private int currentCharacterIndex; @Transient private TeamInfo mpTeam; @Transient private int entityId; @Transient private final List avatars; @Transient private final Set gadgets; @Transient private final IntSet teamResonances; @Transient private final IntSet teamResonancesConfig; @Transient private int useTemporarilyTeamIndex = -1; /** * Temporary Team for tower */ @Transient private List temporaryTeam; public TeamManager() { this.mpTeam = new TeamInfo(); this.avatars = new ArrayList<>(); this.gadgets = new HashSet<>(); this.teamResonances = new IntOpenHashSet(); this.teamResonancesConfig = new IntOpenHashSet(); } public TeamManager(Player player) { this(); this.player = player; this.teams = new HashMap<>(); this.currentTeamIndex = 1; for (int i = 1; i <= GameConstants.MAX_TEAMS; i++) { this.teams.put(i, new TeamInfo()); } } public Player getPlayer() { return player; } public World getWorld() { return player.getWorld(); } public void setPlayer(Player player) { this.player = player; } public Map getTeams() { return this.teams; } public TeamInfo getMpTeam() { return mpTeam; } public void setMpTeam(TeamInfo mpTeam) { this.mpTeam = mpTeam; } public int getCurrentTeamId() { // Starts from 1 return currentTeamIndex; } private void setCurrentTeamId(int currentTeamIndex) { this.currentTeamIndex = currentTeamIndex; } public int getCurrentCharacterIndex() { return currentCharacterIndex; } public void setCurrentCharacterIndex(int currentCharacterIndex) { this.currentCharacterIndex = currentCharacterIndex; } public long getCurrentCharacterGuid() { return getCurrentAvatarEntity().getAvatar().getGuid(); } public TeamInfo getCurrentTeamInfo() { if (useTemporarilyTeamIndex >= 0 && useTemporarilyTeamIndex < temporaryTeam.size()){ return temporaryTeam.get(useTemporarilyTeamIndex); } if (this.getPlayer().isInMultiplayer()) { return this.getMpTeam(); } return this.getTeams().get(this.currentTeamIndex); } public TeamInfo getCurrentSinglePlayerTeamInfo() { return this.getTeams().get(this.currentTeamIndex); } public int getEntityId() { return entityId; } public void setEntityId(int entityId) { this.entityId = entityId; } public Set getGadgets() { return gadgets; } public IntSet getTeamResonances() { return teamResonances; } public IntSet getTeamResonancesConfig() { return teamResonancesConfig; } public List getActiveTeam() { return avatars; } public EntityAvatar getCurrentAvatarEntity() { return getActiveTeam().get(currentCharacterIndex); } public boolean isSpawned() { return getPlayer().getScene() != null && getPlayer().getScene().getEntities().containsKey(getCurrentAvatarEntity().getId()); } public int getMaxTeamSize() { if (getPlayer().isInMultiplayer()) { int max = Grasscutter.getConfig().getGameServerOptions().MaxAvatarsInTeamMultiplayer; if (getPlayer().getWorld().getHost() == this.getPlayer()) { return Math.max(1, (int) Math.ceil(max / (double) getWorld().getPlayerCount())); } return Math.max(1, (int) Math.floor(max / (double) getWorld().getPlayerCount())); } return Grasscutter.getConfig().getGameServerOptions().MaxAvatarsInTeam; } // Methods private void updateTeamResonances() { Int2IntOpenHashMap map = new Int2IntOpenHashMap(); this.getTeamResonances().clear(); this.getTeamResonancesConfig().clear(); for (EntityAvatar entity : getActiveTeam()) { AvatarSkillDepotData skillData = entity.getAvatar().getAvatarData().getSkillDepot(); if (skillData != null) { map.addTo(skillData.getElementType().getValue(), 1); } } for (Int2IntMap.Entry e : map.int2IntEntrySet()) { if (e.getIntValue() >= 2) { ElementType element = ElementType.getTypeByValue(e.getIntKey()); if (element.getTeamResonanceId() != 0) { this.getTeamResonances().add(element.getTeamResonanceId()); this.getTeamResonancesConfig().add(element.getConfigHash()); } } } // No resonances if (this.getTeamResonances().size() == 0) { this.getTeamResonances().add(ElementType.Default.getTeamResonanceId()); this.getTeamResonancesConfig().add(ElementType.Default.getTeamResonanceId()); } } public void updateTeamEntities(BasePacket responsePacket) { // Sanity check - Should never happen if (this.getCurrentTeamInfo().getAvatars().size() <= 0) { return; } // If current team has changed EntityAvatar currentEntity = this.getCurrentAvatarEntity(); Int2ObjectMap existingAvatars = new Int2ObjectOpenHashMap<>(); int prevSelectedAvatarIndex = -1; for (EntityAvatar entity : getActiveTeam()) { existingAvatars.put(entity.getAvatar().getAvatarId(), entity); } // Clear active team entity list this.getActiveTeam().clear(); // Add back entities into team for (int i = 0; i < this.getCurrentTeamInfo().getAvatars().size(); i++) { int avatarId = this.getCurrentTeamInfo().getAvatars().get(i); EntityAvatar entity = null; if (existingAvatars.containsKey(avatarId)) { entity = existingAvatars.get(avatarId); existingAvatars.remove(avatarId); if (entity == currentEntity) { prevSelectedAvatarIndex = i; } } else { entity = new EntityAvatar(getPlayer().getScene(), getPlayer().getAvatars().getAvatarById(avatarId)); } this.getActiveTeam().add(entity); } // Unload removed entities for (EntityAvatar entity : existingAvatars.values()) { getPlayer().getScene().removeEntity(entity); entity.getAvatar().save(); } // Set new selected character index if (prevSelectedAvatarIndex == -1) { // Previous selected avatar is not in the same spot, we will select the current one in the prev slot prevSelectedAvatarIndex = Math.min(this.currentCharacterIndex, this.getActiveTeam().size() - 1); } this.currentCharacterIndex = prevSelectedAvatarIndex; // Update team resonances updateTeamResonances(); // Packets getPlayer().getWorld().broadcastPacket(new PacketSceneTeamUpdateNotify(getPlayer())); // Skill charges packet - Yes, this is official server behavior as of 2.6.0 for (EntityAvatar entity : getActiveTeam()) { if (entity.getAvatar().getSkillExtraChargeMap().size() > 0) { getPlayer().sendPacket(new PacketAvatarSkillInfoNotify(entity.getAvatar())); } } // Run callback if (responsePacket != null) { getPlayer().sendPacket(responsePacket); } // Check if character changed if (currentEntity != getCurrentAvatarEntity()) { // Remove and Add getPlayer().getScene().replaceEntity(currentEntity, getCurrentAvatarEntity()); } } public synchronized void setupAvatarTeam(int teamId, List list) { // Sanity checks if (list.size() == 0 || list.size() > getMaxTeamSize() || getPlayer().isInMultiplayer()) { return; } // Get team TeamInfo teamInfo = this.getTeams().get(teamId); if (teamInfo == null) { return; } // Set team data LinkedHashSet newTeam = new LinkedHashSet<>(); for (int i = 0; i < list.size(); i++) { Avatar avatar = getPlayer().getAvatars().getAvatarByGuid(list.get(i)); if (avatar == null || newTeam.contains(avatar)) { // Should never happen return; } newTeam.add(avatar); } // Clear current team info and add avatars from our new team teamInfo.getAvatars().clear(); for (Avatar avatar : newTeam) { teamInfo.addAvatar(avatar); } // Update packet getPlayer().sendPacket(new PacketAvatarTeamUpdateNotify(getPlayer())); // Update entites if (teamId == this.getCurrentTeamId()) { this.updateTeamEntities(new PacketSetUpAvatarTeamRsp(getPlayer(), teamId, teamInfo)); } else { getPlayer().sendPacket(new PacketSetUpAvatarTeamRsp(getPlayer(), teamId, teamInfo)); } } public void setupMpTeam(List list) { // Sanity checks if (list.size() == 0 || list.size() > getMaxTeamSize() || !getPlayer().isInMultiplayer()) { return; } TeamInfo teamInfo = this.getMpTeam(); // Set team data LinkedHashSet newTeam = new LinkedHashSet<>(); for (int i = 0; i < list.size(); i++) { Avatar avatar = getPlayer().getAvatars().getAvatarByGuid(list.get(i)); if (avatar == null || newTeam.contains(avatar)) { // Should never happen return; } newTeam.add(avatar); } // Clear current team info and add avatars from our new team teamInfo.getAvatars().clear(); for (Avatar avatar : newTeam) { teamInfo.addAvatar(avatar); } // Packet this.updateTeamEntities(new PacketChangeMpTeamAvatarRsp(getPlayer(), teamInfo)); } public void setupTemporaryTeam(List> guidList) { var team = guidList.stream().map(list -> { // Sanity checks if (list.size() == 0 || list.size() > getMaxTeamSize()) { return null; } // Set team data LinkedHashSet newTeam = new LinkedHashSet<>(); for (int i = 0; i < list.size(); i++) { Avatar avatar = getPlayer().getAvatars().getAvatarByGuid(list.get(i)); if (avatar == null || newTeam.contains(avatar)) { // Should never happen return null; } newTeam.add(avatar); } // convert to avatar ids return newTeam.stream() .map(Avatar::getAvatarId) .toList(); }) .filter(Objects::nonNull) .map(TeamInfo::new) .toList(); this.temporaryTeam = team; } public void useTemporaryTeam(int index) { this.useTemporarilyTeamIndex = index; updateTeamEntities(null); } public void cleanTemporaryTeam() { // check if using temporary team if(useTemporarilyTeamIndex < 0){ return; } this.useTemporarilyTeamIndex = -1; this.temporaryTeam = null; updateTeamEntities(null); } public synchronized void setCurrentTeam(int teamId) { // if (getPlayer().isInMultiplayer()) { return; } // Get team TeamInfo teamInfo = this.getTeams().get(teamId); if (teamInfo == null || teamInfo.getAvatars().size() == 0) { return; } // Set this.setCurrentTeamId(teamId); this.updateTeamEntities(new PacketChooseCurAvatarTeamRsp(teamId)); } public synchronized void setTeamName(int teamId, String teamName) { // Get team TeamInfo teamInfo = this.getTeams().get(teamId); if (teamInfo == null) { return; } teamInfo.setName(teamName); // Packet getPlayer().sendPacket(new PacketChangeTeamNameRsp(teamId, teamName)); } public synchronized void changeAvatar(long guid) { EntityAvatar oldEntity = this.getCurrentAvatarEntity(); if (guid == oldEntity.getAvatar().getGuid()) { return; } EntityAvatar newEntity = null; int index = -1; for (int i = 0; i < getActiveTeam().size(); i++) { if (guid == getActiveTeam().get(i).getAvatar().getGuid()) { index = i; newEntity = getActiveTeam().get(i); } } if (index < 0 || newEntity == oldEntity) { return; } // Set index this.setCurrentCharacterIndex(index); // Old entity motion state oldEntity.setMotionState(MotionState.MOTION_STANDBY); // Remove and Add getPlayer().getScene().replaceEntity(oldEntity, newEntity); getPlayer().sendPacket(new PacketChangeAvatarRsp(guid)); } public void onAvatarDie(long dieGuid) { EntityAvatar deadAvatar = this.getCurrentAvatarEntity(); if (deadAvatar.isAlive() || deadAvatar.getId() != dieGuid) { return; } PlayerDieType dieType = deadAvatar.getKilledType(); int killedBy = deadAvatar.getKilledBy(); if (dieType == PlayerDieType.PLAYER_DIE_DRAWN) { // Died in water. Do not replace // The official server has skipped this notify and will just respawn the team immediately after the animation. // TODO: Perhaps find a way to get vanilla experience? getPlayer().sendPacket(new PacketWorldPlayerDieNotify(dieType, killedBy)); } else { // Replacement avatar EntityAvatar replacement = null; int replaceIndex = -1; for (int i = 0; i < this.getActiveTeam().size(); i++) { EntityAvatar entity = this.getActiveTeam().get(i); if (entity.isAlive()) { replaceIndex = i; replacement = entity; break; } } if (replacement == null) { // No more living team members... getPlayer().sendPacket(new PacketWorldPlayerDieNotify(dieType, killedBy)); } else { // Set index and spawn replacement member this.setCurrentCharacterIndex(replaceIndex); getPlayer().getScene().addEntity(replacement); } } // Response packet getPlayer().sendPacket(new PacketAvatarDieAnimationEndRsp(deadAvatar.getId(), 0)); } public boolean reviveAvatar(Avatar avatar) { for (EntityAvatar entity : getActiveTeam()) { if (entity.getAvatar() == avatar) { if (entity.isAlive()) { return false; } entity.setFightProperty( FightProperty.FIGHT_PROP_CUR_HP, entity.getFightProperty(FightProperty.FIGHT_PROP_MAX_HP) * .1f ); getPlayer().sendPacket(new PacketAvatarFightPropUpdateNotify(entity.getAvatar(), FightProperty.FIGHT_PROP_CUR_HP)); getPlayer().sendPacket(new PacketAvatarLifeStateChangeNotify(entity.getAvatar())); return true; } } return false; } public boolean healAvatar(Avatar avatar, int healRate, int healAmount) { for (EntityAvatar entity : getActiveTeam()) { if (entity.getAvatar() == avatar) { if (!entity.isAlive()) { return false; } entity.setFightProperty( FightProperty.FIGHT_PROP_CUR_HP, (float) Math.min( (entity.getFightProperty(FightProperty.FIGHT_PROP_CUR_HP) + entity.getFightProperty(FightProperty.FIGHT_PROP_MAX_HP) * (float) healRate / 100.0 + (float) healAmount / 100.0), entity.getFightProperty(FightProperty.FIGHT_PROP_MAX_HP) ) ); getPlayer().sendPacket(new PacketAvatarFightPropUpdateNotify(entity.getAvatar(), FightProperty.FIGHT_PROP_CUR_HP)); getPlayer().sendPacket(new PacketAvatarLifeStateChangeNotify(entity.getAvatar())); return true; } } return false; } public void respawnTeam() { // Make sure all team members are dead // Drowning needs revive when there may be other team members still alive. // for (EntityAvatar entity : getActiveTeam()) { // if (entity.isAlive()) { // return; // } // } player.getMovementManager().resetTimer(); // prevent drowning immediately after respawn // Revive all team members for (EntityAvatar entity : getActiveTeam()) { entity.setFightProperty( FightProperty.FIGHT_PROP_CUR_HP, entity.getFightProperty(FightProperty.FIGHT_PROP_MAX_HP) * .4f ); getPlayer().sendPacket(new PacketAvatarFightPropUpdateNotify(entity.getAvatar(), FightProperty.FIGHT_PROP_CUR_HP)); getPlayer().sendPacket(new PacketAvatarLifeStateChangeNotify(entity.getAvatar())); } // Teleport player getPlayer().sendPacket(new PacketPlayerEnterSceneNotify(getPlayer(), EnterType.ENTER_SELF, EnterReason.Revival, 3, GameConstants.START_POSITION)); // Set player position player.setSceneId(3); player.getPos().set(GameConstants.START_POSITION); // Packets getPlayer().sendPacket(new BasePacket(PacketOpcodes.WorldPlayerReviveRsp)); } public void saveAvatars() { // Save all avatars from active team for (EntityAvatar entity : getActiveTeam()) { entity.getAvatar().save(); } } }