MaterialEffectsUpdator.h

Go to the documentation of this file.

#ifndef _CR_MATERIALEFFECTSUPDATOR_H_
#define _CR_MATERIALEFFECTSUPDATOR_H_

#include "DataFormats/GeometrySurface/interface/Surface.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "DataFormats/TrajectorySeed/interface/PropagationDirection.h"

namespace materialEffect {
  enum CovIndex { elos = 0, msxx = 1, msxy = 2, msyy = 3 };
  class Covariance {
  public:
    float operator[](CovIndex i) const { return data[i]; }
    float& operator[](CovIndex i) { return data[i]; }
    void add(AlgebraicSymMatrix55& cov) const {
      cov(0, 0) += data[elos];
      cov(1, 1) += data[msxx];
      cov(1, 2) += data[msxy];
      cov(2, 2) += data[msyy];
    }
    Covariance& operator+=(Covariance const& cov) {
      for (int i = 0; i != 4; ++i)
        data[i] += cov.data[i];
      return *this;
    }

  private:
    float data[4] = {0};
  };

  struct Effect {
    float weight = 1.f;
    // Change in |p| from material effects.
    float deltaP = 0;
    // Contribution to covariance matrix (in local co-ordinates) from material effects.
    Covariance deltaCov;
    void combine(Effect const& e1, Effect const& e2) {
      weight *= e1.weight * e2.weight;
      deltaP += e1.deltaP + e2.deltaP;
      deltaCov += e1.deltaCov;
      deltaCov += e2.deltaCov;
    }
  };

}  // namespace materialEffect

class MaterialEffectsUpdator {
public:
  typedef materialEffect::Covariance Covariance;
  typedef materialEffect::Effect Effect;
  typedef materialEffect::CovIndex CovIndex;

  MaterialEffectsUpdator(float mass);
  virtual ~MaterialEffectsUpdator();

  virtual TrajectoryStateOnSurface updateState(const TrajectoryStateOnSurface& TSoS,
                                               const PropagationDirection propDir) const;

  virtual bool updateStateInPlace(TrajectoryStateOnSurface& TSoS, const PropagationDirection propDir) const;

  inline float mass() const { return theMass; }

  virtual MaterialEffectsUpdator* clone() const = 0;

  // here comes the actual computation of the values
  virtual void compute(const TrajectoryStateOnSurface&, const PropagationDirection, Effect& effect) const = 0;

private:
  float theMass;
};

#endif