// BaseDiagram.cxx - part of GUI launcher using Qt5 // // Written by James Turner, started December 2014. // // Copyright (C) 2014 James Turner // // This program is free software; you can redistribute it and/or // modify it under the terms of the GNU General Public License as // published by the Free Software Foundation; either version 2 of the // License, or (at your option) any later version. // // This program is distributed in the hope that it will be useful, but // WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU // General Public License for more details. // // You should have received a copy of the GNU General Public License // along with this program; if not, write to the Free Software // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. #include "BaseDiagram.hxx" #include #include #include #include #include #include #include #include #include #include "QtLauncher_fwd.hxx" /* equatorial and polar earth radius */ const float rec = 6378137; // earth radius, equator (?) const float rpol = 6356752.314f; // earth radius, polar (?) const double MINIMUM_SCALE = 0.002; //Returns Earth radius at a given latitude (Ellipsoide equation with two equal axis) static float earth_radius_lat( float lat ) { double a = cos(lat)/rec; double b = sin(lat)/rpol; return 1.0f / sqrt( a * a + b * b ); } BaseDiagram::BaseDiagram(QWidget* pr) : QWidget(pr), m_autoScalePan(true), m_wheelAngleDeltaAccumulator(0) { setSizePolicy(QSizePolicy::MinimumExpanding, QSizePolicy::MinimumExpanding); setMinimumSize(100, 100); } QTransform BaseDiagram::transform() const { QTransform t; t.translate(width() / 2, height() / 2); // center projection origin in the widget t.scale(m_scale, m_scale); // apply any pan offset that exists t.translate(m_panOffset.x(), m_panOffset.y()); // center the bounding box (may not be at the origin) t.translate(-m_bounds.center().x(), -m_bounds.center().y()); return t; } void BaseDiagram::clearIgnoredNavaids() { m_ignored.clear(); } void BaseDiagram::addIgnoredNavaid(FGPositionedRef pos) { if (isNavaidIgnored(pos)) return; m_ignored.push_back(pos); } void BaseDiagram::extendRect(QRectF &r, const QPointF &p) { if (p.x() < r.left()) { r.setLeft(p.x()); } else if (p.x() > r.right()) { r.setRight(p.x()); } if (p.y() < r.top()) { r.setTop(p.y()); } else if (p.y() > r.bottom()) { r.setBottom(p.y()); } } void BaseDiagram::paintEvent(QPaintEvent* pe) { QPainter p(this); p.setRenderHints(QPainter::Antialiasing); p.fillRect(rect(), QColor(0x3f, 0x3f, 0x3f)); if (m_autoScalePan) { // fit bounds within our available space, allowing for a margin const int MARGIN = 32; // pixels double ratioInX = (width() - MARGIN * 2) / m_bounds.width(); double ratioInY = (height() - MARGIN * 2) / m_bounds.height(); m_scale = std::min(ratioInX, ratioInY); } QTransform t(transform()); p.setTransform(t); paintPolygonData(&p); paintNavaids(&p); paintContents(&p); } void BaseDiagram::paintAirplaneIcon(QPainter* painter, const SGGeod& geod, int headingDeg) { QPointF pos = project(geod); QPixmap pix(":/airplane-icon"); pos = painter->transform().map(pos); painter->resetTransform(); painter->translate(pos.x(), pos.y()); painter->rotate(headingDeg); painter->setRenderHint(QPainter::SmoothPixmapTransform, true); QRect airplaneIconRect = pix.rect(); airplaneIconRect.moveCenter(QPoint(0,0)); painter->drawPixmap(airplaneIconRect, pix); } void BaseDiagram::paintPolygonData(QPainter* painter) { QTransform xf = painter->transform(); QTransform invT = xf.inverted(); SGGeod topLeft = unproject(invT.map(rect().topLeft()), m_projectionCenter); SGGeod viewCenter = unproject(invT.map(rect().center()), m_projectionCenter); SGGeod bottomRight = unproject(invT.map(rect().bottomRight()), m_projectionCenter); double drawRangeNm = std::max(SGGeodesy::distanceNm(viewCenter, topLeft), SGGeodesy::distanceNm(viewCenter, bottomRight)); flightgear::PolyLineList lines(flightgear::PolyLine::linesNearPos(viewCenter, drawRangeNm, flightgear::PolyLine::COASTLINE)); QPen waterPen(QColor(64, 64, 255), 1); waterPen.setCosmetic(true); painter->setPen(waterPen); flightgear::PolyLineList::const_iterator it; for (it=lines.begin(); it != lines.end(); ++it) { paintGeodVec(painter, (*it)->points()); } lines = flightgear::PolyLine::linesNearPos(viewCenter, drawRangeNm, flightgear::PolyLine::URBAN); for (it=lines.begin(); it != lines.end(); ++it) { fillClosedGeodVec(painter, QColor(192, 192, 96), (*it)->points()); } lines = flightgear::PolyLine::linesNearPos(viewCenter, drawRangeNm, flightgear::PolyLine::RIVER); painter->setPen(waterPen); for (it=lines.begin(); it != lines.end(); ++it) { paintGeodVec(painter, (*it)->points()); } lines = flightgear::PolyLine::linesNearPos(viewCenter, drawRangeNm, flightgear::PolyLine::LAKE); for (it=lines.begin(); it != lines.end(); ++it) { fillClosedGeodVec(painter, QColor(128, 128, 255), (*it)->points()); } } void BaseDiagram::paintGeodVec(QPainter* painter, const flightgear::SGGeodVec& vec) { QVector projected; projected.reserve(vec.size()); flightgear::SGGeodVec::const_iterator it; for (it=vec.begin(); it != vec.end(); ++it) { projected.append(project(*it)); } painter->drawPolyline(projected.data(), projected.size()); } void BaseDiagram::fillClosedGeodVec(QPainter* painter, const QColor& color, const flightgear::SGGeodVec& vec) { QVector projected; projected.reserve(vec.size()); flightgear::SGGeodVec::const_iterator it; for (it=vec.begin(); it != vec.end(); ++it) { projected.append(project(*it)); } painter->setPen(Qt::NoPen); painter->setBrush(color); painter->drawPolygon(projected.data(), projected.size()); } class MapFilter : public FGPositioned::TypeFilter { public: MapFilter(LauncherAircraftType aircraft) { // addType(FGPositioned::FIX); addType(FGPositioned::NDB); addType(FGPositioned::VOR); if (aircraft == Helicopter) { addType(FGPositioned::HELIPAD); } if (aircraft == Seaplane) { addType(FGPositioned::SEAPORT); } else { addType(FGPositioned::AIRPORT); } } virtual bool pass(FGPositioned* aPos) const { bool ok = TypeFilter::pass(aPos); // fix-filtering code disabled since fixed are entirely disabled #if 0 if (ok && (aPos->type() == FGPositioned::FIX)) { // ignore fixes which end in digits if (aPos->ident().length() > 4 && isdigit(aPos->ident()[3]) && isdigit(aPos->ident()[4])) { return false; } } #endif return ok; } }; void BaseDiagram::splitItems(const FGPositionedList& in, FGPositionedList& navaids, FGPositionedList& ports) { FGPositionedList::const_iterator it = in.begin(); for (; it != in.end(); ++it) { if (FGAirport::isAirportType(it->ptr())) { ports.push_back(*it); } else { navaids.push_back(*it); } } } bool orderAirportsByRunwayLength(const FGPositionedRef& a, const FGPositionedRef& b) { FGAirport* aptA = static_cast(a.ptr()); FGAirport* aptB = static_cast(b.ptr()); return aptA->longestRunway()->lengthFt() > aptB->longestRunway()->lengthFt(); } void BaseDiagram::paintNavaids(QPainter* painter) { QTransform xf = painter->transform(); painter->setTransform(QTransform()); // reset to identity QTransform invT = xf.inverted(); SGGeod topLeft = unproject(invT.map(rect().topLeft()), m_projectionCenter); SGGeod viewCenter = unproject(invT.map(rect().center()), m_projectionCenter); SGGeod bottomRight = unproject(invT.map(rect().bottomRight()), m_projectionCenter); double drawRangeNm = std::max(SGGeodesy::distanceNm(viewCenter, topLeft), SGGeodesy::distanceNm(viewCenter, bottomRight)); MapFilter f(m_aircraftType); FGPositionedList items = FGPositioned::findWithinRange(viewCenter, drawRangeNm, &f); FGPositionedList navaids, ports; splitItems(items, navaids, ports); if (ports.size() >= 40) { FGPositionedList::iterator middle = ports.begin() + 40; std::partial_sort(ports.begin(), middle, ports.end(), orderAirportsByRunwayLength); ports.resize(40); } m_labelRects.clear(); m_labelRects.reserve(items.size()); FGPositionedList::const_iterator it; for (it = ports.begin(); it != ports.end(); ++it) { paintNavaid(painter, xf, *it); } for (it = navaids.begin(); it != navaids.end(); ++it) { paintNavaid(painter, xf, *it); } // restore transform painter->setTransform(xf); } QRect boundsOfLines(const QVector& lines) { QRect r; Q_FOREACH(const QLineF& l, lines) { r = r.united(QRectF(l.p1(), l.p2()).toRect()); } return r; } void BaseDiagram::paintNavaid(QPainter* painter, const QTransform& t, const FGPositionedRef &pos) { if (isNavaidIgnored(pos)) return; bool drawAsIcon = true; const double minRunwayLengthFt = (16 / m_scale) * SG_METER_TO_FEET; const FGPositioned::Type ty(pos->type()); const bool isNDB = (ty == FGPositioned::NDB); QRect iconRect; if (ty == FGPositioned::AIRPORT) { FGAirport* apt = static_cast(pos.ptr()); if (apt->hasHardRunwayOfLengthFt(minRunwayLengthFt)) { drawAsIcon = false; painter->setTransform(t); QVector lines = projectAirportRuwaysWithCenter(apt, m_projectionCenter); QPen pen(QColor(0x03, 0x83, 0xbf), 8); pen.setCosmetic(true); painter->setPen(pen); painter->drawLines(lines); QPen linePen(Qt::white, 2); linePen.setCosmetic(true); painter->setPen(linePen); painter->drawLines(lines); painter->resetTransform(); iconRect = t.mapRect(boundsOfLines(lines)); } } if (drawAsIcon) { QPixmap pm = iconForPositioned(pos); QPointF loc = t.map(project(pos->geod())); iconRect = pm.rect(); iconRect.moveCenter(loc.toPoint()); painter->drawPixmap(iconRect, pm); } // compute label text so we can measure it QString label; if (FGAirport::isAirportType(pos.ptr())) { label = QString::fromStdString(pos->name()); label = fixNavaidName(label); } else { label = QString::fromStdString(pos->ident()); } if (ty == FGPositioned::NDB) { FGNavRecord* nav = static_cast(pos.ptr()); label.append("\n").append(QString::number(nav->get_freq() / 100)); } else if (ty == FGPositioned::VOR) { FGNavRecord* nav = static_cast(pos.ptr()); label.append("\n").append(QString::number(nav->get_freq() / 100.0, 'f', 1)); } QRect textBounds = painter->boundingRect(QRect(0, 0, 100, 100), Qt::TextWordWrap, label); int textFlags; textBounds = rectAndFlagsForLabel(pos->guid(), iconRect, textBounds.size(), textFlags); painter->setPen(isNDB ? QColor(0x9b, 0x5d, 0xa2) : QColor(0x03, 0x83, 0xbf)); painter->drawText(textBounds, textFlags, label); } bool BaseDiagram::isNavaidIgnored(const FGPositionedRef &pos) const { return m_ignored.contains(pos); } bool BaseDiagram::isLabelRectAvailable(const QRect &r) const { Q_FOREACH(const QRect& lr, m_labelRects) { if (lr.intersects(r)) return false; } return true; } int BaseDiagram::textFlagsForLabelPosition(LabelPosition pos) { #if 0 switch (pos) { case LABEL_RIGHT: return Qt::AlignLeft | Qt::AlignVCenter; case LABEL_ABOVE: return Qt::AlignHCenter | Qt::A } #endif return 0; } QRect BaseDiagram::rectAndFlagsForLabel(PositionedID guid, const QRect& item, const QSize &bounds, int& flags) const { m_labelRects.append(item); int pos = m_labelPositions.value(guid, LABEL_RIGHT); bool firstAttempt = true; flags = Qt::TextWordWrap; while (pos < LAST_POSITION) { QRect r = labelPositioned(item, bounds, static_cast(pos)); if (isLabelRectAvailable(r)) { m_labelRects.append(r); m_labelPositions[guid] = static_cast(pos); flags |= textFlagsForLabelPosition(static_cast(pos)); return r; } else if (firstAttempt && (pos != LABEL_RIGHT)) { pos = LABEL_RIGHT; } else { ++pos; } firstAttempt = false; } return QRect(item.x(), item.y(), bounds.width(), bounds.height()); } QRect BaseDiagram::labelPositioned(const QRect& itemRect, const QSize& bounds, LabelPosition lp) const { const int SHORT_MARGIN = 4; const int DIAGONAL_MARGIN = 12; QPoint topLeft = itemRect.topLeft(); switch (lp) { // cardinal compass points are short (close in) case LABEL_RIGHT: topLeft = QPoint(itemRect.right() + SHORT_MARGIN, itemRect.center().y() - bounds.height() / 2); break; case LABEL_ABOVE: topLeft = QPoint(itemRect.center().x() - (bounds.width() / 2), itemRect.top() - (SHORT_MARGIN + bounds.height())); break; case LABEL_BELOW: topLeft = QPoint(itemRect.center().x() - (bounds.width() / 2), itemRect.bottom() + SHORT_MARGIN); break; case LABEL_LEFT: topLeft = QPoint(itemRect.left() - (SHORT_MARGIN + bounds.width()), itemRect.center().y() - bounds.height() / 2); break; // first diagonals are further out (to hopefully have a better chance // of finding clear space case LABEL_NE: topLeft = QPoint(itemRect.right() + DIAGONAL_MARGIN, itemRect.top() - (DIAGONAL_MARGIN + bounds.height())); break; case LABEL_NW: topLeft = QPoint(itemRect.left() - (DIAGONAL_MARGIN + bounds.width()), itemRect.top() - (DIAGONAL_MARGIN + bounds.height())); break; case LABEL_SE: topLeft = QPoint(itemRect.right() + DIAGONAL_MARGIN, itemRect.bottom() + DIAGONAL_MARGIN); break; case LABEL_SW: topLeft = QPoint(itemRect.left() - (DIAGONAL_MARGIN + bounds.width()), itemRect.bottom() + DIAGONAL_MARGIN); break; default: qWarning() << Q_FUNC_INFO << "Implement me"; } return QRect(topLeft, bounds); } void BaseDiagram::mousePressEvent(QMouseEvent *me) { m_lastMousePos = me->pos(); m_didPan = false; } void BaseDiagram::mouseMoveEvent(QMouseEvent *me) { m_autoScalePan = false; QPointF delta = me->pos() - m_lastMousePos; m_lastMousePos = me->pos(); // offset is stored in metres so we don't have to modify it when // zooming m_panOffset += (delta / m_scale); m_didPan = true; update(); } int intSign(int v) { return (v == 0) ? 0 : (v < 0) ? -1 : 1; } void BaseDiagram::wheelEvent(QWheelEvent *we) { m_autoScalePan = false; int delta = we->angleDelta().y(); if (delta == 0) return; if (intSign(m_wheelAngleDeltaAccumulator) != intSign(delta)) { m_wheelAngleDeltaAccumulator = 0; } m_wheelAngleDeltaAccumulator += delta; if (m_wheelAngleDeltaAccumulator > 120) { m_wheelAngleDeltaAccumulator = 0; m_scale *= 1.5; } else if (m_wheelAngleDeltaAccumulator < -120) { m_wheelAngleDeltaAccumulator = 0; m_scale *= 0.75; } SG_CLAMP_RANGE(m_scale, MINIMUM_SCALE, 1.0); update(); } void BaseDiagram::paintContents(QPainter* painter) { } void BaseDiagram::recomputeBounds(bool resetZoom) { m_bounds = QRectF(); doComputeBounds(); if (resetZoom) { m_autoScalePan = true; m_scale = 1.0; m_panOffset = QPointF(); } update(); } void BaseDiagram::doComputeBounds() { // no-op in the base class } void BaseDiagram::extendBounds(const QPointF& p) { extendRect(m_bounds, p); } QPointF BaseDiagram::project(const SGGeod& geod, const SGGeod& center) { double r = earth_radius_lat(geod.getLatitudeRad()); double ref_lat = center.getLatitudeRad(), ref_lon = center.getLongitudeRad(), lat = geod.getLatitudeRad(), lon = geod.getLongitudeRad(), lonDiff = lon - ref_lon; double c = acos( sin(ref_lat) * sin(lat) + cos(ref_lat) * cos(lat) * cos(lonDiff) ); if (c == 0.0) { // angular distance from center is 0 return QPointF(0.0, 0.0); } double k = c / sin(c); double x, y; if (ref_lat == (90 * SG_DEGREES_TO_RADIANS)) { x = (SGD_PI / 2 - lat) * sin(lonDiff); y = -(SGD_PI / 2 - lat) * cos(lonDiff); } else if (ref_lat == -(90 * SG_DEGREES_TO_RADIANS)) { x = (SGD_PI / 2 + lat) * sin(lonDiff); y = (SGD_PI / 2 + lat) * cos(lonDiff); } else { x = k * cos(lat) * sin(lonDiff); y = k * ( cos(ref_lat) * sin(lat) - sin(ref_lat) * cos(lat) * cos(lonDiff) ); } // flip for top-left origin return QPointF(x, -y) * r; } SGGeod BaseDiagram::unproject(const QPointF& xy, const SGGeod& center) { double r = earth_radius_lat(center.getLatitudeRad()); double lat = 0, lon = 0, ref_lat = center.getLatitudeRad(), ref_lon = center.getLongitudeRad(), rho = QVector2D(xy).length(), c = rho/r; if (rho == 0) { return center; } // invert y to balance the equivalent in project() double x = xy.x(), y = -xy.y(); lat = asin( cos(c) * sin(ref_lat) + (y * sin(c) * cos(ref_lat)) / rho); if (ref_lat == (90 * SG_DEGREES_TO_RADIANS)) // north pole { lon = ref_lon + atan(-x/y); } else if (ref_lat == -(90 * SG_DEGREES_TO_RADIANS)) // south pole { lon = ref_lon + atan(x/y); } else { lon = ref_lon + atan(x* sin(c) / (rho * cos(ref_lat) * cos(c) - y * sin(ref_lat) * sin(c))); } return SGGeod::fromRad(lon, lat); } QPointF BaseDiagram::project(const SGGeod& geod) const { return project(geod, m_projectionCenter); } QPixmap BaseDiagram::iconForPositioned(const FGPositionedRef& pos, const IconOptions& options) { // if airport type, check towered or untowered bool small = options.testFlag(SmallIcons); bool isTowered = false; if (FGAirport::isAirportType(pos)) { FGAirport* apt = static_cast(pos.ptr()); isTowered = apt->hasTower(); } switch (pos->type()) { case FGPositioned::VOR: if (static_cast(pos.ptr())->isVORTAC()) return QPixmap(":/vortac-icon"); if (static_cast(pos.ptr())->hasDME()) return QPixmap(":/vor-dme-icon"); return QPixmap(":/vor-icon"); case FGPositioned::AIRPORT: return iconForAirport(static_cast(pos.ptr()), options); case FGPositioned::HELIPORT: return QPixmap(":/heliport-icon"); case FGPositioned::SEAPORT: return QPixmap(isTowered ? ":/seaport-tower-icon" : ":/seaport-icon"); case FGPositioned::NDB: return QPixmap(small ? ":/ndb-small-icon" : ":/ndb-icon"); case FGPositioned::FIX: return QPixmap(":/waypoint-icon"); default: break; } return QPixmap(); } QPixmap BaseDiagram::iconForAirport(FGAirport* apt, const IconOptions& options) { if (apt->isClosed()) { return QPixmap(":/airport-closed-icon"); } if (!apt->hasHardRunwayOfLengthFt(1500)) { return QPixmap(apt->hasTower() ? ":/airport-tower-icon" : ":/airport-icon"); } if (options.testFlag(LargeAirportPlans) && apt->hasHardRunwayOfLengthFt(8500)) { QPixmap result(32, 32); result.fill(Qt::transparent); { QPainter p(&result); p.setRenderHint(QPainter::Antialiasing, true); QRectF b = result.rect().adjusted(4, 4, -4, -4); QVector lines = projectAirportRuwaysIntoRect(apt, b); p.setPen(QPen(QColor(0x03, 0x83, 0xbf), 8)); p.drawLines(lines); p.setPen(QPen(Qt::white, 2)); p.drawLines(lines); } return result; } QPixmap result(25, 25); result.fill(Qt::transparent); { QPainter p(&result); p.setRenderHint(QPainter::Antialiasing, true); p.setPen(Qt::NoPen); p.setBrush(apt->hasTower() ? QColor(0x03, 0x83, 0xbf) : QColor(0x9b, 0x5d, 0xa2)); p.drawEllipse(QPointF(13, 13), 10, 10); FGRunwayRef r = apt->longestRunway(); p.setPen(QPen(Qt::white, 2)); p.translate(13, 13); p.rotate(r->headingDeg()); p.drawLine(0, -8, 0, 8); } return result; } QVector BaseDiagram::projectAirportRuwaysWithCenter(FGAirportRef apt, const SGGeod& c) { QVector r; const FGRunwayList& runways(apt->getRunwaysWithoutReciprocals()); FGRunwayList::const_iterator it; for (it = runways.begin(); it != runways.end(); ++it) { FGRunwayRef rwy = *it; QPointF p1 = project(rwy->geod(), c); QPointF p2 = project(rwy->end(), c); r.append(QLineF(p1, p2)); } return r; } void BaseDiagram::setAircraftType(LauncherAircraftType type) { m_aircraftType = type; update(); } QVector BaseDiagram::projectAirportRuwaysIntoRect(FGAirportRef apt, const QRectF &bounds) { QVector r = projectAirportRuwaysWithCenter(apt, apt->geod()); QRectF extent; Q_FOREACH(const QLineF& l, r) { extendRect(extent, l.p1()); extendRect(extent, l.p2()); } // find constraining scale factor double ratioInX = bounds.width() / extent.width(); double ratioInY = bounds.height() / extent.height(); QTransform t; t.translate(bounds.left(), bounds.top()); t.scale(std::min(ratioInX, ratioInY), std::min(ratioInX, ratioInY)); t.translate(-extent.left(), -extent.top()); // move unscaled to 0,0 for (int i=0; i