#include "wc.h"

#include "johab.h"

#include "wtf.h"

#ifdef USE_UNICODE

#include "ucs.h"

#endif

#define C0 WC_JOHAB_MAP_C0

#define GL WC_JOHAB_MAP_GL

#define C1 WC_JOHAB_MAP_C1

#define GH WC_JOHAB_MAP_GH

#define GB WC_JOHAB_MAP_GB

#define JJ WC_JOHAB_MAP_JJ

#define JB WC_JOHAB_MAP_JB

#define HB WC_JOHAB_MAP_HB

#define CJ WC_JOHAB_MAP_CJ

#define CB WC_JOHAB_MAP_CB

/*

  00-1F 20-30 31-40 41-7E 7F 80 81-83 84-90 91-D3 D4-D7 D8-DE DF E0-F9 FA-FE FF

  C0    GL    GL    GL    C0 -  -     J     J     -     H     -  H     -     -

  -     -     J     B     -  -  J     J     B     B     B     B  B     B     -

  C0    GL    GH    GB    C0 C1 CJ    JJ    JB    CB    HB    CB HB    CB    C1 

*/

wc_uint8 WC_JOHAB_MAP[ 0x100 ] = {

    C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0,

    C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0, C0,

/*  20 */

    GL, GL, GL, GL, GL, GL, GL, GL, GL, GL, GL, GL, GL, GL, GL, GL,

/*  30  31 */

    GL, GH, GH, GH, GH, GH, GH, GH, GH, GH, GH, GH, GH, GH, GH, GH,

/*  40  41 */

    GH, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, 

    GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, 

    GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, 

    GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, GB, C0,

/*  80          83  84 */

    C1, CJ, CJ, CJ, JJ, JJ, JJ, JJ, JJ, JJ, JJ, JJ, JJ, JJ, JJ, JJ,

/*  90  91 */

    JJ, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, 

    JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, 

    JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, 

    JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, JB, 

/*              D3  D4          D7  D8                          DF */

    JB, JB, JB, JB, CB, CB, CB, CB, HB, HB, HB, HB, HB, HB, HB, CB, 

    HB, HB, HB, HB, HB, HB, HB, HB, HB, HB, HB, HB, HB, HB, HB, HB,

/*                                      F9  FA              FE  FF */

    HB, HB, HB, HB, HB, HB, HB, HB, HB, HB, CB, CB, CB, CB, CB, C1,

};

static wc_uint8 johab1_N_map[ 3 ][ 32 ] = {

  { 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,

   15,16,17,18,19, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 },

  { 0, 0, 0, 1, 2, 3, 4, 5, 0, 0, 6, 7, 8, 9,10,11,

    0, 0,12,13,14,15,16,17, 0, 0,18,19,20,21, 0, 0 },

  { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,15,

   16,17, 0,18,19,20,21,22,23,24,25,26,27,28, 0, 0 }

};

static wc_uint8 N_johab1_map[ 3 ][ 32 ] = {

  { 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,15,16,17,

   18,19,20, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 },

  { 3, 4, 5, 6, 7,10,11,12,13,14,15,18,19,20,21,22,

   23,26,27,28,29, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 },

  { 1, 2, 3, 4, 5, 6, 7, 8, 9,10,11,12,13,14,15,16,

   17,19,20,21,22,23,24,25,26,27,28,29, 0, 0, 0, 0 }

};

wc_wchar_t

wc_johab_to_ksx1001(wc_wchar_t cc)

{

#ifdef USE_UNICODE

    static wc_table *t = NULL;

#endif

    switch (cc.ccs) {

    case WC_CCS_JOHAB:

  return wc_johab_to_ksx1001(wc_johab_to_cs128w(cc));

    case WC_CCS_JOHAB_1:

    case WC_CCS_JOHAB_2:

#ifdef USE_UNICODE

  if (WcOption.ucs_conv) {

      if (t == NULL)

    t = wc_get_ucs_table(WC_CCS_KS_X_1001);

      cc = wc_any_to_any(cc, t);

  } else

#endif

      cc.ccs = WC_CCS_UNKNOWN_W;

  break;

    case WC_CCS_JOHAB_3:

  if (cc.code >= 0x2121)

      cc.ccs = WC_CCS_KS_X_1001;

  else

      cc.ccs = WC_CCS_UNKNOWN_W;

  break;

    }

    return cc;

}

wc_wchar_t

wc_ksx1001_to_johab(wc_wchar_t cc)

{

    cc.code &= 0x7f7f;

    if ((cc.code >= 0x2121 && cc.code <  0x2421) ||

  (cc.code >  0x2453 && cc.code <= 0x2C7E) ||

  (cc.code >= 0x4A21 && cc.code <= 0x7D7E)) {

  cc.ccs = WC_CCS_JOHAB_3;

  return cc;

    }

#ifdef USE_UNICODE

    if (WcOption.ucs_conv)

  cc = wc_ucs_to_johab(wc_any_to_ucs(cc));

    else

#endif

  cc.ccs = WC_CCS_UNKNOWN_W;

    return cc;

}

#ifdef USE_UNICODE

wc_wchar_t

wc_ucs_to_johab(wc_uint32 ucs)

{

    wc_table *t;

    wc_wchar_t cc;

    if (ucs >= WC_C_UCS2_HANGUL && ucs <= WC_C_UCS2_HANGUL_END) {

  ucs -= WC_C_UCS2_HANGUL;

  cc.code = WC_N_JOHAB1(ucs);

  cc.ccs = WC_CCS_JOHAB;

    } else if (ucs >= 0x3131 && ucs <= 0x3163) {

  t = wc_get_ucs_table(WC_CCS_JOHAB_2);

  cc = wc_ucs_to_any(ucs, t);

    } else {

  t = wc_get_ucs_table(WC_CCS_JOHAB_3);

  cc = wc_ucs_to_any(ucs, t);

    }

    return cc;

}

#endif

wc_uint32

wc_johab1_to_N(wc_uint32 code)

{

    wc_uint32 a, b, c;

    a = johab1_N_map[0][(code >> 10) & 0x1F];

    b = johab1_N_map[1][(code >> 5)  & 0x1F];

    c = johab1_N_map[2][ code        & 0x1F];

    if (a && b && c)

  return ((a - 1) * 21 + (b - 1)) * 28 + (c - 1);

    return WC_C_JOHAB_ERROR;

}

wc_uint32

wc_N_to_johab1(wc_uint32 code)

{

    wc_uint32 a, b, c;

    a = N_johab1_map[0][(code / 28) / 21 & 0x1F];

    b = N_johab1_map[1][(code / 28) % 21 & 0x1F];

    c = N_johab1_map[2][ code % 28       & 0x1F];

    return 0x8000 | (a << 10) | (b << 5) | c;

}

/* 0x1F21 - 0x2C7E, 0x4A21 - 0x7C7E

  (0x1F21 - 0x207E are not in KS X 1001) */

#define johab3_to_ksx1001(ub, lb) \

{ \

    if (ub < 0xe0) { \

  ub = ((ub - 0xd8) << 1) + 0x1f; \

    } else { \

  ub = ((ub - 0xe0) << 1) + 0x4a; \

    } \

    if (lb < 0xa1) { \

  lb -= (lb < 0x91) ? 0x10 : 0x22; \

    } else { \

  ub++; \

  lb -= 0x80; \

    } \

}

#define ksx1001_to_johab3(ub, lb) \

{ \

    if (ub < 0x4a) { \

  ub -= 0x1f; \

  lb += (ub & 0x1) ? 0x80 : ((lb < 0x6f) ? 0x10 : 0x22); \

  ub = (ub >> 1) + 0xd8; \

    } else { \

  ub -= 0x4a; \

  lb += (ub & 0x1) ? 0x80 : ((lb < 0x6f) ? 0x10 : 0x22); \

  ub = (ub >> 1) + 0xe0; \

    } \

}

wc_wchar_t

wc_johab_to_cs128w(wc_wchar_t cc)

{

    wc_uint32 n;

    wc_uchar ub, lb;

    if (cc.code < 0xD800) {

  n = WC_JOHAB1_N(cc.code);

  if (n != WC_C_JOHAB_ERROR) {

      cc.code = WC_N_CS94x128(n);

      cc.ccs = WC_CCS_JOHAB_1;

  } else {

      n = WC_JOHAB2_N(cc.code);

      cc.code = WC_N_CS128W(n);

      cc.ccs = WC_CCS_JOHAB_2;

  }

    } else {

  ub = cc.code >> 8;

  lb = cc.code & 0xff;

  johab3_to_ksx1001(ub, lb);

  cc.code = ((wc_uint32)ub << 8) | lb;

  cc.ccs = WC_CCS_JOHAB_3;

    }

    return cc;

}

wc_wchar_t

wc_cs128w_to_johab(wc_wchar_t cc)

{

    wc_uint32 n;

    wc_uchar ub, lb;

    switch (cc.ccs) {

    case WC_CCS_JOHAB_1:

  n = WC_CS94x128_N(cc.code);

  cc.code = WC_N_JOHAB1(n);

  break;

    case WC_CCS_JOHAB_2:

  n = WC_CS128W_N(cc.code);

  cc.code = WC_N_JOHAB2(n);

  break;

    case WC_CCS_JOHAB_3:

  ub = (cc.code >> 8) & 0x7f;

  lb = cc.code & 0x7f;

  ksx1001_to_johab3(ub, lb);

  cc.code = ((wc_uint32)ub << 8) | lb;

    }

    cc.ccs = WC_CCS_JOHAB;

    return cc;

}

Str

wc_conv_from_johab(Str is, wc_ces ces)

{

    Str os;

    wc_uchar *sp = (wc_uchar *)is->ptr;

    wc_uchar *ep = sp + is->length;

    wc_uchar *p;

    int state = WC_JOHAB_NOSTATE;

    for (p = sp; p < ep && *p < 0x80; p++)

        ;

    if (p == ep)

  return is;

    os = Strnew_size(is->length);

    if (p > sp)

  Strcat_charp_n(os, is->ptr, (int)(p - sp));

    for (; p < ep; p++) {

  switch (state) {

  case WC_JOHAB_NOSTATE:

      switch (WC_JOHAB_MAP[*p] & WC_JOHAB_MAP_1) {

      case WC_JOHAB_MAP_UJ:

    state = WC_JOHAB_HANGUL1;

    break;

      case WC_JOHAB_MAP_UH:

    state = WC_JOHAB_HANJA1;

    break;

      case WC_JOHAB_MAP_C1:

    wtf_push_unknown(os, p, 1);

    break;

      default:

    Strcat_char(os, (char)*p);

    break;

      }

      break;

  case WC_JOHAB_HANGUL1:

      if (WC_JOHAB_MAP[*p] & WC_JOHAB_MAP_LJ) 

    wtf_push(os, WC_CCS_JOHAB, ((wc_uint32)*(p-1) << 8) | *p);

      else

    wtf_push_unknown(os, p-1, 2);

      state = WC_JOHAB_NOSTATE;

      break;

  case WC_JOHAB_HANJA1:

      if (WC_JOHAB_MAP[*p] & WC_JOHAB_MAP_LH)

    wtf_push(os, WC_CCS_JOHAB, ((wc_uint32)*(p-1) << 8) | *p);

      else

    wtf_push_unknown(os, p-1, 2);

      state = WC_JOHAB_NOSTATE;

      break;

  }

    }

    switch (state) {

    case WC_JOHAB_HANGUL1:

    case WC_JOHAB_HANJA1:

  wtf_push_unknown(os, p-1, 1);

  break;

    }

    return os;

}

void

wc_push_to_johab(Str os, wc_wchar_t cc, wc_status *st)

{

  while (1) {

    switch (cc.ccs) {

    case WC_CCS_US_ASCII:

  Strcat_char(os, (char)cc.code);

  return;

    case WC_CCS_JOHAB_1:

    case WC_CCS_JOHAB_2:

    case WC_CCS_JOHAB_3:

  cc = wc_cs128w_to_johab(cc);

    case WC_CCS_JOHAB:

  Strcat_char(os, (char)(cc.code >> 8));

  Strcat_char(os, (char)(cc.code & 0xff));

  return;

    case WC_CCS_KS_X_1001:

  cc = wc_ksx1001_to_johab(cc);

  continue;

    case WC_CCS_UNKNOWN_W:

  if (!WcOption.no_replace)

      Strcat_charp(os, WC_REPLACE_W);

  return;

    case WC_CCS_UNKNOWN:

  if (!WcOption.no_replace)

      Strcat_charp(os, WC_REPLACE);

  return;

    default:

#ifdef USE_UNICODE

  if (WcOption.ucs_conv)

      cc = wc_any_to_any_ces(cc, st);

  else

#endif

      cc.ccs = WC_CCS_IS_WIDE(cc.ccs) ? WC_CCS_UNKNOWN_W : WC_CCS_UNKNOWN;

  continue;

    }

  }

}

Str

wc_char_conv_from_johab(wc_uchar c, wc_status *st)

{

    static Str os;

    static wc_uchar johabu;

    if (st->state == -1) {

  st->state = WC_JOHAB_NOSTATE;

  os = Strnew_size(8);

    }

    switch (st->state) {

    case WC_JOHAB_NOSTATE:

  switch (WC_JOHAB_MAP[c] & WC_JOHAB_MAP_1) {

  case WC_JOHAB_MAP_UJ:

      johabu = c;

      st->state = WC_JOHAB_HANGUL1;

      return NULL;

  case WC_JOHAB_MAP_UH:

      johabu = c;

      st->state = WC_JOHAB_HANJA1;

      return NULL;

  case WC_JOHAB_MAP_C1:

      break;

  default:

      Strcat_char(os, (char)c);

      break;

  }

  break;

    case WC_JOHAB_HANGUL1:

  if (WC_JOHAB_MAP[c] & WC_JOHAB_MAP_LJ)

      wtf_push(os, WC_CCS_JOHAB, ((wc_uint32)johabu << 8) | c);

  break;

    case WC_JOHAB_HANJA1:

  if (WC_JOHAB_MAP[c] & WC_JOHAB_MAP_LH)

      wtf_push(os, WC_CCS_JOHAB, ((wc_uint32)johabu << 8) | c);

  break;

    }

    st->state = -1;

    return os;

}