wrapped in main

Author: prathameshnium

pica/keithley/k6517b/Pyroelectricity/Instrument_Control/PyroDataVisualization_Simple.py

@@ -34,55 +34,58 @@ def select_file():
     return tempdir
 
 
-# Call the function to select the file
-selected_file = str(select_file())
+def main():
+    # Call the function to select the file
+    selected_file = str(select_file())
 
-# Print the selected file path
-print(f"Selected file: {selected_file}")
+    # Print the selected file path
+    print(f"Selected file: {selected_file}")
 
-# Load data from CSV file
+    # Load data from CSV file
 
+    # Set up the plot
+    plt.style.use('fivethirtyeight')
+    fig, axs = plt.subplots(3, 1, figsize=(9, 12))
 
-# Set up the plot
-plt.style.use('fivethirtyeight')
-fig, axs = plt.subplots(3, 1, figsize=(9, 12))
 
+    def animate(i):
+        # Reload data from CSV (in case it has changed)
+        data = pd.read_csv(selected_file)
+        x = data['Time (s)']
+        y1 = data['Temperature (K)']
+        y2 = data['Current (A)']
 
-def animate(i):
-    # Reload data from CSV (in case it has changed)
-    data = pd.read_csv(selected_file)
-    x = data['Time (s)']
-    y1 = data['Temperature (K)']
-    y2 = data['Current (A)']
+        # Clear previous plots
+        for ax in axs:
+            ax.clear()
 
-    # Clear previous plots
-    for ax in axs:
-        ax.clear()
+        # Update subplots
+        axs[0].plot(x, y1, label='T', color='b', linewidth=0.8)
+        axs[0].scatter(x, y1, color='b')
+        axs[0].set_title('T vs time', fontsize=13)
+        axs[0].set_xlabel('Time (s)', fontsize=13)
+        axs[0].set_ylabel('Temperature (K)', fontsize=13)
+        axs[0].legend(loc='upper left')
 
-    # Update subplots
-    axs[0].plot(x, y1, label='T', color='b', linewidth=0.8)
-    axs[0].scatter(x, y1, color='b')
-    axs[0].set_title('T vs time', fontsize=13)
-    axs[0].set_xlabel('Time (s)', fontsize=13)
-    axs[0].set_ylabel('Temperature (K)', fontsize=13)
-    axs[0].legend(loc='upper left')
+        axs[1].plot(x, y2, label='I', color='g', linewidth=0.8)
+        axs[1].scatter(x, y2, color='g')
+        axs[1].set_title('I vs time', fontsize=13)
+        axs[1].set_xlabel('Time (s)', fontsize=13)
+        axs[1].set_ylabel('Current (A)', fontsize=13)
+        axs[1].legend(loc='upper left')
 
-    axs[1].plot(x, y2, label='I', color='g', linewidth=0.8)
-    axs[1].scatter(x, y2, color='g')
-    axs[1].set_title('I vs time', fontsize=13)
-    axs[1].set_xlabel('Time (s)', fontsize=13)
-    axs[1].set_ylabel('Current (A)', fontsize=13)
-    axs[1].legend(loc='upper left')
+        axs[2].plot(y1, y2, label='I vs T', color='r', linewidth=0.8)
+        axs[2].scatter(y1, y2, color='r')
+        axs[2].set_title('I vs T', fontsize=13)
+        axs[2].set_xlabel('Temperature (K)', fontsize=13)
+        axs[2].set_ylabel('Current (A)', fontsize=13)
+        axs[2].legend(loc='upper left')
 
-    axs[2].plot(y1, y2, label='I vs T', color='r', linewidth=0.8)
-    axs[2].scatter(y1, y2, color='r')
-    axs[2].set_title('I vs T', fontsize=13)
-    axs[2].set_xlabel('Temperature (K)', fontsize=13)
-    axs[2].set_ylabel('Current (A)', fontsize=13)
-    axs[2].legend(loc='upper left')
 
+    ani = FuncAnimation(plt.gcf(), animate, interval=1000, cache_frame_data=False)
 
-ani = FuncAnimation(plt.gcf(), animate, interval=1000, cache_frame_data=False)
+    plt.tight_layout()
+    plt.show()
 
-plt.tight_layout()
-plt.show()
+if __name__ == "__main__":
+    main()

pica/utils/GPIB_VISA_InterfaceTest_Simple_Instrument_Control.py

@@ -23,27 +23,31 @@
 
 import pyvisa
 
-# Initialize the tool to find connected instruments
-rm = pyvisa.ResourceManager()
-instrument_addresses = rm.list_resources()
-
-# Check if any instruments were found
-if not instrument_addresses:
-    print("No instruments found. Check connections and VISA installation.")
-else:
-    print(
-        f"Found {len(instrument_addresses)} instrument(s). Checking them now...\n")
-
-    # Loop through each instrument and try to get its ID
-    for address in instrument_addresses:
-        try:
-            # Connect to the instrument (connection closes automatically)
-            with rm.open_resource(address) as instrument:
-                instrument.timeout = 2000  # Set a 2-second timeout
-                idn = instrument.query('*IDN?')
-                print(f"Address: {address}\n  ID: {idn.strip()}\n")
-        except Exception as e:
-            # If something goes wrong, print an error and continue
-            print(f"Address: {address}\n  Error: Could not get ID. {e}\n")
-
-print("Scan complete.")
+def main():
+    # Initialize the tool to find connected instruments
+    rm = pyvisa.ResourceManager()
+    instrument_addresses = rm.list_resources()
+
+    # Check if any instruments were found
+    if not instrument_addresses:
+        print("No instruments found. Check connections and VISA installation.")
+    else:
+        print(
+            f"Found {len(instrument_addresses)} instrument(s). Checking them now...\n")
+
+        # Loop through each instrument and try to get its ID
+        for address in instrument_addresses:
+            try:
+                # Connect to the instrument (connection closes automatically)
+                with rm.open_resource(address) as instrument:
+                    instrument.timeout = 2000  # Set a 2-second timeout
+                    idn = instrument.query('*IDN?')
+                    print(f"Address: {address}\n  ID: {idn.strip()}\n")
+            except Exception as e:
+                # If something goes wrong, print an error and continue
+                print(f"Address: {address}\n  Error: Could not get ID. {e}\n")
+
+    print("Scan complete.")
+
+if __name__ == "__main__":
+    main()