float
TrackProjectionTools::getE33Barrel( string detName, float phi, float eta ){
string towernodename = "TOWER_CALIB_" + detName;
// Grab the towers
RawTowerContainer* towerList = findNode::getClass<RawTowerContainer>(_topNode, towernodename.c_str());
if (!towerList)
{
std::cout << PHWHERE << ": Could not find node " << towernodename.c_str() << std::endl;
return -1;
}
string towergeomnodename = "TOWERGEOM_" + detName;
RawTowerGeomContainer *towergeo = findNode::getClass<RawTowerGeomContainer>(_topNode, towergeomnodename.c_str());
if (! towergeo)
{
cout << PHWHERE << ": Could not find node " << towergeomnodename.c_str() << endl;
return -1;
}
// calculate 3x3 and 5x5 tower energies
int binphi = towergeo->get_phibin(phi);
int bineta = towergeo->get_etabin(eta);
float energy_3x3 = 0.0;
float energy_5x5 = 0.0;
for (int iphi = binphi-2; iphi <= binphi+2; ++iphi) {
for (int ieta = bineta-2; ieta <= bineta+2; ++ieta) {
// wrap around
int wrapphi = iphi;
if (wrapphi < 0) {
wrapphi = towergeo->get_phibins() + wrapphi;
}
if (wrapphi >= towergeo->get_phibins()) {
wrapphi = wrapphi - towergeo->get_phibins();
}
// edges
if (ieta < 0) continue;
if (ieta >= towergeo->get_etabins()) continue;
RawTower* tower = towerList->getTower(ieta,wrapphi);
if (tower) {
energy_5x5 += tower->get_energy();
if (abs(iphi - binphi)<=1 and abs(ieta - bineta)<=1 )
energy_3x3 += tower->get_energy();
}
}
}
return energy_3x3;
}
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 PHG4GenFitTrackProjection::process_event(PHCompositeNode *topNode) {
if (verbosity > 1)
cout << "PHG4GenFitTrackProjection::process_event -- entered" << endl;
//---------------------------------
// Get Objects off of the Node Tree
//---------------------------------
// Pull the reconstructed track information off the node tree...
SvtxTrackMap* _g4tracks = findNode::getClass<SvtxTrackMap>(topNode,
"SvtxTrackMap");
if (!_g4tracks) {
cerr << PHWHERE << " ERROR: Can't find SvtxTrackMap." << endl;
return Fun4AllReturnCodes::ABORTRUN;
}
for (int i = 0; i < _num_cal_layers; ++i) {
if (std::isnan(_cal_radii[i]))
continue;
if (verbosity > 1)
cout << "Projecting tracks into: " << _cal_names[i] << endl;
// pull the tower geometry
string towergeonodename = "TOWERGEOM_" + _cal_names[i];
RawTowerGeomContainer *towergeo = findNode::getClass<
RawTowerGeomContainer>(topNode, towergeonodename.c_str());
if (!towergeo) {
cerr << PHWHERE << " ERROR: Can't find node " << towergeonodename
<< endl;
return Fun4AllReturnCodes::ABORTRUN;
}
// pull the towers
string towernodename = "TOWER_CALIB_" + _cal_names[i];
RawTowerContainer *towerList = findNode::getClass<RawTowerContainer>(
topNode, towernodename.c_str());
if (!towerList) {
cerr << PHWHERE << " ERROR: Can't find node " << towernodename
<< endl;
return Fun4AllReturnCodes::ABORTRUN;
}
// pull the clusters
string clusternodename = "CLUSTER_" + _cal_names[i];
RawClusterContainer *clusterList = findNode::getClass<
RawClusterContainer>(topNode, clusternodename.c_str());
if (!clusterList) {
cerr << PHWHERE << " ERROR: Can't find node " << clusternodename
<< endl;
return Fun4AllReturnCodes::ABORTRUN;
}
// loop over all tracks
for (SvtxTrackMap::Iter iter = _g4tracks->begin();
iter != _g4tracks->end(); ++iter) {
SvtxTrack *track = iter->second;
#ifdef DEBUG
cout
<<__LINE__
<<": track->get_charge(): "<<track->get_charge()
<<endl;
#endif
if(!track) {
if(verbosity >= 2) LogWarning("!track");
continue;
}
if (verbosity > 1)
cout << "projecting track id " << track->get_id() << endl;
if (verbosity > 1) {
cout << " track pt = " << track->get_pt() << endl;
}
std::vector<double> point;
point.assign(3, -9999.);
auto last_state_iter = --track->end_states();
SvtxTrackState * trackstate = last_state_iter->second;
if(!trackstate) {
if(verbosity >= 2) LogWarning("!trackstate");
continue;
}
auto pdg = unique_ptr<TDatabasePDG> (TDatabasePDG::Instance());
int reco_charge = track->get_charge();
int gues_charge = pdg->GetParticle(_pid_guess)->Charge();
if(reco_charge*gues_charge<0) _pid_guess *= -1;
#ifdef DEBUG
cout
<<__LINE__
<<": guess charge: " << gues_charge
<<": reco charge: " << reco_charge
<<": pid: " << _pid_guess
<<": pT: " << sqrt(trackstate->get_px()*trackstate->get_px() + trackstate->get_py()*trackstate->get_py())
<<endl;
#endif
auto rep = unique_ptr<genfit::AbsTrackRep> (new genfit::RKTrackRep(_pid_guess));
unique_ptr<genfit::MeasuredStateOnPlane> msop80 = nullptr;
{
TVector3 pos(trackstate->get_x(), trackstate->get_y(), trackstate->get_z());
//pos.SetXYZ(0.01,0,0);
TVector3 mom(trackstate->get_px(), trackstate->get_py(), trackstate->get_pz());
//mom.SetXYZ(1,0,0);
TMatrixDSym cov(6);
for (int i = 0; i < 6; ++i) {
for (int j = 0; j < 6; ++j) {
cov[i][j] = trackstate->get_error(i, j);
}
}
msop80 = unique_ptr<genfit::MeasuredStateOnPlane> (new genfit::MeasuredStateOnPlane(rep.get()));
msop80->setPosMomCov(pos, mom, cov);
}
#ifdef DEBUG
{
double x = msop80->getPos().X();
double y = msop80->getPos().Y();
double z = msop80->getPos().Z();
// double px = msop80->getMom().X();
// double py = msop80->getMom().Y();
double pz = msop80->getMom().Z();
genfit::FieldManager *field_mgr = genfit::FieldManager::getInstance();
double Bx=0, By=0, Bz=0;
field_mgr->getFieldVal(x,y,z,Bx,By,Bz);
cout
<< __LINE__
<< ": { " << msop80->getPos().Perp() << ", " << msop80->getPos().Phi() << ", " << msop80->getPos().Eta() << "} @ "
//<< "{ " << Bx << ", " << By << ", " << Bz << "}"
<< "{ " << msop80->getMom().Perp() << ", " << msop80->getMom().Phi() << ", " << pz << "} "
<<endl;
//msop80->Print();
}
#endif
try {
rep->extrapolateToCylinder(*msop80, _cal_radii[i], TVector3(0,0,0), TVector3(0,0,1));
//rep->extrapolateToCylinder(*msop80, 5., TVector3(0,0,0), TVector3(0,0,1));
} catch (...) {
if(verbosity >= 2) LogWarning("extrapolateToCylinder failed");
continue;
}
#ifdef DEBUG
{
cout<<__LINE__<<endl;
//msop80->Print();
double x = msop80->getPos().X();
double y = msop80->getPos().Y();
double z = msop80->getPos().Z();
// double px = msop80->getMom().X();
// double py = msop80->getMom().Y();
double pz = msop80->getMom().Z();
genfit::FieldManager *field_mgr = genfit::FieldManager::getInstance();
double Bx=0, By=0, Bz=0;
field_mgr->getFieldVal(x,y,z,Bx,By,Bz);
cout
<< __LINE__
<< ": { " << msop80->getPos().Perp() << ", " << msop80->getPos().Phi() << ", " << msop80->getPos().Eta() << "} @ "
//<< "{ " << Bx << ", " << By << ", " << Bz << "}"
<< "{ " << msop80->getMom().Perp() << ", " << msop80->getMom().Phi() << ", " << pz << "} "
<<endl;
}
#endif
point[0] = msop80->getPos().X();
point[1] = msop80->getPos().Y();
point[2] = msop80->getPos().Z();
#ifdef DEBUG
cout
<<__LINE__
<<": GenFit: {"
<< point[0] <<", "
<< point[1] <<", "
<< point[2] <<" }"
<<endl;
#endif
if (std::isnan(point[0]))
continue;
if (std::isnan(point[1]))
continue;
if (std::isnan(point[2]))
continue;
double x = point[0];
double y = point[1];
double z = point[2];
double phi = atan2(y, x);
double eta = asinh(z / sqrt(x * x + y * y));
if (verbosity > 1) {
cout << " initial track phi = " << track->get_phi();
cout << ", eta = " << track->get_eta() << endl;
cout << " calorimeter phi = " << phi << ", eta = " << eta
<< endl;
}
// projection is outside the detector extent
// TODO towergeo doesn't make this easy to extract, but this should be
// fetched from the node tree instead of hardcoded
if (fabs(eta) >= 1.0)
continue;
// calculate 3x3 tower energy
int binphi = towergeo->get_phibin(phi);
int bineta = towergeo->get_etabin(eta);
double energy_3x3 = 0.0;
double energy_5x5 = 0.0;
for (int iphi = binphi - 2; iphi <= binphi + 2; ++iphi) {
for (int ieta = bineta - 2; ieta <= bineta + 2; ++ieta) {
// wrap around
int wrapphi = iphi;
if (wrapphi < 0) {
wrapphi = towergeo->get_phibins() + wrapphi;
}
if (wrapphi >= towergeo->get_phibins()) {
wrapphi = wrapphi - towergeo->get_phibins();
}
// edges
if (ieta < 0)
continue;
if (ieta >= towergeo->get_etabins())
continue;
RawTower* tower = towerList->getTower(ieta, wrapphi);
if (tower) {
energy_5x5 += tower->get_energy();
if (abs(iphi - binphi) <= 1 and abs(ieta - bineta) <= 1)
energy_3x3 += tower->get_energy();
if (verbosity > 1)
cout << " tower " << ieta << " " << wrapphi
<< " energy = " << tower->get_energy()
<< endl;
}
}
}
track->set_cal_energy_3x3(_cal_types[i], energy_3x3);
track->set_cal_energy_5x5(_cal_types[i], energy_5x5);
// loop over all clusters and find nearest
double min_r = DBL_MAX;
double min_index = -9999;
double min_dphi = NAN;
double min_deta = NAN;
double min_e = NAN;
#ifdef DEBUG
double min_cluster_phi = NAN;
#endif
for (unsigned int k = 0; k < clusterList->size(); ++k) {
RawCluster *cluster = clusterList->getCluster(k);
double dphi = atan2(sin(phi - cluster->get_phi()),
cos(phi - cluster->get_phi()));
double deta = eta - cluster->get_eta();
double r = sqrt(pow(dphi, 2) + pow(deta, 2));
if (r < min_r) {
min_index = k;
min_r = r;
min_dphi = dphi;
min_deta = deta;
min_e = cluster->get_energy();
#ifdef DEBUG
min_cluster_phi = cluster->get_phi();
#endif
}
}
if (min_index != -9999) {
track->set_cal_dphi(_cal_types[i], min_dphi);
track->set_cal_deta(_cal_types[i], min_deta);
track->set_cal_cluster_id(_cal_types[i], min_index);
track->set_cal_cluster_e(_cal_types[i], min_e);
#ifdef DEBUG
cout
<<__LINE__
<<": min_cluster_phi: "<<min_cluster_phi
<<endl;
#endif
if (verbosity > 1) {
cout << " nearest cluster dphi = " << min_dphi << " deta = "
<< min_deta << " e = " << min_e << endl;
}
}
} // end track loop
} // end calorimeter layer loop
if (verbosity > 1)
cout << "PHG4GenFitTrackProjection::process_event -- exited" << endl;
return Fun4AllReturnCodes::EVENT_OK;
}
int
QAG4SimulationCalorimeter::process_event_Tower(PHCompositeNode *topNode)
{
const string detector(_calo_name);
if (verbosity > 2)
cout << "QAG4SimulationCalorimeter::process_event_Tower() entered" << endl;
Fun4AllHistoManager *hm = QAHistManagerDef::getHistoManager();
assert(hm);
TH1D* h_norm = dynamic_cast<TH1D*>(hm->getHisto(
get_histo_prefix() + "_Normalization"));
assert(h_norm);
string towernodename = "TOWER_CALIB_" + detector;
// 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_" + detector;
RawTowerGeomContainer *towergeom = findNode::getClass<RawTowerGeomContainer>(
topNode, towergeomnodename.c_str());
if (!towergeom)
{
cout << PHWHERE << ": Could not find node " << towergeomnodename.c_str()
<< endl;
return Fun4AllReturnCodes::ABORTRUN;
}
static const int max_size = 5;
map<int, string> size_label;
size_label[1] = "1x1";
size_label[2] = "2x2";
size_label[3] = "3x3";
size_label[4] = "4x4";
size_label[5] = "5x5";
map<int, double> max_energy;
map<int, TH1F*> energy_hist_list;
map<int, TH1F*> max_energy_hist_list;
for (int size = 1; size <= max_size; ++size)
{
max_energy[size] = 0;
TH1F* h = dynamic_cast<TH1F*>(hm->getHisto(
get_histo_prefix() + "_Tower_" + size_label[size]));
assert(h);
energy_hist_list[size] = h;
h = dynamic_cast<TH1F*>(hm->getHisto(
get_histo_prefix() + "_Tower_" + size_label[size] + "_max"));
assert(h);
max_energy_hist_list[size] = h;
}
h_norm->Fill("Tower", towergeom->size()); // total tower count
h_norm->Fill("Tower Hit", towers->size());
for (int binphi = 0; binphi < towergeom->get_phibins(); ++binphi)
{
for (int bineta = 0; bineta < towergeom->get_etabins(); ++bineta)
{
for (int size = 1; size <= max_size; ++size)
{
// for 2x2 and 4x4 use slide-2 window as implemented in DAQ
if ((size == 2 or size == 4)
and ((binphi % 2 != 0) and (bineta % 2 != 0)))
continue;
double energy = 0;
// sliding window made from 2x2 sums
for (int iphi = binphi; iphi < binphi + size; ++iphi)
{
for (int ieta = bineta; ieta < bineta + size; ++ieta)
{
if (ieta > towergeom->get_etabins())
continue;
// wrap around
int wrapphi = iphi;
assert(wrapphi >= 0);
if (wrapphi >= towergeom->get_phibins())
{
wrapphi = wrapphi - towergeom->get_phibins();
}
RawTower* tower = towers->getTower(ieta, wrapphi);
if (tower)
{
const double e_intput = tower->get_energy();
energy += e_intput;
}
}
}
energy_hist_list[size]->Fill(energy == 0 ? 9.1e-4 : energy); // trick to fill 0 energy tower to the first bin
if (energy > max_energy[size])
max_energy[size] = energy;
} // for (int size = 1; size <= 4; ++size)
}
}
for (int size = 1; size <= max_size; ++size)
{
max_energy_hist_list[size]->Fill(max_energy[size]);
}
return Fun4AllReturnCodes::EVENT_OK;
}
