MdtAsBuiltJsonDumpAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#include "MdtAsBuiltJsonDumpAlg.h"
#include "StoreGate/ReadCondHandle.h"
#include "nlohmann/json.hpp"
#include <fstream>
 
MdtAsBuiltJsonDumpAlg::MdtAsBuiltJsonDumpAlg(const std::string& name, ISvcLocator* pSvcLocator):
      AthAlgorithm{name, pSvcLocator} {}
 
StatusCode MdtAsBuiltJsonDumpAlg::initialize() {
    ATH_CHECK(m_readKey.initialize());   
    ATH_CHECK(m_idHelperSvc.retrieve());
    return StatusCode::SUCCESS;
  }
StatusCode MdtAsBuiltJsonDumpAlg::execute(){
  const EventContext& ctx = Gaudi::Hive::currentContext();
  SG::ReadCondHandle<MdtAsBuiltContainer> asBuiltContainer{m_readKey, ctx};
  if (!asBuiltContainer.isValid()){
    ATH_MSG_FATAL("Failed to load ALine container "<<m_readKey.fullKey());
    return StatusCode::FAILURE;
  }
  std::ofstream ostr{m_jsonFile};
  if (!ostr.good()) {
     ATH_MSG_FATAL("Failed to create output file "<<m_jsonFile);
     return StatusCode::FAILURE;
  }
  ostr<<"["<<std::endl;
  unsigned int nLines{0};
  using multilayer_t = MdtAsBuiltPar::multilayer_t;
  using tubeSide_t = MdtAsBuiltPar::tubeSide_t;
  for (const MdtAsBuiltPar& asBuilt : **asBuiltContainer) {
    ++nLines;
    ostr<<"    {"<<std::endl;
    ostr<<"     \"typ\": \""<<asBuilt.AmdbStation()<<"\","<<std::endl;
    ostr<<"     \"jzz\": "<<asBuilt.AmdbEta()<<", "<<std::endl;
    ostr<<"     \"jff\": "<<asBuilt.AmdbPhi()<<", "<<std::endl;
    for (const multilayer_t ml : {multilayer_t::ML1, multilayer_t::ML2}){
      for (const tubeSide_t side: {tubeSide_t::POS, tubeSide_t::NEG}){
          std::stringstream prefix{};
          prefix<<"Ml"<<(static_cast<unsigned>(ml) + 1);
          prefix<<(side  == tubeSide_t::POS? "Pos" : "Neg")<<"TubeSide";
          
          auto dumpValue = [&prefix,&ostr](const std::string& field, 
                                           const float val,
                                           bool last = false) {
              ostr<<"     \""<<prefix.str()<<field<<"\": ";
              ostr<<val<<(last ? "":", ")<<std::endl;      
          };
          dumpValue("y0", asBuilt.y0(ml, side));
          dumpValue("z0", asBuilt.z0(ml, side));
          dumpValue("alpha", asBuilt.alpha(ml, side));
          dumpValue("ypitch", asBuilt.ypitch(ml, side));
          dumpValue("zpitch", asBuilt.zpitch(ml, side));
          dumpValue("stagg", asBuilt.stagg(ml, side),
                             ml == multilayer_t::ML2 && side == tubeSide_t::NEG);
        }
    }
    ostr<<"    }"<< (nLines != asBuiltContainer->size() ? "," : "")<<std::endl;
  }
  ostr<<"]"<<std::endl;
  ostr.close();
  
  
  return StatusCode::SUCCESS;
}