bifrost

type Arc struct {
	Target   uint64
	Distance uint32 // in ms
}

type Bifrost struct {
	TransferLimit             int
	TransferPaddingMs         uint64  // only search for trips, padded a bit after transitioning
	WalkingSpeed              float64 // in meters per ms
	MaxWalkingMs              uint32  // duration of walks not allowed to be higher than this when transferring
	MaxStopsConnectionSeconds uint32  // max length of added arcs between stops and street graph in deciseconds

	Data *RoutingData
}

type GeoPoint struct {
	Latitude  float64
	Longitude float64
	VertKey   uint64
}

type LoadOptions struct {
	OsmPaths    []string // paths to osm pbf files
	GtfsPaths   []string // path to GTFS zip files
	BifrostPath string   // path to bifrost cache
}

type Progress struct {
	Start     time.Time
	LastPrint time.Time
	Current   uint64
	Total     uint64
}

type Rounds struct {
	Rounds                 []map[uint64]StopArrival
	MarkedStops            map[uint64]bool
	MarkedStopsForTransfer map[uint64]bool
	EarliestArrivals       map[uint64]uint64
	Queue                  map[uint32]uint32
}

type Route struct {
	Stops []uint64
	Trips []uint32
}

type RouteInformation struct {
	ShortName string
}

type RoutingData struct {
	MaxTripDayLength uint32 `json:"maxTripDayLength"` // number of days to go backwards in time (for trips that end after midnight or multiple days later than the start)

	Services []*Service `json:"services"`

	Routes       []*Route          `json:"routes"`
	StopToRoutes [][]StopRoutePair `json:"stopToRoutes"`
	Trips        []*Trip           `json:"trips"`
	StreetGraph  [][]Arc           `json:"streetGraph"`

	Reorders map[uint64][]uint32 `json:"reorders"`

	// for reconstructing journeys after routing
	Vertices         []Vertex            `json:"vertices"`
	StopsIndex       map[string]uint64   `json:"stopsIndex"`     // gtfs stop id -> vertex index
	NodesIndex       map[int64]uint64    `json:"nodesIndex"`     // csv vertex id -> vertex index
	GtfsRouteIndex   []uint32            `json:"gtfsRouteIndex"` // route index -> gtfs route index
	RouteInformation []*RouteInformation `json:"routeInformation"`
	TripInformation  []*TripInformation  `json:"tripInformation"`
	TripToRoute      []uint32            `json:"tripToRoute"` // trip index -> route index

	// for finding vertices by location. points are GeoPoint
	VertexTree *kdtree.KDTree `json:"-"`
}

type Service struct {
	Weekdays uint8  // bitfield, 1 << 0 = monday, 1 << 6 = sunday
	StartDay uint32 // day relative to PivotDate
	EndDay   uint32 // day relative to PivotDate

	AddedExceptions   []uint32 // unix days
	RemovedExceptions []uint32 // unix days
}

type SourceKey struct {
	StopKey   uint64    // stop key in RoutingData
	Departure time.Time // departure time
}

type SourceLocation struct {
	Location  *fptf.Location
	Departure time.Time
}

type StopArrival struct {
	Arrival uint64 // arrival time in unix ms
	Trip    uint32 // trip id, 0xffffffff specifies a transfer, 0xfffffffe specifies no change compared to previous round

	EnterKey  uint64 // stop sequence key in route for trips, vertex key for transfers
	Departure uint64 // departure day for trips, departure time in unix ms for transfers
}

type StopContext struct {
	Id   string
	Name string
}

type StopRoutePair struct {
	Route         uint32
	StopKeyInTrip uint32
}

type Stopover struct {
	Arrival   uint32 // ms time since start of day
	Departure uint32 // ms time since start of day
}

type Trip struct {
	Service   uint32
	StopTimes []Stopover
}

type TripInformation struct {
	Headsign string
}

type Vertex struct {
	Longitude float64
	Latitude  float64
	Stop      *StopContext
}