StFmsEventClusterer.h

Go to the documentation of this file.

// $Id: StFmsEventClusterer.h,v 1.8 2019/06/26 16:49:53 akio Exp $
//
// $Log: StFmsEventClusterer.h,v $
// Revision 1.8  2019/06/26 16:49:53  akio
// shower shape scaling for all shapes
//
// Revision 1.7  2018/03/02 20:27:29  akio
// Big update from      Zhanwen Zhu with new shower shape and six z slices
//
// Revision 1.6  2015/11/05 17:54:57  akio
// Adding option to scale up shower shape function for large cells
//
// Revision 1.5  2015/11/02 22:44:49  akio
// Fix photonEnergyInTower()
//
// Revision 1.4  2015/10/30 21:33:56  akio
// fix parameter initialization
// adding new cluster categorization method
//
// Revision 1.3  2015/10/21 15:58:05  akio
// Code speed up (~x2) by optimizing minimization fuctions and showershape function
// Add option to merge small cells to large, so that it finds cluster at border
// Add option to perform 1photon fit when 2photon fit faield
// Add option to turn on/off global refit
// Moment analysis done without ECUTOFF when no tower in cluster exceed ECUTOFF=0.5GeV
//
// Revision 1.2  2015/09/02 15:01:32  akio
// Removing StFmsGeometry class, and now it uses StFmsDbMaker to get appropriate parameters.
//
// Revision 1.1  2015/03/10 14:38:54  jeromel
// First version of FmsUtil from Yuxi Pan - reviewd 2015/02
//
#ifndef STROOT_STFMSPOINTMAKER_STFMSEVENTCLUSTERER_H_
#define STROOT_STFMSPOINTMAKER_STFMSEVENTCLUSTERER_H_

#include <memory>  // For std::unique_ptr
#include <vector>

#include "TObject.h"

#include "StFmsClusterFinder.h"

class StFmsDbMaker;

namespace FMSCluster {  // $NMSPC
class StFmsClusterFitter;
class StFmsFittedPhoton;
  //class StFmsGeometry;

class StFmsEventClusterer: public TObject {
 public:
  //  StFmsEventClusterer(const StFmsGeometry* geometry, Int_t detectorId);
  StFmsEventClusterer(StFmsDbMaker* db, Int_t detectorId, 
                      Int_t globalrefit, Int_t mergeSmallToLarge, 
                      Int_t try1Photon, Int_t categorizationAlgo,
                      Float_t scaleShowerShapeLarge , Float_t scaleShowerShapeSmall,
                      Int_t showerShapeWithAngle , double vertexz);
  ~StFmsEventClusterer();
  int detector() const { return mDetectorId; }
  Bool_t cluster(std::vector<FMSCluster::StFmsTower>* towers);
#ifndef __CINT__  // Hide ClusterList from CINT as it uses C++11

  ClusterList& clusters() { return mClusters; }
  const ClusterList& clusters() const { return mClusters; }
#endif  // __CINT__

 private:
#ifndef __CINT__  // Hide ClusterList from CINT as it uses C++11

  typedef ClusterList::iterator ClusterIter;
  typedef ClusterList::const_iterator ClusterConstIter;
#endif  // __CINT__

  StFmsEventClusterer(const StFmsEventClusterer&);
  StFmsEventClusterer& operator=(const StFmsEventClusterer&);
  Int_t fitEvent();
  Int_t findClusters();
  Bool_t fitClusters();
  Bool_t refitClusters();
  Double_t photonEnergyInCluster(const StFmsTowerCluster* cluster,
                                 const StFmsFittedPhoton* photon) const;
  Double_t photonEnergyInTower(const StFmsTower* tower,
                               const StFmsFittedPhoton* photon) const;
  Double_t fit1PhotonCluster(StFmsTowerCluster* cluster);
#ifndef __CINT__  // Hide Cluster(Const)Iter from CINT as it uses C++11
  /*
   Special 2-photon fit for a single cluster.

   Cluster moments must have been calculated first

   Returns the &chi;<sup>2</sup> of the fit.
   */
  Double_t fit2PhotonCluster(ClusterIter cluster);
  /*
   Fit an ambiguous cluster (one that isn't obviously 1- or 2-photon).

   First tries a 1-photon fit. If that fit is good enough, it is set as a
   1-photon cluster.
   Otherwise tries a 2-photon fit and chooses the better result.

   Returns the category of the cluster (EFmsClusterCategory in StFmsCluster).
   */
  Int_t fitAmbiguousCluster(ClusterIter cluster);
  /*
   Perform a global fit of all photons in an event.

   Update the (x, y) positions end energies of the photons in each cluster based
   on a global fit including all photons.
   Only makes sense when there is more than one photon in the event.
   Also, should only be done after prior single- or N-photon fits of the
   clusters concerned, in order to provide starting values for the photon
   positions and energies.
   Arguments:
    - nPhotons, number of photons in the event
    - nClusters, number of clusters containing those photons
    - first, iterator to the first cluster

   Returns the &chi;<sup>2</sup> of the fit.
   */
  Double_t fitGlobalClusters(unsigned int nPhotons, unsigned int nClusters,
                            ClusterIter first);
  /*
   Run tests on the lower-energy photon in a 2-photon cluster.

   Return true if the photon passes tests, in which case it is a real photon.
   Return false if it fails, in which case it is a bogus photon due to some
   problem in reconstruction - the cluster is actually a 1-photon cluster.

   Arguments:
        const iterator of ClusterList
   */
  bool validate2ndPhoton(ClusterConstIter cluster) const;
  ClusterList mClusters;  
  std::unique_ptr<StFmsClusterFitter> mFitter;   
#endif  // __CINT__
  StFmsClusterFinder mClusterFinder;   
  //const StFmsGeometry* mGeometry;   ///< FMS geometry for current run
  Int_t mDetectorId;   
  std::vector<FMSCluster::StFmsTower>* mTowers;   
  std::vector<Double_t> mTowerWidthXY;   

  StFmsDbMaker *mFmsDbMaker; 
  Int_t mGlobalRefit;        
  Int_t mMergeSmallToLarge;  
  Int_t mTry1PhotonFitWhen2PhotonFitFailed;
  Int_t mCategorizationAlgo; 
  Float_t mScaleShowerShapeLarge; 
  Float_t mScaleShowerShapeSmall; 
  Int_t mShowerShapeWithAngle; 
  double vertexz;

  ClassDef(StFmsEventClusterer, 0)
};
}  // namespace FMSCluster
#endif  // STROOT_STFMSPOINTMAKER_STFMSEVENTCLUSTERER_H_