Skip to content
Snippets Groups Projects
Commit 244f0eec authored by Alice Brenon's avatar Alice Brenon
Browse files

Take argument-handling out of the CLI function to draw DensityProfiles

parent 6e282546
Branches playground
No related tags found
No related merge requests found
...@@ -42,13 +42,16 @@ def densityProfile(measures, resolution): ...@@ -42,13 +42,16 @@ def densityProfile(measures, resolution):
def drawDensityProfile(measures, outputFile, resolution): def drawDensityProfile(measures, outputFile, resolution):
plotDensity(densityProfile(measures, resolution), outputFile) plotDensity(densityProfile(measures, resolution), outputFile)
def drawDensityProfileCLI(arguments): def getArgs(arguments):
cli = argparse.ArgumentParser( cli = argparse.ArgumentParser(
prog='densityProfile', prog='densityProfile',
description="Draw a density profile from a set of occurrences") description="Draw a density profile from a set of occurrences")
cli.add_argument('inputTSV') cli.add_argument('inputTSV')
cli.add_argument('outputPNG') cli.add_argument('outputPNG')
cli.add_argument('-r', '--resolution', type=int) cli.add_argument('-r', '--resolution', type=int)
args = cli.parse_args(arguments) return cli.parse_args(arguments)
def drawDensityProfileCLI(arguments):
args = getArgs(arguments)
measures = [m[1] for m in tabular(args.inputTSV).iterrows()] measures = [m[1] for m in tabular(args.inputTSV).iterrows()]
drawDensityProfile(measures, args.outputPNG, args.resolution or 100) drawDensityProfile(measures, args.outputPNG, args.resolution or 100)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment