PSD.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 MECURYDPM_PSD_H
#define MECURYDPM_PSD_H
 
#include <fstream>
#include <Logger.h>
#include <vector>
#include <Tools/csvReader.h>
#include <random>
#include <utility>
#include <algorithm>
#include "iostream"
#include "ExtendedMath.h"
#include "RNG.h"
#include "Math/DistributionElements.h"
#include "Helpers/Helpers.h"
 
using mathsFunc::square;
using mathsFunc::cubic;
 
class PSD
{
 
public:
    enum class TYPE
    {
        CUMULATIVE_NUMBER_DISTRIBUTION,
        CUMULATIVE_LENGTH_DISTRIBUTION,
        CUMULATIVE_VOLUME_DISTRIBUTION,
        CUMULATIVE_AREA_DISTRIBUTION,
        PROBABILITYDENSITY_NUMBER_DISTRIBUTION,
        PROBABILITYDENSITY_LENGTH_DISTRIBUTION,
        PROBABILITYDENSITY_AREA_DISTRIBUTION,
        PROBABILITYDENSITY_VOLUME_DISTRIBUTION
    };
    
    PSD();
    
    PSD(std::unique_ptr<PSD> psdPtr) : particleSizeDistribution_(psdPtr->particleSizeDistribution_),
                                       nParticlesPerClass_(psdPtr->nParticlesPerClass_),
                                       volumePerClass_(psdPtr->volumePerClass_),
                                       index_(psdPtr->index_),
                                       random_(psdPtr->random_) {}
 
    PSD(const PSD& other);
 
    ~PSD();
 
    PSD* copy() const;
 
    void printPSD() const;
 
    Mdouble drawSample();
 
    Mdouble insertManuallyByVolume(Mdouble volume);
 
    void validateCumulativeDistribution();
 
    void validateProbabilityDensityDistribution();
 
    void setPSDFromVector(std::vector<DistributionElements> psd, TYPE PSDType);
 
    void setPSDFromCSV(const std::string& fileName, TYPE PSDType, bool headings = false, Mdouble
    unitScalingFactorRadii = 1.0);
 
    void setDistributionUniform(Mdouble radMin, Mdouble radMax, int numberOfBins);
 
    void setDistributionNormal(Mdouble mean, Mdouble standardDeviation, int numberOfBins);
    
    static PSD getDistributionNormal(Mdouble mean, Mdouble standardDeviation, int numberOfBins) {
        PSD psd;
        psd.setDistributionNormal(mean, standardDeviation, numberOfBins);
        return psd;
    }
 
    void setDistributionLogNormal(Mdouble mean, Mdouble standardDeviation, int numberOfBins);
 
    static PSD getDistributionLogNormal(Mdouble mean, Mdouble standardDeviation, int numberOfBins) {
        PSD psd;
        psd.setDistributionLogNormal(mean, standardDeviation, numberOfBins);
        return psd;
    }
    
    void setDistributionPhiNormal(Mdouble D50, Mdouble standardDeviationinPhi, int numberOfBins);
    
    void convertProbabilityDensityToCumulative();
 
    void convertCumulativeToProbabilityDensity();
 
    void convertProbabilityDensityToProbabilityDensityNumberDistribution(TYPE PDFType);
 
    void convertProbabilityDensityNumberDistributionToProbabilityDensity(TYPE PDFType);
 
    MERCURYDPM_DEPRECATED
    void convertProbabilityDensityNumberDistributionToProbabilityDensityVolumeDistribution();
 
    void convertCumulativeToCumulativeNumberDistribution(TYPE CDFType);
 
    void convertCumulativeNumberDistributionToCumulative(TYPE CDFType);
 
    MERCURYDPM_DEPRECATED
    void cutoffCumulativeNumber(Mdouble quantileMin, Mdouble quantileMax, Mdouble minPolydispersity = 0.1)
     {
         cutoffCumulativeDistribution(TYPE::CUMULATIVE_NUMBER_DISTRIBUTION, quantileMin, quantileMax, minPolydispersity);
     }
 
    void cutoffCumulativeDistribution(TYPE CDFType, Mdouble quantileMin, Mdouble quantileMax, Mdouble minPolydispersity = 0.1);
 
    MERCURYDPM_DEPRECATED
    void cutoffAndSqueezeCumulative(Mdouble quantileMin, Mdouble quantileMax, Mdouble squeeze,
                                    Mdouble minPolydispersity = 0.1)
    {
        cutoffAndSqueezeCumulativeDistribution(TYPE::CUMULATIVE_NUMBER_DISTRIBUTION, quantileMin, quantileMax, squeeze, minPolydispersity);
    }
 
    void cutoffAndSqueezeCumulativeDistribution(TYPE CDFType, Mdouble quantileMin, Mdouble quantileMax, Mdouble squeeze, Mdouble minPolydispersity = 0.1);
 
    Mdouble getMinRadius() const;
 
    Mdouble getMaxRadius() const;
    
    std::vector<DistributionElements> getParticleSizeDistribution() const;
 
    std::vector<DistributionElements> getParticleSizeDistributionByType(TYPE psdType, Mdouble scalingFactor = 1.0) const;
    
    MERCURYDPM_DEPRECATED
    void setParticleSizeDistribution(std::vector<DistributionElements>);
 
    void scaleParticleSize(double scale);
 
    double scaleParticleSizeAuto(int numberOfParticles, double targetVolume, bool allowScaleDown = false);
 
    int getInsertedParticleNumber() const;
    
    void decrementNParticlesPerClass();
    
    void decrementVolumePerClass(Mdouble volume);
 
    Mdouble getRadius(int index) const;
    
    Mdouble getNumberDx(Mdouble x) const;
 
    Mdouble getLengthDx(Mdouble x) const;
 
    Mdouble getAreaDx(Mdouble x) const;
    
    Mdouble getVolumeDx(Mdouble x) const;
 
    Mdouble getRadiusByQuantile(Mdouble quantile) const;
 
    Mdouble getQuantileByRadius(Mdouble radius) const;
 
    Mdouble getVolumetricMeanRadius() const;
    
    Mdouble getSizeRatio() const;
    
    void cutHighSizeRatio();
 
//    /*!
//     * \brief compute raw momenta of the user defined PSD.
//     */
//    void computeRawMomenta();
//
//    /*!
//     * \brief compute central momenta of the user defined PSD.
//     */
//    void computeCentralMomenta();
//
//    /*!
//     * \brief compute standardised momenta of the user defined PSD.
//     */
//    void computeStandardisedMomenta();
//
//    /*!
//     * \brief get momenta of the user defined PSD.
//     */
//    std::array<Mdouble, 6> getMomenta() const;
    
 
    std::vector< std::pair <Mdouble,Mdouble> > convertPdfPhiToPdfMeter(Mdouble meaninPhi, Mdouble standardDeviationinPhi, int numberOfBins);
 
    Mdouble computeD50(std::vector< std::pair <Mdouble,Mdouble> > vectorOfPair);
    
    static std::vector<Mdouble> linspace(Mdouble Min, Mdouble Max, int numberOfBins);
    
    void setFixedSeed(int seed)
    {
//        random_.setLinearCongruentialGeneratorParmeters(0,0,1);
        random_.setRandomSeed(seed);
    }
    
    friend bool operator<(const DistributionElements& l, const DistributionElements& r);
    
    friend bool operator<(const DistributionElements& l, Mdouble r);
    
    friend std::ostream& operator<<(std::ostream& os, DistributionElements& p);
    
    friend std::istream& operator>>(std::istream& is, DistributionElements& p);
    
    friend Mdouble operator==(DistributionElements l, Mdouble r);
 
private:
    std::vector<DistributionElements> particleSizeDistribution_;
 
    std::vector<int> nParticlesPerClass_;
    
    std::vector<Mdouble> volumePerClass_;
    
    int index_;
    
    RNG random_;
};
 
 
#endif //MECURYDPM_PSD_H