DigitizationCsvOutput.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file DigitizationCsvOutput.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2020 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Definitions/Algebra.hpp"

#include <fstream>

namespace ActsFatras {

struct DigitizationCsvOutput {
  void writeLine(std::ostream& outf, const Acts::Vector2& p0,
                 const Acts::Vector2& p1) const {
    outf << "l," << p0.x() << "," << p0.y() << "," << p1.x() << "," << p1.y()
         << "\n";
  };

  void writeArc(std::ostream& outf, double r, double phiMin,
                double phiMax) const {
    outf << "a," << r << "," << r << "," << phiMin << "," << phiMax << "\n";
  };

  void writePolygon(std::ostream& outf,
                    const std::vector<Acts::Vector2>& vertices,
                    const Acts::Vector2& shift = Acts::Vector2(0., 0.)) const {
    auto nvertices = vertices.size();
    for (unsigned long iv = 1; iv < nvertices; ++iv) {
      writeLine(outf, vertices[iv - 1] + shift, vertices[iv] + shift);
    }
    writeLine(outf, vertices[nvertices - 1] + shift, vertices[0] + shift);
  };
};

}  // namespace ActsFatras