ENH: Added base class for function objects that generate text files

src/OpenFOAM/db/functionObjects/functionObjectFile/functionObjectFile.C

@@ -0,0 +1,203 @@
+/*---------------------------------------------------------------------------*\
+  =========                 |
+  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
+   \\    /   O peration     |
+    \\  /    A nd           | Copyright (C) 2012 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 "functionObjectFile.H"
+#include "Time.H"
+
+// * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
+
+const Foam::word Foam::functionObjectFile::outputPrefix = "postProcessing";
+
+// * * * * * * * * * * * * Protected Member Functions  * * * * * * * * * * * //
+
+Foam::fileName Foam::functionObjectFile::baseFileDir() const
+{
+    fileName baseDir = obr_.time().path();
+
+    if (Pstream::parRun())
+    {
+        // Put in undecomposed case (Note: gives problems for
+        // distributed data running)
+        baseDir = baseDir/".."/outputPrefix;
+    }
+    else
+    {
+        baseDir = baseDir/outputPrefix;
+    }
+
+    return baseDir;
+}
+
+
+void Foam::functionObjectFile::createFiles()
+{
+    const word startTimeName =
+        obr_.time().timeName(obr_.time().startTime().value());
+
+    label i = 0;
+    forAllConstIter(wordHashSet, names_, iter)
+    {
+        if (Pstream::master() && !filePtrs_.set(i))
+        {
+            fileName outputDir(baseFileDir()/prefix_/startTimeName);
+
+            mkDir(outputDir);
+
+            filePtrs_.set(i, new OFstream(outputDir/(iter.key() + ".dat")));
+
+            writeFileHeader(i);
+        }
+    }
+}
+
+
+void Foam::functionObjectFile::writeFileHeader(const label i)
+{
+    // do nothing
+}
+
+
+void Foam::functionObjectFile::write()
+{
+    if (Pstream::master())
+    {
+        createFiles();
+    }
+}
+
+
+void Foam::functionObjectFile::resetNames(const wordList& names)
+{
+    names_.clear();
+    names_.insert(names);
+
+    filePtrs_.clear();
+    filePtrs_.setSize(names_.toc().size());
+
+    createFiles();
+}
+
+
+void Foam::functionObjectFile::resetName(const word& name)
+{
+    names_.clear();
+    names_.insert(name);
+
+    filePtrs_.clear();
+    filePtrs_.setSize(1);
+
+    createFiles();
+}
+
+
+// * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
+
+Foam::functionObjectFile::functionObjectFile
+(
+    const objectRegistry& obr,
+    const word& prefix
+)
+:
+    obr_(obr),
+    prefix_(prefix),
+    names_(),
+    filePtrs_()
+{}
+
+
+Foam::functionObjectFile::functionObjectFile
+(
+    const objectRegistry& obr,
+    const word& prefix,
+    const word& name
+)
+:
+    obr_(obr),
+    prefix_(prefix),
+    names_(),
+    filePtrs_()
+{
+    names_.clear();
+    names_.insert(name);
+
+    filePtrs_.clear();
+    filePtrs_.setSize(names_.toc().size());
+}
+
+
+Foam::functionObjectFile::functionObjectFile
+(
+    const objectRegistry& obr,
+    const word& prefix,
+    const wordList& names
+)
+:
+    obr_(obr),
+    prefix_(prefix),
+    names_(names),
+    filePtrs_()
+{
+    names_.clear();
+    names_.insert(names);
+
+    filePtrs_.clear();
+    filePtrs_.setSize(names_.toc().size());
+}
+
+
+// * * * * * * * * * * * * * * * * Destructor  * * * * * * * * * * * * * * * //
+
+Foam::functionObjectFile::~functionObjectFile()
+{}
+
+
+// * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
+
+Foam::OFstream& Foam::functionObjectFile::file()
+{
+    if (filePtrs_.size() != 1)
+    {
+        WarningIn("Foam::Ostream& Foam::functionObjectFile::file()")
+            << "Requested single file, but multiple files are present"
+            << endl;
+    }
+
+    return filePtrs_[0];
+}
+
+
+Foam::PtrList<Foam::OFstream>& Foam::functionObjectFile::files()
+{
+    return filePtrs_;
+}
+
+
+Foam::OFstream& Foam::functionObjectFile::file(const label i)
+{
+    return filePtrs_[i];
+}
+
+
+// ************************************************************************* //