1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
/*=========================================================================
*
* Copyright NumFOCUS
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#ifndef itkThreadLogger_h
#define itkThreadLogger_h
#include "itkLogger.h"
#include <mutex>
#include <atomic>
#include <string>
#include <queue>
#include <thread>
namespace itk
{
/** \class ThreadLogger
* \brief Providing logging service as a separate thread.
*
* \author Hee-Su Kim, Compute Science Dept. Kyungpook National University,
* ISIS Center, Georgetown University.
*
* \ingroup OSSystemObjects LoggingObjects
* \ingroup ITKCommon
*/
class ITKCommon_EXPORT ThreadLogger : public Logger
{
public:
using Self = ThreadLogger;
using Superclass = Logger;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(ThreadLogger);
/** New macro for creation of a Smart Pointer */
itkNewMacro(Self);
using OutputType = Logger::OutputType;
using PriorityLevelEnum = Logger::PriorityLevelEnum;
using DelayType = unsigned int;
/** Definition of types of operations for ThreadLogger. */
typedef enum
{
SET_PRIORITY_LEVEL,
SET_LEVEL_FOR_FLUSHING,
ADD_LOG_OUTPUT,
WRITE,
FLUSH
} OperationType;
/** Set the priority level for the current logger. Only messages that have
* priorities equal or greater than the one set here will be posted to the
* current outputs. */
void
SetPriorityLevel(PriorityLevelEnum level) override;
/** Get the priority level for the current logger. Only messages that have
* priorities equal or greater than the one set here will be posted to the
* current outputs. */
PriorityLevelEnum
GetPriorityLevel() const override;
void
SetLevelForFlushing(PriorityLevelEnum level) override;
PriorityLevelEnum
GetLevelForFlushing() const override;
/** Set the delay in milliseconds between checks to see if there are any
* low priority messages to be processed.
*/
virtual void
SetDelay(DelayType delay);
/** Get the delay in milliseconds between checks to see if there are any
* low priority messages to be processed.
*/
virtual DelayType
GetDelay() const;
/** Registers another output stream with the multiple output. */
void
AddLogOutput(OutputType * output) override;
void
Write(PriorityLevelEnum level, std::string const & content) override;
void
Flush() override;
protected:
/** Constructor */
ThreadLogger();
/** Destructor */
~ThreadLogger() override;
/** Print contents of a ThreadLogger */
void
PrintSelf(std::ostream & os, Indent indent) const override;
void
ThreadFunction();
private:
void
InternalFlush();
std::thread m_Thread{};
std::atomic<bool> m_TerminationRequested{};
std::queue<OperationType> m_OperationQ{};
std::queue<std::string> m_MessageQ{};
std::queue<PriorityLevelEnum> m_LevelQ{};
std::queue<OutputType::Pointer> m_OutputQ{};
mutable std::mutex m_Mutex{};
DelayType m_Delay{};
}; // class ThreadLogger
} // namespace itk
#endif // itkThreadLogger_h
|
