src/common/demloader.cpp

   1 #include <QtMath>
   2 #include <QFileInfo>
   3 #include "common/rectc.h"
   4 #include "demloader.h"
   5
   6
   7 static QList<DEM::Tile> tiles(const RectC &rect)
   8 {
   9         QList<DEM::Tile> list;
  10
  11         if (rect.isNull())
  12                 return list;
  13
  14         for (int i = qFloor(rect.top()); i >= qFloor(rect.bottom()); i--)
  15                 for (int j = qFloor(rect.left()); j <= qFloor(rect.right()); j++)
  16                         list.append(DEM::Tile(j, i));
  17
  18         return list;
  19 }
  20
  21 static bool isZip(const QUrl &url)
  22 {
  23         QFileInfo fi(url.fileName());
  24         return (fi.suffix().toLower() == "zip");
  25 }
  26
  27
  28 DEMLoader::DEMLoader(const QString &dir, QObject *parent)
  29   : QObject(parent), _dir(dir)
  30 {
  31         _downloader = new Downloader(this);
  32         connect(_downloader, &Downloader::finished, this, &DEMLoader::finished);
  33 }
  34
  35 int DEMLoader::numTiles(const RectC &rect) const
  36 {
  37         QList<DEM::Tile> tl(tiles(rect));
  38         int cnt = 0;
  39
  40         for (int i = 0; i < tl.size(); i++) {
  41                 const DEM::Tile &t = tl.at(i);
  42                 QString fn(tileFile(t));
  43                 QString zn(fn + ".zip");
  44
  45                 if (!(QFileInfo::exists(zn) || QFileInfo::exists(fn)))
  46                         cnt++;
  47         }
  48
  49         return cnt;
  50 }
  51
  52 bool DEMLoader::loadTiles(const RectC &rect)
  53 {
  54         QList<DEM::Tile> tl(tiles(rect));
  55         QList<Download> dl;
  56
  57         /* Create the user DEM dir only when a download is requested as it will
  58            override the global DEM dir. */
  59         if (!_dir.mkpath(_dir.absolutePath())) {
  60                 qWarning("%s: %s", qPrintable(_dir.canonicalPath()),
  61                   "Error creating DEM directory");
  62                 return false;
  63         }
  64         DEM::setDir(_dir.path());
  65
  66         for (int i = 0; i < tl.size(); i++) {
  67                 const DEM::Tile &t = tl.at(i);
  68                 QString fn(tileFile(t));
  69                 QString zn(fn + ".zip");
  70
  71                 if (!(QFileInfo::exists(zn) || QFileInfo::exists(fn))) {
  72                         QUrl url(tileUrl(t));
  73                         dl.append(Download(url, isZip(url) ? zn : fn));
  74                 }
  75
  76                 if (dl.size() > DEM_DOWNLOAD_LIMIT) {
  77                         qWarning("DEM download limit (%d) exceeded.", DEM_DOWNLOAD_LIMIT);
  78                         return false;
  79                 }
  80         }
  81
  82         return _downloader->get(dl, _headers);
  83 }
  84
  85 bool DEMLoader::checkTiles(const RectC &rect) const
  86 {
  87         QList<DEM::Tile> tl(tiles(rect));
  88
  89         for (int i = 0; i < tl.size(); i++) {
  90                 const DEM::Tile &t = tl.at(i);
  91                 QString fn(tileFile(t));
  92                 QString zn(fn + ".zip");
  93
  94                 if (!(QFileInfo::exists(zn) || QFileInfo::exists(fn)))
  95                         return false;
  96         }
  97
  98         return true;
  99 }
 100
 101 QUrl DEMLoader::tileUrl(const DEM::Tile &tile) const
 102 {
 103         QString url(_url);
 104
 105         url.replace("$lon", tile.lonStr());
 106         url.replace("$lat", tile.latStr());
 107
 108         return QUrl(url);
 109 }
 110
 111 QString DEMLoader::tileFile(const DEM::Tile &tile) const
 112 {
 113         return _dir.absoluteFilePath(tile.fileName());
 114 }
 115
 116 void DEMLoader::setAuthorization(const Authorization &authorization)
 117 {
 118         QList<HTTPHeader> headers;
 119         if (!authorization.isNull())
 120                 headers.append(authorization.header());
 121         _headers = headers;
 122 }