#ifndef LAPLACIAN_HH
#define LAPLACIAN_HH
#include <dune/istl/bcrsmatrix.hh>
#include <dune/common/fvector.hh>

template<class B>
void setupSparsityPattern(Dune::BCRSMatrix<B>& A, int N)
{
  typedef typename Dune::BCRSMatrix<B> Matrix;
  A.setSize(N*N, N*N, N*N*5);
  A.setBuildMode(Matrix::row_wise);
  
  for (typename Dune::BCRSMatrix<B>::CreateIterator i = A.createbegin(); i != A.createend(); ++i){
    int x = i.index()%N; // x coordinate in the 2d field
    int y = i.index()/N;  // y coordinate in the 2d field

    if(y>0)
      // insert lower neighbour
      i.insert(i.index()-N);
    if(x>0)
      // insert left neighbour
      i.insert(i.index()-1);

    // insert diagonal value
    i.insert(i.index());
    
    if(x<N-1)
      //insert right neighbour
      i.insert(i.index()+1);
    if(y<N-1)
      // insert upper neighbour
      i.insert(i.index()+N);
  }
}


template<class B>
void setupLaplacian(Dune::BCRSMatrix<B>& A, int N)
{ 
  typedef typename Dune::BCRSMatrix<B>::field_type FieldType;
  
  setupSparsityPattern(A,N);
  
  B diagonal(static_cast<FieldType>(0)), bone(static_cast<FieldType>(0)),
      beps(static_cast<FieldType>(0));
  for(typename B::RowIterator b = diagonal.begin(); b !=  diagonal.end(); ++b)
    b->operator[](b.index())=4;


  for(typename B::RowIterator b=bone.begin(); b !=  bone.end(); ++b)
    b->operator[](b.index())=-1.0;

  
  for (typename Dune::BCRSMatrix<B>::RowIterator i = A.begin(); i != A.end(); ++i){
    int x = i.index()%N; // x coordinate in the 2d field
    int y = i.index()/N;  // y coordinate in the 2d field
    
    /*    if(x==0 || x==N-1 || y==0||y==N-1){
      
      i->operator[](i.index())=1.0;
    
      if(y>0)
      i->operator[](i.index()-N)=0;

    if(y<N-1)
      i->operator[](i.index()+N)=0.0;
    
    if(x>0)
      i->operator[](i.index()-1)=0.0;
    
    if(x < N-1)
      i->operator[](i.index()+1)=0.0;
    
      }else*/
    {
      
    i->operator[](i.index())=diagonal;
  
    if(y>0)
      i->operator[](i.index()-N)=bone;

    if(y<N-1)
      i->operator[](i.index()+N)=bone;
    
    if(x>0)
      i->operator[](i.index()-1)=bone;
    
    if(x < N-1)
      i->operator[](i.index()+1)=bone;
    }
  }
}
template<int BS>
void setBoundary(Dune::BlockVector<Dune::FieldVector<double,BS> >& lhs, 
		 Dune::BlockVector<Dune::FieldVector<double,BS> >& rhs,
		 int N)
{
  for(int i=0; i < lhs.size(); ++i){
    int x = i/N;
    int y = i%N;

    if(x==0 || y ==0 || x==N-1 || y==N-1){
      lhs[i]=rhs[i]=0;
    }
  }
}  
#endif
