import rtc_plugins
import time
import numpy as np

srcUserId = "srcUser1"
destUserId = "destUser1"

srcDisplayName = "srcDisplayName1"
destDisplayName = "destDisplayName1"
srcRoomId = "srcRoom1"
destRoomId = "destRoomId1"
srcChannelIndex = 42
destChannelIndex = 43

send_time = 0
total_time = 0
total_cnt = 0
def my_callback(npData, dataCount, sampleRate, numChannels, channelIndex):
    global total_time, total_cnt
    recv_time = time.time()
    total_time = total_time + (recv_time - send_time)
    total_cnt = total_cnt + 1
    print(f"dataCount:{dataCount}, sampleRate:{sampleRate}, numChannels:{numChannels}, channelIndex:{channelIndex}")


ret = rtc_plugins.init(srcUserId, srcDisplayName, srcRoomId, my_callback)
if ret != 0:
    print(f"init fail, ret:{ret}")
    exit(1)
ret = rtc_plugins.initSend(destRoomId, destChannelIndex)
if ret != 0:
    print(f"initSend fail, ret:{ret}")
    exit(1)
ret = rtc_plugins.initRecv(srcRoomId, destUserId, srcChannelIndex)
if ret != 0:
    print(f"initRecv fail, ret:{ret}")
    exit(1)

audioData = np.array([0, 1, -1, 0], dtype=np.int16)
for i in range(100):
    send_time = time.time()
    ret = rtc_plugins.sendCustomAudioData(destChannelIndex, audioData, 48000, 1, len(audioData))
    if ret != 0:
        print(f"send fail, ret:{ret}")
    time.sleep(3)

print(f"total_time:{total_time}, total_cnt:{total_cnt}")
print(total_time/total_cnt)