Compare_merge_methods.ipynb

 
 Comparing fusion methods for events
 '''
Created on 15 fev. 2019

@author: Fab
'''
%matplotlib inline

import sys
sys.path.insert(1, "../")
import time

import sqlite3
from lmtanalysis.FileUtil import getFilesToProcess

from lmtanalysis.Measure import *
from lmtanalysis.Event import EventTimeLine
 
 Ask file to process to the user with a dialog window:

note: You can run this step only once, and then keep processing the file(s) with the next cells.
note2: the file window can be hidden by other windows.
 #ask the user for database to process
print( "Select file name in window")
files = getFilesToProcess()
 
 Parameters
 event_name = "Huddling"
start_frame = 0
end_frame = 10*oneHour
merging_distance = 5
filtering_length = 30
id1 = 1
 
 for file in files:

    # connect to database
    connection = sqlite3.connect( file )

    eventTimeLine_1 = EventTimeLine(connection, event_name, idA = 1, minFrame = start_frame, maxFrame = end_frame)
    eventTimeLine_2 = EventTimeLine(connection, event_name, idA = 2, minFrame = start_frame, maxFrame = end_frame)
    eventTimeLine_3 = EventTimeLine(connection, event_name, idA = 3, minFrame = start_frame, maxFrame = end_frame)
    eventTimeLine_4 = EventTimeLine(connection, event_name, idA = 4, minFrame = start_frame, maxFrame = end_frame)

    eventTimeLine_1.printEventList()
    eventTimeLine_2.printEventList()
    eventTimeLine_3.printEventList()
    eventTimeLine_4.printEventList()

 
 for file in files:

    # connect to database
    connection = sqlite3.connect( file )

    eventTimeLine_1 = EventTimeLine( connection, event_name, idA = 1, minFrame = start_frame, maxFrame = end_frame )
    eventTimeLine_2 = EventTimeLine( connection, event_name, idA = 2, minFrame = start_frame, maxFrame = end_frame )
    eventTimeLine_3 = EventTimeLine( connection, event_name, idA = 3, minFrame = start_frame, maxFrame = end_frame )
    eventTimeLine_4 = EventTimeLine( connection, event_name, idA = 4, minFrame = start_frame, maxFrame = end_frame )

    s_time = time.time()
    eventTimeLine_1.mergeCloseEvents(2*merging_distance)
    eventTimeLine_2.mergeCloseEvents(2*merging_distance)
    eventTimeLine_3.mergeCloseEvents(2*merging_distance)
    eventTimeLine_4.mergeCloseEvents(2*merging_distance)
    print(f"Execution_time {time.time() - s_time} s")

    eventTimeLine_1.removeEventsBelowLength(filtering_length)
    eventTimeLine_2.removeEventsBelowLength(filtering_length)
    eventTimeLine_3.removeEventsBelowLength(filtering_length)
    eventTimeLine_4.removeEventsBelowLength(filtering_length)
    eventTimeLine_1.printEventList()
    eventTimeLine_2.printEventList()
    eventTimeLine_3.printEventList()
    eventTimeLine_4.printEventList()
 
 for file in files:

    # connect to database
    connection = sqlite3.connect(file)

    eventTimeLine_1 = EventTimeLine(connection, event_name, idA = 1, minFrame = start_frame, maxFrame = end_frame)
    eventTimeLine_2 = EventTimeLine(connection, event_name, idA = 2, minFrame = start_frame, maxFrame = end_frame)
    eventTimeLine_3 = EventTimeLine(connection, event_name, idA = 3, minFrame = start_frame, maxFrame = end_frame)
    eventTimeLine_4 = EventTimeLine(connection, event_name, idA = 4, minFrame = start_frame, maxFrame = end_frame)

    s_time = time.time()
    eventTimeLine_1.closeEvents(merging_distance)
    eventTimeLine_2.closeEvents(merging_distance)
    eventTimeLine_3.closeEvents(merging_distance)
    eventTimeLine_4.closeEvents(merging_distance)
    print(f"Execution_time {time.time() - s_time} s")

    eventTimeLine_1.removeEventsBelowLength(filtering_length)
    eventTimeLine_2.removeEventsBelowLength(filtering_length)
    eventTimeLine_3.removeEventsBelowLength(filtering_length)
    eventTimeLine_4.removeEventsBelowLength(filtering_length)
    eventTimeLine_1.printEventList()
    eventTimeLine_2.printEventList()
    eventTimeLine_3.printEventList()
    eventTimeLine_4.printEventList()