MvtxPrototype2Align.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef MVTX_P2_ALIGN_H
#define MVTX_P2_ALIGN_H

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

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

class TrkrHitSetContainer;

class MvtxPrototype2Align : public SubsysReco {

public:

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


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

  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; }

  void SetAlignmentParFileName(const char* name) { m_afname = name; }

  void SetIsGlobal(const bool yn) { m_is_global = yn; }
  bool GetIsGlobal() { return m_is_global; }

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
  std::string m_afname; // alignment par file name. If null generated by run by run

  bool m_is_global;

  PHTimeServer::timer _timer;
};

#endif