#include "llvm/CodeGen/InterferenceGraph.h"


InterferenceGraph::InterferenceGraph(RegClass *const RC) : RegCl(RC), 
							   IGNodeList() 
{   
  IG = NULL;         
  Size = 0;            
  if( DEBUG_RA) {
    cout << "Interference graph created!" << endl;
  }
}

InterferenceGraph:: ~InterferenceGraph() {                // destructor
    if( IG )
      delete []IG;
  }




void InterferenceGraph::createGraph()   
{ 
    Size = IGNodeList.size();
    IG = (char **) new char *[Size]; 
    for( unsigned int r=0; r < Size; ++r)
      IG[r] = new char[Size];

    // init IG matrix
    for(unsigned int i=0; i < Size; i++)     
      for( unsigned int j=0; j < Size ; j++)
	IG[i][j] = 0;
}



void InterferenceGraph::addLRToIG(LiveRange *const LR)
{
  IGNode *Node = new IGNode(LR,  IGNodeList.size() );
  IGNodeList.push_back( Node );
  //Node->setRegClass( RegCl );
}


// update both the matrix and AdjLists of nodes.
// If there is already an interference between LR1 and LR2, adj lists
// are not updated. LR1 and LR2 must be distinct since if not, it suggests
// that there is some wrong logic in some other method.

void InterferenceGraph::setInterference(const LiveRange *const LR1,
					const LiveRange *const LR2 ) {
  assert(LR1 != LR2);   

  IGNode *const IGNode1 = LR1->getUserIGNode();
  IGNode *const IGNode2 = LR2->getUserIGNode();

  if( DEBUG_RA) {
    assertIGNode( IGNode1 );   
    assertIGNode( IGNode2 );
  }
  
  const unsigned int row = IGNode1->getIndex();
  const unsigned int col = IGNode2->getIndex();

  char *val;

  if( DEBUG_RA > 1) 
    cout << "setting intf for: [" << row << "][" <<  col << "]" << endl; 

  ( row > col) ?  val = &IG[row][col]: val = &IG[col][row]; 

  if( ! (*val) ) {                      // if this interf is not previously set

    *val = 1;                           // add edges between nodes 
    IGNode1->addAdjIGNode( IGNode2 );   
    IGNode2->addAdjIGNode( IGNode1 );
  }

}



unsigned InterferenceGraph::getInterference(const LiveRange *const LR1,
					   const LiveRange *const LR2 ) const {

  assert(LR1 != LR2);

  if( DEBUG_RA) {
    assertIGNode( LR1->getUserIGNode() );  
    assertIGNode( LR2->getUserIGNode() );
  }

  const unsigned int row = LR1->getUserIGNode()->getIndex();
  const unsigned int col = LR2->getUserIGNode()->getIndex();

  char ret; 
  ( row > col) ?  (ret = IG[row][col]) : (ret = IG[col][row]) ; 
  return ret;

}



// Merge 2 IGNodes. The neighbors of the SrcNode will be added to the DestNode.
// Then the IGNode2L  will be deleted. Necessary for coalescing.
// IMPORTANT: The live ranges are NOT merged by this method. Use 
//            LiveRangeInfo::unionAndUpdateLRs for that purpose.

void InterferenceGraph::mergeIGNodesOfLRs(const LiveRange *const LR1, 
					  LiveRange *const LR2 ) {

  assert( LR1 != LR2);                  // cannot merge the same live range

  IGNode *const DestNode = LR1->getUserIGNode();
  IGNode *SrcNode = LR2->getUserIGNode();

  assertIGNode( DestNode );
  assertIGNode( SrcNode );

  if( DEBUG_RA > 1) {
    cout << "Merging LRs: \""; LR1->printSet(); 
    cout << "\" and \""; LR2->printSet();
    cout << "\"" << endl;
  }

  unsigned SrcDegree = SrcNode->getNumOfNeighbors();
  const unsigned SrcInd = SrcNode->getIndex();


  // for all neighs of SrcNode
  for(unsigned i=0; i < SrcDegree; i++) {        
    IGNode *NeighNode = SrcNode->getAdjIGNode(i); 

    LiveRange *const LROfNeigh = NeighNode->getParentLR();

    // delete edge between src and neigh - even neigh == dest
    NeighNode->delAdjIGNode(SrcNode);  

    // set the matrix posn to 0 betn src and neigh - even neigh == dest
    const unsigned NInd = NeighNode->getIndex();
    ( SrcInd > NInd) ?  (IG[SrcInd][NInd]=0) : (IG[NInd][SrcInd]=0) ; 


    if( LR1 != LROfNeigh) {             // if the neigh != dest 
     
      // add edge betwn Dest and Neigh - if there is no current edge
      setInterference(LR1, LROfNeigh );  
    }
    
    //cout<< "  #Neighs - Neigh: ["<< NeighNode->getIndex()<< "] ";
    //cout << NeighNode->getNumOfNeighbors();
    //cout << " Dest: [" << DestNode->getIndex() << "] ";
    //cout << DestNode->getNumOfNeighbors()  << endl;

  }

  IGNodeList[ SrcInd ] = NULL;

  // SrcNode is no longer necessary - LR2 must be deleted by the caller
  delete( SrcNode );    

}



// must be called after modifications to the graph are over but before
// pushing IGNodes on to the stack for coloring.

void InterferenceGraph::setCurDegreeOfIGNodes()
{
  unsigned Size = IGNodeList.size();

  for( unsigned i=0; i < Size; i++) {
    IGNode *Node = IGNodeList[i];
    if( Node )
      Node->setCurDegree();
  }
}





//--------------------- debugging (Printing) methods -----------------------


void InterferenceGraph::printIG() const
{

  for(unsigned int i=0; i < Size; i++) {   

    const IGNode *const Node = IGNodeList[i];
    if( ! Node )
      continue;                         // skip empty rows

    cout << " [" << i << "] ";

      for( unsigned int j=0; j < Size; j++) {
	if( j >= i) break;
	if( IG[i][j] ) cout << "(" << i << "," << j << ") ";
      }
      cout << endl;
    }
}


void InterferenceGraph::printIGNodeList() const
{
  vector<IGNode *>::const_iterator IGIt = IGNodeList.begin(); // hash map iter

  for(unsigned i=0; i < IGNodeList.size() ; ++i) {

    const IGNode *const Node = IGNodeList[i];

    if( ! Node )
      continue;

    cout << " [" << Node->getIndex() << "] ";
    (Node->getParentLR())->printSet(); 
    //int Deg = Node->getCurDegree();
    cout << "\t <# of Neighs: " << Node->getNumOfNeighbors() << ">" << endl;
    
  }
}