#include "CMapper.h"



//--------------------------- Init ---------------------------------------
//
//  This method needs to be called before you can use an instance of this
//  class. It calculates and sets up all the cells in the defined region.
//------------------------------------------------------------------------
void CMapper::Init(int MaxRangeX, int MaxRangeY)
{
  //if already initialized return
  if(m_NumCellsX) return;

  m_dCellSize = CParams::dCellSize;
  
  //first calculate how many segments we will require
  m_NumCellsX = (int)(MaxRangeX/m_dCellSize)+1;
  m_NumCellsY = (int)(MaxRangeY/m_dCellSize)+1;
  
  //create the 2d vector of blank segments
  for (int x=0; x<m_NumCellsX; ++x)
  {
    vector<SCell> temp;

    for (int y=0; y<m_NumCellsY; ++y)
    {
      temp.push_back(SCell(x*m_dCellSize, (x+1)*m_dCellSize, y*m_dCellSize, (y+1)*m_dCellSize));
    }

    m_2DvecCells.push_back(temp);
  }

  m_iTotalCells = m_NumCellsX * m_NumCellsY;
}

//--------------------- Update -------------------------------------------
//
//  when called with an X/Y position this method increases the frequency
//  count of the cell at that position.
//------------------------------------------------------------------------
void CMapper::Update(double xPos, double yPos)
{
  //check to make sure positions are within range
  if ( (xPos < 0) || (xPos > CParams::WindowWidth) || 
       (yPos < 0) || (yPos > CParams::WindowHeight) )
  {
    return;
  }
  
  int cellX = (int)(xPos / m_dCellSize );
  int cellY = (int)(yPos / m_dCellSize );

  m_2DvecCells[cellX][cellY].Update();

  return;
}

//------------------------ TicksLingered ---------------------------------
//
//  returns the frequency count of the cell at this location
//------------------------------------------------------------------------
int CMapper::TicksLingered(double xPos, double yPos)const
{
  //bounds check the incoming values
  if ( (xPos > CParams::WindowWidth) || (xPos < 0) ||
       (yPos > CParams::WindowHeight)|| (yPos < 0))
  {
    return 999;
  }
  
  int cellX = (int)(xPos / m_dCellSize);
  int cellY = (int)(yPos / m_dCellSize);

  return m_2DvecCells[cellX][cellY].iTicksSpentHere;
}

//------------------------- Visited --------------------------------------
//
//  returns true if the cell has been visited. False if not
//------------------------------------------------------------------------
bool CMapper::BeenVisited(double xPos, double yPos)const
{
  int cellX = (int)(xPos / m_dCellSize);
  int cellY = (int)(yPos / m_dCellSize);

  if (m_2DvecCells[cellX][cellY].iTicksSpentHere > 0)
  {
    return true;
  }

  else
  {
    return false;
  }
}
//--------------------------------- Render -------------------------------
//
//  renders the visited cells. The color gets darker the more frequently
//  the cell has been visited.
//------------------------------------------------------------------------
void CMapper::Render(HDC surface)
{

  for (int x=0; x<m_NumCellsX; ++x)
  {
    for (int y=0; y<m_NumCellsY; ++y)
    {
      if (m_2DvecCells[x][y].iTicksSpentHere > 0)
      {
        int shading = 2 * m_2DvecCells[x][y].iTicksSpentHere;

        if (shading >220)
        {
          shading = 220;
        }

        
        HBRUSH lightbrush = CreateSolidBrush(RGB(240,220-shading,220-shading));
        
        FillRect(surface, &m_2DvecCells[x][y].Cell, lightbrush); 

        DeleteObject(lightbrush);
      }
    }
  } 
}

//----------------------------------- Reset ------------------------------
void CMapper::Reset()
{

  for (int x=0; x<m_NumCellsX; ++x)
  {
    for (int y=0; y<m_NumCellsY; ++y)
    {
     m_2DvecCells[x][y].Reset();
    }
  }
}


//----------------------------- NumCellsVisited --------------------------
//
//  returns the total number of cells in the map which have been visited
//------------------------------------------------------------------------
int CMapper::NumCellsVisited() const
{
  int total = 0;

  for (int x=0; x<m_NumCellsX; ++x)
  {
    for (int y=0; y<m_NumCellsY; ++y)
    {
      if (m_2DvecCells[x][y].iTicksSpentHere > 0)
      {
        ++total;
      }
    }
  }

  return total;
}