def build()

in build_graph/build.py [0:0]

• 21 lines of code
• 11 McCabe index (conditional complexity)

    def build(self):

        last_vid = self.frames[0][0]
        for frame_i in tqdm.tqdm(range(self.window_size//2, len(self.frames)-self.window_size//2)):

            if self.frames[frame_i][1] < self.start_frame or self.frames[frame_i][1] > self.end_frame:
                continue

            # for combined graphs, if we move to the next video, refresh everything
            v_id = self.frames[frame_i][0]
            if v_id!=last_vid:
                self.reset_buffer()
                self.state['last_node'] = None
                print ('Moving from %s --> %s. Clearing buffers.'%(last_vid, v_id))
                last_vid = v_id

            if self.state['phase']=='localize':
                node_i, score_i = self.localize_step(frame_i)
            elif self.state['phase']=='create':
                node_i, score_i = self.create_step(frame_i)

            self.print_state(frame_i)
            self.state.update({'frame':frame_i, 'inactive':self.state['node'] is None, 'node':node_i, 'score':score_i, 'viz_node':node_i or self.state['viz_node']})
            self.log_history(frame_i)

            if frame_i%self.args.skip==0:
                self.viz(frame_i, block_size=256, img_sz=50, nsz=25, out_sz=1024)

        self.save_history()