#!/usr/bin/python

import os, csv, sys
from DenseSubGraph import *

#Example of using the class DenseSubGraph

#Create an object of the class DenseSubGraph
Object = DenseSubGraph()

#Define a bipartite graph
BiGraph = {'RPS24A(**9)': ['AIR1', 'CLN2', 'GLN1', 'IMP2', 'LSB1', 'PGI1', 'PRK1', 'PTK2', 'RGT1', 'RPB5', 'RVS167', 'SRV2', 'SSA1', 'UBX6'], 'FKS1(HAPLOID)': ['ABP1', 'ACT1', 'AKL1', 'ARG80', 'BUL1', 'CLN2', 'GYL1', 'HUA1', 'LAS17', 'LSB1', 'PAN1', 'PTK2', 'RHO1', 'RPC40', 'RVS167', 'SRV2', 'TOR2', 'UBP5', 'UTP9', 'VRP1', 'YPR078C'], 'RPS24A(HAPLOID)': ['AIP1', 'DED1', 'GCN4', 'GLK1', 'GYP5', 'MCM1', 'PTK2', 'SAC6', 'SLA2', 'SSA1', 'YSC84'], 'KAR2(TET PROMOTER)': ['ABP1', 'ACT1', 'AIR1', 'AKL1', 'CBK1', 'CLN2', 'CLN3', 'CRN1', 'GTS1', 'GYP5', 'HRD1', 'HUA1', 'IMP2', 'LSB1', 'MMS2', 'MSB1', 'MYO3', 'PAN1', 'PTK2', 'RGT1', 'RPB5', 'RPC40', 'RUP1', 'RVS167', 'SAC6', 'SRV2', 'TWF1', 'UBP5', 'UTP9', 'YSC84'], 'FKS1(TET PROMOTER)': ['ABP1', 'CLN2', 'CLN3', 'FKS1', 'LSB1', 'MYO3', 'PTK2', 'SCD5', 'SRV2', 'UBP5', 'YPR078C'], 'CDC42(TET PROMOTER)': ['ABP1', 'ACT1', 'AKL1', 'APC5', 'ARC15', 'BEM2', 'BUL1', 'CBK1', 'CDC3', 'CDC42', 'CLN2', 'CLN3', 'CRN1', 'FCY2', 'FKS1', 'HUA1', 'LSB1', 'MCM6', 'MMS2', 'MYO3', 'PRO3', 'PTK2', 'RPB3', 'RPB5', 'RPC40', 'SCP1', 'SLA2', 'SLG1', 'SRV2', 'STE12', 'SUC2', 'TWF1', 'UBP5', 'UTP9', 'YPR078C', 'YSC84'], 'SHE4': ['ACT1', 'AKL1', 'ARC15', 'ARC18', 'ARC35', 'ARC40', 'CAP1', 'CAP2', 'CLN2', 'DUN1', 'ENT1', 'ERB1', 'FCY2', 'FKS1', 'GYP5', 'HXT6', 'MCM6', 'MMS2', 'MYO3', 'PHO84', 'RHO1', 'RPA43', 'RPC40', 'RPO21', 'RVS161', 'SAC6', 'SCD5', 'SLA2', 'STT4', 'TWF1', 'UBP5', 'UBX6', 'UTP9', 'VRP1', 'YPR078C'], 'GLUCOSAMINE': ['ABP1', 'ARG80', 'CRN1', 'HXT6', 'PGI1', 'PRO3', 'PTK2', 'SRV2', 'UBP5', 'VRP1', 'YPR078C'], 'RPL12A': ['ABP1', 'CLN2', 'CLN3', 'CRN1', 'FCY2', 'GCN4', 'GTS1', 'HUA1', 'LOT6', 'PTK2', 'RVS167', 'SLA2', 'SRV2', 'UTP9']}

#Find the highly dense subgrpah
Result = Object.Find_Subgraph(BiGraph)

#print Result
print 'Nodes in upper set:',Result[0]
print 'Nodes in lower set:',Result[1]