#include <iostream>
#include <gsl/gsl_rng.h>

#include "rtnorm.hpp"
#include <gsl/gsl_cdf.h>
#include <gsl/gsl_math.h>

//--génération quantiles--
std::vector<double> quantile_norm(int n, double sigma){
  std::vector<double> q(n);
  for (int i=0; i<n; i++) {
    q[i] = gsl_cdf_gaussian_Pinv ((double)(i+1)/n, sigma);
  }
  return q;
}

//--tirage de normale tronquée entre les quantile de taille 1/n i et i+1--
double quantile_truncate_normal (int i, int n, double mu, 
 double sigma, gsl_rng *gen) {
	 std::vector<double> q;
	 q = quantile_norm(n, sigma);
	 std::pair<double, double> p;
	 p = rtnorm (gen, q[i], q[i+1], mu, sigma);
	 return p.first;
 }

std::pair<double, double> mean_var( std::vector<double> r){
    std::pair<double, double> p;
    for(int i=0; i<r.size(); i++){
        p.first += r[i];
        p.second += r[i]*r[i];
    }
    p.first /= r.size();
    p.second /= r.size();
    p.second -= p.first * p.first;
    return p;
}

int main()
{
  //--- GSL random init ---
  gsl_rng_env_setup();                          // Read variable environnement
  const gsl_rng_type* type = gsl_rng_default;   // Default algorithm 'twister'
  gsl_rng *gen = gsl_rng_alloc (type);          // Rand generator allocation
  std::vector<double> q;
  q = quantile_norm(10, 1);
  std::cout<<q[5]<<std::endl;

  std::pair<double, double> p;
  std::pair<double, double> mv;
  //number of classes
  int I = 10;
  //number of samples
  int N = 10000;
  std::vector<double> r(N);
  double a;
  for (int i=0; i<I; i++){
      if(i==0){
          a = GSL_NEGINF;
      }else{
          a = q[i-1];
      }
      for(int j=0; j<N; j++){
          p = rtnorm (gen, a, q[i], 0, 1);
          r[j] = p.first;
      }
      mv = mean_var(r);
      std::cout<<"mean :"<<mv.first<<" var :"<<mv.second<<std::endl;
  }
  gsl_rng_free(gen);
  return 0;
}