Initial.h

Go to the documentation of this file.

#pragma once
 
 
#include "common/ForwardDecl.h"
#include "objects/containers/ArrayView.h"
#include "objects/geometry/Vector.h"
#include "objects/wrappers/Function.h"
#include "quantities/IMaterial.h"
 
NAMESPACE_SPH_BEGIN
 
class BodyView {
private:
    RawPtr<Storage> storage;
 
    Size bodyIndex;
 
public:
    BodyView(Storage& storage, const Size bodyIndex);
 
    BodyView& displace(const Vector& dr);
 
    BodyView& addVelocity(const Vector& v);
 
    enum class RotationOrigin {
        FRAME_ORIGIN,   
        CENTER_OF_MASS, 
    };
 
    BodyView& addRotation(const Vector& omega, const RotationOrigin origin);
 
    BodyView& addRotation(const Vector& omega, const Vector& origin);
 
private:
    Vector getOrigin(const RotationOrigin origin) const;
};
 
struct PowerLawSfd {
    Float exponent;
 
    Interval interval;
 
    INLINE Float operator()(const Float x) const {
        SPH_ASSERT(x >= 0._f && x <= 1._f);
        SPH_ASSERT(exponent != 1._f);
        const Float Rmin = pow(interval.lower(), 1._f - exponent);
        const Float Rmax = pow(interval.upper(), 1._f - exponent);
        const Float R = pow((Rmax - Rmin) * x + Rmin, 1._f / (1._f - exponent));
        SPH_ASSERT(R >= interval.lower() && R <= interval.upper(), R);
        return R;
    }
};
 
 
class InitialConditions : public Noncopyable {
private:
    MaterialInitialContext context;
 
    Size bodyIndex = 0;
 
public:
    explicit InitialConditions(const RunSettings& settings);
 
    ~InitialConditions();
 
    BodyView addMonolithicBody(Storage& storage, const BodySettings& body);
 
    BodyView addMonolithicBody(Storage& storage, const IDomain& domain, const BodySettings& body);
 
    BodyView addMonolithicBody(Storage& storage, const IDomain& domain, SharedPtr<IMaterial> material);
 
    BodyView addMonolithicBody(Storage& storage,
        const IDomain& domain,
        SharedPtr<IMaterial> material,
        const IDistribution& distribution);
 
 
    struct BodySetup {
        SharedPtr<IDomain> domain;
        SharedPtr<IMaterial> material;
 
        BodySetup();
 
        BodySetup(SharedPtr<IDomain> domain, SharedPtr<IMaterial> material);
 
        BodySetup(SharedPtr<IDomain> domain, const BodySettings& body);
 
        ~BodySetup();
    };
 
    BodyView addHeterogeneousBody(Storage& storage,
        const BodySetup& environment,
        ArrayView<const BodySetup> bodies);
 
    void addRubblePileBody(Storage& storage,
        const IDomain& domain,
        const PowerLawSfd& sfd,
        const BodySettings& bodySettings);
 
private:
    void setQuantities(Storage& storage, IMaterial& material, const Float volume);
};
 
void repelParticles(ArrayView<Vector> r, const Float radius);
 
Vector moveToCenterOfMassSystem(ArrayView<const Float> m, ArrayView<Vector> r);
 
void moveToCenterOfMassSystem(Storage& storage);
 
NAMESPACE_SPH_END