PrimitivesVisualizationBase.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PrimitivesVisualizationBase.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/Surfaces/PlaneSurface.hpp"
#include "Acts/Surfaces/RectangleBounds.hpp"
#include "Acts/Tests/CommonHelpers/FloatComparisons.hpp"
#include "Acts/Utilities/Definitions.hpp"
#include "Acts/Visualization/EventDataVisualization.hpp"
#include "Acts/Visualization/GeometryVisualization.hpp"
#include "Acts/Visualization/IVisualization.hpp"
#include "Acts/Visualization/ObjVisualization.hpp"
#include "Acts/Visualization/PlyVisualization.hpp"

#include <fstream>
#include <sstream>
#include <string>

namespace Acts {
namespace PrimitivesVisualization {

// Test on a plane
auto identity = std::make_shared<Transform3D>(Transform3D::Identity());
auto rectangle = std::make_shared<RectangleBounds>(10., 10.);
auto plane = Surface::makeShared<PlaneSurface>(identity, rectangle);

// Test context
GeometryContext gctx = GeometryContext();

static inline std::string test(IVisualization& helper) {
  std::stringstream ss;

  // Line visualization ------------------------------------------------
  IVisualization::ColorType lineColor = {0, 0, 250};
  Vector3D start = {1., 1., 1.};
  Vector3D end = {4., 4., 4.};
  Acts::Visualization::drawSegment(helper, start, end, 0.1, 72, lineColor);
  helper.write("Primitives_Line");
  helper.write(ss);
  helper.write(ss);
  helper.clear();

  // Arrows visualization ------------------------------------------------
  start = {1., 0., 0.};
  end = {4., 0., 0.};
  Acts::Visualization::drawArrowForward(helper, start, end, 0.1, 0.1, 3., 72,
                                        {0, 75, 0});

  start = {1., 2., 0.};
  end = {4., 2., 0.};
  Acts::Visualization::drawArrowBackward(helper, start, end, 0.1, 0.1, 3., 72,
                                         {0, 150, 0});

  start = {1., 4., 0.};
  end = {4., 4., 0.};
  Acts::Visualization::drawArrowsBoth(helper, start, end, 0.1, 0.1, 3., 72,
                                      {0, 250, 0});
  helper.write("Primitives_Arrows");
  helper.write(ss);
  helper.clear();

  // Error visualization: local ---------------------------------------------
  IVisualization::ColorType surfaceColor = {120, 250, 0};
  Acts::Visualization::drawSurface(
      helper, *plane, gctx, Transform3D::Identity(), 72, false, surfaceColor);

  IVisualization::ColorType errorColor = {250, 0, 0};

  ActsSymMatrixD<2> cov = ActsSymMatrixD<2>::Identity();
  double s0 = 0.45;
  double s1 = 1.99;
  double r01 = 0.78;
  cov << s0 * s0, r01 * s0 * s1, r01 * s0 * s1, s1 * s1;

  Vector2D lcentered{0., 0.};
  Acts::Visualization::drawCovarianceCartesian(
      helper, lcentered, cov, plane->transform(gctx), {3}, 10., 72, errorColor);

  helper.write("Primitives_CartesianError");
  helper.write(ss);
  helper.clear();

  // Error visualization: angular ---------------------------------------------
  Acts::Visualization::drawSurface(
      helper, *plane, gctx, Transform3D::Identity(), 72, false, surfaceColor);

  // Error visualization
  errorColor = {250, 0, 0};

  cov = ActsSymMatrixD<2>::Identity();
  s0 = 0.08;
  s1 = 0.01;
  r01 = 0.3;
  cov << s0 * s0, r01 * s0 * s1, r01 * s0 * s1, s1 * s1;

  Vector3D origin{0., 0., 0.};
  Vector3D direction = Vector3D(1., 3., 15.).normalized();

  double directionScale = 5.;

  Acts::Visualization::drawCovarianceAngular(
      helper, origin, direction, cov, {3}, directionScale, 10., 72, errorColor);

  Acts::Visualization::drawArrowForward(
      helper, origin + 0.5 * directionScale * direction,
      origin + 1.2 * directionScale * direction, 0.02, 0.1, 5., 72,
      {10, 10, 10});

  helper.write("Primitives_AngularError");
  helper.write(ss);
  helper.clear();

  return ss.str();
}

}  // namespace PrimitivesVisualization
}  // namespace Acts