QUAD_POISSON.h

// File : QUAD_POISSON.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 QUAD_POISSON_H
#define QUAD_POISSON_H

#define USINGGLUT

#ifdef USINGGLUT
#include <gl/glut.h>
#endif
#include <cstdlib>
#include "CELL.h"
#include <list>
#include "CG_SOLVER.h"
#include "CG_SOLVER_SSE.h"
#include "BlueNoise/BLUE_NOISE.h"

#include <iostream>

using namespace std;

class QUAD_POISSON  
{
public:
        QUAD_POISSON(int xRes, 
               int yRes,
               int iterations = 10);
  
        virtual ~QUAD_POISSON();
 
#ifdef USINGGLUT  
  void draw(CELL* cell = NULL);

  void drawCell(CELL* cell, 
                float r = 1.0f, 
                float g = 0.0f, 
                float b = 0.0f);
#endif

  int solve();  
 
  CELL* insert(float xPos, float yPos);

  CELL* insert(int xPos, int yPos) { 
    return insert((float)xPos / _maxRes, (float)yPos / _maxRes);
  };
  
  void getAllLeaves(list<CELL*>& leaves, CELL* currentCell = NULL);
  
  list<CELL*>& getSmallestLeaves() { return _smallestLeaves; };

  int& maxRes() { return _maxRes; }

  int& maxDepth() { return _maxDepth; };
  
private:
  CELL* _root;

  int _maxRes;

  int _maxDepth;
  
  list<CELL*> _emptyLeaves;

  list<CELL*> _smallestLeaves;
  
  CG_SOLVER* _solver;
  
  void balance();

  void getEmptyLeaves(list<CELL*>& leaves, CELL* currentCell = NULL);
  
  void buildNeighbors();

  void deleteGhosts(CELL* currentCell = NULL);

  BLUE_NOISE* _noiseFunc;

  bool* _noise;

  void setNoise(CELL* cell);
};

#endif

---