unitSetIO.C

Go to the documentation of this file.

/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   \\    /   O peration     | Website:  https://openfoam.org
    \\  /    A nd           | Copyright (C) 2024-2026 OpenFOAM Foundation
     \\/     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/>.
 
\*---------------------------------------------------------------------------*/
 
#include "units.H"
#include "dictionary.H"
#include "NamedEnum.H"
#include "symbols.H"
 
// * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
 
Foam::unitSet::unitSet(Istream& is)
:
    dimensions_(dimless),
    multiplier_(NaN)
{
    is >> *this;
}
 
 
// * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 
void Foam::unitSet::read(const word& keyword, const dictionary& dict)
{
    const unitSet units(dict.lookup(keyword));
 
    if (!compare(*this, units, false))
    {
        multiplier_ = scalar(1);
 
        FatalIOErrorInFunction(dict)
            << "The units " << units.info() << " of " << keyword
            << " in dictionary " << dict.name() << " do not have "
            << "the required dimensions " << info()
            << abort(FatalIOError);
    }
 
    reset(units);
}
 
 
void Foam::unitSet::read(Istream& is)
{
    const unitSet units(is);
 
    if (!compare(*this, units, false))
    {
        multiplier_ = scalar(1);
 
        FatalIOErrorInFunction(is)
            << "The units " << units.info() << " provided do not have "
            << "the required dimensions " << info()
            << abort(FatalIOError);
    }
 
    reset(units);
}
 
 
void Foam::unitSet::read
(
    const word& keyword,
    const dictionary& dict,
    Istream& is
)
{
    const unitSet units(is);
 
    if (!compare(*this, units, false))
    {
        multiplier_ = scalar(1);
 
        FatalIOErrorInFunction(dict)
            << "The units " << units.info() << " of " << keyword
            << " in dictionary " << dict.name() << " do not have "
            << "the required dimensions " << info()
            << abort(FatalIOError);
    }
 
    reset(units);
}
 
 
bool Foam::unitSet::readIfPresent
(
    const word& keyword,
    const dictionary& dict
)
{
    const entry* entryPtr = dict.lookupEntryPtr(keyword, false, true);
 
    if (entryPtr)
    {
        const unitSet units(entryPtr->stream());
 
        if (!compare(*this, units, false))
        {
            multiplier_ = scalar(1);
 
            FatalIOErrorInFunction(dict)
                << "The units " << units.info() << " of " << keyword
                << " in dictionary " << dict.name() << " do not have "
                << "the required dimensions " << info()
                << abort(FatalIOError);
        }
 
        reset(units);
 
        return true;
    }
    else
    {
        return false;
    }
}
 
 
bool Foam::unitSet::readIfPresent(Istream& is)
{
    token nextToken(is);
    is.putBack(nextToken);
 
    if (nextToken != token::BEGIN_SQR) return false;
 
    const unitSet units(is);
 
    if (!unitSet::compare(units, *this, false))
    {
        multiplier_ = scalar(1);
 
        FatalIOErrorInFunction(is)
            << "The units " << units.info() << " provided do not have "
            << "the required dimensions " << info()
            << abort(FatalIOError);
    }
 
    if (debug && (any() || !unitSet::compare(units, *this, true)))
    {
        Info<< "Unit conversion at line " << is.lineNumber()
            << " of file " << is.name()
            << " with factor " << units.multiplier_ << endl;
    }
 
    reset(units);
 
    return true;
}
 
 
bool Foam::unitSet::readIfPresent
(
    const word& keyword,
    const dictionary& dict,
    Istream& is
)
{
    token nextToken(is);
    is.putBack(nextToken);
 
    if (nextToken != token::BEGIN_SQR) return false;
 
    const unitSet units(is);
 
    if (!unitSet::compare(units, *this, false))
    {
        multiplier_ = scalar(1);
 
        FatalIOErrorInFunction(dict)
            << "The units " << units.info() << " of " << keyword
            << " in dictionary " << dict.name() << " do not have "
            << "the required dimensions " << info()
            << abort(FatalIOError);
    }
 
    if (debug && (any() || !unitSet::compare(units, *this, true)))
    {
        Info<< "Unit conversion of " << keyword
            << " in dictionary " << dict.name()
            << " with factor " << units.multiplier_ << endl;
    }
 
    reset(units);
 
    return true;
}
 
 
// * * * * * * * * * * * * * * Friend Operators * * * * * * * * * * * * * * //
 
Foam::Istream& Foam::operator>>(Istream& is, unitSet& units)
{
    token nextToken;
 
    // Read the next delimiting token. This must be the start bracket.
    is >> nextToken;
    if (nextToken != token::BEGIN_SQR)
    {
        FatalIOErrorInFunction(is)
            << "expected a " << token::BEGIN_SQR << " in unitSet"
            << endl << "in stream " << is.info() << ", got a "
            << nextToken << exit(FatalIOError);
    }
 
    // Peek at the next token
    is >> nextToken;
    is.putBack(nextToken);
 
    // If not a number or separator, then these are named units. Parse.
    if (!nextToken.isNumber() && nextToken != token::COLON)
    {
        // Named units. Parse. Note: Use an explicit construction of the
        // identity unit instead of 'unitless' because this function may be
        // used in a static context before 'unitless' is available.
        units.reset
        (
            symbols::parseNoBeginOrEnd
            (
                is,
                unitSet(dimless, 0, 0, 1),
                Foam::units::table()
            )
        );
 
        // Read the next delimiting token. This must be the end bracket.
        is >> nextToken;
        if (nextToken != token::END_SQR)
        {
            FatalIOErrorInFunction(is)
                << "expected a " << token::END_SQR << " in unitSet "
                << endl << "in stream " << is.info() << ", got a "
                << nextToken << exit(FatalIOError);
        }
 
        // Check state of Istream
        is.check("Istream& operator>>(Istream&, unitSet&)");
 
        return is;
    }
 
    // Otherwise these are numbered units. Read directly...
 
    // Read the dimensions
    units.dimensions_.readNoBeginOrEnd(is);
 
    // Read the next delimiting token. If a separator, then there are
    // dimensionless units and a multiplier to read. If it is an end bracket,
    // then the dimensionless units are zero and the multiplier is one and the
    // parsing is finished. Otherwise the parsing has failed.
    is >> nextToken;
    if (nextToken == token::COLON)
    {
        // Peek at the next token
        is >> nextToken;
        is.putBack(nextToken);
 
        // Read the dimensionless units if present, or set to zero
        if (!nextToken.isNumber())
        {
            for (int i = 0; i < unitSet::nDimlessUnits; ++ i)
            {
                units.exponents_[i] = 0;
            }
        }
        else
        {
            for (int i = 0; i < unitSet::nDimlessUnits; ++ i)
            {
                is  >> units.exponents_[i];
            }
        }
 
        // Read the next delimiting token. If a separator then there is a
        // multiplier to read. If it is an end bracket then the multiplier is
        // one and the parsing is finished. Otherwise the parsing has failed.
        is >> nextToken;
        if (nextToken == token::COLON)
        {
            // Peek at the next token
            is >> nextToken;
            is.putBack(nextToken);
 
            // Read the multiplier if present, or set to unity
            if (!nextToken.isNumber())
            {
                units.multiplier_ = 1;
            }
            else
            {
                is >> units.multiplier_;
            }
 
            // Read the next delimiting token. This must be the end bracket.
            is >> nextToken;
            if (nextToken != token::END_SQR)
            {
                FatalIOErrorInFunction(is)
                    << "expected a " << token::END_SQR << " in unitSet "
                    << endl << "in stream " << is.info() << ", got a "
                    << nextToken << exit(FatalIOError);
            }
        }
        else if (nextToken == token::END_SQR)
        {
            units.multiplier_ = 1;
        }
        else
        {
            FatalIOErrorInFunction(is)
                << "expected a " << token::END_SQR << " or a " << token::COLON
                << " in unitSet " << endl << "in stream " << is.info()
                << ", got a " << nextToken << exit(FatalIOError);
        }
    }
    else if (nextToken == token::END_SQR)
    {
        for (int i = 0; i < unitSet::nDimlessUnits; ++ i)
        {
            units.exponents_[i] = 0;
        }
 
        units.multiplier_ = 1;
    }
    else
    {
        FatalIOErrorInFunction(is)
            << "expected a " << token::END_SQR << " or a " << token::COLON
            << " in unitSet " << endl << "in stream " << is.info()
            << ", got a " << nextToken << exit(FatalIOError);
    }
 
    // Check state of Istream
    is.check("Istream& operator>>(Istream&, unitSet&)");
 
    return is;
}
 
 
Foam::Ostream& Foam::operator<<(Ostream& os, const unitSet& units)
{
    // Write the start
    os << token::BEGIN_SQR;
 
    // Write the dimensions
    units.dimensions_.writeNoBeginOrEnd(os);
 
    // Determine if any dimensionless units are non-zero
    bool nonZeroDimlessUnits = false;
    for (int i = 0; i < unitSet::nDimlessUnits; ++ i)
    {
        nonZeroDimlessUnits =
            nonZeroDimlessUnits
         || mag(units.exponents_[i]) > unitSet::smallExponent;
    }
 
    // Determine if the multiplier is non-unity
    bool nonUnityMultiplier = units.multiplier_ != 1;
 
    // Write a separator if there is anything to follow
    if (nonZeroDimlessUnits || nonUnityMultiplier)
    {
        os  << token::SPACE << token::COLON;
    }
 
    // Write the dimensionless units if any are non-zero
    if (nonZeroDimlessUnits)
    {
        for (int i = 0; i < unitSet::nDimlessUnits; ++ i)
        {
            os  << token::SPACE << units.exponents_[i];
        }
    }
 
    // Write a separator if there is anything to follow
    if (nonUnityMultiplier)
    {
        os  << token::SPACE << token::COLON;
    }
 
    // Write the multiplier if it is non-unity
    if (nonUnityMultiplier)
    {
        os  << token::SPACE << units.multiplier_;
    }
 
    // Write the end
    os  << token::END_SQR;
 
    // Check state of Ostream
    os.check("Ostream& operator<<(Ostream&, const unitSet&)");
 
    return os;
}
 
 
Foam::Ostream& Foam::operator<<
(
    Ostream& os,
    const InfoProxy<unitSet>& ip
)
{
    const unitSet& units = ip.t_;
 
    // Filter out special cases
    if (units.any())
    {
        return os << token::BEGIN_SQR << "<any>" << token::END_SQR;
    }
    if (units.none())
    {
        return os << token::BEGIN_SQR << "<none>" << token::END_SQR;
    }
 
    // Write the start
    os << token::BEGIN_SQR;
 
    // Write the dimensions
    units.dimensions_.writeInfoNoBeginOrEnd(os);
 
    // Determine if any dimensionless units are non-zero
    bool nonZeroDimlessUnits = false;
    for (int i = 0; i < unitSet::nDimlessUnits; ++ i)
    {
        nonZeroDimlessUnits =
            nonZeroDimlessUnits
         || mag(units.exponents_[i]) > unitSet::smallExponent;
    }
 
    // Determine if the multiplier is non-unity
    bool nonUnityMultiplier = units.multiplier_ != 1;
 
    // Write a separator if there is anything to follow
    if (nonZeroDimlessUnits || nonUnityMultiplier)
    {
        os  << token::SPACE << token::COLON;
    }
 
    // Write the dimensionless units if any are non-zero
    if (nonZeroDimlessUnits)
    {
        for (int i=0; i<unitSet::nDimlessUnits; i++)
        {
            if (mag(units.exponents_[i]) > unitSet::smallExponent)
            {
                os << token::SPACE << unitSet::dimlessUnitTypeNames_
                      [static_cast<unitSet::dimlessUnitType>(i)];
 
                if (units.exponents_[i] != 1)
                {
                    os << '^' << units.exponents_[i];
                }
            }
        }
    }
 
    // Write a separator if there is anything to follow
    if (nonUnityMultiplier)
    {
        os  << token::SPACE << token::COLON;
    }
 
    // Write the multiplier if it is non-unity
    if (nonUnityMultiplier)
    {
        os  << token::SPACE << units.multiplier_;
    }
 
    // Write the end
    os  << token::END_SQR;
 
    // Check state of Ostream
    os.check("Ostream& operator<<(Ostream&, const InfoProxy<unitSet>&)");
 
    return os;
}
 
 
// ************************************************************************* //