|
|
|
Definition at line 35 of file TrackClusterRemover.cc.
| TrackClusterRemover::TrackClusterRemover | ( | const edm::ParameterSet & | iConfig | ) |
Definition at line 113 of file TrackClusterRemover.cc.
References clusterWasteSolution_, doPixel_, doStrip_, Exception, edm::ParameterSet::exists(), lumiContext::fill, filterTracks_, edm::ParameterSet::getParameter(), i, NumberOfParamBlocks, overrideTrkQuals_, pblocks_, reco::TrackBase::qualityByName(), readPSet(), trackQuality_, and reco::TrackBase::undefQuality.
:
trajectories_(iConfig.getParameter<InputTag>("trajectories")),
doStrip_(iConfig.existsAs<bool>("doStrip") ? iConfig.getParameter<bool>("doStrip") : true),
doPixel_(iConfig.existsAs<bool>("doPixel") ? iConfig.getParameter<bool>("doPixel") : true),
stripClusters_(doStrip_ ? iConfig.getParameter<InputTag>("stripClusters") : InputTag("NONE")),
pixelClusters_(doPixel_ ? iConfig.getParameter<InputTag>("pixelClusters") : InputTag("NONE")),
mergeOld_(iConfig.exists("oldClusterRemovalInfo")),
oldRemovalInfo_(mergeOld_ ? iConfig.getParameter<InputTag>("oldClusterRemovalInfo") : InputTag("NONE")),
clusterWasteSolution_(true)
{
if (iConfig.exists("overrideTrkQuals"))
overrideTrkQuals_.push_back(iConfig.getParameter<edm::InputTag>("overrideTrkQuals"));
if (iConfig.exists("clusterLessSolution"))
clusterWasteSolution_=!iConfig.getParameter<bool>("clusterLessSolution");
if (doPixel_ && clusterWasteSolution_) produces< edmNew::DetSetVector<SiPixelCluster> >();
if (doStrip_ && clusterWasteSolution_) produces< edmNew::DetSetVector<SiStripCluster> >();
if (clusterWasteSolution_) produces< ClusterRemovalInfo >();
fill(pblocks_, pblocks_+NumberOfParamBlocks, ParamBlock());
readPSet(iConfig, "Common",-1);
if (doPixel_) {
readPSet(iConfig, "Pixel" ,0,1);
readPSet(iConfig, "PXB" ,0);
readPSet(iConfig, "PXE" ,1);
}
if (doStrip_) {
readPSet(iConfig, "Strip" ,2,3,4,5);
readPSet(iConfig, "StripInner" ,2,3);
readPSet(iConfig, "StripOuter" ,4,5);
readPSet(iConfig, "TIB" ,2);
readPSet(iConfig, "TID" ,3);
readPSet(iConfig, "TOB" ,4);
readPSet(iConfig, "TEC" ,5);
}
bool usingCharge = false;
for (size_t i = 0; i < NumberOfParamBlocks; ++i) {
if (!pblocks_[i].isSet_) throw cms::Exception("Configuration Error") << "TrackClusterRemover: Missing configuration for detector with subDetID = " << (i+1);
if (pblocks_[i].usesCharge_ && !usingCharge) {
throw cms::Exception("Configuration Error") << "TrackClusterRemover: Configuration for subDetID = " << (i+1) << " uses cluster charge, which is not enabled.";
}
}
if (!clusterWasteSolution_){
produces<edm::ContainerMask<edmNew::DetSetVector<SiPixelCluster> > >();
produces<edm::ContainerMask<edmNew::DetSetVector<SiStripCluster> > >();
}
trackQuality_=reco::TrackBase::undefQuality;
filterTracks_=false;
if (iConfig.exists("TrackQuality")){
filterTracks_=true;
trackQuality_=reco::TrackBase::qualityByName(iConfig.getParameter<std::string>("TrackQuality"));
}
}
| TrackClusterRemover::~TrackClusterRemover | ( | ) |
Definition at line 170 of file TrackClusterRemover.cc.
{
}
| auto_ptr< edmNew::DetSetVector< T > > TrackClusterRemover::cleanup | ( | const edmNew::DetSetVector< T > & | oldClusters, |
| const std::vector< uint8_t > & | isGood, | ||
| reco::ClusterRemovalInfo::Indices & | refs, | ||
| const reco::ClusterRemovalInfo::Indices * | oldRefs | ||
| ) | [private] |
Definition at line 185 of file TrackClusterRemover.cc.
References edmNew::DetSetVector< T >::begin(), edmNew::DetSetVector< T >::data(), edmNew::DetSetVector< T >::dataSize(), edmNew::DetSetVector< T >::end(), getHLTprescales::index, mergeOld(), convertSQLitetoXML_cfg::output, and edmNew::DetSetVector< T >::size().
Referenced by produce().
{
typedef typename edmNew::DetSetVector<T> DSV;
typedef typename edmNew::DetSetVector<T>::FastFiller DSF;
typedef typename edmNew::DetSet<T> DS;
auto_ptr<DSV> output(new DSV());
output->reserve(oldClusters.size(), oldClusters.dataSize());
unsigned int countOld=0;
unsigned int countNew=0;
// cluster removal loop
const T * firstOffset = & oldClusters.data().front();
for (typename DSV::const_iterator itdet = oldClusters.begin(), enddet = oldClusters.end(); itdet != enddet; ++itdet) {
DS oldDS = *itdet;
if (oldDS.empty()) continue; // skip empty detsets
uint32_t id = oldDS.detId();
DSF outds(*output, id);
for (typename DS::const_iterator it = oldDS.begin(), ed = oldDS.end(); it != ed; ++it) {
uint32_t index = ((&*it) - firstOffset);
countOld++;
if (isGood[index]) {
outds.push_back(*it);
countNew++;
refs.push_back(index);
//std::cout << "TrackClusterRemover::cleanup " << typeid(T).name() << " reference " << index << " to " << (refs.size() - 1) << std::endl;
}
}
if (outds.empty()) outds.abort(); // not write in an empty DSV
}
if (oldRefs != 0) mergeOld(refs, *oldRefs);
return output;
}
| void TrackClusterRemover::mergeOld | ( | reco::ClusterRemovalInfo::Indices & | refs, |
| const reco::ClusterRemovalInfo::Indices & | oldRefs | ||
| ) | [private] |
| void TrackClusterRemover::process | ( | const OmniClusterRef & | cluRef, |
| uint32_t | subdet | ||
| ) | [inline, private] |
Definition at line 225 of file TrackClusterRemover.cc.
References OmniClusterRef::cluster_strip(), clusterWasteSolution_, collectedStrips_, edm::Ref< C, T, F >::id(), edm::Ref< C, T, F >::key(), TrackClusterRemover::ParamBlock::maxSize_, pblocks_, strips, and stripSourceProdID.
{
SiStripRecHit2D::ClusterRef cluster = ocluster.cluster_strip();
if (cluster.id() != stripSourceProdID) throw cms::Exception("Inconsistent Data") <<
"TrackClusterRemover: strip cluster ref from Product ID = " << cluster.id() <<
" does not match with source cluster collection (ID = " << stripSourceProdID << ")\n.";
assert(cluster.id() == stripSourceProdID);
if (pblocks_[subdet-1].usesSize_ && (cluster->amplitudes().size() > pblocks_[subdet-1].maxSize_)) return;
strips[cluster.key()] = false;
//if (!clusterWasteSolution_) collectedStrip[hit->geographicalId()].insert(cluster);
assert(collectedStrips_.size() > cluster.key());
//assert(hit->geographicalId() == cluster->geographicalId()); //This condition fails
if (!clusterWasteSolution_) collectedStrips_[cluster.key()]=true;
}
| void TrackClusterRemover::process | ( | const TrackingRecHit * | hit, |
| float | chi2 | ||
| ) | [inline, private] |
Definition at line 242 of file TrackClusterRemover.cc.
References SiPixelRecHit::cluster(), clusterWasteSolution_, collectedPixels_, cond::rpcobgas::detid, doPixel_, doStrip_, Exception, TrackingRecHit::geographicalId(), edm::Ref< C, T, F >::id(), edm::Ref< C, T, F >::key(), TrackClusterRemover::ParamBlock::maxSize_, SiStripMatchedRecHit2D::monoClusterRef(), NumberOfParamBlocks, TrackerSingleRecHit::omniClusterRef(), ProjectedSiStripRecHit2D::originalHit(), pblocks_, PixelSubdetector::PixelBarrel, PixelSubdetector::PixelEndcap, pixels, pixelSourceProdID, DetId::rawId(), SiStripMatchedRecHit2D::stereoClusterRef(), and DetId::subdetId().
Referenced by produce().
{
DetId detid = hit->geographicalId();
uint32_t subdet = detid.subdetId();
assert ((subdet > 0) && (subdet <= NumberOfParamBlocks));
// chi2 cut
if (chi2 > pblocks_[subdet-1].maxChi2_) return;
if ((subdet == PixelSubdetector::PixelBarrel) || (subdet == PixelSubdetector::PixelEndcap)) {
if (!doPixel_) return;
// this is a pixel, and i *know* it is
const SiPixelRecHit *pixelHit = static_cast<const SiPixelRecHit *>(hit);
SiPixelRecHit::ClusterRef cluster = pixelHit->cluster();
if (cluster.id() != pixelSourceProdID) throw cms::Exception("Inconsistent Data") <<
"TrackClusterRemover: pixel cluster ref from Product ID = " << cluster.id() <<
" does not match with source cluster collection (ID = " << pixelSourceProdID << ")\n.";
assert(cluster.id() == pixelSourceProdID);
//DBG// cout << "HIT NEW PIXEL DETID = " << detid.rawId() << ", Cluster [ " << cluster.key().first << " / " << cluster.key().second << " ] " << endl;
// if requested, cut on cluster size
if (pblocks_[subdet-1].usesSize_ && (cluster->pixels().size() > pblocks_[subdet-1].maxSize_)) return;
// mark as used
pixels[cluster.key()] = false;
//if(!clusterWasteSolution_) collectedPixel[detid.rawId()].insert(cluster);
assert(collectedPixels_.size() > cluster.key());
//assert(detid.rawId() == cluster->geographicalId()); //This condition fails
if(!clusterWasteSolution_) collectedPixels_[cluster.key()]=true;
} else { // aka Strip
if (!doStrip_) return;
const type_info &hitType = typeid(*hit);
if (hitType == typeid(SiStripRecHit2D)) {
const SiStripRecHit2D *stripHit = static_cast<const SiStripRecHit2D *>(hit);
//DBG// cout << "Plain RecHit 2D: " << endl;
process(stripHit->omniClusterRef(),subdet);}
else if (hitType == typeid(SiStripRecHit1D)) {
const SiStripRecHit1D *hit1D = static_cast<const SiStripRecHit1D *>(hit);
process(hit1D->omniClusterRef(),subdet);
} else if (hitType == typeid(SiStripMatchedRecHit2D)) {
const SiStripMatchedRecHit2D *matchHit = static_cast<const SiStripMatchedRecHit2D *>(hit);
//DBG// cout << "Matched RecHit 2D: " << endl;
process(matchHit->monoClusterRef(),subdet);
process(matchHit->stereoClusterRef(),subdet);
} else if (hitType == typeid(ProjectedSiStripRecHit2D)) {
const ProjectedSiStripRecHit2D *projHit = static_cast<const ProjectedSiStripRecHit2D *>(hit);
//DBG// cout << "Projected RecHit 2D: " << endl;
process(projHit->originalHit().omniClusterRef(),subdet);
} else throw cms::Exception("NOT IMPLEMENTED") << "Don't know how to handle " << hitType.name() << " on detid " << detid.rawId() << "\n";
}
}
| void TrackClusterRemover::produce | ( | edm::Event & | iEvent, |
| const edm::EventSetup & | iSetup | ||
| ) | [virtual] |
Implements edm::EDProducer.
Definition at line 311 of file TrackClusterRemover.cc.
References cleanup(), clusterWasteSolution_, collectedPixels_, collectedStrips_, prof2calltree::count, doPixel_, doStrip_, edm::InputTag::encode(), Exception, lumiContext::fill, filterTracks_, edm::Event::getByLabel(), edm::HandleBase::id(), TrackingRecHit::isValid(), LogDebug, Trajectory::measurements(), mergeOld_, oldRemovalInfo_, overrideTrkQuals_, clustersummaryproducer_cfg::pixelClusters, pixelClusters_, pixels, pixelSourceProdID, process(), edm::Event::put(), reco::TrackBase::quality(), edm::ProductID::reset(), clustersummaryproducer_cfg::stripClusters, stripClusters_, strips, stripSourceProdID, trackQuality_, and trajectories_.
{
ProductID pixelOldProdID, stripOldProdID;
Handle<edmNew::DetSetVector<SiPixelCluster> > pixelClusters;
if (doPixel_) {
iEvent.getByLabel(pixelClusters_, pixelClusters);
pixelSourceProdID = pixelClusters.id();
}
//DBG// std::cout << "TrackClusterRemover: Read pixel " << pixelClusters_.encode() << " = ID " << pixelSourceProdID << std::endl;
Handle<edmNew::DetSetVector<SiStripCluster> > stripClusters;
if (doStrip_) {
iEvent.getByLabel(stripClusters_, stripClusters);
stripSourceProdID = stripClusters.id();
}
//DBG// std::cout << "TrackClusterRemover: Read strip " << stripClusters_.encode() << " = ID " << stripSourceProdID << std::endl;
auto_ptr<ClusterRemovalInfo> cri;
if (clusterWasteSolution_){
if (doStrip_ && doPixel_) cri.reset(new ClusterRemovalInfo(pixelClusters, stripClusters));
else if (doStrip_) cri.reset(new ClusterRemovalInfo(stripClusters));
else if (doPixel_) cri.reset(new ClusterRemovalInfo(pixelClusters));
}
Handle<ClusterRemovalInfo> oldRemovalInfo;
if (mergeOld_ && clusterWasteSolution_) {
iEvent.getByLabel(oldRemovalInfo_, oldRemovalInfo);
// Check ProductIDs
if ( (oldRemovalInfo->stripNewRefProd().id() == stripClusters.id()) &&
(oldRemovalInfo->pixelNewRefProd().id() == pixelClusters.id()) ) {
cri->getOldClustersFrom(*oldRemovalInfo);
pixelOldProdID = oldRemovalInfo->pixelRefProd().id();
stripOldProdID = oldRemovalInfo->stripRefProd().id();
} else {
throw cms::Exception("Inconsistent Data") << "TrackClusterRemover: " <<
"Input collection product IDs are [pixel: " << pixelClusters.id() << ", strip: " << stripClusters.id() << "] \n" <<
"\t but the *old* ClusterRemovalInfo " << oldRemovalInfo_.encode() << " refers as 'new product ids' to " <<
"[pixel: " << oldRemovalInfo->pixelNewRefProd().id() << ", strip: " << oldRemovalInfo->stripNewRefProd().id() << "]\n" <<
"NOTA BENE: when running TrackClusterRemover with an old ClusterRemovalInfo the hits in the trajectory MUST be already re-keyed.\n";
}
} else { // then Old == Source
pixelOldProdID = pixelSourceProdID;
stripOldProdID = stripSourceProdID;
}
Handle<TrajTrackAssociationCollection> trajectories_totrack;
iEvent.getByLabel(trajectories_,trajectories_totrack);
std::vector<Handle<edm::ValueMap<int> > > quals;
if ( overrideTrkQuals_.size() > 0) {
quals.resize(1);
iEvent.getByLabel(overrideTrkQuals_[0],quals[0]);
}
if (doStrip_) {
strips.resize(stripClusters->dataSize()); fill(strips.begin(), strips.end(), true);
}
if (doPixel_) {
pixels.resize(pixelClusters->dataSize()); fill(pixels.begin(), pixels.end(), true);
}
typedef edm::ContainerMask<edmNew::DetSetVector<SiPixelCluster> > PixelMaskContainer;
typedef edm::ContainerMask<edmNew::DetSetVector<SiStripCluster> > StripMaskContainer;
if(mergeOld_) {
edm::Handle<PixelMaskContainer> oldPxlMask;
edm::Handle<StripMaskContainer> oldStrMask;
iEvent.getByLabel(oldRemovalInfo_,oldPxlMask);
iEvent.getByLabel(oldRemovalInfo_,oldStrMask);
LogDebug("TrackClusterRemover")<<"to merge in, "<<oldStrMask->size()<<" strp and "<<oldPxlMask->size()<<" pxl";
oldStrMask->copyMaskTo(collectedStrips_);
oldPxlMask->copyMaskTo(collectedPixels_);
}else {
collectedStrips_.resize(stripClusters->dataSize()); fill(collectedStrips_.begin(), collectedStrips_.end(), false);
collectedPixels_.resize(pixelClusters->dataSize()); fill(collectedPixels_.begin(), collectedPixels_.end(), false);
}
TrajTrackAssociationCollection::const_iterator asst=trajectories_totrack->begin();
for ( ; asst!=trajectories_totrack->end();++asst){
const Track & track = *(asst->val);
if (filterTracks_) {
bool goodTk = true;
if ( quals.size()!=0) {
int qual=(*(quals[0]))[asst->val];
if ( qual < 0 ) {goodTk=false;}
//note that this does not work for some trackquals (goodIterative or undefQuality)
else
goodTk = ( qual & (1<<trackQuality_))>>trackQuality_;
}
else
goodTk=(track.quality(trackQuality_));
if ( !goodTk) continue;
}
const Trajectory &tj = *(asst->key);
const vector<TrajectoryMeasurement> &tms = tj.measurements();
vector<TrajectoryMeasurement>::const_iterator itm, endtm;
for (itm = tms.begin(), endtm = tms.end(); itm != endtm; ++itm) {
const TrackingRecHit *hit = itm->recHit()->hit();
if (!hit->isValid()) continue;
process( hit, itm->estimate() );
}
}
if (doPixel_ && clusterWasteSolution_) {
auto_ptr<edmNew::DetSetVector<SiPixelCluster> > newPixelClusters = cleanup(*pixelClusters, pixels,
cri->pixelIndices(), mergeOld_ ? &oldRemovalInfo->pixelIndices() : 0);
OrphanHandle<edmNew::DetSetVector<SiPixelCluster> > newPixels = iEvent.put(newPixelClusters);
//DBG// std::cout << "TrackClusterRemover: Wrote pixel " << newPixels.id() << " from " << pixelSourceProdID << std::endl;
cri->setNewPixelClusters(newPixels);
}
if (doStrip_ && clusterWasteSolution_) {
auto_ptr<edmNew::DetSetVector<SiStripCluster> > newStripClusters = cleanup(*stripClusters, strips,
cri->stripIndices(), mergeOld_ ? &oldRemovalInfo->stripIndices() : 0);
OrphanHandle<edmNew::DetSetVector<SiStripCluster> > newStrips = iEvent.put(newStripClusters);
//DBG// std::cout << "TrackClusterRemover: Wrote strip " << newStrips.id() << " from " << stripSourceProdID << std::endl;
cri->setNewStripClusters(newStrips);
}
if (clusterWasteSolution_) {
// double fraction_pxl= cri->pixelIndices().size() / (double) pixels.size();
// double fraction_strp= cri->stripIndices().size() / (double) strips.size();
// edm::LogWarning("TrackClusterRemover")<<" fraction: " << fraction_pxl <<" "<<fraction_strp;
iEvent.put(cri);
}
pixels.clear(); strips.clear();
if (!clusterWasteSolution_){
//auto_ptr<edmNew::DetSetVector<SiPixelClusterRefNew> > removedPixelClsuterRefs(new edmNew::DetSetVector<SiPixelClusterRefNew>());
//auto_ptr<edmNew::DetSetVector<SiStripRecHit1D::ClusterRef> > removedStripClsuterRefs(new edmNew::DetSetVector<SiStripRecHit1D::ClusterRef>());
std::auto_ptr<StripMaskContainer> removedStripClusterMask(
new StripMaskContainer(edm::RefProd<edmNew::DetSetVector<SiStripCluster> >(stripClusters),collectedStrips_));
LogDebug("TrackClusterRemover")<<"total strip to skip: "<<std::count(collectedStrips_.begin(),collectedStrips_.end(),true);
iEvent.put( removedStripClusterMask );
std::auto_ptr<PixelMaskContainer> removedPixelClusterMask(
new PixelMaskContainer(edm::RefProd<edmNew::DetSetVector<SiPixelCluster> >(pixelClusters),collectedPixels_));
LogDebug("TrackClusterRemover")<<"total pxl to skip: "<<std::count(collectedPixels_.begin(),collectedPixels_.end(),true);
iEvent.put( removedPixelClusterMask );
}
}
| void TrackClusterRemover::readPSet | ( | const edm::ParameterSet & | iConfig, |
| const std::string & | name, | ||
| int | id1 = -1, |
||
| int | id2 = -1, |
||
| int | id3 = -1, |
||
| int | id4 = -1, |
||
| int | id5 = -1, |
||
| int | id6 = -1 |
||
| ) | [private] |
Definition at line 95 of file TrackClusterRemover.cc.
References edm::ParameterSet::exists(), lumiContext::fill, and edm::ParameterSet::getParameter().
Referenced by TrackClusterRemover().
{
if (iConfig.exists(name)) {
ParamBlock pblock(iConfig.getParameter<ParameterSet>(name));
if (id1 == -1) {
fill(pblocks_, pblocks_+NumberOfParamBlocks, pblock);
} else {
pblocks_[id1] = pblock;
if (id2 != -1) pblocks_[id2] = pblock;
if (id3 != -1) pblocks_[id3] = pblock;
if (id4 != -1) pblocks_[id4] = pblock;
if (id5 != -1) pblocks_[id5] = pblock;
if (id6 != -1) pblocks_[id6] = pblock;
}
}
}
bool TrackClusterRemover::clusterWasteSolution_ [private] |
Definition at line 82 of file TrackClusterRemover.cc.
Referenced by process(), produce(), and TrackClusterRemover().
std::vector<bool> TrackClusterRemover::collectedPixels_ [private] |
Definition at line 86 of file TrackClusterRemover.cc.
std::vector<bool> TrackClusterRemover::collectedStrips_ [private] |
Definition at line 85 of file TrackClusterRemover.cc.
bool TrackClusterRemover::doPixel_ [private] |
Definition at line 58 of file TrackClusterRemover.cc.
Referenced by process(), produce(), and TrackClusterRemover().
bool TrackClusterRemover::doStrip_ [private] |
Definition at line 58 of file TrackClusterRemover.cc.
Referenced by process(), produce(), and TrackClusterRemover().
bool TrackClusterRemover::filterTracks_ [private] |
Definition at line 83 of file TrackClusterRemover.cc.
Referenced by produce(), and TrackClusterRemover().
bool TrackClusterRemover::mergeOld_ [private] |
Definition at line 60 of file TrackClusterRemover.cc.
Referenced by produce().
const unsigned int TrackClusterRemover::NumberOfParamBlocks = 6 [static, private] |
Definition at line 54 of file TrackClusterRemover.cc.
Referenced by process(), and TrackClusterRemover().
Definition at line 61 of file TrackClusterRemover.cc.
Referenced by produce().
std::vector<edm::InputTag> TrackClusterRemover::overrideTrkQuals_ [private] |
Definition at line 57 of file TrackClusterRemover.cc.
Referenced by produce(), and TrackClusterRemover().
Definition at line 63 of file TrackClusterRemover.cc.
Referenced by process(), and TrackClusterRemover().
Definition at line 59 of file TrackClusterRemover.cc.
Referenced by produce().
std::vector<uint8_t> TrackClusterRemover::pixels [private] |
Definition at line 67 of file TrackClusterRemover.cc.
Definition at line 68 of file TrackClusterRemover.cc.
Definition at line 59 of file TrackClusterRemover.cc.
Referenced by produce().
std::vector<uint8_t> TrackClusterRemover::strips [private] |
Definition at line 67 of file TrackClusterRemover.cc.
Definition at line 68 of file TrackClusterRemover.cc.
Definition at line 84 of file TrackClusterRemover.cc.
Referenced by produce(), and TrackClusterRemover().
Definition at line 56 of file TrackClusterRemover.cc.
Referenced by produce().
1.7.3