eliot/game/bag.cpp
2013-01-17 17:59:28 +01:00

150 lines
3.4 KiB
C++

/*****************************************************************************
* Eliot
* Copyright (C) 1999-2012 Antoine Fraboulet & Olivier Teulière
* Authors: Antoine Fraboulet <antoine.fraboulet @@ free.fr>
* Olivier Teulière <ipkiss @@ gmail.com>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*****************************************************************************/
#include <boost/foreach.hpp>
#include <cstdlib> // For rand()
#include <dic.h>
#include "bag.h"
#include "debug.h"
#include "encoding.h"
INIT_LOGGER(game, Bag);
Bag::Bag(const Dictionary &iDic)
: m_dic(iDic), m_nbTiles(0)
{
BOOST_FOREACH(const Tile &tile, m_dic.getAllTiles())
{
m_tilesMap[tile] = tile.maxNumber();
m_nbTiles += tile.maxNumber();
}
}
unsigned Bag::count(const Tile &iTile) const
{
map<Tile, int>::const_iterator it = m_tilesMap.find(iTile);
if (it != m_tilesMap.end())
return it->second;
return 0;
}
unsigned Bag::getNbVowels() const
{
int v = 0;
std::pair<Tile, int> p;
BOOST_FOREACH(p, m_tilesMap)
{
if (p.first.isVowel())
v += p.second;
}
return v;
}
unsigned Bag::getNbConsonants() const
{
int c = 0;
std::pair<Tile, int> p;
BOOST_FOREACH(p, m_tilesMap)
{
if (p.first.isConsonant())
c += p.second;
}
return c;
}
void Bag::takeTile(const Tile &iTile)
{
ASSERT(contains(iTile),
"The bag does not contain the letter " + lfw(iTile.getDisplayStr()));
m_tilesMap[iTile]--;
m_nbTiles--;
}
void Bag::replaceTile(const Tile &iTile)
{
ASSERT(count(iTile) < iTile.maxNumber(),
"Cannot replace tile: " + lfw(iTile.getDisplayStr()));
m_tilesMap[iTile]++;
m_nbTiles++;
}
Tile Bag::selectRandom() const
{
return selectRandomTile(m_nbTiles, false, false);
}
Tile Bag::selectRandomVowel() const
{
return selectRandomTile(getNbVowels(), true, false);
}
Tile Bag::selectRandomConsonant() const
{
return selectRandomTile(getNbConsonants(), false, true);
}
Tile Bag::selectRandomTile(unsigned total,
bool onlyVowels, bool onlyConsonants) const
{
ASSERT(total > 0, "Not enough tiles (of the requested kind) in the bag");
int n = (int)((double)total * rand() / (RAND_MAX + 1.0));
std::pair<Tile, int> p;
BOOST_FOREACH(p, m_tilesMap)
{
if (onlyVowels && !p.first.isVowel())
continue;
if (onlyConsonants && !p.first.isConsonant())
continue;
if (n < p.second)
return p.first;
n -= p.second;
}
ASSERT(false, "We should not come here");
return Tile();
}
Bag & Bag::operator=(const Bag &iOther)
{
m_tilesMap = iOther.m_tilesMap;
m_nbTiles = iOther.m_nbTiles;
return *this;
}