import ferrobus
import time

# Create a transit model from OpenStreetMap and GTFS data
model = ferrobus.create_transit_model(
    osm_path="path/to/city.osm.pbf", # OSM pbf file, preferably filtered by region and tag
    gtfs_dirs=["path/to/gtfs_data", "path/to/another_gtfs"], # feeds, operating in the same region
    date=datetime.date.today() # date to use when filtering GTFS data / None for all dates
)

# Initializes and pre-calculates transit points for efficient geographic operations.
#
# Transit points represent specific geographic locations and should be used as inputs for all route calculations and related
# operations. By creating and reusing these pre-initialized points, the system avoids redundant computations of geographic
# positions, resulting in significantly improved performance for routing and spatial queries.
origin = ferrobus.create_transit_point(52.52, 13.40, model)
destination = ferrobus.create_transit_point(52.53, 13.42, model)_

# Find route (departure at noon)
departure_time = 12 * 3600  # 12:00 noon in seconds since midnight
start_time = time.perf_counter()
route = ferrobus.find_route(
    start_point=origin,
    end_point=destination,
    departure_time=departure_time,
    max_transfers=3  # Allow up to 3 transfers
)
end_time = time.perf_counter()

# Display route information
print(f"Route found in {end_time - start_time:.3f} seconds")
print(f"Travel time: {route['travel_time_seconds'] / 60:.1f} minutes")
print(f"Transit time: {route['transit_time_seconds'] / 60:.1f} minutes")
print(f"Walking time: {route['walking_time_seconds'] / 60:.1f} minutes")
print(f"Number of transfers: {route['transfers']}")

# Get detailed journey information with all legs (walking, transit)
journey = ferrobus.detailed_journey(
    transit_model=model,
    start_point=origin,
    end_point=destination,
    departure_time=departure_time,
    max_transfers=3
)

# Calculate travel times between multiple points
points = [origin, destination, point3, point4]
matrix = ferrobus.travel_time_matrix(
    transit_model=model,
    points=points,
    departure_time=departure_time,
    max_transfers=3
)

# Create an isochrone index for a specific area
index = ferrobus.create_isochrone_index(model, area_wkt, 8)

# Calculate isochrone (areas reachable within 30 minutes)
isochrone = ferrobus.calculate_isochrone(
    transit_model=model,
    origin=origin,
    departure_time=departure_time,
    max_transfers=2,
    max_travel_time=1800,  # 30 minutes in seconds
    isochrone_index=index
)