src/OpenFOAM/db/IOobjectList/IOobjectList.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 1991-2009 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 \*---------------------------------------------------------------------------*/
  26
  27 #include "IOobjectList.H"
  28 #include "Time.H"
  29 #include "OSspecific.H"
  30
  31
  32 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
  33
  34 Foam::IOobjectList::IOobjectList(const label nIoObjects)
  35 :
  36     HashPtrTable<IOobject>(nIoObjects)
  37 {}
  38
  39
  40 Foam::IOobjectList::IOobjectList
  41 (
  42     const objectRegistry& db,
  43     const fileName& instance,
  44     const fileName& local
  45 )
  46 :
  47     HashPtrTable<IOobject>()
  48 {
  49     word newInstance = instance;
  50
  51     if (!isDir(db.path(instance)))
  52     {
  53         newInstance = db.time().findInstancePath(instant(instance));
  54
  55         if (newInstance.empty())
  56         {
  57             return;
  58         }
  59     }
  60
  61     // Create list file names in directory
  62     fileNameList ObjectNames =
  63         readDir(db.path(newInstance, db.dbDir()/local), fileName::FILE);
  64
  65     forAll(ObjectNames, i)
  66     {
  67         IOobject* objectPtr = new IOobject
  68         (
  69             ObjectNames[i],
  70             newInstance,
  71             local,
  72             db,
  73             IOobject::MUST_READ,
  74             IOobject::NO_WRITE
  75         );
  76
  77         if (objectPtr->headerOk())
  78         {
  79             insert(ObjectNames[i], objectPtr);
  80         }
  81         else
  82         {
  83             delete objectPtr;
  84         }
  85     }
  86 }
  87
  88
  89 Foam::IOobjectList::IOobjectList(const IOobjectList& ioOL)
  90 :
  91     HashPtrTable<IOobject>(ioOL)
  92 {}
  93
  94
  95 // * * * * * * * * * * * * * * * * Destructors * * * * * * * * * * * * * * * //
  96
  97 Foam::IOobjectList::~IOobjectList()
  98 {}
  99
 100
 101 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 102
 103 bool Foam::IOobjectList::add(IOobject& io)
 104 {
 105     return insert(io.name(), &io);
 106 }
 107
 108
 109 bool Foam::IOobjectList::remove(IOobject& io)
 110 {
 111     HashPtrTable<IOobject>::iterator iter =
 112         HashPtrTable<IOobject>::find(io.name());
 113
 114     if (iter != end())
 115     {
 116         return erase(iter);
 117     }
 118     else
 119     {
 120         return false;
 121     }
 122 }
 123
 124
 125 Foam::IOobject* Foam::IOobjectList::lookup(const word& name) const
 126 {
 127     HashPtrTable<IOobject>::const_iterator iter = find(name);
 128
 129     if (iter != end())
 130     {
 131         if (IOobject::debug)
 132         {
 133             Info<< "IOobjectList::lookup : found " << name
 134                 << endl;
 135         }
 136
 137         return const_cast<IOobject*>(*iter);
 138     }
 139     else
 140     {
 141         if (IOobject::debug)
 142         {
 143             Info<< "IOobjectList::lookup : could not find " << name
 144                 << endl;
 145         }
 146
 147         return NULL;
 148     }
 149 }
 150
 151
 152 Foam::IOobjectList Foam::IOobjectList::lookupClass(const word& ClassName) const
 153 {
 154     IOobjectList IOobjectsOfClass(size());
 155
 156     for
 157     (
 158         HashPtrTable<IOobject>::const_iterator iter = begin();
 159         iter != end();
 160         ++iter
 161     )
 162     {
 163         if (iter()->headerClassName() == ClassName)
 164         {
 165             if (IOobject::debug)
 166             {
 167                 Info<< "IOobjectList::lookupClass : found "
 168                     << iter()->name()
 169                     << endl;
 170             }
 171
 172             IOobjectsOfClass.insert(iter()->name(), new IOobject(*iter()));
 173         }
 174     }
 175
 176     return IOobjectsOfClass;
 177 }
 178
 179
 180 Foam::wordList Foam::IOobjectList::names() const
 181 {
 182     wordList objectNames(size());
 183
 184     label count = 0;
 185     for
 186     (
 187         HashPtrTable<IOobject>::const_iterator iter = begin();
 188         iter != end();
 189         ++iter
 190     )
 191     {
 192         objectNames[count++] = iter()->name();
 193     }
 194
 195     return objectNames;
 196 }
 197
 198
 199 Foam::wordList Foam::IOobjectList::names(const word& ClassName) const
 200 {
 201     wordList objectNames(size());
 202
 203     label count = 0;
 204     for
 205     (
 206         HashPtrTable<IOobject>::const_iterator iter = begin();
 207         iter != end();
 208         ++iter
 209     )
 210     {
 211         if (iter()->headerClassName() == ClassName)
 212         {
 213             objectNames[count++] = iter()->name();
 214         }
 215     }
 216
 217     objectNames.setSize(count);
 218
 219     return objectNames;
 220 }
 221
 222
 223 // ************************************************************************* //