//  Class examples to generate mass distribution 
// R.A. for A. Abate , Nov. 2008

#ifndef SPECPK_SEEN
#define SPECPK_SEEN

#include "machdefs.h"      
#include "sopnamsp.h"       
#include <math.h>
#include <iostream>
#include <vector>
#include <string> 

#include "genericfunc.h"  
#include "array.h"         
#include "histats.h"       
#include "fftwserver.h"    
#include "randinterf.h"      

#include "mdish.h"       

#define DeuxPI 2.*M_PI 

// -- SpectralShape class : test P(k) class
class SpectralShape : public GenericFunc {
public:
  SpectralShape(int typ);
// Return the value of power spectrum for wave number wk   
  virtual double operator() (double wk); 
  inline  double Value(double wk) { return((*this)(wk)); }
// Return a vector representing the power spectrum (for checking) 
  Histo GetPk(int n=256);      	
  int typ_;
};


#define TF  r_4 

// -- Four3DPk class :  3D fourier amplitudes and power spectrum 
class Four3DPk {
public:
// Constructor
  Four3DPk(TArray< complex<TF> > & fourcoedd, RandomGeneratorInterface& rg);
  Four3DPk(RandomGeneratorInterface& rg, sa_size_t szx=128, sa_size_t szy=256, sa_size_t szz=128);
  inline void SetCellSize(double dkx=DeuxPI, double dky=DeuxPI, double dkz=DeuxPI) 
  { dkx_=dkx;  dky_=dky;  dkz_=dkz; } 
  inline int SetPrtLevel(int lev=0) 
  { int olev=prtlev_; prtlev_=lev; return olev; }
  void ComputeFourierAmp(SpectralShape& pk);
  void ComputeNoiseFourierAmp(Four2DResponse& resp, bool crmask=false);
// Return the array size 
  inline sa_size_t NCells() { return fourAmp.Size(); }
// Set the cell size/step in Fourier Space
// Return the fourier amplitude matrix  
  TArray< complex<TF> > GetFourierAmp()
    { return fourAmp; }
// Return the mass density matrix    
  TArray<TF> ComputeMassDens(); 

// Return the reconstructed power spectrum as a profile histogram   
  HProf ComputePk(double s2cut=0., int nbin=256, double kmin=0., double kmax=-1.);
  void  ComputePkCumul(HProf& hp, double s2cut=0.);

protected:
  // member attribute
  RandomGeneratorInterface& rg_;
  TArray< complex<TF> > fourAmp;  // complex array of fourier coefficients
  double dkx_, dky_, dkz_;
  int prtlev_;
};


// -- MassDist2D class :  2D mass distribution 
class MassDist2D {
public:
// Constructor
  MassDist2D(GenericFunc& pk, int size=1024, double meandens=1.);
// Do the computation 
  void Compute();
// Return the array size 
  inline sa_size_t ArrSize() { return sizeA; }
// Return the fourier amplitude matrix  
  TMatrix< complex<r_8> > GetFourierAmp()
    { if (!fg_fourAmp) ComputeFourierAmp(); return fourAmp; }
// Return the mass density matrix    
  Matrix GetMassDens()
    { if (!fg_massDens) ComputeMassDens(); return massDens; }

// Return the reconstructed power spectrum as a profile histogram   
  HProf ReconstructPk(int nbin=0);
protected:
  void ComputeFourierAmp();
  void ComputeMassDens();
  
// member attribute
  GenericFunc& pkSpec;   // The spectralShape
  sa_size_t sizeA;       // 2D array size
  double meanRho;       // Mean Density
  bool fg_fourAmp;  // true -> fourAmp computed 
  TMatrix< complex<r_8> > fourAmp;  // complex array of fourier coefficients
  bool fg_massDens;  // true -> MassDens computed 
  TMatrix< r_8 > massDens;  	// real array of d rho/rho 
};


#endif