File indexing completed on 2024-04-28 04:41:39

 /*
     SPDX-FileCopyrightText: 2021 Volker Krause <vkrause@kde.org>
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "uicrailwaycoach.h"
 
 #include <QDebug>
 
 using namespace KPublicTransport;
 
 QStringView UicRailwayCoach::countryCode(QStringView coachNumber)
 {
     // TODO handle different formatting of the coach number
     if (coachNumber.size() > 4) {
         return coachNumber.mid(2, 2);
     }
     return {};
 }
 
 static QStringView classificationDigit(QStringView coachNumber)
 {
     // TODO handle different formatting
     if (coachNumber.size() > 5) {
         return coachNumber.mid(4, 1);
     }
     return {};
 }
 
 // see https://en.wikipedia.org/wiki/UIC_classification_of_railway_coaches
 struct {
     const char prefix[5];
     VehicleSection::Classes classes;
     VehicleSection::Features features;
     VehicleSection::Type type;
     int deckCount;
 } static constexpr const class_prefix_table[] = {
     { "AB", VehicleSection::FirstClass | VehicleSection::SecondClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 1 },
     { "AR", VehicleSection::FirstClass, VehicleSection::Restaurant, VehicleSection::PassengerCar, 1 },
     { "A", VehicleSection::FirstClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 1 },
     { "BR", VehicleSection::SecondClass,  VehicleSection::Restaurant, VehicleSection::PassengerCar, 1 },
     { "B", VehicleSection::SecondClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 1 },
     { "DAB", VehicleSection::FirstClass | VehicleSection::SecondClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 2 },
     { "DA", VehicleSection::FirstClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 2 },
     { "DB", VehicleSection::SecondClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 2 },
     { "DD", VehicleSection::UnknownClass, VehicleSection::NoFeatures, VehicleSection::CarTransportCar, 2 },
     { "WLAB", VehicleSection::FirstClass | VehicleSection::SecondClass, VehicleSection::NoFeatures, VehicleSection::SleepingCar, 1 },
     { "WLA", VehicleSection::FirstClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 1 },
     { "WLB", VehicleSection::SecondClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 1 },
     { "WR", VehicleSection::UnknownClass, VehicleSection::Restaurant, VehicleSection::RestaurantCar, 1 },
     { "KA", VehicleSection::FirstClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 1 },
     { "KB", VehicleSection::SecondClass, VehicleSection::NoFeatures, VehicleSection::UnknownType, 1 },
 };
 
 VehicleSection::Classes UicRailwayCoach::coachClass(QStringView coachNumber, QStringView coachClassification)
 {
     const auto it = std::find_if(std::begin(class_prefix_table), std::end(class_prefix_table), [coachClassification](const auto &prefix) {
         return coachClassification.startsWith(QLatin1String(prefix.prefix));
     });
     if (it != std::end(class_prefix_table)) {
         return (*it).classes;
     }
 
     const auto cls = classificationDigit(coachNumber);
     if (!cls.empty()) {
         switch (cls.at(0).cell()) {
             case '1':
                 return VehicleSection::FirstClass;
             case '2':
             case '5':
                 return VehicleSection::SecondClass;
             case '3':
                 return VehicleSection::FirstClass | VehicleSection::SecondClass;
         }
     }
 
     return {};
 }
 
 // see https://de.wikipedia.org/wiki/Code_f%C3%BCr_das_Austauschverfahren
 struct {
     const char prefix[3];
     VehicleSection::Type type;
     VehicleSection::Features features;
 } static constexpr const number_prefix_table[] = {
     { "50", VehicleSection::PassengerCar, VehicleSection::NoFeatures },
     { "70", VehicleSection::PassengerCar, VehicleSection::AirConditioning },
     { "71", VehicleSection::SleepingCar, VehicleSection::NoFeatures },
     { "73", VehicleSection::PassengerCar, VehicleSection::AirConditioning },
     { "91", VehicleSection::Engine, VehicleSection::NoFeatures },
     { "92", VehicleSection::Engine, VehicleSection::NoFeatures },
 };
 
 // see https://en.wikipedia.org/wiki/UIC_classification_of_railway_coaches
 struct UicClassificationSecondary {
     const char code[4];
     VehicleSection::Features features;
     VehicleSection::Type type;
     int deckCount;
 };
 
 // 54: Czech Republic
 static constexpr const UicClassificationSecondary secondary_54_table[] = {
     { "b", VehicleSection::WheelchairAccessible, VehicleSection::UnknownType, 1 },
     { "c", VehicleSection::NoFeatures, VehicleSection::CouchetteCar, 1 },
     { "d", VehicleSection::BikeStorage, VehicleSection::UnknownType, 1 },
     { "f", VehicleSection::NoFeatures, VehicleSection::ControlCar, 1 },
     { "o", VehicleSection::NoFeatures, VehicleSection::UnknownType, 2 }, // ### could also be 't'?
     { "z", VehicleSection::AirConditioning, VehicleSection::PassengerCar, 1 },
 };
 
 // 80: Germany
 static constexpr const UicClassificationSecondary secondary_80_table[] = {
     { "b", VehicleSection::WheelchairAccessible, VehicleSection::UnknownType, 1 },
     { "c", VehicleSection::NoFeatures, VehicleSection::CouchetteCar, 1 },
     { "d", VehicleSection::BikeStorage, VehicleSection::UnknownType, 1 },
     { "f", VehicleSection::NoFeatures, VehicleSection::ControlCar, 1 },
     { "k", VehicleSection::Restaurant, VehicleSection::UnknownType, 1 },
     { "p", VehicleSection::AirConditioning, VehicleSection::PassengerCar, 1 },
     { "q", VehicleSection::NoFeatures, VehicleSection::ControlCar, 1 },
 };
 
 // 81: Austria
 static constexpr const UicClassificationSecondary secondary_81_table[] = {
     { "b", VehicleSection::WheelchairAccessible, VehicleSection::UnknownType, 1 }, // TODO wheelchair accessible toilets specifically
     { "c", VehicleSection::NoFeatures, VehicleSection::CouchetteCar, 1 },
     { "f", VehicleSection::NoFeatures, VehicleSection::ControlCar, 1 },
     { "p", VehicleSection::NoFeatures, VehicleSection::PassengerCar, 1 },
     { "-s", VehicleSection::NoFeatures, VehicleSection::ControlCar, 1 },
     { "-dl", VehicleSection::NoFeatures, VehicleSection::PassengerCar, 2 },
     { "-ds", VehicleSection::NoFeatures, VehicleSection::ControlCar, 2 },
 };
 
 // 85: Switzerland
 static constexpr const UicClassificationSecondary secondary_85_table[] = {
     { "c", VehicleSection::NoFeatures, VehicleSection::CouchetteCar, 1 },
     { "r", VehicleSection::Restaurant, VehicleSection::UnknownType, 1 },
     { "t", VehicleSection::NoFeatures, VehicleSection::ControlCar, 1 },
 };
 
 // 87: France
 static constexpr const UicClassificationSecondary secondary_87_table[] = {
     { "c", VehicleSection::NoFeatures, VehicleSection::CouchetteCar, 1 },
     { "e", VehicleSection::NoFeatures, VehicleSection::UnknownType, 2 },
     { "h", VehicleSection::WheelchairAccessible, VehicleSection::UnknownType, 1 },
     { "u", VehicleSection::AirConditioning, VehicleSection::UnknownType, 1 },
 };
 
 struct {
     const char country[3];
     const UicClassificationSecondary *begin;
     const UicClassificationSecondary *end;
 } static constexpr const secondary_tables[] = {
     { "54", std::begin(secondary_54_table), std::end(secondary_54_table) },
     { "80", std::begin(secondary_80_table), std::end(secondary_80_table) },
     { "81", std::begin(secondary_81_table), std::end(secondary_81_table) },
     { "85", std::begin(secondary_85_table), std::end(secondary_85_table) },
     { "87", std::begin(secondary_87_table), std::end(secondary_87_table) },
 };
 
 int UicRailwayCoach::deckCount(QStringView coachNumber, QStringView coachClassification)
 {
     int decks = 1;
     const auto it = std::find_if(std::begin(class_prefix_table), std::end(class_prefix_table), [coachClassification](const auto &prefix) {
         return coachClassification.startsWith(QLatin1String(prefix.prefix));
     });
     if (it != std::end(class_prefix_table)) {
         decks = std::max(decks, (*it).deckCount);
     }
 
     const auto country = UicRailwayCoach::countryCode(coachNumber);
     for (const auto &tab : secondary_tables) {
         if (country != QLatin1String(tab.country)) {
             continue;
         }
         for (auto it = tab.begin; it != tab.end; ++it) {
             if (coachClassification.contains(QLatin1String((*it).code))) {
                 decks = std::max(decks, (*it).deckCount);
             }
         }
     }
 
     return decks;
 }
 
 VehicleSection::Features UicRailwayCoach::features(QStringView coachNumber, QStringView coachClassification)
 {
     VehicleSection::Features f = {};
     const auto it = std::find_if(std::begin(class_prefix_table), std::end(class_prefix_table), [coachClassification](const auto &prefix) {
         return coachClassification.startsWith(QLatin1String(prefix.prefix));
     });
     if (it != std::end(class_prefix_table)) {
         f |= (*it).features;
     }
 
     const auto it2 = std::find_if(std::begin(number_prefix_table), std::end(number_prefix_table), [coachNumber](const auto &prefix) {
         return coachNumber.startsWith(QLatin1String(prefix.prefix));
     });
     if (it2 != std::end(number_prefix_table)) {
         f |= (*it2).features;
     }
 
     const auto country = UicRailwayCoach::countryCode(coachNumber);
     for (const auto &tab : secondary_tables) {
         if (country != QLatin1String(tab.country)) {
             continue;
         }
         for (auto it = tab.begin; it != tab.end; ++it) {
             if (coachClassification.contains(QLatin1String((*it).code))) {
                 f |= (*it).features;
             }
         }
     }
 
     return f;
 }
 
 VehicleSection::Type UicRailwayCoach::type(QStringView coachNumber, QStringView coachClassification)
 {
     bool seenPassengerCar = false;
     const auto it = std::find_if(std::begin(class_prefix_table), std::end(class_prefix_table), [coachClassification](const auto &prefix) {
         return prefix.type != VehicleSection::UnknownType && coachClassification.startsWith(QLatin1String(prefix.prefix));
     });
     if (it != std::end(class_prefix_table)) {
         if ((*it).type == VehicleSection::PassengerCar) {
             seenPassengerCar = true;
         } else {
             return (*it).type;
         }
     }
 
     const auto it2 = std::find_if(std::begin(number_prefix_table), std::end(number_prefix_table), [coachNumber](const auto &prefix) {
         return prefix.type != VehicleSection::UnknownType && coachNumber.startsWith(QLatin1String(prefix.prefix));
     });
     if (it2 != std::end(number_prefix_table)) {
         if ((*it2).type == VehicleSection::PassengerCar) {
             seenPassengerCar = true;
         } else {
             return (*it2).type;
         }
     }
 
     const auto country = UicRailwayCoach::countryCode(coachNumber);
     for (const auto &tab : secondary_tables) {
         if (country != QLatin1String(tab.country)) {
             continue;
         }
         const auto it = std::find_if(tab.begin, tab.end, [coachClassification](const auto &prefix) {
             return prefix.type != VehicleSection::UnknownType && coachClassification.contains(QLatin1String(prefix.code));
         });
         if (it != tab.end) {
             if ((*it).type == VehicleSection::PassengerCar) {
                 seenPassengerCar = true;
             } else {
                 return (*it).type;
             }
         }
     }
 
     return seenPassengerCar ? VehicleSection::PassengerCar : VehicleSection::UnknownType;
 }