"""
This code adds Python signatures to the docs.

TODO:
* clarify when there are several C++ signatures corresponding to a single Python function.
    i.e: calcHist():
    http://docs.opencv.org/3.2.0/d6/dc7/group__imgproc__hist.html#ga4b2b5fd75503ff9e6844cc4dcdaed35d
* clarify special case:
    http://docs.opencv.org/3.2.0/db/de0/group__core__utils.html#ga4910d7f86336cd4eff9dd05575667e41
"""
from __future__ import print_function
import os
import re
import sys
import logging

loglevel=os.environ.get("LOGLEVEL", None)
if loglevel:
    logging.basicConfig(level=loglevel)

ADD_JAVA = False
ADD_PYTHON = True
ROOT_DIR = sys.argv[1]
PYTHON_SIGNATURES_FILE = sys.argv[2]

import json
python_signatures = dict()
with open(PYTHON_SIGNATURES_FILE, "rt") as f:
    python_signatures = json.load(f)
    print("Loaded Python signatures: %d" % len(python_signatures))

class Configuration():
    def __init__(self):
        self.ADD_PYTHON = ADD_PYTHON
        self.python_signatures = python_signatures
        self.ADD_JAVA = ADD_JAVA

config = Configuration()


import html_functions

soup = html_functions.load_html_file(ROOT_DIR + "index.html")
href_list = html_functions.get_links_list(soup, True)

for link in href_list:
    # add python signatures to the module
    soup = html_functions.load_html_file(ROOT_DIR + link)
    sub_href_list = html_functions.get_links_list(soup, True)
    module_name = html_functions.get_text_between_substrings(link, "group__", ".html")
    html_functions.add_signatures(soup, ROOT_DIR + link, module_name, config)

    # add python signatures to the sub-modules
    link = re.sub(r"group__.+html", "", link)
    for sub_link in sub_href_list:
        tmp_dir = ROOT_DIR + link + sub_link
        soup = html_functions.load_html_file(tmp_dir)
        html_functions.add_signatures(soup, tmp_dir, module_name, config)