refs #707, allow to resolve airline by callsign directly in airline ICAO list

src/blackmisc/aviation/airlineicaocode.cpp

@@ -7,7 +7,8 @@
  * contained in the LICENSE file.
  */
 
-#include "blackmisc/aviation/airlineicaocode.h"
+#include "airlineicaocode.h"
+#include "callsign.h"
 #include "blackmisc/comparefunctions.h"
 #include "blackmisc/db/datastoreutility.h"
 #include "blackmisc/icons.h"
@@ -33,11 +34,21 @@ namespace BlackMisc
     {
         CAirlineIcaoCode::CAirlineIcaoCode(const QString &airlineDesignator)
             : m_designator(airlineDesignator.trimmed().toUpper())
-        {}
+        {
+            if (this->m_designator.length() == 4)
+            {
+                this->setDesignator(this->m_designator);
+            }
+        }
 
         CAirlineIcaoCode::CAirlineIcaoCode(const QString &airlineDesignator, const QString &airlineName, const BlackMisc::CCountry &country, const QString &telephony, bool virtualAirline, bool operating)
             : m_designator(airlineDesignator.trimmed().toUpper()), m_name(airlineName), m_country(country), m_telephonyDesignator(telephony), m_isVa(virtualAirline), m_isOperating(operating)
-        {}
+        {
+            if (this->m_designator.length() == 4)
+            {
+                this->setDesignator(this->m_designator);
+            }
+        }
 
         const QString CAirlineIcaoCode::getVDesignator() const
         {
@@ -281,6 +292,30 @@ namespace BlackMisc
             return s.append(" ").append(getDbKeyAsStringInParentheses());
         }
 
+        CAirlineIcaoCode CAirlineIcaoCode::thisOrCallsignCode(const CCallsign &callsign) const
+        {
+            if (this->hasValidDbKey()) { return *this; }
+            if (callsign.isEmpty()) { return *this; }
+            const QString callsignAirline = callsign.getAirlineSuffix();
+            if (callsignAirline.isEmpty()) { return *this; }
+            if (callsignAirline == this->m_designator) { return *this; }
+
+            const CAirlineIcaoCode callsignIcao(callsignAirline);
+            if (this->m_designator.isEmpty()) { return callsignIcao; }
+
+            // here we have 2 possible codes
+            if (callsignIcao.isVirtualAirline())
+            {
+
+                if (callsignIcao.getDesignator().endsWith(this->m_designator))
+                {
+                    // callsign ICAO is virtual airline of myself, this is more accurate
+                    return callsignIcao;
+                }
+            }
+            return *this;
+        }
+
         QString CAirlineIcaoCode::getNameWithKey() const
         {
             if (!hasValidDbKey()) { return getName(); }

src/blackmisc/aviation/airlineicaocode.h

@@ -30,6 +30,8 @@ namespace BlackMisc
 {
     namespace Aviation
     {
+        class CCallsign;
+
         //! Value object for ICAO classification
         class BLACKMISC_EXPORT CAirlineIcaoCode :
             public CValueObject<CAirlineIcaoCode>,
@@ -156,6 +158,9 @@ namespace BlackMisc
             //! Comined string with key
             QString getCombinedStringWithKey() const;
 
+            //! What is better, the callsign airline code or this code
+            CAirlineIcaoCode thisOrCallsignCode(const CCallsign &callsign) const;
+
             //! \copydoc BlackMisc::Mixin::Icon::toIcon
             CIcon toIcon() const;
 

src/blackmisc/aviation/airlineicaocodelist.cpp

@@ -88,35 +88,36 @@ namespace BlackMisc
             });
         }
 
-        CAirlineIcaoCode CAirlineIcaoCodeList::smartAirlineIcaoSelector(const CAirlineIcaoCode &icaoPattern) const
+        CAirlineIcaoCode CAirlineIcaoCodeList::smartAirlineIcaoSelector(const CAirlineIcaoCode &icaoPattern, const CCallsign &callsign) const
         {
-            if (icaoPattern.hasValidDbKey())
+            const CAirlineIcaoCode patternUsed = icaoPattern.thisOrCallsignCode(callsign);
+            if (patternUsed.hasValidDbKey())
             {
                 return this->findByKey(icaoPattern.getDbKey(), icaoPattern);
             }
 
             // search by parts
             CAirlineIcaoCodeList codes;
-            if (icaoPattern.hasValidDesignator())
+            if (patternUsed.hasValidDesignator())
             {
-                codes = this->findByVDesignator(icaoPattern.getVDesignator());
+                codes = this->findByVDesignator(patternUsed.getVDesignator());
             }
             else
             {
-                codes = this->findByIataCode(icaoPattern.getIataCode());
+                codes = this->findByIataCode(patternUsed.getIataCode());
             }
 
             if (codes.size() == 1) { return codes.front(); }
 
             // further reduce
-            if (icaoPattern.hasValidCountry())
+            if (patternUsed.hasValidCountry())
             {
-                CAirlineIcaoCodeList countryCodes = codes.findByCountryIsoCode(icaoPattern.getCountry().getIsoCode());
+                CAirlineIcaoCodeList countryCodes = codes.findByCountryIsoCode(patternUsed.getCountry().getIsoCode());
                 if (!countryCodes.isEmpty()) { return countryCodes.front(); }
             }
 
             if (!codes.isEmpty()) { return codes.front(); }
-            return icaoPattern;
+            return patternUsed;
         }
 
         CAirlineIcaoCode CAirlineIcaoCodeList::findBestMatchByCallsign(const CCallsign &callsign) const

src/blackmisc/aviation/airlineicaocodelist.h

@@ -70,7 +70,7 @@ namespace BlackMisc
             CAirlineIcaoCodeList findByMilitary(bool military) const;
 
             //! Best selection by given pattern
-            CAirlineIcaoCode smartAirlineIcaoSelector(const CAirlineIcaoCode &icaoPattern) const;
+            CAirlineIcaoCode smartAirlineIcaoSelector(const CAirlineIcaoCode &icaoPattern, const CCallsign &callsign) const;
 
             //! Use callsign to conclude airline
             CAirlineIcaoCode findBestMatchByCallsign(const CCallsign &callsign) const;