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 = 9
sample_name = 'qNSC_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 = 9
sample_name = 'qNSC_Young'

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: 1497 SequenceContrib ScoresHyp_Contrib Scores	Ctcf, Ctcfl
metacluster_1/pattern_1 # seqlets: 1145 SequenceContrib ScoresHyp_Contrib Scores	Sp2, Sp3, Sp1, Klf3, Sp4, Sp5, Klf6, Wt1, Klf15, Maz, Zbtb17, Klf1, Zfx, Egr1, Zfp281, Klf5, Klf4, Klf7, E2f4, Egr2, Rela, Klf8, Klf12, Usf2, E2f1, Bcl6, E2f7, Fli1, Mbd2, Tfap2c, Sall4, Nr1h4, Ptf1a, E2f6, Tfap2b
metacluster_1/pattern_2 # seqlets: 1101 SequenceContrib ScoresHyp_Contrib Scores	Nfib, Nfic
metacluster_1/pattern_3 # seqlets: 582 SequenceContrib ScoresHyp_Contrib Scores	Nrf1
metacluster_1/pattern_4 # seqlets: 535 SequenceContrib ScoresHyp_Contrib Scores	Foxi1, Nfyb, Nfya, Nfyc, Pbx3
metacluster_1/pattern_5 # seqlets: 535 SequenceContrib ScoresHyp_Contrib Scores	Gabpa, Elf1, Elk1, Elk4, Erg, Ets1, Elk3, Fev, Etv5, Elf5, Etv1, Etv4, Elf2, Elf4, Ehf, Etv6, XP_911724.4, Etv3, Etv2, Spi1, Elf3, Spdef, Stat3
metacluster_1/pattern_6 # seqlets: 465 SequenceContrib ScoresHyp_Contrib Scores	Vsx2, Prrx2, Lhx8, Lhx2, Shox2, Alx1, En2, Rhox6, Lhx9, Rax, Arx, Pax6, Pou1f1, Uncx, Lhx3, Emx2, Vsx1, Lbx2, Hoxa2, Gbx1, Lhx4, Noto, Alx4, Sebox, Alx3, Gbx2, Nkx1-2, Gsx2, Msx3, Phox2a, Prop1, Pax4, Hoxd3, Vax2, Phox2b, Evx1, Esx1, Lhx6, Otp, Prrx1, Hoxa1, En1, Nkx1-1, Vax1, Lmx1b, Nobox, Pax7, Msx1, Pou2f3, Evx2, Pou2f2, Isx, Mnx1, Pdx1, Hoxc5, Hoxb3, Hoxa3, Msx2, Hoxd1, Pou2f1, Lhx1, Lhx5, Pou3f2, Pou3f1, Dlx3, Lmx1a, Hoxa5, Barhl1, Barhl2, Hoxc6, Dlx5, Hlx, Hoxb5, Isl1, Dbx1, Tlx2, Meox2, Dbx2, Pou4f3, Pou3f4, Meox1, Nkx6-3, Isl2, Bsx, Hnf1b, Pou6f1, Nkx6-1, Arid3b, Hoxb4, Hmbox1, Barx1, Hnf1a, Dlx2, Duxbl1
metacluster_1/pattern_7 # seqlets: 433 SequenceContrib ScoresHyp_Contrib Scores	Creb1, Atf7, Creb5, Atf2, Jdp2, Crem, Atf1, Mafb, Atf3, Fosl2, Nfil3, Fosl1, Tef, Tfe3, Jun
metacluster_1/pattern_8 # seqlets: 314 SequenceContrib ScoresHyp_Contrib Scores	Usf1, Tfeb, Tfec, Mitf, Arntl, Bhlhe40, Srebf1, Arnt, Clock, Mlx, Npas2, Mycn, Creb3l2, Srebf2, Tcfl5, Max, Myc, Bhlhe41, Rarg, Hlf, Mlxip
metacluster_1/pattern_9 # seqlets: 290 SequenceContrib ScoresHyp_Contrib Scores	Rfx3, Rfx2, Rfx1, Rfx4, Rfx6, Rfx7, Sox2
metacluster_1/pattern_10 # seqlets: 267 SequenceContrib ScoresHyp_Contrib Scores	Sox9, Sox3, Sox10, Sox4, Sox6, Sox17
metacluster_1/pattern_11 # seqlets: 255 SequenceContrib ScoresHyp_Contrib Scores	Zfp143, Thap11, Tbx2
metacluster_1/pattern_12 # seqlets: 114 SequenceContrib ScoresHyp_Contrib Scores	Dbp, Cebpb, Cebpa, Cebpg, Cebpd, Ddit3, Atf4, Cebpe
metacluster_1/pattern_13 # seqlets: 56 SequenceContrib ScoresHyp_Contrib Scores	Taf1, Yy1, Zfp42
metacluster_1/pattern_14 # seqlets: 51 SequenceContrib ScoresHyp_Contrib Scores	Prdm1, Irf2, Irf1, Stat2, Irf7, Irf8, Stat1, Irf9, Irf3, Irf4
metacluster_1/pattern_15 # seqlets: 51 SequenceContrib ScoresHyp_Contrib Scores
metacluster_1/pattern_16 # seqlets: 44 SequenceContrib ScoresHyp_Contrib Scores	Bach2, Junb, Mafk, Fosb, Mafa, Fos, Batf, Jund, Nfe2l2
metacluster_1/pattern_17 # seqlets: 35 SequenceContrib ScoresHyp_Contrib Scores