concentrationset.h

Go to the documentation of this file.

#ifndef CONCENTRATIONSET_H
#define CONCENTRATIONSET_H
 
#include <vector>
#include <string>
#include "Distribution.h"
#include "cmbtimeseries.h"
#include "cmbtimeseriesset.h"
 
using namespace std;
 
class ConcentrationSet : public vector<double>
{
public:
    // ========== Construction and Assignment ==========
 
    ConcentrationSet();
    ConcentrationSet(const ConcentrationSet& other);
    ConcentrationSet(int n);
    ConcentrationSet& operator=(const ConcentrationSet& other);
 
    // ========== Data Management ==========
 
    void AppendValue(double value);
 
    void AppendSet(const ConcentrationSet& other);
    void AppendSet(ConcentrationSet* other);
 
    // ========== Basic Statistics ==========
 
    double CalculateMean() const;
 
    double CalculateStdDev(double mean_value = -999) const;
 
    double CalculateMeanLog() const;
 
    double CalculateStdDevLog(double mean_value = -999) const;
 
    double CalculateSSE(double mean_value = -999) const;
 
    double CalculateSSELog(double mean_value = -999) const;
 
    double CalculateNorm(double power = 2.0) const;
 
    double CalculateNormLog(double power = 2.0) const;
 
    double GetMinimum() const;
 
    double GetMaximum() const;
 
    // ========== Distribution Fitting ==========
 
    vector<double> EstimateDistributionParameters(
        distribution_type dist_type = distribution_type::none
    );
 
    double CalculateLogLikelihood(
        const vector<double>& params = vector<double>(),
        distribution_type dist_type = distribution_type::none
    ) const;
 
    distribution_type SelectBestDistribution();
 
    Distribution* GetFittedDistribution() { return &fitted_distribution_; }
    const Distribution* GetFittedDistribution() const { return &fitted_distribution_; }
 
    Distribution* GetEstimatedDistribution() { return &estimated_distribution_; }
    const Distribution* GetEstimatedDistribution() const { return &estimated_distribution_; }
 
    double GetEstimatedMu() const { return estimated_distribution_.parameters.size() > 0 ? estimated_distribution_.parameters[0] : 0.0; }
    void SetEstimatedMu(double value) {
        if (estimated_distribution_.parameters.empty()) estimated_distribution_.parameters.resize(2);
        estimated_distribution_.parameters[0] = value;
    }
 
    double GetEstimatedSigma() const { return estimated_distribution_.parameters.size() > 1 ? estimated_distribution_.parameters[1] : 0.0; }
    void SetEstimatedSigma(double value) {
        if (estimated_distribution_.parameters.empty()) estimated_distribution_.parameters.resize(2);
        estimated_distribution_.parameters[1] = value;
    }
 
    // ========== Goodness-of-Fit Testing ==========
 
    CMBTimeSeries CreateDataCDF() const;
 
    CMBTimeSeriesSet CreateCDFComparison(distribution_type dist_type) const;
 
    CMBTimeSeriesSet CreateFittedDistribution(distribution_type dist_type) const;
 
    double CalculateKolmogorovSmirnovStatistic(distribution_type dist_type) const;
 
    // ========== Transformations ==========
 
    ConcentrationSet CreateLogTransformed() const;
 
    double CalculateBoxCoxLogLikelihood(double lambda) const;
 
    ConcentrationSet ApplyBoxCoxTransform(double lambda, bool normalize) const;
 
    double FindOptimalBoxCoxParameter(
        double min_lambda,
        double max_lambda,
        int n_intervals
    ) const;
 
    // ========== Utility Functions ==========
 
    vector<unsigned int> CalculateRanks() const;
 
private:
    Distribution fitted_distribution_;      
    Distribution estimated_distribution_;   
};
 
#endif // CONCENTRATIONSET_H