#!/usr/bin/python
# -*- coding: utf-8 -*-
def MutableNamedTuple(name, fields):
  def tostr(value):
    if type(value) == unicode:
      return "'%s'" % (value.encode('utf8'))
    return repr(value)
  class TemporaryClass(object):
    __name__ = name
    def __init__(self, *args):
      assert(len(args) == len(fields))
      for (key, value) in zip(fields, args):
        self.__dict__[key] = value
    def __str__(self):
      return '%s(%s)' % (name, ', '.join(
          tostr(self.__dict__[key]) for key in fields))
  return TemporaryClass

RADICAL_VARIANTS_TO_SKIP = (u'𠆢', u'𠘨')
SIMPLIFIED_RADICALS_TO_SKIP = (27,)

def in_cjk_block(character):
  if not (len(character) == 1 and 0x4e00 <= ord(character) <= 0x9fff):
    print '%s is U+%s' % (character, hex(ord(character))[2:].upper())
    return False
  return True

with open('scripts/glyphs') as f:
  glyphs = f.readlines()[0].strip().decode('utf8')
  assert(all(in_cjk_block(glyph) for glyph in glyphs))
  glyph_set = set(glyphs)
assert(len(glyphs) == len(glyph_set) == 6763)

ArchRadical = MutableNamedTuple(
    'Radical', ['number', 'character', 'definition', 'pinyin', 'strokes'])

with open('scripts/arch_radicals') as f:
  rows = [line.strip().decode('utf8').split('  ') for line in f.readlines()]
  arch_radicals = [ArchRadical(*row) for row in rows]
  arch_radical_map = dict((radical.character, radical)
                          for radical in arch_radicals)
assert(len(arch_radicals) == len(arch_radical_map) == 214)

WikiRadical = MutableNamedTuple(
    'WikiRadical', ['number', 'character', 'strokes', 'pinyin',
                    'unused1', 'unused2', 'unused3', 'definition',
                    'frequency', 'simplified', 'examples'])

with open('scripts/wiki_radicals') as f:
  rows = [line.strip().decode('utf8').split('\t') for line in f.readlines()[2:]]
  wiki_radicals = [WikiRadical(*row) for row in rows]
  wiki_radical_map = dict((radical.character, radical)
                          for radical in wiki_radicals)
assert(len(wiki_radicals) == len(wiki_radical_map) == 214)

print 'Homogenizing Arch radicals:'
for radical in arch_radicals:
  radical.number = int(radical.number)
  radical.variants = ''
  if ' ' in radical.strokes:
    index = radical.strokes.find(' ')
    radical.variants = radical.strokes[index + 1:]
    radical.strokes = radical.strokes[:index]
  radical.strokes = int(radical.strokes)
  if radical.variants.startswith('('):
    assert(radical.variants.endswith(')'))
    radical.traditional = radical.variants[1:-1]
    radical.variants = ''
  else:
    radical.traditional = None
  if radical.variants:
    radical.variants = tuple(sorted(radical.variants.split()))
  else:
    radical.variants = ()
  in_cjk_block(radical.character)
  if radical.traditional is not None:
    in_cjk_block(radical.traditional)
  [in_cjk_block(variant) for variant in radical.variants]
  assert(radical.definition)
  assert(radical.pinyin)

print 'Homogenizing Wiki radicals:'
for radical in wiki_radicals:
  radical.number = int(radical.number)
  radical.strokes = int(radical.strokes)
  radical.variants = ()
  if ' ' in radical.character:
    index = radical.character.find(' ')
    assert(radical.character[index + 1] == '(')
    assert(radical.character[-1] == ')')
    radical.variants = radical.character[index + 2:-1].split(',')
    radical.variants = [variant.strip() for variant in radical.variants
                        if variant.strip() not in RADICAL_VARIANTS_TO_SKIP]
    radical.variants = tuple(sorted(radical.variants))
    radical.character = radical.character[:index]
  radical.traditional = None
  if radical.simplified and radical.number not in SIMPLIFIED_RADICALS_TO_SKIP:
    if radical.simplified.startswith('(pr. '):
      radical.pinyin = radical.simplified[5:-1]
    else:
      radical.traditional = radical.character
      radical.character = radical.simplified
  in_cjk_block(radical.character)
  if radical.traditional is not None:
    in_cjk_block(radical.traditional)
  [in_cjk_block(variant) for variant in radical.variants]
  assert(radical.definition)
  assert(radical.pinyin)

for (arch_radical, wiki_radical) in zip(arch_radicals, wiki_radicals):
  assert(arch_radical.number == wiki_radical.number)
  if arch_radical.character != wiki_radical.character:
    print 'Different characters for radical %s: %s vs. %s' % (
        arch_radical.number, arch_radical.character, wiki_radical.character)
  if arch_radical.definition != wiki_radical.definition:
    print 'Different definitions for radical %s: "%s" vs. "%s"' % (
        arch_radical.number, arch_radical.definition, wiki_radical.definition)
  if arch_radical.pinyin != wiki_radical.pinyin:
    print 'Different pronunciation for radical %s: "%s" vs. "%s"' % (
        arch_radical.number, arch_radical.pinyin, wiki_radical.pinyin)
  if arch_radical.traditional != wiki_radical.traditional:
    print 'Different variants for radical %s: "%s" vs. "%s"' % (
        arch_radical.number, arch_radical.traditional, wiki_radical.traditional)
  if arch_radical.variants != wiki_radical.variants:
    print 'Different variants for radical %s: (%s) vs. (%s)' % (
        arch_radical.number,
        ', '.join(variant.encode('utf8') for variant in arch_radical.variants),
        ', '.join(variant.encode('utf8') for variant in wiki_radical.variants))

Decomposition = MutableNamedTuple(
  'Decomposition', ['character', 'strokes', 'type', 'part1', 'strokes1',
                    'warning1', 'part2', 'strokes2', 'warning2',
                    'cangjie', 'radical'])

with open('data/decomposition/data') as f:
  lines = [line for line in f.readlines() if line.startswith('\t')]
  rows = [line.strip().decode('utf8').split('\t') for line in lines]
  decompositions = [Decomposition(*row)  for row in rows if len(row) == 11]
  decomposition_map = dict((decomposition.character, decomposition)
                            for decomposition in decompositions)
assert(len(decomposition_map) == 21166)

#print 'Checking decompositions:'
for glyph in glyphs:
  assert(glyph in decomposition_map), 'Missing glyph: %s' % (glyph,)
  decomposition = decomposition_map[glyph]
  for part in decomposition.part1 + decomposition.part2:
    if part != '*' and part not in glyph_set:
      #print 'Extra glyph needed for %s: %s' % (glyph, part)
      #in_cjk_block(part)
      continue