Ejemplo n.º 1
0
void JetEvaluator::fillOutputNtuples(PHCompositeNode *topNode) {
  
  if (Verbosity() > 2) cout << "JetEvaluator::fillOutputNtuples() entered" << endl;

  JetRecoEval*   recoeval = _jetevalstack->get_reco_eval();
  //JetTruthEval* trutheval = _jetevalstack->get_truth_eval();
 
  //-------------------------
  // fill the reco jet ntuple
  //-------------------------

  if (_do_recojet_eval) {
    if (Verbosity() > 1) cout << "JetEvaluator::filling recojet ntuple..." << endl;

    JetMap* recojets = findNode::getClass<JetMap>(topNode,_recojetname.c_str());
    if (!recojets) {
      cerr << PHWHERE << " ERROR: Can't find " << _recojetname << endl;
      exit(-1);
    }
  
    // for every recojet
    for (JetMap::Iter iter = recojets->begin();
	 iter != recojets->end();
	 ++iter) {
      Jet* recojet = iter->second;
      Jet* truthjet = recoeval->max_truth_jet_by_energy(recojet);

      float id    = recojet->get_id();
      float ncomp = recojet->size_comp();
      float eta   = recojet->get_eta();
      float phi   = recojet->get_phi();
      float e     = recojet->get_e();
      float pt    = recojet->get_pt();

      float gid        = NAN;
      float gncomp     = NAN;
      float geta       = NAN;
      float gphi       = NAN;
      float ge         = NAN;
      float gpt        = NAN;
      float efromtruth = NAN;

      if (truthjet) {
	gid    = truthjet->get_id();
	gncomp = truthjet->size_comp();
	geta   = truthjet->get_eta();
	gphi   = truthjet->get_phi();
	ge     = truthjet->get_e();
	gpt    = truthjet->get_pt();
	efromtruth = recoeval->get_energy_contribution(recojet,truthjet);
      }
      
      float recojet_data[14] = {(float) _ievent,
				id,
				ncomp,
				eta,
				phi,
				e,
				pt,
				gid,
				gncomp,
				geta,
				gphi,
				ge,
				gpt,
				efromtruth
      };

      _ntp_recojet->Fill(recojet_data);
    }
  }

  //-------------------------
  // fill the truth jet ntuple
  //-------------------------

  if (_do_truthjet_eval) {
    if (Verbosity() > 1) cout << "JetEvaluator::filling truthjet ntuple..." << endl;

    JetMap* truthjets = findNode::getClass<JetMap>(topNode,_truthjetname.c_str());
    if (!truthjets) {
      cerr << PHWHERE << " ERROR: Can't find " << _truthjetname << endl;
      exit(-1);
    }
  
    // for every truthjet
    for (JetMap::Iter iter = truthjets->begin();
	 iter != truthjets->end();
	 ++iter) {
      Jet* truthjet = iter->second;
      Jet* recojet = recoeval->best_jet_from(truthjet);

      float gid    = truthjet->get_id();
      float gncomp = truthjet->size_comp();
      float geta   = truthjet->get_eta();
      float gphi   = truthjet->get_phi();
      float ge     = truthjet->get_e();
      float gpt    = truthjet->get_pt();

      float id         = NAN;
      float ncomp      = NAN;
      float eta        = NAN;
      float phi        = NAN;
      float e          = NAN;
      float pt         = NAN;
      float efromtruth = NAN;

      if (recojet) {
	id         = recojet->get_id();
	ncomp      = recojet->size_comp();
	eta        = recojet->get_eta();
	phi        = recojet->get_phi();
	e          = recojet->get_e();
	pt         = recojet->get_pt();
	efromtruth = recoeval->get_energy_contribution(recojet,truthjet);
      }
      
      float truthjet_data[14] = {(float) _ievent,
				 gid,
				 gncomp,
				 geta,
				 gphi,
				 ge,
				 gpt,
				 id,
				 ncomp,
				 eta,
				 phi,
				 e,
				 pt,
				 efromtruth
      };

      _ntp_truthjet->Fill(truthjet_data);
    }
  }

  return;
}
int
LeptoquarksReco::process_event(PHCompositeNode *topNode)
{
  /* Reset branch map */
  ResetBranchMap();

  /* Create map to collect tau candidates.
   * Use energy as 'key' to the map because energy is unique for each jet, while there are sometimes multiple jets (same energy,
   * different jet ID) in the input jet collection. Also, map automatically sorts entries by key, i.e. this gives list of tau candidates
   * sorted by energy. */
  type_map_tcan tauCandidateMap;

  /* Get reco jets collection */
  JetMap* recojets = findNode::getClass<JetMap>(topNode,_jetcolname.c_str());
  if (!recojets)
    {
      cerr << PHWHERE << " ERROR: Can't find " << _jetcolname << endl;
      return Fun4AllReturnCodes::ABORTEVENT;
    }

  /* Get track collection with all tracks in this event */
  SvtxTrackMap* trackmap =
    findNode::getClass<SvtxTrackMap>(topNode,"SvtxTrackMap");
  if (!trackmap)
    {
      cout<< PHWHERE << "SvtxTrackMap node not found on node tree"
           << endl;
      return Fun4AllReturnCodes::ABORTEVENT;
    }

  /* Get vertex collection with all vertices in this event */
  SvtxVertexMap* vertexmap =
    findNode::getClass<SvtxVertexMap>(topNode,"SvtxVertexMap");
  if (!vertexmap)
    {
      cout<< PHWHERE << "SvtxVertexMap node not found on node tree"
          << endl;
      return Fun4AllReturnCodes::ABORTEVENT;
    }
    
  
    /* Get collection of truth particles from event generator */
  PHHepMCGenEventMap *genevtmap = findNode::getClass<PHHepMCGenEventMap>(topNode,"PHHepMCGenEventMap");
  if (!genevtmap) {
    cerr << PHWHERE << " ERROR: Can't find PHHepMCGenEventMap" << endl;
    return Fun4AllReturnCodes::ABORTEVENT;
  }

  /* create map of all tower collections and tower geometry collections used */
  type_map_cdata map_calotower;

  /* Select calorimeter to include */
  vector< RawTowerDefs::CalorimeterId > v_caloids;
  v_caloids.push_back( RawTowerDefs::CEMC );
  v_caloids.push_back( RawTowerDefs::HCALIN );
  v_caloids.push_back( RawTowerDefs::HCALOUT );
  v_caloids.push_back( RawTowerDefs::FEMC );
  v_caloids.push_back( RawTowerDefs::FHCAL );
  v_caloids.push_back( RawTowerDefs::EEMC );

  /* Fill map with calorimeter data */
  for ( unsigned i = 0; i < v_caloids.size(); i++ )
    {
      /* Get collection of towers */
      string towers_name = "TOWER_CALIB_";
      towers_name.append( RawTowerDefs::convert_caloid_to_name( v_caloids.at( i ) ) );

      RawTowerContainer *towers = findNode::getClass<RawTowerContainer>(topNode,towers_name);
      if (!towers) {
        cerr << PHWHERE << " ERROR: Can't find " << towers_name << endl;
        return Fun4AllReturnCodes::ABORTEVENT;
      }

      /* get geometry collection */
      string towergeom_name = "TOWERGEOM_";
      towergeom_name.append( RawTowerDefs::convert_caloid_to_name( v_caloids.at( i ) ) );

      RawTowerGeomContainer *geom = findNode::getClass<RawTowerGeomContainer>(topNode,towergeom_name);
      if (!geom) {
        cerr << PHWHERE << " ERROR: Can't find " << towergeom_name << endl;
        return Fun4AllReturnCodes::ABORTEVENT;
      }

      /* Insert tower and geometry collections in map */
      map_calotower.insert( make_pair( v_caloids.at( i ), make_pair( towers, geom ) ) );
    }


  /* Loop over all jets to collect tau candidate objects */
  for (JetMap::Iter iter = recojets->begin();
       iter != recojets->end();
       ++iter)
    {
      /* skip jets below energy threshold */
      if ( (iter->second)->get_e() < _tau_jet_emin )
        continue;

      /* create new tau candidate */
      PidCandidatev1 *tc = new PidCandidatev1();
      tc->set_candidate_id( (iter->second)->get_id() );
      tc->set_property( PidCandidate::jet_id , (iter->second)->get_id() );

      tc->set_property( PidCandidate::evtgen_pid, (int)0 );

      /* add tau candidate to collection */
      tauCandidateMap.insert( make_pair( (iter->second)->get_e(), tc ) );
    }
  
  /* vertex information */
  SvtxEvalStack *svtxevalstack = new SvtxEvalStack(topNode);
  
  /* Add jet information to tau candidates */
  AddJetInformation( tauCandidateMap, recojets, &map_calotower );

  /* Add jet structure information to tau candidates */
  AddJetStructureInformation( tauCandidateMap, &map_calotower );

  /* Add track information to tau candidates */
  AddTrackInformation( tauCandidateMap, trackmap, vertexmap, svtxevalstack, recojets->get_par());  

  /* Add tag for true Tau particle jet to tau candidates */
  AddTrueTauTag( tauCandidateMap, genevtmap );

  /* Add information about tau candidats to output tree */
  WritePidCandidatesToTree( tauCandidateMap );
  
  /* Add global event information to separate tree */
  AddGlobalEventInformation( tauCandidateMap, &map_calotower );

  /* fill event information tree */
  _t_event->Fill();

  /* count up event number */
  _ievent ++;

  return 0;
}