fn_BZ_linspace.hpp

Go to the documentation of this file.

/*
 * BELFEM -- The Berkeley Lab Finite Element Framework
 * Copyright (c) 2026, The Regents of the University of California,
 * through Lawrence Berkeley National Laboratory (subject to receipt of any required
 * approvals from the U.S. Dept. of Energy).  All rights reserved.
 *
 * Developers: Christian Messe, Gregory Giard
 *
 * See the top-level LICENSE file for the complete license and disclaimer.
 */
 
#ifndef BELFEM_FN_BZ_LINSPACE_HPP
#define BELFEM_FN_BZ_LINSPACE_HPP
 
#include "cl_BZ_Vector.hpp"
 
namespace belfem
{
//------------------------------------------------------------------------------
 
    template< typename T >
    void
    linspace( const T & aStart,
              const T & aEnd,
              const belfem::size_t & aN,
              Vector <T> & aValues )
    {
        // yes, this can be done a lot better using std::for_each
 
        aValues.set_size( aN );
 
        belfem::size_t tN = aN - 1;
 
        // stepsize
        T tDX = ( aEnd - aStart ) / ( aN - 1 );
 
        // first step
        aValues( 0 ) = aStart;
 
        // intermediate steps
        for ( uint k = 1; k < tN; ++k )
        {
            aValues( k ) = aValues( k - 1 ) + tDX;
        }
 
        // last step
        aValues( tN ) = aEnd;
    }
 
//------------------------------------------------------------------------------
 
    template< typename T >
    Vector <T>
    linspace( const T & aStart,
              const T & aEnd,
              const belfem::size_t & aN )
    {
        Vector <T> aValues;
        linspace( aStart, aEnd, aN, aValues );
        return aValues;
    }
 
//------------------------------------------------------------------------------
 
}
#endif //BELFEM_FN_BZ_LINSPACE_HPP