From 93945f6a2deb3fd66b050dc76c942671c2f4567d Mon Sep 17 00:00:00 2001
From: Jeremy Fee <jmfee@usgs.gov>
Date: Thu, 28 Jan 2016 12:47:58 -0700
Subject: [PATCH] Add alpha, beta, gamma, and m command line parameters to
sqdist
---
geomagio/algorithm/SqDistAlgorithm.py | 33 +++++++++++++++++++++++++++
1 file changed, 33 insertions(+)
diff --git a/geomagio/algorithm/SqDistAlgorithm.py b/geomagio/algorithm/SqDistAlgorithm.py
index e22338aa6..99e279728 100644
--- a/geomagio/algorithm/SqDistAlgorithm.py
+++ b/geomagio/algorithm/SqDistAlgorithm.py
@@ -457,3 +457,36 @@ class SqDistAlgorithm(Algorithm):
alpha, beta, gamma = parameters[0]
rmse = parameters[1]
return (alpha, beta, gamma, rmse)
+
+
+ @classmethod
+ def add_arguments(cls, parser):
+ """Add command line arguments to argparse parser.
+
+ Parameters
+ ----------
+ parser: ArgumentParser
+ command line argument parser
+ """
+ parser.add_argument('--sqdist-alpha',
+ type=float,
+ help='SqDist alpha parameter')
+ parser.add_argument('--sqdist-beta',
+ type=float,
+ help='SqDist alpha parameter')
+ parser.add_argument('--sqdist-gamma',
+ type=float,
+ help='SqDist alpha parameter')
+
+ def configure(self, arguments):
+ """Configure algorithm using comand line arguments.
+
+ Parameters
+ ----------
+ arguments: Namespace
+ parsed command line arguments
+ """
+ Algorithm.configure(self, arguments)
+ self.alpha = arguments.sqdist_alpha
+ self.beta = arguments.sqdist_beta
+ self.gamma = arguments.sqdist_gamma
--
GitLab