src/foam/graph/graph.H

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | foam-extend: Open Source CFD
   4    \\    /   O peration     | Version:     3.2
   5     \\  /    A nd           | Web:         http://www.foam-extend.org
   6      \\/     M anipulation  | For copyright notice see file Copyright
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of foam-extend.
  10
  11     foam-extend 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 3 of the License, or (at your
  14     option) any later version.
  15
  16     foam-extend is distributed in the hope that it will be useful, but
  17     WITHOUT ANY WARRANTY; without even the implied warranty of
  18     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19     General Public License for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with foam-extend.  If not, see <http://www.gnu.org/licenses/>.
  23
  24 Class
  25     Foam::graph
  26
  27 Description
  28     Class to create, store and output qgraph files.
  29
  30 SourceFiles
  31     graph.C
  32
  33 \*---------------------------------------------------------------------------*/
  34
  35 #ifndef graph_H
  36 #define graph_H
  37
  38 #include "foamString.H"
  39 #include "point.H"
  40 #include "HashPtrTable.H"
  41 #include "curve.H"
  42
  43 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  44
  45 namespace Foam
  46 {
  47
  48 // Forward declaration of friend functions and operators
  49
  50 class graph;
  51
  52 Ostream& operator<<(Ostream&, const graph&);
  53
  54
  55 /*---------------------------------------------------------------------------*\
  56                            Class graph Declaration
  57 \*---------------------------------------------------------------------------*/
  58
  59 class graph
  60 :
  61     public HashPtrTable<curve>
  62 {
  63     // Private Data
  64
  65         string title_;
  66         string xName_;
  67         string yName_;
  68
  69         scalarField x_;
  70
  71
  72         struct xy
  73         {
  74             scalar x_, y_;
  75
  76             xy()
  77             {}
  78
  79         // Friend Operators
  80
  81             friend bool operator==(const xy& a, const xy& b)
  82             {
  83                 return equal(a.x_, b.x_) && equal(a.y_, b.y_);
  84             }
  85
  86             friend bool operator!=(const xy& a, const xy& b)
  87             {
  88                 return !(a == b);
  89             }
  90
  91             friend Istream& operator>>(Istream& is, xy& xyd)
  92             {
  93                 is >> xyd.x_ >> xyd.y_;
  94                 return is;
  95             }
  96
  97             friend Ostream& operator<<(Ostream& os, const xy& xyd)
  98             {
  99                 os << xyd.x_ << ' ' << xyd.y_;
 100                 return os;
 101             }
 102         };
 103
 104
 105     // Private Member Functions
 106
 107         void readCurves(Istream&);
 108
 109
 110 public:
 111
 112     // Constructors
 113
 114         //- Construct from title and labels (no curves)
 115         graph
 116         (
 117             const string& title,
 118             const string& xName,
 119             const string& yName,
 120             const scalarField& x
 121         );
 122
 123         //- Construct from title, labels and y data for 1 curve
 124         graph
 125         (
 126             const string& title,
 127             const string& xName,
 128             const string& yName,
 129             const scalarField& x,
 130             const scalarField& y
 131         );
 132
 133         //- Construct from Istream given title and labels
 134         graph
 135         (
 136             const string& title,
 137             const string& xName,
 138             const string& yName,
 139             Istream& is
 140         );
 141
 142         //- Construct from Istream
 143         graph(Istream& is);
 144
 145
 146     // Member functions
 147
 148         // Access
 149
 150             const string& title() const
 151             {
 152                 return title_;
 153             }
 154
 155             const string& xName() const
 156             {
 157                 return xName_;
 158             }
 159
 160             const string& yName() const
 161             {
 162                 return yName_;
 163             }
 164
 165
 166             const scalarField& x() const
 167             {
 168                 return x_;
 169             }
 170
 171             scalarField& x()
 172             {
 173                 return x_;
 174             }
 175
 176
 177             const scalarField& y() const;
 178
 179             scalarField& y();
 180
 181
 182         // Write
 183
 184             //- Abstract base class for a graph writer
 185             class writer
 186             {
 187
 188             protected:
 189
 190                 void writeXY
 191                 (
 192                     const scalarField& x,
 193                     const scalarField& y,
 194                     Ostream&
 195                 ) const;
 196
 197             public:
 198
 199                 //- Runtime type information
 200                 TypeName("writer");
 201
 202                 //- Declare run-time constructor selection table
 203                 declareRunTimeSelectionTable
 204                 (
 205                     autoPtr,
 206                     writer,
 207                     word,
 208                     (),
 209                     ()
 210                 );
 211
 212                 // Selectors
 213
 214                     //- Return a reference to the selected writer
 215                     static autoPtr<writer> New
 216                     (
 217                         const word& writeFormat
 218                     );
 219
 220
 221                 // Constructors
 222
 223                     //- Construct null
 224                     writer()
 225                     {}
 226
 227
 228                 // Destructor
 229
 230                     virtual ~writer()
 231                     {}
 232
 233
 234                 // Member Functions
 235
 236                     // Access
 237
 238                         //- Return the appropriate fileName extension
 239                         //  for this graph format
 240                         virtual const word& ext() const = 0;
 241
 242
 243                     // Write
 244
 245                         //- Write graph in appropriate format
 246                         virtual void write(const graph&, Ostream&) const = 0;
 247             };
 248
 249             //- Write out graph data as a simple table
 250             void writeTable(Ostream&) const;
 251
 252             //- Write graph to stream in given format
 253             void write(Ostream&, const word& format) const;
 254
 255             //- Write graph to file in given format
 256             void write(const fileName& fName, const word& format) const;
 257
 258
 259     // Friend operators
 260
 261         //- Ostream Operator
 262         friend Ostream& operator<<(Ostream&, const graph&);
 263 };
 264
 265
 266 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 267
 268 } // End namespace Foam
 269
 270 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 271
 272 #endif
 273
 274 // ************************************************************************* //