source: JEM-EUSO/esaf_cc_at_lal/packages/reconstruction/modules/base/clustering/include/RobustModule.hh @ 114

// ESAF : Euso Simulation and Analysis Framework
// $Id$
// Author: Svetlana Biktemerova   Okt, 20 2011

#ifndef __ROBUSTMODULE_HH_
#define __ROBUSTMODULE_HH_

#include "euso.hh"
#include "RecoModule.hh"

#include <TROOT.h>
#include <TVector2.h>
#include <TF1.h>
#include <vector>
#include <map>

#ifdef USE_VISUALIZATION
class EnergyViewer;
#endif
class RecoEvent;
class RecoRootEvent;
class EOpticsResponse;
class ERunParameters;
class RecoPixelData;
class RecoShowerTrack;
class EnergyModule;
class HmaxByShapeMethod;
class TVector3;
class TH1D;
class TH2D;
class TF1;
class TGraph;
class TFile;
class EusoCluster;

class RobustModule : public RecoModule  {
public:
        // ctor
        RobustModule();
        // dtor
        virtual ~RobustModule();
        
        
        // Init method: called at the beginning of a run
        Bool_t Init();
        
        // called before each event process
        Bool_t PreProcess();
        
        // event processing
        Bool_t Process( RecoEvent* );
        
        // called after processing an event
        Bool_t PostProcess();
        
        // save root output
        virtual Bool_t SaveRootData( RecoRootEvent* );
        
        // save module output
        Bool_t SaveGlobalData();
        
        // called at the end of each run
        Bool_t Done();
        
        // memory clean
        void UserMemoryClean();
        
        static RobustModule *Get();

        class TrackInformation: public TObject {
        public:
                TrackInformation() : fLine("trackline", "[0]*x+[1]", 0., 1.) {};
                ~TrackInformation() {};
                
                void Clear(){   fLine_point1.Set(0.,0.);
                fLine_point2.Set(0.,0.);
                fSig_point1=fSig_point2=0.;
                fSignalPerGtu.clear();} 
                TVector2 fLine_point1;
                TVector2 fLine_point2;
                double fSig_point1;
                double fSig_point2;
                map< int, vector<RecoPixelData*> > fSignalPerGtu;
                TF1 fLine;
                //      ClassDef(TrackInformation,1);
        };
        
private:
        // Private methods
        Bool_t FindTrackOnFocalPlane(double *a=0);
        void Coefficients(double &a, double &b, const map <int, TVector3 >& c);
        double chi(const map <int, TVector3 >& c,double a,double b);
        Bool_t GetSignalPixels(double a, double b, map <int, TVector3 >& pixel_map);
        void PointsIntersection(double *px, double *py, double *x);
        void GetLimits(TH1D* h,double par, double opt, double& l1, double& l2);
        void SubtractBg(TH1D* h, double bg, double power);
        void GetProjectionX(TH2D*, TH1D*);
        void GetProjectionY(TH2D*, TH1D*);
        
        // Private variables
        RecoEvent                    *fEvent;              // pointer to the current reco event object
        ERunParameters               *fRunpars;            // ERunParameters
        
        double fx1;  // the coordinates of the box where our signal pixels are
        double fx2;  //
        double fy1;  //
        double fy2;  //
        double* fWeights; 
        
        static RobustModule *fMe;
                
        TrackInformation ftrack_info;
        vector<int> sigpix;
        ClassDef(RobustModule,0)
};

#endif  /* __ROBUSTMETHOD_HH_ */