ContactLawTestHelpers.h

Go to the documentation of this file.

// This file is part of the MercuryDPM project (https://www.mercurydpm.org).
// Copyright (c), The MercuryDPM Developers Team. All rights reserved.
// License: BSD 3-Clause License; see the LICENSE file in the root directory.
 
#ifndef CONTACT_LAW_TEST_HELPERS
#define CONTACT_LAW_TEST_HELPERS
 
#include "DPMBase.h"
#include "Helpers/FileIOHelpers.h"
#include "Logger.h"
#include "Particles/BaseParticle.h"
#include "Species/BaseSpecies.h"
#include "Walls/InfiniteWall.h"
 
void loadingTest(const ParticleSpecies* species, Mdouble displacement, Mdouble velocity, Mdouble radius,
                          std::string name)
{
    class LoadingTest : public DPMBase
    {
        const ParticleSpecies* species;
        Mdouble displacement;
        Mdouble velocity;
        Mdouble radius;
    public:
        //public variables
        LoadingTest(const ParticleSpecies* species, Mdouble displacement, Mdouble velocity, Mdouble radius)
                : species(species), displacement(displacement), velocity(velocity), radius(radius)
        {}
 
        void setupInitialConditions() override
        {
            //setName("LoadingTest"+species->getName());
            setTimeMax(2.0 * displacement / velocity);
            setTimeStep(2e-3 * getTimeMax());
            setSaveCount(1);
            setFileType(FileType::NO_FILE);
            fStatFile.setFileType(FileType::ONE_FILE);
 
            setMax({radius, radius, radius + radius});
            setMin({-radius, -radius, 0});
            setSystemDimensions(3);
            setParticleDimensions(3);
 
            speciesHandler.copyAndAddObject(*species);
 
            SphericalParticle p;
            p.setSpecies(speciesHandler.getObject(0));
            p.setRadius(radius);
            p.setPosition({0, 0, radius});
            particleHandler.copyAndAddObject(p);
 
            InfiniteWall w;
            w.setSpecies(speciesHandler.getObject(0));
            w.set(Vec3D(0, 0, -1), Vec3D(0.0, 0.0, 0.0));
            wallHandler.copyAndAddObject(w);
        }
 
        void actionsBeforeTimeStep() override
        {
            BaseParticle* p = particleHandler.getLastObject();
            logger.assert_debug(p,"Empty particle handler");
            p->setAngularVelocity({0, 0, 0});
 
            //Moving particle normally into surface
            if (getTime() <= displacement / velocity)
            {
                p->setVelocity({0, 0, velocity});
                p->setPosition({0, 0, radius - velocity * getTime()});
            }
            else
            {
                p->setVelocity({0, 0, -velocity});
                p->setPosition({0, 0, radius - displacement - displacement + velocity * getTime()});
            }
        }
    } test(species, displacement, velocity, radius);
    test.setName(name);
    test.solve();
    helpers::writeToFile(test.getName() + ".gnu", "plot '" + test.getName() + ".fstat' u 7:9 w lp");
    logger(INFO, "finished loading test: run 'gnuplot %.gnu' to view output", test.getName());
}
 
void normalAndTangentialLoadingTest(const ParticleSpecies* species, Mdouble displacement,
                                             Mdouble tangentialDisplacement, Mdouble velocity, Mdouble radius,
                                             std::string name)
{
    class LoadingTest : public DPMBase
    {
        const ParticleSpecies* species;
        Mdouble displacement;
        Mdouble tangentialDisplacement;
        Mdouble velocity;
        Mdouble radius;
    public:
        //public variables
        LoadingTest(const ParticleSpecies* species, Mdouble displacement, Mdouble tangentialDisplacement,
                    Mdouble velocity, Mdouble radius)
                : species(species), displacement(displacement), tangentialDisplacement(tangentialDisplacement),
                  velocity(velocity), radius(radius)
        {}
 
        void setupInitialConditions() override
        {
            //setName("TangentialLoadingTest"+species->getName());
            setTimeMax(4.0 * tangentialDisplacement / velocity);
            setTimeStep(4e-4 * getTimeMax());
            setSaveCount(1);
            setFileType(FileType::NO_FILE);
            fStatFile.setFileType(FileType::ONE_FILE);
 
            setMax({radius, radius, radius + radius});
            setMin({-radius, -radius, 0});
            setSystemDimensions(3);
            setParticleDimensions(3);
 
            speciesHandler.copyAndAddObject(*species);
 
            SphericalParticle p;
            p.setSpecies(speciesHandler.getObject(0));
            p.setRadius(radius);
            p.setPosition({0, 0, radius - displacement});
            particleHandler.copyAndAddObject(p);
 
            InfiniteWall w;
            w.setSpecies(speciesHandler.getObject(0));
            w.set(Vec3D(0, 0, -1), Vec3D(0.0, 0.0, 0.0));
            wallHandler.copyAndAddObject(w);
        }
 
        void actionsBeforeTimeStep() override
        {
            BaseParticle* p = particleHandler.getLastObject();
            logger.assert_debug(p,"Empty particle handler");
            p->setAngularVelocity({0, 0, 0});
    
            //Moving particle cyclically right and left between +-tangentialDisplacement
            bool moveRight = static_cast<int>(getTime() / (2.0*tangentialDisplacement / velocity) +0.5)%2==0;
            if (moveRight)
            {
                p->setVelocity({-velocity, 0, 0});
                p->setPosition({tangentialDisplacement - velocity * getTime(), 0, radius - displacement});
            }
            else
            {
                p->setVelocity({velocity, 0, 0});
                p->setPosition({-2*tangentialDisplacement + velocity * getTime(), 0, radius - displacement});
            }
        }
 
    } test(species, displacement, tangentialDisplacement, velocity, radius);
    test.setName(name);
    test.solve();
    helpers::writeToFile(test.getName() + ".gnu", "plot '" + test.getName() + ".fstat' u 8:($10*$14) w lp");
    logger(INFO, "finished tangential loading test: run 'gnuplot %.gnu' to view output", test.getName());
}
 
void objectivenessTest(const ParticleSpecies* species, Mdouble displacement, Mdouble tangentialDisplacement,
                                Mdouble velocity, Mdouble radius, std::string name)
{
    class ObjectivenessTest : public DPMBase
    {
        const ParticleSpecies* species;
        Mdouble displacement;
        Mdouble tangentialDisplacement;
        Mdouble velocity;
        Mdouble radius;
    public:
        //public variables
        ObjectivenessTest(const ParticleSpecies* species, Mdouble displacement, Mdouble tangentialDisplacement,
                          Mdouble velocity, Mdouble radius)
                : species(species), displacement(displacement), tangentialDisplacement(tangentialDisplacement),
                  velocity(velocity), radius(radius)
        {}
 
        void setupInitialConditions() override
        {
            //setName("ObjectivenessTest"+species->getName());
            setTimeMax((tangentialDisplacement + 0.5 * constants::pi * radius) / velocity);
            setTimeStep(1e-4 * getTimeMax());
            setSaveCount(20);
            setFileType(FileType::NO_FILE);
            dataFile.setFileType(FileType::ONE_FILE);
            fStatFile.setFileType(FileType::ONE_FILE);
 
            setMax(radius * Vec3D(2, 2, 2));
            setMin(radius * Vec3D(-2, -2, -2));
            setSystemDimensions(2);
            setParticleDimensions(3);
 
            speciesHandler.copyAndAddObject(*species);
    
            SphericalParticle p;
            p.setSpecies(speciesHandler.getObject(0));
            p.setRadius(radius);
            p.setPosition({0, radius - displacement, 0});
            particleHandler.copyAndAddObject(p);
            p.setPosition({0, -radius + displacement, 0});
            particleHandler.copyAndAddObject(p);
        }
 
        void actionsBeforeTimeStep() override
        {
            BaseParticle* p = particleHandler.getObject(0);
            BaseParticle* q = particleHandler.getLastObject();
            logger.assert_debug(p,"Empty particle handler");
            logger.assert_debug(q,"Empty particle handler");
 
            //Moving particle normally into surface
            if (getTime() <= tangentialDisplacement / velocity)
            {
                p->setAngularVelocity({0, 0, 0});
                p->setVelocity({velocity, 0, 0});
                p->setPosition({-tangentialDisplacement + velocity * getTime(), radius - displacement, 0});
                q->setAngularVelocity({0, 0, 0});
                q->setVelocity({-velocity, 0, 0});
                q->setPosition({tangentialDisplacement - velocity * getTime(), -radius + displacement, 0});
            }
            else
            {
                Mdouble angle = velocity / (radius - displacement) * (getTime() - tangentialDisplacement / velocity);
                Mdouble s = sin(angle);
                Mdouble c = cos(angle);
                p->setAngularVelocity(velocity / (radius - displacement) * Vec3D(0, 0, -1));
                //p->setAngularVelocity(Vec3D(0,0,0));
                p->setOrientation({1, 0, 0, 0});
                p->setVelocity(velocity * Vec3D(c, -s, 0));
                //p->setVelocity(Vec3D(0,0,0));
                p->setPosition((radius - displacement) * Vec3D(s, c, 0));
                q->setAngularVelocity(-p->getAngularVelocity());
                q->setOrientation(-p->getOrientation());
                q->setVelocity(-p->getVelocity());
                q->setPosition(-p->getPosition());
            }
        }
 
    } test(species, displacement, tangentialDisplacement, velocity, radius);
    test.setName(name);
    test.solve();
    helpers::writeToFile(test.getName() + ".gnu", "set size ratio -1; plot '" + test.getName() + ".fstat' u 14:15 every 2 w lp");
    logger(INFO, "finished objectiveness test: run 'gnuplot %.gnu' to view output", test.getName());
}
 
#endif // CONTACT_LAW_TEST_HELPERS