| Geant4 Cross Reference |
1 // 1
2 // *******************************************
3 // * License and Disclaimer
4 // *
5 // * The Geant4 software is copyright of th
6 // * the Geant4 Collaboration. It is provided
7 // * conditions of the Geant4 Software License
8 // * LICENSE and available at http://cern.ch/
9 // * include a list of copyright holders.
10 // *
11 // * Neither the authors of this software syst
12 // * institutes,nor the agencies providing fin
13 // * work make any representation or warran
14 // * regarding this software system or assum
15 // * use. Please see the license in the file
16 // * for the full disclaimer and the limitatio
17 // *
18 // * This code implementation is the result
19 // * technical work of the GEANT4 collaboratio
20 // * By using, copying, modifying or distri
21 // * any work based on the software) you ag
22 // * use in resulting scientific publicati
23 // * acceptance of all terms of the Geant4 Sof
24 // *******************************************
25 //
26 // G4BuffercoutDestination class implementatio
27 //
28 // Author: A.Dotti (SLAC), 14 April 2017
29 // -------------------------------------------
30
31 #include "G4BuffercoutDestination.hh"
32
33 #include "G4AutoLock.hh"
34
35 #include <iostream>
36
37 // Private class to implement buffering of log
38 class G4BuffercoutDestination::BufferImpl
39 {
40 public:
41 using FlushFn_t = std::function<void(const s
42
43 public:
44 explicit BufferImpl(std::size_t maxSize) : m
45 explicit BufferImpl(std::size_t maxSize, Flu
46
47 ~BufferImpl() = default;
48
49 // Set number of characters to hold before F
50 // If buffer exceeds new maximum, Flush() wi
51 void SetMaxSize(std::size_t n) { m_maxSize =
52
53 // Reset buffer without flushing
54 void Reset()
55 {
56 m_buffer.str("");
57 m_buffer.clear();
58 m_currentSize = 0;
59 }
60
61 G4int Receive(const G4String& msg)
62 {
63 m_currentSize += msg.size();
64 m_buffer << msg;
65
66 if (m_maxSize > 0 && m_currentSize > m_max
67 return Flush();
68 }
69 return 0;
70 }
71
72 // Flush buffer to destination and reset it
73 G4int Flush()
74 {
75 m_flushFn(m_buffer.str());
76 Reset();
77 return 0;
78 }
79
80 private:
81 std::size_t m_maxSize = 0;
82 std::ostringstream m_buffer;
83 std::size_t m_currentSize = 0;
84 FlushFn_t m_flushFn = [](auto& s) { std::cou
85 };
86
87 // -------------------------------------------
88 G4BuffercoutDestination::G4BuffercoutDestinati
89 : m_maxSize(max),
90 m_buffer_dbg(std::make_unique<BufferImpl>(
91 m_buffer_out(std::make_unique<BufferImpl>(
92 m_buffer_err(std::make_unique<BufferImpl>(
93 {}
94
95 // -------------------------------------------
96 G4BuffercoutDestination::~G4BuffercoutDestinat
97
98 // -------------------------------------------
99 void G4BuffercoutDestination::Finalize()
100 {
101 FlushG4cerr();
102 FlushG4cout();
103 FlushG4debug();
104 }
105
106 // -------------------------------------------
107 G4int G4BuffercoutDestination::ReceiveG4debug(
108 {
109 return m_buffer_dbg->Receive(msg);
110 }
111
112 // -------------------------------------------
113 G4int G4BuffercoutDestination::ReceiveG4cout(c
114 {
115 return m_buffer_out->Receive(msg);
116 }
117
118 // -------------------------------------------
119 G4int G4BuffercoutDestination::ReceiveG4cerr(c
120 {
121 return m_buffer_err->Receive(msg);
122 }
123
124 // -------------------------------------------
125 G4int G4BuffercoutDestination::FlushG4debug()
126 {
127 return m_buffer_dbg->Flush();
128 }
129
130 // -------------------------------------------
131 G4int G4BuffercoutDestination::FlushG4cout()
132 {
133 return m_buffer_out->Flush();
134 }
135
136 // -------------------------------------------
137 G4int G4BuffercoutDestination::FlushG4cerr()
138 {
139 return m_buffer_err->Flush();
140 }
141
142 // -------------------------------------------
143 void G4BuffercoutDestination::SetMaxSize(std::
144 {
145 m_maxSize = max;
146 m_buffer_dbg->SetMaxSize(m_maxSize);
147 m_buffer_out->SetMaxSize(m_maxSize);
148 m_buffer_err->SetMaxSize(m_maxSize);
149 }
150