HGridUnitTest.cpp File Reference

#include <Species/LinearViscoelasticSpecies.h>
#include "Mercury3D.h"
#include "DPMBase.h"
#include "Boundaries/PeriodicBoundary.h"
#include "Particles/BaseParticle.h"

Go to the source code of this file.

Classes
class	MD_demo
class	HGrid_demo

Functions
int	main (int argc UNUSED, char *argv[] UNUSED)

Function Documentation

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

Definition at line 163 of file HGridUnitTest.cpp.

References MD_demo::alpha, BaseHandler< T >::begin(), BOTTOMUP, BaseHandler< T >::end(), EXPONENTIAL, LINEAR, MD_demo::N, MD_demo::nu, MD_demo::omega, DPMBase::particleHandler, ParticleSpecies::setDensity(), MercuryBase::setHGridDistribution(), MercuryBase::setHGridMaxLevels(), MercuryBase::setHGridMethod(), Files::setName(), DPMBase::setParticleDimensions(), Files::setSaveCount(), DPMBase::setSystemDimensions(), DPMBase::setTimeMax(), DPMBase::setTimeStep(), MD_demo::setupInitialConditions(), DPMBase::solve(), MD_demo::species, and TOPDOWN.

{
    MD_demo MD_problem;
    MD_problem.species->setDensity(2000);

    MD_problem.omega = 40;
    MD_problem.alpha = -2;
    MD_problem.nu = 0.7;
    MD_problem.N = 1000;
    MD_problem.setSystemDimensions(3);
    MD_problem.setParticleDimensions(3);
    MD_problem.setName("HGridUnitTest_MD");
    MD_problem.setTimeStep(0.1);
    MD_problem.setTimeMax(0.09);
    MD_problem.setSaveCount(1);
    MD_problem.setupInitialConditions();

    HGrid_demo HGrid_problem1(MD_problem);
    HGrid_problem1.setHGridMethod(TOPDOWN);
    HGrid_problem1.setHGridMaxLevels(3);
    HGrid_problem1.setHGridDistribution(EXPONENTIAL);
    HGrid_problem1.setName("HGridUnitTest_HGrid1");
    
    HGrid_demo HGrid_problem2(MD_problem);
    HGrid_problem2.setHGridMethod(BOTTOMUP);
    HGrid_problem2.setHGridMaxLevels(8);
    HGrid_problem2.setHGridDistribution(LINEAR);
    HGrid_problem2.setName("HGridUnitTest_HGrid2");

    std::cout << "Solving the MD problem" << std::endl;
    MD_problem.solve();
    std::cout << "Solving the first HGrid problem" << std::endl;
    HGrid_problem1.solve();
    std::cout << "Solving the second HGrid problem" << std::endl;
    HGrid_problem2.solve();
    
    // Check the particles are in the same place for all three problem i.e. no HGrid and two different HGrid settings 
    std::vector<BaseParticle*>::iterator hGrid1It = HGrid_problem1.particleHandler.begin();
    std::vector<BaseParticle*>::iterator hGrid2It = HGrid_problem2.particleHandler.begin();
    for (std::vector<BaseParticle*>::iterator mdIt = MD_problem.particleHandler.begin(); mdIt != MD_problem.particleHandler.end(); ++mdIt)
    {
        if (!(*mdIt)->getPosition().isEqualTo((*hGrid1It)->getPosition(),1e-10))
        {
            exit(-1);
        }
        if (!(*mdIt)->getPosition().isEqualTo((*hGrid2It)->getPosition(), 1e-10))
        {
            exit(-1);
        }
        ++hGrid1It;
        ++hGrid2It;
    }
}