weights.h

Go to the documentation of this file.

#pragma once
 
#include <thrust/host_vector.h>
#include "grid.h"
#include "../functors.h"
#include "../blas1.h"
#include "../enums.h"
 
namespace dg{
namespace create{
template< class Topology, size_t N, size_t ...I>
auto do_weights( const Topology& g, std::array<bool, N> coo, std::index_sequence<I...>)
{
    std::array< decltype(g.weights(0)),N> weights;
    for( unsigned u=0; u<N; u++)
        weights[u] = g.weights(u);
    for( unsigned u=0; u<N; u++)
        if( !coo[u])
            dg::blas1::copy( 1, weights[u]);
    return dg::kronecker( dg::Product(), weights[I]...);
}
template< class Topology, size_t ...I>
auto do_weights( const Topology& g, std::index_sequence<I...>)
{
    return dg::kronecker( dg::Product(), g.weights(I)...);
}
 
 
 
 
template<class Topology>
auto weights( const Topology& g)
{
    return do_weights( g, std::make_index_sequence<Topology::ndim()>());
}
 
template<class Topology>
auto weights( const Topology& g, std::array<bool,Topology::ndim()> remains)
{
    return do_weights( g, remains, std::make_index_sequence<Topology::ndim()>());
}
 
 
template<class Topology>
auto inv_weights( const Topology& g)
{
    auto v = weights( g);
    dg::blas1::transform( v, v, dg::INVERT());
    return v;
}
 
 
}//namespace create
}//namespace dg