PFourierCanvas.h

Go to the documentation of this file.

/***************************************************************************

  PFourierCanvas.h

  Author: Andreas Suter
  e-mail: andreas.suter@psi.ch

***************************************************************************/

/***************************************************************************
 *   Copyright (C) 2007-2023 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_