/***************************************************************************
 *
 * $Id: StHiAnalysis.h,v 1.1 2002/04/02 20:05:18 jklay Exp $                                    
 *
 * Author: Bum Choi, UT Austin, Apr 2002
 *
 ***************************************************************************
 *
 * Description:  Class to perform highpt Analysis on highpt uDSTs
 *
 *
 ***************************************************************************
 * 
 * $Log: StHiAnalysis.h,v $
 * Revision 1.1  2002/04/02 20:05:18  jklay
 * Bums analysis tools for highpt uDSTs
 *
 *
 **************************************************************************/
#ifndef StHiAnalysis_H
#define StHiAnalysis_H

#include "StHiBaseAnalysis.h"

class StHiAnalysis : public StHiBaseAnalysis{
 public:
  StHiAnalysis(const char* inputDir="./", const char* match="st_physics",
	       const char* outRootName="hianalysis.hist.root");
  virtual ~StHiAnalysis();

  void setMomentumShiftFile(char *fname);
  void setApplyMomentumShift(int doShift) {mApplyMomShift = doShift;}
 private:

  float correctPtPr(StHiMicroTrack *track, float vertexZ, float bkgRate);
  float correctPtGl(StHiMicroTrack *track, float vertexZ, float bkgRate);
  void initHistograms();
  void fillEventHistograms();
  void finishHistograms();
  Int_t findSector(float phi, float z);

  void trackLoop();
  
  // Flags etc for the momentum shift correction (year2 space charge)
  TString mMomentumShiftFileName;
  int mApplyMomShift;

  TF2 *mMomentumShiftPrMap;
  TF2 *mMomentumShiftGlMap;


  //***************************************************************
  // histograms
  //

  //Event level histograms
  TH3D* h3VertexXYZ;  
  TH2D* h2ZDCSumVsCTB;

  TH2D* h2NGoodGlobalsVsNch;
  TH1D* h1FlowCent;

  TH1D* h1NEvent; //! # of events for the centrality class
  TH1D* h1EtaCut; //! just in case, store the eta cut  
  
//Track level histograms
  TH1D* h1FitPts;

  //Before cut on specified variable, but all cut on |eta|<0.5
  TH2D* h2DcaGlVsSector;  
  TH2D* h2DcaXYGlVsSector;  
  TH2D* h2FitPtsVsSector;  
  TH3D* h3FitPtsSectorPtPr;  
  TH2D* h2MaxPtsVsSector;  
  TH2D* h2AllPtsVsSector;  

  //After all track cuts
  TH1D* h1YieldVsSector;
  TH2D* h2PrPtVsSector;  
  TH2D* h2GlPtVsSector;  
  TH2D* h2ResPrPtVsSector;  
  TH2D* h2ResGlPtVsSector;  

  // 0 is plus, 1 is minus 2 is plus+minus  
  struct PlusMinus{
    TH3D* h3PhiPrDcaXYGlPtPr; // look at phi counts
    TH3D* h3PhiGlDcaXYGlPtGl;
  
    TH3D* h3DcaGlDcaXYGlPtPr; // look at dca's for different pt bins
    TH3D* h3DcaGlDcaXYGlPtGl;

    // --backgrounds
    TH2D* h2SDcaGlPtPrRebin;
    TH2D* h2DcaXYGlPtPrRebin;
    TH2D* h2DcaGlPtPrRebin;
    // --
    
    TH3D* h3PhiPrFitPtsPtPr; // phi,fit pts,pt
    TH3D* h3VtxZFitPtsPtPr; // vtx z, fit pts, pt
    TH3D* h3VtxZFitPtsEtaPr; // vtx z, fit pts, eta

    TH3D* h3PhiPrAllPtsPtPr; // phi, all pts, pt
    TH3D* h3PhiPrMaxPtsPtPr; // phi, all pts, pt
    
    TH3D* h3FlowCentFitPtsPtPr; // flow cent, fit pts, pt

    //TH2D* h2ZDCCentralityPtPr; // pt yields and centrality
    TH2D* h2CentralityPtPr;
    TH3D* h3VtxZEtaPrPtPr; // reality check of the eta vs vertex
    TH3D* h3EtaPhiPtPr; 
    TH3D* h3VtxZEtaGlPtGl;

//Hey, this is everything we need to do the spectra analysis!
    TH1D* h1RawPtGlVarBin0; //
    TH1D* h1RawPtGlVarBin1; //
    TH1D* h1RawXtGl; //
    TH1D* h1RawPtPrVarBin0; //
    TH1D* h1RawPtPrVarBin1; //
    TH1D* h1RawXtPr; //

    TH1D* h1OneOverPtGlVarBin0; //
    TH1D* h1OneOverPtGlVarBin1; //
    TH1D* h1OneOverXtGl; //
    TH1D* h1OneOverPtPrVarBin0; //
    TH1D* h1OneOverPtPrVarBin1; //
    TH1D* h1OneOverXtPr; //

    TH1D* h1WeightedMeanPtGlVarBin0; //	h1Raw/h1OneOverPt
    TH1D* h1WeightedMeanPtGlVarBin1; //
    TH1D* h1WeightedMeanXtGl; //
    TH1D* h1WeightedMeanPtPrVarBin0; //
    TH1D* h1WeightedMeanPtPrVarBin1; //
    TH1D* h1WeightedMeanXtPr; //

    TH1D* h1RawSpectraNarrowBins; //

  };

  //0 is East 1 is West 2 is Both 
  struct EastWest{
    PlusMinus pm[3];   
  };

  EastWest ew[3]; //!

//Note:  East and West, where track and vtx Z are on same side can be obtained by
// setting the vertex Z cut to one side or the other and comparing the East from VtxZ<0 with
// West from VtxZ>0 runs.

  TH3D* h3ResPtPrGlPtPrDcaXYGl; //!
  TH3D* h3ResPtPrGlPtGlDcaXYGl; //!
 

  ClassDef(StHiAnalysis,1)
};

#endif