Files
openfoam/src/OpenFOAM/primitives/Tensor/Tensor.H
Mark Olesen 1c4e32fb6a ENH: add iterators to VectorSpace (#1265)
- this adds support for various STL operations including

    * sorting, filling, find min/max element etc.
    * for-range iteration

STYLE: use constexpr for VectorSpace rank
2019-04-04 17:14:27 +02:00

307 lines
8.7 KiB
C++

/*---------------------------------------------------------------------------*\
========= |
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
\\ / O peration |
\\ / A nd | Copyright (C) 2018 OpenCFD Ltd.
\\/ M anipulation |
-------------------------------------------------------------------------------
| Copyright (C) 2011-2016 OpenFOAM Foundation
-------------------------------------------------------------------------------
License
This file is part of OpenFOAM.
OpenFOAM 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.
OpenFOAM 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 OpenFOAM. If not, see <http://www.gnu.org/licenses/>.
Class
Foam::Tensor
Description
Templated 3D tensor derived from MatrixSpace adding construction from
9 components, element access using xx(), xy() etc. member functions and
the inner-product (dot-product) and outer-product of two Vectors
(tensor-product) operators.
SourceFiles
TensorI.H
See also
Foam::MatrixSpace
Foam::Vector
\*---------------------------------------------------------------------------*/
#ifndef Tensor_H
#define Tensor_H
#include "MatrixSpace.H"
#include "Vector.H"
#include "SphericalTensor.H"
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
namespace Foam
{
// Forward Declarations
template<class Cmpt> class SymmTensor;
/*---------------------------------------------------------------------------*\
Class Tensor Declaration
\*---------------------------------------------------------------------------*/
template<class Cmpt>
class Tensor
:
public MatrixSpace<Tensor<Cmpt>, Cmpt, 3, 3>
{
public:
//- Equivalent type of labels used for valid component indexing
typedef Tensor<label> labelType;
// Member constants
//- Rank of Tensor is 2
static constexpr direction rank = 2;
// Static data members
static const Tensor I;
//- Component labeling enumeration
enum components { XX, XY, XZ, YX, YY, YZ, ZX, ZY, ZZ };
// Constructors
//- Construct null
inline Tensor();
//- Construct initialized to zero
inline Tensor(const Foam::zero);
//- Construct given MatrixSpace of the same rank
template<class Cmpt2>
inline Tensor(const MatrixSpace<Tensor<Cmpt2>, Cmpt2, 3, 3>& vs);
//- Construct given VectorSpace of the same rank
template<class Cmpt2>
inline Tensor(const VectorSpace<Tensor<Cmpt2>, Cmpt2, 9>& vs);
//- Construct given SphericalTensor
inline Tensor(const SphericalTensor<Cmpt>& st);
//- Construct given SymmTensor
inline Tensor(const SymmTensor<Cmpt>& st);
//- Construct given triad of row vectors
inline Tensor(const Vector<Vector<Cmpt>>& tr);
//- Construct given the three row vectors
inline Tensor
(
const Vector<Cmpt>& x,
const Vector<Cmpt>& y,
const Vector<Cmpt>& z
);
//- Construct given the nine components
inline Tensor
(
const Cmpt txx, const Cmpt txy, const Cmpt txz,
const Cmpt tyx, const Cmpt tyy, const Cmpt tyz,
const Cmpt tzx, const Cmpt tzy, const Cmpt tzz
);
//- Construct from a block of another matrix space
template
<
template<class, direction, direction> class Block2,
direction BRowStart,
direction BColStart
>
Tensor
(
const Block2<Tensor<Cmpt>, BRowStart, BColStart>& block
);
//- Construct from Istream
inline Tensor(Istream& is);
// Member Functions
// Component access
inline const Cmpt& xx() const;
inline const Cmpt& xy() const;
inline const Cmpt& xz() const;
inline const Cmpt& yx() const;
inline const Cmpt& yy() const;
inline const Cmpt& yz() const;
inline const Cmpt& zx() const;
inline const Cmpt& zy() const;
inline const Cmpt& zz() const;
inline Cmpt& xx();
inline Cmpt& xy();
inline Cmpt& xz();
inline Cmpt& yx();
inline Cmpt& yy();
inline Cmpt& yz();
inline Cmpt& zx();
inline Cmpt& zy();
inline Cmpt& zz();
// Column-vector access.
//- Extract vector for column 0
inline Vector<Cmpt> cx() const;
//- Extract vector for column 1
inline Vector<Cmpt> cy() const;
//- Extract vector for column 2
inline Vector<Cmpt> cz() const;
//- Extract vector for given column.
// Compile-time check of column index.
template<direction Col>
inline Vector<Cmpt> col() const;
//- Extract vector for given column (0,1,2).
// Runtime check of column index.
inline Vector<Cmpt> col(const direction c) const;
//- Set values of given column
// Compile-time check of column index.
template<direction Col>
inline void col(const Vector<Cmpt>& v);
//- Set values of given column (0,1,2)
// Runtime check of column index.
inline void col(const direction c, const Vector<Cmpt>& v);
// Row-vector access.
//- Extract vector for row 0
inline Vector<Cmpt> x() const;
//- Extract vector for row 1
inline Vector<Cmpt> y() const;
//- Extract vector for row 2
inline Vector<Cmpt> z() const;
//- Extract vector for given row.
// Compile-time check of row index.
template<direction Row>
inline Vector<Cmpt> row() const;
//- Extract vector for given row (0,1,2)
// Runtime check of row index.
inline Vector<Cmpt> row(const direction r) const;
//- Set values of given row
// Compile-time check of row index.
template<direction Row>
inline void row(const Vector<Cmpt>& v);
//- Set values of given row (0,1,2)
// Runtime check of row index.
inline void row(const direction r, const Vector<Cmpt>& v);
//- Return vector for given row (0,1,2)
// Runtime check of row index.
inline Vector<Cmpt> vectorComponent(const direction cmpt) const;
// Tensor Operations
//- Return transpose
inline Tensor<Cmpt> T() const;
//- Return inverse
inline Tensor<Cmpt> inv() const;
//- Inner-product of this with another Tensor.
inline Tensor<Cmpt> inner(const Tensor<Cmpt>& t2) const;
//- Schur-product of this with another Tensor.
inline Tensor<Cmpt> schur(const Tensor<Cmpt>& t2) const;
// Member Operators
//- Assign inner-product of this with another Tensor.
inline void operator&=(const Tensor<Cmpt>& t);
//- Inherit MatrixSpace assignment operators
using Tensor::msType::operator=;
//- Assign to an equivalent vector space
template<class Cmpt2>
inline void operator=(const VectorSpace<Tensor<Cmpt2>, Cmpt2, 9>&);
//- Assign to a SphericalTensor
inline void operator=(const SphericalTensor<Cmpt>&);
//- Assign to a SymmTensor
inline void operator=(const SymmTensor<Cmpt>&);
//- Assign to a triad of row vectors
inline void operator=(const Vector<Vector<Cmpt>>&);
};
template<class Cmpt>
class typeOfRank<Cmpt, 2>
{
public:
typedef Tensor<Cmpt> type;
};
template<class Cmpt>
class typeOfTranspose<Cmpt, Tensor<Cmpt>>
{
public:
typedef Tensor<Cmpt> type;
};
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
} // End namespace Foam
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
// Include inline implementations
#include "TensorI.H"
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
#endif
// ************************************************************************* //