ObjTestWriter.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2017-2018 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/.

#include <fstream>
#include <vector>
#include "Acts/Geometry/GeometryContext.hpp"
#include "Acts/Geometry/Polyhedron.hpp"
#include "Acts/Geometry/Volume.hpp"
#include "Acts/Surfaces/PlaneSurface.hpp"
#include "Acts/Surfaces/RectangleBounds.hpp"
#include "Acts/Utilities/BoundingBox.hpp"
#include "Acts/Utilities/Definitions.hpp"
#include "Acts/Visualization/ObjVisualization.hpp"

namespace Acts {

using IdentifiedPolyderon = std::tuple<std::string, bool, Polyhedron>;

namespace Test {

struct ObjTestWriter {
  static void writeSectorLinesObj(const std::string& name,
                                  const std::pair<Vector3D, Vector3D>& lineA,
                                  const std::pair<Vector3D, Vector3D>& lineB) {
    std::ofstream ostream;
    ostream.open(name + ".obj");
    ObjVisualization objH;
    objH.line(lineA.first, lineA.second);
    objH.line(lineB.first, lineB.second);
    objH.write(ostream);
    ostream.close();
  }

  static void writeSectorPlanesObj(const std::string& name, double phiSec,
                                   double phiAvg, double hX, double hY) {
    // Construct the helper planes for sectoral building
    auto sectorBounds = std::make_shared<RectangleBounds>(hX, hY);

    Vector3D helperColX(0., 0., 1.);
    Vector3D helperColY(1., 0., 0.);
    Vector3D helperColZ(0., 1., 0.);
    RotationMatrix3D helperRotation;
    helperRotation.col(0) = helperColX;
    helperRotation.col(1) = helperColY;
    helperRotation.col(2) = helperColZ;
    // curvilinear surfaces are boundless
    Transform3D helperTransform{helperRotation};

    auto sectorTransformM = std::make_shared<Transform3D>(helperTransform);
    sectorTransformM->prerotate(AngleAxis3D(phiAvg - phiSec, helperColX));

    auto sectorTransformP = std::make_shared<Transform3D>(helperTransform);
    sectorTransformP->prerotate(AngleAxis3D(phiAvg + phiSec, helperColX));

    auto sectorPlaneM =
        Surface::makeShared<PlaneSurface>(sectorTransformM, sectorBounds);

    auto sectorPlaneP =
        Surface::makeShared<PlaneSurface>(sectorTransformP, sectorBounds);

    std::ofstream ostream;
    ostream.open(name + ".obj");
    ObjVisualization objH;
    sectorPlaneM->polyhedronRepresentation(GeometryContext(), 1).draw(objH);
    sectorPlaneP->polyhedronRepresentation(GeometryContext(), 1).draw(objH);
    objH.write(ostream);
    ostream.close();
  }

  static void writeObj(const std::vector<IdentifiedPolyderon>& iphs) {
    for (const auto& iph : iphs) {
      std::ofstream ostream;
      ostream.open(std::get<std::string>(iph) + ".obj");
      ObjVisualization objH;
      std::get<Polyhedron>(iph).draw(objH, std::get<bool>(iph));
      objH.write(ostream);
      ostream.close();
    }
  }

  static void writeObj(const std::string& name,
                       const Volume::BoundingBox& bBox) {
    std::ofstream ostream;
    ostream.open(name + std::string(".obj"));
    ObjVisualization objH;
    bBox.draw(objH);
    objH.write(ostream);
    ostream.close();
  }
};

}  // namespace Test

}  // namespace Acts