test120.cpp

#include "preprocessor_defines.dat"
#include "Class_Global.hpp"
#include "Class_Para_Tree.hpp"
#include "User_Data_Comm.hpp"
#include "User_Data_LB.hpp"

using namespace std;

// =================================================================================== //

int main(int argc, char *argv[]) {

#if NOMPI==0
    MPI::Init(argc, argv);

    {
#endif
        int iter = 0;

        Class_Para_Tree<3> pablo120;

        int idx = 0;
        pablo120.setBalance(idx,false);

        for (iter=1; iter<6; iter++){
            pablo120.adaptGlobalRefine();
        }

#if NOMPI==0

        pablo120.loadBalance();
#endif

        double xc, yc, zc;
        double radius = 0.15;
        double t0 = 0;
        double Dt = 0.000025;
        double omega = 2.0*3.14/0.001;

        xc = 0.25*cos(omega* Dt) + 0.5 ;
        yc = 0.25*sin(omega* Dt) + 0.5;
        zc = 100*Dt;

        uint32_t nocts = pablo120.getNumOctants();
        vector<double> oct_data(nocts, 0.0);

        int itstart = 1;
        int itend = 460;

        for (iter=itstart; iter<itend; iter++){

            xc = 0.25*cos(omega* Dt* (double) iter) + 0.5 ;
            yc = 0.25*sin(omega* Dt* (double) iter) + 0.5;
            zc = 100*Dt*iter;

            for (int i=0; i<nocts; i++){
                bool inside = false;
                vector<vector<double> > nodes = pablo120.getNodes(i);
                vector<double> center = pablo120.getCenter(i);
                oct_data[i] = sqrt((pow((center[0]-xc),2.0)+pow((center[1]-yc),2.0)+pow((center[2]-zc),2.0)));
                for (int j=0; j<global3D.nnodes; j++){
                    double x = nodes[j][0];
                    double y = nodes[j][1];
                    double z = nodes[j][2];
                    if (pow((x-xc),2.0)+pow((y-yc),2.0)+pow((z-zc),2.0) <= pow(radius,2.0)){
                        if (pablo120.getLevel(i) < 6){
                            pablo120.setMarker(i,1);
                        }
                        else{
                            pablo120.setMarker(i,0);
                        }
                        inside = true;
                    }
                }
                if (pablo120.getLevel(i) > 5 && !inside){
                    pablo120.setMarker(i,-1);
                }
            }

            bool adapt = pablo120.adapt();

#if NOMPI==0

            pablo120.loadBalance();
#endif

            nocts = pablo120.getNumOctants();
            vector<double> oct_data_new(nocts, 0.0);
            for (int i=0; i<nocts; i++){
                vector<double> center = pablo120.getCenter(i);
                oct_data_new[i] = sqrt((pow((center[0]-xc),2.0)+pow((center[1]-yc),2.0)+pow((center[2]-zc),2.0)));
            }

            oct_data.resize(nocts);
            oct_data = oct_data_new;
            oct_data_new.clear();

            pablo120.updateConnectivity();
            pablo120.writeTest("Pablo120_iter"+to_string(static_cast<unsigned long long>(iter)), oct_data);
        }
#if NOMPI==0
    }

    MPI::Finalize();
#endif
}