#include "stratified_sampling.hpp"
#include <algorithm>
#include <iostream>

std::pair<double, double> mean_var( std::vector<double> r){
    std::pair<double, double> p;
    for(auto &x: r){
        p.first += x;
        p.second += x*x;
    }
    p.first /= r.size();
    p.second /= r.size();
    p.second -= p.first * p.first;
    return p;
}

//actualisation du nombre de tirages à faire par strates
template <typename Gen>
void stratified_sampling<Gen>::update(int Nk) {
    int I = p.size();
    //reinitialistation du vecteur M du nombre de tirages par strates
    if (M.empty()) {
        M.resize(I,1);
    }
    else {
        for(int i=0; i<I; i++){
            M[i]=1;
        }
    }

    std::vector<double> m(I, 0); //le vecteur des m_i idéals

    if (sigma.empty()) {
        for (int i=0; i<I; i++) {
            m[i] = (Nk-I)*p[i];
        }
    }
    else {
        double scal = std::inner_product(p.begin(), p.end(), sigma.begin(), (double) 0);
        for (int i=0; i < I; i++) {
            m[i] = (Nk-I)*p[i]*sigma[i]/scal;
            std::cout<<m[i]<<std::endl;
        }
    }
    M[0]+=floor(m[0]);
    double current = m[0];
    for (int i=1; i<I; i++){
        M[i] += floor(current+m[i]) - floor(current);
        current += m[i];
    }
}