sid
/
NazaraEngine
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Core/CallOnExit: Rework to use template instead of std::function

This commit is contained in:
Jérôme Leclercq 2022-01-23 00:16:09 +01:00
parent 29786765c6
commit 2ebcddf9de
5 changed files with 85 additions and 100 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
include/Nazara/Core/CallOnExit.hpp
View File

@ -8,29 +8,32 @@
#define NAZARA_CORE_CALLONEXIT_HPP
#include <Nazara/Prerequisites.hpp>
#include <functional>
#include <optional>
namespace Nz
{
template<typename F>
class CallOnExit
{
using Func = std::function<void()>;
public:
CallOnExit(Func func = nullptr);
CallOnExit() = default;
CallOnExit(F&& functor);
CallOnExit(const CallOnExit&) = delete;
CallOnExit(CallOnExit&&) noexcept = delete;
~CallOnExit();
void CallAndReset(Func func = nullptr);
void Reset(Func func = nullptr);
void CallAndReset();
void Reset();
CallOnExit& operator=(const CallOnExit&) = delete;
CallOnExit& operator=(CallOnExit&&) noexcept = default;
private:
Func m_func;
std::optional<F> m_functor;
};
template<typename F>
CallOnExit(F) -> CallOnExit<F>;
}
#include <Nazara/Core/CallOnExit.inl>

32
include/Nazara/Core/CallOnExit.inl
View File

@ -15,24 +15,24 @@ namespace Nz
*/
/*!
* \brief Constructs a CallOnExit object with a function
* \brief Constructs a CallOnExit object with a functor
*
* \param func Function to call on exit
*/
inline CallOnExit::CallOnExit(Func func) :
m_func(func)
template<typename F>
CallOnExit<F>::CallOnExit(F&& functor) :
m_functor(std::move(functor))
{
}
/*!
* \brief Destructs the object and calls the function
*/
inline CallOnExit::~CallOnExit()
template<typename F>
CallOnExit<F>::~CallOnExit()
{
if (m_func)
m_func();
if (m_functor)
(*m_functor)();
}
/*!
@ -40,13 +40,13 @@ namespace Nz
*
* \param func Function to call on exit
*/
inline void CallOnExit::CallAndReset(Func func)
template<typename F>
void CallOnExit<F>::CallAndReset()
{
if (m_func)
m_func();
if (m_functor)
(*m_functor)();
Reset(func);
m_functor.reset();
}
/*!
@ -54,10 +54,10 @@ namespace Nz
*
* \param func Function to call on exit
*/
inline void CallOnExit::Reset(Func func)
template<typename F>
void CallOnExit<F>::Reset()
{
m_func = func;
m_functor.reset();
}
}

51
src/Nazara/Network/TcpClient.cpp
View File

@ -45,15 +45,11 @@ namespace Nz
Disconnect();
Open(remoteAddress.GetProtocol());
CallOnExit restoreBlocking;
CallOnExit restoreBlocking([this] { SocketImpl::SetBlocking(m_handle, true); });
if (m_isBlockingEnabled)
{
SocketImpl::SetBlocking(m_handle, false);
restoreBlocking.Reset([this] ()
{
SocketImpl::SetBlocking(m_handle, true);
});
}
else
restoreBlocking.Reset();
SocketState state = SocketImpl::Connect(m_handle, remoteAddress, &m_lastError);
m_peerAddress = (state != SocketState::NotConnected) ? remoteAddress : IpAddress::Invalid;
@ -363,14 +359,13 @@ namespace Nz
std::size_t totalByteSent = 0;
CallOnExit updateSent;
if (sent)
CallOnExit updateSent([sent, &totalByteSent]
{
updateSent.Reset([sent, &totalByteSent] ()
{
*sent = totalByteSent;
});
}
*sent = totalByteSent;
});
if (!sent)
updateSent.Reset();
while (totalByteSent < size || !IsBlockingEnabled())
{
@ -582,15 +577,11 @@ namespace Nz
{
NazaraAssert(m_handle != SocketImpl::InvalidHandle, "Invalid handle");
CallOnExit restoreBlocking;
if (!m_isBlockingEnabled)
{
SocketImpl::SetBlocking(m_handle, true);
restoreBlocking.Reset([this] ()
{
SocketImpl::SetBlocking(m_handle, false);
});
}
CallOnExit restoreBlocking([this] { SocketImpl::SetBlocking(m_handle, true); });
if (m_isBlockingEnabled)
SocketImpl::SetBlocking(m_handle, false);
else
restoreBlocking.Reset();
std::size_t received;
if (!Receive(buffer, size, &received))
@ -630,15 +621,11 @@ namespace Nz
NazaraAssert(buffer, "Invalid buffer");
NazaraAssert(m_handle != SocketImpl::InvalidHandle, "Invalid handle");
CallOnExit restoreBlocking;
if (!m_isBlockingEnabled)
{
SocketImpl::SetBlocking(m_handle, true);
restoreBlocking.Reset([this] ()
{
SocketImpl::SetBlocking(m_handle, false);
});
}
CallOnExit restoreBlocking([this] { SocketImpl::SetBlocking(m_handle, true); });
if (m_isBlockingEnabled)
SocketImpl::SetBlocking(m_handle, false);
else
restoreBlocking.Reset();
std::size_t sent;
if (!Send(buffer, size, &sent))

34
src/Nazara/Utility/Formats/MTLParser.cpp
View File

@ -33,17 +33,16 @@ namespace Nz
{
m_currentStream = &stream;
// Force stream in text mode, reset it at the end
Nz::CallOnExit resetTextMode;
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
// force stream in text mode, reset it at the end
CallOnExit resetTextMode([&stream]
{
stream.EnableTextMode(true);
stream.EnableTextMode(false);
});
resetTextMode.Reset([&stream] ()
{
stream.EnableTextMode(false);
});
}
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
stream.EnableTextMode(true);
else
resetTextMode.Reset();
m_keepLastLine = false;
m_lineCount = 0;
@ -489,17 +488,16 @@ namespace Nz
{
m_currentStream = &stream;
// Force stream in text mode, reset it at the end
Nz::CallOnExit resetTextMode;
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
// force stream in text mode, reset it at the end
CallOnExit resetTextMode([&stream]
{
stream.EnableTextMode(true);
stream.EnableTextMode(false);
});
resetTextMode.Reset([&stream] ()
{
stream.EnableTextMode(false);
});
}
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
stream.EnableTextMode(true);
else
resetTextMode.Reset();
m_outputStream.str({});

51
src/Nazara/Utility/Formats/OBJParser.cpp
View File

@ -21,17 +21,16 @@ namespace Nz
m_keepLastLine = false;
m_lineCount = 0;
// Force stream in text mode, reset it at the end
Nz::CallOnExit resetTextMode;
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
// force stream in text mode, reset it at the end
CallOnExit resetTextMode([&stream]
{
stream.EnableTextMode(true);
stream.EnableTextMode(false);
});
resetTextMode.Reset([&stream] ()
{
stream.EnableTextMode(false);
});
}
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
stream.EnableTextMode(true);
else
resetTextMode.Reset();
unsigned int failureCount = 0;
while (Advance(false))
@ -93,17 +92,16 @@ namespace Nz
m_keepLastLine = false;
m_lineCount = 0;
// Force stream in text mode, reset it at the end
Nz::CallOnExit resetTextMode;
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
// force stream in text mode, reset it at the end
CallOnExit resetTextMode([&stream]
{
stream.EnableTextMode(true);
stream.EnableTextMode(false);
});
resetTextMode.Reset([&stream] ()
{
stream.EnableTextMode(false);
});
}
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
stream.EnableTextMode(true);
else
resetTextMode.Reset();
std::string matName, meshName;
matName = meshName = "default";
@ -491,17 +489,16 @@ namespace Nz
{
m_currentStream = &stream;
// Force stream in text mode, reset it at the end
Nz::CallOnExit resetTextMode;
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
// force stream in text mode, reset it at the end
CallOnExit resetTextMode([&stream]
{
stream.EnableTextMode(true);
stream.EnableTextMode(false);
});
resetTextMode.Reset([&stream] ()
{
stream.EnableTextMode(false);
});
}
if ((stream.GetStreamOptions() & StreamOption::Text) == 0)
stream.EnableTextMode(true);
else
resetTextMode.Reset();
m_outputStream.str({});