import changes, test changes, tool instructions added to readme, new environment variable JETSON_TESTING_MODEL_NAME overrides device models entirely, pinmux as one word, addressing comments

Co-authored-by: Tharun Ganeshram <tharun.ganeshram@gmail.com>

Author: tganeshram

README.md

@@ -82,6 +82,23 @@ rules by running:
 sudo udevadm control --reload-rules && sudo udevadm trigger
 ```
 
+# Pinmux Lookup Tool
+
+The `jetson-gpio-pinmux-lookup` command-line tool helps you find the pinmux register address for GPIO pins. This is useful for debugging or setting up pinmux configurations.
+
+**Usage:**
+```shell
+jetson-gpio-pinmux-lookup <gpio_pin_number>
+```
+
+**Example:**
+```shell
+jetson-gpio-pinmux-lookup 7
+# Output on Orin Device: GPIO Pin 7: Mux Register Address = 0x2430070
+```
+
+The tool accepts BOARD mode GPIO pin numbers (1-40) and returns the corresponding pinmux register address in hexadecimal format.
+
 # Running the sample scripts
 
 With the permissions set as needed, the sample applications provided in the

lib/python/Jetson/GPIO/gpio_pin_data.py

@@ -573,6 +573,16 @@ def get_compatibles(compatible_path):
 def get_model():
     compatible_path = '/proc/device-tree/compatible'
 
+    # check first to see whether a testing model name is set
+    model_name = os.environ.get("JETSON_TESTING_MODEL_NAME")
+    if model_name is not None:
+        model_name = model_name.strip()
+        if model_name in JETSON_MODELS:
+            return model_name
+        else:
+            msg = f"Environment variable 'JETSON_TESTING_MODEL_NAME={model_name}' is invalid."
+            sys.stderr.write(msg)
+    
     # get model info from compatible_path
     if os.path.exists(compatible_path):
         compatibles = get_compatibles(compatible_path)

…ython/Jetson/GPIO/gpio_pin_mux_lookup.py …python/Jetson/GPIO/gpio_pinmux_lookup.pylib/python/Jetson/GPIO/gpio_pin_mux_lookup.py renamed to lib/python/Jetson/GPIO/gpio_pinmux_lookup.py lib/python/Jetson/GPIO/gpio_pin_mux_lookup.py renamed to lib/python/Jetson/GPIO/gpio_pinmux_lookup.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python
 # Copyright (c) 2019-2023, NVIDIA CORPORATION. All rights reserved.
 #
 # Permission is hereby granted, free of charge, to any person obtaining a
@@ -18,14 +19,14 @@
 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 # DEALINGS IN THE SOFTWARE.
 
-# @File name: gpio_pin_mux_lookup.py
+# @File name: gpio_pinmux_lookup.py
 # @Date:
 # @Last modified by:
 # @Last Modified time: 9/17/2025
-# @Description: Simple tool to lookup pin mux register address for GPIO pins. User must specify the BOARD mode GPIO pin number.
+# @Description: Simple tool to lookup pinmux register address for GPIO pins. User must specify the BOARD mode GPIO pin number.
 
 import sys
-import gpio_pin_data
+from Jetson.GPIO import gpio_pin_data
 
 # @brief Get pin register address for a given BOARD mode GPIO pin number.
 # @param[in] gpio_pin: BOARD mode GPIO pin number
@@ -44,11 +45,11 @@ def lookup_mux_register(gpio_pin, pin_defs):
 # @brief Main function to handle command line interface.
 def main():
     if len(sys.argv) != 2:
-        print("Usage: jetson-gpio-pin-mux-lookup <gpio_pin_number>")
-        print("\nLookup pin mux register address for GPIO pins.")
+        print("Usage: jetson-gpio-pinmux-lookup <gpio_pin_number>")
+        print("\nLookup pinmux register address for GPIO pins.")
         print("Specify the Board Mode GPIO pin number (e.g., 7, 11, 40, etc.)")
         print("\nExample:")
-        print("  jetson-gpio-pin-mux-lookup 7")
+        print("  jetson-gpio-pinmux-lookup 7")
         sys.exit(1)
 
     try:

samples/test_gpio_pin_mux_lookup.py samples/test_gpio_pinmux_lookup.pysamples/test_gpio_pin_mux_lookup.py renamed to samples/test_gpio_pinmux_lookup.py

@@ -89,10 +89,10 @@ def test(f):
     return f
 
 # command to run the script
-script_path = 'jetson-gpio-pin-mux-lookup'
+script_command = 'jetson-gpio-pinmux-lookup'
 
 def run_gpio_tool(args, env=None):
-    cmd = [sys.executable, script_path] + args
+    cmd = [script_command] + args
     current_env = os.environ.copy()
     if env:
         current_env.update(env)
@@ -114,7 +114,7 @@ def test_no_arguments():
     returncode, stdout, stderr = run_gpio_tool([])
     assert returncode == 1
     assert "Usage:" in stdout
-    assert "jetson-gpio-pin-mux-lookup <gpio_pin_number>" in stdout
+    assert "jetson-gpio-pinmux-lookup <gpio_pin_number>" in stdout
     print("✓ No arguments test passed")
 
 @test
@@ -133,7 +133,7 @@ def test_non_integer_argument():
 
 @test
 def test_negative_pin():
-    test_env = {'JETSON_MODEL_NAME': 'JETSON_ORIN'}
+    test_env = {'JETSON_TESTING_MODEL_NAME': 'JETSON_ORIN'}
     returncode, stdout, stderr = run_gpio_tool(['-1'], env=test_env)
     assert returncode == 1
     error_msg = stderr + stdout
@@ -142,7 +142,7 @@ def test_negative_pin():
 
 @test
 def test_pin_too_high():
-    test_env = {'JETSON_MODEL_NAME': 'JETSON_ORIN'}
+    test_env = {'JETSON_TESTING_MODEL_NAME': 'JETSON_ORIN'}
     returncode, stdout, stderr = run_gpio_tool(['41'], env=test_env)
     assert returncode == 1
     error_msg = stderr + stdout
@@ -153,7 +153,7 @@ def test_pin_too_high():
 
 @test
 def test_jetson_orin_valid_pins():
-    test_env = {'JETSON_MODEL_NAME': 'JETSON_ORIN'}
+    test_env = {'JETSON_TESTING_MODEL_NAME': 'JETSON_ORIN'}
 
     # Test all valid GPIO pins for Jetson Orin
     for gpio_pin in test_data['JETSON_ORIN']['expected_addresses'].keys():
@@ -166,7 +166,7 @@ def test_jetson_orin_valid_pins():
 
 @test
 def test_jetson_orin_nx_valid_pins():
-    test_env = {'JETSON_MODEL_NAME': 'JETSON_ORIN_NX'}
+    test_env = {'JETSON_TESTING_MODEL_NAME': 'JETSON_ORIN_NX'}
 
     # Test all valid GPIO pins for Jetson Orin NX
     for gpio_pin in test_data['JETSON_ORIN_NX']['expected_addresses'].keys():
@@ -179,7 +179,7 @@ def test_jetson_orin_nx_valid_pins():
 
 @test
 def test_jetson_orin_nano_valid_pins():
-    test_env = {'JETSON_MODEL_NAME': 'JETSON_ORIN_NANO'}
+    test_env = {'JETSON_TESTING_MODEL_NAME': 'JETSON_ORIN_NANO'}
 
     # Test all valid GPIO pins for Jetson Orin Nano
     for gpio_pin in test_data['JETSON_ORIN_NX']['expected_addresses'].keys():
@@ -194,7 +194,7 @@ def test_jetson_orin_nano_valid_pins():
 
 @test
 def test_invalid_pins_jetson_orin():
-    test_env = {'JETSON_MODEL_NAME': 'JETSON_ORIN'}
+    test_env = {'JETSON_TESTING_MODEL_NAME': 'JETSON_ORIN'}
 
     # Test a few invalid pins
     for pin in [1, 2, 3, 4, 5, 6]:
@@ -208,7 +208,7 @@ def test_invalid_pins_jetson_orin():
 @test
 def test_no_model_environment():
     # Remove model environment variable
-    test_env = {k: v for k, v in os.environ.items() if k != 'JETSON_MODEL_NAME'}
+    test_env = {k: v for k, v in os.environ.items() if k != 'JETSON_TESTING_MODEL_NAME'}
 
     returncode, stdout, stderr = run_gpio_tool(['7'], env=test_env)
 
@@ -227,7 +227,7 @@ def test_no_model_environment():
 def run_all_tests():
     # Run all tests and report results.
     print("=" * 60)
-    print("GPIO Pin Mux Lookup Tool - Test Suite")
+    print("GPIO Pinmux Lookup Tool - Test Suite")
     print("=" * 60)
 
     passed = 0

setup.py

@@ -45,7 +45,7 @@
       include_package_data          = True,
       entry_points={
           'console_scripts': [
-              'jetson-gpio-pin-mux-lookup=Jetson.GPIO.gpio_pin_mux_lookup:main'
+              'jetson-gpio-pinmux-lookup=Jetson.GPIO.gpio_pinmux_lookup:main'
           ]
       },
 )