PyramidLaw.hpp

/*
 * This file is part of bogus, a C++ sparse block matrix library.
 *
 * Copyright 2016 Gilles Daviet <gdaviet@gmail.com>
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/

#ifndef BOGUS_PYRAMIDLAW_HPP
#define BOGUS_PYRAMIDLAW_HPP

#include <cmath>

namespace bogus
{


template < DenseIndexType Dimension, typename Scalar, bool DeSaxceCOV >
class PyramidLaw
{
public:
    typedef LocalProblemTraits< Dimension, Scalar > Traits ;
    enum{ dimension = Dimension } ;


    PyramidLaw( const unsigned n, const double * mu ) ;

    Scalar eval( const unsigned problemIndex,
                 const typename Traits::Vector &x,
                 const typename Traits::Vector &y ) const ;

    bool solveLocal(
            const unsigned problemIndex,
            const typename Traits::Matrix &A,
            const typename Traits::Vector &b,
            typename Traits::Vector &x,
            const Scalar scaling
            ) const ;

    void projectOnConstraint( const unsigned problemIndex, typename Traits::Vector &x ) const ;


    template< typename Segment >
    void dualityCOV( const unsigned problemIndex, const Segment& y,
                     typename Traits::Vector& s ) const
    {
        if( DeSaxceCOV ) {
            Traits::np( s ) = m_mu[problemIndex] * Traits::tp(y).template lpNorm<1>() ;
            Traits::tp( s ).setZero() ;
        } else  s.setZero() ;
    }

private:

    const double * m_mu ;
    const unsigned m_n ;

} ;


}

#endif