95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package pokercore
|
|
|
|
import (
|
|
"fmt"
|
|
rand "math/rand"
|
|
"sync"
|
|
)
|
|
|
|
type Deck struct {
|
|
mu sync.Mutex
|
|
Cards Cards
|
|
ShuffleSeedVal int64
|
|
Dealt int
|
|
}
|
|
|
|
func NewDeckFromCards(cards Cards) *Deck {
|
|
d := new(Deck)
|
|
d.Cards = make([]Card, len(cards))
|
|
copy(d.Cards, cards)
|
|
return d
|
|
}
|
|
|
|
func NewEmptyDeck() *Deck {
|
|
d := new(Deck)
|
|
d.Cards = make([]Card, 0)
|
|
return d
|
|
}
|
|
|
|
// NewDeck returns a new deck of 52 sorted cards.
|
|
func NewDeck() *Deck {
|
|
d := NewEmptyDeck()
|
|
|
|
ranks := []Rank{
|
|
ACE, DEUCE, THREE, FOUR, FIVE,
|
|
SIX, SEVEN, EIGHT, NINE, TEN, JACK,
|
|
QUEEN, KING}
|
|
|
|
// This is the suit order used by dealers at
|
|
// The Golden Nugget in Las Vegas.
|
|
suits := []Suit{SPADE, HEART, DIAMOND, CLUB}
|
|
|
|
for _, s := range suits {
|
|
for _, r := range ranks {
|
|
d.AddCard(Card{Rank: r, Suit: s})
|
|
}
|
|
}
|
|
|
|
return d
|
|
}
|
|
|
|
func (d *Deck) AddCard(c Card) {
|
|
d.mu.Lock()
|
|
defer d.mu.Unlock()
|
|
d.Cards = append(d.Cards, c)
|
|
}
|
|
|
|
// ShuffleRandomly shuffles the deck using cryptographically random numbers.
|
|
func (d *Deck) ShuffleRandomly() {
|
|
d.mu.Lock()
|
|
defer d.mu.Unlock()
|
|
rnd := rand.New(rand.NewSource(int64(cryptoUint64())))
|
|
//FIXME(sneak) not sure if this is constant time or not
|
|
rnd.Shuffle(len(d.Cards), func(i, j int) { d.Cards[i], d.Cards[j] = d.Cards[j], d.Cards[i] })
|
|
}
|
|
|
|
// ShuffleDeterministically shuffles the deck using a deterministic seed for testing.
|
|
func (d *Deck) ShuffleDeterministically(seed int64) {
|
|
d.mu.Lock()
|
|
defer d.mu.Unlock()
|
|
r := rand.New(rand.NewSource(seed))
|
|
//FIXME(sneak) not sure if this is constant time or not
|
|
r.Shuffle(len(d.Cards), func(i, j int) { d.Cards[i], d.Cards[j] = d.Cards[j], d.Cards[i] })
|
|
}
|
|
|
|
// Deal removes n cards from the top of the deck and returns them. The Deck is
|
|
// modified in place.to remove the dealt cards, just like in real life.
|
|
func (d *Deck) Deal(n int) (output Cards) {
|
|
d.mu.Lock()
|
|
defer d.mu.Unlock()
|
|
for i := 0; i < n; i++ {
|
|
output = append(output, d.Cards[i])
|
|
}
|
|
d.Cards = d.Cards[n:]
|
|
d.Dealt += n
|
|
return output
|
|
}
|
|
|
|
func (d *Deck) String() string {
|
|
return fmt.Sprintf("Deck{%s size=%d dealt=%d}", d.Cards, d.Count(), d.Dealt)
|
|
}
|
|
|
|
func (d *Deck) Count() int {
|
|
return len(d.Cards)
|
|
}
|