/*
  Copyright (c) 2011-2015 Ronald de Man
*/

#ifndef TBCORE_H
#  define TBCORE_H

#  ifndef __WIN32__
#    include <pthread.h>
#    define SEP_CHAR ':'
#    define FD int
#    define FD_ERR -1
#  else
#    include <windows.h>
#    define SEP_CHAR ';'
#    define FD HANDLE
#    define FD_ERR INVALID_HANDLE_VALUE
#  endif

#  ifdef TB_HAVE_THREADS
#    ifndef __WIN32__
#      define LOCK_T pthread_mutex_t
#      define LOCK_INIT(x) pthread_mutex_init(&(x), NULL)
#      define LOCK(x) pthread_mutex_lock(&(x))
#      define UNLOCK(x) pthread_mutex_unlock(&(x))
#    else
#      define LOCK_T HANDLE
#      define LOCK_INIT(x) do { x = CreateMutex(NULL, FALSE, NULL); } while (0)
#      define LOCK(x) WaitForSingleObject(x, INFINITE)
#      define UNLOCK(x) ReleaseMutex(x)
#    endif
#  else                         /* !TB_HAVE_THREADS */
#    define LOCK_T          int
#    define LOCK_INIT(x)
                     /* NOP */
#    define LOCK(x)
                /* NOP */
#    define UNLOCK(x)
                  /* NOP */
#  endif

#  define WDLSUFFIX ".rtbw"
#  define DTZSUFFIX ".rtbz"
#  define WDLDIR "RTBWDIR"
#  define DTZDIR "RTBZDIR"
#  define TBPIECES 6

#  define WDL_MAGIC 0x5d23e871
#  define DTZ_MAGIC 0xa50c66d7

#  define TBHASHBITS 10

typedef unsigned long long uint64;
typedef unsigned int uint32;
typedef unsigned char ubyte;
typedef unsigned short ushort;

struct TBHashEntry;

#  ifdef DECOMP64
typedef uint64 base_t;
#  else
typedef uint32 base_t;
#  endif

struct PairsData {
  char *indextable;
  ushort *sizetable;
  ubyte *data;
  ushort *offset;
  ubyte *symlen;
  ubyte *sympat;
  int blocksize;
  int idxbits;
  int min_len;
  base_t base[1];               // C++ complains about base[]...
};

struct TBEntry {
  char *data;
  uint64 key;
  uint64 mapping;
  ubyte ready;
  ubyte num;
  ubyte symmetric;
  ubyte has_pawns;
} __attribute__ ((__may_alias__));

struct TBEntry_piece {
  char *data;
  uint64 key;
  uint64 mapping;
  ubyte ready;
  ubyte num;
  ubyte symmetric;
  ubyte has_pawns;
  ubyte enc_type;
  struct PairsData *precomp[2];
  int factor[2][TBPIECES];
  ubyte pieces[2][TBPIECES];
  ubyte norm[2][TBPIECES];
};

struct TBEntry_pawn {
  char *data;
  uint64 key;
  uint64 mapping;
  ubyte ready;
  ubyte num;
  ubyte symmetric;
  ubyte has_pawns;
  ubyte pawns[2];
  struct {
    struct PairsData *precomp[2];
    int factor[2][TBPIECES];
    ubyte pieces[2][TBPIECES];
    ubyte norm[2][TBPIECES];
  } file[4];
};

struct DTZEntry_piece {
  char *data;
  uint64 key;
  uint64 mapping;
  ubyte ready;
  ubyte num;
  ubyte symmetric;
  ubyte has_pawns;
  ubyte enc_type;
  struct PairsData *precomp;
  int factor[TBPIECES];
  ubyte pieces[TBPIECES];
  ubyte norm[TBPIECES];
  ubyte flags;                  // accurate, mapped, side
  ushort map_idx[4];
  ubyte *map;
};

struct DTZEntry_pawn {
  char *data;
  uint64 key;
  uint64 mapping;
  ubyte ready;
  ubyte num;
  ubyte symmetric;
  ubyte has_pawns;
  ubyte pawns[2];
  struct {
    struct PairsData *precomp;
    int factor[TBPIECES];
    ubyte pieces[TBPIECES];
    ubyte norm[TBPIECES];
  } file[4];
  ubyte flags[4];
  ushort map_idx[4][4];
  ubyte *map;
};

struct TBHashEntry {
  uint64 key;
  struct TBEntry *ptr;
};

struct DTZTableEntry {
  uint64 key1;
  uint64 key2;
  struct TBEntry *entry;
};

#endif