from OldHangeul import text_to_jamo
import torch
import string




# 한글 자모의 로마자 대응 값
initials = {
    'ᄀ': 'K', 'ᄁ': 'KK', 'ᄂ': 'N', 'ᄃ': 'T', 'ᄄ': 'TT', 'ᄅ': 'R', 'ᄆ': 'M', 'ᄇ': 'P', 'ᄈ': 'PP', 'ᄉ': 'S', 'ᄊ': 'SS',
    'ᄋ': 'NG', 'ᄌ': 'C', 'ᄍ': 'CC', 'ᄎ': 'CH', 'ᄏ': 'KH', 'ᄐ': 'TH', 'ᄑ': 'PH', 'ᄒ': 'H'
}

medials = {
    'ᅡ': 'a', 'ᅢ': 'ae', 'ᅣ': 'ya', 'ᅤ': 'yae', 'ᅥ': 'eo', 'ᅦ': 'e', 'ᅧ': 'yeo', 'ᅨ': 'ye', 'ᅩ': 'o', 'ᅪ': 'wa',
    'ᅫ': 'wae', 'ᅬ': 'oe', 'ᅭ': 'yo', 'ᅮ': 'u', 'ᅯ': 'wo', 'ᅰ': 'we', 'ᅱ': 'wi', 'ᅲ': 'yu', 'ᅳ': 'eu', 'ᅴ': 'ui', 'ᅵ': 'i'
}

finals = {
    '': '', 'ᆨ': 'k', 'ᆩ': 'kk', 'ᆪ': 'ks', 'ᆫ': 'n', 'ᆬ': 'nj', 'ᆭ': 'nh', 'ᆮ': 't', 'ᆯ': 'r', 'ᆰ': 'rk', 'ᆱ': 'rm',
    'ᆲ': 'rb', 'ᆳ': 'rs', 'ᆴ': 'rt', 'ᆵ': 'rp', 'ᆶ': 'rh', 'ᆷ': 'm', 'ᆸ': 'p', 'ᆹ': 'ps', 'ᆺ': 's', 'ᆻ': 'ss', 'ᆼ': 'ng',
    'ᆽ': 'c', 'ᆾ': 'ch', 'ᆿ': 'kh', 'ᇀ': 'th', 'ᇁ': 'ph', 'ᇂ': 'h'
}

# 역방향 매핑을 위한 사전 생성
rev_initials = {v: k for k, v in initials.items()}
rev_medials = {v: k for k, v in medials.items()}
rev_finals = {v: k for k, v in finals.items()}

def hangul_to_roman(hangul):
    result = []
    for char in hangul:
        if '가' <= char <= '힣':
            jamos=text_to_jamo(char, compatibility=False, spacing=False)
            initial = initials[jamos[0]]
            medial = medials[jamos[2]]
            final = finals[jamos[4]] if len(jamos) == 5 else ''
            result.append(initial + medial + final)
        else:
            result.append(char)
    return ''.join(result)




all_letters = string.ascii_letters + " .,;'"
n_letters = len(all_letters)

# all_letters 로 문자의 주소 찾기, 예시 "a" = 0
def letterToIndex(letter):
    return all_letters.find(letter)


# 한 줄(이름)을  <line_length x 1 x n_letters>,
# 또는 One-Hot 문자 벡터의 Array로 변경
def lineToTensor(line):
    tensor = torch.zeros(len(line), 1, n_letters)
    for li, letter in enumerate(line):
        tensor[li][0][letterToIndex(letter)] = 1
    return tensor