applications/utilities/postProcessing/miscellaneous/ptot/ptot.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 2011 OpenFOAM Foundation
   6      \\/     M anipulation  |
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of OpenFOAM.
  10
  11     OpenFOAM is free software: you can redistribute it and/or modify it
  12     under the terms of the GNU General Public License as published by
  13     the Free Software Foundation, either version 3 of the License, or
  14     (at your option) any later version.
  15
  16     OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
  17     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  18     FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  19     for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with OpenFOAM.  If not, see <http://www.gnu.org/licenses/>.
  23
  24 Application
  25     ptot
  26
  27 Description
  28     For each time: calculate the total pressure.
  29
  30 \*---------------------------------------------------------------------------*/
  31
  32 #include "fvCFD.H"
  33
  34 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  35
  36 int main(int argc, char *argv[])
  37 {
  38     timeSelector::addOptions();
  39
  40 #   include "setRootCase.H"
  41 #   include "createTime.H"
  42
  43     instantList timeDirs = timeSelector::select0(runTime, args);
  44
  45 #   include "createMesh.H"
  46
  47     forAll(timeDirs, timeI)
  48     {
  49         runTime.setTime(timeDirs[timeI], timeI);
  50
  51         Info<< "Time = " << runTime.timeName() << endl;
  52
  53         IOobject pheader
  54         (
  55             "p",
  56             runTime.timeName(),
  57             mesh,
  58             IOobject::MUST_READ
  59         );
  60
  61         IOobject Uheader
  62         (
  63             "U",
  64             runTime.timeName(),
  65             mesh,
  66             IOobject::MUST_READ
  67         );
  68
  69
  70         // Check p and U exist
  71         if (pheader.headerOk() && Uheader.headerOk())
  72         {
  73             mesh.readUpdate();
  74
  75             Info<< "    Reading p" << endl;
  76             volScalarField p(pheader, mesh);
  77
  78             Info<< "    Reading U" << endl;
  79             volVectorField U(Uheader, mesh);
  80
  81             Info<< "    Calculating ptot" << endl;
  82             if (p.dimensions() == dimensionSet(0, 2, -2, 0, 0))
  83             {
  84                 volScalarField ptot
  85                 (
  86                     IOobject
  87                     (
  88                         "ptot",
  89                         runTime.timeName(),
  90                         mesh,
  91                         IOobject::NO_READ
  92                     ),
  93                     p + 0.5*magSqr(U)
  94                 );
  95                 ptot.write();
  96             }
  97             else
  98             {
  99                 IOobject rhoheader
 100                 (
 101                     "rho",
 102                     runTime.timeName(),
 103                     mesh,
 104                     IOobject::MUST_READ
 105                 );
 106
 107                 // Check rho exists
 108                 if (rhoheader.headerOk())
 109                 {
 110                     Info<< "    Reading rho" << endl;
 111                     volScalarField rho(rhoheader, mesh);
 112
 113                     volScalarField ptot
 114                     (
 115                         IOobject
 116                         (
 117                             "ptot",
 118                             runTime.timeName(),
 119                             mesh,
 120                             IOobject::NO_READ
 121                         ),
 122                         p + 0.5*rho*magSqr(U)
 123                     );
 124                     ptot.write();
 125                 }
 126                 else
 127                 {
 128                     Info<< "    No rho" << endl;
 129                 }
 130             }
 131         }
 132         else
 133         {
 134             Info<< "    No p or U" << endl;
 135         }
 136
 137         Info<< endl;
 138     }
 139
 140     return 0;
 141 }
 142
 143
 144 // ************************************************************************* //