mirror of
https://github.com/swift-project/pilotclient.git
synced 2026-04-03 15:45:46 +08:00
a9198e4f13
* some started with get/some not (removed get, get is normally used for our getters) * some used swift, getSwiftXZY dir (removed swift) * used full word "directory", not dir/directory mixed
511 lines
22 KiB
C++
511 lines
22 KiB
C++
/* Copyright (C) 2013
|
|
* swift project Community / Contributors
|
|
*
|
|
* This file is part of swift project. It is subject to the license terms in the LICENSE file found in the top-level
|
|
* directory of this distribution and at http://www.swift-project.org/license.html. No part of swift project,
|
|
* including this file, may be copied, modified, propagated, or distributed except according to the terms
|
|
* contained in the LICENSE file.
|
|
*/
|
|
|
|
#include "airlineicaocode.h"
|
|
#include "callsign.h"
|
|
#include "blackmisc/db/datastoreutility.h"
|
|
#include "blackmisc/comparefunctions.h"
|
|
#include "blackmisc/icons.h"
|
|
#include "blackmisc/logcategory.h"
|
|
#include "blackmisc/logcategorylist.h"
|
|
#include "blackmisc/propertyindex.h"
|
|
#include "blackmisc/statusmessage.h"
|
|
#include "blackmisc/stringutils.h"
|
|
#include "blackmisc/directoryutils.h"
|
|
#include "blackmisc/variant.h"
|
|
#include "blackmisc/verify.h"
|
|
|
|
#include <QJsonValue>
|
|
#include <QRegularExpression>
|
|
#include <QRegularExpressionMatch>
|
|
#include <QThreadStorage>
|
|
#include <QStringBuilder>
|
|
#include <QDir>
|
|
#include <Qt>
|
|
#include <QtGlobal>
|
|
|
|
using namespace BlackMisc;
|
|
using namespace BlackMisc::Db;
|
|
|
|
namespace BlackMisc
|
|
{
|
|
namespace Aviation
|
|
{
|
|
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_telephonyDesignator(telephony), m_country(country), m_isVa(virtualAirline), m_isOperating(operating)
|
|
{
|
|
if (this->m_designator.length() == 4)
|
|
{
|
|
this->setDesignator(this->m_designator);
|
|
}
|
|
}
|
|
|
|
const QString CAirlineIcaoCode::getVDesignator() const
|
|
{
|
|
if (!isVirtualAirline()) { return this->m_designator; }
|
|
return QLatin1Char('V') % this->m_designator;
|
|
}
|
|
|
|
QString CAirlineIcaoCode::getVDesignatorDbKey() const
|
|
{
|
|
return this->isLoadedFromDb() ?
|
|
this->getVDesignator() % this->getDbKeyAsStringInParentheses(" ") :
|
|
this->getVDesignator();
|
|
}
|
|
|
|
void CAirlineIcaoCode::setDesignator(const QString &icaoDesignator)
|
|
{
|
|
this->m_designator = icaoDesignator.trimmed().toUpper();
|
|
if (this->m_designator.length() == 4 && this->m_designator.startsWith("V"))
|
|
{
|
|
// a virtual designator was provided
|
|
this->setVirtualAirline(true);
|
|
this->m_designator = this->m_designator.right(3);
|
|
}
|
|
}
|
|
|
|
QString CAirlineIcaoCode::getDesignatorNameCountry() const
|
|
{
|
|
QString s(this->getDesignator());
|
|
if (this->hasName()) { s = s.append(" ").append(this->getName()); }
|
|
if (this->hasValidCountry()) { s = s.append(" ").append(this->getCountryIso()); }
|
|
return s.trimmed();
|
|
}
|
|
|
|
QString CAirlineIcaoCode::getSimplifiedName() const
|
|
{
|
|
return BlackMisc::simplifyNameForSearch(this->getName());
|
|
}
|
|
|
|
bool CAirlineIcaoCode::hasValidCountry() const
|
|
{
|
|
return this->m_country.isValid();
|
|
}
|
|
|
|
bool CAirlineIcaoCode::hasValidDesignator() const
|
|
{
|
|
return isValidAirlineDesignator(m_designator);
|
|
}
|
|
|
|
bool CAirlineIcaoCode::hasIataCode() const
|
|
{
|
|
return !this->m_iataCode.isEmpty();
|
|
}
|
|
|
|
bool CAirlineIcaoCode::matchesDesignator(const QString &designator) const
|
|
{
|
|
if (designator.isEmpty()) { return false; }
|
|
return designator.trimmed().toUpper() == this->m_designator;
|
|
}
|
|
|
|
bool CAirlineIcaoCode::matchesVDesignator(const QString &designator) const
|
|
{
|
|
if (designator.isEmpty()) { return false; }
|
|
return designator.trimmed().toUpper() == this->getVDesignator();
|
|
}
|
|
|
|
bool CAirlineIcaoCode::matchesIataCode(const QString &iata) const
|
|
{
|
|
if (iata.isEmpty()) { return false; }
|
|
return iata.trimmed().toUpper() == this->m_iataCode;
|
|
}
|
|
|
|
bool CAirlineIcaoCode::matchesDesignatorOrIataCode(const QString &candidate) const
|
|
{
|
|
if (candidate.isEmpty()) { return false; }
|
|
return this->matchesDesignator(candidate) || this->matchesIataCode(candidate);
|
|
}
|
|
|
|
bool CAirlineIcaoCode::matchesVDesignatorOrIataCode(const QString &candidate) const
|
|
{
|
|
if (candidate.isEmpty()) { return false; }
|
|
return this->matchesVDesignator(candidate) || this->matchesIataCode(candidate);
|
|
}
|
|
|
|
bool CAirlineIcaoCode::matchesNamesOrTelephonyDesignator(const QString &candidate) const
|
|
{
|
|
const QString cand(candidate.toUpper().trimmed());
|
|
if (this->getName().contains(cand, Qt::CaseInsensitive) || this->getTelephonyDesignator().contains(cand, Qt::CaseInsensitive)) { return true; }
|
|
return this->isContainedInSimplifiedName(candidate);
|
|
}
|
|
|
|
bool CAirlineIcaoCode::isContainedInSimplifiedName(const QString &candidate) const
|
|
{
|
|
if (candidate.isEmpty() || !this->hasName()) { return false; }
|
|
auto simplifiedName = makeRange(getName().begin(), getName().end()).findBy([](QChar c) { return c.isLetter(); });
|
|
auto it = std::search(simplifiedName.begin(), simplifiedName.end(), candidate.begin(), candidate.end(), [](QChar a, QChar b) { return a.toUpper() == b.toUpper(); });
|
|
return it != simplifiedName.end();
|
|
}
|
|
|
|
bool CAirlineIcaoCode::hasSimplifiedName() const
|
|
{
|
|
return this->hasName() && !this->getSimplifiedName().isEmpty();
|
|
}
|
|
|
|
bool CAirlineIcaoCode::hasCompleteData() const
|
|
{
|
|
return this->hasValidDesignator() && this->hasValidCountry() && this->hasName();
|
|
}
|
|
|
|
CIcon CAirlineIcaoCode::toIcon() const
|
|
{
|
|
if (this->hasValidDbKey() && CAirlineIcaoCode::iconIds().contains(this->getDbKey()))
|
|
{
|
|
static const QString p("airlines/%1_%2.png");
|
|
const QString n(p.arg(this->getDbKey(), 5, 10, QChar('0')).arg(this->getDesignator()));
|
|
return CIcon(n, this->convertToQString());
|
|
}
|
|
return CIcon::iconByIndex(CIcons::StandardIconEmpty);
|
|
}
|
|
|
|
QString CAirlineIcaoCode::convertToQString(bool i18n) const
|
|
{
|
|
Q_UNUSED(i18n);
|
|
return m_designator %
|
|
QLatin1String(" (") % m_name % QLatin1String(")") %
|
|
QLatin1String(" Op: ") % boolToYesNo(this->isOperating()) %
|
|
QLatin1String(" VA: ") % boolToYesNo(this->isVirtualAirline()) %
|
|
QLatin1String(" Mil: ") % boolToYesNo(this->isMilitary());
|
|
}
|
|
|
|
CVariant CAirlineIcaoCode::propertyByIndex(const BlackMisc::CPropertyIndex &index) const
|
|
{
|
|
if (index.isMyself()) { return CVariant::from(*this); }
|
|
if (IDatastoreObjectWithIntegerKey::canHandleIndex(index)) { return IDatastoreObjectWithIntegerKey::propertyByIndex(index); }
|
|
const ColumnIndex i = index.frontCasted<ColumnIndex>();
|
|
switch (i)
|
|
{
|
|
case IndexAirlineDesignator:
|
|
return CVariant::fromValue(this->m_designator);
|
|
case IndexIataCode:
|
|
return CVariant::fromValue(this->m_iataCode);
|
|
case IndexAirlineCountryIso:
|
|
return CVariant::fromValue(this->getCountryIso());
|
|
case IndexAirlineCountry:
|
|
return this->m_country.propertyByIndex(index.copyFrontRemoved());
|
|
case IndexAirlineName:
|
|
return CVariant::fromValue(this->m_name);
|
|
case IndexTelephonyDesignator:
|
|
return CVariant::fromValue(this->m_telephonyDesignator);
|
|
case IndexIsVirtualAirline:
|
|
return CVariant::fromValue(this->m_isVa);
|
|
case IndexIsOperating:
|
|
return CVariant::fromValue(this->m_isOperating);
|
|
case IndexIsMilitary:
|
|
return CVariant::fromValue(this->m_isMilitary);
|
|
case IndexDesignatorNameCountry:
|
|
return CVariant::fromValue(this->getDesignatorNameCountry());
|
|
case IndexGroupDesignator:
|
|
return CVariant::fromValue(this->getGroupDesignator());
|
|
case IndexGroupName:
|
|
return CVariant::fromValue(this->getGroupName());
|
|
case IndexGroupId:
|
|
return CVariant::fromValue(this->getGroupId());
|
|
default:
|
|
return CValueObject::propertyByIndex(index);
|
|
}
|
|
}
|
|
|
|
void CAirlineIcaoCode::setPropertyByIndex(const CPropertyIndex &index, const CVariant &variant)
|
|
{
|
|
if (index.isMyself()) { (*this) = variant.to<CAirlineIcaoCode>(); return; }
|
|
if (IDatastoreObjectWithIntegerKey::canHandleIndex(index)) { IDatastoreObjectWithIntegerKey::setPropertyByIndex(index, variant); return; }
|
|
const ColumnIndex i = index.frontCasted<ColumnIndex>();
|
|
switch (i)
|
|
{
|
|
case IndexAirlineDesignator:
|
|
this->setDesignator(variant.value<QString>());
|
|
break;
|
|
case IndexIataCode:
|
|
this->setIataCode(variant.value<QString>());
|
|
break;
|
|
case IndexAirlineCountry:
|
|
this->setCountry(variant.value<CCountry>());
|
|
break;
|
|
case IndexAirlineName:
|
|
this->setName(variant.value<QString>());
|
|
break;
|
|
case IndexTelephonyDesignator:
|
|
this->setTelephonyDesignator(variant.value<QString>());
|
|
break;
|
|
case IndexIsVirtualAirline:
|
|
this->setVirtualAirline(variant.toBool());
|
|
break;
|
|
case IndexIsOperating:
|
|
this->setOperating(variant.toBool());
|
|
break;
|
|
case IndexIsMilitary:
|
|
this->setMilitary(variant.toBool());
|
|
break;
|
|
case IndexGroupDesignator:
|
|
this->setGroupDesignator(variant.toQString());
|
|
break;
|
|
case IndexGroupName:
|
|
this->setGroupName(variant.toQString());
|
|
break;
|
|
case IndexGroupId:
|
|
this->setGroupId(variant.toInt());
|
|
break;
|
|
default:
|
|
CValueObject::setPropertyByIndex(index, variant);
|
|
break;
|
|
}
|
|
}
|
|
|
|
int CAirlineIcaoCode::comparePropertyByIndex(const CPropertyIndex &index, const CAirlineIcaoCode &compareValue) const
|
|
{
|
|
if (index.isMyself()) { return m_designator.compare(compareValue.getDesignator(), Qt::CaseInsensitive); }
|
|
if (IDatastoreObjectWithIntegerKey::canHandleIndex(index)) { return IDatastoreObjectWithIntegerKey::comparePropertyByIndex(index, compareValue);}
|
|
const ColumnIndex i = index.frontCasted<ColumnIndex>();
|
|
switch (i)
|
|
{
|
|
case IndexAirlineDesignator:
|
|
return this->m_designator.compare(compareValue.getDesignator());
|
|
case IndexIataCode:
|
|
return this->m_iataCode.compare(compareValue.getIataCode());
|
|
case IndexAirlineCountry:
|
|
return this->m_country.comparePropertyByIndex(index.copyFrontRemoved(), compareValue.getCountry());
|
|
case IndexDesignatorNameCountry:
|
|
return this->m_country.getName().compare(compareValue.getCountry().getName(), Qt::CaseInsensitive);
|
|
case IndexAirlineName:
|
|
return this->m_name.compare(compareValue.getName(), Qt::CaseInsensitive);
|
|
case IndexTelephonyDesignator:
|
|
return this->m_telephonyDesignator.compare(compareValue.getTelephonyDesignator(), Qt::CaseInsensitive);
|
|
case IndexIsVirtualAirline:
|
|
return Compare::compare(this->isVirtualAirline(), compareValue.isVirtualAirline());
|
|
case IndexIsOperating:
|
|
return Compare::compare(this->isOperating(), compareValue.isOperating());
|
|
case IndexIsMilitary:
|
|
return Compare::compare(this->isMilitary(), compareValue.isMilitary());
|
|
case IndexGroupDesignator:
|
|
return this->m_groupDesignator.compare(compareValue.getGroupDesignator(), Qt::CaseInsensitive);
|
|
case IndexGroupName:
|
|
return this->m_groupName.compare(compareValue.getGroupName(), Qt::CaseInsensitive);
|
|
case IndexGroupId:
|
|
return Compare::compare(this->m_groupId, compareValue.getGroupId());
|
|
default:
|
|
break;
|
|
}
|
|
Q_ASSERT_X(false, Q_FUNC_INFO, "No compare function");
|
|
return 0;
|
|
}
|
|
|
|
CStatusMessageList CAirlineIcaoCode::validate() const
|
|
{
|
|
static const CLogCategoryList cats(CLogCategoryList(this).join({ CLogCategory::validation() }));
|
|
CStatusMessageList msgs;
|
|
if (!hasValidDesignator()) { msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "Airline: missing designator")); }
|
|
if (!hasValidCountry()) { msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "Airline: missing country")); }
|
|
if (!hasName()) { msgs.push_back(CStatusMessage(cats, CStatusMessage::SeverityError, "Airline: no name")); }
|
|
return msgs;
|
|
}
|
|
|
|
bool CAirlineIcaoCode::isValidAirlineDesignator(const QString &airline)
|
|
{
|
|
// allow 2 chars for IATA
|
|
if (airline.length() < 2 || airline.length() > 5) { return false; }
|
|
const auto chars = makeRange(airline.begin(), airline.end());
|
|
if (chars.containsBy([](QChar c) { return !c.isUpper() && !c.isDigit(); })) { return false; }
|
|
return true;
|
|
}
|
|
|
|
QSet<QString> CAirlineIcaoCode::specialValidDesignators()
|
|
{
|
|
static const QSet<QString> valid({ "VV", "VM"});
|
|
return valid;
|
|
}
|
|
|
|
QString CAirlineIcaoCode::normalizeDesignator(const QString &candidate)
|
|
{
|
|
QString n(candidate.trimmed().toUpper());
|
|
n = n.left(indexOfChar(n, [](QChar c) { return c.isSpace(); }));
|
|
return removeChars(n, [](QChar c) { return !c.isLetterOrNumber(); });
|
|
}
|
|
|
|
QString CAirlineIcaoCode::getCombinedStringWithKey() const
|
|
{
|
|
QString s(getVDesignator());
|
|
if (s.isEmpty()) s = "????";
|
|
if (hasName()) { s = s.append(" ").append(getName()); }
|
|
return s.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 (!this->hasValidDbKey()) { return this->getName(); }
|
|
return this->hasName() ?
|
|
QString(this->getName()).append(" ").append(this->getDbKeyAsStringInParentheses()) :
|
|
this->getDbKeyAsStringInParentheses();
|
|
}
|
|
|
|
void CAirlineIcaoCode::updateMissingParts(const CAirlineIcaoCode &otherIcaoCode)
|
|
{
|
|
if (!this->hasValidDbKey() && otherIcaoCode.hasValidDbKey())
|
|
{
|
|
// we have no DB data, but the other one has
|
|
// so we change roles. We take the DB object as base, and update our parts
|
|
CAirlineIcaoCode copy(otherIcaoCode);
|
|
copy.updateMissingParts(*this);
|
|
*this = copy;
|
|
return;
|
|
}
|
|
|
|
if (!this->hasValidDesignator()) { this->setDesignator(otherIcaoCode.getDesignator()); }
|
|
if (!this->hasValidCountry()) { this->setCountry(otherIcaoCode.getCountry()); }
|
|
if (!this->hasName()) { this->setName(otherIcaoCode.getName()); }
|
|
if (!this->hasTelephonyDesignator()) { this->setTelephonyDesignator(otherIcaoCode.getTelephonyDesignator()); }
|
|
if (!this->hasValidDbKey())
|
|
{
|
|
this->setDbKey(otherIcaoCode.getDbKey());
|
|
this->setUtcTimestamp(otherIcaoCode.getUtcTimestamp());
|
|
}
|
|
}
|
|
|
|
QString CAirlineIcaoCode::asHtmlSummary() const
|
|
{
|
|
return this->getCombinedStringWithKey();
|
|
}
|
|
|
|
int CAirlineIcaoCode::calculateScore(const CAirlineIcaoCode &otherCode) const
|
|
{
|
|
const bool bothFromDb = otherCode.isLoadedFromDb() && this->isLoadedFromDb();
|
|
if (bothFromDb && this->getDbKey() == otherCode.getDbKey())
|
|
{
|
|
return 100;
|
|
}
|
|
int score = 0;
|
|
if (otherCode.hasValidDesignator() && this->getDesignator() == otherCode.getDesignator())
|
|
{
|
|
score += 60;
|
|
}
|
|
|
|
if (bothFromDb && this->isVirtualAirline() == otherCode.isVirtualAirline())
|
|
{
|
|
score += 20;
|
|
}
|
|
if (this->hasName() && this->getName() == otherCode.getName())
|
|
{
|
|
score += 20;
|
|
}
|
|
else if (this->hasTelephonyDesignator() && this->getTelephonyDesignator() == otherCode.getTelephonyDesignator())
|
|
{
|
|
score += 15;
|
|
}
|
|
else if (this->hasSimplifiedName() && this->getSimplifiedName() == otherCode.getSimplifiedName())
|
|
{
|
|
score += 10;
|
|
}
|
|
return score;
|
|
}
|
|
|
|
CAirlineIcaoCode CAirlineIcaoCode::fromDatabaseJson(const QJsonObject &json, const QString &prefix)
|
|
{
|
|
if (!existsKey(json, prefix))
|
|
{
|
|
// when using relationship, this can be null (e.g. for color liveries)
|
|
return CAirlineIcaoCode();
|
|
}
|
|
|
|
QString designator(json.value(prefix % QLatin1String("designator")).toString());
|
|
if (!CAirlineIcaoCode::isValidAirlineDesignator(designator))
|
|
{
|
|
designator = CAirlineIcaoCode::normalizeDesignator(designator);
|
|
}
|
|
|
|
const QString iata(json.value(prefix % QLatin1String("iata")).toString());
|
|
const QString telephony(json.value(prefix % QLatin1String("callsign")).toString());
|
|
const QString name(json.value(prefix % QLatin1String("name")).toString());
|
|
const QString countryIso(json.value(prefix % QLatin1String("country")).toString());
|
|
const QString countryName(json.value(prefix % QLatin1String("countryname")).toString());
|
|
const QString groupName(json.value(prefix % QLatin1String("groupname")).toString());
|
|
const QString groupDesignator(json.value(prefix % QLatin1String("groupdesignator")).toString());
|
|
const int groupId(json.value(prefix % QLatin1String("groupid")).toInt(-1));
|
|
const bool va = CDatastoreUtility::dbBoolStringToBool(json.value(prefix % QLatin1String("va")).toString());
|
|
const bool operating = CDatastoreUtility::dbBoolStringToBool(json.value(prefix % QLatin1String("operating")).toString());
|
|
const bool military = CDatastoreUtility::dbBoolStringToBool(json.value(prefix % QLatin1String("military")).toString());
|
|
|
|
CAirlineIcaoCode code(
|
|
designator, name,
|
|
CCountry(countryIso, countryName),
|
|
telephony, va, operating
|
|
);
|
|
code.setIataCode(iata);
|
|
code.setMilitary(military);
|
|
code.setGroupDesignator(groupDesignator);
|
|
code.setGroupId(groupId);
|
|
code.setGroupName(groupName);
|
|
code.setKeyAndTimestampFromDatabaseJson(json, prefix);
|
|
return code;
|
|
}
|
|
|
|
//! \private
|
|
QSet<int> iconIdsImpl()
|
|
{
|
|
QDir dir(CDirectoryUtils::imagesAirlinesDirectory());
|
|
Q_ASSERT_X(dir.exists(), Q_FUNC_INFO, "image directory missing");
|
|
|
|
QSet<int> ids;
|
|
dir.setFilter(QDir::Files | QDir::NoSymLinks);
|
|
dir.setSorting(QDir::Name);
|
|
bool ok = false;
|
|
for (const QFileInfo &fileInfo : dir.entryInfoList())
|
|
{
|
|
const QString fn(fileInfo.fileName());
|
|
ok = fn.size() > 5;
|
|
if (!ok) { continue; }
|
|
BLACK_VERIFY_X(ok, Q_FUNC_INFO, "wrong file name");
|
|
const int id = fn.left(5).toInt(&ok);
|
|
BLACK_VERIFY_X(ok, Q_FUNC_INFO, "wrong id format");
|
|
if (!ok) { continue; }
|
|
ids.insert(id);
|
|
}
|
|
return ids;
|
|
}
|
|
|
|
const QSet<int> &CAirlineIcaoCode::iconIds()
|
|
{
|
|
static const QSet<int> ids = iconIdsImpl();
|
|
return ids;
|
|
}
|
|
} // namespace
|
|
} // namespace
|