TaskFunctor.h

Go to the documentation of this file.

/*
 * Copyright (c) 2015-2024, EPFL/Blue Brain Project
 *
 * Responsible Author: Daniel.Nachbaur@epfl.ch
 *
 * This file is part of Blue Brain BioExplorer <https://github.com/BlueBrain/BioExplorer>
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License version 3.0 as published
 * by the Free Software Foundation.
 *
 * This library 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 Lesser General Public License for more
 * details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
 
#pragma once
 
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wshadow"
#endif
#include <async++.h>
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
 
#include <string>
 
namespace core
{
class TaskFunctor
{
public:
    using ProgressFunc = std::function<void(std::string, float, float)>;
 
    void setProgressFunc(const ProgressFunc& progressFunc) { _progressFunc = progressFunc; }
 
    void progress(const std::string& message, const float increment, const float amount)
    {
        cancelCheck();
        if (_progressFunc)
            _progressFunc(message, increment, amount);
    }
 
    void setCancelToken(async::cancellation_token& cancelToken) { _cancelToken = &cancelToken; }
 
    void cancelCheck() const
    {
        if (_cancelToken)
            async::interruption_point(*_cancelToken);
    }
 
private:
    async::cancellation_token* _cancelToken{nullptr};
    ProgressFunc _progressFunc;
};
} // namespace core