SpacePointUtility.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2022 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 "Acts/Definitions/TrackParametrization.hpp"
#include "Acts/EventData/SourceLink.hpp"
#include "Acts/Geometry/GeometryContext.hpp"
#include "Acts/Geometry/GeometryIdentifier.hpp"
#include "Acts/Geometry/TrackingGeometry.hpp"
#include "Acts/SpacePointFormation/SpacePointBuilderConfig.hpp"
#include "Acts/Utilities/Result.hpp"

#include <array>
#include <cstddef>
#include <functional>
#include <iostream>
#include <memory>
#include <system_error>
#include <utility>
#include <vector>

namespace Acts {
class SourceLink;

struct SpacePointParameters {
  Vector3 firstBtmToTop;
  Vector3 secondBtmToTop;
  Vector3 vtxToFirstMid2;
  Vector3 vtxToSecondMid2;
  Vector3 firstBtmToTopXvtxToFirstMid2;
  Vector3 secondBtmToTopXvtxToSecondMid2;
  double mag_firstBtmToTop = 0.;
  double m = 0.;
  double n = 0.;
  double limit = 1.;
  double limitExtended = 0.;
};

class SpacePointUtility {
 public:
  SpacePointUtility(SpacePointBuilderConfig cfg) : m_config(std::move(cfg)) {}

  std::pair<Vector3, Vector2> globalCoords(
      const GeometryContext& gctx, const SourceLink& slink,
      const SourceLinkSurfaceAccessor& surfaceAccessor, const BoundVector& par,
      const BoundSquareMatrix& cov) const;

  Vector2 rhoZCovariance(const GeometryContext& gctx, const Surface& surface,
                         const Vector3& globalPos,
                         const SquareMatrix2& localCov) const;

  Vector2 calcRhoZVars(
      const GeometryContext& gctx, const SourceLink& slinkFront,
      const SourceLink& slinkBack,
      const SourceLinkSurfaceAccessor& surfaceAccessor,
      const std::function<std::pair<const BoundVector, const BoundSquareMatrix>(
          SourceLink)>& paramCovAccessor,
      const Vector3& globalPos, const double theta) const;

  Result<void> calculateStripSPPosition(
      const std::pair<Vector3, Vector3>& stripEnds1,
      const std::pair<Vector3, Vector3>& stripEnds2, const Vector3& posVertex,
      SpacePointParameters& spParams, const double stripLengthTolerance) const;

  Result<void> recoverSpacePoint(SpacePointParameters& spParams,
                                 double stripLengthGapTolerance) const;

  Result<double> differenceOfMeasurementsChecked(
      const Vector3& pos1, const Vector3& pos2, const Vector3& posVertex,
      const double maxDistance, const double maxAngleTheta2,
      const double maxAnglePhi2) const;

  Result<double> calcPerpendicularProjection(
      const std::pair<Vector3, Vector3>& stripEnds1,
      const std::pair<Vector3, Vector3>& stripEnds2,
      SpacePointParameters& spParams) const;

 private:
  SpacePointBuilderConfig m_config;
  std::error_code m_error;
};

}  // namespace Acts