/*

 * Copyright 2007 Sun Microsystems, Inc.  All Rights Reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 *

 */

#include "incls/_precompiled.incl"

#include "incls/_idealGraphPrinter.cpp.incl"

#ifndef PRODUCT

// Constants

// Keep consistent with Java constants

const char *IdealGraphPrinter::INDENT = "  ";

const char *IdealGraphPrinter::TOP_ELEMENT = "graphDocument";

const char *IdealGraphPrinter::GROUP_ELEMENT = "group";

const char *IdealGraphPrinter::GRAPH_ELEMENT = "graph";

const char *IdealGraphPrinter::PROPERTIES_ELEMENT = "properties";

const char *IdealGraphPrinter::EDGES_ELEMENT = "edges";

const char *IdealGraphPrinter::PROPERTY_ELEMENT = "p";

const char *IdealGraphPrinter::EDGE_ELEMENT = "edge";

const char *IdealGraphPrinter::NODE_ELEMENT = "node";

const char *IdealGraphPrinter::NODES_ELEMENT = "nodes";

const char *IdealGraphPrinter::REMOVE_EDGE_ELEMENT = "removeEdge";

const char *IdealGraphPrinter::REMOVE_NODE_ELEMENT = "removeNode";

const char *IdealGraphPrinter::METHOD_NAME_PROPERTY = "name";

const char *IdealGraphPrinter::METHOD_IS_PUBLIC_PROPERTY = "public";

const char *IdealGraphPrinter::METHOD_IS_STATIC_PROPERTY = "static";

const char *IdealGraphPrinter::TRUE_VALUE = "true";

const char *IdealGraphPrinter::NODE_NAME_PROPERTY = "name";

const char *IdealGraphPrinter::EDGE_NAME_PROPERTY = "name";

const char *IdealGraphPrinter::NODE_ID_PROPERTY = "id";

const char *IdealGraphPrinter::FROM_PROPERTY = "from";

const char *IdealGraphPrinter::TO_PROPERTY = "to";

const char *IdealGraphPrinter::PROPERTY_NAME_PROPERTY = "name";

const char *IdealGraphPrinter::GRAPH_NAME_PROPERTY = "name";

const char *IdealGraphPrinter::INDEX_PROPERTY = "index";

const char *IdealGraphPrinter::METHOD_ELEMENT = "method";

const char *IdealGraphPrinter::INLINE_ELEMENT = "inline";

const char *IdealGraphPrinter::BYTECODES_ELEMENT = "bytecodes";

const char *IdealGraphPrinter::METHOD_BCI_PROPERTY = "bci";

const char *IdealGraphPrinter::METHOD_SHORT_NAME_PROPERTY = "shortName";

const char *IdealGraphPrinter::CONTROL_FLOW_ELEMENT = "controlFlow";

const char *IdealGraphPrinter::BLOCK_NAME_PROPERTY = "name";

const char *IdealGraphPrinter::BLOCK_DOMINATOR_PROPERTY = "dom";

const char *IdealGraphPrinter::BLOCK_ELEMENT = "block";

const char *IdealGraphPrinter::SUCCESSORS_ELEMENT = "successors";

const char *IdealGraphPrinter::SUCCESSOR_ELEMENT = "successor";

const char *IdealGraphPrinter::ASSEMBLY_ELEMENT = "assembly";

int IdealGraphPrinter::_file_count = 0;

IdealGraphPrinter *IdealGraphPrinter::printer() {

  if (PrintIdealGraphLevel == 0) return NULL;

  JavaThread *thread = JavaThread::current();

  if (!thread->is_Compiler_thread()) return NULL;

  CompilerThread *compiler_thread = (CompilerThread *)thread;

  if (compiler_thread->ideal_graph_printer() == NULL) {

    IdealGraphPrinter *printer = new IdealGraphPrinter();

    compiler_thread->set_ideal_graph_printer(printer);

  }

  return compiler_thread->ideal_graph_printer();

}

void IdealGraphPrinter::clean_up() {

  JavaThread *p;

  for (p = Threads::first(); p; p = p->next()) {

    if (p->is_Compiler_thread()) {

      CompilerThread *c = (CompilerThread *)p;

      IdealGraphPrinter *printer = c->ideal_graph_printer();

      if (printer) {

        delete printer;

      }

      c->set_ideal_graph_printer(NULL);

    }

  }

}

// Constructor, either file or network output

IdealGraphPrinter::IdealGraphPrinter() {

  _should_send_method = true;

  _output = NULL;

  buffer[0] = 0;

  _depth = 0;

  _current_method = NULL;

  assert(!_current_method, "current method must be initialized to NULL");

  if (PrintIdealGraphFile != NULL) {

    ThreadCritical tc;

    // User wants all output to go to files

    if (_file_count != 0) {

      ResourceMark rm;

      stringStream st;

      const char* dot = strrchr(PrintIdealGraphFile, '.');

      if (dot) {

        st.write(PrintIdealGraphFile, dot - PrintIdealGraphFile);

        st.print("%d%s", _file_count, dot);

      } else {

        st.print("%s%d", PrintIdealGraphFile, _file_count);

      }

    } else {

    }

    _file_count++;

  } else {

    // Try to connect to visualizer

    if (_stream->connect(PrintIdealGraphAddress, PrintIdealGraphPort)) {

      char c = 0;

      _stream->read(&c, 1);

      if (c != 'y') {

        tty->print_cr("Client available, but does not want to receive data!");

        _stream->close();

        delete _stream;

        _stream = NULL;

        return;

      }

      _output = _stream;

    } else {

      // It would be nice if we could shut down cleanly but it should

      // be an error if we can't connect to the visualizer.

      fatal2("Couldn't connect to visualizer at %s:%d", PrintIdealGraphAddress, PrintIdealGraphPort);

    }

  }

}

// Destructor, close file or network stream

IdealGraphPrinter::~IdealGraphPrinter() {

  if (_stream) {

    delete _stream;

    if (_stream == _output) {

      _output = NULL;

    }

    _stream = NULL;

  }

  if (_output) {

    delete _output;

    _output = NULL;

  }

}

}

void IdealGraphPrinter::print_method(ciMethod *method, int bci, InlineTree *tree) {

  stringStream str;

  method->print_name(&str);

  stringStream shortStr;

  method->print_short_name(&shortStr);

  output()->print_cr("<![CDATA[");

  method->print_codes_on(output());

  output()->print_cr("]]>");

  if (tree != NULL) {

    GrowableArray<InlineTree *> subtrees = tree->subtrees();

    for (int i = 0; i < subtrees.length(); i++) {

      print_inline_tree(subtrees.at(i));

    }

  }

  output()->flush();

}

void IdealGraphPrinter::print_inline_tree(InlineTree *tree) {

  if (tree == NULL) return;

  ciMethod *method = tree->method();

  print_method(tree->method(), tree->caller_bci(), tree);

}

void IdealGraphPrinter::print_inlining(Compile* compile) {

  // Print inline tree

  if (_should_send_method) {

    InlineTree *inlineTree = compile->ilt();

    if (inlineTree != NULL) {

      print_inline_tree(inlineTree);

    } else {

      // print this method only

    }

  }

}

// Has to be called whenever a method is compiled

void IdealGraphPrinter::begin_method(Compile* compile) {

  ciMethod *method = compile->method();

  assert(_output, "output stream must exist!");

  assert(method, "null methods are not allowed!");

  assert(!_current_method, "current method must be null!");

  // Print properties

  // Add method name

  stringStream strStream;

  method->print_name(&strStream);

  if (method->flags().is_public()) {

  }

  if (method->flags().is_static()) {

  }

  if (_stream) {

    char answer = 0;

    int result = _stream->read(&answer, 1);

    _should_send_method = (answer == 'y');

  }

  this->_current_method = method;

}

// Has to be called whenever a method has finished compilation

void IdealGraphPrinter::end_method() {

  nmethod* method = (nmethod*)this->_current_method->code();

  _current_method = NULL;

}

// Print indent

void IdealGraphPrinter::print_indent() {

  for (int i = 0; i < _depth; i++) {

  }

}

bool IdealGraphPrinter::traverse_outs() {

  return _traverse_outs;

}

void IdealGraphPrinter::set_traverse_outs(bool b) {

  _traverse_outs = b;

}