89 lines
3.4 KiB
C++
89 lines
3.4 KiB
C++
#include <mbgl/util/worker.hpp>
|
|
#include <mbgl/util/work_task.hpp>
|
|
#include <mbgl/util/work_request.hpp>
|
|
#include <mbgl/platform/platform.hpp>
|
|
#include <mbgl/map/vector_tile.hpp>
|
|
#include <mbgl/map/live_tile.hpp>
|
|
#include <mbgl/util/pbf.hpp>
|
|
#include <mbgl/renderer/raster_bucket.hpp>
|
|
|
|
#include <cassert>
|
|
#include <future>
|
|
|
|
namespace mbgl {
|
|
|
|
class Worker::Impl {
|
|
public:
|
|
explicit Impl(FileSource* fs) {
|
|
// FIXME: Workers should not access the FileSource but it
|
|
// is currently needed because of GlyphsPBF. See #1664.
|
|
util::ThreadContext::setFileSource(fs);
|
|
}
|
|
|
|
void parseRasterTile(RasterBucket* bucket, std::string data, std::function<void (TileParseResult)> callback) {
|
|
std::unique_ptr<util::Image> image(new util::Image(data));
|
|
if (!(*image)) {
|
|
callback(TileParseResult("error parsing raster image"));
|
|
}
|
|
|
|
if (!bucket->setImage(std::move(image))) {
|
|
callback(TileParseResult("error setting raster image to bucket"));
|
|
}
|
|
|
|
callback(TileParseResult(TileData::State::parsed));
|
|
}
|
|
|
|
void parseVectorTile(TileWorker* worker, std::string data, std::function<void (TileParseResult)> callback) {
|
|
try {
|
|
pbf tilePBF(reinterpret_cast<const unsigned char *>(data.data()), data.size());
|
|
callback(worker->parse(VectorTile(tilePBF)));
|
|
} catch (const std::exception& ex) {
|
|
callback(TileParseResult(ex.what()));
|
|
}
|
|
}
|
|
|
|
void parseLiveTile(TileWorker* worker, const LiveTile* tile, std::function<void (TileParseResult)> callback) {
|
|
try {
|
|
callback(worker->parse(*tile));
|
|
} catch (const std::exception& ex) {
|
|
callback(TileParseResult(ex.what()));
|
|
}
|
|
}
|
|
|
|
void redoPlacement(TileWorker* worker, float angle, bool collisionDebug, std::function<void ()> callback) {
|
|
worker->redoPlacement(angle, collisionDebug);
|
|
callback();
|
|
}
|
|
};
|
|
|
|
Worker::Worker(std::size_t count) {
|
|
util::ThreadContext context = {"Worker", util::ThreadType::Worker, util::ThreadPriority::Low};
|
|
for (std::size_t i = 0; i < count; i++) {
|
|
threads.emplace_back(std::make_unique<util::Thread<Impl>>(context, util::ThreadContext::getFileSource()));
|
|
}
|
|
}
|
|
|
|
Worker::~Worker() = default;
|
|
|
|
std::unique_ptr<WorkRequest> Worker::parseRasterTile(RasterBucket& bucket, std::string data, std::function<void (TileParseResult)> callback) {
|
|
current = (current + 1) % threads.size();
|
|
return threads[current]->invokeWithCallback(&Worker::Impl::parseRasterTile, callback, &bucket, data);
|
|
}
|
|
|
|
std::unique_ptr<WorkRequest> Worker::parseVectorTile(TileWorker& worker, std::string data, std::function<void (TileParseResult)> callback) {
|
|
current = (current + 1) % threads.size();
|
|
return threads[current]->invokeWithCallback(&Worker::Impl::parseVectorTile, callback, &worker, data);
|
|
}
|
|
|
|
std::unique_ptr<WorkRequest> Worker::parseLiveTile(TileWorker& worker, const LiveTile& tile, std::function<void (TileParseResult)> callback) {
|
|
current = (current + 1) % threads.size();
|
|
return threads[current]->invokeWithCallback(&Worker::Impl::parseLiveTile, callback, &worker, &tile);
|
|
}
|
|
|
|
std::unique_ptr<WorkRequest> Worker::redoPlacement(TileWorker& worker, float angle, bool collisionDebug, std::function<void ()> callback) {
|
|
current = (current + 1) % threads.size();
|
|
return threads[current]->invokeWithCallback(&Worker::Impl::redoPlacement, callback, &worker, angle, collisionDebug);
|
|
}
|
|
|
|
} // end namespace mbgl
|