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) 1991-2008 OpenCFD Ltd.
   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 the
  13     Free Software Foundation; either version 2 of the License, or (at your
  14     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, write to the Free Software Foundation,
  23     Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  24
  25 Application
  26     ptot
  27
  28 Description
  29     For each time: calculate the total pressure.
  30
  31 \*---------------------------------------------------------------------------*/
  32
  33 #include "fvCFD.H"
  34
  35 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  36
  37 int main(int argc, char *argv[])
  38 {
  39
  40 #   include "addTimeOptions.H"
  41 #   include "setRootCase.H"
  42
  43 #   include "createTime.H"
  44
  45     // Get times list
  46     instantList Times = runTime.times();
  47
  48     // set startTime and endTime depending on -time and -latestTime options
  49 #   include "checkTimeOptions.H"
  50
  51     runTime.setTime(Times[startTime], startTime);
  52
  53 #   include "createMesh.H"
  54
  55     for (label i=startTime; i<endTime; i++)
  56     {
  57         runTime.setTime(Times[i], i);
  58
  59         Info<< "Time = " << runTime.timeName() << endl;
  60
  61         IOobject pheader
  62         (
  63             "p",
  64             runTime.timeName(),
  65             mesh,
  66             IOobject::MUST_READ
  67         );
  68
  69         IOobject Uheader
  70         (
  71             "U",
  72             runTime.timeName(),
  73             mesh,
  74             IOobject::MUST_READ
  75         );
  76
  77
  78         // Check p and U exist
  79         if (pheader.headerOk() && Uheader.headerOk())
  80         {
  81             mesh.readUpdate();
  82
  83             Info<< "    Reading p" << endl;
  84             volScalarField p(pheader, mesh);
  85
  86             Info<< "    Reading U" << endl;
  87             volVectorField U(Uheader, mesh);
  88
  89             Info<< "    Calculating ptot" << endl;
  90             if (p.dimensions() == dimensionSet(0, 2, -2, 0, 0))
  91             {
  92                 volScalarField ptot
  93                 (
  94                     IOobject
  95                     (
  96                         "ptot",
  97                         runTime.timeName(),
  98                         mesh,
  99                         IOobject::NO_READ
 100                     ),
 101                     p + 0.5*magSqr(U)
 102                 );
 103                 ptot.write();
 104             }
 105             else
 106             {
 107                 IOobject rhoheader
 108                 (
 109                     "rho",
 110                     runTime.timeName(),
 111                     mesh,
 112                     IOobject::MUST_READ
 113                 );
 114
 115                 // Check rho exists
 116                 if (rhoheader.headerOk())
 117                 {
 118                     Info<< "    Reading rho" << endl;
 119                     volScalarField rho(rhoheader, mesh);
 120
 121                     volScalarField ptot
 122                     (
 123                         IOobject
 124                         (
 125                             "ptot",
 126                             runTime.timeName(),
 127                             mesh,
 128                             IOobject::NO_READ
 129                         ),
 130                         p + 0.5*rho*magSqr(U)
 131                     );
 132                     ptot.write();
 133                 }
 134                 else
 135                 {
 136                     Info<< "    No rho" << endl;
 137                 }
 138             }
 139         }
 140         else
 141         {
 142             Info<< "    No p or U" << endl;
 143         }
 144
 145         Info<< endl;
 146     }
 147
 148     return(0);
 149 }
 150
 151
 152 // ************************************************************************* //