GFTrackCand.cxx

Go to the documentation of this file.

/* Copyright 2008-2010, Technische Universitaet Muenchen,
   Authors: Christian Hoeppner & Sebastian Neubert

   This file is part of GENFIT.

   GENFIT is free software: you can redistribute it and/or modify
   it under the terms of the GNU Lesser General Public License as published
   by the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   GENFIT is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public License
   along with GENFIT.  If not, see <http://www.gnu.org/licenses/>.
*/
#include "larreco/Genfit/GFTrackCand.h"
#include "larreco/Genfit/GFException.h"

#include <algorithm>
#include <ostream>
#include <vector>

//ClassImp(GFTrackCand)

genf::GFTrackCand::GFTrackCand() : fCurv(0), fDip(0), fInv(false), fQoverpSeed(0.), fMcTrackId(-1)
{}

genf::GFTrackCand::~GFTrackCand() {}

genf::GFTrackCand::GFTrackCand(double curv,
                               double dip,
                               double inv,
                               std::vector<unsigned int> detIDs,
                               std::vector<unsigned int> hitIDs)
  : fDetId(detIDs)
  , fHitId(hitIDs)
  , fCurv(curv)
  , fDip(dip)
  , fInv(inv)
  , fQoverpSeed(0.)
  , fMcTrackId(-1)
{
  if (fDetId.size() != fHitId.size())
    throw GFException("genf::GFTrackCand::GFTrackCand(): hit/det size mismatch", __LINE__, __FILE__)
      .setFatal();
  fRho.resize(fDetId.size(), 0.);
}
genf::GFTrackCand::GFTrackCand(double curv,
                               double dip,
                               double inv,
                               std::vector<unsigned int> detIDs,
                               std::vector<unsigned int> hitIDs,
                               std::vector<double> rhos)
  : fDetId(detIDs)
  , fHitId(hitIDs)
  , fRho(rhos)
  , fCurv(curv)
  , fDip(dip)
  , fInv(inv)
  , fQoverpSeed(0.)
  , fMcTrackId(-1)
{
  if (fDetId.size() != fHitId.size())
    throw GFException("genf::GFTrackCand::GFTrackCand(): hit/det size mismatch", __LINE__, __FILE__)
      .setFatal();
  if (fDetId.size() != fHitId.size())
    throw GFException("genf::GFTrackCand::GFTrackCand(): rho/det size mismatch", __LINE__, __FILE__)
      .setFatal();
}

void genf::GFTrackCand::addHit(unsigned int detId,
                               unsigned int hitId,
                               double rho,
                               unsigned int planeId)
{
  fDetId.push_back(detId);
  fHitId.push_back(hitId);
  fPlaneId.push_back(planeId);
  fRho.push_back(rho);
}

std::vector<unsigned int> genf::GFTrackCand::GetHitIDs(int detId)
{
  if (detId < 0) { // return hits from all detectors
    return fHitId;
  }
  else {
    std::vector<unsigned int> result;
    unsigned int n = fHitId.size();
    for (unsigned int i = 0; i < n; ++i) {
      if (fDetId[i] == (unsigned int)detId) result.push_back(fHitId[i]);
    }
    return result;
  }
}

void genf::GFTrackCand::reset()
{
  fDetId.clear();
  fHitId.clear();
}

bool genf::GFTrackCand::HitInTrack(unsigned int detId, unsigned int hitId)
{
  for (unsigned int i = 0; i < fDetId.size(); i++) {
    if (detId == fDetId[i])
      if (hitId == fHitId[i]) return true;
  }
  return false;
}

bool genf::operator==(const GFTrackCand& lhs, const GFTrackCand& rhs)
{
  if (lhs.getNHits() != rhs.getNHits()) return false;
  bool result = std::equal(lhs.fDetId.begin(), lhs.fDetId.end(), rhs.fDetId.begin());
  result &= std::equal(lhs.fHitId.begin(), lhs.fHitId.end(), rhs.fHitId.begin());
  return result;
}

void genf::GFTrackCand::Print(std::ostream& out /* = std::cout */) const
{
  out << "======== GFTrackCand::print ========";
  if (fMcTrackId >= 0) out << "\nmcTrackId=" << fMcTrackId;
  out << "\nseed values for pos,direction, and q/p: " << std::endl;
  PrintROOTobject(out, fPosSeed);
  PrintROOTobject(out, fDirSeed);
  out << "q/p=" << fQoverpSeed << std::endl;
  if (fDetId.size() != fHitId.size())
    throw std::runtime_error("genf::GFTrackCand::GFTrackCand(): hit/det size mismatch");
  out << "detId|hitId|rho ";
  for (unsigned int i = 0; i < fDetId.size(); ++i)
    out << fDetId.at(i) << "|" << fHitId.at(i) << "|" << fRho.at(i) << " ";
  out << std::endl;
}

void genf::GFTrackCand::append(const GFTrackCand& rhs)
{
  unsigned int detId, hitId;
  double rho;
  for (unsigned int i = 0; i < rhs.getNHits(); ++i) {
    rhs.getHit(i, detId, hitId, rho);
    addHit(detId, hitId, rho);
  }
}