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_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 = 9
sample_name = 'qNSC_Young_Q_A'


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