Method my_unit_mut

battlecode-engine/src/world.rs:798–810

Gets a mutable version of this unit from space or the current planet. Checks that its team is the same as the current team. NoSuchUnit - the unit does not exist (inside the vision range). TeamNotAllowed - the unit is not on the current player's team.

(&mut self, id: UnitID)

    /// * NoSuchUnit - the unit does not exist (inside the vision range).
    /// * TeamNotAllowed - the unit is not on the current player's team.
    fn my_unit_mut(&mut self, id: UnitID) -> Result<&mut Unit, GameError> {
        if self.my_unit(id)?.team() == self.team() {
            if self.my_planet().units.contains_key(&id) {
                Ok(self.my_planet_mut().units.get_mut(&id).expect("key exists"))
            } else if self.my_team().units_in_space.contains_key(&id) {
                Ok(self.my_team_mut().units_in_space.get_mut(&id).expect("key exists"))
            } else {
                Err(GameError::NoSuchUnit)?
            }
        } else {
            Err(GameError::TeamNotAllowed)?
        }
    }

    /// Gets this unit from space or the current planet.
    ///