dataset.py 文件源码

def gen_tracking_db(database, tracking_stats):
    """Generate TrackingDataset structure.

    Parameters
    ----------
    database : h5py.File
        HDF5 file object
    tracking_stats : dictionary
        the dictionary that contains TrackingDataset's stats

    Returns
    -------
    database : h5py.File
        HDF5 file object with multiple groups
    """
    primary_list = tracking_stats["primary_list"]

    for pc in primary_list:
        if pc not in database:
            database.create_group(pc)
            print "[MESSAGE] Primary group %s is created" % (pc)

    print "[MESSAGE] TrackingDataset HDF5 structure is generated."