include/root/TFFTRealComplex.h

0001 // @(#)root/fft:$Id$
0002 // Author: Anna Kreshuk   07/4/2006
0003
0004 /*************************************************************************
0005  * Copyright (C) 1995-2006, Rene Brun and Fons Rademakers.               *
0006  * All rights reserved.                                                  *
0007  *                                                                       *
0008  * For the licensing terms see $ROOTSYS/LICENSE.                         *
0009  * For the list of contributors see $ROOTSYS/README/CREDITS.             *
0010  *************************************************************************/
0011
0012 #ifndef ROOT_TFFTRealComplex
0013 #define ROOT_TFFTRealComplex
0014
0015 #include "TVirtualFFT.h"
0016 #include "TString.h"
0017
0018 class TComplex;
0019
0020 class TFFTRealComplex: public TVirtualFFT {
0021  protected:
0022    void     *fIn;        //input array
0023    void     *fOut;       //output array
0024    void     *fPlan;      //fftw plan (the plan how to compute the transform)
0025    Int_t     fNdim;      //number of dimensions
0026    Int_t     fTotalSize; //total size of the transform
0027    Int_t    *fN;         //transform sizes in each dimension
0028    TString   fFlags;     //transform flags
0029
0030    UInt_t MapFlag(Option_t *flag);
0031
0032  public:
0033    TFFTRealComplex();
0034    TFFTRealComplex(Int_t n, Bool_t inPlace);
0035    TFFTRealComplex(Int_t ndim, Int_t *n, Bool_t inPlace);
0036    ~TFFTRealComplex() override;
0037
0038    void       Init( Option_t *flags, Int_t /*sign*/,const Int_t* /*kind*/) override;
0039
0040    virtual Int_t      GetSize() const {return fTotalSize;}
0041    Int_t     *GetN()    const override {return fN;}
0042    Int_t      GetNdim() const override {return fNdim;}
0043    Option_t  *GetType() const override {return "R2C";}
0044    Int_t      GetSign() const override {return 1;}
0045    Option_t  *GetTransformFlag() const override {return fFlags;}
0046    Bool_t     IsInplace() const override {if (fOut) return kTRUE; else return kFALSE;};
0047
0048    void       GetPoints(Double_t *data, Bool_t fromInput = kFALSE) const override;
0049    Double_t   GetPointReal(Int_t ipoint, Bool_t fromInput = kFALSE) const override;
0050    Double_t   GetPointReal(const Int_t *ipoint, Bool_t fromInput = kFALSE) const override;
0051    void       GetPointComplex(Int_t ipoint, Double_t &re, Double_t &im, Bool_t fromInput=kFALSE) const override;
0052    void       GetPointComplex(const Int_t *ipoint, Double_t &re, Double_t &im, Bool_t fromInput=kFALSE) const override;
0053    Double_t  *GetPointsReal(Bool_t fromInput=kFALSE) const override;
0054    void       GetPointsComplex(Double_t *re, Double_t *im, Bool_t fromInput = kFALSE) const override ;
0055     void      GetPointsComplex(Double_t *data, Bool_t fromInput = kFALSE) const override ;
0056
0057    void       SetPoint(Int_t ipoint, Double_t re, Double_t im = 0) override;
0058    void       SetPoint(const Int_t *ipoint, Double_t re, Double_t im = 0) override;
0059    void       SetPoints(const Double_t *data) override;
0060    void       SetPointComplex(Int_t ipoint, TComplex &c) override;
0061    void       SetPointsComplex(const Double_t *re, const Double_t *im) override;
0062    void       Transform() override;
0063
0064    ClassDefOverride(TFFTRealComplex,0);
0065 };
0066
0067 #endif