for i in range(4):
  for j in range(i):
    k+=1;
    plt.subplot(2,3,k);
    plt.plot(dt[:,j], dt[:,i], '.');
    plt.title('L%d vs L%d' % (j+1,i+1))


fig.savefig(os.path.join(fig_directory, 'stage_durations_correlation.pdf'))



#%% PCA on stage durations

fig = plt.figure(412); plt.clf();
fplt.plot_pca(dt[:,:-1])

fig.savefig(os.path.join(fig_directory, 'stage_durations_pca.pdf'))






#%% Get stage durations from Roaming Dwelling data set

strain = 'N2';

rd_data = rexp.load_data(strain);

rd_stage_ids = rd_data.stage_switch;
for i in range(4):
  for j in range(i):
    k+=1;
    plt.subplot(2,3,k);
    plt.plot(dt[:,j], dt[:,i], '.');
    plt.title('L%d vs L%d' % (j+1,i+1))


fig.savefig(os.path.join(fig_directory, 'stage_durations_correlation.pdf'))



#%% PCA on stage durations

fig = plt.figure(412); plt.clf();
fplt.plot_pca(dt[:,:-1])

fig.savefig(os.path.join(fig_directory, 'stage_durations_pca.pdf'))






#%% Get stage durations from Roaming Dwelling data set

strain = 'N2';

rd_data = rexp.load_data(strain);

rd_stage_ids = rd_data.stage_switch;
plt.hist(bend, bins=256)

# resolve for stages etc

mt = np.max(np.abs(theta), axis=1)
idx = mt < 1

theta_red = theta[idx]
theta_red.shape

aplt.plot_array(theta_red.T)

### PCA
plt.figure(12)
plt.clf()
aplt.plot_pca(theta)

pca = aplt.PCA(theta)

pca_comp = pca.Wt

import worm.model as wm
import worm.geometry as wgeo
w = wm.WormModel(npoints=22)

fig = plt.figure(17)
plt.clf()
for i in range(len(pca_comp)):
    tt = pca_comp[i] * 21
    tta = tt - np.mean(tt)
    if i == 0: