PFourierCanvas.h

Go to the documentation of this file.

/***************************************************************************
 
  PFourierCanvas.h
 
  Author: Andreas Suter
  e-mail: andreas.suter@psi.ch
 
***************************************************************************/
 
/***************************************************************************
 *   Copyright (C) 2007-2026 by Andreas Suter                              *
 *   andreas.suter@psi.ch                                                  *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
 
#ifndef _PFOURIERCANVAS_H_
#define _PFOURIERCANVAS_H_
 
#include <vector>
#include <memory>
 
#include <TObject.h>
#include <TQObject.h>
#include <TTimer.h>
#include <TStyle.h>
#include <TRootCanvas.h>
#include <TGMenu.h>
#include <TCanvas.h>
#include <TPaveText.h>
#include <TLegend.h>
#include <TPad.h>
#include <TH1F.h>
#include <TLatex.h>
 
#include "PMusr.h"
#include "PFourier.h"
 
// Canvas menu id's
#define P_MENU_ID_FOURIER              10001
#define P_MENU_ID_AVERAGE              10002
#define P_MENU_ID_AVERAGE_PER_DATA_SET 10003
#define P_MENU_ID_EXPORT_DATA          10004
 
#define P_MENU_ID_FOURIER_REAL           100
#define P_MENU_ID_FOURIER_IMAG           101
#define P_MENU_ID_FOURIER_REAL_AND_IMAG  102
#define P_MENU_ID_FOURIER_PWR            103
#define P_MENU_ID_FOURIER_PHASE          104
#define P_MENU_ID_FOURIER_PHASE_OPT_REAL 105
#define P_MENU_ID_FOURIER_PHASE_PLUS     106
#define P_MENU_ID_FOURIER_PHASE_MINUS    107
 
//------------------------------------------------------------------------
struct PFourierCanvasDataSet {
  TH1F *dataFourierRe;           
  TH1F *dataFourierIm;           
  TH1F *dataFourierPwr;          
  TH1F *dataFourierPhase;        
  TH1F *dataFourierPhaseOptReal; 
  std::vector<Double_t> optPhase;     
};
 
//------------------------------------------------------------------------
typedef std::vector<PFourierCanvasDataSet> PFourierCanvasDataList;
 
//--------------------------------------------------------------------------
class PFourierCanvas : public TObject, public TQObject
{
  public:
    PFourierCanvas();

    PFourierCanvas(std::vector<PFourier*> &fourier, PIntVector dataSetTag, const Char_t* title,
                   const Bool_t showAverage, const Bool_t showAveragePerDataSet,
                   const Int_t fourierPlotOpt, Double_t fourierXrange[2], Double_t phase,
                   Int_t wtopx, Int_t wtopy, Int_t ww, Int_t wh, const Bool_t batch);

    PFourierCanvas(std::vector<PFourier*> &fourier, PIntVector dataSetTag, const Char_t* title,
                   const Bool_t showAverage, const Bool_t showAveragePerDataSet,
                   const Int_t fourierPlotOpt, Double_t fourierXrange[2], Double_t phase,
                   Int_t wtopx, Int_t wtopy, Int_t ww, Int_t wh,
                   const PIntVector markerList, const PIntVector colorList, const Bool_t batch);

    virtual void Done(Int_t status=0); // *SIGNAL*

    virtual void HandleCmdKey(Int_t event, Int_t x, Int_t y, TObject *selected); // SLOT

    virtual void HandleMenuPopup(Int_t id); // SLOT

    virtual void LastCanvasClosed(); // SLOT

    virtual void UpdateFourierPad();

    virtual void UpdateInfoPad();

    virtual Bool_t IsValid() { return fValid; }

    virtual void SetTimeout(Int_t ival);

    virtual void SaveGraphicsAndQuit(const Char_t *fileName);

    virtual void ExportData(const Char_t *pathFileName);
 
  private:
    Int_t  fTimeout;         
    Bool_t fBatchMode;       
    Bool_t fValid;           
    Bool_t fAveragedView;    
    Bool_t fAveragedViewPerDataSet; 
    PIntVector fDataSetTag;  
    Int_t  fCurrentPlotView; 
    Double_t fInitialXRange[2]; 
    Double_t fInitialYRange[2]; 
 
    TString fTitle;
    TString fXaxisTitle;
    std::vector<PFourier*> fFourier; 
    PFourierCanvasDataList fFourierHistos; 
    PFourierCanvasDataList fFourierAverage; 
    Double_t fCurrentFourierPhase; 
    std::unique_ptr<TLatex> fCurrentFourierPhaseText; 
 
    std::unique_ptr<TStyle> fStyle; 
 
    std::unique_ptr<TTimer> fTimeoutTimer;  
 
    PIntVector fMarkerList; 
    PIntVector fColorList;  
 
    // canvas menu related variables
    TRootCanvas *fImp;       
    TGMenuBar   *fBar;       
    TGPopupMenu *fPopupMain; 
    std::unique_ptr<TGPopupMenu> fPopupFourier; 
 
    // canvas related variables
    std::unique_ptr<TCanvas>   fMainCanvas; 
    std::unique_ptr<TPaveText> fTitlePad;   
    std::unique_ptr<TPad>      fFourierPad; 
    std::unique_ptr<TLegend>   fInfoPad;    
    std::unique_ptr<TLegend>   fLegAvgPerDataSet; 

    virtual void CreateXaxisTitle();

    virtual void CreateStyle();

    virtual void InitFourierDataSets();

    virtual void InitFourierCanvas(const Char_t* title, Int_t wtopx, Int_t wtopy, Int_t ww, Int_t wh);

    virtual void CleanupAverage();

    virtual void HandleAverage();

    virtual void CalcPhaseOptReal();

    virtual void PlotFourier();

    virtual void PlotFourierPhaseValue();

    virtual void PlotAverage();

    virtual void IncrementFourierPhase();

    virtual void DecrementFourierPhase();

    virtual Double_t GetMaximum(TH1F* histo, Double_t xmin=-1.0, Double_t xmax=-1.0);

    virtual Double_t GetMinimum(TH1F* histo, Double_t xmin=-1.0, Double_t xmax=-1.0);

    virtual Double_t GetInterpolatedValue(TH1F* histo, Double_t xVal);

    virtual TString GetDataSetName(TString title);
 
    ClassDef(PFourierCanvas, 1)
};
 
#endif // _PFOURIERCANVAS_H_