From 244f0eecfc5eecd82efce3b157d55b851f3364ca Mon Sep 17 00:00:00 2001
From: Alice BRENON <alice.brenon@ens-lyon.fr>
Date: Mon, 25 Nov 2024 16:31:44 +0100
Subject: [PATCH] Take argument-handling out of the CLI function to draw
DensityProfiles
---
GEODE/Visualisation/DensityProfile.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/GEODE/Visualisation/DensityProfile.py b/GEODE/Visualisation/DensityProfile.py
index 4e8c4bd..36745a9 100644
--- a/GEODE/Visualisation/DensityProfile.py
+++ b/GEODE/Visualisation/DensityProfile.py
@@ -42,13 +42,16 @@ def densityProfile(measures, resolution):
def drawDensityProfile(measures, outputFile, resolution):
plotDensity(densityProfile(measures, resolution), outputFile)
-def drawDensityProfileCLI(arguments):
+def getArgs(arguments):
cli = argparse.ArgumentParser(
prog='densityProfile',
description="Draw a density profile from a set of occurrences")
cli.add_argument('inputTSV')
cli.add_argument('outputPNG')
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()]
drawDensityProfile(measures, args.outputPNG, args.resolution or 100)
--
GitLab