functionObjects::writeRegisteredObject -> functionObjects::writeObjects

Added configuration file etc/caseDicts/postProcessing/fields/writeObjects

src/functionObjects/utilities/writeObjects/writeObjects.C

@@ -0,0 +1,164 @@
+/*---------------------------------------------------------------------------*\
+  =========                 |
+  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
+   \\    /   O peration     |
+    \\  /    A nd           | Copyright (C) 2011-2016 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 "writeObjects.H"
+#include "Time.H"
+#include "polyMesh.H"
+#include "addToRunTimeSelectionTable.H"
+
+// * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
+
+namespace Foam
+{
+namespace functionObjects
+{
+    defineTypeNameAndDebug(writeObjects, 0);
+
+    addToRunTimeSelectionTable
+    (
+        functionObject,
+        writeObjects,
+        dictionary
+    );
+}
+}
+
+
+// * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
+
+Foam::functionObjects::writeObjects::writeObjects
+(
+    const word& name,
+    const Time& runTime,
+    const dictionary& dict
+)
+:
+    functionObject(name),
+    obr_
+    (
+        runTime.lookupObject<objectRegistry>
+        (
+            dict.lookupOrDefault("region", polyMesh::defaultRegion)
+        )
+    ),
+    exclusiveWriting_(false),
+    objectNames_()
+{
+    read(dict);
+}
+
+
+// * * * * * * * * * * * * * * * * Destructor  * * * * * * * * * * * * * * * //
+
+Foam::functionObjects::writeObjects::~writeObjects()
+{}
+
+
+// * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
+
+bool Foam::functionObjects::writeObjects::read(const dictionary& dict)
+{
+    if (dict.found("field"))
+    {
+        objectNames_.setSize(1);
+        dict.lookup("field") >> objectNames_[0];
+    }
+    else if (dict.found("fields"))
+    {
+        dict.lookup("fields") >> objectNames_;
+    }
+    else
+    {
+        dict.lookup("objects") >> objectNames_;
+    }
+
+    dict.readIfPresent("exclusiveWriting", exclusiveWriting_);
+
+    return true;
+}
+
+
+bool Foam::functionObjects::writeObjects::execute
+(
+    const bool postProcess
+)
+{
+    return true;
+}
+
+
+bool Foam::functionObjects::writeObjects::write
+(
+    const bool postProcess
+)
+{
+    Info<< type() << " " << name() << " write:" << nl;
+
+    if (!obr_.time().writeTime())
+    {
+        obr_.time().writeTimeDict();
+    }
+
+    DynamicList<word> allNames(obr_.toc().size());
+    forAll(objectNames_, i)
+    {
+        wordList names(obr_.names<regIOobject>(objectNames_[i]));
+
+        if (names.size())
+        {
+            allNames.append(names);
+        }
+        else
+        {
+            WarningInFunction
+                << "Object " << objectNames_[i] << " not found in "
+                << "database. Available objects:" << nl << obr_.sortedToc()
+                << endl;
+        }
+    }
+
+    forAll(allNames, i)
+    {
+        regIOobject& obj = const_cast<regIOobject&>
+        (
+            obr_.lookupObject<regIOobject>(allNames[i])
+        );
+
+        if (exclusiveWriting_)
+        {
+            // Switch off automatic writing to prevent double write
+            obj.writeOpt() = IOobject::NO_WRITE;
+        }
+
+        Info<< "    writing object " << obj.name() << endl;
+
+        obj.write();
+    }
+
+    return true;
+}
+
+
+// ************************************************************************* //

src/functionObjects/utilities/writeObjects/writeObjects.H

@@ -0,0 +1,163 @@
+/*---------------------------------------------------------------------------*\
+  =========                 |
+  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
+   \\    /   O peration     |
+    \\  /    A nd           | Copyright (C) 2011-2016 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/>.
+
+Class
+    Foam::functionObjects::writeObjects
+
+Group
+    grpUtilitiesFunctionObjects
+
+Description
+    This function object allows specification of different writing frequency
+    of objects registered to the database. It has similar functionality
+    as the main time database through the writeControl setting:
+
+    \li \c timeStep
+    \li \c writeTime
+    \li \c adjustableRunTime
+    \li \c runTime
+    \li \c clockTime
+    \li \c cpuTime
+
+    Example of function object specification:
+    \verbatim
+    writeObjects1
+    {
+        type        writeObjects;
+        libs ("libutilityFunctionObjects.so");
+        exclusiveWriting     true;
+        ...
+        objectNames (obj1 obj2);
+    }
+    \endverbatim
+
+    \heading Function object usage
+    \table
+        Property     | Description             | Required    | Default value
+        type         | type name: writeObjects | yes |
+        objectNames  | objects to write        | yes         |
+        exclusiveWriting    | Takes over object writing | no | yes
+    \endtable
+
+    \c exclusiveWriting disables automatic writing (i.e through database) of the
+    objects to avoid duplicate writing.
+
+SeeAlso
+    Foam::functionObject
+    Foam::functionObjects::timeControl
+
+SourceFiles
+    writeObjects.C
+
+\*---------------------------------------------------------------------------*/
+
+#ifndef functionObjects_writeObjects_H
+#define functionObjects_writeObjects_H
+
+#include "functionObject.H"
+#include "wordReList.H"
+
+// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
+
+namespace Foam
+{
+
+// Forward declaration of classes
+class objectRegistry;
+
+namespace functionObjects
+{
+
+/*---------------------------------------------------------------------------*\
+                   Class writeObjects Declaration
+\*---------------------------------------------------------------------------*/
+
+class writeObjects
+:
+    public functionObject
+{
+    // Private data
+
+        //- Refererence to Db
+        const objectRegistry& obr_;
+
+        //- Takes over the writing from Db
+        bool exclusiveWriting_;
+
+        //- Names of objects to control
+        wordReList objectNames_;
+
+
+    // Private Member Functions
+
+        //- Disallow default bitwise copy construct
+        writeObjects(const writeObjects&);
+
+        //- Disallow default bitwise assignment
+        void operator=(const writeObjects&);
+
+
+public:
+
+    //- Runtime type information
+    TypeName("writeObjects");
+
+
+    // Constructors
+
+        //- Construct from Time and dictionary
+        writeObjects
+        (
+            const word& name,
+            const Time& runTime,
+            const dictionary& dict
+        );
+
+
+    //- Destructor
+    virtual ~writeObjects();
+
+
+    // Member Functions
+
+        //- Read the writeObjects data
+        virtual bool read(const dictionary&);
+
+        //- Do nothing
+        virtual bool execute(const bool postProcess = false);
+
+        //- Write the registered objects
+        virtual bool write(const bool postProcess = false);
+};
+
+
+// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
+
+} // End namespace functionObjects
+} // End namespace Foam
+
+// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
+
+#endif
+
+// ************************************************************************* //