void CaloEvaluator::fillOutputNtuples(PHCompositeNode *topNode) {
if (verbosity > 2) cout << "CaloEvaluator::fillOutputNtuples() entered" << endl;
CaloRawClusterEval* clustereval = _caloevalstack->get_rawcluster_eval();
CaloRawTowerEval* towereval = _caloevalstack->get_rawtower_eval();
CaloTruthEval* trutheval = _caloevalstack->get_truth_eval();
//----------------------
// fill the Event NTuple
//----------------------
if (_do_gpoint_eval) {
// need things off of the DST...
PHG4TruthInfoContainer* truthinfo = findNode::getClass<PHG4TruthInfoContainer>(topNode,"G4TruthInfo");
if (!truthinfo) {
cerr << PHWHERE << " ERROR: Can't find G4TruthInfo" << endl;
exit(-1);
}
// need things off of the DST...
SvtxVertexMap* vertexmap = findNode::getClass<SvtxVertexMap>(topNode,"SvtxVertexMap");
PHG4VtxPoint *gvertex = truthinfo->GetPrimaryVtx( truthinfo->GetPrimaryVertexIndex() );
float gvx = gvertex->get_x();
float gvy = gvertex->get_y();
float gvz = gvertex->get_z();
float vx = NAN;
float vy = NAN;
float vz = NAN;
if (vertexmap) {
if (!vertexmap->empty()) {
SvtxVertex* vertex = (vertexmap->begin()->second);
vx = vertex->get_x();
vy = vertex->get_y();
vz = vertex->get_z();
}
}
float gpoint_data[7] = {_ievent,
gvx,
gvy,
gvz,
vx,
vy,
vz
};
_ntp_gpoint->Fill(gpoint_data);
}
//------------------------
// fill the Gshower NTuple
//------------------------
if (_ntp_gshower) {
if (verbosity > 1) cout << "CaloEvaluator::filling gshower ntuple..." << endl;
PHG4TruthInfoContainer* truthinfo = findNode::getClass<PHG4TruthInfoContainer>(topNode,"G4TruthInfo");
if (!truthinfo) {
cerr << PHWHERE << " ERROR: Can't find G4TruthInfo" << endl;
exit(-1);
}
PHG4TruthInfoContainer::ConstRange range = truthinfo->GetPrimaryParticleRange();
for (PHG4TruthInfoContainer::ConstIterator iter = range.first;
iter != range.second;
++iter) {
PHG4Particle* primary = iter->second;
if (primary->get_e() < _truth_e_threshold) continue;
if (!_truth_trace_embed_flags.empty()) {
if (_truth_trace_embed_flags.find(trutheval->get_embed(primary)) ==
_truth_trace_embed_flags.end()) continue;
}
float gparticleID = primary->get_track_id();
float gflavor = primary->get_pid();
std::set<PHG4Hit*> g4hits = trutheval->get_shower_from_primary(primary);
float gnhits = g4hits.size();
float gpx = primary->get_px();
float gpy = primary->get_py();
float gpz = primary->get_pz();
float ge = primary->get_e();
float gpt = sqrt(gpx*gpx+gpy*gpy);
float geta = NAN;
if (gpt != 0.0) geta = asinh(gpz/gpt);
float gphi = atan2(gpy,gpx);
PHG4VtxPoint* vtx = trutheval->get_vertex(primary);
float gvx = vtx->get_x();
float gvy = vtx->get_y();
float gvz = vtx->get_z();
float gembed = trutheval->get_embed(primary);
float gedep = trutheval->get_shower_energy_deposit(primary);
float gmrad = trutheval->get_shower_moliere_radius(primary);
RawCluster* cluster = clustereval->best_cluster_from(primary);
float clusterID = NAN;
float ntowers = NAN;
float eta = NAN;
float phi = NAN;
float e = NAN;
float efromtruth = NAN;
if (cluster) {
clusterID = cluster->get_id();
ntowers = cluster->getNTowers();
eta = cluster->get_eta();
phi = cluster->get_phi();
e = cluster->get_energy();
efromtruth = clustereval->get_energy_contribution(cluster, primary);
}
float shower_data[20] = {_ievent,
gparticleID,
gflavor,
gnhits,
geta,
gphi,
ge,
gpt,
gvx,
gvy,
gvz,
gembed,
gedep,
gmrad,
clusterID,
ntowers,
eta,
phi,
e,
efromtruth
};
_ntp_gshower->Fill(shower_data);
}
}
//----------------------
// fill the Tower NTuple
//----------------------
if (_do_tower_eval) {
if (verbosity > 1) cout << "CaloEvaluator::filling tower ntuple..." << endl;
string towernode = "TOWER_CALIB_" + _caloname;
RawTowerContainer* towers = findNode::getClass<RawTowerContainer>(topNode,towernode.c_str());
if (!towers) {
cerr << PHWHERE << " ERROR: Can't find " << towernode << endl;
exit(-1);
}
string towergeomnode = "TOWERGEOM_" + _caloname;
RawTowerGeomContainer* towergeom = findNode::getClass<RawTowerGeomContainer>(topNode,towergeomnode.c_str());
if (!towergeom) {
cerr << PHWHERE << " ERROR: Can't find " << towergeomnode << endl;
exit(-1);
}
RawTowerContainer::ConstRange begin_end = towers->getTowers();
RawTowerContainer::ConstIterator rtiter;
for (rtiter = begin_end.first; rtiter != begin_end.second; ++rtiter) {
RawTower *tower = rtiter->second;
if (tower->get_energy() < _reco_e_threshold) continue;
float towerid = tower->get_id();
float ieta = tower->get_bineta();
float iphi = tower->get_binphi();
float eta = towergeom->get_etacenter(tower->get_bineta());
float phi = towergeom->get_phicenter(tower->get_binphi());
float e = tower->get_energy();
PHG4Particle* primary = towereval->max_truth_primary_by_energy(tower);
float gparticleID = NAN;
float gflavor = NAN;
float gnhits = NAN;
float gpx = NAN;
float gpy = NAN;
float gpz = NAN;
float ge = NAN;
float gpt = NAN;
float geta = NAN;
float gphi = NAN;
float gvx = NAN;
float gvy = NAN;
float gvz = NAN;
float gembed = NAN;
float gedep = NAN;
float gmrad = NAN;
float efromtruth = NAN;
if (primary) {
gparticleID = primary->get_track_id();
gflavor = primary->get_pid();
std::set<PHG4Hit*> g4hits = trutheval->get_shower_from_primary(primary);
gnhits = g4hits.size();
gpx = primary->get_px();
gpy = primary->get_py();
gpz = primary->get_pz();
ge = primary->get_e();
gpt = sqrt(gpx * gpx + gpy * gpy);
if (gpt != 0.0) geta = asinh(gpz / gpt);
gphi = atan2(gpy, gpx);
PHG4VtxPoint* vtx = trutheval->get_vertex(primary);
if (vtx) {
gvx = vtx->get_x();
gvy = vtx->get_y();
gvz = vtx->get_z();
}
gembed = trutheval->get_embed(primary);
gedep = trutheval->get_shower_energy_deposit(primary);
gmrad = trutheval->get_shower_moliere_radius(primary);
efromtruth = towereval->get_energy_contribution(tower, primary);
}
float tower_data[21] = {_ievent,
towerid,
ieta,
iphi,
eta,
phi,
e,
gparticleID,
gflavor,
gnhits,
geta,
gphi,
ge,
gpt,
gvx,
gvy,
gvz,
gembed,
gedep,
gmrad,
efromtruth
};
_ntp_tower->Fill(tower_data);
}
}
//------------------------
// fill the Cluster NTuple
//------------------------
if (_do_cluster_eval) {
if (verbosity > 1) cout << "CaloEvaluator::filling gcluster ntuple..." << endl;
string clusternode = "CLUSTER_" + _caloname;
RawClusterContainer* clusters = findNode::getClass<RawClusterContainer>(topNode,clusternode.c_str());
if (!clusters) {
cerr << PHWHERE << " ERROR: Can't find " << clusternode << endl;
exit(-1);
}
// for every cluster
for (unsigned int icluster = 0; icluster < clusters->size(); icluster++) {
RawCluster *cluster = clusters->getCluster(icluster);
if (cluster->get_energy() < _reco_e_threshold) continue;
float clusterID = cluster->get_id();
float ntowers = cluster->getNTowers();
float eta = cluster->get_eta();
float phi = cluster->get_phi();
float e = cluster->get_energy();
PHG4Particle* primary = clustereval->max_truth_primary_by_energy(cluster);
float gparticleID = NAN;
float gflavor = NAN;
float gnhits = NAN;
float gpx = NAN;
float gpy = NAN;
float gpz = NAN;
float ge = NAN;
float gpt = NAN;
float geta = NAN;
float gphi = NAN;
float gvx = NAN;
float gvy = NAN;
float gvz = NAN;
float gembed = NAN;
float gedep = NAN;
float gmrad = NAN;
float efromtruth = NAN;
if (primary) {
gparticleID = primary->get_track_id();
gflavor = primary->get_pid();
std::set<PHG4Hit*> g4hits = trutheval->get_shower_from_primary(primary);
gnhits = g4hits.size();
gpx = primary->get_px();
gpy = primary->get_py();
gpz = primary->get_pz();
ge = primary->get_e();
gpt = sqrt(gpx * gpx + gpy * gpy);
if (gpt != 0.0) geta = asinh(gpz / gpt);
gphi = atan2(gpy, gpx);
PHG4VtxPoint* vtx = trutheval->get_vertex(primary);
if (vtx) {
gvx = vtx->get_x();
gvy = vtx->get_y();
gvz = vtx->get_z();
}
gembed = trutheval->get_embed(primary);
gedep = trutheval->get_shower_energy_deposit(primary);
gmrad = trutheval->get_shower_moliere_radius(primary);
efromtruth = clustereval->get_energy_contribution(cluster,
primary);
}
float cluster_data[20] = {_ievent,
clusterID,
ntowers,
eta,
phi,
e,
gparticleID,
gflavor,
gnhits,
geta,
gphi,
ge,
gpt,
gvx,
gvy,
gvz,
gembed,
gedep,
gmrad,
efromtruth
};
_ntp_cluster->Fill(cluster_data);
}
}
return;
}
int CaloTriggerSim::process_event(PHCompositeNode *topNode)
{
if (verbosity > 0)
std::cout << "CaloTriggerSim::process_event: entering" << std::endl;
// pull out the tower containers and geometry objects at the start
RawTowerContainer *towersEM3 = findNode::getClass<RawTowerContainer>(topNode, "TOWER_CALIB_CEMC");
RawTowerContainer *towersIH3 = findNode::getClass<RawTowerContainer>(topNode, "TOWER_CALIB_HCALIN");
RawTowerContainer *towersOH3 = findNode::getClass<RawTowerContainer>(topNode, "TOWER_CALIB_HCALOUT");
if (verbosity > 0) {
std::cout << "CaloTriggerSim::process_event: " << towersEM3->size() << " TOWER_CALIB_CEMC towers" << std::endl;
std::cout << "CaloTriggerSim::process_event: " << towersIH3->size() << " TOWER_CALIB_HCALIN towers" << std::endl;
std::cout << "CaloTriggerSim::process_event: " << towersOH3->size() << " TOWER_CALIB_HCALOUT towers" << std::endl;
}
RawTowerGeomContainer_Cylinderv1 *geomEM = findNode::getClass<RawTowerGeomContainer_Cylinderv1>(topNode, "TOWERGEOM_CEMC");
RawTowerGeomContainer *geomIH = findNode::getClass<RawTowerGeomContainer>(topNode, "TOWERGEOM_HCALIN");
RawTowerGeomContainer *geomOH = findNode::getClass<RawTowerGeomContainer>(topNode, "TOWERGEOM_HCALOUT");
// get the binning from the geometry (different for 1D vs 2D...)
int geom_etabins = geomEM->get_etabins();
int geom_phibins = geomEM->get_phibins();
// if internal knowledge of geometry is unset, set it now (should
// only happen once, on the first event)
if (_EMCAL_1x1_NETA < 0)
{
_EMCAL_1x1_NETA = geom_etabins;
_EMCAL_1x1_NPHI = geom_phibins;
// half as many 2x2 windows along each axis as 1x1
_EMCAL_2x2_NETA = geom_etabins / 2;
_EMCAL_2x2_NPHI = geom_phibins / 2;
// each 2x2 window defines a 4x4 window for which that 2x2 window
// is the upper-left corner, so there are as many 4x4's as 2x2's
// (except in eta, where the edge effect means there is 1 fewer)
_EMCAL_4x4_NETA = geom_etabins / 2 - 1;
_EMCAL_4x4_NPHI = geom_phibins / 2;
// reset all maps
_EMCAL_1x1_MAP.resize(_EMCAL_1x1_NETA, std::vector<float>(_EMCAL_1x1_NPHI, 0));
_EMCAL_2x2_MAP.resize(_EMCAL_2x2_NETA, std::vector<float>(_EMCAL_2x2_NPHI, 0));
_EMCAL_4x4_MAP.resize(_EMCAL_4x4_NETA, std::vector<float>(_EMCAL_4x4_NPHI, 0));
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: setting number of window in eta / phi,";
std::cout << "1x1 are " << _EMCAL_1x1_NETA << " / " << _EMCAL_1x1_NPHI << ", ";
std::cout << "2x2 are " << _EMCAL_2x2_NETA << " / " << _EMCAL_2x2_NPHI << ", ";
std::cout << "4x4 are " << _EMCAL_4x4_NETA << " / " << _EMCAL_4x4_NPHI << std::endl;
}
}
// reset 1x1 map
for (int ieta = 0; ieta < _EMCAL_1x1_NETA; ieta++)
{
for (int iphi = 0; iphi < _EMCAL_1x1_NPHI; iphi++)
{
_EMCAL_1x1_MAP[ieta][iphi] = 0;
}
}
// iterate over EMCal towers, constructing 1x1's
RawTowerContainer::ConstRange begin_end = towersEM3->getTowers();
for (RawTowerContainer::ConstIterator rtiter = begin_end.first; rtiter != begin_end.second; ++rtiter)
{
RawTower *tower = rtiter->second;
RawTowerGeom *tower_geom = geomEM->get_tower_geometry(tower->get_key());
float this_eta = tower_geom->get_eta();
float this_phi = tower_geom->get_phi();
int this_etabin = geomEM->get_etabin(this_eta);
int this_phibin = geomEM->get_phibin(this_phi);
float this_E = tower->get_energy();
_EMCAL_1x1_MAP[this_etabin][this_phibin] += this_E;
if (verbosity > 1 && tower->get_energy() > 1)
{
std::cout << "CaloTriggerSim::process_event: EMCal 1x1 tower eta ( bin ) / phi ( bin ) / E = " << std::setprecision(6) << this_eta << " ( " << this_etabin << " ) / " << this_phi << " ( " << this_phibin << " ) / " << this_E << std::endl;
}
}
// reset 2x2 map and best
for (int ieta = 0; ieta < _EMCAL_2x2_NETA; ieta++)
{
for (int iphi = 0; iphi < _EMCAL_2x2_NPHI; iphi++)
{
_EMCAL_2x2_MAP[ieta][iphi] = 0;
}
}
_EMCAL_2x2_BEST_E = 0;
_EMCAL_2x2_BEST_PHI = 0;
_EMCAL_2x2_BEST_ETA = 0;
// now reconstruct 2x2 map from 1x1 map
for (int ieta = 0; ieta < _EMCAL_2x2_NETA; ieta++)
{
for (int iphi = 0; iphi < _EMCAL_2x2_NPHI; iphi++)
{
float this_sum = 0;
this_sum += _EMCAL_1x1_MAP[2 * ieta][2 * iphi];
this_sum += _EMCAL_1x1_MAP[2 * ieta][2 * iphi + 1]; // 2 * iphi + 1 is safe, since _EMCAL_2x2_NPHI = _EMCAL_1x1_NPHI / 2
this_sum += _EMCAL_1x1_MAP[2 * ieta + 1][2 * iphi]; // 2 * ieta + 1 is safe, since _EMCAL_2x2_NETA = _EMCAL_1x1_NETA / 2
this_sum += _EMCAL_1x1_MAP[2 * ieta + 1][2 * iphi + 1];
if (_emulate_truncation) {
this_sum = truncate_8bit( this_sum );
}
// populate 2x2 map
_EMCAL_2x2_MAP[ieta][iphi] = this_sum;
// to calculate the eta, phi position, take the average of that of the 1x1's
float this_eta = 0.5 * (geomEM->get_etacenter(2 * ieta) + geomEM->get_etacenter(2 * ieta + 1));
float this_phi = 0.5 * (geomEM->get_phicenter(2 * iphi) + geomEM->get_phicenter(2 * iphi + 1));
// wrap-around phi (apparently needed for 2D geometry?)
if (this_phi > 3.14159) this_phi -= 2 * 3.14159;
if (this_phi < -3.14159) this_phi += 2 * 3.14159;
if (verbosity > 1 && this_sum > 1)
{
std::cout << "CaloTriggerSim::process_event: EMCal 2x2 tower eta ( bin ) / phi ( bin ) / E = " << std::setprecision(6) << this_eta << " ( " << ieta << " ) / " << this_phi << " ( " << iphi << " ) / " << this_sum << std::endl;
}
if (this_sum > _EMCAL_2x2_BEST_E)
{
_EMCAL_2x2_BEST_E = this_sum;
_EMCAL_2x2_BEST_PHI = this_phi;
_EMCAL_2x2_BEST_ETA = this_eta;
}
}
}
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: best EMCal 2x2 window is at eta / phi = " << _EMCAL_2x2_BEST_ETA << " / " << _EMCAL_2x2_BEST_PHI << " and E = " << _EMCAL_2x2_BEST_E << std::endl;
}
// reset 4x4 map & best
for (int ieta = 0; ieta < _EMCAL_4x4_NETA; ieta++)
{
for (int iphi = 0; iphi < _EMCAL_4x4_NPHI; iphi++)
{
_EMCAL_4x4_MAP[ieta][iphi] = 0;
}
}
_EMCAL_4x4_BEST_E = 0;
_EMCAL_4x4_BEST_PHI = 0;
_EMCAL_4x4_BEST_ETA = 0;
int emcal_4x4_best_iphi = -1;
int emcal_4x4_best_ieta = -1;
// now reconstruct (sliding) 4x4 map from 2x2 map
for (int ieta = 0; ieta < _EMCAL_4x4_NETA; ieta++)
{
for (int iphi = 0; iphi < _EMCAL_4x4_NPHI; iphi++)
{
// for eta calculation (since eta distribution is potentially
// non-uniform), average positions of all four towers
float this_eta = 0.25 * (geomEM->get_etacenter(2 * ieta) + geomEM->get_etacenter(2 * ieta + 1) + geomEM->get_etacenter(2 * ieta + 2) + geomEM->get_etacenter(2 * ieta + 3));
// for phi calculation (since phi distribution is uniform), take
// first tower and add 1.5 tower widths
float this_phi = geomEM->get_phicenter(2 * iphi) + 1.5 * (geomEM->get_phicenter(2 * iphi + 1) - geomEM->get_phicenter(2 * iphi));
// wrap-around phi (apparently needed for 2D geometry?)
if (this_phi > 3.14159) this_phi -= 2 * 3.14159;
if (this_phi < -3.14159) this_phi += 2 * 3.14159;
float this_sum = 0;
this_sum += _EMCAL_2x2_MAP[ieta][iphi];
this_sum += _EMCAL_2x2_MAP[ieta + 1][iphi]; // ieta + 1 is safe, since _EMCAL_4x4_NETA = _EMCAL_2x2_NETA - 1
if (iphi != _EMCAL_4x4_NPHI - 1)
{
// if we are not in the last phi row, can safely access 'iphi+1'
this_sum += _EMCAL_2x2_MAP[ieta][iphi + 1];
this_sum += _EMCAL_2x2_MAP[ieta + 1][iphi + 1];
}
else
{
// if we are in the last phi row, wrap back around to zero
this_sum += _EMCAL_2x2_MAP[ieta][0];
this_sum += _EMCAL_2x2_MAP[ieta + 1][0];
}
_EMCAL_4x4_MAP[ieta][iphi] = this_sum;
if (verbosity > 1 && this_sum > 1)
{
std::cout << "CaloTriggerSim::process_event: EMCal 4x4 tower eta ( bin ) / phi ( bin ) / E = " << std::setprecision(6) << this_eta << " ( " << ieta << " ) / " << this_phi << " ( " << iphi << " ) / " << this_sum << std::endl;
}
if (this_sum > _EMCAL_4x4_BEST_E) {
_EMCAL_4x4_BEST_E = this_sum;
_EMCAL_4x4_BEST_PHI = this_phi;
_EMCAL_4x4_BEST_ETA = this_eta;
emcal_4x4_best_iphi = iphi;
emcal_4x4_best_ieta = ieta;
}
}
}
_EMCAL_4x4_BEST2_E = 0;
_EMCAL_4x4_BEST2_PHI = 0;
_EMCAL_4x4_BEST2_ETA = 0;
// find second-largest 4x4 which is > 1 tower away...
for (int ieta = 0; ieta < _EMCAL_4x4_NETA; ieta++)
{
for (int iphi = 0; iphi < _EMCAL_4x4_NPHI; iphi++)
{
int deta = ieta - emcal_4x4_best_ieta;
int dphi = ( iphi - emcal_4x4_best_iphi ) % _EMCAL_4x4_NPHI ;
if ( abs( deta ) < 1.5 && abs( dphi ) < 1.5 )
continue;
float this_eta = 0.25 * (geomEM->get_etacenter(2 * ieta) + geomEM->get_etacenter(2 * ieta + 1) + geomEM->get_etacenter(2 * ieta + 2) + geomEM->get_etacenter(2 * ieta + 3));
float this_phi = geomEM->get_phicenter(2 * iphi) + 1.5 * (geomEM->get_phicenter(2 * iphi + 1) - geomEM->get_phicenter(2 * iphi));
if (this_phi > 3.14159) this_phi -= 2 * 3.14159;
if (this_phi < -3.14159) this_phi += 2 * 3.14159;
float this_sum = _EMCAL_4x4_MAP[ieta][iphi];
if (this_sum > _EMCAL_4x4_BEST2_E) {
_EMCAL_4x4_BEST2_E = this_sum;
_EMCAL_4x4_BEST2_PHI = this_phi;
_EMCAL_4x4_BEST2_ETA = this_eta;
}
}
}
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: best EMCal 4x4 window is at eta / phi = " << _EMCAL_4x4_BEST_ETA << " / " << _EMCAL_4x4_BEST_PHI << " and E = " << _EMCAL_4x4_BEST_E << std::endl;
std::cout << "CaloTriggerSim::process_event: 2nd best EMCal 4x4 window is at eta / phi = " << _EMCAL_4x4_BEST2_ETA << " / " << _EMCAL_4x4_BEST2_PHI << " and E = " << _EMCAL_4x4_BEST2_E << std::endl;
}
// begin full calo sim
// get the 0.1x0.1 binning from the OHCal geometry
int geomOH_etabins = geomOH->get_etabins();
int geomOH_phibins = geomOH->get_phibins();
// if internal knowledge of geometry is unset, set it now
if (_FULLCALO_0p1x0p1_NETA < 0)
{
_FULLCALO_PHI_START = geomOH->get_phibounds( 0 ).first;
_FULLCALO_PHI_END = geomOH->get_phibounds( geomOH_phibins - 1 ).second;
_FULLCALO_0p1x0p1_NETA = geomOH_etabins;
_FULLCALO_0p1x0p1_NPHI = geomOH_phibins;
// half as many 0.2x0.2 windows along each axis as 0.1x0.1
_FULLCALO_0p2x0p2_NETA = geomOH_etabins / 2;
_FULLCALO_0p2x0p2_NPHI = geomOH_phibins / 2;
// each 0.2x0.2 window defines a 0.4x0.4 window for which that
// 0.2x0.2 window is the upper-left corner, so there are as many
// 0.4x0.4's as 0.2x0.2's (except in eta, where the edge effect
// means there is 1 fewer)
_FULLCALO_0p4x0p4_NETA = geomOH_etabins / 2 - 1;
_FULLCALO_0p4x0p4_NPHI = geomOH_phibins / 2;
// for 0.6x0.6 windows, the above logic applies, except that the
// edge effect causes there to be 2 fewer less in eta
_FULLCALO_0p6x0p6_NETA = geomOH_etabins / 2 - 2;
_FULLCALO_0p6x0p6_NPHI = geomOH_phibins / 2;
// for 0.8x0.8 windows, the above logic applies, except that the
// edge effect causes there to be 3 fewer less in eta
_FULLCALO_0p8x0p8_NETA = geomOH_etabins / 2 - 3;
_FULLCALO_0p8x0p8_NPHI = geomOH_phibins / 2;
// for 1.0x1.0 windows, the above logic applies, except that the
// edge effect causes there to be 4 fewer less in eta
_FULLCALO_1p0x1p0_NETA = geomOH_etabins / 2 - 4;
_FULLCALO_1p0x1p0_NPHI = geomOH_phibins / 2;
// reset all maps
_FULLCALO_0p1x0p1_MAP.resize(_FULLCALO_0p1x0p1_NETA, std::vector<float>(_FULLCALO_0p1x0p1_NPHI, 0));
_FULLCALO_0p2x0p2_MAP.resize(_FULLCALO_0p2x0p2_NETA, std::vector<float>(_FULLCALO_0p2x0p2_NPHI, 0));
_FULLCALO_0p4x0p4_MAP.resize(_FULLCALO_0p4x0p4_NETA, std::vector<float>(_FULLCALO_0p4x0p4_NPHI, 0));
_FULLCALO_0p6x0p6_MAP.resize(_FULLCALO_0p6x0p6_NETA, std::vector<float>(_FULLCALO_0p6x0p6_NPHI, 0));
_FULLCALO_0p8x0p8_MAP.resize(_FULLCALO_0p8x0p8_NETA, std::vector<float>(_FULLCALO_0p8x0p8_NPHI, 0));
_FULLCALO_1p0x1p0_MAP.resize(_FULLCALO_1p0x1p0_NETA, std::vector<float>(_FULLCALO_1p0x1p0_NPHI, 0));
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: determining phi range for 0.1x0.1 full calo map: " << _FULLCALO_PHI_START << " to " << _FULLCALO_PHI_END << std::endl;
std::cout << "CaloTriggerSim::process_event: setting number of full calo window in eta / phi:" << std::endl;
std::cout << " 0.1x0.1 are " << _FULLCALO_0p1x0p1_NETA << " / " << _FULLCALO_0p1x0p1_NPHI << ", ";
std::cout << "0.2x0.2 are " << _FULLCALO_0p2x0p2_NETA << " / " << _FULLCALO_0p2x0p2_NPHI << ", ";
std::cout << "0.4x0.4 are " << _FULLCALO_0p4x0p4_NETA << " / " << _FULLCALO_0p4x0p4_NPHI << ", ";
std::cout << "0.6x0.6 are " << _FULLCALO_0p6x0p6_NETA << " / " << _FULLCALO_0p6x0p6_NPHI << ", ";
std::cout << "0.8x0.8 are " << _FULLCALO_0p8x0p8_NETA << " / " << _FULLCALO_0p8x0p8_NPHI << ", ";
std::cout << "1.0x1.0 are " << _FULLCALO_1p0x1p0_NETA << " / " << _FULLCALO_1p0x1p0_NPHI << std::endl;
}
}
// reset 0.1x0.1 map
for (int ieta = 0; ieta < _FULLCALO_0p1x0p1_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p1x0p1_NPHI; iphi++)
{
_FULLCALO_0p1x0p1_MAP[ieta][iphi] = 0;
}
}
// iterate over EMCal towers, filling in the 0.1x0.1 region they contribute to
RawTowerContainer::ConstRange begin_end_EM = towersEM3->getTowers();
for (RawTowerContainer::ConstIterator rtiter = begin_end_EM.first; rtiter != begin_end_EM.second; ++rtiter)
{
RawTower *tower = rtiter->second;
RawTowerGeom *tower_geom = geomEM->get_tower_geometry(tower->get_key());
float this_eta = tower_geom->get_eta();
float this_phi = tower_geom->get_phi();
if (this_phi < _FULLCALO_PHI_START) this_phi += 2*3.14159;
if (this_phi > _FULLCALO_PHI_END) this_phi -= 2*3.14159;
// note: look up eta/phi index based on OHCal geometry, since this
// defines the 0.1x0.1 regions
int this_etabin = geomOH->get_etabin( this_eta );
int this_phibin = geomOH->get_phibin( this_phi );
float this_E = tower->get_energy();
_FULLCALO_0p1x0p1_MAP[ this_etabin ][ this_phibin ] += this_E;
if (verbosity > 1 && tower->get_energy() > 1)
{
std::cout << "CaloTriggerSim::process_event: EMCal tower at eta / phi (added to fullcalo map with etabin / phibin ) / E = " << std::setprecision(6) << this_eta << " / " << this_phi << " ( " << this_etabin << " / " << this_phibin << " ) / " << this_E << std::endl;
}
}
// iterate over IHCal towers, filling in the 0.1x0.1 region they contribute to
RawTowerContainer::ConstRange begin_end_IH = towersIH3->getTowers();
for (RawTowerContainer::ConstIterator rtiter = begin_end_IH.first; rtiter != begin_end_IH.second; ++rtiter)
{
RawTower *tower = rtiter->second;
RawTowerGeom *tower_geom = geomIH->get_tower_geometry(tower->get_key());
float this_eta = tower_geom->get_eta();
float this_phi = tower_geom->get_phi();
if (this_phi < _FULLCALO_PHI_START) this_phi += 2*3.14159;
if (this_phi > _FULLCALO_PHI_END) this_phi -= 2*3.14159;
// note: look up eta/phi index based on OHCal geometry, even though I
// think it is by construction the same as the IHCal geometry...
int this_etabin = geomOH->get_etabin( this_eta );
int this_phibin = geomOH->get_phibin( this_phi );
float this_E = tower->get_energy();
_FULLCALO_0p1x0p1_MAP[ this_etabin ][ this_phibin ] += this_E;
if (verbosity > 1 && tower->get_energy() > 0.5)
{
std::cout << "CaloTriggerSim::process_event: IHCal tower at eta / phi (added to fullcalo map with etabin / phibin ) / E = " << std::setprecision(6) << this_eta << " / " << this_phi << " ( " << this_etabin << " / " << this_phibin << " ) / " << this_E << std::endl;
}
}
// iterate over OHCal towers, filling in the 0.1x0.1 region they contribute to
RawTowerContainer::ConstRange begin_end_OH = towersOH3->getTowers();
for (RawTowerContainer::ConstIterator rtiter = begin_end_OH.first; rtiter != begin_end_OH.second; ++rtiter)
{
RawTower *tower = rtiter->second;
RawTowerGeom *tower_geom = geomOH->get_tower_geometry(tower->get_key());
float this_eta = tower_geom->get_eta();
float this_phi = tower_geom->get_phi();
if (this_phi < _FULLCALO_PHI_START) this_phi += 2*3.14159;
if (this_phi > _FULLCALO_PHI_END) this_phi -= 2*3.14159;
// note: use the nominal eta/phi index, since the fullcalo 0.1x0.1
// map is defined by the OHCal geometry itself
int this_etabin = geomOH->get_etabin( this_eta );
int this_phibin = geomOH->get_phibin( this_phi );
float this_E = tower->get_energy();
_FULLCALO_0p1x0p1_MAP[ this_etabin ][ this_phibin ] += this_E;
if (verbosity > 1 && tower->get_energy() > 0.5)
{
std::cout << "CaloTriggerSim::process_event: OHCal tower at eta / phi (added to fullcalo map with etabin / phibin ) / E = " << std::setprecision(6) << this_eta << " / " << this_phi << " ( " << this_etabin << " / " << this_phibin << " ) / " << this_E << std::endl;
}
}
// reset 0.2x0.2 map and best
for (int ieta = 0; ieta < _FULLCALO_0p2x0p2_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p2x0p2_NPHI; iphi++)
{
_FULLCALO_0p2x0p2_MAP[ ieta ][ iphi ] = 0;
}
}
_FULLCALO_0p2x0p2_BEST_E = 0;
_FULLCALO_0p2x0p2_BEST_PHI = 0;
_FULLCALO_0p2x0p2_BEST_ETA = 0;
// now reconstruct (non-sliding) 0.2x0.2 map from 0.1x0.1 map
for (int ieta = 0; ieta < _FULLCALO_0p2x0p2_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p2x0p2_NPHI; iphi++)
{
float this_sum = 0;
this_sum += _FULLCALO_0p1x0p1_MAP[2 * ieta][2 * iphi];
this_sum += _FULLCALO_0p1x0p1_MAP[2 * ieta][2 * iphi + 1]; // 2 * iphi + 1 is safe, since _FULLCALO_0p2x0p2_NPHI = _FULLCALO_0p1x0p1_NPHI / 2
this_sum += _FULLCALO_0p1x0p1_MAP[2 * ieta + 1][2 * iphi]; // 2 * ieta + 1 is safe, since _FULLCALO_0p2x0p2_NETA = _FULLCALO_0p1x0p1_NETA / 2
this_sum += _FULLCALO_0p1x0p1_MAP[2 * ieta + 1][2 * iphi + 1];
// populate 0.2x0.2 map
_FULLCALO_0p2x0p2_MAP[ieta][iphi] = this_sum;
// to calculate the eta, phi position, take the average of that
// of the contributing 0.1x0.1's (which are defined by the OHCal geometry)
float this_eta = 0.5 * (geomOH->get_etacenter(2 * ieta) + geomOH->get_etacenter(2 * ieta + 1));
float this_phi = 0.5 * (geomOH->get_phicenter(2 * iphi) + geomOH->get_phicenter(2 * iphi + 1));
if (verbosity > 1 && this_sum > 1)
{
std::cout << "CaloTriggerSim::process_event: FullCalo 0.2x0.2 window eta ( bin ) / phi ( bin ) / E = " << std::setprecision(6) << this_eta << " ( " << ieta << " ) / " << this_phi << " ( " << iphi << " ) / " << this_sum << std::endl;
}
if (this_sum > _FULLCALO_0p2x0p2_BEST_E)
{
_FULLCALO_0p2x0p2_BEST_E = this_sum;
_FULLCALO_0p2x0p2_BEST_PHI = this_phi;
_FULLCALO_0p2x0p2_BEST_ETA = this_eta;
}
}
}
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: best FullCalo 0.2x0.2 window is at eta / phi = " << _FULLCALO_0p2x0p2_BEST_ETA << " / " << _FULLCALO_0p2x0p2_BEST_PHI << " and E = " << _FULLCALO_0p2x0p2_BEST_E << std::endl;
}
// reset fullcalo 0.4x0.4 map & best
for (int ieta = 0; ieta < _FULLCALO_0p4x0p4_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p4x0p4_NPHI; iphi++)
{
_FULLCALO_0p4x0p4_MAP[ ieta ][ iphi ] = 0;
}
}
_FULLCALO_0p4x0p4_BEST_E = 0;
_FULLCALO_0p4x0p4_BEST_PHI = 0;
_FULLCALO_0p4x0p4_BEST_ETA = 0;
// now reconstruct (sliding) 0.4x0.4 map from 0.2x0.2 map
for (int ieta = 0; ieta < _FULLCALO_0p4x0p4_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p4x0p4_NPHI; iphi++)
{
// for eta calculation, use position of corner tower and add 1.5
// tower widths
float this_eta = geomOH->get_etacenter(2 * ieta) + 1.5 * ( geomOH->get_etacenter( 1 ) - geomOH->get_etacenter( 0 ) );
// for phi calculation, use position of corner tower and add 1.5
// tower widths
float this_phi = geomOH->get_phicenter(2 * iphi) + 1.5 * (geomOH->get_phicenter( 1 ) - geomOH->get_phicenter( 0 ) );
float this_sum = 0;
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][iphi];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][iphi]; // 2 * ieta + 1 is safe, since _FULLCALO_0p4x0p4_NETA = _FULLCALO_0p4x0p4_NETA - 1
// add 1 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
_FULLCALO_0p4x0p4_MAP[ieta][iphi] = this_sum;
if (verbosity > 1 && this_sum > 2)
{
std::cout << "CaloTriggerSim::process_event: FullCalo 0.4x0.4 tower eta ( bin ) / phi ( bin ) / E = " << std::setprecision(6) << this_eta << " ( " << ieta << " ) / " << this_phi << " ( " << iphi << " ) / " << this_sum << std::endl;
}
if (this_sum > _FULLCALO_0p4x0p4_BEST_E)
{
_FULLCALO_0p4x0p4_BEST_E = this_sum;
_FULLCALO_0p4x0p4_BEST_PHI = this_phi;
_FULLCALO_0p4x0p4_BEST_ETA = this_eta;
}
}
}
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: best FullCalo 0.4x0.4 window is at eta / phi = " << _FULLCALO_0p4x0p4_BEST_ETA << " / " << _FULLCALO_0p4x0p4_BEST_PHI << " and E = " << _FULLCALO_0p4x0p4_BEST_E << std::endl;
}
// reset fullcalo 0.6x0.6 map & best
for (int ieta = 0; ieta < _FULLCALO_0p6x0p6_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p6x0p6_NPHI; iphi++)
{
_FULLCALO_0p6x0p6_MAP[ ieta ][ iphi ] = 0;
}
}
_FULLCALO_0p6x0p6_BEST_E = 0;
_FULLCALO_0p6x0p6_BEST_PHI = 0;
_FULLCALO_0p6x0p6_BEST_ETA = 0;
// now reconstruct (sliding) 0.6x0.6 map from 0.2x0.2 map
for (int ieta = 0; ieta < _FULLCALO_0p6x0p6_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p6x0p6_NPHI; iphi++)
{
// for eta calculation, use position of corner tower and add 2.5
// tower widths
float this_eta = geomOH->get_etacenter(2 * ieta) + 2.5 * ( geomOH->get_etacenter( 1 ) - geomOH->get_etacenter( 0 ) );
// for phi calculation, use position of corner tower and add 2.5
// tower widths
float this_phi = geomOH->get_phicenter(2 * iphi) + 2.5 * (geomOH->get_phicenter( 1 ) - geomOH->get_phicenter( 0 ) );
float this_sum = 0;
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][iphi];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][iphi]; // ieta + 1 is safe, since _FULLCALO_0p6x0p6_NETA = _FULLCALO_0p2x0p2_NETA - 2
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][iphi]; // ieta + 2 is safe, since _FULLCALO_0p6x0p6_NETA = _FULLCALO_0p2x0p2_NETA - 2
// add 1 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
// add 2 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
_FULLCALO_0p6x0p6_MAP[ieta][iphi] = this_sum;
if (verbosity > 1 && this_sum > 3)
{
std::cout << "CaloTriggerSim::process_event: FullCalo 0.6x0.6 tower eta ( bin ) / phi ( bin ) / E = " << std::setprecision(6) << this_eta << " ( " << ieta << " ) / " << this_phi << " ( " << iphi << " ) / " << this_sum << std::endl;
}
if (this_sum > _FULLCALO_0p6x0p6_BEST_E)
{
_FULLCALO_0p6x0p6_BEST_E = this_sum;
_FULLCALO_0p6x0p6_BEST_PHI = this_phi;
_FULLCALO_0p6x0p6_BEST_ETA = this_eta;
}
}
}
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: best FullCalo 0.6x0.6 window is at eta / phi = " << _FULLCALO_0p6x0p6_BEST_ETA << " / " << _FULLCALO_0p6x0p6_BEST_PHI << " and E = " << _FULLCALO_0p6x0p6_BEST_E << std::endl;
}
// reset fullcalo 0.8x0.8 map & best
for (int ieta = 0; ieta < _FULLCALO_0p8x0p8_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p8x0p8_NPHI; iphi++)
{
_FULLCALO_0p8x0p8_MAP[ ieta ][ iphi ] = 0;
}
}
_FULLCALO_0p8x0p8_BEST_E = 0;
_FULLCALO_0p8x0p8_BEST_PHI = 0;
_FULLCALO_0p8x0p8_BEST_ETA = 0;
// now reconstruct (sliding) 0.8x0.8 map from 0.2x0.2 map
for (int ieta = 0; ieta < _FULLCALO_0p8x0p8_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_0p8x0p8_NPHI; iphi++)
{
// for eta calculation, use position of corner tower and add 3.5
// tower widths
float this_eta = geomOH->get_etacenter(2 * ieta) + 3.5 * ( geomOH->get_etacenter( 1 ) - geomOH->get_etacenter( 0 ) );
// for phi calculation, use position of corner tower and add 3.5
// tower widths
float this_phi = geomOH->get_phicenter(2 * iphi) + 3.5 * (geomOH->get_phicenter( 1 ) - geomOH->get_phicenter( 0 ) );
float this_sum = 0;
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][iphi];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][iphi]; // ieta + 1 is safe, since _FULLCALO_0p8x0p8_NETA = _FULLCALO_0p2x0p2_NETA - 3
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][iphi]; // ieta + 2 is safe, since _FULLCALO_0p8x0p8_NETA = _FULLCALO_0p2x0p2_NETA - 3
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][iphi]; // ieta + 3 is safe, since _FULLCALO_0p8x0p8_NETA = _FULLCALO_0p2x0p2_NETA - 3
// add 1 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
// add 2 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
// add 3 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
_FULLCALO_0p8x0p8_MAP[ieta][iphi] = this_sum;
if (verbosity > 1 && this_sum > 4)
{
std::cout << "CaloTriggerSim::process_event: FullCalo 0.8x0.8 tower eta ( bin ) / phi ( bin ) / E = " << std::setprecision(6) << this_eta << " ( " << ieta << " ) / " << this_phi << " ( " << iphi << " ) / " << this_sum << std::endl;
}
if (this_sum > _FULLCALO_0p8x0p8_BEST_E)
{
_FULLCALO_0p8x0p8_BEST_E = this_sum;
_FULLCALO_0p8x0p8_BEST_PHI = this_phi;
_FULLCALO_0p8x0p8_BEST_ETA = this_eta;
}
}
}
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: best FullCalo 0.8x0.8 window is at eta / phi = " << _FULLCALO_0p8x0p8_BEST_ETA << " / " << _FULLCALO_0p8x0p8_BEST_PHI << " and E = " << _FULLCALO_0p8x0p8_BEST_E << std::endl;
}
// reset fullcalo 1.0x1.0 map & best
for (int ieta = 0; ieta < _FULLCALO_1p0x1p0_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_1p0x1p0_NPHI; iphi++)
{
_FULLCALO_1p0x1p0_MAP[ ieta ][ iphi ] = 0;
}
}
_FULLCALO_1p0x1p0_BEST_E = 0;
_FULLCALO_1p0x1p0_BEST_PHI = 0;
_FULLCALO_1p0x1p0_BEST_ETA = 0;
// now reconstruct (sliding) 1.0x1.0 map from 0.2x0.2 map
for (int ieta = 0; ieta < _FULLCALO_1p0x1p0_NETA; ieta++)
{
for (int iphi = 0; iphi < _FULLCALO_1p0x1p0_NPHI; iphi++)
{
// for eta calculation, use position of corner tower and add 4.5
// tower widths
float this_eta = geomOH->get_etacenter(2 * ieta) + 4.5 * ( geomOH->get_etacenter( 1 ) - geomOH->get_etacenter( 0 ) );
// for phi calculation, use position of corner tower and add 4.5
// tower widths
float this_phi = geomOH->get_phicenter(2 * iphi) + 4.5 * (geomOH->get_phicenter( 1 ) - geomOH->get_phicenter( 0 ) );
float this_sum = 0;
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][iphi];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][iphi]; // ieta + 1 is safe, since _FULLCALO_1p0x1p0_NETA = _FULLCALO_0p2x0p2_NETA - 4
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][iphi]; // ieta + 2 is safe, since _FULLCALO_1p0x1p0_NETA = _FULLCALO_0p2x0p2_NETA - 4
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][iphi]; // ieta + 3 is safe, since _FULLCALO_1p0x1p0_NETA = _FULLCALO_0p2x0p2_NETA - 4
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 4][iphi]; // ieta + 4 is safe, since _FULLCALO_1p0x1p0_NETA = _FULLCALO_0p2x0p2_NETA - 4
// add 1 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 4][ ( iphi + 1 ) % _FULLCALO_0p2x0p2_NPHI ];
// add 2 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 4][ ( iphi + 2 ) % _FULLCALO_0p2x0p2_NPHI ];
// add 3 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 4][ ( iphi + 3 ) % _FULLCALO_0p2x0p2_NPHI ];
// add 4 to phi, but take modulus w.r.t. _FULLCALO_0p2x0p2_NPHI
// in case we have wrapped back around
this_sum += _FULLCALO_0p2x0p2_MAP[ieta][ ( iphi + 4 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 1][ ( iphi + 4 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 2][ ( iphi + 4 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 3][ ( iphi + 4 ) % _FULLCALO_0p2x0p2_NPHI ];
this_sum += _FULLCALO_0p2x0p2_MAP[ieta + 4][ ( iphi + 4 ) % _FULLCALO_0p2x0p2_NPHI ];
_FULLCALO_1p0x1p0_MAP[ieta][iphi] = this_sum;
if (verbosity > 1 && this_sum > 5)
{
std::cout << "CaloTriggerSim::process_event: FullCalo 1.0x1.0 tower eta ( bin ) / phi ( bin ) / E = " << std::setprecision(6) << this_eta << " ( " << ieta << " ) / " << this_phi << " ( " << iphi << " ) / " << this_sum << std::endl;
}
if (this_sum > _FULLCALO_1p0x1p0_BEST_E)
{
_FULLCALO_1p0x1p0_BEST_E = this_sum;
_FULLCALO_1p0x1p0_BEST_PHI = this_phi;
_FULLCALO_1p0x1p0_BEST_ETA = this_eta;
}
}
}
if (verbosity > 0)
{
std::cout << "CaloTriggerSim::process_event: best FullCalo 1.0x1.0 window is at eta / phi = " << _FULLCALO_1p0x1p0_BEST_ETA << " / " << _FULLCALO_1p0x1p0_BEST_PHI << " and E = " << _FULLCALO_1p0x1p0_BEST_E << std::endl;
}
FillNode(topNode);
if (verbosity > 0) std::cout << "CaloTriggerSim::process_event: exiting" << std::endl;
return Fun4AllReturnCodes::EVENT_OK;
}
int
RecoInfoExport::process_event(PHCompositeNode *topNode)
{
++_event;
stringstream fname;
fname << _file_prefix << "_Event" << _event << ".dat";
fstream fdata(fname.str(), ios_base::out);
for (auto & calo_name : _calo_names)
{
string towernodename = "TOWER_CALIB_" + calo_name;
// Grab the towers
RawTowerContainer* towers = findNode::getClass<RawTowerContainer>(topNode,
towernodename.c_str());
if (!towers)
{
std::cout << PHWHERE << ": Could not find node "
<< towernodename.c_str() << std::endl;
return Fun4AllReturnCodes::ABORTRUN;
}
string towergeomnodename = "TOWERGEOM_" + calo_name;
RawTowerGeomContainer *towergeom = findNode::getClass<
RawTowerGeomContainer>(topNode, towergeomnodename.c_str());
if (!towergeom)
{
cout << PHWHERE << ": Could not find node "
<< towergeomnodename.c_str() << endl;
return Fun4AllReturnCodes::ABORTRUN;
}
set<const RawTower *> good_towers;
RawTowerContainer::ConstRange begin_end = towers->getTowers();
RawTowerContainer::ConstIterator rtiter;
for (rtiter = begin_end.first; rtiter != begin_end.second; ++rtiter)
{
const RawTower *tower = rtiter->second;
assert(tower);
if (tower->get_energy() > _tower_threshold)
{
good_towers.insert(tower);
}
}
fdata
<< (boost::format("%1% (1..%2% hits)") % calo_name
% good_towers.size()) << endl;
bool first = true;
for (const auto & tower : good_towers)
{
assert(tower);
float eta = towergeom->get_etacenter(tower->get_bineta());
float phi = towergeom->get_phicenter(tower->get_binphi());
phi = atan2(cos(phi), sin(phi));
if (first)
{
first = false;
}
else
fdata << ",";
fdata
<< (boost::format("[%1%,%2%,%3%]") % eta % phi
% tower->get_energy());
}
fdata << endl;
}
{
fdata << "Track list" << endl;
// need things off of the DST...
PHG4TruthInfoContainer* truthinfo = findNode::getClass<
PHG4TruthInfoContainer>(topNode, "G4TruthInfo");
if (!truthinfo)
{
cerr << PHWHERE << " ERROR: Can't find G4TruthInfo" << endl;
exit(-1);
}
// create SVTX eval stack
SvtxEvalStack svtxevalstack(topNode);
// SvtxVertexEval* vertexeval = svtxevalstack.get_vertex_eval();
// SvtxTrackEval* trackeval = svtxevalstack.get_track_eval();
SvtxTruthEval* trutheval = svtxevalstack.get_truth_eval();
// loop over all truth particles
PHG4TruthInfoContainer::Range range =
truthinfo->GetPrimaryParticleRange();
for (PHG4TruthInfoContainer::ConstIterator iter = range.first;
iter != range.second; ++iter)
{
PHG4Particle* g4particle = iter->second;
const TVector3 mom(g4particle->get_px(), g4particle->get_py(),
g4particle->get_pz());
std::set<PHG4Hit*> g4hits = trutheval->all_truth_hits(g4particle);
map<float, PHG4Hit *> time_sort;
map<float, PHG4Hit *> layer_sort;
for (auto & hit : g4hits)
{
if (hit)
{
time_sort[hit->get_avg_t()] = hit;
}
}
for (auto & hit_pair : time_sort)
{
if (hit_pair.second->get_layer() != UINT_MAX
and layer_sort.find(hit_pair.second->get_layer())
== layer_sort.end())
{
layer_sort[hit_pair.second->get_layer()] = hit_pair.second;
}
}
if (layer_sort.size() > 5 and mom.Pt() > _pT_threshold) // minimal track length cut
{
stringstream spts;
TVector3 last_pos(0, 0, 0);
bool first = true;
for (auto & hit_pair : layer_sort)
{
TVector3 pos(hit_pair.second->get_avg_x(),
hit_pair.second->get_avg_y(),
hit_pair.second->get_avg_z());
// hit step cuts
if ((pos - last_pos).Mag() < _min_track_hit_dist
and hit_pair.first != (layer_sort.rbegin()->first)
and hit_pair.first != (layer_sort.begin()->first))
continue;
last_pos = pos;
if (first)
{
first = false;
}
else
spts << ",";
spts << "[";
spts << pos.x();
spts << ",";
spts << pos.y();
spts << ",";
spts << pos.z();
spts << "]";
}
const int abs_pid = abs(g4particle->get_pid());
int t = 5;
if (abs_pid
== TDatabasePDG::Instance()->GetParticle("pi+")->PdgCode())
{
t = 1;
}
else if (abs_pid
== TDatabasePDG::Instance()->GetParticle("proton")->PdgCode())
{
t = 2;
}
else if (abs_pid
== TDatabasePDG::Instance()->GetParticle("K+")->PdgCode())
{
t = 3;
}
else if (abs_pid
== TDatabasePDG::Instance()->GetParticle("e-")->PdgCode())
{
t = 3;
}
const TParticlePDG * pdg_part =
TDatabasePDG::Instance()->GetParticle(11);
const int c =
(pdg_part != nullptr) ? (copysign(1, pdg_part->Charge())) : 0;
fdata
<< (boost::format(
"{ \"pt\": %1%, \"t\": %2%, \"e\": %3%, \"p\": %4%, \"c\": %5%, \"pts\":[ %6% ]}")
% mom.Pt() % t % mom.PseudoRapidity() % mom.Phi() % c
% spts.str()) << endl;
}
}
}
fdata.close();
return 0;
}
