int PHG4TPCClusterizer::process_event(PHCompositeNode* topNode) { PHNodeIterator iter(topNode); PHCompositeNode* dstNode = static_cast<PHCompositeNode*>(iter.findFirst("PHCompositeNode", "DST")); if (!dstNode) { cout << PHWHERE << "DST Node missing, doing nothing." << endl; return Fun4AllReturnCodes::ABORTRUN; } PHNodeIterator iter_dst(dstNode); SvtxHitMap* hits = findNode::getClass<SvtxHitMap>(dstNode, "SvtxHitMap"); if (!hits) { cout << PHWHERE << "ERROR: Can't find node SvtxHitMap" << endl; return Fun4AllReturnCodes::ABORTRUN; } PHCompositeNode* svxNode = dynamic_cast<PHCompositeNode*>(iter_dst.findFirst("PHCompositeNode", "SVTX")); if (!svxNode) { svxNode = new PHCompositeNode("SVTX"); dstNode->addNode(svxNode); } SvtxClusterMap* svxclusters = findNode::getClass<SvtxClusterMap>(dstNode, "SvtxClusterMap"); if (!svxclusters) { svxclusters = new SvtxClusterMap_v1(); PHIODataNode<PHObject>* SvtxClusterMapNode = new PHIODataNode<PHObject>(svxclusters, "SvtxClusterMap", "PHObject"); svxNode->addNode(SvtxClusterMapNode); } PHG4CylinderCellGeomContainer* geom_container = findNode::getClass<PHG4CylinderCellGeomContainer>(topNode,"CYLINDERCELLGEOM_SVTX"); if (!geom_container) return Fun4AllReturnCodes::ABORTRUN; PHG4CylinderCellContainer* cells = findNode::getClass<PHG4CylinderCellContainer>(dstNode,"G4CELL_SVTX"); if (!cells) return Fun4AllReturnCodes::ABORTRUN; std::vector<std::vector<const SvtxHit*> > layer_sorted; PHG4CylinderCellGeomContainer::ConstRange layerrange = geom_container->get_begin_end(); for (PHG4CylinderCellGeomContainer::ConstIterator layeriter = layerrange.first; layeriter != layerrange.second; ++layeriter) { // We only need TPC layers here, so skip the layers below _min_layer // This if statement is needed because although the maps ladder layers are not included in the cylinder cell geom container, // the cylinder Svx layers are, so they have to be dropped here if they are present if( (unsigned int) layeriter->second->get_layer() < _min_layer) continue; layer_sorted.push_back(std::vector<const SvtxHit*>()); } for (SvtxHitMap::Iter iter = hits->begin(); iter != hits->end(); ++iter) { SvtxHit* hit = iter->second; if( (unsigned int) hit->get_layer() < _min_layer) continue; layer_sorted[hit->get_layer() - _min_layer].push_back(hit); } for (PHG4CylinderCellGeomContainer::ConstIterator layeriter = layerrange.first; layeriter != layerrange.second; ++layeriter) { unsigned int layer = (unsigned int)layeriter->second->get_layer(); // exit on the MAPS layers... // needed in case cylinder svtx layers are present if (layer < _min_layer) continue; if (layer > _max_layer) continue; PHG4CylinderCellGeom* geo = geom_container->GetLayerCellGeom(layer); const int nphibins = layeriter->second->get_phibins(); const int nzbins = layeriter->second->get_zbins(); nhits.clear(); nhits.assign(nzbins, 0); amps.clear(); amps.assign(nphibins * nzbins, 0.); cellids.clear(); cellids.assign(nphibins * nzbins, 0); for (unsigned int i = 0; i < layer_sorted[layer - _min_layer].size(); ++i) { const SvtxHit* hit = layer_sorted[layer - _min_layer][i]; if (hit->get_e() <= 0.) continue; PHG4CylinderCell* cell = cells->findCylinderCell(hit->get_cellid()); int phibin = cell->get_binphi(); int zbin = cell->get_binz(); nhits[zbin] += 1; amps[zbin * nphibins + phibin] += hit->get_e(); cellids[zbin * nphibins + phibin] = hit->get_id(); } int nhits_tot = 0; for (int zbin = 0; zbin < nzbins; ++zbin) { nhits_tot += nhits[zbin]; } while (nhits_tot > 0) { for (int zbin = 0; zbin < nzbins; ++zbin) { if (nhits[zbin] <= 0) continue; for (int phibin = 0; phibin < nphibins; ++phibin) { if (is_local_maximum(amps, nphibins, nzbins, phibin, zbin) == false) { continue; } float phi = 0.; float z = 0.; float e = 0.; fit_cluster(amps, nphibins, nzbins, nhits_tot, nhits, phibin, zbin, geo, phi, z, e); if ((layer > 2) && (e < energy_cut)) { continue; } SvtxCluster_v1 clus; clus.set_layer(layer); clus.set_e(e); double radius = geo->get_radius() + 0.5*geo->get_thickness(); clus.set_position(0, radius * cos(phi)); clus.set_position(1, radius * sin(phi)); clus.set_position(2, z); clus.insert_hit(cellids[zbin * nphibins + phibin]); float invsqrt12 = 1.0/sqrt(12.); TMatrixF DIM(3,3); DIM[0][0] = 0.0;//pow(0.0*0.5*thickness,2); DIM[0][1] = 0.0; DIM[0][2] = 0.0; DIM[1][0] = 0.0; DIM[1][1] = pow(0.5*0.011,2); DIM[1][2] = 0.0; DIM[2][0] = 0.0; DIM[2][1] = 0.0; DIM[2][2] = pow(0.5*0.03,2); TMatrixF ERR(3,3); ERR[0][0] = 0.0;//pow(0.0*0.5*thickness*invsqrt12,2); ERR[0][1] = 0.0; ERR[0][2] = 0.0; ERR[1][0] = 0.0; ERR[1][1] = pow(0.5*0.011*invsqrt12,2); ERR[1][2] = 0.0; ERR[2][0] = 0.0; ERR[2][1] = 0.0; ERR[2][2] = pow(0.5*0.03*invsqrt12,2); TMatrixF ROT(3,3); ROT[0][0] = cos(phi); ROT[0][1] = -sin(phi); ROT[0][2] = 0.0; ROT[1][0] = sin(phi); ROT[1][1] = cos(phi); ROT[1][2] = 0.0; ROT[2][0] = 0.0; ROT[2][1] = 0.0; ROT[2][2] = 1.0; TMatrixF ROT_T(3,3); ROT_T.Transpose(ROT); TMatrixF COVAR_DIM(3,3); COVAR_DIM = ROT * DIM * ROT_T; clus.set_size( 0 , 0 , COVAR_DIM[0][0] ); clus.set_size( 0 , 1 , COVAR_DIM[0][1] ); clus.set_size( 0 , 2 , COVAR_DIM[0][2] ); clus.set_size( 1 , 0 , COVAR_DIM[1][0] ); clus.set_size( 1 , 1 , COVAR_DIM[1][1] ); clus.set_size( 1 , 2 , COVAR_DIM[1][2] ); clus.set_size( 2 , 0 , COVAR_DIM[2][0] ); clus.set_size( 2 , 1 , COVAR_DIM[2][1] ); clus.set_size( 2 , 2 , COVAR_DIM[2][2] ); TMatrixF COVAR_ERR(3,3); COVAR_ERR = ROT * ERR * ROT_T; clus.set_error( 0 , 0 , COVAR_ERR[0][0] ); clus.set_error( 0 , 1 , COVAR_ERR[0][1] ); clus.set_error( 0 , 2 , COVAR_ERR[0][2] ); clus.set_error( 1 , 0 , COVAR_ERR[1][0] ); clus.set_error( 1 , 1 , COVAR_ERR[1][1] ); clus.set_error( 1 , 2 , COVAR_ERR[1][2] ); clus.set_error( 2 , 0 , COVAR_ERR[2][0] ); clus.set_error( 2 , 1 , COVAR_ERR[2][1] ); clus.set_error( 2 , 2 , COVAR_ERR[2][2] ); svxclusters->insert(&clus); } } } } reset(); return Fun4AllReturnCodes::EVENT_OK; }