From 4225ae52fc217870455953fe4a9895fb5480bdb8 Mon Sep 17 00:00:00 2001
From: sangge <2251250136@qq.com>
Date: Fri, 20 Oct 2023 17:15:31 +0800
Subject: [PATCH] feat: add timer

---
 src/demo.py | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/src/demo.py b/src/demo.py
index f68dc8a..b8a2b46 100644
--- a/src/demo.py
+++ b/src/demo.py
@@ -1,30 +1,50 @@
 from tpre import *
+import time
 
 # 1
+start_time = time.time()
 pk_a, sk_a = GenerateKeyPair()
 m = b"hello world"
+end_time = time.time()
+elapsed_time = end_time - start_time
+print(f"代码块1运行时间:{elapsed_time}秒")
 
 # 2
+start_time = time.time()
 capsule_ct = Encrypt(pk_a, m)
+end_time = time.time()
+elapsed_time = end_time - start_time
+print(f"代码块2运行时间:{elapsed_time}秒")
 
 # 3
 pk_b, sk_b = GenerateKeyPair()
 
-N = 70
-T = 49
+N = 10
+T = 5
 
 # 5
+start_time = time.time()
 rekeys = GenerateReKey(sk_a, pk_b, N, T)
+end_time = time.time()
+elapsed_time = end_time - start_time
+print(f"代码块5运行时间:{elapsed_time}秒")
 
 # 7
+start_time = time.time()
 cfrag_cts = []
 
 for rekey in rekeys:
     cfrag_ct = ReEncrypt(rekey, capsule_ct)
     cfrag_cts.append(cfrag_ct)
+end_time = time.time()
+elapsed_time = end_time - start_time
+print(f"代码块7运行时间:{elapsed_time}秒")
 
 # 9
+start_time = time.time()
 cfrags = mergecfrag(cfrag_cts)
 m = DecryptFrags(sk_b, pk_b, pk_a, cfrags)
-
+end_time = time.time()
+elapsed_time = end_time - start_time
+print(f"代码块9运行时间:{elapsed_time}秒")
 print(m)