ObjSurfaceWriter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// This file was largely imported from the Acts testing framework
 
#pragma once
 
#include "Acts/Geometry/GeometryContext.hpp"
#include <fstream>
#include <iostream>
#include <mutex>
#include "ObjHelper.h"
#include "Acts/Surfaces/Surface.hpp"
#include "Acts/Utilities/Logger.hpp"
 
namespace Acts {
 
class ObjSurfaceWriter
{
public:
  // @class Config
  //
  // The nested config class
  class Config
  {
  public:
    std::shared_ptr<const Acts::Logger> logger;
    std::string name;
    unsigned int outputPhiSegments = 72;
    double outputThickness = 2.;
    bool outputSensitive = true;
    bool outputLayerSurface = true;
    double outputScalor = 1.;
    unsigned int outputPrecision = 6;
    std::string filePrefix = "";
    std::string planarPrefix   = "";
    std::string cylinderPrefix = "";
    std::string diskPrefix     = "";
    std::shared_ptr<std::ofstream> outputStream = nullptr;
 
    Config(const std::string&   lname = "ObjSurfaceWriter",
           Acts::Logging::Level lvl   = Acts::Logging::INFO)
      : logger(Acts::getDefaultLogger(lname, lvl)), name(lname)
    {
    }
  };
 
  ObjSurfaceWriter(const Config& cfg);
 
  const std::string &
  name() const;
 
  void
  write(const Acts::GeometryContext &gctx, const Acts::Surface &surface);
 
  void
  write(const std::string& sinfo);
 
private:
  Config                  m_cfg;         
  ObjHelper::VtnCounter m_vtnCounter;  
  std::mutex m_write_mutex;  
 
  const Acts::Logger&
  logger() const
  {
    return *m_cfg.logger;
  }
};
 
}