54 lines
1.3 KiB
C++
54 lines
1.3 KiB
C++
/*
|
|
* This file is part of OpenTTD.
|
|
* OpenTTD 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, version 2.
|
|
* OpenTTD 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 OpenTTD. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
/** @file scope.h Simple scope guard... */
|
|
|
|
#ifndef SCOPE_H
|
|
#define SCOPE_H
|
|
|
|
#include <utility>
|
|
|
|
template <typename T>
|
|
class scope_exit_obj {
|
|
T f;
|
|
bool shouldexec;
|
|
|
|
|
|
public:
|
|
|
|
scope_exit_obj(T &&func)
|
|
: f(std::move(func)), shouldexec(true) { }
|
|
|
|
scope_exit_obj(const scope_exit_obj ©src) = delete;
|
|
scope_exit_obj(scope_exit_obj &&movesrc)
|
|
: f(std::move(movesrc.f)), shouldexec(movesrc.shouldexec) {
|
|
movesrc.shouldexec = false;
|
|
}
|
|
|
|
~scope_exit_obj() {
|
|
exec();
|
|
}
|
|
|
|
void exec() {
|
|
if (shouldexec) {
|
|
f();
|
|
shouldexec = false;
|
|
}
|
|
}
|
|
|
|
void cancel() {
|
|
shouldexec = false;
|
|
}
|
|
};
|
|
|
|
template <typename T>
|
|
scope_exit_obj<typename std::decay<T>::type> scope_guard(T &&func) {
|
|
return scope_exit_obj<typename std::decay<T>::type>(std::forward<T>(func));
|
|
}
|
|
|
|
#endif /* SCOPE_H */
|