From 8f500085dcee42ee6e56c6e44fd52ded1d6a3c91 Mon Sep 17 00:00:00 2001 From: Samuel Sloniker Date: Wed, 18 Aug 2021 13:44:50 -0700 Subject: [PATCH] Pythonify main() --- worm.c | 13 ------------- worm.py | 9 +++++++++ 2 files changed, 9 insertions(+), 13 deletions(-) create mode 100644 worm.py diff --git a/worm.c b/worm.c index 67176e3..61b26a1 100644 --- a/worm.c +++ b/worm.c @@ -87,7 +87,6 @@ char outbuf[BUFSIZ]; void crash(void) __attribute__((__noreturn__)); void display(const struct body *, char); -int main(int, char **); int run_size(int); int run_worm(); void leave(int) __attribute__((__noreturn__)); @@ -99,18 +98,6 @@ int rnd(int); void setup(void); void wake(int); -int main(int argc, char **argv) { - if (argc == 2) - run_size(atoi(argv[1])); - if ((start_len <= 0) || (start_len > ((LINES-3) * (COLS-2)) / 3)) - run_size(LENGTH); - - int retval = run_worm(); - - return retval; - -} - int run_size(int sl) { start_len = sl; run_worm(); diff --git a/worm.py b/worm.py new file mode 100644 index 0000000..4b2f603 --- /dev/null +++ b/worm.py @@ -0,0 +1,9 @@ +import ctypes +import sys + +worm = ctypes.cdll.LoadLibrary('./libworm.so') + +if len(sys.argv) == 2: + worm.run_size(int(sys.argv[1])) +else: + worm.run_size(10)