import numpy as np
np.random.seed(42)
import random
random.seed(1)
import matplotlib.pyplot as plt
%matplotlib inline
import sys
# import anhima
# dev imports
sys.path.insert(0, '..')
%reload_ext autoreload
%autoreload 1
%aimport anhima.loc

# simulate non-uniform variant positions
n_variants = 1000
p = 0
pos = []
for i in range(n_variants):
    gap = int(np.abs(np.cos(i/100))*100)
    p += gap
    pos.append(p)
pos = np.array(pos)

# plot variant locations
anhima.loc.plot_variant_locator(pos, step=10);

# plot variant locations with y axis inverted
anhima.loc.plot_variant_locator(pos, step=10, flip=True);

loc = anhima.loc.locate_interval(pos, 11000, 20000)
loc

pos[loc.start-1], pos[loc.start]

pos[loc.stop-1], pos[loc.stop]

# plot variant counts
anhima.loc.plot_windowed_variant_counts(pos, window_size=1000);

# plot variant counts
anhima.loc.plot_windowed_variant_counts(pos, window_size=5000);

# plot variant density
anhima.loc.plot_windowed_variant_density(pos, window_size=1000);

# plot variant density
anhima.loc.plot_windowed_variant_density(pos, window_size=5000);