CoordinateTransformer.h

Go to the documentation of this file.

// Copyright 2019-2023 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#ifndef ALICEO2_TRD_COORDINATE_TRANSFORMER_H_
#define ALICEO2_TRD_COORDINATE_TRANSFORMER_H_
 
 
#include "DataFormatsTRD/Hit.h"
 
#include <array>
 
namespace o2::trd
{
 
class Geometry;
class CoordinateTransformer;
 
class ChamberSpacePoint
{
 public:
  ChamberSpacePoint(int det = -999) : mDetector(det){};
  ChamberSpacePoint(int id, int detector, float x, float y, float z, std::array<float, 3> rct, bool inDrift)
    : mID(id), mDetector(detector), mX(x), mY(y), mZ(z), mPadrow(rct[0]), mPadcol(rct[1]), mTimebin(rct[2]), mInDrift(inDrift){};
 
  bool isValid() const { return mDetector >= 0; }
 
  bool isFromDriftRegion() const { return mInDrift; }
 
  int getID() const { return mID; }
 
  float getX() const { return mX; }
 
  float getY() const { return mY; }
 
  float getZ() const { return mZ; }
 
  int getDetector() const { return mDetector; }
 
  int getPadRow() const { return int(floor(mPadrow)); }
  float getPadRowF() const { return mPadrow; }
 
  float getPadCol() const { return mPadcol; }
 
  float getTimeBin() const { return mTimebin; }
 
  float getMCMChannel(int mcmcol) const;
 
  bool isInMCM(int detector, int padrow, int mcmcol) const;
 
  // int getMCM() const { return o2::trd::HelperMethods::getMCMfromPad(mPadrow, mPadcol); }
 
  // int getROB() const { return o2::trd::HelperMethods::getROBfromPad(mPadrow, mPadcol); }
 
 protected:
  float mX, mY, mZ;
  bool mInDrift;
  int mID;
  int mDetector;
  float mPadrow, mPadcol, mTimebin;
 
  // static constexpr float xscale = 1.0 / (o2::trd::Geometry::cheight() + o2::trd::Geometry::cspace());
  // static constexpr float xoffset = o2::trd::Geometry::getTime0(0);
  // static constexpr float alphascale = 1.0 / o2::trd::Geometry::getAlpha();
};
 
std::ostream& operator<<(std::ostream& os, const ChamberSpacePoint& p);
 
class HitPoint : public ChamberSpacePoint
{
 public:
  HitPoint(ChamberSpacePoint position, float charge)
    : ChamberSpacePoint(position), mCharge(charge)
  {
  }
 
  HitPoint(){};
 
  float getCharge() { return mCharge; }
 
 private:
  float mCharge{0.0};
};
 
class TrackSegment
{
 public:
  TrackSegment(){}; // default constructor, will create invalid start and end points
  TrackSegment(ChamberSpacePoint start, ChamberSpacePoint end, int id)
    : mStartPoint(start), mEndPoint(end), mTrackID(id)
  {
    assert(start.getDetector() == end.getDetector());
  }
 
  bool isValid() const { return mStartPoint.isValid(); }
 
  float getPadColAtTimeBin(float timebin = 0) const
  {
    return mStartPoint.getPadCol() - getSlope() * (mStartPoint.getTimeBin() - timebin);
  }
 
  float getSlope() const
  {
    return (mEndPoint.getPadCol() - mStartPoint.getPadCol()) / (mEndPoint.getTimeBin() - mStartPoint.getTimeBin());
  }
 
  int getDetector() const { return mStartPoint.getDetector(); }
 
  ChamberSpacePoint& getStartPoint() { return mStartPoint; }
  ChamberSpacePoint& getEndPoint() { return mEndPoint; }
 
 protected:
  ChamberSpacePoint mStartPoint, mEndPoint;
  int mTrackID;
 
  // int mDetector;
};
 
// std::ostream& operator<<(std::ostream& os, const ChamberSpacePoint& p);
 
class CoordinateTransformer
{
 public:
  static CoordinateTransformer* instance()
  {
    static CoordinateTransformer mCTrans;
    return &mCTrans;
  }
 
  std::array<float, 3> Local2RCT(int det, float x, float y, float z);
 
  std::array<float, 3> Local2RCT(o2::trd::Hit& hit)
  {
    return Local2RCT(hit.GetDetectorID(), hit.getLocalT(), hit.getLocalC(), hit.getLocalR());
  }
 
  o2::trd::ChamberSpacePoint MakeSpacePoint(o2::trd::Hit& hit);
 
  std::array<float, 3> OrigLocal2RCT(int det, float x, float y, float z);
 
  std::array<float, 3> OrigLocal2RCT(o2::trd::Hit& hit)
  {
    return OrigLocal2RCT(hit.GetDetectorID(), hit.getLocalT(), hit.getLocalC(), hit.getLocalR());
  }
 
  float GetVdrift() { return mVdrift; }
  void SetVdrift(float x) { mVdrift = x; }
 
  float GetT0() { return mT0; }
  void SetT0(float x) { mT0 = x; }
 
  float GetExB() { return mExB; }
  void SetExB(float x) { mExB = x; }
 
 protected:
  o2::trd::Geometry* mGeo;
  float mVdrift{1.5625}; 
  float mT0{4.0};        
  float mExB{0.140};     
 
 private:
  CoordinateTransformer();
};
 
} // namespace o2::trd
 
#endif // ALICEO2_TRD_RAWDISPLAY_H_