MuonAlignmentCondAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCONDALG_MUONALIGNMENTCONDALG_H
#define MUONCONDALG_MUONALIGNMENTCONDALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "AthenaPoolUtilities/CondAttrListCollection.h"
#include "CoralBase/Blob.h"
#include "StoreGate/CondHandleKeyArray.h"
#include "StoreGate/ReadCondHandleKey.h"
// typedefs for A/BLineMapContainer
#include "CoralUtilities/blobaccess.h"
#include "GaudiKernel/ServiceHandle.h"
#include "MuonAlignmentData/CorrContainer.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "nlohmann/json.hpp"
 
 
class MuonAlignmentCondAlg : public AthReentrantAlgorithm {
public:
    MuonAlignmentCondAlg(const std::string& name, ISvcLocator* pSvcLocator);
    virtual ~MuonAlignmentCondAlg() = default;
    virtual StatusCode initialize() override;
    virtual StatusCode execute(const EventContext& ctx) const override;
    virtual bool isReEntrant() const override { return false; }
 
private:
    template <class ContType>
        StatusCode writeContainer(const EventContext& ctx,
                                  const SG::WriteCondHandleKey<ContType>& writeKey,
                                  std::unique_ptr<ContType>&& container) const;
    
    StatusCode loadDataFromLegacy(const std::string& data, nlohmann::json& json, bool loadBLines) const;
    StatusCode parseDataFromJSON(const nlohmann::json& lines,
                                ALineContainer& writeALineCdo, 
                                BLineContainer& writeBLineCdo) const;
    StatusCode loadCoolFolder(const EventContext& ctx,
                              const SG::ReadCondHandleKey<CondAttrListCollection>& key,
                              ALineContainer& writeALineCdo, 
                              BLineContainer& writeBLineCdo) const;
    
    // Read Handles
    SG::ReadCondHandleKeyArray<CondAttrListCollection> m_alignKeys{this, "ParlineFolders", {
                                                                  "/MUONALIGN/MDT/BARREL",
                                                                  "/MUONALIGN/MDT/ENDCAP/SIDEA",
                                                                  "/MUONALIGN/MDT/ENDCAP/SIDEC",
                                                                  "/MUONALIGN/TGC/SIDEA",
                                                                  "/MUONALIGN/TGC/SIDEC",
                                                                  },
                                                                  "Folder names where the alignment paramters are stored"};
 
    // Write Handles
    SG::WriteCondHandleKey<ALineContainer> m_writeALineKey{this, "WriteALineKey", "ALineContainer",
                                                              "Key of output muon alignment ALine condition data"};
    SG::WriteCondHandleKey<BLineContainer> m_writeBLineKey{this, "WriteBLineKey", "BLineContainer",
                                                              "Key of output muon alignment BLine condition data"};
    
    ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
    
    
    Gaudi::Property<bool> m_loadALines{this, "LoadALines", true, "Setup whether Alines shall be written"};
    Gaudi::Property<bool> m_loadBLines{this, "LoadBLines", true, "Setup whether BLines shall be written"};
    // new folder format 2020
    Gaudi::Property<bool> m_newFormat2020 {this, "NewFormat2020", false, 
                          "The database folders are given in the new JSON format"};
    Gaudi::Property<std::string> m_readFromJSON{this,"readFromJSON", "",
                                 "Reads the A & B lines parameters from a JSON file instead from COOL"};
};
 
#endif