firefight's simulator

#include <stdlib.h>
#include <string.h>

typedef struct {
  long double hitChance;
  unsigned char maxDamage;
  unsigned char hitPoints;
} PlayerInfo;

typedef struct {
  long double win;
  long double lose;
  long double draw;
} Odds;

const Odds error = {0, 0, 0};

static void turnsOdds(unsigned char, long double *, long double **, unsigned *, long double);
static void hitsOdds(unsigned char, long double, long double *);
static void multiplyVector(long double *, long double *, long double);
static void addVector(long double *, long double *);
static long double totalOdds(long double *);
static long double currentError(long double *, unsigned char);
static void expand(long double **, unsigned, unsigned);
static Odds calculateFor(long double *, long double *, unsigned);

Odds calculateOdds (PlayerInfo p1, PlayerInfo p2, long double threshold) {
  unsigned t1, t2;
  long double * odds1;
  long double * odds2;
  long double hitodds1[12];
  long double hitodds2[12];
  unsigned turns;
  if ((p1.hitChance < 0) || (p1.hitChance > 1)) return error;
  if ((p2.hitChance < 0) || (p2.hitChance > 1)) return error;
  if ((!p1.maxDamage) || (p1.maxDamage > 11)) return error;
  if ((!p2.maxDamage) || (p2.maxDamage > 11)) return error;
  if (p1.hitPoints > 99) return error;
  if (p2.hitPoints > 99) return error;
  if ((threshold <= 0) || (threshold >= 1)) return error;
  hitsOdds(p1.maxDamage, p1.hitChance, hitodds1);
  hitsOdds(p2.maxDamage, p2.hitChance, hitodds2);
  turnsOdds(p2.hitPoints, hitodds1, &odds1, &t1, threshold);
  turnsOdds(p1.hitPoints, hitodds2, &odds2, &t2, threshold);
  if (t1 < t2) {
    expand(&odds1, t1, t2);
    turns = t2;
  } else if (t2 < t1) {
    expand(&odds2, t2, t1);
    turns = t1;
  } else
    turns = t1;
  Odds result = calculateFor(odds1, odds2, turns);
  free(odds1);
  free(odds2);
  long double totalError = 1;
  totalError -= result.win;
  totalError -= result.lose;
  totalError -= result.draw;
  if ((totalError * totalError) > (threshold * threshold * 10)) return calculateOdds(p1, p2, threshold / 10);
  result.draw += totalError;
  return result;
}

static void turnsOdds (unsigned char opponentHP, long double * hitOdds, long double ** result, unsigned * resultSize, long double threshold) {
  unsigned turns = 0;
  long double * turnOdds = malloc(sizeof(long double));
  *turnOdds = 1;
  long double damageDealtA[111];
  long double damageDealtB[111];
  unsigned char current = 0;
  long double * damageDealt;
  long double * damageDealtPrev;
  memset(damageDealtA, 0, 111 * sizeof(long double));
  *damageDealtA = 1;
  unsigned currentTurn;
  long double curDamage[12];
  unsigned char curHP;
  do {
    currentTurn = turns;
    turns += 20;
    turnOdds = realloc(turnOdds, sizeof(long double) * (turns + 1));
    for (currentTurn ++; currentTurn <= turns; currentTurn ++) {
      damageDealtPrev = current ? damageDealtB : damageDealtA;
      damageDealt = current ? damageDealtA : damageDealtB;
      memset(damageDealt, 0, 111 * sizeof(long double));
      for (curHP = 0; curHP < opponentHP; curHP ++) {
        multiplyVector(curDamage, hitOdds, damageDealtPrev[curHP]);
        addVector(damageDealt + curHP, curDamage);
      }
      turnOdds[currentTurn] = totalOdds(damageDealt + opponentHP);
      current = !current;
    }
    damageDealt = current ? damageDealtA : damageDealtB;
    *turnOdds = currentError(damageDealt, opponentHP);
  } while (*turnOdds >= threshold);
  *result = turnOdds;
  *resultSize = turns;
}

static void hitsOdds (unsigned char maxDamage, long double hitChance, long double * result) {
  *result = 1 - hitChance;
  long double x = hitChance / maxDamage;
  unsigned char pos;
  for (pos = 1; pos <= maxDamage; pos ++) result[pos] = x;
  for (; pos <= 11; pos ++) result[pos] = 0;
}

static void multiplyVector (long double * result, long double * src, long double value) {
  unsigned char pos;
  for (pos = 0; pos < 12; pos ++)
    result[pos] = src[pos] * value;
}

static void addVector (long double * dst, long double * src) {
  unsigned char pos;
  for (pos = 0; pos < 12; pos ++)
    dst[pos] += src[pos];
}

static long double totalOdds (long double * damageOdds) {
  long double total = 0;
  unsigned char pos;
  for (pos = 0; pos < 12; pos ++)
    total += damageOdds[pos];
  return total;
}

static long double currentError (long double * damageOdds, unsigned char maxDamage) {
  long double total = 0;
  unsigned char pos;
  for (pos = 0; pos < maxDamage; pos ++)
    total += damageOdds[pos];
  return total;
}

static void expand (long double ** turnOdds, unsigned from, unsigned to) {
  *turnOdds = realloc(*turnOdds, sizeof(long double) * (to + 1));
  unsigned pos;
  for (pos = from + 1; pos <= to; pos ++)
    (*turnOdds)[pos] = 0;
}

static Odds calculateFor (long double * turnOdds1, long double * turnOdds2, unsigned turns) {
  long double r0 = 0, r1 = 0, r2 = 0;
  unsigned turn1, turn2;
  for (turn1 = 1; turn1 <= turns; turn1 ++) {
    for (turn2 = 1; turn2 < turn1; turn2 ++)
      r2 += turnOdds1[turn1] * turnOdds2[turn2];
    r0 += turnOdds1[turn1] * turnOdds2[turn1];
    for (turn2 = turn1 + 1; turn2 <= turns; turn2 ++)
      r1 += turnOdds1[turn1] * turnOdds2[turn2];
    r1 += turnOdds1[turn1] * (*turnOdds2);
  }
  for (turn2 = 1; turn2 <= turns; turn2 ++)
    r2 += turnOdds2[turn2] * (*turnOdds1);
  r0 += (*turnOdds1) * (*turnOdds2);
  return (Odds) {.win = r1, .lose = r2, .draw = r0};
}