TreeFiller/src/FillerPixelHits.cc

// $Id: FillerPixelHits.cc,v 1.4 2009/11/19 14:35:23 loizides Exp $

#include "MitProd/TreeFiller/interface/FillerPixelHits.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "DataFormats/SiPixelDetId/interface/PXBDetId.h"
#include "DataFormats/SiPixelDetId/interface/PXFDetId.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "Geometry/CommonDetUnit/interface/GeomDet.h"
#include "Geometry/TrackerTopology/interface/RectangularPixelTopology.h" 
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "MitAna/DataTree/interface/Names.h"
#include "MitAna/DataTree/interface/PixelHitCol.h"
#include "MitProd/ObjectService/interface/ObjectService.h"

using namespace std;
using namespace edm;
using namespace mithep;

//--------------------------------------------------------------------------------------------------
FillerPixelHits::FillerPixelHits(const ParameterSet &cfg, const char *name, bool active) :
  BaseFiller(cfg,name,active),
  edmName_(Conf().getUntrackedParameter<string>("edmName","siPixelRecHits")),
  mitName_(Conf().getUntrackedParameter<string>("mitName",Names::gkPixelHitBrn)),
  phits_(new mithep::PixelHitArr(1000))
{
  // Constructor.
}

//--------------------------------------------------------------------------------------------------
FillerPixelHits::~FillerPixelHits()
{
  // Destructor.

  delete phits_;
}

//--------------------------------------------------------------------------------------------------
void FillerPixelHits::BookDataBlock(TreeWriter &tws, const edm::EventSetup &es)
{
  // Add tracks branch to tree, publish and get tracker geometry.

  tws.AddBranch(mitName_,&phits_);
  OS()->add<PixelHitArr>(phits_,mitName_);
}

//--------------------------------------------------------------------------------------------------
void FillerPixelHits::FillDataBlock(const edm::Event      &event, 
                                    const edm::EventSetup &setup)
{
  // Fill pixel hits from edm collection into our collection.

  phits_->Delete();

  // initialize handle and get product
  Handle<SiPixelRecHitCollection> hRecHits;
  GetProduct(edmName_, hRecHits, event);  

  const SiPixelRecHitCollection *hits = hRecHits.product();
  if (!hits->size())
    return;

  // get tracker geometry
  edm::ESHandle<TrackerGeometry> trackerHandle;
  setup.get<TrackerDigiGeometryRecord>().get(trackerHandle);
  const TrackerGeometry *tgeo = trackerHandle.product();

  for(SiPixelRecHitCollection::DataContainer::const_iterator hit = hits->data().begin(), 
        end = hits->data().end(); hit != end; ++hit) {

    if (!hit->isValid())
      continue;

    DetId id(hit->geographicalId());

    int type = 0;
    if(id.subdetId() == int(PixelSubdetector::PixelBarrel)) {
      PXBDetId pid(id);
      type = pid.layer();
    } else if (id.subdetId() == int(PixelSubdetector::PixelEndcap)) {
      PXFDetId pid(id);
      if (pid.disk()==0)
        type = 11;
      else
        type = 12;
      if (pid.side()!=0)
        type = -type;
    } else {
      continue;
    }

    bool isAnyPixelOnEdge = false;
    const PixelGeomDetUnit *pgdu = 
      static_cast<const PixelGeomDetUnit*>(tgeo->idToDetUnit(id));
    if (1) {
      const RectangularPixelTopology *pixTopo = 
        static_cast<const RectangularPixelTopology*>(&pgdu->specificTopology());
      const vector<SiPixelCluster::Pixel> pixels(hit->cluster()->pixels());
      for(std::vector<SiPixelCluster::Pixel>::const_iterator pixel = pixels.begin(); 
          pixel != pixels.end(); ++pixel) {
        int pixelX = pixel->x;
        int pixelY = pixel->y;
        if(pixTopo->isItEdgePixelInX(pixelX) || pixTopo->isItEdgePixelInY(pixelY)) {
          isAnyPixelOnEdge = true;
          break;
        }
      }
    }

    LocalPoint lpos = LocalPoint(hit->localPosition().x(),
                                 hit->localPosition().y(),
                                 hit->localPosition().z());
    GlobalPoint gpos = pgdu->toGlobal(lpos);
    mithep::PixelHit *newhit = phits_->Allocate();
    new (newhit) mithep::PixelHit(gpos.x(),gpos.y(),gpos.z());
    newhit->SetType(type);
    newhit->SetQuality(hit->rawQualityWord());
    newhit->SetCharge(static_cast<int>(hit->cluster()->charge()));
    newhit->SetSizeX(hit->cluster()->sizeX());
    newhit->SetSizeY(hit->cluster()->sizeY());
    newhit->SetAnyPixelIsOnEdge(isAnyPixelOnEdge);
  }
  phits_->Trim();
}
Revision:	1.5
Committed:	Wed Nov 25 14:45:40 2009 UTC (15 years, 5 months ago) by loizides
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Mit_012c
Changes since 1.4:	+5 -5 lines
Log Message:	Added pixels
#	User	Rev	Content
1	loizides	1.5	// $Id: FillerPixelHits.cc,v 1.4 2009/11/19 14:35:23 loizides Exp $
2	loizides	1.1
3			#include "MitProd/TreeFiller/interface/FillerPixelHits.h"
4			#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
5			#include "DataFormats/GeometryVector/interface/LocalPoint.h"
6			#include "DataFormats/SiPixelDetId/interface/PXBDetId.h"
7			#include "DataFormats/SiPixelDetId/interface/PXFDetId.h"
8			#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
9			#include "FWCore/Framework/interface/ESHandle.h"
10			#include "Geometry/CommonDetUnit/interface/GeomDet.h"
11	loizides	1.3	#include "Geometry/TrackerTopology/interface/RectangularPixelTopology.h"
12			#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
13	loizides	1.1	#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
14			#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
15			#include "MitAna/DataTree/interface/Names.h"
16			#include "MitAna/DataTree/interface/PixelHitCol.h"
17			#include "MitProd/ObjectService/interface/ObjectService.h"
18
19			using namespace std;
20			using namespace edm;
21			using namespace mithep;
22
23			//--------------------------------------------------------------------------------------------------
24			FillerPixelHits::FillerPixelHits(const ParameterSet &cfg, const char *name, bool active) :
25			BaseFiller(cfg,name,active),
26			edmName_(Conf().getUntrackedParameter<string>("edmName","siPixelRecHits")),
27			mitName_(Conf().getUntrackedParameter<string>("mitName",Names::gkPixelHitBrn)),
28	loizides	1.5	phits_(new mithep::PixelHitArr(1000))
29	loizides	1.1	{
30			// Constructor.
31			}
32
33			//--------------------------------------------------------------------------------------------------
34			FillerPixelHits::~FillerPixelHits()
35			{
36			// Destructor.
37
38			delete phits_;
39			}
40
41			//--------------------------------------------------------------------------------------------------
42			void FillerPixelHits::BookDataBlock(TreeWriter &tws, const edm::EventSetup &es)
43			{
44			// Add tracks branch to tree, publish and get tracker geometry.
45
46			tws.AddBranch(mitName_,&phits_);
47			OS()->add<PixelHitArr>(phits_,mitName_);
48			}
49
50			//--------------------------------------------------------------------------------------------------
51			void FillerPixelHits::FillDataBlock(const edm::Event &event,
52			const edm::EventSetup &setup)
53			{
54			// Fill pixel hits from edm collection into our collection.
55
56			phits_->Delete();
57
58			// initialize handle and get product
59			Handle<SiPixelRecHitCollection> hRecHits;
60			GetProduct(edmName_, hRecHits, event);
61
62			const SiPixelRecHitCollection *hits = hRecHits.product();
63	loizides	1.4	if (!hits->size())
64			return;
65
66			// get tracker geometry
67			edm::ESHandle<TrackerGeometry> trackerHandle;
68			setup.get<TrackerDigiGeometryRecord>().get(trackerHandle);
69	loizides	1.5	const TrackerGeometry *tgeo = trackerHandle.product();
70	loizides	1.4
71	loizides	1.1	for(SiPixelRecHitCollection::DataContainer::const_iterator hit = hits->data().begin(),
72			end = hits->data().end(); hit != end; ++hit) {
73
74	loizides	1.2	if (!hit->isValid())
75			continue;
76
77	loizides	1.1	DetId id(hit->geographicalId());
78
79	loizides	1.4	int type = 0;
80	loizides	1.1	if(id.subdetId() == int(PixelSubdetector::PixelBarrel)) {
81			PXBDetId pid(id);
82			type = pid.layer();
83			} else if (id.subdetId() == int(PixelSubdetector::PixelEndcap)) {
84			PXFDetId pid(id);
85			if (pid.disk()==0)
86			type = 11;
87			else
88			type = 12;
89			if (pid.side()!=0)
90			type = -type;
91	loizides	1.3	} else {
92			continue;
93			}
94
95			bool isAnyPixelOnEdge = false;
96	loizides	1.5	const PixelGeomDetUnit *pgdu =
97			static_cast<const PixelGeomDetUnit*>(tgeo->idToDetUnit(id));
98	loizides	1.3	if (1) {
99			const RectangularPixelTopology *pixTopo =
100	loizides	1.4	static_cast<const RectangularPixelTopology*>(&pgdu->specificTopology());
101			const vector<SiPixelCluster::Pixel> pixels(hit->cluster()->pixels());
102	loizides	1.3	for(std::vector<SiPixelCluster::Pixel>::const_iterator pixel = pixels.begin();
103			pixel != pixels.end(); ++pixel) {
104			int pixelX = pixel->x;
105			int pixelY = pixel->y;
106			if(pixTopo->isItEdgePixelInX(pixelX) \|\| pixTopo->isItEdgePixelInY(pixelY)) {
107			isAnyPixelOnEdge = true;
108			break;
109			}
110			}
111	loizides	1.1	}
112
113	loizides	1.4	LocalPoint lpos = LocalPoint(hit->localPosition().x(),
114			hit->localPosition().y(),
115			hit->localPosition().z());
116			GlobalPoint gpos = pgdu->toGlobal(lpos);
117	loizides	1.1	mithep::PixelHit *newhit = phits_->Allocate();
118			new (newhit) mithep::PixelHit(gpos.x(),gpos.y(),gpos.z());
119			newhit->SetType(type);
120			newhit->SetQuality(hit->rawQualityWord());
121			newhit->SetCharge(static_cast<int>(hit->cluster()->charge()));
122	loizides	1.3	newhit->SetSizeX(hit->cluster()->sizeX());
123			newhit->SetSizeY(hit->cluster()->sizeY());
124			newhit->SetAnyPixelIsOnEdge(isAnyPixelOnEdge);
125	loizides	1.1	}
126			phits_->Trim();
127			}