/*
 * Sub-phase averaging tools
 */

#ifndef SubPhase_INC
#define SubPhase_INC

#include <vector>
#include "common/Domain.h"
#include "common/Communication.h"
#include "analysis/analysis.h"
#include "analysis/distance.h"
#include "analysis/Minkowski.h"
#include "common/Utilities.h"
#include "common/MPI.h"
#include "IO/MeshDatabase.h"
#include "IO/Reader.h"
#include "IO/Writer.h"

class phase {
public:
    int Nc;
    double p;
    double M, Px, Py, Pz, K, visc;
    double V, A, H, X;
    void reset() {
        p = M = Px = Py = Pz = K = 0.0;
        visc = 0.0;
        V = A = H = X = 0.0;
        Nc = 1;
    }

private:
};

class interface {
public:
    int Nc;
    double M, Px, Py, Pz, K;
    double Mw, Mn, Pnx, Pny, Pnz, Pwx, Pwy, Pwz, Kw, Kn;
    double V, A, H, X;
    void reset() {
        Nc = 0;
        M = Px = Py = Pz = K = 0.0;
        V = A = H = X = 0.0;
        Mw = Mn = Pnx = Pny = Pnz = Pwx = Pwy = Pwz = Kw = Kn = 0.0;
    }

private:
};

class SubPhase {
public:
    std::shared_ptr<Domain> Dm;
    double Volume;
    // input variables
    double rho_n, rho_w;
    double nu_n, nu_w;
    double gamma_wn, beta;
    double Fx, Fy, Fz;
    /*
	 * indices 
	 * 	    w - water phase
	 * 	    n - not water phase
	 * 		c - connected part
	 * 		d - disconnected part
	 * 		i - interface region
	 * 		b - bulk (total)
	 */
    // local entities
    phase wc, wd, wb, nc, nd, nb, solid;
    interface iwn, iwnc;
    interface ifs;

    // global entities
    phase gwc, gwd, gwb, gnc, gnd, gnb, gsolid;
    interface giwn, giwnc;
    interface gifs;
    /* fluid-solid wetting interaction */
    double total_wetting_interaction, count_wetting_interaction;
    double total_wetting_interaction_global, count_wetting_interaction_global;

    //...........................................................................
    int Nx, Ny, Nz;
    IntArray PhaseID;      // Phase ID array (solid=0, non-wetting=1, wetting=2)
    BlobIDArray Label_WP;  // Wetting phase label
    BlobIDArray Label_NWP; // Non-wetting phase label index (0:nblobs-1)
    std::vector<BlobIDType>
        Label_NWP_map;    // Non-wetting phase label for each index
    DoubleArray Rho_n;    // density field
    DoubleArray Rho_w;    // density field
    DoubleArray Phi;      // phase indicator field
    DoubleArray DelPhi;   // Magnitude of Gradient of the phase indicator field
    DoubleArray Pressure; // pressure field
    DoubleArray Vel_x;    // velocity field
    DoubleArray Vel_y;
    DoubleArray Vel_z;
    DoubleArray Dissipation;
    DoubleArray SDs;

    std::shared_ptr<Minkowski> morph_w;
    std::shared_ptr<Minkowski> morph_n;
    std::shared_ptr<Minkowski> morph_i;

    SubPhase(std::shared_ptr<Domain> Dm);
    ~SubPhase();

    void SetParams(double rhoA, double rhoB, double tauA, double tauB,
                   double force_x, double force_y, double force_z, double alpha,
                   double beta);
    void Basic();
    void Full();
    void Write(int time);
    void AggregateLabels(const std::string &filename);

private:
    FILE *TIMELOG;
    FILE *SUBPHASE;
};

#endif