def embedAllFunctions(projectRoot,
filterName='DefaultFilter',
ngramN=None,
smallerNgramsToo=None):
functionEmbedder = Embedder(projectRoot)
if ngramN != None:
functionEmbedder.configureNgramCalculator(ngramN, smallerNgramsToo)
if filterName != 'DefaultFilter':
""" if filterName == 'OnlyConditionsFilter':
functionEmbedder.setFilter(OnlyConditionsFilter())
elif filterName == 'DefaultPlusConditions':
print 'DefaultPlusConditions'
functionEmbedder.setFilter(DefaultPlusConditionsFilter())
elif filterName == 'IdentifiersInConditions':
functionEmbedder.setFilter(IdentifiersInConditionsFilter())
elif filterName == 'WaterOnly':
functionEmbedder.setFilter(WaterOnlyFilter())
elif filterName == 'ConditionalExpressions':
functionEmbedder.setFilter(ConditionalExpressionsFilter())
"""
if filterName == 'Symbols':
functionEmbedder.setFilter(SymbolFilter())
elif filterName == 'APISymbols':
functionEmbedder.setFilter(APIUsageFilter())
functionEmbedder.embedAllFunctions()
name = '%s_%d.pickl' % (filterName, ngramN)
return functionEmbedder.save(name)
def __init__(self, projectRoot, ngramN, smallerNgramsToo):
self.projectRoot = projectRoot
self.nCalls = self._determineTotalNumberOfCalls()
self.callAreaExtractor = SinkSnippetExtractor()
self.embedder = Embedder(projectRoot)
self.embedder.configureNgramCalculator(ngramN, smallerNgramsToo)
self.nameDictMapToMatrix = NameDictMapToMatrix()
