TMLink.h

Go to the documentation of this file.

//-----------------------------------------------------------------------------
// $Id: TMLink.h,v 1.8 2010/03/31 09:58:59 liucy Exp $
//-----------------------------------------------------------------------------
// Filename : TMLink.h
// Section  : Tracking
// Owner    : Yoshi Iwasaki
// Email    : yoshihito.iwasaki@kek.jp
//-----------------------------------------------------------------------------
// Description : A class to relate TMDCWireHit and TTrack objects.
//               See http://bsunsrv1.kek.jp/~yiwasaki/tracking/
//-----------------------------------------------------------------------------

#ifndef TMLink_FLAG_
#define TMLink_FLAG_

#include <iostream>

#ifdef TRKRECO_DEBUG_DETAIL
#ifndef TRKRECO_DEBUG
#define TRKRECO_DEBUG
#endif
#endif
#define HEP_SHORT_NAMES
#include <string>
#ifndef CLHEP_POINT3D_H
#include "CLHEP/Geometry/Point3D.h"
#endif
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
    typedef HepGeom::Point3D<double> HepPoint3D;
#endif

#include "CLHEP/Alist/ConstAList.h"
#include "TrkReco/TMDCWire.h"
#include "TrkReco/TMDCWireHit.h"
//#include "TrkReco/TMDCWire.h"
//#include "TrkReco/TMDCClust.h"

class TTrackHEP;

class TTrack;

class TMLink {

  public:
/*    TMLink(TTrack * track = 0,
          const TMDCWireHit * hit = 0,
          const HepPoint3D & position = HepPoint3D());*/
    TMLink(TTrack * track = 0,
          const TMDCWireHit * hit = 0,
          const HepPoint3D & position = HepPoint3D(),
          const HepPoint3D & positionD = HepPoint3D(),
          const double = 0.0);

    TMLink(const TMLink &);

    virtual ~TMLink();

  public:// Selectors
    void dump(const std::string & message = std::string(""),
              const std::string & prefix = std::string("")) const;

    const TMDCWireHit * hit(void) const;

    const TMDCWire * const wire(void) const;

    TTrack * track(void) const;

    double pull(void) const;

    const HepPoint3D & xyPosition(void) const;

    const HepPoint3D & positionOnWire(void) const;

    const HepPoint3D & positionOnTrack(void) const;

    double dPhi(void) const;

    unsigned leftRight(void) const;

    int zStatus(void) const;

    int zPair(void) const;

    const HepPoint3D & position(void) const;
    const HepPoint3D & positionD(void) const;

  
    double distance(void) const;

    TMLink * neighbor(unsigned int) const;

    TMLink * link(void) const;

// S.Suzuki added
    //    int usecathode(void) const;

    //    TMDCClust * getmclust(void) const;
// end S.Suzuki

    const HepPoint3D & arcZ(const unsigned = 0) const;

  public:// tmp
//    const HepPoint3D & conf(void) const;
    const HepPoint3D & conf(const HepPoint3D & conf);

  public:// Modifiers
    void update(const HepPoint3D & onTrack,
                const HepPoint3D & onWire,
                unsigned leftRight,
                double pull);

    const TMDCWireHit * hit(const TMDCWireHit *);

    TTrack * track(TTrack *);

    const HepPoint3D & positionOnWire(const HepPoint3D &);
    const HepPoint3D & positionOnWire(double p[3]);

    const HepPoint3D & positionOnTrack(const HepPoint3D &);
    const HepPoint3D & positionOnTrack(double p[3]);

    double dPhi(double);

    const HepPoint3D & position(const HepPoint3D &);
    const HepPoint3D & positionD(const HepPoint3D &);

    void neighbor(unsigned int, TMLink *);

    TMLink * link(TMLink *);

    unsigned leftRight(unsigned);

    double pull(double);

    int zStatus(int);

    int zPair(int);

// S.Suzuki added

    //    void setusecathode(int);

    //    void setmclust(TMDCClust *);

    //    void setZphiBeforeCathode(float);

// end S.Suzuki

    const HepPoint3D & arcZ(const HepPoint3D &, const unsigned = 0);
    const unsigned fit2D(const unsigned &);
    unsigned fit2D(void);

  public:// Drift distance
    float drift(void) const;
    float drift(unsigned) const;
    float drift(float, unsigned);
    double cDrift(void) const;
    double cDrift(unsigned) const;
    double cDrift(double, unsigned);

    float dDrift(void) const;
    float dDrift(unsigned) const;
    float dDrift(float, unsigned);

    unsigned tsfTag(void) const;
    unsigned tsfTag(unsigned);

    double tof(void) const;
    double tof(double);

    void setDriftTime(double);
    double getDriftTime(void);

   double DriftTime(double,double) const;
  private:
    TTrack * _track;
    const TMDCWireHit * _hit;
    HepPoint3D _onTrack;
    HepPoint3D _onWire;
    HepPoint3D _position;
    HepPoint3D _positionD;
    double _dPhi;
    unsigned _leftRight;
    float _drift[2];
    float _dDrift[2];

    double _cDrift[2];

    int _zStatus;
    int _zPair;
    double _pull;
    TMLink * _neighbor[6];
    TMLink * _link;

    //...tmp...
    HepPoint3D _conf;

    double _tof;
    double _drifttime;

   //-- S. Suzuki added -------
    //    int _usecathode;
    //    float _ZphiBeforeCathode;
    //    TMDCClust * _mclust;
    //-- S. Suzuki added end ---

    HepPoint3D _arcZ[4];
    unsigned _fit2D;
    unsigned _tsfTag;
};

// Utility functions

const TTrackHEP &
Links2HEP(const AList<TMLink> & links);

unsigned
NSuperLayers(const AList<TMLink> & links);

unsigned
NSuperLayers(const AList<TMLink> & links, unsigned minNHits);

unsigned
NMissingAxialSuperLayers(const AList<TMLink> & links);

unsigned
NLayers(const AList<TMLink> & links);

void
NHits(const AList<TMLink> & links, unsigned nHits[50]);

void
NHitsSuperLayer(const AList<TMLink> & links, unsigned nHits[11]);

AList<TMLink>
AxialHits(const AList<TMLink> & links);

AList<TMLink>
StereoHits(const AList<TMLink> & links);

unsigned
NAxialHits(const AList<TMLink> & links);

unsigned
NStereoHits(const AList<TMLink> & links);

unsigned
Width(const AList<TMLink> &);

AList<TMLink>
Edges(const AList<TMLink> &);

AList<TMLink>
SameLayer(const AList<TMLink> & list, const TMLink & a);
AList<TMLink>
SameLayer(const AList<TMLink> & list, unsigned id);

AList<TMLink>
SameSuperLayer(const AList<TMLink> & list, const TMLink & a);
AList<TMLink> 
SameSuperLayer(const AList<TMLink> & list, unsigned id);

unsigned
SuperLayer(const AList<TMLink> & list);
unsigned
SuperLayer(const AList<TMLink> & list, unsigned minNHits);

TMLink *
InnerMost(const AList<TMLink> & links);
TMLink *
OuterMost(const AList<TMLink> & links);

AList<TMLink>
InOut(const AList<TMLink> &);

template <class T> class CAList;

void
SeparateCores(const AList<TMLink> & input,
              AList<TMLink> & cores,
              AList<TMLink> & nonCores);

AList<TMLink>
Cores(const AList<TMLink> & input);

void
Dump(const CAList<TMLink> & links,
     const std::string & message = std::string(""),
     const std::string & prefix = std::string(""));
void
Dump(const TMLink & link,
     const std::string & message = std::string(""),
     const std::string & prefix = std::string(""));

#if defined(__GNUG__)
int
SortByWireId(const TMLink ** a, const TMLink ** b);
int
SortByX(const TMLink ** a, const TMLink ** b);
#else
extern "C" int
SortByWireId(const void* a, const void* b);
extern "C" int
SortByX(const void* a, const void* b);
#endif

//-----------------------------------------------------------------------------

#ifdef TMLink_NO_INLINE
#define inline
#else
#undef inline
#define TMLink_INLINE_DEFINE_HERE
#endif

#ifdef TMLink_INLINE_DEFINE_HERE

inline
const TMDCWireHit *
TMLink::hit(void) const {
    return _hit;
}

inline
TTrack *
TMLink::track(void) const {
    return _track;
}

inline
const TMDCWireHit *
TMLink::hit(const TMDCWireHit * a) {
    return _hit = a;
}

inline
TTrack *
TMLink::track(TTrack * a) {
    return _track = a;
}

inline
void
TMLink::update(const HepPoint3D & onTrack,
              const HepPoint3D & onWire,
              unsigned leftRight,
              double pull) {
    _onTrack = onTrack;
    _onWire = onWire;
    _leftRight = leftRight;
    _pull = pull;
}

inline
double
TMLink::pull(void) const {
    return _pull;
}

inline
double
TMLink::pull(double a) {
    return _pull = a;
}

inline
const HepPoint3D &
TMLink::positionOnWire(void) const {
    return _onWire;
}

inline
const HepPoint3D &
TMLink::positionOnTrack(void) const {
    return _onTrack;
}

inline
const HepPoint3D &
TMLink::positionOnWire(const HepPoint3D & a) {
    return _onWire = a;
}

inline
const HepPoint3D &
TMLink::positionOnWire(double p[3]) {
    _onWire.setX(p[0]);
    _onWire.setY(p[1]);
    _onWire.setZ(p[2]);
    return _onWire;
}

inline
const HepPoint3D &
TMLink::positionOnTrack(const HepPoint3D & a) {
    return _onTrack = a;
}

inline
const HepPoint3D &
TMLink::positionOnTrack(double p[3]) {
    _onTrack.setX(p[0]);
    _onTrack.setY(p[1]);
    _onTrack.setZ(p[2]);
    return _onTrack;
}

inline
unsigned
TMLink::leftRight(void) const {
    return _leftRight;
}

inline
unsigned
TMLink::leftRight(unsigned a) {
    return _leftRight = a;
}

inline
double
TMLink::dPhi(void) const {
    return _dPhi;
}

inline
double
TMLink::dPhi(double a) {
    return _dPhi = a;
}

inline
const HepPoint3D &
TMLink::position(void) const {
    return _position;
}

inline
const HepPoint3D &
TMLink::position(const HepPoint3D & a) {
    return _position = a;
}

inline
const HepPoint3D &
TMLink::positionD(void) const {
    return _positionD;
}
                                                                                                                             
inline
const HepPoint3D &
TMLink::positionD(const HepPoint3D & a) {
    return _positionD = a;
}

// inline
// const HepPoint3D &
// TMLink::conf(void) const {
//     return _conf;
// }

inline
const HepPoint3D &
TMLink::conf(const HepPoint3D & a) {
    return _conf = a;
}

inline
void 
TMLink::neighbor(unsigned n, TMLink * neighbor){
  if(n <= 6)
    _neighbor[n] = neighbor;
}

inline
TMLink *
TMLink::neighbor(unsigned n) const {
  if(n <= 6)
    return _neighbor[n];
  return NULL;
}

inline
TMLink *
TMLink::link(void) const {
    return _link;
}

inline
TMLink *
TMLink::link(TMLink * a) {
    return _link = a;
}

inline
double
TMLink::distance(void) const {
    return (_onTrack - _onWire).mag();
}

//inline
//const TMDCWire * const
//TMLink::wire(void) const {
//    if (_hit) return _hit->wire();
//    return NULL;
//}

//zangsl 040518 move the following two fun. to TMLink.cxx file
/*
inline
const TMDCWire * const
TMLink::wire(void) const {
    if (_hit) return _hit->wire();
    return NULL;
}

inline
const HepPoint3D &
TMLink::xyPosition(void) const {
    return _hit->wire()->xyPosition();
}
*/

inline
int
TMLink::zStatus(void) const {
    return _zStatus;
}

inline
int
TMLink::zStatus(int a) {
    return _zStatus = a;
}

inline
int
TMLink::zPair(void) const {
    return _zPair;
}

inline
int
TMLink::zPair(int a) {
    return _zPair = a;
}

/*
//-- S. Suzuki added -------
inline
int 
TMLink::usecathode(void) const {
    return _usecathode;
}

inline
void
TMLink::setusecathode(int a) {
    _usecathode = a ;
}

inline 
TMDCClust *
TMLink::getmclust(void) const {
    return _mclust;
}

inline
void
TMLink::setmclust(TMDCClust * a) {
    _mclust = a ;
}

//-- S. Suzuki added end ----
*/

inline
const HepPoint3D &
TMLink::arcZ(const unsigned i) const {
  if(i < 4)return _arcZ[i];
  std::cerr << "Error!! Please stop!!.....arcZ of TMLink!!! in TrkReco." << std::endl;
  return _arcZ[0];
}

inline
const HepPoint3D & 
TMLink::arcZ(const HepPoint3D &az, const unsigned i){
  if(i < 4)return _arcZ[i] = az;
  std::cerr << "Error!! Please stop!!.....arcZ of TMLink!!! in TrkReco." << std::endl;
  return _arcZ[0];
}

//inline
//void
//TMLink::setZphiBeforeCathode(float a) {
//      _ZphiBeforeCathode = a;
//}

inline
const unsigned 
TMLink::fit2D(const unsigned &f) {
  return _fit2D = f;
}

inline
unsigned 
TMLink::fit2D(void) {
  return _fit2D;
}

inline
float
TMLink::drift(unsigned a) const {
    return _drift[a];
}

inline
float
TMLink::drift(float b, unsigned a) {
    return _drift[a] = b;
}

inline
double
TMLink::cDrift(unsigned a) const {
    return _cDrift[a];
}
                                                                                                                                               
inline
double
TMLink::cDrift(double b, unsigned a) {
    return _cDrift[a] = b;
}

inline
float
TMLink::dDrift(unsigned a) const {
    return _dDrift[a];
}

inline
float
TMLink::dDrift(float b, unsigned a) {
    return _dDrift[a] = b;
}

inline
float
TMLink::drift(void) const {
    return (_drift[0] + _drift[1]) / 2.;
}
                                                                                                                                               
inline
double
TMLink::cDrift(void) const {
    return (_cDrift[0] + _cDrift[1]) / 2.;
}

inline
float
TMLink::dDrift(void) const {
    return (_dDrift[0] + _dDrift[1]) / 2.;
}

inline
unsigned
TMLink::tsfTag(void) const {
        return _tsfTag;
}

inline
unsigned
TMLink::tsfTag(unsigned a) {
        return _tsfTag = a;
}

inline
double
TMLink::tof(void) const {
  return _tof;
}

inline
double
TMLink::tof(double a) {
  return _tof = a;
}

inline
double
TMLink::getDriftTime(void) {
  return _drifttime;
}

inline
void
TMLink::setDriftTime(double time){
_drifttime = time;

 }
#endif

#undef inline

#endif /* TMLink_FLAG_ */

---