git_forlog/script.py at main · VanuatuN/git_forlog · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
# Stage 1: Handle Command Line

import sys
import numpy as np
import matplotlib.pyplot as plt
import math


def parse_command_line():
    if len(sys.argv) != 2:
        print("Usage: python script.py <function_number>")
        sys.exit(1)

    try:
        function_number = int(sys.argv[1])
        return function_number
    except ValueError:
        print("Please provide a valid integer for function number.")
        sys.exit(1)


def fill_lists_irrational(xval, yval, function_number):
    if function_number == 1:
        yval.extend(xval)  # y = f(x) = x
    elif function_number == 2:
        yval.extend(x**2 for x in xval)  # y = f(x) = x**2
    elif function_number == 3:
        yval.extend(x**3 for x in xval)  # y = f(x) = x**3
    elif function_number == 4:
        yval.extend(math.sin(x) for x in xval)  # y = f(x) = sin(x)
    elif function_number == 5:
        yval.extend(math.cos(x) for x in xval)  # y = f(x) = cos(x)
    elif function_number == 6:
        yval.extend(math.tan(x) for x in xval)  # y = f(x) = tan(x)
    elif function_number == 7:
        yval.extend(math.exp(x) for x in xval)  # y = f(x) = exp(x)
    elif function_number == 8:
        yval.extend(math.sqrt(abs(x)) for x in xval)  # y = f(x) = sqrt(|x|)
    else:
        print("Function not implemented.")
        print_usage()
        sys.exit(1)

# Stage 6: Development Branch

def print_usage():
    print("Usage: python script.py <function_number>")
    print("Supported functions:")
    print("  1 - f(x) = x")
    print("  2 - f(x) = x^2 (polynomial)")
    print("  3 - f(x) = x^3 (polynomial)")
    print("  4 - f(x) = sin(x) (trigonometric)")
    print("  5 - f(x) = cos(x) (trigonometric)")
    print("  6 - f(x) = tan(x) (trigonometric)")
    print("  7 - f(x) = exp(x) (irrational)")
    print("  8 - f(x) = sqrt(|x|) (irrational)")


if __name__ == "__main__":
    function_number = parse_command_line()
    print(f"Selected function number: {function_number}")

    # Define x values
    xval = np.arange(-3.0, 3.0, 0.1).tolist()

    # Initialize y values
    yval = []

    # Fill lists based on function number
    fill_lists_irrational(xval, yval, function_number)

    print("xval:", xval)
    print("yval:", yval)