MvtxAlign.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file MvtxAlign.h

#ifndef __PHG4SVTXCLUSTERIZER_H__
#define __PHG4SVTXCLUSTERIZER_H__

#include <fun4all/SubsysReco.h>
#include <phool/PHTimeServer.h>
#include <map>
#include <limits.h>

#include <trackbase/TrkrClusterContainer.h>
#include <trackbase/TrkrDefUtil.h>

class TrkrHitSetContainer;

class MvtxAlign : public SubsysReco {

public:

  struct AlignmentPar {
    double dx;
    double dy;
    double dz;
  };


  MvtxAlign(const std::string &name = "MvtxAlign");
  virtual ~MvtxAlign() {}

  int Init(PHCompositeNode *topNode) {return 0;}

  int InitRun(PHCompositeNode *topNode);

  int process_event(PHCompositeNode *topNode);

  int End(PHCompositeNode *topNode) {return 0;}

  void AddAlignmentPar(TrkrDefs::hitsetkey key, double dx, double dy, double dz);

  void PrintAlignmentPars(std::ostream &os = std::cout) const;

  void SetAlignmentParFileDir(const std::string fdir) { fdir_ = fdir; }

  void SetAlignmentParFromFile(const bool yn) { apff_ = yn; }

private:

  // read the alignment parameters from file based on run number
  int ReadAlignmentParFile();

  // node tree storage pointers
  TrkrClusterContainer* clusters_;

  // storage object for misalignments
  std::map<TrkrDefs::hitsetkey, AlignmentPar> alignmap_;

  // directory for alignment parameter files
  std::string fdir_;
  int runnumber_;
  bool apff_; // alignment par from file

  PHTimeServer::timer _timer;
};

#endif