CG_SOLVER.h

// File : CG_SOLVER.h
//
// Copyright 2006
// The University of North Carolina at Chapel Hill
// 
//
// Permission to use, copy, modify, distribute and sell this software and its 
// documentation for any purpose is hereby granted without fee, provided that 
// the above copyright notice appear in all copies and that both that copyright 
// notice and this permission notice appear in supporting documentation. 
// Binaries may be compiled with this software without any royalties or 
// restrictions. 
//
// The University of North Carolina at Chapel Hill makes no representations 
// about the suitability of this software for any purpose. It is provided 
// "as is" without express or implied warranty.
//

#ifndef CG_SOLVER_H
#define CG_SOLVER_H

#include "CELL.h"
#include <cmath>
#include <list>

using namespace std;

class CG_SOLVER
{
public:
        CG_SOLVER(int maxDepth, int iterations = 10, int digits = 8);
        virtual ~CG_SOLVER();

  virtual int solve(list<CELL*> cells);

  float calcResidual(list<CELL*> cells);

  int& iterations() { return _iterations; };

protected:  
  int _iterations;  
  int _digits;      

  // conjugate gradient arrays
  float* _direction;  
  float* _potential;  
  float* _residual;   
  float* _q;          
  
  int _arraySize;     
  int _listSize;      

  void calcStencils(list<CELL*> cells);

  virtual void reallocate();

  float* _dx;
};

#endif

---