diff --git a/python/main.py b/python/main.py index 1ac029d..c47b434 100644 --- a/python/main.py +++ b/python/main.py @@ -1,195 +1,193 @@ from common.commons import * -from python.settings import * if __name__ == '__main__': - try: - args = getRun() - setLogg() + args = getRun() + setLogg() - setEnv(args) + setEnv(args) - job = args.job - # job = "cluster" - pd.options.mode.chained_assignment = None + job = args.job - # subject = 'ALL' - # rootType = 'if' + from python.settings import * + # job = "cluster" + pd.options.mode.chained_assignment = None - print(job) + # subject = 'ALL' + # rootType = 'if' - if job == 'dataset4j': - from javaDS import createDS + print(job) - createDS() + if job == 'dataset4j': + from javaDS import createDS - elif job == 'dataset4c': - from otherDatasets import core + createDS() - 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 == '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 + elif job == 'actionSI': + from pairs import actionPairs - matches = actionPairs() + matches = actionPairs() - from pairs import createPairs + from pairs import createPairs - createPairs(matches) + createPairs(matches) - from pairs import importAction + from pairs import importAction - 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 == '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 + 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') + 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 + elif job == 'tokenSI': + from pairs import tokenPairs - tokenPairs() + tokenPairs() - from pairs import importTokens + from pairs import importTokens - importTokens() + importTokens() - elif job == 'clusterTokens': - from abstractPatch import cluster + 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') + 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 + elif job == 'codeflaws': + from otherDatasets import codeflaws - codeflaws() + codeflaws() - elif job == 'indexClusters': - from sprinferIndex import runSpinfer + elif job == 'indexClusters': + from sprinferIndex import runSpinfer - runSpinfer() + runSpinfer() - from sprinferIndex import test + from sprinferIndex import test - test() - from sprinferIndex import divideCoccis + test() + from sprinferIndex import divideCoccis - divideCoccis() - from sprinferIndex import removeDuplicates + divideCoccis() + from sprinferIndex import removeDuplicates - removeDuplicates() + removeDuplicates() - # from patchManyBugs import patchCore - # patchCore() - # # from patchManyBugs import patched - # # patched() - # from patchManyBugs import exportSosPatches - # exportSosPatches() - # from validate_manybugs import validate - # - # validate() + # 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 + elif job == 'patternOperations': + from sprinferIndex import patternOperations - patternOperations() - elif job == 'patchManyBugs': - from patchManyBugs import buildAll + patternOperations() + elif job == 'patchManyBugs': + from patchManyBugs import buildAll - 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() + # 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 + elif job == 'patchIntro': + from sprinferIndex import patchCoreIntro - patchCoreIntro() - # from sprinferIndex import patched - # patched() + 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 + 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 + patch_validate() + elif job == 'checkCorrectIntro': + from test_patched_file import checkCorrect - checkCorrect() - elif job == 'manybugs': - from getManybugs import export + checkCorrect() + elif job == 'manybugs': + from getManybugs import export - export() + export() - elif job == 'validateMany': - from patch_validate import patch_validate + elif job == 'validateMany': + from patch_validate import patch_validate - patch_validate() + patch_validate() - elif job == 'introclass': - from getIntroClass import export + elif job == 'introclass': + from getIntroClass import export - export() + export() - elif job == 'stats': - from stats import statsNormal + elif job == 'stats': + from stats import statsNormal - statsNormal(True) + statsNormal(True) - elif job == 'datasetDefects4J': - from defects4JDataset import core + elif job == 'datasetDefects4J': + from defects4JDataset import core - core() + core() - elif job == 'bug': - from bugstats import bStats + elif job == 'bug': + from bugstats import bStats - bStats() - elif job == 'defects4j': - from stats import defects4jStats + bStats() + elif job == 'defects4j': + from stats import defects4jStats - defects4jStats() - elif job == 'patterns': - from stats import exportAbstractPatterns + defects4jStats() + elif job == 'patterns': + from stats import exportAbstractPatterns - exportAbstractPatterns() + exportAbstractPatterns() - else: - logging.error('Unknown job %s', job) - except Exception as e: - logging.error(e) + else: + logging.error('Unknown job %s', job)