coef_i.cpp

/*
    Copyright 2017 Philippe Grandclement
 
    This file is part of Kadath.
 
    Kadath is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.
 
    Kadath is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
 
    You should have received a copy of the GNU General Public License
    along with Kadath.  If not, see <http://www.gnu.org/licenses/>.
*/
 
#include "base_spectral.hpp"
#include "array.hpp"
#include "array.hpp"
#include "array_math.hpp"
 
namespace Kadath {
void coef_i_1d (int, Array<double>&) ;
 
void Base_spectral::coef_i_dim (int dim, int nbr_conf, Array<double> *& inout) const {
    
    Dim_array res_out(inout->get_dimensions()) ;
    res_out.set(dim) = nbr_conf ;
    Array<double> res (res_out) ;
 
    int after = 1 ;
    for (int i=0 ; i<dim ; i++)
        after *= inout->get_size(i) ;
 
    int before = 1 ;
    for (int i=dim+1 ; i<ndim ; i++)
        before *= inout->get_size(i) ;
 
    int nbr_coef = inout->get_size(dim) ;
    int nbr = (nbr_coef > nbr_conf) ? nbr_coef : nbr_conf ;
 
    Array_iterator index_base (bases_1d[dim]->get_dimensions()) ;
    
    Array_iterator demarre_coef (inout->get_dimensions()) ;
    Array_iterator demarre_conf (res_out) ;
 
    Array_iterator loop_before_coef (inout->get_dimensions()) ;
    Array_iterator loop_before_out (res_out) ;
 
    Array_iterator lit_in (inout->get_dimensions()) ;
    Array_iterator put_out (res_out) ;
    
    Array<double> tab_1d (nbr) ;
    
    // Loop on dimensions before 
    for (int i=0 ; i<before ; i++) {
        
         demarre_coef.set(loop_before_coef) ;
         demarre_conf.set(loop_before_out) ;
        // On get la base
        
        int base = (*bases_1d[dim])(index_base) ;
        // Loop on dimensions after :
        for (int j=0 ; j<after ; j++) {
        
            lit_in = demarre_coef ;
            for (int k=0 ; k<nbr_coef ; k++) {
                tab_1d.set(k) = (*inout)(lit_in) ;
                lit_in.inc(after) ;
            }
            
            // Transformation
            coef_i_1d(base, tab_1d) ;
 
            // On range :
            put_out.set(demarre_conf) ;
            for (int k=0 ; k<nbr_conf ; k++) {
                res.set(put_out) = tab_1d(k) ;
                put_out.inc(after) ;
            }
            demarre_conf.inc() ;
            demarre_coef.inc() ;
        }
        index_base.inc() ;
        loop_before_coef.inc1( dim+1) ;
        loop_before_out.inc1( dim+1) ;
    }
    
    delete inout ;
    inout = new Array<double>(res) ;
}
 
Array<double> Base_spectral::coef_i (const Dim_array& in_conf, const Array<double>& cof) const {
    // Recopie
    Array<double>* conf = new Array<double> (cof) ;
    for (int d=0 ; d<ndim ; d++)
          coef_i_dim(d, in_conf(d), conf) ;
    Array<double> res (*conf) ;
    delete conf ;
    return res ;
}
 
}