SA_DiceBag.m

   1 //
   2 //  SA_DiceBag.m
   3 //
   4 //  Copyright 2016-2021 Said Achmiz.
   5 //  See LICENSE and README.md for more info.
   6
   7 #import "SA_DiceBag.h"
   8 #import <GameplayKit/GameplayKit.h>
   9
  10 /*******************************************/
  11 #pragma mark SA_DiceBag class implementation
  12 /*******************************************/
  13
  14 @implementation SA_DiceBag {
  15 //      GKRandomSource *_randomSource;
  16
  17         NSMutableDictionary <NSNumber *, GKRandomDistribution *> *_dice;
  18 }
  19
  20 -(instancetype) init {
  21         if (!(self = [super init]))
  22                 return nil;
  23
  24 //      _randomSource = [GKMersenneTwisterRandomSource new];
  25
  26         _dice = [NSMutableDictionary dictionary];
  27
  28         return self;
  29 }
  30
  31 /****************************/
  32 #pragma mark - Public methods
  33 /****************************/
  34
  35 -(NSUInteger) biggestPossibleDieSize {
  36         return NSUIntegerMax;
  37 }
  38
  39 -(NSUInteger) rollDie:(NSUInteger)dieSize {
  40 //      return [_randomSource nextIntWithUpperBound:dieSize] + 1;
  41         return [[self dieOfSize:dieSize] nextInt];
  42 }
  43
  44 -(NSArray <NSNumber *> *) rollNumber:(NSUInteger)number
  45                                                           ofDice:(NSUInteger)dieSize {
  46         return [self rollNumber:number
  47                                          ofDice:dieSize
  48                                 withOptions:0];
  49 }
  50
  51 -(NSArray <NSNumber *> *) rollNumber:(NSUInteger)number
  52                                                           ofDice:(NSUInteger)dieSize
  53                                                  withOptions:(SA_DiceRollingOptions)options {
  54         NSMutableArray *rollsArray = [NSMutableArray arrayWithCapacity:number];
  55
  56         for (NSUInteger i = 0; i < number; i++) {
  57                 NSUInteger dieRoll;
  58                 do {
  59                         dieRoll = [self rollDie:dieSize];
  60                         [rollsArray addObject:@(dieRoll)];
  61                 } while (   options & SA_DiceRollingExplodingDice
  62                                  && dieSize > 1
  63                                  && dieRoll == dieSize);
  64         }
  65
  66         return rollsArray;
  67 }
  68
  69 -(char) rollFudgeDie {
  70         NSInteger d3roll = [self rollDie:3];
  71         return (char) (d3roll - 2);
  72 }
  73
  74 -(NSArray <NSNumber *> *) rollFudgeDice:(NSUInteger)number {
  75         NSMutableArray *rollsArray = [NSMutableArray arrayWithCapacity:number];
  76
  77         for (NSUInteger i = 0; i < number; i++) {
  78                 [rollsArray addObject:@([self rollFudgeDie])];
  79         }
  80
  81         return rollsArray;
  82 }
  83
  84 /****************************/
  85 #pragma mark - Helper methods
  86 /****************************/
  87
  88 -(GKRandomDistribution *) dieOfSize:(NSUInteger) dieSize {
  89         if (_dice[@(dieSize)] == nil)
  90                 _dice[@(dieSize)] = [[GKRandomDistribution alloc] initWithRandomSource:[GKMersenneTwisterRandomSource new]
  91                                                                                                                                    lowestValue:1
  92                                                                                                                                   highestValue:dieSize];
  93
  94         return _dice[@(dieSize)];
  95 }
  96
  97 @end