# Author: qutools GmbH
# Last edited: Sep 2025
#
# Tested with Windows 11 (64bit), python 3.11.9 (64bit)
#
# This is demo code. Use at your own risk. No warranties.
#
# It may be used and modified with no restriction; raw copies as well as 
# modified versions may be distributed without limitation.

# for sleep
import time

# This code shows how to get event rates and coincidences from a quTAG connected via USB.

# Import the python wrapper which wraps the DLL functions.
# The wrapper should be in the same directory like this code in the folder '..\QUTAG-V1.5.0\userlib\src'.
try:
        import QuTAG_LC
except:
        print("Time Tagger wrapper QuTAG_LC.py is not in the search path.")

# Initialize the quTAG device
qutag = QuTAG_LC.QuTAG()

time.sleep(1)

# Set the exposure time (or integration time) of the internal coincidence counters in milliseconds, range = 0...65535
qutag.setExposureTime(100) # 100 ms exposure time

qutag.setCoincidenceWindow(2000)

# Give some time to accumulate data
time.sleep(1) # 1 second sleep time with 100ms exposure time would give ~10 times data we don't get

# device type
print("getDeviceType: ", qutag.getDeviceType())

# channel count
print("getChannelCount: ", qutag.getChannelCount())

# Now let's retrieve the most recent values of the built-in coincidence counters from quTAG.
# The array contains count rates for all 5 channels and rates for coincidences of events detected on different channels. Events are coincident if they happen within the coincidence window (qutag.setCoincidenceWindow).
# The coincidence counters are not accumulated, i.e. the counter values for the last exposure (see setExposureTime ) are returned.
data,updates = qutag.getCoincCounters()

# updates   Output: Number of data updates by the device since the last call. Pointer may be NULL.
# data      Output: Counter Values. The array must have at least 16 elements.
#           The Counters come in the following channel order with single counts and coincidences:
#           0(Start), 1, 2, 3, 4, 1/2, 1/3, 2/3, 1/4, 2/4, 3/4, 1/2/3, 1/2/4, 1/3/4, 2/3/4, 1/2/3/4 
### see 'tdcbase.h' file reference for more info: function TDC_getCoincCounters(Int32 *data, Int32 *updates)

# Array of channel names
CoincCounter_names = ['0(Start)','1','2','3','4','1/2','1/3','2/3','1/4','2/4','3/4','1/2/3','1/2/4','1/3/4','2/3/4','1/2/3/4']

# Let's print the collected count rates and coincidence rates
print("Channel/Coincidence : Counts ")
for i in range(len(CoincCounter_names)):
        print(CoincCounter_names[i],": ", data[i])

# Disconnects a connected device and stops the internal event loop.
qutag.deInitialize()