ParticleHandler.h

Go to the documentation of this file.

/*
 * ParticleHandler.h
 *
 *  Created on: 10 Mar 2014
 *      Author: ttitsche
 */

#pragma once


#include <vector>
#include <random>
#include <Eigen/Core>

namespace NuTo
{

class CollidableParticleBase;
class CollidableParticleSphere;
class VisualizeUnstructuredGrid;
class Specimen;
typedef std::vector<CollidableParticleSphere*> ParticleContainer;

class ParticleHandler
{
public:
    ParticleHandler(int rNumParticles, Eigen::MatrixXd rParticleBoundingBox, double rVelocityRange, double rGrowthRate,
                    int rSeed = 0);

    ParticleHandler(Eigen::MatrixXd rSpheres, double rVelocityRange, double rRelativeGrowthRate,
                    double rAbsoluteGrowthRate, int rSeed = 0);

    ParticleHandler(const std::string& rFileName, double rVelocityRange, double rRelativeGrowthRate,
                    double rAbsoluteGrowthRate, int rSeed = 0);


    ~ParticleHandler();

    void Sync(const double rTime);

    double GetKineticEnergy() const;

    double GetVolume() const;

    void ExportParticlesToVTU3D(std::string rOutputDirectory, int rTimeStep, double rGlobalTime, bool rFinal) const;

    void ExportParticlesToVTU2D(std::string rOutputFile, double rZCoord) const;

    void ExportParticlesToGmsh3D(std::string rOutputFile, Specimen& rSpecimen, double rMeshSize) const;

    void ExportParticlesToGmsh2D(std::string rOutputFile, Specimen& rSpecimen, double rMeshSize, double rZCoord,
                                 double rMinRadius) const;


    void ResetVelocities();

    Eigen::MatrixXd GetParticles(bool rInitialRadius = false) const;

    Eigen::MatrixXd GetParticles2D(double rZCoord, double rMinRadius) const;

    void ExportParticlesToFile(const std::string& rExportFileName, bool rInitialRadius) const;

    CollidableParticleSphere* GetParticle(const int rIndex) const;

    int GetNumParticles() const;

    double GetAbsoluteMininimalDistance(Specimen& rSpecimen);

    void SetVisualizationFileName(const std::string& rVisualizationFileName);

    Eigen::VectorXi GetSubBoxDivisions(Specimen& rSpecimen, const int rParticlesPerBox);

private:
    ParticleContainer mParticles;

    Eigen::VectorXd GetRandomVector(double rStart, double rEnd);

    Eigen::VectorXd GetRandomVector(const Eigen::MatrixXd& rBounds);

    void CreateParticlesFromMatrix(const Eigen::MatrixXd& rSpheres, double rVelocityRange, double rRelativeGrowthRate,
                                   double rAbsoluteGrowthRate);

    std::string mVisualizationFileName;

    std::mt19937 mRNG;
};


} /* namespace NuTo */