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_Q_A'

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_Q_A'

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

Pattern Name	TF Name(s)	Modisco
metacluster_1/pattern_0 # seqlets: 769 SequenceContrib ScoresHyp_Contrib Scores	Ctcf, Ctcfl
metacluster_1/pattern_1 # seqlets: 706 SequenceContrib ScoresHyp_Contrib Scores	Sp3, Sp2, Sp1, Sp4, Klf3, Klf6, Sp5, Klf1, Zbtb17, Wt1, Klf5, E2f4, Egr1, Maz, Zfp281, Klf4, Klf7, Klf8, Klf12, Klf15, Zfx, Egr2, E2f1, Fli1, Sall4
metacluster_1/pattern_2 # seqlets: 578 SequenceContrib ScoresHyp_Contrib Scores	Nfib, Nfic, Nfia, Zic2, Zic3, Insm1
metacluster_1/pattern_3 # seqlets: 452 SequenceContrib ScoresHyp_Contrib Scores	Nfyb, Nfyc, Nfya, Foxi1, Pbx3, Nkx1-1, Emx2, Lhx8, A1JVI6_MOUSE, Gbx1, Msx2, Nobox, Gbx2, En2, Msx3, Barhl1, Lhx6, Prrx1, Hoxd1, Shox2, Rax, Nkx1-2, Vax2, Gsx2, Evx2, Vsx1, Barhl2, En1, Arx, Hoxa2
metacluster_1/pattern_4 # seqlets: 316 SequenceContrib ScoresHyp_Contrib Scores	Nrf1
metacluster_1/pattern_5 # seqlets: 295 SequenceContrib ScoresHyp_Contrib Scores	Ascl2, Ascl1, Nhlh1, Myf6, Tcf3, Ptf1a, Myog, Tcf12, Myod1, Nhlh2, Bhlha15, Tal1, Tcf7l1, Atoh1, Tcf4, Snai2, Myf5
metacluster_1/pattern_6 # seqlets: 213 SequenceContrib ScoresHyp_Contrib Scores	Sox9, Sox10, Sox3, Sox2, Sox4, Sox17, Sox6
metacluster_1/pattern_7 # seqlets: 208 SequenceContrib ScoresHyp_Contrib Scores	Atf2, Atf7, Creb1, Crem, Jdp2, Atf1, Creb5, Atf3, Fosl2, Usf2, Fosl1, Mafb, Nfil3, Jun, Hlf, Tef, Cebpb, Gmeb2, Dbp, Cebpa
metacluster_1/pattern_8 # seqlets: 193 SequenceContrib ScoresHyp_Contrib Scores	Rfx2, Rfx3, Rfx1, Rfx4, Rfx7, Rfx6, Pou5f1
metacluster_1/pattern_9 # seqlets: 173 SequenceContrib ScoresHyp_Contrib Scores	Gabpa, Elk1, Elk4, Ets1, Elf1, Erg, Elk3, Etv5, Etv1, Etv4, Etv3, Etv6, XP_911724.4, Elf4, Etv2, Elf2, Ehf, Mxi1, Fev, Elf5, Spdef, Spi1, Elf3
metacluster_1/pattern_10 # seqlets: 142 SequenceContrib ScoresHyp_Contrib Scores	Thap11, Zfp143, Tbx2
metacluster_1/pattern_11 # seqlets: 123 SequenceContrib ScoresHyp_Contrib Scores	Tfeb, Usf1, Tfe3, Mitf, Tfec, Arntl, Clock, Arnt, Npas2, Mlx, Srebf1, Mycn, Bhlhe40, Myc, Max, Creb3l2, Rxrg, Tcfl5, Srebf2, Rxrb, Mbd2, Xbp1, Vdr
metacluster_1/pattern_12 # seqlets: 108 SequenceContrib ScoresHyp_Contrib Scores	Zeb1, Snai1
metacluster_1/pattern_13 # seqlets: 45 SequenceContrib ScoresHyp_Contrib Scores	Bach2, Mlxip, Arnt2, Zbtb33, Pax2, Bach1
metacluster_1/pattern_14 # seqlets: 31 SequenceContrib ScoresHyp_Contrib Scores