diff --git a/k4FWCore/components/Writer.cpp b/k4FWCore/components/Writer.cpp index 64d02d35..7ee461c6 100644 --- a/k4FWCore/components/Writer.cpp +++ b/k4FWCore/components/Writer.cpp @@ -21,8 +21,10 @@ #include "GaudiKernel/AnyDataWrapper.h" #include "GaudiKernel/IDataManagerSvc.h" #include "GaudiKernel/IDataProviderSvc.h" +#include "GaudiKernel/IHiveWhiteBoard.h" #include "GaudiKernel/SmartDataPtr.h" #include "GaudiKernel/StatusCode.h" + #include "podio/Frame.h" #include "IIOSvc.h" @@ -31,7 +33,7 @@ #include "k4FWCore/FunctionalUtils.h" #include "k4FWCore/IMetadataSvc.h" -#include +#include #include #include @@ -156,7 +158,7 @@ class Writer final : public Gaudi::Functional::Consumername() == k4FWCore::frameLocation) { continue; } @@ -190,7 +192,7 @@ class Writer final : public Gaudi::Functional::ConsumerunregisterObject(p); + const auto sc = m_dataSvc->unregisterObject(p); if (!sc.isSuccess()) { error() << "Failed to unregister object" << endmsg; return; @@ -221,7 +223,7 @@ class Writer final : public Gaudi::Functional::ConsumerretrieveObject("/Event/" + coll, storeCollection).isFailure()) { error() << "Failed to retrieve collection " << coll << endmsg; @@ -233,12 +235,12 @@ class Writer final : public Gaudi::Functional::Consumer>*>(storeCollection); - storePtr->getData().release(); + const auto storePtr = dynamic_cast>*>(storeCollection); + [[maybe_unused]] auto releasedPtr = storePtr->getData().release(); delete storePtr; } - for (auto& coll : m_collectionsToAdd) { + for (const auto& coll : m_collectionsToAdd) { DataObject* storeCollection; if (m_dataSvc->retrieveObject("/Event/" + coll, storeCollection).isFailure()) { error() << "Failed to retrieve collection " << coll << endmsg; @@ -257,8 +259,15 @@ class Writer final : public Gaudi::Functional::Consumer(storeCollection); if (!old_collection) { - error() << "Failed to cast collection " << coll << endmsg; - return; + // This can happen for objects that are not collections like in the + // MarlinWrapper for converter maps or a LCEvent, or, in general, + // anything else + info() << "Object in the store with name " << coll + << " does not look like a collection so it can not be written to the output file" << endmsg; + m_collectionsToSave.erase(std::remove(m_collectionsToSave.begin(), m_collectionsToSave.end(), coll), + m_collectionsToSave.end()); + m_collectionsToAdd.erase(std::remove(m_collectionsToAdd.begin(), m_collectionsToAdd.end(), coll), + m_collectionsToAdd.end()); } else { std::unique_ptr uptr( const_cast(old_collection->collectionBase())); diff --git a/test/k4FWCoreTest/CMakeLists.txt b/test/k4FWCoreTest/CMakeLists.txt index 9340b296..0e7291e4 100644 --- a/test/k4FWCoreTest/CMakeLists.txt +++ b/test/k4FWCoreTest/CMakeLists.txt @@ -188,6 +188,7 @@ set_tests_properties(FunctionalWrongImport PROPERTIES PASS_REGULAR_EXPRESSION "I add_test_with_env(FunctionalReadNthEvent options/ExampleFunctionalReadNthEvent.py PROPERTIES DEPENDS FunctionalProducer ADD_TO_CHECK_FILES) add_test_with_env(FunctionalProducerRNTuple options/ExampleFunctionalProducerRNTuple.py ADD_TO_CHECK_FILES) add_test_with_env(FunctionalTTreeToRNTuple options/ExampleFunctionalTTreeToRNTuple.py PROPERTIES DEPENDS FunctionalProducer ADD_TO_CHECK_FILES) +add_test_with_env(GaudiFunctional options/ExampleGaudiFunctional.py PROPERTIES DEPENDS FunctionalProducer ADD_TO_CHECK_FILES) # The following is done to make the tests work without installing the files in diff --git a/test/k4FWCoreTest/options/ExampleGaudiFunctional.py b/test/k4FWCoreTest/options/ExampleGaudiFunctional.py new file mode 100644 index 00000000..ae7ea0b1 --- /dev/null +++ b/test/k4FWCoreTest/options/ExampleGaudiFunctional.py @@ -0,0 +1,43 @@ +# +# Copyright (c) 2014-2024 Key4hep-Project. +# +# This file is part of Key4hep. +# See https://key4hep.github.io/key4hep-doc/ for further info. +# +# 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 +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# 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. +# + +# This is an example reading from a file and using a producer to create new +# data + +from Gaudi.Configuration import INFO +from Configurables import ExampleGaudiFunctionalProducer, ExampleFunctionalTransformer +from Configurables import EventDataSvc +from k4FWCore import ApplicationMgr, IOSvc + +svc = IOSvc("IOSvc") +svc.Input = "functional_producer.root" +svc.Output = "gaudi_functional.root" + +gaudi_producer = ExampleGaudiFunctionalProducer("GaudiProducer", OutputCollectionName="Output") +transformer = ExampleFunctionalTransformer( + "Transformer", InputCollection=["MCParticles"], OutputCollection=["NewMCParticles"] +) + +mgr = ApplicationMgr( + TopAlg=[gaudi_producer, transformer], + EvtSel="NONE", + EvtMax=-1, + ExtSvc=[EventDataSvc("EventDataSvc")], + OutputLevel=INFO, +) diff --git a/test/k4FWCoreTest/scripts/CheckOutputFiles.py b/test/k4FWCoreTest/scripts/CheckOutputFiles.py index 67c98238..518eb087 100644 --- a/test/k4FWCoreTest/scripts/CheckOutputFiles.py +++ b/test/k4FWCoreTest/scripts/CheckOutputFiles.py @@ -70,6 +70,7 @@ def check_metadata(filename, expected_metadata): check_collections("functional_transformer.root", ["MCParticles", "NewMCParticles"]) +check_collections("gaudi_functional.root", ["MCParticles", "NewMCParticles"]) check_collections("functional_transformer_cli.root", ["MCParticles", "NewMCParticles"]) check_collections( "functional_transformer_multiple.root", diff --git a/test/k4FWCoreTest/src/components/ExampleGaudiFunctionalProducer.cpp b/test/k4FWCoreTest/src/components/ExampleGaudiFunctionalProducer.cpp new file mode 100644 index 00000000..8d1250c4 --- /dev/null +++ b/test/k4FWCoreTest/src/components/ExampleGaudiFunctionalProducer.cpp @@ -0,0 +1,36 @@ +/* + * Copyright (c) 2014-2024 Key4hep-Project. + * + * This file is part of Key4hep. + * See https://key4hep.github.io/key4hep-doc/ for further info. + * + * 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 + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * 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. + */ + +#include "Gaudi/Functional/Producer.h" + +#include + +using BaseClass_t = Gaudi::Functional::Traits::BaseClass_t; + +struct ExampleGaudiFunctionalProducer final : Gaudi::Functional::Producer { + // The pair in KeyValues can be changed from python and it corresponds + // to the name of the output collection + ExampleGaudiFunctionalProducer(const std::string& name, ISvcLocator* svcLoc) + : Producer(name, svcLoc, KeyValue{"OutputCollectionName", "OutputCollection"}) {} + + // This is the function that will be called to produce the data + int operator()() const override { return 3; } +}; + +DECLARE_COMPONENT(ExampleGaudiFunctionalProducer)