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