2 WARNING: This code uses internal Qt API - the QZipReader class for reading
3 ZIP files - and things may break if Qt changes the API. For Qt5 this is not
4 a problem as we can "see the future" now and there are no changes in all
5 the supported Qt5 versions up to the last one (5.15). In Qt6 the class
6 might change or even disappear in the future, but this is very unlikely
7 as there were no changes for several years and The Qt Company's policy
8 is: "do not invest any resources into any desktop related stuff unless
9 absolutely necessary". There is an issue (QTBUG-3897) since the year 2009 to
10 include the ZIP reader into the public API, which aptly illustrates the
11 effort The Qt Company is willing to make about anything desktop related...
18 #include <QRegularExpression>
20 #include <private/qzipreader_p.h>
21 #include "common/rectc.h"
24 #define SRTM3_SAMPLES 1201
25 #define SRTM1_SAMPLES 3601
26 #define SRTM05_SAMPLES 7201
28 #define SRTM_SIZE(samples) \
29 ((samples) * (samples) * 2)
31 static qreal
interpolate(qreal dx
, qreal dy
, qreal p0
, qreal p1
, qreal p2
,
34 return p0
* (1 - dx
) * (1 - dy
) + p1
* dx
* (1 - dy
) + p2
* dy
* (1 - dx
)
38 static qreal
value(int col
, int row
, int samples
, const QByteArray
*data
)
40 int pos
= ((samples
- 1 - row
) * samples
+ col
) * 2;
41 qint16 val
= qFromBigEndian(*((const qint16
*)(data
->constData() + pos
)));
43 return (val
== -32768) ? NAN
: val
;
46 static qreal
height(const Coordinates
&c
, const QByteArray
*data
)
50 if (data
->size() == SRTM_SIZE(SRTM3_SAMPLES
))
51 samples
= SRTM3_SAMPLES
;
52 else if (data
->size() == SRTM_SIZE(SRTM1_SAMPLES
))
53 samples
= SRTM1_SAMPLES
;
54 else if (data
->size() == SRTM_SIZE(SRTM05_SAMPLES
))
55 samples
= SRTM05_SAMPLES
;
59 int row
= (int)((c
.lat() - qFloor(c
.lat())) * (samples
- 1));
60 int col
= (int)((c
.lon() - qFloor(c
.lon())) * (samples
- 1));
61 qreal dx
= ((c
.lon() - qFloor(c
.lon())) * (samples
- 1)) - col
;
62 qreal dy
= ((c
.lat() - qFloor(c
.lat())) * (samples
- 1)) - row
;
64 qreal p0
= value(col
, row
, samples
, data
);
65 qreal p1
= value(col
+ 1, row
, samples
, data
);
66 qreal p2
= value(col
, row
+ 1, samples
, data
);
67 qreal p3
= value(col
+ 1, row
+ 1, samples
, data
);
69 return interpolate(dx
, dy
, p0
, p1
, p2
, p3
);
74 QString
DEM::Tile::latStr() const
76 const char ns
= (_lat
>= 0) ? 'N' : 'S';
77 return QString("%1%2").arg(ns
).arg(qAbs(_lat
), 2, 10, QChar('0'));
80 QString
DEM::Tile::lonStr() const
82 const char ew
= (_lon
>= 0) ? 'E' : 'W';
83 return QString("%1%2").arg(ew
).arg(qAbs(_lon
), 3, 10, QChar('0'));
86 QString
DEM::Tile::baseName() const
88 return QString("%1%2.hgt").arg(latStr(), lonStr());
92 DEM::TileCache
DEM::_data
;
94 QString
DEM::fileName(const QString
&baseName
)
96 return QDir(_dir
).absoluteFilePath(baseName
);
99 void DEM::setCacheSize(int size
)
101 _data
.setMaxCost(size
* 1024);
104 void DEM::setDir(const QString
&path
)
109 void DEM::clearCache()
114 qreal
DEM::elevation(const Coordinates
&c
)
119 Tile
tile(qFloor(c
.lon()), qFloor(c
.lat()));
121 QByteArray
*ba
= _data
[tile
];
123 QString
bn(tile
.baseName());
124 QString
fn(fileName(bn
));
125 QString
zn(fn
+ ".zip");
127 if (QFileInfo::exists(zn
)) {
128 QZipReader
zip(zn
, QIODevice::ReadOnly
);
129 ba
= new QByteArray(zip
.fileData(bn
));
130 qreal ele
= height(c
, ba
);
131 _data
.insert(tile
, ba
, ba
->size());
135 if (!file
.open(QIODevice::ReadOnly
)) {
136 qWarning("%s: %s", qPrintable(file
.fileName()),
137 qPrintable(file
.errorString()));
138 _data
.insert(tile
, new QByteArray());
141 ba
= new QByteArray(file
.readAll());
142 qreal ele
= height(c
, ba
);
143 _data
.insert(tile
, ba
, ba
->size());
148 return height(c
, ba
);
151 QList
<Area
> DEM::tiles()
153 static const QRegularExpression
re("([NS])([0-9]{2})([EW])([0-9]{3})");
155 QFileInfoList
files(dir
.entryInfoList(QDir::Files
| QDir::Readable
));
156 QLocale
l(QLocale::system());
159 for (int i
= 0; i
< files
.size(); i
++) {
160 QString
basename(files
.at(i
).baseName());
161 QRegularExpressionMatch
match(re
.match(basename
));
162 if (!match
.hasMatch())
165 int lat
= match
.captured(2).toInt();
166 int lon
= match
.captured(4).toInt();
167 if (match
.captured(1) == "S")
169 if (match
.captured(3) == "W")
172 Area
area(RectC(Coordinates(lon
, lat
+ 1), Coordinates(lon
+ 1, lat
)));
173 area
.setName(basename
);
174 area
.setDescription(files
.at(i
).suffix().toUpper() + ", "
175 + l
.formattedDataSize(files
.at(i
).size()));
184 QDebug
operator<<(QDebug dbg
, const DEM::Tile
&tile
)
186 dbg
.nospace() << "Tile(" << tile
.baseName() << ")";
189 #endif // QT_NO_DEBUG