SourceDiscretization.h

Go to the documentation of this file.

// This file os part of FVM
// Copyright (c) 2012 FVM Authors
// See LICENSE file for terms.

#ifndef _SOURCEDISCRETIZATION_H_
#define _SOURCEDISCRETIZATION_H_

#include "Field.h"
#include "MultiField.h"
#include "MultiFieldMatrix.h"
#include "Mesh.h"
#include "Discretization.h"
#include "StorageSite.h"
#include "GeomFields.h"
#include "CRMatrixRect.h"

template<class X>
class SourceDiscretization : public Discretization
{
public:

  typedef typename NumTypeTraits<X>::T_Scalar T_Scalar;
  
  typedef Array<X> XArray;

  typedef Array<T_Scalar> TArray;
 
  SourceDiscretization(const MeshList& meshes,
                       const GeomFields& geomFields,
                       const Field& varField,
                       const Field& sourceField)  :
    Discretization(meshes),
    _geomFields(geomFields),
    _varField(varField),
    _sourceField(sourceField)
   {}

                          
  void discretize(const Mesh& mesh, MultiFieldMatrix& mfmatrix,
                  MultiField&, MultiField& rField)
  {
    const StorageSite& cells = mesh.getCells();

    const TArray& cellVolume = dynamic_cast<const TArray&>(_geomFields.volume[cells]);

    const XArray& source = dynamic_cast<const XArray&>(_sourceField[cells]);

    const MultiField::ArrayIndex cVarIndex(&_varField, &cells);

    XArray& rCell = dynamic_cast<XArray&>(rField[cVarIndex]);   
    
    const int nCells = cells.getSelfCount();
    
    for(int c=0; c<nCells; c++)
    {
        rCell[c] += cellVolume[c]*source[c];
    }
  }
    

private:
  const GeomFields& _geomFields;
  const Field& _varField;
  const Field& _sourceField;
};

#endif