invariants.hh

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
/*                                                                             */
/*   This file is part of the C++-library RFFGen.                              */
/*   Copyright 2015 Lars Lubkoll                                               */
/*                                                                             */
/*   RFFGen 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.                                       */
/*                                                                             */
/*   RFFGen 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 RFFGen.  If not, see <http://www.gnu.org/licenses/>.           */
/*                                                                             */
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#ifndef RFFGEN_LINEAR_ALGEBRA_INVARIANTS_HH
#define RFFGEN_LINEAR_ALGEBRA_INVARIANTS_HH

#include "mixedInvariants.hh"
#include "modifiedMixedInvariants.hh"
#include "modifiedPrincipalInvariants.hh"
#include "principalInvariants.hh"

namespace RFFGen
{
  namespace LinearAlgebra
  {
    enum class Invariant { PRINCIPAL, MIXED, MODIFIED, MODIFIED_MIXED };

    template <Invariant> struct InvariantTraits;
    template <>
    struct InvariantTraits<Invariant::PRINCIPAL>
    {
      template <class Matrix>
      using FirstInvariant          = FirstPrincipalInvariant<Matrix>;

      template <class Matrix>
      using SecondInvariant         = SecondPrincipalInvariant<Matrix>;

      template <class Matrix>
      using ThirdInvariant          = ThirdPrincipalInvariant<Matrix>;


      template < class Matrix , int offset = LinearAlgebra::dimension<Matrix>() >
      using ShiftedFirstInvariant   = ShiftedFirstPrincipalInvariant<Matrix,offset>;

      template < class Matrix , int offset = LinearAlgebra::dimension<Matrix>() >
      using ShiftedSecondInvariant  = ShiftedSecondPrincipalInvariant<Matrix,offset>;

      template <class Matrix>
      using ShiftedThirdInvariant   = ShiftedThirdPrincipalInvariant<Matrix>;
    };

    template <>
    struct InvariantTraits<Invariant::MODIFIED>
    {
      template <class Matrix>
      using FirstInvariant          = FirstModifiedPrincipalInvariant<Matrix>;

      template <class Matrix>
      using SecondInvariant         = SecondModifiedPrincipalInvariant<Matrix>;

      template <class Matrix>
      using ThirdInvariant          = ThirdModifiedPrincipalInvariant<Matrix>;


      template < class Matrix , int offset = LinearAlgebra::dimension<Matrix>() >
      using ShiftedFirstInvariant   = ShiftedFirstModifiedPrincipalInvariant<Matrix,offset>;

      template < class Matrix , int offset = LinearAlgebra::dimension<Matrix>() >
      using ShiftedSecondInvariant  = ShiftedSecondModifiedPrincipalInvariant<Matrix,offset>;

      template <class Matrix>
      using ShiftedThirdInvariant   = ShiftedThirdModifiedPrincipalInvariant<Matrix>;
    };

    template <>
    struct InvariantTraits<Invariant::MIXED>
    {
      template <class Matrix>
      using FirstInvariant          = FirstMixedInvariant<Matrix>;

      template <class Matrix>
      using SecondInvariant         = SecondMixedInvariant<Matrix>;

      template <class Matrix>
      using ThirdInvariant          = ThirdMixedInvariant<Matrix>;

      template <class Matrix>
      using ShiftedFirstInvariant   = ShiftedFirstMixedInvariant<Matrix>;

      template <class Matrix>
      using ShiftedSecondInvariant  = ShiftedSecondMixedInvariant<Matrix>;

      template <class Matrix>
      using ShiftedThirdInvariant   = ShiftedThirdMixedInvariant<Matrix>;
    };

    template <>
    struct InvariantTraits<Invariant::MODIFIED_MIXED>
    {
      template <class Matrix>
      using FirstInvariant          = FirstModifiedMixedInvariant<Matrix>;

      template <class Matrix>
      using SecondInvariant         = SecondModifiedMixedInvariant<Matrix>;

      template <class Matrix>
      using ThirdInvariant          = ThirdModifiedMixedInvariant<Matrix>;


      template <class Matrix>
      using ShiftedFirstInvariant   = ShiftedFirstModifiedMixedInvariant<Matrix>;

      template <class Matrix>
      using ShiftedSecondInvariant  = ShiftedSecondModifiedMixedInvariant<Matrix>;

      template <class Matrix>
      using ShiftedThirdInvariant   = ShiftedThirdModifiedMixedInvariant<Matrix>;
    };
  }
}

#endif // RFFGEN_LINEAR_ALGEBRA_INVARIANTS_HH