ATLAS Offline Software
Loading...
Searching...
No Matches
AthenaPoolCnvSvc Class Reference

This class provides the interface between Athena and PoolSvc. More...

#include <AthenaPoolCnvSvc.h>

Inheritance diagram for AthenaPoolCnvSvc:
Collaboration diagram for AthenaPoolCnvSvc:

Public Member Functions

virtual StatusCode initialize () override
 Required of all Gaudi Services.
virtual StatusCode io_reinit () override
virtual StatusCode finalize () override
 Required of all Gaudi Services.
virtual StatusCode io_finalize () override
virtual StatusCode stop () override
virtual StatusCode createObj (IOpaqueAddress *pAddress, DataObject *&refpObject) override
 Implementation of IConversionSvc: Create the transient representation of an object from persistent state.
virtual StatusCode createRep (DataObject *pObject, IOpaqueAddress *&refpAddress) override
 Implementation of IConversionSvc: Convert the transient object to the requested representation.
virtual StatusCode fillRepRefs (IOpaqueAddress *pAddress, DataObject *pObject) override
 Implementation of IConversionSvc: Resolve the references of the converted object.
virtual StatusCode connectOutput (const std::string &outputConnectionSpec, const std::string &openMode) override
 Implementation of IConversionSvc: Connect to the output connection specification with open mode.
virtual StatusCode connectOutput (const std::string &outputConnectionSpec) override
 Implementation of IConversionSvc: Connect to the output connection specification with open mode.
virtual StatusCode commitOutput (const std::string &outputConnectionSpec, bool doCommit) override
 Implementation of IConversionSvc: Commit pending output.
virtual StatusCode disconnectOutput (const std::string &outputConnectionSpec) override
 Disconnect to the output connection.
virtual IPoolSvcgetPoolSvc () override
virtual TokenregisterForWrite (Placement *placement, const void *obj, const RootType &classDesc) override
virtual void setObjPtr (void *&obj, const Token *token) override
virtual bool useDetailChronoStat () const override
StatusCode createAddress (long svcType, const CLID &clid, const std::string *par, const unsigned long *ip, IOpaqueAddress *&refpAddress) override
 Create a Generic address using explicit arguments to identify a single object.
virtual StatusCode createAddress (long svcType, const CLID &clid, const std::string &refAddress, IOpaqueAddress *&refpAddress) override
 Create address from string form.
virtual StatusCode convertAddress (const IOpaqueAddress *pAddress, std::string &refAddress) override
 Convert address to string form.
virtual StatusCode registerCleanUp (IAthenaPoolCleanUp *cnv) override
 Implement registerCleanUp to register a IAthenaPoolCleanUp to be called during cleanUp.
virtual StatusCode cleanUp (const std::string &connection) override
 Implement cleanUp to call all registered IAthenaPoolCleanUp cleanUp() function.
virtual StatusCode setInputAttributes (const std::string &fileName) override
 Set the input file attributes, if any are requested from jobOpts.
virtual void handle (const Incident &incident) override
 Implementation of IIncidentListener: Handle for EndEvent incidence.
void flushDataHeaderForms (const std::string &streamName="*")
 Tell DataHeaderCnv to write out all DataHeaderForms for a given streamName (default is all)
 AthenaPoolCnvSvc (const std::string &name, ISvcLocator *pSvcLocator)
 Standard Service Constructor.
virtual ~AthenaPoolCnvSvc ()=default
 Destructor.

Protected Member Functions

unsigned outputContextId (const std::string &outputConnection)

Protected Attributes

PMonUtils::BasicStopWatchResultMap_t m_chronoMap {}
 Map that holds chrono information.
Gaudi::Property< bool > m_persSvcPerOutput {this,"PersSvcPerOutput",true}
 PersSvcPerOutput, boolean property to use multiple persistency services, one per output stream.
Gaudi::Property< std::string > m_persSvcPerInputType {this,"PersSvcPerInputType",""}
 PersSvcPerInputType, string property, tree name to use multiple persistency services, one per input type.
std::mutex m_mutex
std::set< unsigned int > m_processedContextIds
 Track context IDs for which extractPoolAttributes has been called.
Gaudi::Property< bool > m_oneDataHeaderForm { this, "OneDataHeaderForm", false }
 If true, use only one DataHeaderForm per Stream.
Gaudi::Property< int > m_DHFormCacheSize { this, "maxDHFormCacheSize", 100 }
 Property for DataHeaderCnv input DHForm cache size.
Gaudi::Property< bool > m_DHFilterAliases { this, "doFilterDHAliases", true }
 Flag to control SG alias filtering when writing out DataHeader (see DataHeaderCnv_p6)

Private Member Functions

void extractPoolAttributes (const Gaudi::Property< std::vector< std::string > > &property, std::vector< std::vector< std::string > > *contAttr, std::vector< std::vector< std::string > > *dbAttr, std::vector< std::vector< std::string > > *domAttr=0) const
 Extract POOL ItechnologySpecificAttributes for Domain, Database and Container from property.
StatusCode processPoolAttributes (std::vector< std::vector< std::string > > &attr, const std::string &fileName, unsigned long contextId, bool doGet=true, bool doSet=true, bool doClear=true) const
 Set/get technology dependent POOL attributes.

Private Attributes

std::string m_lastInputFileName
 decoded storage tech requested in "StorageTechnology" property
ServiceHandle< IPoolSvcm_poolSvc {this,"PoolSvc","PoolSvc"}
ServiceHandle< IClassIDSvc > m_clidSvc {this,"ClassIDSvc","ClassIDSvc"}
Gaudi::Property< bool > m_useDetailChronoStat {this,"UseDetailChronoStat",false}
 UseDetailChronoStat, enable detailed output for time and size statistics for AthenaPOOL: default = false.
Gaudi::Property< std::string > m_containerNamingSchemeProp {this, "PoolContainerNamingScheme", "Historical"}
 POOL container naming scheme selection.
Gaudi::Property< std::vector< std::string > > m_poolAttr {this,"PoolAttributes",{},"Pool Attributes","OrderedSet<std::string>"}
 Output PoolAttributes, vector with names and values of technology specific attributes for POOL.
std::vector< std::vector< std::string > > m_domainAttr
std::vector< std::vector< std::string > > m_databaseAttr
std::vector< std::vector< std::string > > m_containerAttr
Gaudi::Property< std::vector< std::string > > m_inputPoolAttr {this,"InputPoolAttributes",{}}
 Input PoolAttributes, vector with names and values of technology specific attributes for POOL.
std::vector< std::vector< std::string > > m_inputAttr
Gaudi::Property< std::vector< std::string > > m_inputPoolAttrPerEvent {this,"PrintInputAttrPerEvt",{}}
 Print input PoolAttributes per event, vector with names of technology specific attributes for POOL to be printed each event.
std::vector< std::vector< std::string > > m_inputAttrPerEvent

Friends

class SvcFactory< AthenaPoolCnvSvc >

Detailed Description

This class provides the interface between Athena and PoolSvc.

Definition at line 40 of file AthenaPoolCnvSvc.h.

Constructor & Destructor Documentation

◆ AthenaPoolCnvSvc()

AthenaPoolCnvSvc::AthenaPoolCnvSvc ( const std::string & name,
ISvcLocator * pSvcLocator )

Standard Service Constructor.

Definition at line 462 of file AthenaPoolCnvSvc.cxx.

462 :
463 base_class(name, pSvcLocator, pool::POOL_StorageType.type()) {
464}
static const DbType POOL_StorageType
Definition DbType.h:84

◆ ~AthenaPoolCnvSvc()

virtual AthenaPoolCnvSvc::~AthenaPoolCnvSvc ( )
virtualdefault

Destructor.

Member Function Documentation

◆ cleanUp()

StatusCode AthenaPoolCnvSvc::cleanUp ( const std::string & connection)
overridevirtual

Implement cleanUp to call all registered IAthenaPoolCleanUp cleanUp() function.

Definition at line 411 of file AthenaPoolCnvSvc.cxx.

411 {
412 bool retError = false;
413 std::size_t cpos = connection.find(':');
414 std::size_t bpos = connection.find('[');
415 if (cpos == std::string::npos) {
416 cpos = 0;
417 } else {
418 cpos++;
419 }
420 if (bpos != std::string::npos) bpos = bpos - cpos;
421 const std::string conn = connection.substr(cpos, bpos);
422 ATH_MSG_VERBOSE("Cleanup for Connection='"<< conn <<"'");
423 for (auto converter : m_cnvs) {
424 if (!converter->cleanUp(conn).isSuccess()) {
425 ATH_MSG_WARNING("AthenaPoolConverter cleanUp failed.");
426 retError = true;
427 }
428 }
429 return(retError ? StatusCode::FAILURE : StatusCode::SUCCESS);
430}
#define ATH_MSG_VERBOSE(x)
#define ATH_MSG_WARNING(x)

◆ commitOutput()

StatusCode AthenaPoolCnvSvc::commitOutput ( const std::string & outputConnectionSpec,
bool doCommit )
overridevirtual

Implementation of IConversionSvc: Commit pending output.

Parameters
doCommit[IN] boolean to force full commit

Definition at line 266 of file AthenaPoolCnvSvc.cxx.

266 {
267 // This is called after all DataObjects are converted.
268 std::string outputConnection = outputConnectionSpec.substr(0, outputConnectionSpec.find('['));
269 // StopWatch listens from here until the end of this current scope
270 PMonUtils::BasicStopWatch stopWatch("commitOutput", m_chronoMap);
271 std::unique_lock<std::mutex> lock(m_mutex);
272 unsigned int contextId = outputContextId(outputConnection);
273 if (!processPoolAttributes(m_domainAttr, outputConnection, contextId).isSuccess()) {
274 ATH_MSG_DEBUG("commitOutput failed process POOL domain attributes.");
275 }
276 if (!processPoolAttributes(m_databaseAttr, outputConnection, contextId).isSuccess()) {
277 ATH_MSG_DEBUG("commitOutput failed process POOL database attributes.");
278 }
279 if (!processPoolAttributes(m_containerAttr, outputConnection, contextId).isSuccess()) {
280 ATH_MSG_DEBUG("commitOutput failed process POOL container attributes.");
281 }
282
283 // lock.unlock(); //MN: first need to make commitCache slot-specific
284 try {
285 if (doCommit) {
286 if (!m_poolSvc->commit(contextId).isSuccess()) {
287 ATH_MSG_ERROR("commitOutput FAILED to commit OutputStream.");
288 return(StatusCode::FAILURE);
289 }
290 } else {
291 if (!m_poolSvc->commitAndHold(contextId).isSuccess()) {
292 ATH_MSG_ERROR("commitOutput FAILED to commitAndHold OutputStream.");
293 return(StatusCode::FAILURE);
294 }
295 }
296 } catch (std::exception& e) {
297 ATH_MSG_ERROR("commitOutput - caught exception: " << e.what());
298 return(StatusCode::FAILURE);
299 }
300 if (!this->cleanUp(outputConnection).isSuccess()) {
301 ATH_MSG_ERROR("commitOutput FAILED to cleanup converters.");
302 return(StatusCode::FAILURE);
303 }
304
305 return(StatusCode::SUCCESS);
306}
#define ATH_MSG_ERROR(x)
#define ATH_MSG_DEBUG(x)
StatusCode processPoolAttributes(std::vector< std::vector< std::string > > &attr, const std::string &fileName, unsigned long contextId, bool doGet=true, bool doSet=true, bool doClear=true) const
Set/get technology dependent POOL attributes.
ServiceHandle< IPoolSvc > m_poolSvc
virtual StatusCode cleanUp(const std::string &connection) override
Implement cleanUp to call all registered IAthenaPoolCleanUp cleanUp() function.

◆ connectOutput() [1/2]

StatusCode AthenaPoolCnvSvc::connectOutput ( const std::string & outputConnectionSpec)
overridevirtual

Implementation of IConversionSvc: Connect to the output connection specification with open mode.

Parameters
outputConnectionSpec[IN] the name of the output connection specification as string.

Definition at line 238 of file AthenaPoolCnvSvc.cxx.

238 {
239// This is called before DataObjects are being converted.
240 std::string outputConnection = outputConnectionSpec.substr(0, outputConnectionSpec.find('['));
241 unsigned int contextId = outputContextId(outputConnection);
242 try {
243 if (!m_poolSvc->connect(pool::ITransaction::UPDATE, contextId).isSuccess()) {
244 ATH_MSG_ERROR("connectOutput FAILED to open an UPDATE transaction.");
245 return(StatusCode::FAILURE);
246 }
247 } catch (std::exception& e) {
248 ATH_MSG_ERROR("connectOutput - caught exception: " << e.what());
249 return(StatusCode::FAILURE);
250 }
251 std::unique_lock<std::mutex> lock(m_mutex);
252 if (m_processedContextIds.insert(contextId).second) {
253 // Extracting OUTPUT POOL ItechnologySpecificAttributes for Domain, Database and Container.
254 extractPoolAttributes(m_poolAttr, &m_containerAttr, &m_databaseAttr, &m_domainAttr);
255 }
256 if (!processPoolAttributes(m_domainAttr, outputConnection, contextId).isSuccess()) {
257 ATH_MSG_DEBUG("connectOutput failed process POOL domain attributes.");
258 }
259 if (!processPoolAttributes(m_databaseAttr, outputConnection, contextId).isSuccess()) {
260 ATH_MSG_DEBUG("connectOutput failed process POOL database attributes.");
261 }
262 return(StatusCode::SUCCESS);
263}
void extractPoolAttributes(const Gaudi::Property< std::vector< std::string > > &property, std::vector< std::vector< std::string > > *contAttr, std::vector< std::vector< std::string > > *dbAttr, std::vector< std::vector< std::string > > *domAttr=0) const
Extract POOL ItechnologySpecificAttributes for Domain, Database and Container from property.
std::set< unsigned int > m_processedContextIds
Track context IDs for which extractPoolAttributes has been called.

◆ connectOutput() [2/2]

StatusCode AthenaPoolCnvSvc::connectOutput ( const std::string & outputConnectionSpec,
const std::string & openMode )
overridevirtual

Implementation of IConversionSvc: Connect to the output connection specification with open mode.

Parameters
outputConnectionSpec[IN] the name of the output connection specification as string.
openMode[IN] the open mode of the file as string.

Definition at line 233 of file AthenaPoolCnvSvc.cxx.

234 {
235 return(connectOutput(outputConnectionSpec));
236}

◆ convertAddress()

StatusCode AthenaPoolCnvSvc::convertAddress ( const IOpaqueAddress * pAddress,
std::string & refAddress )
overridevirtual

Convert address to string form.

Parameters
pAddress[IN] address to be converted.
refAddress[OUT] converted string form.

Definition at line 394 of file AthenaPoolCnvSvc.cxx.

395 {
396 assert(pAddress);
397 const TokenAddress* tokAddr = dynamic_cast<const TokenAddress*>(pAddress);
398 if (tokAddr != nullptr && tokAddr->getToken() != nullptr) {
399 refAddress = tokAddr->getToken()->toString();
400 } else {
401 refAddress = *pAddress->par();
402 }
403 return(StatusCode::SUCCESS);
404}
This class provides a Generic Transient Address for POOL tokens.
Token * getToken()
virtual const std::string toString() const
Retrieve the string representation of the token.
Definition Token.cxx:134

◆ createAddress() [1/2]

StatusCode AthenaPoolCnvSvc::createAddress ( long svcType,
const CLID & clid,
const std::string & refAddress,
IOpaqueAddress *& refpAddress )
overridevirtual

Create address from string form.

Parameters
svcType[IN] service type of the address.
clid[IN] class id for the address.
refAddress[IN] string form to be converted.
refpAddress[OUT] converted address.

Definition at line 382 of file AthenaPoolCnvSvc.cxx.

385 {
386 if (svcType != repSvcType()) {
387 ATH_MSG_ERROR("createAddress: svcType != POOL_StorageType " << svcType << " " << repSvcType());
388 return(StatusCode::FAILURE);
389 }
390 refpAddress = new GenericAddress(repSvcType(), clid, refAddress);
391 return(StatusCode::SUCCESS);
392}

◆ createAddress() [2/2]

StatusCode AthenaPoolCnvSvc::createAddress ( long svcType,
const CLID & clid,
const std::string * par,
const unsigned long * ip,
IOpaqueAddress *& refpAddress )
override

Create a Generic address using explicit arguments to identify a single object.

Parameters
svcType[IN] service type of the address.
clid[IN] class id for the address.
par[IN] string containing the database name.
ip[IN] object identifier.
refpAddress[OUT] converted address.

Definition at line 356 of file AthenaPoolCnvSvc.cxx.

360 {
361 if( svcType != repSvcType() ) {
362 ATH_MSG_ERROR("createAddress: svcType != POOL_StorageType " << svcType << " " << repSvcType());
363 return(StatusCode::FAILURE);
364 }
365 std::unique_ptr<Token> token;
366 if (par[0].compare(0, 3, "SHM") == 0) {
367 token = std::make_unique<Token>();
368 token->setOid(Token::OID_t(ip[0], ip[1]));
369 token->setAuxString("[PNAME=" + par[2] + "]");
370 RootType classDesc = RootType::ByNameNoQuiet(par[2]);
371 token->setClassID(pool::DbReflex::guid(classDesc));
372 } else {
373 token.reset(m_poolSvc->getToken(par[0], par[1], ip[0]));
374 }
375 if (token == nullptr) {
376 return(StatusCode::RECOVERABLE);
377 }
378 refpAddress = new TokenAddress(repSvcType(), clid, "", par[1], IPoolSvc::kInputStream, std::move(token));
379 return(StatusCode::SUCCESS);
380}
TTypeAdapter RootType
Definition RootType.h:211
@ kInputStream
Definition IPoolSvc.h:40
static TScopeAdapter ByNameNoQuiet(const std::string &name, Bool_t load=kTRUE)
Definition RootType.cxx:586
static Guid guid(const TypeH &id)
Determine Guid (normalized string form) from reflection type.

◆ createObj()

StatusCode AthenaPoolCnvSvc::createObj ( IOpaqueAddress * pAddress,
DataObject *& refpObject )
overridevirtual

Implementation of IConversionSvc: Create the transient representation of an object from persistent state.

Parameters
pAddress[IN] pointer to IOpaqueAddress of the representation.
refpObject[OUT] pointer to DataObject to be created.

Definition at line 137 of file AthenaPoolCnvSvc.cxx.

137 {
138 assert(pAddress);
139 std::string objName = "ALL";
140 if (m_useDetailChronoStat.value()) {
141 if (m_clidSvc->getTypeNameOfID(pAddress->clID(), objName).isFailure()) {
142 objName = std::to_string(pAddress->clID());
143 }
144 objName += '#';
145 objName += *(pAddress->par() + 1);
146 }
147 // StopWatch listens from here until the end of this current scope
148 PMonUtils::BasicStopWatch stopWatch("cObj_" + objName, m_chronoMap);
149 if (!m_persSvcPerInputType.value().empty()) { // Use separate PersistencySvc for each input data type
150 TokenAddress* tokAddr = dynamic_cast<TokenAddress*>(pAddress);
151 if (tokAddr != nullptr && tokAddr->getToken() != nullptr && (tokAddr->getToken()->contID().starts_with(m_persSvcPerInputType.value() + "(") || tokAddr->getToken()->contID().starts_with(m_persSvcPerInputType.value() + "_"))) {
152 const unsigned int maxContext = m_poolSvc->getInputContextMapSize();
153 const unsigned int auxContext = m_poolSvc->getInputContext(tokAddr->getToken()->classID().toString() + tokAddr->getToken()->dbID().toString(), 1);
154 if (m_poolSvc->getInputContextMapSize() > maxContext) {
155 if (!processPoolAttributes(m_inputAttr, m_lastInputFileName, auxContext, false, true, false).isSuccess()) {
156 ATH_MSG_DEBUG("setInputAttribute failed setting POOL database/container attributes.");
157 }
158 }
159 char text[32];
160 const std::string contextStr = std::format("[CTXT={:08X}]", auxContext);
161 std::strncpy(text, contextStr.c_str(), sizeof(text) - 1);
162 text[sizeof(text) - 1] = '\0';
163 tokAddr->getToken()->setAuxString(text);
164 }
165 }
166 // Forward to base class createObj
167 StatusCode status = ::AthCnvSvc::createObj(pAddress, refpObject);
168 return(status);
169}
virtual StatusCode createObj(IOpaqueAddress *pAddress, DataObject *&refpObject) override
Implementation of IConverter: Create the transient representation of an object.
Gaudi::Property< std::string > m_persSvcPerInputType
PersSvcPerInputType, string property, tree name to use multiple persistency services,...
constexpr void toString(std::span< char, StrLen > buf, bool uppercase=true) const noexcept
Automatic conversion to string representation.
Token & setAuxString(const std::string &auxString)
Set auxiliary string.
Definition Token.h:93
const std::string & contID() const
Access container identifier.
Definition Token.h:69
const Guid & classID() const
Access database identifier.
Definition Token.h:73
const Guid & dbID() const
Access database identifier.
Definition Token.h:64
::StatusCode StatusCode
StatusCode definition for legacy code.
status
Definition merge.py:16

◆ createRep()

StatusCode AthenaPoolCnvSvc::createRep ( DataObject * pObject,
IOpaqueAddress *& refpAddress )
overridevirtual

Implementation of IConversionSvc: Convert the transient object to the requested representation.

Parameters
pObject[IN] pointer to DataObject.
refpAddress[OUT] pointer to IOpaqueAddress of the representation to be created.

Definition at line 171 of file AthenaPoolCnvSvc.cxx.

171 {
172 assert(pObject);
173 std::string objName = "ALL";
174 if (m_useDetailChronoStat.value()) {
175 if (m_clidSvc->getTypeNameOfID(pObject->clID(), objName).isFailure()) {
176 objName = std::to_string(pObject->clID());
177 }
178 objName += '#';
179 objName += pObject->registry()->name();
180 }
181 // StopWatch listens from here until the end of this current scope
182 PMonUtils::BasicStopWatch stopWatch("cRep_" + objName, m_chronoMap);
183 StatusCode status = StatusCode::FAILURE;
184 if (pObject->clID() == 1) {
185 // No transient object was found use cnv to write default persistent object
186 SG::DataProxy* proxy = dynamic_cast<SG::DataProxy*>(pObject->registry());
187 if (proxy != nullptr) {
188 IConverter* cnv = converter(proxy->clID());
189 status = cnv->createRep(pObject, refpAddress);
190 }
191 } else {
192 // Forward to base class createRep
193 try {
194 status = ::AthCnvSvc::createRep(pObject, refpAddress);
195 } catch(std::runtime_error& e) {
196 ATH_MSG_FATAL(e.what());
197 }
198 }
199 return(status);
200}
#define ATH_MSG_FATAL(x)
virtual StatusCode createRep(DataObject *pObject, IOpaqueAddress *&refpAddress) override
Implementation of IConverter: Convert the transient object to the requested representation.

◆ disconnectOutput()

StatusCode AthenaPoolCnvSvc::disconnectOutput ( const std::string & outputConnectionSpec)
overridevirtual

Disconnect to the output connection.

Definition at line 309 of file AthenaPoolCnvSvc.cxx.

309 {
310 std::string outputConnection = outputConnectionSpec.substr(0, outputConnectionSpec.find('['));
311 unsigned int contextId = outputContextId(outputConnection);
312 StatusCode sc = m_poolSvc->disconnect(contextId);
313 return sc;
314}
static Double_t sc

◆ extractPoolAttributes()

void AthenaPoolCnvSvc::extractPoolAttributes ( const Gaudi::Property< std::vector< std::string > > & property,
std::vector< std::vector< std::string > > * contAttr,
std::vector< std::vector< std::string > > * dbAttr,
std::vector< std::vector< std::string > > * domAttr = 0 ) const
private

Extract POOL ItechnologySpecificAttributes for Domain, Database and Container from property.

Definition at line 466 of file AthenaPoolCnvSvc.cxx.

469 {
470 std::vector<std::string> opt;
471 std::string attributeName, containerName, databaseName, valueString;
472 for (const auto& propertyValue : property.value()) {
473 opt.clear();
474 attributeName.clear();
475 containerName.clear();
476 databaseName.clear();
477 valueString.clear();
478 using Gaudi::Utils::AttribStringParser;
479 for (const AttribStringParser::Attrib& attrib : AttribStringParser (propertyValue)) {
480 if (attrib.tag == "DatabaseName") {
481 databaseName = attrib.value;
482 } else if (attrib.tag == "ContainerName") {
483 if (databaseName.empty()) {
484 databaseName = "*";
485 }
486 containerName = attrib.value;
487 } else {
488 attributeName = attrib.tag;
489 valueString = attrib.value;
490 }
491 }
492 if (!attributeName.empty() && !valueString.empty()) {
493 opt.push_back(attributeName);
494 opt.push_back(valueString);
495 if (!databaseName.empty()) {
496 opt.push_back(databaseName);
497 if (!containerName.empty()) {
498 opt.push_back(containerName);
499 if (containerName.compare(0, 6, "TTree=") == 0) {
500 dbAttr->push_back(opt);
501 } else {
502 contAttr->push_back(opt);
503 }
504 } else {
505 opt.push_back("");
506 dbAttr->push_back(opt);
507 }
508 } else if (domAttr != 0) {
509 domAttr->push_back(opt);
510 } else {
511 opt.push_back("*");
512 opt.push_back("");
513 dbAttr->push_back(opt);
514 }
515 }
516 }
517}

◆ fillRepRefs()

StatusCode AthenaPoolCnvSvc::fillRepRefs ( IOpaqueAddress * pAddress,
DataObject * pObject )
overridevirtual

Implementation of IConversionSvc: Resolve the references of the converted object.

Parameters
pAddress[IN] pointer to IOpaqueAddress of the representation to be resolved.
pObject[IN] pointer to DataObject to be created.

Definition at line 202 of file AthenaPoolCnvSvc.cxx.

202 {
203 assert(pObject);
204 std::string objName = "ALL";
205 if (m_useDetailChronoStat.value()) {
206 if (m_clidSvc->getTypeNameOfID(pObject->clID(), objName).isFailure()) {
207 objName = std::to_string(pObject->clID());
208 }
209 objName += '#';
210 objName += pObject->registry()->name();
211 }
212 // StopWatch listens from here until the end of this current scope
213 PMonUtils::BasicStopWatch stopWatch("fRep_" + objName, m_chronoMap);
214 StatusCode status = StatusCode::FAILURE;
215 if (pObject->clID() == 1) {
216 // No transient object was found use cnv to write default persistent object
217 SG::DataProxy* proxy = dynamic_cast<SG::DataProxy*>(pObject->registry());
218 if (proxy != nullptr) {
219 IConverter* cnv = converter(proxy->clID());
220 status = cnv->fillRepRefs(pAddress, pObject);
221 }
222 } else {
223 // Forward to base class fillRepRefs
224 try {
225 status = ::AthCnvSvc::fillRepRefs(pAddress, pObject);
226 } catch(std::runtime_error& e) {
227 ATH_MSG_FATAL(e.what());
228 }
229 }
230 return(status);
231}
virtual StatusCode fillRepRefs(IOpaqueAddress *pAddress, DataObject *pObject) override
Implementation of IConverter: Resolve the references of the converted object.

◆ finalize()

StatusCode AthenaPoolCnvSvc::finalize ( )
overridevirtual

Required of all Gaudi Services.

Definition at line 103 of file AthenaPoolCnvSvc.cxx.

103 {
104 ATH_MSG_VERBOSE("Finalizing...");
105 // Some algorithms write in finalize(), flush DHForms if any are left
106 flushDataHeaderForms();
107 // Release ClassIDSvc
108 if (!m_clidSvc.release().isSuccess()) {
109 ATH_MSG_WARNING("Cannot release ClassIDSvc.");
110 }
111 // Release PoolSvc
112 if (!m_poolSvc.release().isSuccess()) {
113 ATH_MSG_WARNING("Cannot release PoolSvc.");
114 }
115 // Print Performance Statistics
116 // The pattern AthenaPoolCnvSvc.*PerfStats is ignored in AtlasTest/TestTools/share/post.sh
117 const std::string msgPrefix{"PerfStats "};
118 ATH_MSG_INFO(msgPrefix << std::string(40, '-'));
119 ATH_MSG_INFO(msgPrefix << "Timing Measurements for AthenaPoolCnvSvc");
120 ATH_MSG_INFO(msgPrefix << std::string(40, '-'));
121 for(const auto& [key, value] : m_chronoMap) {
122 ATH_MSG_INFO(msgPrefix << "| " << std::left << std::setw(15) << key << " | "
123 << std::right << std::setw(15) << std::fixed << std::setprecision(0) << value << " ms |");
124 }
125 ATH_MSG_INFO(msgPrefix << std::string(40, '-'));
126
127 m_cnvs.clear();
128 m_cnvs.shrink_to_fit();
129 return(StatusCode::SUCCESS);
130}
#define ATH_MSG_INFO(x)

◆ flushDataHeaderForms()

void AthenaPoolCnvSvc::flushDataHeaderForms ( const std::string & streamName = "*")

Tell DataHeaderCnv to write out all DataHeaderForms for a given streamName (default is all)

Definition at line 89 of file AthenaPoolCnvSvc.cxx.

89 {
90 // Write remaining DataHeaderForms for a given streamName, "*"" means all
91 auto DHCnvListener = dynamic_cast<IIncidentListener*>( converter( ClassID_traits<DataHeader>::ID() ) );
92 FileIncident incident(name(), "WriteDataHeaderForms", streamName);
93 if( DHCnvListener ) DHCnvListener->handle(incident);
94}

◆ getPoolSvc()

IPoolSvc * AthenaPoolCnvSvc::getPoolSvc ( )
overridevirtual
Returns
pointer to PoolSvc instance.

Definition at line 323 of file AthenaPoolCnvSvc.cxx.

323 {
324 return(&*m_poolSvc);
325}

◆ handle()

void AthenaPoolCnvSvc::handle ( const Incident & incident)
overridevirtual

Implementation of IIncidentListener: Handle for EndEvent incidence.

Definition at line 454 of file AthenaPoolCnvSvc.cxx.

454 {
455 if (incident.type() == "EndEvent") {
456 if (!processPoolAttributes(m_inputAttrPerEvent, m_lastInputFileName, IPoolSvc::kInputStream).isSuccess()) {
457 ATH_MSG_DEBUG("handle EndEvent failed process POOL database attributes.");
458 }
459 }
460}

◆ initialize()

StatusCode AthenaPoolCnvSvc::initialize ( )
overridevirtual

Required of all Gaudi Services.

Definition at line 35 of file AthenaPoolCnvSvc.cxx.

35 {
36 // Initialize DataModelCompatSvc
37 ServiceHandle<IService> dmcsvc("DataModelCompatSvc", this->name());
38 ATH_CHECK(dmcsvc.retrieve());
39 // Retrieve PoolSvc
40 ATH_CHECK(m_poolSvc.retrieve());
41 // Retrieve ClassIDSvc
42 ATH_CHECK(m_clidSvc.retrieve());
43 // Register this service for 'I/O' events
44 ServiceHandle<IIoComponentMgr> iomgr("IoComponentMgr", name());
45 ATH_CHECK(iomgr.retrieve());
46 if (!iomgr->io_register(this).isSuccess()) {
47 ATH_MSG_FATAL("Could not register myself with the IoComponentMgr !");
48 return(StatusCode::FAILURE);
49 }
50 // Global POOL container naming scheme
51 if (auto scheme = APRDefaults::parseNamingScheme(m_containerNamingSchemeProp.value())) {
53 } else {
54 ATH_MSG_ERROR(std::format("Invalid PoolContainerNamingScheme: {}, see APRDefaults.h for the full list.", m_containerNamingSchemeProp.value()));
55 return StatusCode::FAILURE;
56 }
57 // Extracting INPUT POOL ItechnologySpecificAttributes for Domain, Database and Container.
58 extractPoolAttributes(m_inputPoolAttr, &m_inputAttr, &m_inputAttr, &m_inputAttr);
59 // Extracting the INPUT POOL ItechnologySpecificAttributes which are to be printed for each event
60 extractPoolAttributes(m_inputPoolAttrPerEvent, &m_inputAttrPerEvent, &m_inputAttrPerEvent, &m_inputAttrPerEvent);
61 // Setup incident for EndEvent to print out attributes each event
62 ServiceHandle<IIncidentSvc> incSvc("IncidentSvc", name());
63 long int pri = 1000;
64 if (!m_inputPoolAttrPerEvent.value().empty()) {
65 // Set to be listener for EndEvent
66 incSvc->addListener(this, "EndEvent", pri);
67 ATH_MSG_DEBUG("Subscribed to EndEvent for printing out input file attributes.");
68 }
69 if (!processPoolAttributes(m_inputAttr, "", IPoolSvc::kInputStream, false, true, true).isSuccess()) {
70 ATH_MSG_DEBUG("setInputAttribute failed setting POOL domain attributes.");
71 }
72
73 // Load these dictionaries now, so we don't need to try to do so
74 // while multiple threads are running.
75 TClass::GetClass ("TLeafI");
76 TClass::GetClass ("TLeafL");
77 TClass::GetClass ("TLeafD");
78 TClass::GetClass ("TLeafF");
79
80 return(StatusCode::SUCCESS);
81}
#define ATH_CHECK
Evaluate an expression and check for errors.
Gaudi::Property< std::string > m_containerNamingSchemeProp
POOL container naming scheme selection.
Gaudi::Property< std::vector< std::string > > m_inputPoolAttrPerEvent
Print input PoolAttributes per event, vector with names of technology specific attributes for POOL to...
void setNamingScheme(NamingScheme scheme)
Definition APRDefaults.h:72
std::optional< NamingScheme > parseNamingScheme(std::string_view name)
Definition APRDefaults.h:87

◆ io_finalize()

StatusCode AthenaPoolCnvSvc::io_finalize ( )
overridevirtual

Definition at line 132 of file AthenaPoolCnvSvc.cxx.

132 {
133 ATH_MSG_DEBUG("I/O finalization...");
134 return(StatusCode::SUCCESS);
135}

◆ io_reinit()

StatusCode AthenaPoolCnvSvc::io_reinit ( )
overridevirtual

Definition at line 83 of file AthenaPoolCnvSvc.cxx.

83 {
84 ATH_MSG_DEBUG("I/O reinitialization...");
85 m_processedContextIds.clear();
86 return(StatusCode::SUCCESS);
87}

◆ outputContextId()

unsigned int AthenaPoolCnvSvc::outputContextId ( const std::string & outputConnection)
protected

Definition at line 317 of file AthenaPoolCnvSvc.cxx.

317 {
318 return m_persSvcPerOutput?
319 m_poolSvc->getOutputContext(outputConnection) : (unsigned int)IPoolSvc::kOutputStream;
320}
@ kOutputStream
Definition IPoolSvc.h:40

◆ processPoolAttributes()

StatusCode AthenaPoolCnvSvc::processPoolAttributes ( std::vector< std::vector< std::string > > & attr,
const std::string & fileName,
unsigned long contextId,
bool doGet = true,
bool doSet = true,
bool doClear = true ) const
private

Set/get technology dependent POOL attributes.

Definition at line 519 of file AthenaPoolCnvSvc.cxx.

524 {
525 bool retError = false;
526 for (auto& attrEntry : attr) {
527 if (attrEntry.size() == 2) {
528 const std::string& opt = attrEntry[0];
529 std::string data = attrEntry[1];
530 if (data == "int" || data == "DbLonglong" || data == "double" || data == "string") {
531 if (doGet) {
532 if (!m_poolSvc->getAttribute(opt, data, pool::DbType(pool::ROOTTREE_StorageType).type(), contextId).isSuccess()) {
533 ATH_MSG_DEBUG("getAttribute failed for domain attr " << opt);
534 retError = true;
535 }
536 }
537 } else if (doSet) {
538 if (m_poolSvc->setAttribute(opt, data, pool::DbType(pool::ROOTTREE_StorageType).type(), contextId).isSuccess()) {
539 ATH_MSG_DEBUG("setAttribute " << opt << " to " << data);
540 if (doClear) {
541 attrEntry.clear();
542 }
543 } else {
544 ATH_MSG_DEBUG("setAttribute failed for domain attr " << opt << " to " << data);
545 retError = true;
546 }
547 }
548 }
549 if (attrEntry.size() == 4) {
550 const std::string& opt = attrEntry[0];
551 std::string data = attrEntry[1];
552 const std::string& file = attrEntry[2];
553 const std::string& cont = attrEntry[3];
554 if (!fileName.empty() && (0 == fileName.compare(0, fileName.find('?'), file)
555 || (file[0] == '*' && file.find("," + fileName + ",") == std::string::npos))) {
556 if (data == "int" || data == "DbLonglong" || data == "double" || data == "string") {
557 if (doGet) {
558 if (!m_poolSvc->getAttribute(opt, data, pool::DbType(pool::ROOTTREE_StorageType).type(), fileName, cont, contextId).isSuccess()) {
559 ATH_MSG_DEBUG("getAttribute failed for database/container attr " << opt);
560 retError = true;
561 }
562 }
563 } else if (doSet) {
564 if (m_poolSvc->setAttribute(opt, data, pool::DbType(pool::ROOTTREE_StorageType).type(), fileName, cont, contextId).isSuccess()) {
565 ATH_MSG_DEBUG("setAttribute " << opt << " to " << data << " for db: " << fileName << " and cont: " << cont);
566 if (doClear) {
567 if (file[0] == '*' && !m_persSvcPerOutput) {
568 attrEntry[2] += "," + fileName + ",";
569 } else {
570 attrEntry.clear();
571 }
572 }
573 } else {
574 ATH_MSG_DEBUG("setAttribute failed for " << opt << " to " << data << " for db: " << fileName << " and cont: " << cont);
575 retError = true;
576 }
577 }
578 }
579 }
580 }
581 std::erase_if(attr, [](const auto& entry) { return entry.empty(); });
582 return(retError ? StatusCode::FAILURE : StatusCode::SUCCESS);
583}
char data[hepevt_bytes_allocation_ATLAS]
Definition HepEvt.cxx:11
int type() const
Access to full type.
Definition DbType.h:65
static const DbType ROOTTREE_StorageType
Definition DbType.h:87
std::size_t erase_if(T_container &container, T_Func pred)
TFile * file

◆ registerCleanUp()

StatusCode AthenaPoolCnvSvc::registerCleanUp ( IAthenaPoolCleanUp * cnv)
overridevirtual

Implement registerCleanUp to register a IAthenaPoolCleanUp to be called during cleanUp.

Definition at line 406 of file AthenaPoolCnvSvc.cxx.

406 {
407 m_cnvs.push_back(cnv);
408 return(StatusCode::SUCCESS);
409}

◆ registerForWrite()

Token * AthenaPoolCnvSvc::registerForWrite ( Placement * placement,
const void * obj,
const RootType & classDesc )
overridevirtual
Returns
a string token to a Data Object written to Pool
Parameters
placement[IN] pointer to the placement hint
obj[IN] pointer to the Data Object to be written to Pool
classDesc[IN] pointer to the Seal class description for the Data Object.

Definition at line 327 of file AthenaPoolCnvSvc.cxx.

327 {
328 // StopWatch listens from here until the end of this current scope
329 PMonUtils::BasicStopWatch stopWatch("cRepR_ALL", m_chronoMap);
330 Token* token = nullptr;
331 if (m_persSvcPerOutput) { // Use separate PersistencySvc for each output stream/file
332 char text[32];
333 const std::string contextStr = std::format("[CTXT={:08X}]", m_poolSvc->getOutputContext(placement->fileName()));
334 std::strncpy(text, contextStr.c_str(), sizeof(text) - 1);
335 text[sizeof(text) - 1] = '\0';
336 placement->setAuxString(text);
337 }
338 token = m_poolSvc->registerForWrite(placement, obj, classDesc);
339 return(token);
340}
Placement & setAuxString(const std::string &auxString)
Set auxiliary string.
Definition Placement.h:42
const std::string & fileName() const
Access file name.
Definition Placement.h:28
This class provides a token that identifies in a unique way objects on the persistent storage.
Definition Token.h:21

◆ setInputAttributes()

StatusCode AthenaPoolCnvSvc::setInputAttributes ( const std::string & fileName)
overridevirtual

Set the input file attributes, if any are requested from jobOpts.

Parameters
fileName[IN] name of the input file

Definition at line 432 of file AthenaPoolCnvSvc.cxx.

432 {
433 // Set attributes for input file
434 m_lastInputFileName = fileName; // Save file name for printing attributes per event
435 if (!m_persSvcPerInputType.empty()) {
436// Loop over all extra event input contexts
437 const auto& extraInputContextMap = m_poolSvc->getInputContextMap();
438 for (const auto& [label, id]: extraInputContextMap) {
439 if (!processPoolAttributes(m_inputAttr, m_lastInputFileName, id, false, true, false).isSuccess()) {
440 ATH_MSG_DEBUG("setInputAttribute failed setting POOL database/container attributes.");
441 }
442 }
443 }
444 if (!processPoolAttributes(m_inputAttr, m_lastInputFileName, IPoolSvc::kInputStream, false, true, false).isSuccess()) {
445 ATH_MSG_DEBUG("setInputAttribute failed setting POOL database/container attributes.");
446 }
447 if (!processPoolAttributes(m_inputAttr, m_lastInputFileName, IPoolSvc::kInputStream, true, false).isSuccess()) {
448 ATH_MSG_DEBUG("setInputAttribute failed getting POOL database/container attributes.");
449 }
450 return(StatusCode::SUCCESS);
451}
std::string label(const std::string &format, int i)
Definition label.h:19

◆ setObjPtr()

void AthenaPoolCnvSvc::setObjPtr ( void *& obj,
const Token * token )
overridevirtual
Parameters
obj[OUT] pointer to the Data Object.
token[IN] string token of the Data Object for which a Pool Ref is filled.

Definition at line 342 of file AthenaPoolCnvSvc.cxx.

342 {
343 ATH_MSG_VERBOSE("Requesting object for: " << token->toString());
344 // StopWatch listens from here until the end of this current scope
345 PMonUtils::BasicStopWatch stopWatch("cObjR_ALL", m_chronoMap);
346 if (token->dbID() != Guid::null()) {
347 ATH_MSG_VERBOSE("Requesting object for: " << token->toString());
348 m_poolSvc->setObjPtr(obj, token);
349 }
350}
static const Guid & null() noexcept
NULL-Guid: static class method.
Definition Guid.cxx:14

◆ stop()

StatusCode AthenaPoolCnvSvc::stop ( )
overridevirtual

Definition at line 96 of file AthenaPoolCnvSvc.cxx.

96 {
97 ATH_MSG_VERBOSE("stop()");
98 // In case of direct writing without an OutputStream, this should be a good time to flush DHForms
99 flushDataHeaderForms();
100 return StatusCode::SUCCESS;
101}

◆ useDetailChronoStat()

bool AthenaPoolCnvSvc::useDetailChronoStat ( ) const
overridevirtual
Returns
a boolean for using detailed time and size statistics.

Definition at line 352 of file AthenaPoolCnvSvc.cxx.

352 {
353 return(m_useDetailChronoStat.value());
354}

◆ SvcFactory< AthenaPoolCnvSvc >

friend class SvcFactory< AthenaPoolCnvSvc >
friend

Definition at line 1 of file AthenaPoolCnvSvc.h.

Member Data Documentation

◆ m_chronoMap

PMonUtils::BasicStopWatchResultMap_t AthenaPoolCnvSvc::m_chronoMap {}
protected

Map that holds chrono information.

Definition at line 177 of file AthenaPoolCnvSvc.h.

177{};

◆ m_clidSvc

ServiceHandle<IClassIDSvc> AthenaPoolCnvSvc::m_clidSvc {this,"ClassIDSvc","ClassIDSvc"}
private

Definition at line 173 of file AthenaPoolCnvSvc.h.

173{this,"ClassIDSvc","ClassIDSvc"};

◆ m_containerAttr

std::vector<std::vector<std::string> > AthenaPoolCnvSvc::m_containerAttr
private

Definition at line 191 of file AthenaPoolCnvSvc.h.

◆ m_containerNamingSchemeProp

Gaudi::Property<std::string> AthenaPoolCnvSvc::m_containerNamingSchemeProp {this, "PoolContainerNamingScheme", "Historical"}
private

POOL container naming scheme selection.

Definition at line 185 of file AthenaPoolCnvSvc.h.

185{this, "PoolContainerNamingScheme", "Historical"};

◆ m_databaseAttr

std::vector<std::vector<std::string> > AthenaPoolCnvSvc::m_databaseAttr
private

Definition at line 190 of file AthenaPoolCnvSvc.h.

◆ m_DHFilterAliases

Gaudi::Property<bool> AthenaPoolCnvSvc::m_DHFilterAliases { this, "doFilterDHAliases", true }
protected

Flag to control SG alias filtering when writing out DataHeader (see DataHeaderCnv_p6)

Definition at line 222 of file AthenaPoolCnvSvc.h.

222{ this, "doFilterDHAliases", true };

◆ m_DHFormCacheSize

Gaudi::Property<int> AthenaPoolCnvSvc::m_DHFormCacheSize { this, "maxDHFormCacheSize", 100 }
protected

Property for DataHeaderCnv input DHForm cache size.

Definition at line 219 of file AthenaPoolCnvSvc.h.

219{ this, "maxDHFormCacheSize", 100 };

◆ m_domainAttr

std::vector<std::vector<std::string> > AthenaPoolCnvSvc::m_domainAttr
private

Definition at line 189 of file AthenaPoolCnvSvc.h.

◆ m_inputAttr

std::vector<std::vector<std::string> > AthenaPoolCnvSvc::m_inputAttr
private

Definition at line 195 of file AthenaPoolCnvSvc.h.

◆ m_inputAttrPerEvent

std::vector<std::vector<std::string> > AthenaPoolCnvSvc::m_inputAttrPerEvent
private

Definition at line 199 of file AthenaPoolCnvSvc.h.

◆ m_inputPoolAttr

Gaudi::Property<std::vector<std::string> > AthenaPoolCnvSvc::m_inputPoolAttr {this,"InputPoolAttributes",{}}
private

Input PoolAttributes, vector with names and values of technology specific attributes for POOL.

Definition at line 194 of file AthenaPoolCnvSvc.h.

194{this,"InputPoolAttributes",{}};

◆ m_inputPoolAttrPerEvent

Gaudi::Property<std::vector<std::string> > AthenaPoolCnvSvc::m_inputPoolAttrPerEvent {this,"PrintInputAttrPerEvt",{}}
private

Print input PoolAttributes per event, vector with names of technology specific attributes for POOL to be printed each event.

Definition at line 198 of file AthenaPoolCnvSvc.h.

198{this,"PrintInputAttrPerEvt",{}};

◆ m_lastInputFileName

std::string AthenaPoolCnvSvc::m_lastInputFileName
private

decoded storage tech requested in "StorageTechnology" property

Definition at line 171 of file AthenaPoolCnvSvc.h.

◆ m_mutex

std::mutex AthenaPoolCnvSvc::m_mutex
protected

Definition at line 210 of file AthenaPoolCnvSvc.h.

◆ m_oneDataHeaderForm

Gaudi::Property<bool> AthenaPoolCnvSvc::m_oneDataHeaderForm { this, "OneDataHeaderForm", false }
protected

If true, use only one DataHeaderForm per Stream.

Definition at line 216 of file AthenaPoolCnvSvc.h.

216{ this, "OneDataHeaderForm", false };

◆ m_persSvcPerInputType

Gaudi::Property<std::string> AthenaPoolCnvSvc::m_persSvcPerInputType {this,"PersSvcPerInputType",""}
protected

PersSvcPerInputType, string property, tree name to use multiple persistency services, one per input type.

default = "", no tree name results in a single persistency service.

Definition at line 209 of file AthenaPoolCnvSvc.h.

209{this,"PersSvcPerInputType",""};

◆ m_persSvcPerOutput

Gaudi::Property<bool> AthenaPoolCnvSvc::m_persSvcPerOutput {this,"PersSvcPerOutput",true}
protected

PersSvcPerOutput, boolean property to use multiple persistency services, one per output stream.

default = true.

Definition at line 204 of file AthenaPoolCnvSvc.h.

204{this,"PersSvcPerOutput",true};

◆ m_poolAttr

Gaudi::Property<std::vector<std::string> > AthenaPoolCnvSvc::m_poolAttr {this,"PoolAttributes",{},"Pool Attributes","OrderedSet<std::string>"}
private

Output PoolAttributes, vector with names and values of technology specific attributes for POOL.

Definition at line 188 of file AthenaPoolCnvSvc.h.

188{this,"PoolAttributes",{},"Pool Attributes","OrderedSet<std::string>"};

◆ m_poolSvc

ServiceHandle<IPoolSvc> AthenaPoolCnvSvc::m_poolSvc {this,"PoolSvc","PoolSvc"}
private

Definition at line 172 of file AthenaPoolCnvSvc.h.

172{this,"PoolSvc","PoolSvc"};

◆ m_processedContextIds

std::set<unsigned int> AthenaPoolCnvSvc::m_processedContextIds
protected

Track context IDs for which extractPoolAttributes has been called.

Definition at line 213 of file AthenaPoolCnvSvc.h.

◆ m_useDetailChronoStat

Gaudi::Property<bool> AthenaPoolCnvSvc::m_useDetailChronoStat {this,"UseDetailChronoStat",false}
private

UseDetailChronoStat, enable detailed output for time and size statistics for AthenaPOOL: default = false.

Definition at line 182 of file AthenaPoolCnvSvc.h.

182{this,"UseDetailChronoStat",false};

The documentation for this class was generated from the following files: