#!/usr/bin/env python3

# --- Modules --- #
import drawBot as dB
from pathlib import Path
from collections import defaultdict

# --- Constants --- #
DICT_FOLDER = Path('dictionaries')

# --- Objects & Methods --- #
def calcWordsWidth(words, fontName):
    dB.font(fontName, 1)
    width_2_words = defaultdict(list)
    for eachWord in words:
        txtWdt, txtHgt = dB.textSize(eachWord)
        width_2_words[txtWdt].append(eachWord)
    return width_2_words


# --- Variables --- #
fontName = 'Skia'
language = 'italian'

# --- Instructions --- #
if __name__ == '__main__':
    from loadWords_v4 import loadWords
    words = loadWords(DICT_FOLDER / f'{language}.txt')
    width_2_words = calcWordsWidth(words, fontName)
    print(f'words: {len(words)}')
    # words: 16458
    print(f'groups: {len(width_2_words)}')
    # groups: 6813
    print(f'words/groups ratio: {len(words)/len(width_2_words):.3f}')
    # words/groups ratio: 2.416