configs.py

import os, pickle
from enum import Enum

venue_address = '/app/' if os.path.isdir('/app/') else os.path.dirname(os.path.abspath(__file__))
data_address = venue_address + '/Data/'

version = "0.2.0"

def get_version():
    return version


def config_url_venu():
    return venue_address


def config_url_data():
    return data_address


def config_url_log():
    return os.path.join(config_tmp_dir(), 'logfile.log')


def config_dirname_pdfs_no_ocr():
    return '0-pdfs_no_ocr/'


def config_url_pdfs_no_ocr():
    return os.path.join(data_address, config_dirname_pdfs_no_ocr())


def config_dirname_pdfs():
    return '1-pdfs/'


def config_url_pdfs():
    return os.path.join(data_address, config_dirname_pdfs())


def config_dirname_layouts():
    return '2-layouts/'


def config_url_layouts():
    return os.path.join(data_address, config_dirname_layouts())


def config_dirname_refs():
    return '3-refs/'


def config_url_refs():
    return os.path.join(data_address, config_dirname_refs())


def config_dirname_refs_seg():
    return '3-refs_seg/'


def config_url_refs_segment():
    return os.path.join(data_address, config_dirname_refs_seg())


def config_dirname_bibtex():
    return '3-refs_bibtex/'


def config_url_refs_bibtex():
    return os.path.join(data_address, config_dirname_bibtex())


def config_dirname_seg_prob():
    return '3-refs_seg_prob/'


def config_url_refs_segment_prob():
    return os.path.join(data_address, config_dirname_seg_prob())


def config_dirname_seg_dict():
    return '3-refs_seg_dict/'


def config_url_refs_segment_dict():
    return os.path.join(data_address, config_dirname_seg_dict())


def config_dirname_crossref():
    return '4-refs_crossref/'


def config_url_refs_crossref():
    return os.path.join(data_address, config_dirname_crossref())


def email_address():
    return 'boulanger@lhlt.mpg.de'


def config_data_dirnames():
    return [
        config_dirname_pdfs_no_ocr(),
        config_dirname_pdfs(),
        config_dirname_layouts(),
        config_dirname_refs(),
        config_dirname_refs_seg(),
        config_dirname_seg_prob(),
        config_dirname_seg_dict(),
        config_dirname_bibtex(),
        config_dirname_crossref()
    ]


def config_url_git_repo():
    return "https://github.com/cboulanger/excite-docker"


def config_exparser_dir(version=None):
    if version:
        return os.path.join(config_tmp_dir(), os.path.basename(config_url_git_repo()) + "-" + version)
    return os.path.join(venue_address, "EXparser")


def config_exparser_version_file():
    return os.path.join(config_tmp_dir(), "exparser-version.txt")


def config_dataset_dir(model_name=None):
    if model_name:
        return os.path.join(config_dataset_dir(), model_name)
    return os.path.join(venue_address, "Dataset")


def config_model_dir(model_name=None, version=None):
    """
    Returns the path to the dir containing the trained model files. If no argument is passed,
    return the root path where all trained model data is located. If the model name is passed,
    return the path to this model's data, for the current engine version. If a version is passed,
    return the path to the model data for this version. No check is done if the path exists.
    :param model_name:str
    :param version:str
    :return:str
    """
    if model_name:
        return os.path.join(config_model_dir(None, version), version or get_version(), model_name)
    return os.path.join(venue_address, "Models")


def config_lists_dir():
    return os.path.join(venue_address, "Lists")


def config_tmp_dir():
    return os.path.join(venue_address, "tmp")


class DatasetDirs(Enum):
    FEATURES = "Features"
    LYT = "LYT"
    LRT = "LRT"
    REFLD = "RefLD"
    SEG = "SEG"
    TEST_LYT = "TEST_LYT"
    TEST_REFS = "TEST_REFS"
    TEST_SEG = "TEST_SEG"

# Model data in global vars
class ModelObjects:
    rf = None
    crf = None
    kde_ntag = None
    kde_atag = None
    kde_wtag = None
    # kde_ltag = None
    # kde_dtag = None
    # kde_gtag = None
    # kde_llen = None
    # kde_tlen = None