In [1]:
%run ../../shared_setup.ipynb
In [2]:
dup_samples
Out[2]:
In [3]:
gatk_raw = dict()
gatk_filtered = dict()
for cross in CROSSES:
log('== %s ==' % cross)
callset = np.load(GATK_CALLSET_FN_TEMPLATE.format(cross=cross))
gatk_raw[cross] = callset
log('filter variants')
callset_pass = filter_variants(callset, 'FILTER_PASS')
log('filter calls')
callset_cflt = filter_calls(callset_pass, gatk_conf_calls)
gatk_filtered[cross] = callset_pass
In [4]:
cortex_raw = dict()
cortex_filtered = dict()
for cross in CROSSES:
log('== %s ==' % cross)
callset = np.load(CORTEX_CALLSET_FN_TEMPLATE.format(cross=cross))
cortex_raw[cross] = callset
log('filter variants')
callset_pass = filter_variants(callset, 'FILTER_PASS')
log('filter calls')
callset_cflt = filter_calls(callset_pass, cortex_conf_calls)
cortex_filtered[cross] = callset_cflt
In [5]:
accessibility_colors['InternalHypervariable'] = accessibility_colors['SubtelomericHypervariable']
In [ ]:
In [6]:
fig, ax = plt.subplots()
callset = gatk_raw['hb3_dd2']
chrom = 'Pf3D7_12_v3'
plot_inheritance(ax, callset, chrom)
In [7]:
fig, ax = plt.subplots(figsize=(7, 1))
pos = gatk_raw['hb3_dd2']['variants']['POS']
chrom = 'Pf3D7_12_v3'
plot_variants_locator(ax, gatk_raw['hb3_dd2'], chrom, step=100)
In [8]:
def fig_inheritance(fig, callsets, cross, chrom, filename=None, step=100):
callset = callsets[cross]
ax = plt.subplot2grid((9, 1), (0, 0), rowspan=6)
# don't include parents
n_samples = len(callset['calldata'].dtype.names)
plot_inheritance(ax, callset, chrom, sample_indices=range(2, n_samples))
ax.set_xticks([])
# ax.set_title('%s, chromosome %s' % (LABELS[cross], int(chrom[6:8])), fontsize=8)
ax.set_ylabel('progeny clones', rotation=90)
ax.yaxis.tick_left()
ax.set_yticklabels([])
ax = plt.subplot2grid((9, 1), (6, 0))
plot_variants_locator(ax, callset, chrom, step=step)
#ax.set_xticklabels([])
ax.set_xticks([])
ax = plt.subplot2grid((9, 1), (7, 0))
plot_accessibility(ax, chrom, cross, linewidth=.5)
ax.set_ylabel('accessibility', rotation=0, ha='right', va='center')
ax = plt.subplot2grid((9, 1), (8, 0))
plot_genes(ax, chrom)
# ax.set_xticks([])
# ax.spines['bottom'].set_visible(False)
ax.set_ylabel('genes', rotation=0, ha='right', va='center')
ax.set_xlabel('position (bp)')
fig.subplots_adjust(hspace=0.02, left=.12, top=.9, right=.98, bottom=.13)
In [9]:
chrom = 'Pf3D7_12_v3'
callsets = gatk_raw
cross = 'hb3_dd2'
fig = plt.figure(figsize=(7, 4))
fig.suptitle('%s, chromosome %s, alignment-based calling method (BWA/GATK), unfiltered' % (LABELS[cross], int(chrom[6:8])),
fontsize=10, fontweight='bold')
fig_inheritance(fig, callsets, cross, chrom, step=100)
fn = '../../artwork/supp/filtering_alignment_unfiltered.{dpi}.{fmt}'
for fmt in 'jpeg', 'png':
for dpi in 120, 300:
fig.savefig(fn.format(dpi=dpi, fmt=fmt), dpi=dpi, jpeg_quality=100)
In [10]:
chrom = 'Pf3D7_12_v3'
callsets = gatk_filtered
cross = 'hb3_dd2'
fig = plt.figure(figsize=(7, 4))
fig.suptitle('%s, chromosome %s, alignment-based calling method (BWA/GATK), filtered' % (LABELS[cross], int(chrom[6:8])),
fontsize=10, fontweight='bold')
fig_inheritance(fig, callsets, cross, chrom, step=10, filename=None)
fn = '../../artwork/supp/filtering_alignment_filtered.{dpi}.{fmt}'
for fmt in 'jpeg', 'png':
for dpi in 120, 300:
fig.savefig(fn.format(dpi=dpi, fmt=fmt), dpi=dpi, jpeg_quality=100)
In [11]:
chrom = 'Pf3D7_12_v3'
callsets = cortex_raw
cross = 'hb3_dd2'
fig = plt.figure(figsize=(7, 4))
fig.suptitle('%s, chromosome %s, assembly-based calling method (Cortex), unfiltered' % (LABELS[cross], int(chrom[6:8])),
fontsize=10, fontweight='bold')
fig_inheritance(fig, callsets, cross, chrom, step=100, filename=None)
fn = '../../artwork/supp/filtering_assembly_unfiltered.{dpi}.{fmt}'
for fmt in 'jpeg', 'png':
for dpi in 120, 300:
fig.savefig(fn.format(dpi=dpi, fmt=fmt), dpi=dpi, jpeg_quality=100)
In [12]:
chrom = 'Pf3D7_12_v3'
callsets = cortex_filtered
cross = 'hb3_dd2'
fig = plt.figure(figsize=(7, 4))
fig.suptitle('%s, chromosome %s, assembly-based calling method (Cortex), filtered' % (LABELS[cross], int(chrom[6:8])),
fontsize=10, fontweight='bold')
fig_inheritance(fig, callsets, cross, chrom, step=10, filename=None)
fn = '../../artwork/supp/filtering_assembly_filtered.{dpi}.{fmt}'
for fmt in 'jpeg', 'png':
for dpi in 120, 300:
fig.savefig(fn.format(dpi=dpi, fmt=fmt), dpi=dpi, jpeg_quality=100)
In [ ]: