NoKickRTSel.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#include <tracking/trackFindingVXD/sectorMapTools/NoKickRTSel.h>
#include <tracking/trackFindingVXD/sectorMapTools/NoKickCuts.h>
 
#include <tracking/dataobjects/hitXPDerivate.h>
 
#include <mdst/dataobjects/MCParticle.h>
#include <pxd/dataobjects/PXDTrueHit.h>
#include <svd/dataobjects/SVDTrueHit.h>
 
using namespace Belle2;
 
 
void NoKickRTSel::hitXPBuilder(const RecoTrack& track)
{
  StoreArray<SVDCluster> SVDClusters;
  StoreArray<SVDTrueHit> SVDTrueHits;
  StoreArray<MCParticle> MCParticles;
  StoreArray<RecoTrack> recoTracks;
 
  if (track.getRelationsTo<MCParticle>().size() > 0) {
 
    const MCParticle* particle = track.getRelationsTo<MCParticle>()[0];
 
    std::vector<Belle2::RecoHitInformation::UsedSVDHit*> clusterListSVD = track.getSVDHitList();
    for (const SVDCluster* cluster : clusterListSVD) {
      for (const SVDTrueHit& hit : cluster->getRelationsTo<SVDTrueHit>()) {
        RelationVector<MCParticle> hitFromParticle = hit.getRelationsFrom<MCParticle>();
        if (hitFromParticle.size() > 0) {
          if (hitFromParticle[0]->getIndex() != particle->getIndex()) {
            continue;
          }
        } else continue;
        VxdID trueHitSensorID = hit.getSensorID();
        const VXD::SensorInfoBase& sensorInfo = VXD::GeoCache::getInstance().getSensorInfo(trueHitSensorID);
        hitXPDerivate entry(hit, *cluster, *particle, sensorInfo);
        int NClusterU = 0;
        int NClusterV = 0;
        for (const SVDCluster& Ncluster : hit.getRelationsFrom<SVDCluster>()) {
          if (Ncluster.isUCluster()) NClusterU++;
          else NClusterV++;
        }
        entry.setClusterU(NClusterU);
        entry.setClusterV(NClusterV);
 
        bool isReconstructed(false);
        for (const RecoTrack& aRecoTrack : particle->getRelationsFrom<RecoTrack>())
          isReconstructed |= aRecoTrack.hasSVDHits();
        entry.setReconstructed(isReconstructed);
 
        m_setHitXP.insert(entry);
      }
    }
 
 
    StoreArray<PXDCluster> PXDClusters;
    StoreArray<PXDTrueHit> PXDTrueHits;
 
    std::vector<Belle2::RecoHitInformation::UsedPXDHit*> clusterListPXD = track.getPXDHitList();
    for (const PXDCluster* cluster : clusterListPXD) {
      for (const PXDTrueHit& hit : cluster->getRelationsTo<PXDTrueHit>()) {
        RelationVector<MCParticle> hitFromParticle = hit.getRelationsFrom<MCParticle>();
        if (hitFromParticle.size() > 0) {
          if (hitFromParticle[0]->getIndex() != particle->getIndex()) {
            continue;
          }
        } else continue;
        VxdID trueHitSensorID = hit.getSensorID();
        const VXD::SensorInfoBase& sensorInfo = VXD::GeoCache::getInstance().getSensorInfo(trueHitSensorID);
        hitXPDerivate entry(hit, *particle, sensorInfo);
 
        bool isReconstructed(false);
        for (const RecoTrack& aRecoTrack : particle->getRelationsFrom<RecoTrack>())
          isReconstructed |= aRecoTrack.hasPXDHits();
        entry.setReconstructed(isReconstructed);
 
        m_setHitXP.insert(entry);
      }
    }
 
    for (const auto& element : m_setHitXP) {
      m_hitXP.push_back(element);
    }
 
  }
 
}
 
void NoKickRTSel::hit8TrackBuilder(const RecoTrack& track)
{
  hitXPBuilder(track);
  for (const hitXP& XP : m_hitXP) {
    if (m_8hitTrack.size() < 1) {
      m_8hitTrack.push_back(XP);
    } else if (XP.m_sensorLayer != m_8hitTrack.back().m_sensorLayer ||
               XP.m_sensorLadder != m_8hitTrack.back().m_sensorLadder ||
               XP.m_sensorSensor != m_8hitTrack.back().m_sensorSensor) {
      m_8hitTrack.push_back(XP);
    }
  }
 
}
 
bool NoKickRTSel::globalCut(const std::vector<hitXP>& track8)
{
  int flagd0 = 1;
  int flagz0 = 1;
  int lay3 = 0;
  int lay4 = 0;
  int lay5 = 0;
  int lay6 = 0;
  for (const hitXP& XP : track8) {
    if (XP.getSensorLayer() == 3) lay3 = 1;
    if (XP.getSensorLayer() == 4) lay4 = 1;
    if (XP.getSensorLayer() == 5) lay5 = 1;
    if (XP.getSensorLayer() == 6) lay6 = 1;
    //    if (fabs(XP.getD0Entry()) > 1.) flagd0 = 0;
    //    if (fabs(XP.getZ0Entry()) > 1.) flagz0 = 0;
  }
  int N_lay = lay3 + lay4 + lay5 + lay6;
  if (N_lay >= 3) N_lay = 1;
  else N_lay = 0;
  int flagTot = flagd0 * flagz0 * N_lay;
  if (flagTot == 1) return true;
  else return false;
}
 
bool NoKickRTSel::segmentSelector(const hitXP& hit1, const hitXP& hit2, const std::vector<double>& selCut,
                                  NoKickCuts::EParameters par, bool is0)
{
  if (hit2.m_sensorLayer - hit1.m_sensorLayer > 1) return true;
  else {
    double deltaPar = 0;
    switch (par) {
      case NoKickCuts::c_Omega:
        deltaPar = fabs(hit1.getOmegaEntry() - hit2.getOmegaEntry());
        if (is0) deltaPar = fabs(hit1.getOmega0() - hit2.getOmegaEntry());
        //    selCutPXD =0.4;
        break;
 
      case NoKickCuts::c_D0:
        deltaPar = hit1.getD0Entry() - hit2.getD0Entry();
        if (is0) deltaPar = hit1.getD00() - hit2.getD0Entry();
        //  selCutPXD =1;
        break;
 
      case NoKickCuts::c_Phi0:
        deltaPar = asin(sin(hit1.getPhi0Entry())) - asin(sin(hit2.getPhi0Entry()));
        if (is0) deltaPar = asin(sin(hit1.getPhi00())) - asin(sin(hit2.getPhi0Entry()));
        //  selCutPXD =0.3;
        break;
 
      case NoKickCuts::c_Z0:
        deltaPar = hit1.getZ0Entry() - hit2.getZ0Entry();
        if (is0) deltaPar = hit1.getZ00() - hit2.getZ0Entry();
        //  selCutPXD =1;
        break;
 
      case NoKickCuts::c_Tanlambda:
        deltaPar = hit1.getTanLambdaEntry() - hit2.getTanLambdaEntry();
        if (is0) deltaPar = hit1.getTanLambda0() - hit2.getTanLambdaEntry();
        //  selCutPXD =0.3;
        break;
    }
 
    double usedCut = 0;
    if (fabs(selCut.at(0)) > fabs(selCut.at(1))) {
      usedCut = fabs(selCut.at(0));
    } else usedCut = fabs(selCut.at(1));
 
    if (deltaPar > -usedCut && deltaPar < usedCut) return true;
    else {
      B2DEBUG(20, "--------------------------");
      B2DEBUG(20, "lay1=" << hit1.m_sensorLayer);
      B2DEBUG(20, "lay2=" << hit2.m_sensorLayer);
      B2DEBUG(20, "parameter=" << par);
      B2DEBUG(20, "Min=" << selCut.at(0));
      B2DEBUG(20, "Max=" << selCut.at(1));
      B2DEBUG(20, "deltaPar=" << deltaPar);
      B2DEBUG(20, "momentum=" << hit1.m_momentum0.R());
      return false;
    }
  }
}
 
 
 
bool NoKickRTSel::trackSelector(const RecoTrack& track)
{
  initNoKickRTSel();
  hit8TrackBuilder(track);
 
  if (m_outputFlag) {
    m_pdgID = m_8hitTrack[0].getPDGID();
    m_pMag = track.getMomentumSeed().R();
    m_pt = sqrt(track.getMomentumSeed().X() * track.getMomentumSeed().X() + track.getMomentumSeed().Y() * track.getMomentumSeed().Y());
  }
 
  bool good = globalCut(m_8hitTrack);
  if (good == false) {
    if (m_outputFlag) {
      m_numberOfCuts = -1; //it means "no specific cuts applied, but rejected for global cuts"
      m_Ncuts = m_numberOfCuts;
      m_isCutted = true;
      m_nCutHit->Fill(m_numberOfCuts);
      m_momCut->Fill(track.getMomentumSeed().R());
      m_PDGIDCut->Fill(track.getRelationsTo<MCParticle>()[0]->getPDG());
      m_noKickTree->Fill();
    }
    return false;
  }
 
  if (track.getMomentumSeed().R() > m_pmax) {
    if (m_outputFlag) {
      m_Ncuts = m_numberOfCuts;
      m_isCutted = false;
      m_nCutHit->Fill(m_numberOfCuts);
      m_momSel->Fill(track.getMomentumSeed().R());
      m_PDGIDSel->Fill(track.getRelationsTo<MCParticle>()[0]->getPDG());
      m_noKickTree->Fill();
    }
    return good;
  }
  for (int i = 0; i < (int)(m_8hitTrack.size() - 2); i++) {
    double sinTheta = fabs(m_8hitTrack.at(i).m_momentum0.Y()) /
                      sqrt(pow(m_8hitTrack.at(i).m_momentum0.Y(), 2) +
                           pow(m_8hitTrack.at(i).m_momentum0.Z(), 2));
 
    double momentum =
      sqrt(pow(m_8hitTrack.at(i).m_momentum0.X(), 2) +
           pow(m_8hitTrack.at(i).m_momentum0.Y(), 2) +
           pow(m_8hitTrack.at(i).m_momentum0.Z(), 2));
 
    for (int j = NoKickCuts::c_Omega; j <= NoKickCuts::c_Tanlambda; j++) { //track parameters loop
      std::vector<double> selCut = m_trackCuts.cutSelector(sinTheta, momentum, m_8hitTrack.at(i).m_sensorLayer,
                                                           m_8hitTrack.at(i + 1).m_sensorLayer, (NoKickCuts::EParameters) j);
      bool goodSeg = segmentSelector(m_8hitTrack.at(i), m_8hitTrack.at(i + 1), selCut, (NoKickCuts::EParameters) j);
 
      if (!goodSeg) {
        good = false;
        m_numberOfCuts++;
      }
      if (i == 0) { //beampipe crossing
        bool goodSeg0 = segmentSelector(m_8hitTrack.at(i), m_8hitTrack.at(i), selCut, (NoKickCuts::EParameters) j, true);
        if (!goodSeg0) {
          good = false;
          m_numberOfCuts++;
        }
      }
 
    }
  }
  if (m_outputFlag) {
    m_Ncuts = m_numberOfCuts;
    m_nCutHit->Fill(m_numberOfCuts);
    if (good) {
      m_isCutted = false;
      m_momSel->Fill(track.getMomentumSeed().R());
      m_PDGIDSel->Fill(track.getRelationsTo<MCParticle>()[0]->getPDG());
    } else {
      m_isCutted = true;
      m_momCut->Fill(track.getMomentumSeed().R());
      m_PDGIDCut->Fill(track.getRelationsTo<MCParticle>()[0]->getPDG());
    }
    m_noKickTree->Fill();
  }
  return good;
}
 
 
void NoKickRTSel::produceHistoNoKick()
{
  if (m_outputFlag) {
    m_noKickOutputTFile->cd();
    m_momSel->Write();
    m_momCut->Write();
 
    m_momEff->Add(m_momSel, 1);
    m_momEff->Add(m_momCut, 1);
    m_momEff->Divide(m_momSel, m_momEff, 1, 1);
    m_momEff->Write();
 
    m_PDGIDSel->Write();
    m_PDGIDCut->Write();
 
    m_PDGIDEff->Add(m_PDGIDSel, 1);
    m_PDGIDEff->Add(m_PDGIDCut, 1);
    m_PDGIDEff->Divide(m_PDGIDSel, m_PDGIDEff, 1, 1);
    m_PDGIDEff->Write();
 
    m_nCutHit->Write();
 
    m_noKickTree->Write();
 
    delete m_noKickOutputTFile;
  }
}