ObjExCellWriter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
// This file was largely imported from the Acts testing framework
 
#pragma once
 
#include <TTree.h>
#include <mutex>
#include "Acts/Extrapolation/ExtrapolationCell.hpp"
#include "Acts/Utilities/Logger.hpp"
 
 
 
  template <class T>
  class ObjExCellWriter 
  {
  public:
    // The nested configuration struct
    struct Config
    {
    public:
      double outputScalor = 1.;
      size_t outputPrecision = 4;
      std::shared_ptr<std::ofstream> outputStream = nullptr;
    };
 
    ObjExCellWriter(const Config&        cfg
        //Acts::Logging::Level lvl = Acts::Logging::INFO
        )
    : m_cfg(cfg), m_vCounter(1)
    {
      if (!m_cfg.outputStream) {
        throw std::invalid_argument("Missing output stream");
      }
    }
 
    void
    write(const std::vector<Acts::ExtrapolationCell<T>>& ecells) {
 
      // loop over the cells
      for (auto& eCell : ecells) {
        // remember the first counter
        size_t fCounter = m_vCounter;
 
        // increase the vertex counter
        ++m_vCounter;
        // the event paramters
        auto sPosition = eCell.startParameters->position();
        // write the space point
        (*(m_cfg.outputStream))
          << "v " << m_cfg.outputScalor * sPosition.x() << ", "
          << m_cfg.outputScalor * sPosition.y() << ", "
          << m_cfg.outputScalor * sPosition.z() << '\n';
 
        // loop over extrapolation steps
        for (auto& es : eCell.extrapolationSteps) {
          if (es.parameters) {
            const T& pars      = (*es.parameters);
            auto     tPosition = pars.position();
            // increase the counter
            ++m_vCounter;
            // write the space point
            (*(m_cfg.outputStream))
              << "v " << m_cfg.outputScalor * tPosition.x() << ", "
              << m_cfg.outputScalor * tPosition.y() << ", "
              << m_cfg.outputScalor * tPosition.z() << '\n';
          }
        }
        // write out the line
        for (size_t i=fCounter+1;i<m_vCounter;i++) {
          (*(m_cfg.outputStream)) << "l ";
          (*(m_cfg.outputStream)) << (i-1) << " " << i;
          (*(m_cfg.outputStream)) << '\n';
        }
        //(*(m_cfg.outputStream)) << "l ";
        //for (size_t iv = fCounter; iv < m_vCounter; ++iv)
          //(*(m_cfg.outputStream)) << iv << " ";
 
        //(*(m_cfg.outputStream)) << '\n';
        // new line
        (*(m_cfg.outputStream)) << '\n';
      }
    }
 
  private:
    Config m_cfg;
    size_t m_vCounter;
};