some custom stuff

2025-11-15 00:23:48 +01:00
parent 4dbbd30d4d
commit c189e3f59a
19 changed files with 1116 additions and 831 deletions
--- a/pkg/reader/loader.go
+++ b/pkg/reader/loader.go
@@ -0,0 +1,103 @@
+package reader
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+
+	"git.tornberg.me/go-gtfs/pkg/types"
+)
+
+type TripData struct {
+	Stops    map[string]*types.Stop
+	Trips    map[string]*types.Trip
+	Routes   map[string]*types.Route
+	Agencies map[string]*types.Agency
+}
+
+func LoadTripData(path string) (*TripData, error) {
+	tp := &TripData{
+		Stops:    make(map[string]*types.Stop),
+		Trips:    make(map[string]*types.Trip),
+		Routes:   make(map[string]*types.Route),
+		Agencies: make(map[string]*types.Agency),
+	}
+
+	files := []string{"agency", "routes", "stops", "trips", "stop_times", "transfers"}
+	for _, file := range files {
+
+		f, err := os.Open(filepath.Join(path, file+".txt"))
+		if err != nil {
+			log.Fatalf("failed to open %s: %v", file, err)
+		}
+		defer f.Close()
+
+		switch file {
+		case "agency":
+			err = ParseAgencies(f, func(a types.Agency) {
+				tp.Agencies[a.AgencyID] = &a
+			})
+		case "routes":
+			err = ParseRoutes(f, func(r types.Route) {
+				tp.Routes[r.RouteID] = &r
+				if ag, ok := tp.Agencies[r.AgencyID]; ok {
+					r.Agency = ag
+					ag.AddRoute(&r)
+				}
+			})
+		case "stops":
+			err = ParseStops(f, func(s types.Stop) {
+				tp.Stops[s.StopId] = &s
+			})
+		case "trips":
+			err = ParseTrips(f, func(t types.Trip) {
+				trip := t
+				if route, ok := tp.Routes[trip.RouteId]; ok {
+					trip.SetRoute(route)
+					route.AddTrip(&trip)
+				} else {
+					log.Printf("route %s not found", trip.RouteId)
+				}
+				if agency, ok := tp.Agencies[trip.AgencyID]; ok {
+					trip.Agency = agency
+				} else {
+					log.Printf("agency %s not found", trip.AgencyID)
+				}
+				tp.Trips[trip.TripId] = &trip
+			})
+		case "stop_times":
+			err = ParseStopTimes(f, func(st types.StopTime) {
+				stop, ok := tp.Stops[st.StopId]
+				if ok {
+					st.SetStop(stop)
+				} else {
+					log.Printf("stop %s not found", st.StopId)
+				}
+				trp, ok := tp.Trips[st.TripId]
+				if !ok {
+					log.Printf("trip %s not found", st.TripId)
+
+				} else {
+					stop.AddTrip(trp)
+					trp.AddStopTime(&st)
+				}
+
+			})
+		case "transfers":
+			err = ParseTransfers(f, func(tr types.Transfer) {
+				//transfers = append(transfers, tr)
+				stop, ok := tp.Stops[tr.FromStopId]
+				if ok {
+					stop.AddTransfer(&tr)
+				} else {
+					log.Printf("stop %s not found for transfer", tr.FromStopId)
+				}
+
+			})
+		}
+		if err != nil {
+			return tp, err
+		}
+	}
+	return tp, nil
+}