def saveCommonLog(type, immSeed, otherArray, intersect):
immFile = tools.generatePath(args, "imm")
otherFile = ""
commonFile = ""
if (type == "rtim"):
otherFile = tools.generatePath(args, "rtim")
commonFile = tools.generatePath(args, "rtim_intersect", len(intersect))
elif (type == "stream"):
otherFile = tools.generatePath(args, "stream")
commonFile = tools.generatePath(args, "stream_intersect", len(intersect))
path = '../data/{0}/logs/intersect.log'.format(args.dataset)
with open(path, 'a') as f:
f.write("IMM seed set")
f.write(" - file_name: {}".format(immFile))
f.write(" - size: {}".format(len(immSeed)))
if (type == "rtim"):
f.write("RTIM seed set")
elif (type == "stream"):
f.write("Stream")
f.write(" - file_name: {}".format(otherFile))
f.write(" - size: {}".format(len(otherArray)))
f.write("Intersection")
f.write(" - file_name: {}".format(commonFile))
f.write(" - size: {}".format(len(intersect)))
f.write("----------------------------------------------------")
print("Common log saved at: {}".format(path))
def saveRTIMIntersect(intersect):
path = tools.generatePath(args, "rtim_intersect", len(intersect))
with open(path, 'w') as f:
writer = csv.writer(f)
for val in intersect:
writer.writerow([val])
print("IMM ∩ RTIM saved successfully to {}".format(path))
def initiateProgressLog():
# data/nethept/imm/live/progress/NE_k50_e0,5_urr_v1_s15229_prg.txt
path = tools.generatePath(args, "imm_progress")
with open(path, 'w') as f:
writer = csv.writer(f)
writer.writerow(['seen','user_index','seed_size'])
print("Progress log initiated at {}:".format(path))
def saveIMMStreamNoRtimIntersect(intersect):
path = tools.generatePath(args, "no_rtim")
with open(path, 'w') as f:
writer = csv.writer(f)
for val in intersect:
writer.writerow([val])
print("(IMM ∩ Stream)\RTIM saved successfully to {}".format(path))
def saveImmStreamIntersect(intersect):
# data/nethept/imm/live/NE_k50_e0,01_urr_v1_s15229_ss.txt
path = tools.generatePath(args, "stream_intersect", len(intersect))
with open(path, 'w') as f:
writer = csv.writer(f)
for val in intersect:
writer.writerow([val])
print("IMM ∩ Stream intersect saved successfully to {}".format(path))
def importStream():
# data/nethept/streams/urr/v1/NE_urr_v1_s15229_st.txt
path = tools.generatePath(args, "stream")
print("Importing stream from: {}".format(path))
with open(path, 'r') as f:
content = f.readlines()
content = [x.strip() for x in content]
print("Stream length: {}".format(len(content)))
return content
def saveProgress(seen, user_index, seed_size):
# data/nethept/imm/live/progress/NE_k50_e0,5_urr_v1_s15229_prg.txt
path = tools.generatePath(args, "imm_progress")
with open(path, 'a') as f:
writer = csv.writer(f)
writer.writerow([seen, user_index, seed_size])