Method get_unit_mut

battlecode-engine/src/world.rs:832–844

Gets a mutable version of this unit from space or the current planet. NoSuchUnit - the unit does not exist.

(&mut self, id: UnitID)

    ///
    /// * NoSuchUnit - the unit does not exist.
    fn get_unit_mut(&mut self, id: UnitID) -> Result<&mut Unit, GameError> {
        if self.get_planet(Planet::Earth).units.contains_key(&id) {
            Ok(self.get_planet_mut(Planet::Earth).units.get_mut(&id).expect("key exists"))
        } else if self.get_planet(Planet::Mars).units.contains_key(&id) {
            Ok(self.get_planet_mut(Planet::Mars).units.get_mut(&id).expect("key exists"))
        } else if self.get_team(Team::Red).units_in_space.contains_key(&id) {
            Ok(self.get_team_mut(Team::Red).units_in_space.get_mut(&id).expect("key exists"))
        } else if self.get_team(Team::Blue).units_in_space.contains_key(&id) {
            Ok(self.get_team_mut(Team::Blue).units_in_space.get_mut(&id).expect("key exists"))
        } else {
            Err(GameError::NoSuchUnit)?
        }
    }

    // ************************************************************************
    // **************** UNIT CREATION / DESTRUCTION METHODS *******************