bed_utils.py 文件源码

def overlap(bed1, bed2, bedOut):
    if not bed2:
        shutil.copyfile(bed1,bedOut)
        return

    with open(bed1) as f:
        bed_dict1 = tk_io.get_target_regions(f)

    with open(bed2) as f:
        bed_dict2 = tk_io.get_target_regions(f)

    bed_dict_overlap = {}
    for chrom in bed_dict1:
        if not chrom in bed_dict_overlap:
            bed_dict_overlap[chrom] = tk_regions.Regions([])
        for start, end in bed_dict1[chrom]:
            if chrom in bed_dict2 and \
                bed_dict2[chrom].overlaps_region(start, end):
                bed_dict_overlap[chrom].add_region((start,end))

    writeOut(bed_dict_overlap, bedOut)