4580ec74800db2a9e4288d49c36c41b1c2077c42,donkeycar/management/tub.py,TubApi,clips_of_tub,#TubApi#Any#,81

Before Change

        return os.path.join(tub_path, "record_" + frame_id + ".json")

    def clips_of_tub(self, tub_path):
        seqs = [ int(f.split("_")[0]) for f in os.listdir(tub_path) if f.endswith(".jpg") ]
        seqs.sort()

        entries = ((os.stat(self.image_path(tub_path, seq))[ST_ATIME], seq) for seq in seqs)

        (last_ts, seq) = next(entries)
        clips = [[seq]]
        for next_ts, next_seq in entries:
            //if next_ts - last_ts > 100:  //greater than 1s apart
            //    clips.append([next_seq])
            //else:
            //    clips[-1].append(next_seq)
            clips[-1].append(next_seq)
            last_ts = next_ts

        return clips

    def get(self, tub_id):
        clips = self.clips_of_tub(os.path.join(self.data_path, tub_id))

After Change

        self.data_path = path.absolute()

    def clips_of_tub(self, tub_path):
        tub = Tub(tub_path)

        clips = []
        for record in tub:
            index = record["_index"]
            images_relative_path = os.path.join(Tub.images(), record["cam/image_array"])
            record["cam/image_array"] = images_relative_path
            clips.append(record)

        return [clips]

    def get(self, tub_id):
        base_path = os.path.join(self.data_path, tub_id)

Instances