from common.commons import *

if __name__ == '__main__':
    args = getRun()
    setLogg()

    setEnv(args)

    job = args.job

    from python.settings import *
    # job = "cluster"
    pd.options.mode.chained_assignment = None

    # subject = 'ALL'
    # rootType = 'if'

    print(job)

    if job == 'dataset4j':
        from dataset4j import create_dataset

        create_dataset()

    elif job == 'dataset4c':
        from otherDatasets import core

        core()
    elif job == 'richedit':
        dbDir = join(DATA_PATH, 'redis')
        stopDB(dbDir, REDIS_PORT)
        cmd = "JAVA_HOME='" + jdk8 + "' java  -jar " + join(Path(ROOT_DIR).parent, 'target',
                                                            'FixPatternMiner-1.0.0-jar-with-dependencies.jar') + " " + args.prop + " RICHEDITSCRIPT "
        output = shellCallTemplate(cmd)
        logging.info(output)


    elif job == 'actionSI':
        from pairs import actionPairs

        matches = actionPairs()

        from pairs import createPairs

        createPairs(matches)

        from pairs import importAction

        importAction()

    elif job == 'compare':
        # cmd = "mvn exec:java -f '/data/fixminer_source/'
        # -Dexec.mainClass='edu.lu.uni.serval.richedit.akka.compare.CompareTrees'
        # -Dexec.args='"+ " shape " + join(DATA_PATH,"redis") +" ALLdumps-gumInput.rdb " +
        # "clusterl0-gumInputALL.rdb /data/richedit-core/python/data/richEditScript'"
        cmd = "JAVA_HOME='" + jdk8 + "' java  -jar " + join(Path(ROOT_DIR).parent, 'target',
                                                            'FixPatternMiner-1.0.0-jar-with-dependencies.jar') + " " + args.prop + " COMPARE "
        output = shellCallTemplate4jar(cmd)
        logging.info(output)


    elif job == 'cluster':
        from abstractPatch import cluster

        dbDir = join(DATA_PATH, 'redis')
        startDB(dbDir, REDIS_PORT, PROJECT_TYPE)
        cluster(join(DATA_PATH, 'actions'), join(DATA_PATH, 'pairs'), 'actions')

    elif job == 'tokenSI':
        from pairs import tokenPairs

        tokenPairs()

        from pairs import importTokens

        importTokens()

    elif job == 'clusterTokens':
        from abstractPatch import cluster

        dbDir = join(DATA_PATH, 'redis')
        startDB(dbDir, REDIS_PORT, PROJECT_TYPE)
        cluster(join(DATA_PATH, 'tokens'), join(DATA_PATH, 'pairsToken'), 'tokens')


    elif job == 'codeflaws':
        from otherDatasets import codeflaws

        codeflaws()


    elif job == 'indexClusters':
        from sprinferIndex import runSpinfer

        runSpinfer()

        from sprinferIndex import test

        test()
        from sprinferIndex import divideCoccis

        divideCoccis()
        from sprinferIndex import removeDuplicates

        removeDuplicates()

        # from patchManyBugs import patchCore
        # patchCore()
        # # from patchManyBugs import patched
        # # patched()
        # from patchManyBugs import exportSosPatches
        # exportSosPatches()
        # from validate_manybugs import validate
        #
        # validate()

    elif job == 'patternOperations':
        from sprinferIndex import patternOperations

        patternOperations()
    elif job == 'patchManyBugs':
        from patchManyBugs import buildAll

        buildAll()

        # from patchManyBugs import patchCore
        # patchCore()
        # # from patch_validate import patch_validate_mine
        # # patch_validate_mine()
        # from patchManyBugs import patched
        # patched()
        # from patchManyBugs import exportSosPatches
        # exportSosPatches()

    elif job == 'patchIntro':
        from sprinferIndex import patchCoreIntro

        patchCoreIntro()
        # from sprinferIndex import patched
        # patched()

    elif job == 'validateIntro':
        # from patch_validate_introClass2 import patch_validate
        # patch_validate()
        from test_patched_file import patch_validate

        patch_validate()
    elif job == 'checkCorrectIntro':
        from test_patched_file import checkCorrect

        checkCorrect()
    elif job == 'manybugs':
        from getManybugs import export

        export()

    elif job == 'validateMany':
        from patch_validate import patch_validate

        patch_validate()

    elif job == 'introclass':
        from getIntroClass import export

        export()

    elif job == 'stats':
        from stats import statsNormal

        statsNormal(True)


    elif job == 'datasetDefects4J':
        from defects4JDataset import core

        core()

    elif job == 'bug':
        from bugstats import bStats

        bStats()
    elif job == 'defects4j':
        from stats import defects4jStats

        defects4jStats()
    elif job == 'patterns':
        from stats import exportAbstractPatterns

        exportAbstractPatterns()


    else:
        logging.error('Unknown job %s', job)