e42cc287de
ENH: define addition/subtraction operations for scalar and complex - required since construct complex from scalar is explicit - additional tests in Test-complex
287 lines
7.5 KiB
C
287 lines
7.5 KiB
C
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
|
|
\\ / O peration |
|
|
\\ / A nd | Copyright (C) 2019 OpenCFD Ltd.
|
|
\\/ M anipulation |
|
|
-------------------------------------------------------------------------------
|
|
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/>.
|
|
|
|
Application
|
|
|
|
Description
|
|
Tests for complex numbers
|
|
|
|
\*---------------------------------------------------------------------------*/
|
|
|
|
#include "argList.H"
|
|
#include "complex.H"
|
|
#include "complexFields.H"
|
|
#include "ops.H"
|
|
#include "ListOps.H"
|
|
|
|
using namespace Foam;
|
|
|
|
void print1(const complex& z)
|
|
{
|
|
Info<<"r: " << z.real() << " i: " << z.imag() << nl;
|
|
}
|
|
|
|
|
|
// * * * * * * * * * * * * * * * Main Program * * * * * * * * * * * * * * * //
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
Info<< "complex() : " << complex() << nl
|
|
<< "complex(zero) : " << complex(Zero) << nl
|
|
<< "pTraits<complex>::zero : " << pTraits<complex>::zero << nl
|
|
<< "pTraits<complex>::one : " << pTraits<complex>::one << nl
|
|
<< "complex(scalar) : " << complex(3.14519) << nl
|
|
<< nl;
|
|
|
|
std::complex<scalar> c1(10, -3);
|
|
Info<< "std::complex : " << c1 << nl;
|
|
Info<< "sin: " << std::sin(c1) << nl;
|
|
|
|
Info<< "complexVector::zero : " << complexVector::zero << nl
|
|
<< "complexVector::one : " << complexVector::one << nl
|
|
<< nl;
|
|
|
|
for (complex c : { complex{1, 0}, complex{1, 2}} )
|
|
{
|
|
Info<< nl;
|
|
print1(c);
|
|
|
|
Info<< "sin: " << sin(c) << nl;
|
|
Info<< "pow(3): " << pow(c, 3) << nl;
|
|
Info<< "pow3: " << pow3(c) << nl;
|
|
Info<< "log: " << log(c) << nl;
|
|
Info<< "pow025: " << pow025(c) << nl;
|
|
|
|
// TDB: allow implicit construct from scalar?
|
|
//
|
|
// if (c == 1.0)
|
|
// {
|
|
// Info<< c << " == " << 1 << nl;
|
|
// }
|
|
}
|
|
|
|
complexField fld1(3, complex(2.0, 1.0));
|
|
complexField fld2(fld1);
|
|
|
|
for (complex& c : fld2)
|
|
{
|
|
c = ~c;
|
|
}
|
|
|
|
Info<< "Field " << flatOutput(fld1) << nl;
|
|
Info<< "Conjugate: " << flatOutput(fld2) << nl;
|
|
|
|
// Some arbitrary change
|
|
for (complex& c : fld2)
|
|
{
|
|
c.Im() *= 5;
|
|
}
|
|
|
|
Info<< "sumProd: " << sumProd(fld1, fld2) << nl;
|
|
|
|
fld1 *= 10;
|
|
Info<< "scalar multiply: " << flatOutput(fld1) << nl;
|
|
|
|
fld1 /= 10;
|
|
Info<< "scalar divide: " << flatOutput(fld1) << nl;
|
|
|
|
Info<< "sin: " << sin(fld1) << nl;
|
|
|
|
Info<< "operator + : " << (fld1 + fld2) << nl;
|
|
|
|
// Some operators are still incomplete
|
|
|
|
// Info<< "operator * : " << (fld1 * fld2) << nl;
|
|
// Info<< "operator / : " << (fld1 / fld2) << nl;
|
|
// Info<< "operator / : " << (fld1 / 2) << nl;
|
|
// Info<< "operator / : " << (fld1 / fld2) << nl;
|
|
// Info<< "sqrt : " << sqrt(fld1) << nl;
|
|
// Info<< "pow(2) : " << pow(fld1, 2) << nl;
|
|
|
|
#if 1
|
|
Info<< nl << "## Elementary complex-complex arithmetic operations:" << nl;
|
|
{
|
|
const complex a(6, 1);
|
|
complex b = a;
|
|
|
|
Info << "# Compound assignment operations:" << nl;
|
|
|
|
Info<< "a = " << a << ", b = " << b << nl;
|
|
|
|
// Addition
|
|
b += a;
|
|
Info<< "b += a:" << tab << "b =" << b << nl;
|
|
|
|
// Subtraction
|
|
b -= a;
|
|
Info<< "b -= a:" << tab << "b =" << b << nl;
|
|
|
|
// Multiplication
|
|
b *= a;
|
|
Info<< "b *= a:" << tab << "b =" << b << nl;
|
|
|
|
// Division
|
|
b /= a;
|
|
Info<< "b /= a:" << tab << "b =" << b << nl;
|
|
}
|
|
#endif
|
|
|
|
|
|
#if 1
|
|
Info<< nl << "## Elementary complex-scalar arithmetic operations:" << nl;
|
|
{
|
|
const scalar a = 5;
|
|
complex b(6, 1);
|
|
|
|
Info << "# Non-assignment operations:" << nl;
|
|
|
|
Info<< "(scalar) a = " << a << ", b = " << b << nl;
|
|
|
|
// Addition
|
|
b = a + b;
|
|
Info<< "b = a + b: " << tab << b << nl;
|
|
|
|
b = b + a;
|
|
Info<< "b = b + a: " << tab << b << nl;
|
|
|
|
// Subtraction
|
|
b = a - b;
|
|
Info<< "b = a - b: " << tab << b << nl;
|
|
|
|
b = b - a;
|
|
Info<< "b = b - a: " << tab << b << nl;
|
|
|
|
// Multiplication
|
|
b = a*b;
|
|
Info<< "b = a*b: " << tab << b << nl;
|
|
|
|
b = b*a;
|
|
Info<< "b = b*a: " << tab << b << nl;
|
|
|
|
// Division
|
|
b = a/b;
|
|
Info<< "b = a/b = scalar(a)/b = complex(a)/b:" << tab << b << nl;
|
|
|
|
b = b/a;
|
|
Info<< "b = b/a: " << tab << b << nl;
|
|
|
|
|
|
Info << "# Compound assignment operations:" << nl;
|
|
|
|
Info<< "(scalar) a = " << a << ", b = " << b << nl;
|
|
|
|
// Addition: complex+scalar
|
|
b += a;
|
|
Info<< "b += a (only real part):" << tab << b << nl;
|
|
|
|
// Subtraction: complex-scalar
|
|
b -= a;
|
|
Info<< "b -= a (only real part):" << tab << b << nl;
|
|
|
|
// Multiplication: complex*scalar
|
|
b *= a;
|
|
Info<< "b *= a (real and imag parts):" << tab << b << nl;
|
|
|
|
// Division: complex/scalar
|
|
b /= a;
|
|
Info<< "b /= a (real and imag parts):" << tab << b << nl;
|
|
|
|
}
|
|
#endif
|
|
|
|
|
|
#if 1
|
|
Info<< nl << "## Other mathematical expressions:" << nl;
|
|
{
|
|
const complex a(4.3, -3.14);
|
|
const complex b(0, -4.3);
|
|
const complex c(-4.3, 0);
|
|
|
|
Info<< "a = " << a << ", b = " << b << ", c = " << c << nl;
|
|
|
|
// Square-root
|
|
Info<< "sqrt(a) = " << Foam::sqrt(a) << ", "
|
|
<< "sqrt(b) = " << Foam::sqrt(b) << ", "
|
|
<< "sqrt(c) = " << Foam::sqrt(c) << nl;
|
|
|
|
// Square
|
|
Info<< "sqr(a) = " << sqr(a) << ", "
|
|
<< "sqr(b) = " << sqr(b) << ", "
|
|
<< "sqr(c) = " << sqr(c) << nl;
|
|
|
|
// n^th power
|
|
Info<< "pow(a, -1) = " << pow(a, -1) << ", "
|
|
<< "pow(b, -1) = " << pow(b, -1) << ", "
|
|
<< "pow(c, -1) = " << pow(c, -1) << nl;
|
|
|
|
// Exponential
|
|
Info<< "exp(a) = " << exp(a) << ", "
|
|
<< "exp(b) = " << exp(b) << ", "
|
|
<< "exp(c) = " << exp(c) << nl;
|
|
|
|
// Natural logarithm
|
|
Info<< "log(a) = " << log(a) << ", "
|
|
<< "log(b) = " << log(b) << ", "
|
|
<< "log(c) = " << log(c) << nl;
|
|
|
|
}
|
|
#endif
|
|
|
|
|
|
// Make some changes
|
|
{
|
|
label i = 1;
|
|
for (complex& c : fld1)
|
|
{
|
|
c.Re() += i;
|
|
c.Im() -= 10 - i;
|
|
++i;
|
|
}
|
|
}
|
|
|
|
Info<< nl
|
|
<< "field = " << fld1 << nl;
|
|
|
|
Info<< "magSqr = "
|
|
<< ListOps::create<scalar>
|
|
(
|
|
fld1,
|
|
[](const complex& c) { return magSqr(c); }
|
|
)
|
|
<< nl;
|
|
|
|
Info
|
|
<< "sum = " << sum(fld1) << nl
|
|
<< "min = " << min(fld1) << nl
|
|
<< "max = " << max(fld1) << nl;
|
|
|
|
// MinMax fails since there is no less comparison operator
|
|
// Info<< "min/max = " << MinMax<complex>(fld1) << nl;
|
|
|
|
Info<< "\nEnd\n" << endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
// ************************************************************************* //
|