aboutsummaryrefslogtreecommitdiff
path: root/src/Costasdev.Busurbano.Backend/Services/Processors/ShapeProcessor.cs
diff options
context:
space:
mode:
Diffstat (limited to 'src/Costasdev.Busurbano.Backend/Services/Processors/ShapeProcessor.cs')
-rw-r--r--src/Costasdev.Busurbano.Backend/Services/Processors/ShapeProcessor.cs97
1 files changed, 97 insertions, 0 deletions
diff --git a/src/Costasdev.Busurbano.Backend/Services/Processors/ShapeProcessor.cs b/src/Costasdev.Busurbano.Backend/Services/Processors/ShapeProcessor.cs
new file mode 100644
index 0000000..300ce70
--- /dev/null
+++ b/src/Costasdev.Busurbano.Backend/Services/Processors/ShapeProcessor.cs
@@ -0,0 +1,97 @@
+using Costasdev.Busurbano.Backend.GraphClient.App;
+
+namespace Costasdev.Busurbano.Backend.Services.Processors;
+
+public class ShapeProcessor : IArrivalsProcessor
+{
+ private readonly ILogger<ShapeProcessor> _logger;
+
+ public ShapeProcessor(ILogger<ShapeProcessor> logger)
+ {
+ _logger = logger;
+ }
+
+ public Task ProcessAsync(ArrivalsContext context)
+ {
+ if (context.IsReduced)
+ {
+ return Task.CompletedTask;
+ }
+
+ foreach (var arrival in context.Arrivals)
+ {
+ if (arrival.RawOtpTrip is not ArrivalsAtStopResponse.Arrival otpArrival) continue;
+
+ var encodedPoints = otpArrival.Trip.Geometry?.Points;
+ if (string.IsNullOrEmpty(encodedPoints))
+ {
+ _logger.LogDebug("No geometry found for trip {TripId}", arrival.TripId);
+ continue;
+ }
+
+ try
+ {
+ var points = Decode(encodedPoints);
+ if (points.Count == 0) continue;
+
+ arrival.Shape = new
+ {
+ type = "Feature",
+ geometry = new
+ {
+ type = "LineString",
+ coordinates = points.Select(p => new[] { p.Lon, p.Lat }).ToList()
+ }
+ };
+ }
+ catch (Exception ex)
+ {
+ _logger.LogError(ex, "Error decoding shape for trip {TripId}", arrival.TripId);
+ }
+ }
+
+ return Task.CompletedTask;
+ }
+
+ private static List<(double Lat, double Lon)> Decode(string encodedPoints)
+ {
+ var poly = new List<(double, double)>();
+ char[] polylineChars = encodedPoints.ToCharArray();
+ int index = 0;
+
+ int currentLat = 0;
+ int currentLng = 0;
+ int next5bits;
+ int sum;
+ int shifter;
+
+ while (index < polylineChars.Length)
+ {
+ sum = 0;
+ shifter = 0;
+ do
+ {
+ next5bits = (int)polylineChars[index++] - 63;
+ sum |= (next5bits & 31) << shifter;
+ shifter += 5;
+ } while (next5bits >= 32 && index < polylineChars.Length);
+
+ currentLat += (sum & 1) == 1 ? ~(sum >> 1) : (sum >> 1);
+
+ sum = 0;
+ shifter = 0;
+ do
+ {
+ next5bits = (int)polylineChars[index++] - 63;
+ sum |= (next5bits & 31) << shifter;
+ shifter += 5;
+ } while (next5bits >= 32 && index < polylineChars.Length);
+
+ currentLng += (sum & 1) == 1 ? ~(sum >> 1) : (sum >> 1);
+
+ poly.Add((Convert.ToDouble(currentLat) / 100000.0, Convert.ToDouble(currentLng) / 100000.0));
+ }
+
+ return poly;
+ }
+}