aboutsummaryrefslogtreecommitdiff
path: root/src/zenutil/progress.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/zenutil/progress.cpp')
-rw-r--r--src/zenutil/progress.cpp102
1 files changed, 102 insertions, 0 deletions
diff --git a/src/zenutil/progress.cpp b/src/zenutil/progress.cpp
new file mode 100644
index 000000000..01f6529f2
--- /dev/null
+++ b/src/zenutil/progress.cpp
@@ -0,0 +1,102 @@
+// Copyright Epic Games, Inc. All Rights Reserved.
+
+#include <zenutil/progress.h>
+
+#include <zencore/logging.h>
+
+namespace zen {
+
+class StandardProgressBase;
+
+class StandardProgressBar : public ProgressBase::ProgressBar
+{
+public:
+ StandardProgressBar(StandardProgressBase& Owner, std::string_view InSubTask) : m_Owner(Owner), m_SubTask(InSubTask) {}
+
+ virtual void UpdateState(const State& NewState, bool DoLinebreak) override;
+ virtual void ForceLinebreak() override {}
+ virtual void Finish() override;
+
+private:
+ LoggerRef Log();
+ StandardProgressBase& m_Owner;
+ std::string m_SubTask;
+ State m_State;
+};
+
+class StandardProgressBase : public ProgressBase
+{
+public:
+ StandardProgressBase(LoggerRef Log) : m_Log(Log) {}
+
+ virtual void SetLogOperationName(std::string_view Name) override
+ {
+ m_LogOperationName = Name;
+ ZEN_INFO("{}", m_LogOperationName);
+ }
+ virtual void SetLogOperationProgress(uint32_t StepIndex, uint32_t StepCount) override
+ {
+ const size_t PercentDone = StepCount > 0u ? (100u * StepIndex) / StepCount : 0u;
+ ZEN_INFO("{}: {}%", m_LogOperationName, PercentDone);
+ }
+ virtual void PushLogOperation(std::string_view Name) override { ZEN_UNUSED(Name); }
+ virtual void PopLogOperation() override {}
+ virtual uint32_t GetProgressUpdateDelayMS() const override { return 2000; }
+ virtual std::unique_ptr<ProgressBar> CreateProgressBar(std::string_view InSubTask) override
+ {
+ return std::make_unique<StandardProgressBar>(*this, InSubTask);
+ }
+
+private:
+ friend class StandardProgressBar;
+ LoggerRef m_Log;
+ std::string m_LogOperationName;
+ LoggerRef Log() { return m_Log; }
+};
+
+LoggerRef
+StandardProgressBar::Log()
+{
+ return m_Owner.Log();
+}
+
+void
+StandardProgressBar::UpdateState(const State& NewState, bool DoLinebreak)
+{
+ ZEN_UNUSED(DoLinebreak);
+ const size_t PercentDone =
+ NewState.TotalCount > 0u ? (100u * (NewState.TotalCount - NewState.RemainingCount)) / NewState.TotalCount : 0u;
+ std::string Task = NewState.Task;
+ switch (NewState.Status)
+ {
+ case State::EStatus::Aborted:
+ Task = "Aborting";
+ break;
+ case State::EStatus::Paused:
+ Task = "Paused";
+ break;
+ default:
+ break;
+ }
+ ZEN_INFO("{}: {}%{}", Task, PercentDone, NewState.Details.empty() ? "" : fmt::format(" {}", NewState.Details));
+ m_State = NewState;
+}
+void
+StandardProgressBar::Finish()
+{
+ if (m_State.RemainingCount > 0)
+ {
+ State NewState = m_State;
+ NewState.RemainingCount = 0;
+ NewState.Details = "";
+ UpdateState(NewState, /*DoLinebreak*/ true);
+ }
+}
+
+ProgressBase*
+CreateStandardProgress(LoggerRef Log)
+{
+ return new StandardProgressBase(Log);
+}
+
+} // namespace zen