FWTrajectorySeedProxyBuilder.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     TrajectorySeeds
// Class  :     FWTrajectorySeedProxyBuilder
//
// Implementation:
//     <Notes on implementation>
//
// Original Author:  Chris Jones
//         Created:  Tue Nov 25 14:42:13 EST 2008
//

// system include files
#include "TEveStraightLineSet.h"
#include "TEvePointSet.h"
#include "TEveLine.h"

// user include files
#include "Fireworks/Core/interface/FWSimpleProxyBuilderTemplate.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/FWMagField.h"
#include "Fireworks/Tracks/interface/TrackUtils.h"
#include "Fireworks/Tracks/interface/estimate_field.h"
#include "Fireworks/Core/interface/FWGeometry.h"
#include "Fireworks/Core/interface/fwLog.h"

#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHit.h"
#include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"

class FWTrajectorySeedProxyBuilder : public FWSimpleProxyBuilderTemplate<TrajectorySeed> {
public:
  FWTrajectorySeedProxyBuilder();
  ~FWTrajectorySeedProxyBuilder() override;

  REGISTER_PROXYBUILDER_METHODS();

  FWTrajectorySeedProxyBuilder(const FWTrajectorySeedProxyBuilder&) = delete;  // stop default

  const FWTrajectorySeedProxyBuilder& operator=(const FWTrajectorySeedProxyBuilder&) = delete;  // stop default

private:
  using FWSimpleProxyBuilderTemplate<TrajectorySeed>::build;
  void build(const TrajectorySeed& iData, unsigned int iIndex, TEveElement& oItemHolder, const FWViewContext*) override;
};

FWTrajectorySeedProxyBuilder::FWTrajectorySeedProxyBuilder() {}

FWTrajectorySeedProxyBuilder::~FWTrajectorySeedProxyBuilder() {}

void FWTrajectorySeedProxyBuilder::build(const TrajectorySeed& iData,
                                         unsigned int iIndex,
                                         TEveElement& itemHolder,
                                         const FWViewContext*) {
  // LocalPoint pnt = iData.startingState().parameters().position();
  // std::cout << pnt << std::endl;
  // std::cout << dynamic_cast<const SiPixelRecHit *>(&(*iData.recHits().first)) << std::endl;
  // TEveVector startPos(pnt.x(), pnt.y(), pnt.z());

  TEvePointSet* pointSet = new TEvePointSet;
  TEveLine* line = new TEveLine;
  TEveStraightLineSet* lineSet = new TEveStraightLineSet;

  for (auto const& hit : iData.recHits()) {
    unsigned int id = hit.geographicalId();
    const FWGeometry* geom = item()->getGeom();
    const float* pars = geom->getParameters(id);
    const SiPixelRecHit* rh = dynamic_cast<const SiPixelRecHit*>(&hit);
    // std::cout << id << "id "<<   std::endl;
    if (rh) {
      const SiPixelCluster* itc = rh->cluster().get();
      if (!geom->contains(id)) {
        fwLog(fwlog::kWarning) << "failed get geometry of SiPixelCluster with detid: " << id << std::endl;
        continue;
      }

      float localPoint[3] = {
          fireworks::pixelLocalX((*itc).minPixelRow(), pars), fireworks::pixelLocalY((*itc).minPixelCol(), pars), 0.0};

      float globalPoint[3];
      geom->localToGlobal(id, localPoint, globalPoint);

      pointSet->SetNextPoint(globalPoint[0], globalPoint[1], globalPoint[2]);
      line->SetNextPoint(globalPoint[0], globalPoint[1], globalPoint[2]);

    }

    else {
      const SiStripCluster* cluster = fireworks::extractClusterFromTrackingRecHit(&hit);

      if (cluster) {
        short firststrip = cluster->firstStrip();
        float localTop[3] = {0.0, 0.0, 0.0};
        float localBottom[3] = {0.0, 0.0, 0.0};

        fireworks::localSiStrip(firststrip, localTop, localBottom, pars, id);

        float globalTop[3];
        float globalBottom[3];
        geom->localToGlobal(id, localTop, globalTop, localBottom, globalBottom);

        lineSet->AddLine(globalTop[0], globalTop[1], globalTop[2], globalBottom[0], globalBottom[1], globalBottom[2]);
      }
    }
  }

  setupAddElement(pointSet, &itemHolder);
  setupAddElement(line, &itemHolder);
  setupAddElement(lineSet, &itemHolder);
}

//
// static member functions
//
REGISTER_FWPROXYBUILDER(FWTrajectorySeedProxyBuilder,
                        TrajectorySeed,
                        "TrajectorySeeds",
                        FWViewType::kAll3DBits | FWViewType::kAllRPZBits);