/*
Implementation of Lee et al JCP 2016 lattice boltzmann model
 */
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <iostream>
#include <exception>
#include <stdexcept>
#include <fstream>

#include "common/Communication.h"
#include "common/MPI.h"
#include "ProfilerApp.h"
#include "threadpool/thread_pool.h"
#include "common/ScaLBL.h"
#include "common/WideHalo.h"

#ifndef ScaLBL_FreeLeeModel_INC
#define ScaLBL_FreeLeeModel_INC

class ScaLBL_FreeLeeModel{
public:
  ScaLBL_FreeLeeModel(int RANK, int NP, const Utilities::MPI& COMM);
	~ScaLBL_FreeLeeModel();	
	
	// functions in they should be run
	void ReadParams(string filename);
	void ReadParams(std::shared_ptr<Database> db0);
	void SetDomain();
	void ReadInput();
	void Create_TwoFluid();
	void Initialize_TwoFluid();
	double Run_TwoFluid(int returntime);

	void WriteDebug_TwoFluid();
	void Create_SingleFluid();
	void Initialize_SingleFluid();
	void Run_SingleFluid();
	
	void WriteDebug_SingleFluid();
    // test utilities
    void Create_DummyPhase_MGTest();
    void MGTest();
	
	bool Restart,pBC;
	int timestep,timestepMax;
	int BoundaryCondition;
	double tauA,tauB,rhoA,rhoB;
    double tau, rho0;//only for single-fluid Lee model
    double tauM;//relaxation time for phase field (or mass)
	double W,gamma,kappa,beta;
	double Fx,Fy,Fz,flux;
	double din,dout,inletA,inletB,outletA,outletB;
	
	int Nx,Ny,Nz,N,Np;
    int Nxh,Nyh,Nzh,Nh; // extra halo width
	int rank,nprocx,nprocy,nprocz,nprocs;
	double Lx,Ly,Lz;

	std::shared_ptr<Domain> Dm;   // this domain is for analysis
	std::shared_ptr<Domain> Mask; // this domain is for lbm
	std::shared_ptr<ScaLBL_Communicator> ScaLBL_Comm;
	std::shared_ptr<ScaLBL_Communicator> ScaLBL_Comm_Regular;
	std::shared_ptr<ScaLBLWideHalo_Communicator> ScaLBL_Comm_WideHalo;

    // input database
    std::shared_ptr<Database> db;
    std::shared_ptr<Database> domain_db;
    std::shared_ptr<Database> freelee_db;
    std::shared_ptr<Database> analysis_db;
    std::shared_ptr<Database> vis_db;

    IntArray Map;
    signed char *id;    
	int *NeighborList;
	int *dvcMap;
	double *gqbar, *hq;
	double *mu_phi, *Den, *Phi;
	double *ColorGrad;
	double *Velocity;
	double *Pressure;
	
	void getPhase(DoubleArray &PhaseValues);
	void getPotential(DoubleArray &PressureValues, DoubleArray &MuValues);
	void getVelocity(DoubleArray &Vx, DoubleArray &Vy, DoubleArray &Vz);
    void getData_RegularLayout(const double *data, DoubleArray &regdata);
	
	DoubleArray SignDist;
	
private:
	Utilities::MPI comm;
    
	int dist_mem_size;
	int neighborSize;
	// filenames
    char LocalRankString[8];
    char LocalRankFilename[40];
    char LocalRestartFile[40];
   
    //int rank,nprocs;
    void LoadParams(std::shared_ptr<Database> db0);
	void AssignComponentLabels_ChemPotential_ColorGrad();

};
#endif