import carla
import time
import argparse

# town_idx = '13'

def load_map(town_idx):
    try:
        client = carla.Client("localhost", 2000)
        client.set_timeout(120.0)

        world = client.get_world()
        print("✅ CARLA запущена, текущая карта:", world.get_map().name)

        print(f"Загружаю Town{town_idx}...")
        world = client.load_world(f'Town{town_idx}')

        time.sleep(5.0)
        world = client.get_world()  

        settings = world.get_settings()
        settings.tile_stream_distance  = 800.0
        settings.actor_active_distance = 600.0
        world.apply_settings(settings)

        print(f"✅ Town{town_idx} загружена")
        print("Карта:", world.get_map().name)

    except Exception as e:
        print("❌ Ошибка:", e)


def parse_args():
    p = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)

    p.add_argument('--map_idx', default='01')

    return p.parse_args()

if __name__ == "__main__":
    args = parse_args()
    load_map(town_idx=args.map_idx)