from matlas.matches import DenovoModisco, DenovoHomer
from vdom.helpers import (b, summary)
from IPython.display import display
import numpy as np


def display_paiwise_pattern_comparison(task_idx, sample_name, modiscodir, homerdir):
    display(summary(b(sample_name + " (differential regions)")))
    
    homer_ob = DenovoHomer(homerdir)
#     homer_ob.fetch_tomtom_matches(save_report=True, 
#                                   tomtom_dir= "{0}/{1}_tomtomout".format(homerdir, "CISBP_2.00"))
    homer_ob.load_matched_motifs()
    homer_ob.get_motif_per_celltype()

    modisco_ob = DenovoModisco(modiscodir)
    modisco_ob.load_matched_motifs()
    modisco_ob.get_motif_per_celltype()

    modisco_ob.display_pairwise_tf_table(homer_ob)
    return None

def display_denovo_patterns(sample_name, modiscodir, match_threshold=0.05):
    display(summary(b(sample_name)))
    
    ob = DenovoModisco(modiscodir)
    
    ob.load_matched_motifs()
    ob.get_motif_per_celltype(match_threshold=match_threshold)
    ob.display_individual_table()
    
    return None

root = "/mnt/lab_data/kundaje/users/msharmin/NSC_ATAC_PEAKS_to_share"
task_idx = 8
sample_name = 'aNSC_Young'

display_denovo_patterns(
    sample_name,
    modiscodir="{0}/gw_modisco/multi_tasks/{1}".format(root, sample_name)
)

root = "/mnt/lab_data/kundaje/users/msharmin/NSC_ATAC_PEAKS_to_share"
task_idx = 8
sample_name = 'aNSC_Young'

display_paiwise_pattern_comparison(
    task_idx, sample_name, 
    modiscodir="{0}/gw_modisco/multi_tasks/{1}".format(root, sample_name),
    homerdir="{0}/gw_homer/{1}".format(root, sample_name),
)

Pattern Name	TF Name(s)	Modisco
metacluster_1/pattern_0 # seqlets: 73 SequenceContrib ScoresHyp_Contrib Scores	Sp2, Sp3, Sp1, Klf3, Sp4, Sp5, Klf6, Wt1, Klf1, Maz, Zbtb17, Klf15, Egr1, Klf5, Zfx, Zfp281, Rela, Klf4, Egr2, Klf7, E2f4, Klf8, Klf12, Usf2, Mxi1, E2f1, Sall4, Nr1h4, Srebf2, Fli1, Ctcfl, E2f7, Egr3, Ptf1a
metacluster_1/pattern_1 # seqlets: 68 SequenceContrib ScoresHyp_Contrib Scores
metacluster_1/pattern_2 # seqlets: 55 SequenceContrib ScoresHyp_Contrib Scores	Ctcf