130 lines
3.0 KiB
C++
130 lines
3.0 KiB
C++
#include "../include/DeckDataWrapper.h"
|
|
#include "../include/MTGDeck.h"
|
|
|
|
DeckDataWrapper::DeckDataWrapper(MTGDeck * deck){
|
|
parent = deck;
|
|
for (int i = 0; i <= Constants::MTG_NB_COLORS; i++){
|
|
colors[i] = 0;
|
|
}
|
|
for (int i = 0; i < deck->totalCards(); i++){
|
|
MTGCard * card = deck->_(i);
|
|
Add(card);
|
|
}
|
|
currentposition = 0;
|
|
currentColor = -1;
|
|
}
|
|
|
|
void DeckDataWrapper::save(){
|
|
parent->removeAll();
|
|
map<MTGCard *,int,Cmp1>::iterator it;
|
|
for ( it=cards.begin() ; it != cards.end(); it++ ){
|
|
MTGCard * current = (*it).first;
|
|
for (int i = 0; i < (*it).second; i++){
|
|
parent->add(current);
|
|
}
|
|
}
|
|
parent->save();
|
|
}
|
|
|
|
|
|
DeckDataWrapper::~DeckDataWrapper(){
|
|
SAFE_DELETE(parent);
|
|
}
|
|
|
|
void DeckDataWrapper::updateCounts(MTGCard * card, int removed){
|
|
if (!card){
|
|
for (int i = 0; i < Constants::MTG_NB_COLORS+1; i++){
|
|
colors[i] = 0;
|
|
}
|
|
map<MTGCard *,int,Cmp1>::iterator it;
|
|
for ( it=cards.begin() ; it != cards.end(); it++ ){
|
|
MTGCard * current = (*it).first;
|
|
colors[Constants::MTG_NB_COLORS] += (*it).second;
|
|
for (int i = 0; i < Constants::MTG_NB_COLORS; i++){
|
|
if (current->hasColor(i)) colors[i]+=(*it).second;
|
|
}
|
|
}
|
|
}else{
|
|
int increment = 1;
|
|
if (removed) increment = -1;
|
|
colors[Constants::MTG_NB_COLORS] += increment;
|
|
for (int i = 0; i < Constants::MTG_NB_COLORS; i++){
|
|
if (card->hasColor(i)) colors[i]+=increment;
|
|
}
|
|
}
|
|
}
|
|
|
|
int DeckDataWrapper::Add(MTGCard * card){
|
|
if(cards.find(card) == cards.end()){
|
|
cards[card] = 1;
|
|
}else{
|
|
cards[card]++;
|
|
}
|
|
updateCounts(card);
|
|
return cards[card];
|
|
}
|
|
|
|
int DeckDataWrapper::Remove(MTGCard * card){
|
|
if(cards.find(card) == cards.end() || cards[card] == 0) return 0;
|
|
cards[card]--;
|
|
updateCounts(card,1);
|
|
return 1;
|
|
}
|
|
|
|
MTGCard * DeckDataWrapper::getNext(MTGCard * previous, int color){
|
|
map<MTGCard *,int,Cmp1>::iterator it;
|
|
|
|
it = cards.find(previous);
|
|
|
|
while(1){
|
|
if (it == cards.end()){
|
|
it = cards.begin();
|
|
}else{
|
|
it++;
|
|
}
|
|
if (it == cards.end()) return NULL;
|
|
MTGCard * card = (*it).first;
|
|
if (card == previous) return NULL;
|
|
if ((*it).second >0 && (color ==-1 || card->hasColor(color))){
|
|
return card;
|
|
}
|
|
}
|
|
}
|
|
|
|
MTGCard * DeckDataWrapper::getPrevious(MTGCard * next, int color){
|
|
map<MTGCard *,int,Cmp1>::iterator it;
|
|
it = cards.find(next);
|
|
|
|
while(1){
|
|
if (it == cards.begin()){
|
|
it = cards.end();
|
|
}else{
|
|
it--;
|
|
}
|
|
if (it == cards.end()) return NULL;
|
|
MTGCard * card = (*it).first;
|
|
if (card == next) return NULL;
|
|
if ((*it).second >0 && (color ==-1 || card->hasColor(color))){
|
|
return card;
|
|
}
|
|
}
|
|
}
|
|
|
|
void DeckDataWrapper::updateCurrentPosition(MTGCard * currentCard, int color){
|
|
currentposition = 0;
|
|
MTGCard * next = getNext(NULL,color);
|
|
while (next){
|
|
currentposition+=cards[next];
|
|
if (next == currentCard){
|
|
next = NULL;
|
|
}else{
|
|
next = getNext(next,color);
|
|
}
|
|
}
|
|
}
|
|
|
|
int DeckDataWrapper::getCount(int color){
|
|
if (color == -1) return colors[Constants::MTG_NB_COLORS];
|
|
return colors[color];
|
|
}
|