OSDN Git Service

2004-09-16 Frank Ch. Eigler <fche@redhat.com>
[pf3gnuchains/gcc-fork.git] / gcc / profile.c
index 17f1f74..77c5d03 100644 (file)
-/* Calculate branch probabilities, and basic block execution counts. 
-   Copyright (C) 1990, 1991, 1992, 1993, 1994, 1996, 1997, 1998, 1999, 2000
-   Free Software Foundation, Inc.
+/* Calculate branch probabilities, and basic block execution counts.
+   Copyright (C) 1990, 1991, 1992, 1993, 1994, 1996, 1997, 1998, 1999,
+   2000, 2001, 2002, 2003, 2004  Free Software Foundation, Inc.
    Contributed by James E. Wilson, UC Berkeley/Cygnus Support;
    based on some ideas from Dain Samples of UC Berkeley.
    Further mangling by Bob Manson, Cygnus Support.
 
-This file is part of GNU CC.
+This file is part of GCC.
 
-GNU CC is free software; you can redistribute it and/or modify
-it under the terms of the GNU General Public License as published by
-the Free Software Foundation; either version 2, or (at your option)
-any later version.
+GCC is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 2, or (at your option) any later
+version.
 
-GNU CC 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 for more details.
+GCC 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
+for more details.
 
 You should have received a copy of the GNU General Public License
-along with GNU CC; see the file COPYING.  If not, write to
-the Free Software Foundation, 59 Temple Place - Suite 330,
-Boston, MA 02111-1307, USA.  */
+along with GCC; see the file COPYING.  If not, write to the Free
+Software Foundation, 59 Temple Place - Suite 330, Boston, MA
+02111-1307, USA.  */
+
+/* Generate basic block profile instrumentation and auxiliary files.
+   Profile generation is optimized, so that not all arcs in the basic
+   block graph need instrumenting. First, the BB graph is closed with
+   one entry (function start), and one exit (function exit).  Any
+   ABNORMAL_EDGE cannot be instrumented (because there is no control
+   path to place the code). We close the graph by inserting fake
+   EDGE_FAKE edges to the EXIT_BLOCK, from the sources of abnormal
+   edges that do not go to the exit_block. We ignore such abnormal
+   edges.  Naturally these fake edges are never directly traversed,
+   and so *cannot* be directly instrumented.  Some other graph
+   massaging is done. To optimize the instrumentation we generate the
+   BB minimal span tree, only edges that are not on the span tree
+   (plus the entry point) need instrumenting. From that information
+   all other edge counts can be deduced.  By construction all fake
+   edges must be on the spanning tree. We also attempt to place
+   EDGE_CRITICAL edges on the spanning tree.
+
+   The auxiliary files generated are <dumpbase>.gcno (at compile time)
+   and <dumpbase>.gcda (at run time).  The format is
+   described in full in gcov-io.h.  */
 
 /* ??? Register allocation should use basic block execution counts to
    give preference to the most commonly executed blocks.  */
 
-/* ??? The .da files are not safe.  Changing the program after creating .da
-   files or using different options when compiling with -fbranch-probabilities
-   can result the arc data not matching the program.  Maybe add instrumented
-   arc count to .bbg file?  Maybe check whether PFG matches the .bbg file?  */
-
 /* ??? Should calculate branch probabilities before instrumenting code, since
    then we can use arc counts to help decide which arcs to instrument.  */
 
 #include "config.h"
 #include "system.h"
+#include "coretypes.h"
+#include "tm.h"
 #include "rtl.h"
-#include "tree.h"
 #include "flags.h"
-#include "insn-flags.h"
-#include "insn-config.h"
 #include "output.h"
 #include "regs.h"
 #include "expr.h"
 #include "function.h"
-#include "gcov-io.h"
 #include "toplev.h"
-#include "ggc.h"
-#include "hard-reg-set.h"
-#include "basic-block.h"
-#include "defaults.h"
-
-/* Additional information about the edges we need.  */
-struct edge_info
-  {
-    unsigned int count_valid : 1;
-    unsigned int on_tree : 1;
-    unsigned int ignore : 1;
-  };
-struct bb_info
-  {
-    unsigned int count_valid : 1;
-    int succ_count;
-    int pred_count;
-  };
-
-#define EDGE_INFO(e)  ((struct edge_info *) (e)->aux)
-#define BB_INFO(b)  ((struct bb_info *) (b)->aux)
+#include "coverage.h"
+#include "value-prof.h"
+#include "tree.h"
+#include "cfghooks.h"
+#include "tree-flow.h"
 
-/* Keep all basic block indexes nonnegative in the gcov output.  Index 0
-   is used for entry block, last block exit block.   */
-#define GCOV_INDEX_TO_BB(i)  ((i) == 0 ? ENTRY_BLOCK_PTR               \
-                             : (((i) == n_basic_blocks + 1)            \
-                                ? EXIT_BLOCK_PTR : BASIC_BLOCK ((i)-1)))
-#define BB_TO_GCOV_INDEX(bb)  ((bb) == ENTRY_BLOCK_PTR ? 0             \
-                              : ((bb) == EXIT_BLOCK_PTR                \
-                                 ? n_basic_blocks + 1 : (bb)->index + 1))
+/* Hooks for profiling.  */
+static struct profile_hooks* profile_hooks;
 
-/* Name and file pointer of the output file for the basic block graph.  */
+/* File for profiling debug output.  */
+static inline FILE*
+profile_dump_file (void) {
+  return profile_hooks->profile_dump_file ();
+}
 
-static FILE *bbg_file;
+/* Additional information about the edges we need.  */
+struct edge_info {
+  unsigned int count_valid : 1;
 
-/* Name and file pointer of the input file for the arc count data.  */
+  /* Is on the spanning tree.  */
+  unsigned int on_tree : 1;
 
-static FILE *da_file;
+  /* Pretend this edge does not exist (it is abnormal and we've
+     inserted a fake to compensate).  */
+  unsigned int ignore : 1;
+};
 
-/* Pointer of the output file for the basic block/line number map. */
-static FILE *bb_file;
+struct bb_info {
+  unsigned int count_valid : 1;
 
-/* Last source file name written to bb_file. */
+  /* Number of successor and predecessor edges.  */
+  gcov_type succ_count;
+  gcov_type pred_count;
+};
 
-static char *last_bb_file_name;
+#define EDGE_INFO(e)  ((struct edge_info *) (e)->aux)
+#define BB_INFO(b)  ((struct bb_info *) (b)->aux)
 
-/* Used by final, for allocating the proper amount of storage for the
-   instrumented arc execution counts.  */
+/* Counter summary from the last set of coverage counts read.  */
 
-int count_instrumented_edges;
+const struct gcov_ctr_summary *profile_info;
 
 /* Collect statistics on the performance of this pass for the entire source
    file.  */
 
 static int total_num_blocks;
 static int total_num_edges;
+static int total_num_edges_ignored;
 static int total_num_edges_instrumented;
 static int total_num_blocks_created;
 static int total_num_passes;
@@ -111,96 +117,142 @@ static int total_num_never_executed;
 static int total_num_branches;
 
 /* Forward declarations.  */
-static void find_spanning_tree PARAMS ((struct edge_list *));
-static void init_edge_profiler PARAMS ((void));
-static rtx gen_edge_profiler PARAMS ((int));
-static void instrument_edges PARAMS ((struct edge_list *));
-static void output_gcov_string PARAMS ((const char *, long));
-static void compute_branch_probabilities PARAMS ((void));
-static basic_block find_group PARAMS ((basic_block));
-static void union_groups PARAMS ((basic_block, basic_block));
-
-/* If non-zero, we need to output a constructor to set up the
-   per-object-file data. */
-static int need_func_profiler = 0;
+static void find_spanning_tree (struct edge_list *);
+static unsigned instrument_edges (struct edge_list *);
+static void instrument_values (histogram_values);
+static void compute_branch_probabilities (void);
+static void compute_value_histograms (histogram_values);
+static gcov_type * get_exec_counts (void);
+static basic_block find_group (basic_block);
+static void union_groups (basic_block, basic_block);
+
 \f
 /* Add edge instrumentation code to the entire insn chain.
 
    F is the first insn of the chain.
    NUM_BLOCKS is the number of basic blocks found in F.  */
 
-static void
-instrument_edges (el)
-     struct edge_list *el;
+static unsigned
+instrument_edges (struct edge_list *el)
 {
-  int i;
-  int num_instr_edges = 0;
+  unsigned num_instr_edges = 0;
   int num_edges = NUM_EDGES (el);
-  remove_fake_edges ();
+  basic_block bb;
 
-  for (i = 0; i < n_basic_blocks + 2; i++)
+  FOR_BB_BETWEEN (bb, ENTRY_BLOCK_PTR, NULL, next_bb)
     {
-      basic_block bb = GCOV_INDEX_TO_BB (i);
-      edge e = bb->succ;
-      while (e)
+      edge e;
+
+      for (e = bb->succ; e; e = e->succ_next)
        {
          struct edge_info *inf = EDGE_INFO (e);
+
          if (!inf->ignore && !inf->on_tree)
            {
              if (e->flags & EDGE_ABNORMAL)
                abort ();
-             if (rtl_dump_file)
-               fprintf (rtl_dump_file, "Edge %d to %d instrumented%s\n",
+             if (dump_file)
+               fprintf (dump_file, "Edge %d to %d instrumented%s\n",
                         e->src->index, e->dest->index,
-                        e->flags & EDGE_CRITICAL ? " (and split)" : "");
-             need_func_profiler = 1;
-             insert_insn_on_edge (
-                        gen_edge_profiler (total_num_edges_instrumented
-                                           + num_instr_edges++), e);
+                        EDGE_CRITICAL_P (e) ? " (and split)" : "");
+             (profile_hooks->gen_edge_profiler) (num_instr_edges++, e);
            }
-         e = e->succ_next;
        }
     }
 
-  total_num_edges_instrumented += num_instr_edges;
-  count_instrumented_edges = total_num_edges_instrumented;
-
   total_num_blocks_created += num_edges;
-  if (rtl_dump_file)
-    fprintf (rtl_dump_file, "%d edges instrumented\n", num_instr_edges);
-
-  commit_edge_insertions ();
+  if (dump_file)
+    fprintf (dump_file, "%d edges instrumented\n", num_instr_edges);
+  return num_instr_edges;
 }
 
-/* Output STRING to bb_file, surrounded by DELIMITER.  */
-
+/* Add code to measure histograms for values in list VALUES.  */
 static void
-output_gcov_string (string, delimiter)
-     const char *string;
-     long delimiter;
+instrument_values (histogram_values values)
 {
-  long temp;
-                       
-  /* Write a delimiter to indicate that a file name follows.  */
-  __write_long (delimiter, bb_file, 4);
-
-  /* Write the string.  */
-  temp = strlen (string) + 1;
-  fwrite (string, temp, 1, bb_file);
-
-  /* Append a few zeros, to align the output to a 4 byte boundary.  */
-  temp = temp & 0x3;
-  if (temp)
+  unsigned i, t;
+
+  /* Emit code to generate the histograms before the insns.  */
+
+  for (i = 0; i < VEC_length (histogram_value, values); i++)
     {
-      char c[4];
+      histogram_value hist = VEC_index (histogram_value, values, i);
+      switch (hist->type)
+       {
+       case HIST_TYPE_INTERVAL:
+         t = GCOV_COUNTER_V_INTERVAL;
+         break;
+
+       case HIST_TYPE_POW2:
+         t = GCOV_COUNTER_V_POW2;
+         break;
+
+       case HIST_TYPE_SINGLE_VALUE:
+         t = GCOV_COUNTER_V_SINGLE;
+         break;
+
+       case HIST_TYPE_CONST_DELTA:
+         t = GCOV_COUNTER_V_DELTA;
+         break;
+
+       default:
+         abort ();
+       }
+      if (!coverage_counter_alloc (t, hist->n_counters))
+       continue;
+
+      switch (hist->type)
+       {
+       case HIST_TYPE_INTERVAL:
+         (profile_hooks->gen_interval_profiler) (hist, t, 0);
+         break;
+
+       case HIST_TYPE_POW2:
+         (profile_hooks->gen_pow2_profiler) (hist, t, 0);
+         break;
+
+       case HIST_TYPE_SINGLE_VALUE:
+         (profile_hooks->gen_one_value_profiler) (hist, t, 0);
+         break;
+
+       case HIST_TYPE_CONST_DELTA:
+         (profile_hooks->gen_const_delta_profiler) (hist, t, 0);
+         break;
 
-      c[0] = c[1] = c[2] = c[3] = 0;
-      fwrite (c, sizeof (char), 4 - temp, bb_file);
+       default:
+         abort ();
+       }
     }
+}
+\f
+
+/* Computes hybrid profile for all matching entries in da_file.  */
+
+static gcov_type *
+get_exec_counts (void)
+{
+  unsigned num_edges = 0;
+  basic_block bb;
+  gcov_type *counts;
 
-  /* Store another delimiter in the .bb file, just to make it easy to find
-     the end of the file name.  */
-  __write_long (delimiter, bb_file, 4);
+  /* Count the edges to be (possibly) instrumented.  */
+  FOR_BB_BETWEEN (bb, ENTRY_BLOCK_PTR, NULL, next_bb)
+    {
+      edge e;
+      for (e = bb->succ; e; e = e->succ_next)
+       if (!EDGE_INFO (e)->ignore && !EDGE_INFO (e)->on_tree)
+         num_edges++;
+    }
+
+  counts = get_coverage_counts (GCOV_COUNTER_ARCS, num_edges, &profile_info);
+  if (!counts)
+    return NULL;
+
+  if (dump_file && profile_info)
+    fprintf(dump_file, "Merged %u profiles with maximal count %u.\n",
+           profile_info->runs, (unsigned) profile_info->sum_max);
+
+  return counts;
 }
 \f
 
@@ -208,8 +260,9 @@ output_gcov_string (string, delimiter)
    Annotate them accordingly.  */
 
 static void
-compute_branch_probabilities ()
+compute_branch_probabilities (void)
 {
+  basic_block bb;
   int i;
   int num_edges = 0;
   int changes;
@@ -217,19 +270,32 @@ compute_branch_probabilities ()
   int hist_br_prob[20];
   int num_never_executed;
   int num_branches;
-  int bad_counts = 0;
-  struct bb_info *bb_infos;
+  gcov_type *exec_counts = get_exec_counts ();
+  int exec_counts_pos = 0;
+
+  /* Very simple sanity checks so we catch bugs in our profiling code.  */
+  if (profile_info)
+    {
+      if (profile_info->run_max * profile_info->runs < profile_info->sum_max)
+       {
+         error ("corrupted profile info: run_max * runs < sum_max");
+         exec_counts = NULL;
+       }
+
+      if (profile_info->sum_all < profile_info->sum_max)
+       {
+         error ("corrupted profile info: sum_all is smaller than sum_max");
+         exec_counts = NULL;
+       }
+    }
 
   /* Attach extra info block to each bb.  */
 
-  bb_infos = (struct bb_info *)
-    xcalloc (n_basic_blocks + 2, sizeof (struct bb_info));
-  for (i = 0; i < n_basic_blocks + 2; i++)
+  alloc_aux_for_blocks (sizeof (struct bb_info));
+  FOR_BB_BETWEEN (bb, ENTRY_BLOCK_PTR, NULL, next_bb)
     {
-      basic_block bb = GCOV_INDEX_TO_BB (i);
       edge e;
 
-      bb->aux = &bb_infos[i];
       for (e = bb->succ; e; e = e->succ_next)
        if (!EDGE_INFO (e)->ignore)
          BB_INFO (bb)->succ_count++;
@@ -248,30 +314,40 @@ compute_branch_probabilities ()
   /* The first count in the .da file is the number of times that the function
      was entered.  This is the exec_count for block zero.  */
 
-  for (i = 0; i < n_basic_blocks + 2; i++)
+  FOR_BB_BETWEEN (bb, ENTRY_BLOCK_PTR, NULL, next_bb)
     {
-      basic_block bb = GCOV_INDEX_TO_BB (i);
       edge e;
       for (e = bb->succ; e; e = e->succ_next)
        if (!EDGE_INFO (e)->ignore && !EDGE_INFO (e)->on_tree)
          {
            num_edges++;
-           if (da_file)
+           if (exec_counts)
              {
-               long value;
-               __read_long (&value, da_file, 8);
-               e->count = value;
+               e->count = exec_counts[exec_counts_pos++];
+               if (e->count > profile_info->sum_max)
+                 {
+                   error ("corrupted profile info: edge from %i to %i exceeds maximal count",
+                          bb->index, e->dest->index);
+                 }
              }
            else
              e->count = 0;
+
            EDGE_INFO (e)->count_valid = 1;
            BB_INFO (bb)->succ_count--;
            BB_INFO (e->dest)->pred_count--;
+           if (dump_file)
+             {
+               fprintf (dump_file, "\nRead edge from %i to %i, count:",
+                        bb->index, e->dest->index);
+               fprintf (dump_file, HOST_WIDEST_INT_PRINT_DEC,
+                        (HOST_WIDEST_INT) e->count);
+             }
          }
     }
 
-  if (rtl_dump_file)
-    fprintf (rtl_dump_file, "%d edge counts read\n", num_edges);
+  if (dump_file)
+    fprintf (dump_file, "\n%d edge counts read\n", num_edges);
 
   /* For every block in the file,
      - if every exit/entrance edge has a known count, then set the block count
@@ -296,16 +372,15 @@ compute_branch_probabilities ()
     {
       passes++;
       changes = 0;
-      for (i = n_basic_blocks + 1; i >= 0; i--)
+      FOR_BB_BETWEEN (bb, EXIT_BLOCK_PTR, NULL, prev_bb)
        {
-         basic_block bb = GCOV_INDEX_TO_BB (i);
          struct bb_info *bi = BB_INFO (bb);
          if (! bi->count_valid)
            {
              if (bi->succ_count == 0)
                {
                  edge e;
-                 int total = 0;
+                 gcov_type total = 0;
 
                  for (e = bb->succ; e; e = e->succ_next)
                    total += e->count;
@@ -316,7 +391,7 @@ compute_branch_probabilities ()
              else if (bi->pred_count == 0)
                {
                  edge e;
-                 int total = 0;
+                 gcov_type total = 0;
 
                  for (e = bb->pred; e; e = e->pred_next)
                    total += e->count;
@@ -330,7 +405,7 @@ compute_branch_probabilities ()
              if (bi->succ_count == 1)
                {
                  edge e;
-                 int total = 0;
+                 gcov_type total = 0;
 
                  /* One of the counts will be invalid, but it is zero,
                     so adding it in also doesn't hurt.  */
@@ -350,23 +425,23 @@ compute_branch_probabilities ()
                  EDGE_INFO (e)->count_valid = 1;
                  e->count = total;
                  bi->succ_count--;
-                 
+
                  BB_INFO (e->dest)->pred_count--;
                  changes = 1;
                }
              if (bi->pred_count == 1)
                {
                  edge e;
-                 int total = 0;
+                 gcov_type total = 0;
 
                  /* One of the counts will be invalid, but it is zero,
                     so adding it in also doesn't hurt.  */
                  for (e = bb->pred; e; e = e->pred_next)
                    total += e->count;
 
-                 /* Seedgeh for the invalid edge, and set its count.  */
+                 /* Search for the invalid edge, and set its count.  */
                  for (e = bb->pred; e; e = e->pred_next)
-                   if (! EDGE_INFO (e)->count_valid && ! EDGE_INFO (e)->ignore)
+                   if (!EDGE_INFO (e)->count_valid && !EDGE_INFO (e)->ignore)
                      break;
 
                  /* Calculate count for remaining edge by conservation.  */
@@ -377,25 +452,24 @@ compute_branch_probabilities ()
                  EDGE_INFO (e)->count_valid = 1;
                  e->count = total;
                  bi->pred_count--;
-                 
+
                  BB_INFO (e->src)->succ_count--;
                  changes = 1;
                }
            }
        }
     }
-  if (rtl_dump_file)
-    dump_flow_info (rtl_dump_file);
+  if (dump_file)
+    dump_flow_info (dump_file);
 
   total_num_passes += passes;
-  if (rtl_dump_file)
-    fprintf (rtl_dump_file, "Graph solving took %d passes.\n\n", passes);
+  if (dump_file)
+    fprintf (dump_file, "Graph solving took %d passes.\n\n", passes);
 
   /* If the graph has been correctly solved, every block will have a
      succ and pred count of zero.  */
-  for (i = 0; i < n_basic_blocks; i++)
+  FOR_EACH_BB (bb)
     {
-      basic_block bb = BASIC_BLOCK (i);
       if (BB_INFO (bb)->succ_count || BB_INFO (bb)->pred_count)
        abort ();
     }
@@ -408,97 +482,123 @@ compute_branch_probabilities ()
   num_never_executed = 0;
   num_branches = 0;
 
-  for (i = 0; i < n_basic_blocks; i++)
+  FOR_BB_BETWEEN (bb, ENTRY_BLOCK_PTR, NULL, next_bb)
     {
-      basic_block bb = BASIC_BLOCK (i);
       edge e;
-      rtx insn;
-      int total;
       rtx note;
 
-      total = bb->count;
-      if (!total)
-       continue;
+      if (bb->count < 0)
+       {
+         error ("corrupted profile info: number of iterations for basic block %d thought to be %i",
+                bb->index, (int)bb->count);
+         bb->count = 0;
+       }
       for (e = bb->succ; e; e = e->succ_next)
        {
-         if (any_condjump_p (e->src->end)
-             && !EDGE_INFO (e)->ignore
-             && !(e->flags & (EDGE_ABNORMAL | EDGE_ABNORMAL_CALL | EDGE_FAKE)))
+         /* Function may return twice in the cased the called function is
+            setjmp or calls fork, but we can't represent this by extra
+            edge from the entry, since extra edge from the exit is
+            already present.  We get negative frequency from the entry
+            point.  */
+         if ((e->count < 0
+              && e->dest == EXIT_BLOCK_PTR)
+             || (e->count > bb->count
+                 && e->dest != EXIT_BLOCK_PTR))
+           {
+             if (block_ends_with_call_p (bb))
+               e->count = e->count < 0 ? 0 : bb->count;
+           }
+         if (e->count < 0 || e->count > bb->count)
+           {
+             error ("corrupted profile info: number of executions for edge %d-%d thought to be %i",
+                    e->src->index, e->dest->index,
+                    (int)e->count);
+             e->count = bb->count / 2;
+           }
+       }
+      if (bb->count)
+       {
+         for (e = bb->succ; e; e = e->succ_next)
+           e->probability = (e->count * REG_BR_PROB_BASE + bb->count / 2) / bb->count;
+         if (bb->index >= 0
+             && block_ends_with_condjump_p (bb)
+             && bb->succ->succ_next)
            {
              int prob;
-             /* This calculates the branch probability as an integer between
-                0 and REG_BR_PROB_BASE, properly rounded to the nearest
-                integer.  Perform the arithmetic in double to avoid
-                overflowing the range of ints.  */
-             if (total == 0)
-               prob = -1;
-             else
-               {
-                 prob = (((double)e->count * REG_BR_PROB_BASE)
-                         + (total >> 1)) / total;
-                 if (prob < 0 || prob > REG_BR_PROB_BASE)
-                   {
-                     if (rtl_dump_file)
-                       fprintf (rtl_dump_file, "bad count: prob for %d-%d thought to be %d (forcibly normalized)\n",
-                                e->src->index, e->dest->index, prob);
+             edge e;
+             int index;
 
-                     bad_counts = 1;
-                     prob = REG_BR_PROB_BASE / 2;
-                   }
-                 
-                 /* Match up probability with JUMP pattern.  */
-                 if (e->flags & EDGE_FALLTHRU)
-                   prob = REG_BR_PROB_BASE - prob;
-               }
-             
-             if (prob == -1)
-               num_never_executed++;
-             else
+             /* Find the branch edge.  It is possible that we do have fake
+                edges here.  */
+             for (e = bb->succ; e->flags & (EDGE_FAKE | EDGE_FALLTHRU);
+                  e = e->succ_next)
+               continue; /* Loop body has been intentionally left blank.  */
+
+             prob = e->probability;
+             index = prob * 20 / REG_BR_PROB_BASE;
+
+             if (index == 20)
+               index = 19;
+             hist_br_prob[index]++;
+
+             /* Do this for RTL only.  */
+             if (!ir_type ())
                {
-                 int index = prob * 20 / REG_BR_PROB_BASE;
-                 if (index == 20)
-                   index = 19;
-                 hist_br_prob[index]++;
+                 note = find_reg_note (BB_END (bb), REG_BR_PROB, 0);
+                 /* There may be already note put by some other pass, such
+                    as builtin_expect expander.  */
+                 if (note)
+                   XEXP (note, 0) = GEN_INT (prob);
+                 else
+                   REG_NOTES (BB_END (bb))
+                     = gen_rtx_EXPR_LIST (REG_BR_PROB, GEN_INT (prob),
+                                          REG_NOTES (BB_END (bb)));
                }
              num_branches++;
-             
-             note = find_reg_note (e->src->end, REG_BR_PROB, 0);
-             /* There may be already note put by some other pass, such
-                as builtin_expect expander.  */
-             if (note)
-               XEXP (note, 0) = GEN_INT (prob);
-             else
-               REG_NOTES (e->src->end)
-                 = gen_rtx_EXPR_LIST (REG_BR_PROB, GEN_INT (prob),
-                                      REG_NOTES (e->src->end));
            }
        }
+      /* Otherwise distribute the probabilities evenly so we get sane
+        sum.  Use simple heuristics that if there are normal edges,
+        give all abnormals frequency of 0, otherwise distribute the
+        frequency over abnormals (this is the case of noreturn
+        calls).  */
+      else
+       {
+         int total = 0;
 
-      /* Add a REG_EXEC_COUNT note to the first instruction of this block.  */
-      insn = next_nonnote_insn (bb->head);
-
-      if (GET_CODE (bb->head) == CODE_LABEL)
-       insn = next_nonnote_insn (insn);
-
-      /* Avoid crash on empty basic blocks.  */
-      if (insn && INSN_P (insn))
-       REG_NOTES (insn)
-         = gen_rtx_EXPR_LIST (REG_EXEC_COUNT, GEN_INT (total),
-                              REG_NOTES (insn));
+         for (e = bb->succ; e; e = e->succ_next)
+           if (!(e->flags & (EDGE_COMPLEX | EDGE_FAKE)))
+             total ++;
+         if (total)
+           {
+             for (e = bb->succ; e; e = e->succ_next)
+               if (!(e->flags & (EDGE_COMPLEX | EDGE_FAKE)))
+                 e->probability = REG_BR_PROB_BASE / total;
+               else
+                 e->probability = 0;
+           }
+         else
+           {
+             for (e = bb->succ; e; e = e->succ_next)
+               total ++;
+             for (e = bb->succ; e; e = e->succ_next)
+               e->probability = REG_BR_PROB_BASE / total;
+           }
+         if (bb->index >= 0
+             && block_ends_with_condjump_p (bb)
+             && bb->succ->succ_next)
+           num_branches++, num_never_executed;
+       }
     }
-  
-  /* This should never happen.  */
-  if (bad_counts)
-    warning ("Arc profiling: some edge counts were bad.");
 
-  if (rtl_dump_file)
+  if (dump_file)
     {
-      fprintf (rtl_dump_file, "%d branches\n", num_branches);
-      fprintf (rtl_dump_file, "%d branches never executed\n",
+      fprintf (dump_file, "%d branches\n", num_branches);
+      fprintf (dump_file, "%d branches never executed\n",
               num_never_executed);
       if (num_branches)
        for (i = 0; i < 10; i++)
-         fprintf (rtl_dump_file, "%d%% branches in range %d-%d%%\n",
+         fprintf (dump_file, "%d%% branches in range %d-%d%%\n",
                   (hist_br_prob[i] + hist_br_prob[19-i]) * 100 / num_branches,
                   5 * i, 5 * i + 5);
 
@@ -507,11 +607,128 @@ compute_branch_probabilities ()
       for (i = 0; i < 20; i++)
        total_hist_br_prob[i] += hist_br_prob[i];
 
-      fputc ('\n', rtl_dump_file);
-      fputc ('\n', rtl_dump_file);
+      fputc ('\n', dump_file);
+      fputc ('\n', dump_file);
     }
 
-  free (bb_infos);
+  free_aux_for_blocks ();
+}
+
+/* Load value histograms values whose description is stored in VALUES array
+   from .da file.  */
+
+static void
+compute_value_histograms (histogram_values values)
+{
+  unsigned i, j, t, any;
+  unsigned n_histogram_counters[GCOV_N_VALUE_COUNTERS];
+  gcov_type *histogram_counts[GCOV_N_VALUE_COUNTERS];
+  gcov_type *act_count[GCOV_N_VALUE_COUNTERS];
+  gcov_type *aact_count;
+  histogram_value hist;
+  for (t = 0; t < GCOV_N_VALUE_COUNTERS; t++)
+    n_histogram_counters[t] = 0;
+
+  for (i = 0; i < VEC_length (histogram_value, values); i++)
+    {
+      hist = VEC_index (histogram_value, values, i);
+      n_histogram_counters[(int) hist->type] += hist->n_counters;
+    }
+
+  any = 0;
+  for (t = 0; t < GCOV_N_VALUE_COUNTERS; t++)
+    {
+      if (!n_histogram_counters[t])
+       {
+         histogram_counts[t] = NULL;
+         continue;
+       }
+
+      histogram_counts[t] =
+       get_coverage_counts (COUNTER_FOR_HIST_TYPE (t),
+                            n_histogram_counters[t], NULL);
+      if (histogram_counts[t])
+       any = 1;
+      act_count[t] = histogram_counts[t];
+    }
+  if (!any)
+    return;
+
+  for (i = 0; i < VEC_length (histogram_value, values); i++)
+    {
+      rtx hist_list = NULL_RTX;
+
+      hist = VEC_index (histogram_value, values, i);
+      t = (int) hist->type;
+
+      /* FIXME: make this work for trees.  */
+      if (!ir_type ())
+       {
+         aact_count = act_count[t];
+         act_count[t] += hist->n_counters;
+         for (j = hist->n_counters; j > 0; j--)
+           hist_list = alloc_EXPR_LIST (0, GEN_INT (aact_count[j - 1]), 
+                                       hist_list);
+             hist_list = alloc_EXPR_LIST (0, 
+                           copy_rtx ((rtx) hist->value), hist_list);
+         hist_list = alloc_EXPR_LIST (0, GEN_INT (hist->type), hist_list);
+             REG_NOTES ((rtx) hist->insn) =
+                 alloc_EXPR_LIST (REG_VALUE_PROFILE, hist_list,
+                                  REG_NOTES ((rtx) hist->insn));
+       }
+    }
+
+  for (t = 0; t < GCOV_N_VALUE_COUNTERS; t++)
+    if (histogram_counts[t])
+      free (histogram_counts[t]);
+}
+
+#define BB_TO_GCOV_INDEX(bb)  ((bb)->index + 1)
+/* When passed NULL as file_name, initialize.
+   When passed something else, output the necessary commands to change
+   line to LINE and offset to FILE_NAME.  */
+static void
+output_location (char const *file_name, int line,
+                gcov_position_t *offset, basic_block bb)
+{
+  static char const *prev_file_name;
+  static int prev_line;
+  bool name_differs, line_differs;
+
+  if (!file_name)
+    {
+      prev_file_name = NULL;
+      prev_line = -1;
+      return;
+    }
+
+  name_differs = !prev_file_name || strcmp (file_name, prev_file_name);
+  line_differs = prev_line != line;
+
+  if (name_differs || line_differs)
+    {
+      if (!*offset)
+       {
+         *offset = gcov_write_tag (GCOV_TAG_LINES);
+         gcov_write_unsigned (BB_TO_GCOV_INDEX (bb));
+         name_differs = line_differs=true;
+       }
+
+      /* If this is a new source file, then output the
+        file's name to the .bb file.  */
+      if (name_differs)
+       {
+         prev_file_name = file_name;
+         gcov_write_unsigned (0);
+         gcov_write_string (prev_file_name);
+       }
+      if (line_differs)
+       {
+         gcov_write_unsigned (line);
+         prev_line = line;
+       }
+     }
 }
 
 /* Instrument and/or analyze program behavior based on program flow graph.
@@ -531,36 +748,41 @@ compute_branch_probabilities ()
    Main entry point of this file.  */
 
 void
-branch_prob ()
+branch_prob (void)
 {
-  int i;
-  int num_edges;
-  struct edge_info *edge_infos;
+  basic_block bb;
+  unsigned i;
+  unsigned num_edges, ignored_edges;
+  unsigned num_instrumented;
   struct edge_list *el;
-
-  /* Start of a function.  */
-  if (flag_test_coverage)
-    output_gcov_string (current_function_name, (long) -2);
+  histogram_values values = NULL;
 
   total_num_times_called++;
 
+  flow_call_edges_add (NULL);
+  add_noreturn_fake_exit_edges ();
+
   /* We can't handle cyclic regions constructed using abnormal edges.
      To avoid these we replace every source of abnormal edge by a fake
      edge from entry node and every destination by fake edge to exit.
      This keeps graph acyclic and our calculation exact for all normal
      edges except for exit and entrance ones.
-   
+
      We also add fake exit edges for each call and asm statement in the
      basic, since it may not return.  */
 
-  for (i = 0; i < n_basic_blocks ; i++)
+  FOR_EACH_BB (bb)
     {
-      rtx insn;
       int need_exit_edge = 0, need_entry_edge = 0;
       int have_exit_edge = 0, have_entry_edge = 0;
-      basic_block bb = BASIC_BLOCK (i);
       edge e;
 
+      /* Functions returning multiple times are not handled by extra edges.
+         Instead we simply allow negative counts on edges from exit to the
+         block past call and corresponding probabilities.  We can't go
+         with the extra edges because that would result in flowgraph that
+        needs to have fake edges outside the spanning tree.  */
+
       for (e = bb->succ; e; e = e->succ_next)
        {
          if ((e->flags & (EDGE_ABNORMAL | EDGE_ABNORMAL_CALL))
@@ -578,220 +800,318 @@ branch_prob ()
            have_entry_edge = 1;
        }
 
-      /* ??? Not strictly needed unless flag_test_coverage, but adding
-        them anyway keeps the .da file consistent.  */
-      /* ??? Currently inexact for basic blocks with multiple calls. 
-        We need to split blocks here.  */
-      for (insn = bb->head;
-          insn != NEXT_INSN (bb->end);
-          insn = NEXT_INSN (insn))
-       {
-         rtx set;
-         if (GET_CODE (insn) == CALL_INSN && !CONST_CALL_P (insn))
-           need_exit_edge = 1;
-         else if (GET_CODE (insn) == INSN)
-           {
-             set = PATTERN (insn);
-             if (GET_CODE (set) == PARALLEL)
-               set = XVECEXP (set, 0, 0);
-             if ((GET_CODE (set) == ASM_OPERANDS && MEM_VOLATILE_P (set))
-                 || GET_CODE (set) == ASM_INPUT)
-               need_exit_edge = 1;
-           }
-       }
-
       if (need_exit_edge && !have_exit_edge)
        {
-         if (rtl_dump_file)
-           fprintf (rtl_dump_file, "Adding fake exit edge to bb %i\n",
+         if (dump_file)
+           fprintf (dump_file, "Adding fake exit edge to bb %i\n",
                     bb->index);
-          make_edge (NULL, bb, EXIT_BLOCK_PTR, EDGE_FAKE);
+         make_edge (bb, EXIT_BLOCK_PTR, EDGE_FAKE);
        }
       if (need_entry_edge && !have_entry_edge)
        {
-         if (rtl_dump_file)
-           fprintf (rtl_dump_file, "Adding fake entry edge to bb %i\n",
+         if (dump_file)
+           fprintf (dump_file, "Adding fake entry edge to bb %i\n",
                     bb->index);
-          make_edge (NULL, ENTRY_BLOCK_PTR, bb, EDGE_FAKE);
+         make_edge (ENTRY_BLOCK_PTR, bb, EDGE_FAKE);
        }
     }
 
   el = create_edge_list ();
   num_edges = NUM_EDGES (el);
-  edge_infos = (struct edge_info *)
-    xcalloc (num_edges, sizeof (struct edge_info));
+  alloc_aux_for_edges (sizeof (struct edge_info));
 
+  /* The basic blocks are expected to be numbered sequentially.  */
+  compact_blocks ();
+
+  ignored_edges = 0;
   for (i = 0 ; i < num_edges ; i++)
     {
       edge e = INDEX_EDGE (el, i);
       e->count = 0;
-      e->aux = &edge_infos[i];
 
       /* Mark edges we've replaced by fake edges above as ignored.  */
       if ((e->flags & (EDGE_ABNORMAL | EDGE_ABNORMAL_CALL))
          && e->src != ENTRY_BLOCK_PTR && e->dest != EXIT_BLOCK_PTR)
-       EDGE_INFO (e)->ignore = 1;
+       {
+         EDGE_INFO (e)->ignore = 1;
+         ignored_edges++;
+       }
     }
 
-  total_num_blocks += n_basic_blocks + 2;
-  if (rtl_dump_file)
-    fprintf (rtl_dump_file, "%d basic blocks\n", n_basic_blocks);
-
-  total_num_edges += num_edges;
-  if (rtl_dump_file)
-    fprintf (rtl_dump_file, "%d edges\n", num_edges);
-
 #ifdef ENABLE_CHECKING
   verify_flow_info ();
 #endif
 
-  /* Output line number information about each basic block for
-     GCOV utility.  */
-  if (flag_test_coverage)
-    {
-      int i = 0;
-      for (i = 0 ; i < n_basic_blocks; i++)
-        {
-         basic_block bb = BASIC_BLOCK (i);
-         rtx insn = bb->head;
-          static int ignore_next_note = 0;
-
-         /* We are looking for line number notes.  Search backward before
-            basic block to find correct ones.  */
-         insn = prev_nonnote_insn (insn);
-         if (!insn)
-           insn = get_insns ();
-         else
-           insn = NEXT_INSN (insn);
+  /* Create spanning tree from basic block graph, mark each edge that is
+     on the spanning tree.  We insert as many abnormal and critical edges
+     as possible to minimize number of edge splits necessary.  */
 
-         /* Output a zero to the .bb file to indicate that a new
-            block list is starting.  */
-         __write_long (0, bb_file, 4);
+  find_spanning_tree (el);
 
-         while (insn != bb->end)
-           {
-             if (GET_CODE (insn) == NOTE)
-               {
-                 /* Must ignore the line number notes that immediately
-                    follow the end of an inline function to avoid counting
-                    it twice.  There is a note before the call, and one
-                    after the call.  */
-                 if (NOTE_LINE_NUMBER (insn) == NOTE_INSN_REPEATED_LINE_NUMBER)
-                   ignore_next_note = 1;
-                 else if (NOTE_LINE_NUMBER (insn) > 0)
-                   {
-                     if (ignore_next_note)
-                       ignore_next_note = 0;
-                     else
-                       {
-                         /* If this is a new source file, then output the
-                            file's name to the .bb file.  */
-                         if (! last_bb_file_name
-                             || strcmp (NOTE_SOURCE_FILE (insn),
-                                        last_bb_file_name))
-                           {
-                             if (last_bb_file_name)
-                               free (last_bb_file_name);
-                             last_bb_file_name
-                               = xstrdup (NOTE_SOURCE_FILE (insn));
-                             output_gcov_string (NOTE_SOURCE_FILE (insn),
-                                                 (long)-1);
-                           }
-                         /* Output the line number to the .bb file.  Must be
-                            done after the output_bb_profile_data() call, and
-                            after the file name is written, to ensure that it
-                            is correctly handled by gcov.  */
-                         __write_long (NOTE_LINE_NUMBER (insn), bb_file, 4);
-                       }
-                   }
-               }
-             insn = NEXT_INSN (insn);
-           }
-        }
-      __write_long (0, bb_file, 4);
+  /* Fake edges that are not on the tree will not be instrumented, so
+     mark them ignored.  */
+  for (num_instrumented = i = 0; i < num_edges; i++)
+    {
+      edge e = INDEX_EDGE (el, i);
+      struct edge_info *inf = EDGE_INFO (e);
+
+      if (inf->ignore || inf->on_tree)
+       /*NOP*/;
+      else if (e->flags & EDGE_FAKE)
+       {
+         inf->ignore = 1;
+         ignored_edges++;
+       }
+      else
+       num_instrumented++;
     }
 
-  /* Create spanning tree from basic block graph, mark each edge that is
-     on the spanning tree.  We insert as many abnormal and critical edges
-     as possible to minimize number of edge splits necesary. */
+  total_num_blocks += n_basic_blocks + 2;
+  if (dump_file)
+    fprintf (dump_file, "%d basic blocks\n", n_basic_blocks);
 
-  find_spanning_tree (el);
+  total_num_edges += num_edges;
+  if (dump_file)
+    fprintf (dump_file, "%d edges\n", num_edges);
+
+  total_num_edges_ignored += ignored_edges;
+  if (dump_file)
+    fprintf (dump_file, "%d ignored edges\n", ignored_edges);
 
-  /* Create a .bbg file from which gcov can reconstruct the basic block
-     graph.  First output the number of basic blocks, and then for every
-     edge output the source and target basic block numbers.
-     NOTE: The format of this file must be compatible with gcov.  */
+  /* Write the data from which gcov can reconstruct the basic block
+     graph.  */
 
-  if (flag_test_coverage)
+  /* Basic block flags */
+  if (coverage_begin_output ())
     {
-      int flag_bits;
+      gcov_position_t offset;
 
-      /* The plus 2 stands for entry and exit block.  */
-      __write_long (n_basic_blocks + 2, bbg_file, 4);
-      __write_long (num_edges + 1, bbg_file, 4);
+      offset = gcov_write_tag (GCOV_TAG_BLOCKS);
+      for (i = 0; i != (unsigned) (n_basic_blocks + 2); i++)
+       gcov_write_unsigned (0);
+      gcov_write_length (offset);
+    }
+
+   /* Keep all basic block indexes nonnegative in the gcov output.
+      Index 0 is used for entry block, last index is for exit block.
+      */
+  ENTRY_BLOCK_PTR->index = -1;
+  EXIT_BLOCK_PTR->index = last_basic_block;
+
+  /* Arcs */
+  if (coverage_begin_output ())
+    {
+      gcov_position_t offset;
 
-      for (i = 0; i < n_basic_blocks + 1; i++)
+      FOR_BB_BETWEEN (bb, ENTRY_BLOCK_PTR, EXIT_BLOCK_PTR, next_bb)
        {
-         basic_block bb = GCOV_INDEX_TO_BB (i);
          edge e;
-         long count = 0;
 
-         for (e = bb->succ; e; e = e->succ_next)
-           if (!EDGE_INFO (e)->ignore)
-             count++;
-         __write_long (count, bbg_file, 4);
+         offset = gcov_write_tag (GCOV_TAG_ARCS);
+         gcov_write_unsigned (BB_TO_GCOV_INDEX (bb));
 
          for (e = bb->succ; e; e = e->succ_next)
            {
              struct edge_info *i = EDGE_INFO (e);
              if (!i->ignore)
                {
-                 flag_bits = 0;
+                 unsigned flag_bits = 0;
+
                  if (i->on_tree)
-                   flag_bits |= 0x1;
-                 if (e->flags & EDGE_ABNORMAL)
-                   flag_bits |= 0x2;
+                   flag_bits |= GCOV_ARC_ON_TREE;
+                 if (e->flags & EDGE_FAKE)
+                   flag_bits |= GCOV_ARC_FAKE;
                  if (e->flags & EDGE_FALLTHRU)
-                   flag_bits |= 0x4;
-
-                 __write_long (BB_TO_GCOV_INDEX (e->dest), bbg_file, 4);
-                 __write_long (flag_bits, bbg_file, 4);
+                   flag_bits |= GCOV_ARC_FALLTHROUGH;
+                 /* On trees we don't have fallthru flags, but we can
+                    recompute them from CFG shape.  */
+                 if (ir_type ()
+                     && e->flags & (EDGE_TRUE_VALUE | EDGE_FALSE_VALUE)
+                     && e->src->next_bb == e->dest)
+                   flag_bits |= GCOV_ARC_FALLTHROUGH;
+
+                 gcov_write_unsigned (BB_TO_GCOV_INDEX (e->dest));
+                 gcov_write_unsigned (flag_bits);
                }
            }
+
+         gcov_write_length (offset);
        }
-      /* Emit fake loopback edge for EXIT block to maitain compatibility with
-         old gcov format.  */
-      __write_long (1, bbg_file, 4);
-      __write_long (0, bbg_file, 4);
-      __write_long (0x1, bbg_file, 4);
-
-      /* Emit a -1 to separate the list of all edges from the list of
-        loop back edges that follows.  */
-      __write_long (-1, bbg_file, 4);
     }
 
-  if (flag_branch_probabilities)
-    compute_branch_probabilities ();
+  /* Line numbers.  */
+  if (coverage_begin_output ())
+    {
+      /* Initialize the output.  */
+      output_location (NULL, 0, NULL, NULL);
+
+      if (!ir_type ())
+       {
+         gcov_position_t offset;
+
+         FOR_EACH_BB (bb)
+           {
+             rtx insn = BB_HEAD (bb);
+             int ignore_next_note = 0;
+
+             offset = 0;
+
+             /* We are looking for line number notes.  Search backward
+                before basic block to find correct ones.  */
+             insn = prev_nonnote_insn (insn);
+             if (!insn)
+               insn = get_insns ();
+             else
+               insn = NEXT_INSN (insn);
+
+             while (insn != BB_END (bb))
+               {
+                 if (NOTE_P (insn))
+                   {
+                     /* Must ignore the line number notes that
+                        immediately follow the end of an inline function
+                        to avoid counting it twice.  There is a note
+                        before the call, and one after the call.  */
+                     if (NOTE_LINE_NUMBER (insn)
+                         == NOTE_INSN_REPEATED_LINE_NUMBER)
+                       ignore_next_note = 1;
+                     else if (NOTE_LINE_NUMBER (insn) <= 0)
+                       /*NOP*/;
+                     else if (ignore_next_note)
+                       ignore_next_note = 0;
+                     else
+                       {
+                         expanded_location s;
+                         NOTE_EXPANDED_LOCATION (s, insn);
+                         output_location (s.file, NOTE_LINE_NUMBER (insn), &offset, bb);
+                       }
+                   }
+                 insn = NEXT_INSN (insn);
+               }
+
+             if (offset)
+               {
+                 /* A file of NULL indicates the end of run.  */
+                 gcov_write_unsigned (0);
+                 gcov_write_string (NULL);
+                 gcov_write_length (offset);
+               }
+           }
+       }
+      else
+       {
+         gcov_position_t offset;
+
+         FOR_EACH_BB (bb)
+           {
+             block_stmt_iterator bsi;
+
+             offset = 0;
+
+             if (bb == ENTRY_BLOCK_PTR->next_bb)
+               {
+                 expanded_location curr_location = 
+                   expand_location (DECL_SOURCE_LOCATION
+                                    (current_function_decl));
+                 output_location (curr_location.file, curr_location.line,
+                                  &offset, bb);
+               }
+
+             for (bsi = bsi_start (bb); !bsi_end_p (bsi); bsi_next (&bsi))
+               {
+                 tree stmt = bsi_stmt (bsi);
+                 if (EXPR_HAS_LOCATION (stmt))
+                   output_location (EXPR_FILENAME (stmt), 
+                                    EXPR_LINENO (stmt),
+                                    &offset, bb);
+               }
+
+             /* Notice GOTO expressions we eliminated while constructing the
+                CFG.  */
+             if (bb->succ && !bb->succ->succ_next && bb->succ->goto_locus)
+               {
+                 /* ??? source_locus type is marked deprecated in input.h.  */
+                 source_locus curr_location = bb->succ->goto_locus;
+                 /* ??? The FILE/LINE API is inconsistent for these cases.  */
+#ifdef USE_MAPPED_LOCATION 
+                 output_location (LOCATION_FILE (curr_location),
+                                  LOCATION_LINE (curr_location),
+                                  &offset, bb);
+#else
+                 output_location (curr_location->file, curr_location->line,
+                                  &offset, bb);
+#endif
+               }
 
-  /* For each edge not on the spanning tree, add counting code as rtl.  */
+             if (offset)
+               {
+                 /* A file of NULL indicates the end of run.  */
+                 gcov_write_unsigned (0);
+                 gcov_write_string (NULL);
+                 gcov_write_length (offset);
+               }
+           }
+        }
+    }
 
-  if (profile_arc_flag)
+  ENTRY_BLOCK_PTR->index = ENTRY_BLOCK;
+  EXIT_BLOCK_PTR->index = EXIT_BLOCK;
+#undef BB_TO_GCOV_INDEX
+
+  if (flag_profile_values)
+    find_values_to_profile (&values);
+
+  if (flag_branch_probabilities)
     {
-      instrument_edges (el);
-      allocate_reg_info (max_reg_num (), FALSE, FALSE);
+      compute_branch_probabilities ();
+      if (flag_profile_values)
+       compute_value_histograms (values);
     }
 
   remove_fake_edges ();
-  free (edge_infos);
+
+  /* For each edge not on the spanning tree, add counting code.  */
+  if (profile_arc_flag
+      && coverage_counter_alloc (GCOV_COUNTER_ARCS, num_instrumented))
+    {
+      unsigned n_instrumented = instrument_edges (el);
+
+      if (n_instrumented != num_instrumented)
+       abort ();
+
+      if (flag_profile_values)
+       instrument_values (values);
+
+      /* Commit changes done by instrumentation.  */
+      if (ir_type ())
+       bsi_commit_edge_inserts ((int *)NULL);
+      else
+       {
+          commit_edge_insertions_watch_calls ();
+         allocate_reg_info (max_reg_num (), FALSE, FALSE);
+       }
+    }
+
+  free_aux_for_edges ();
+
+  if (!ir_type ())
+    {
+      /* Re-merge split basic blocks and the mess introduced by
+        insert_insn_on_edge.  */
+      cleanup_cfg (profile_arc_flag ? CLEANUP_EXPENSIVE : 0);
+      if (profile_dump_file())
+       dump_flow_info (profile_dump_file());
+    }
+
   free_edge_list (el);
+  profile_status = PROFILE_READ;
 }
 \f
 /* Union find algorithm implementation for the basic blocks using
-   aux fields. */
+   aux fields.  */
 
 static basic_block
-find_group (bb)
-     basic_block bb;
+find_group (basic_block bb)
 {
   basic_block group = bb, bb1;
 
@@ -809,8 +1129,7 @@ find_group (bb)
 }
 
 static void
-union_groups (bb1, bb2)
-     basic_block bb1, bb2;
+union_groups (basic_block bb1, basic_block bb2)
 {
   basic_block bb1g = find_group (bb1);
   basic_block bb2g = find_group (bb2);
@@ -826,47 +1145,53 @@ union_groups (bb1, bb2)
 /* This function searches all of the edges in the program flow graph, and puts
    as many bad edges as possible onto the spanning tree.  Bad edges include
    abnormals edges, which can't be instrumented at the moment.  Since it is
-   possible for fake edges to form an cycle, we will have to develop some
+   possible for fake edges to form a cycle, we will have to develop some
    better way in the future.  Also put critical edges to the tree, since they
    are more expensive to instrument.  */
 
 static void
-find_spanning_tree (el)
-     struct edge_list *el;
+find_spanning_tree (struct edge_list *el)
 {
   int i;
   int num_edges = NUM_EDGES (el);
+  basic_block bb;
 
   /* We use aux field for standard union-find algorithm.  */
-  EXIT_BLOCK_PTR->aux = EXIT_BLOCK_PTR;
-  ENTRY_BLOCK_PTR->aux = ENTRY_BLOCK_PTR;
-  for (i = 0; i < n_basic_blocks; i++)
-    BASIC_BLOCK (i)->aux = BASIC_BLOCK (i);
+  FOR_BB_BETWEEN (bb, ENTRY_BLOCK_PTR, NULL, next_bb)
+    bb->aux = bb;
 
   /* Add fake edge exit to entry we can't instrument.  */
   union_groups (EXIT_BLOCK_PTR, ENTRY_BLOCK_PTR);
 
-  /* First add all abnormal edges to the tree unless they form an cycle.  */
+  /* First add all abnormal edges to the tree unless they form a cycle. Also
+     add all edges to EXIT_BLOCK_PTR to avoid inserting profiling code behind
+     setting return value from function.  */
   for (i = 0; i < num_edges; i++)
     {
       edge e = INDEX_EDGE (el, i);
-      if ((e->flags & (EDGE_ABNORMAL | EDGE_ABNORMAL_CALL | EDGE_FAKE))
+      if (((e->flags & (EDGE_ABNORMAL | EDGE_ABNORMAL_CALL | EDGE_FAKE))
+          || e->dest == EXIT_BLOCK_PTR)
          && !EDGE_INFO (e)->ignore
          && (find_group (e->src) != find_group (e->dest)))
        {
+         if (dump_file)
+           fprintf (dump_file, "Abnormal edge %d to %d put to tree\n",
+                    e->src->index, e->dest->index);
          EDGE_INFO (e)->on_tree = 1;
          union_groups (e->src, e->dest);
        }
     }
 
-  /* Now insert all critical edges to the tree unless they form an cycle.  */
+  /* Now insert all critical edges to the tree unless they form a cycle.  */
   for (i = 0; i < num_edges; i++)
     {
       edge e = INDEX_EDGE (el, i);
-      if ((e->flags & EDGE_CRITICAL)
-         && !EDGE_INFO (e)->ignore
-         && (find_group (e->src) != find_group (e->dest)))
+      if (EDGE_CRITICAL_P (e) && !EDGE_INFO (e)->ignore
+         && find_group (e->src) != find_group (e->dest))
        {
+         if (dump_file)
+           fprintf (dump_file, "Critical edge %d to %d put to tree\n",
+                    e->src->index, e->dest->index);
          EDGE_INFO (e)->on_tree = 1;
          union_groups (e->src, e->dest);
        }
@@ -876,75 +1201,31 @@ find_spanning_tree (el)
   for (i = 0; i < num_edges; i++)
     {
       edge e = INDEX_EDGE (el, i);
-      if (find_group (e->src) != find_group (e->dest)
-         && !EDGE_INFO (e)->ignore)
+      if (!EDGE_INFO (e)->ignore
+         && find_group (e->src) != find_group (e->dest))
        {
+         if (dump_file)
+           fprintf (dump_file, "Normal edge %d to %d put to tree\n",
+                    e->src->index, e->dest->index);
          EDGE_INFO (e)->on_tree = 1;
          union_groups (e->src, e->dest);
        }
     }
+
+  FOR_BB_BETWEEN (bb, ENTRY_BLOCK_PTR, NULL, next_bb)
+    bb->aux = NULL;
 }
 \f
 /* Perform file-level initialization for branch-prob processing.  */
 
 void
-init_branch_prob (filename)
-  const char *filename;
+init_branch_prob (void)
 {
-  long len;
   int i;
 
-  if (flag_test_coverage)
-    {
-      int len = strlen (filename);
-      char *data_file, *bbg_file_name;
-
-      /* Open an output file for the basic block/line number map.  */
-      data_file = (char *) alloca (len + 4);
-      strcpy (data_file, filename);
-      strip_off_ending (data_file, len);
-      strcat (data_file, ".bb");
-      if ((bb_file = fopen (data_file, "wb")) == 0)
-       pfatal_with_name (data_file);
-
-      /* Open an output file for the program flow graph.  */
-      bbg_file_name = (char *) alloca (len + 5);
-      strcpy (bbg_file_name, filename);
-      strip_off_ending (bbg_file_name, len);
-      strcat (bbg_file_name, ".bbg");
-      if ((bbg_file = fopen (bbg_file_name, "wb")) == 0)
-       pfatal_with_name (bbg_file_name);
-
-      /* Initialize to zero, to ensure that the first file name will be
-        written to the .bb file.  */
-      last_bb_file_name = 0;
-    }
-
-  if (flag_branch_probabilities)
-    {
-      char *da_file_name;
-
-      len = strlen (filename);
-      da_file_name = (char *) alloca (len + 4);
-      strcpy (da_file_name, filename);
-      strip_off_ending (da_file_name, len);
-      strcat (da_file_name, ".da");
-      if ((da_file = fopen (da_file_name, "rb")) == 0)
-       warning ("file %s not found, execution counts assumed to be zero.",
-                da_file_name);
-
-      /* The first word in the .da file gives the number of instrumented
-        edges, which is not needed for our purposes.  */
-
-      if (da_file)
-       __read_long (&len, da_file, 8);
-    }
-
-  if (profile_arc_flag)
-    init_edge_profiler ();
-
   total_num_blocks = 0;
   total_num_edges = 0;
+  total_num_edges_ignored = 0;
   total_num_edges_instrumented = 0;
   total_num_blocks_created = 0;
   total_num_passes = 0;
@@ -959,202 +1240,58 @@ init_branch_prob (filename)
    is completed.  */
 
 void
-end_branch_prob ()
+end_branch_prob (void)
 {
-  if (flag_test_coverage)
-    {
-      fclose (bb_file);
-      fclose (bbg_file);
-    }
-
-  if (flag_branch_probabilities)
-    {
-      if (da_file)
-       {
-         long temp;
-         /* This seems slightly dangerous, as it presumes the EOF
-            flag will not be set until an attempt is made to read
-            past the end of the file. */
-         if (feof (da_file))
-           warning (".da file contents exhausted too early\n");
-         /* Should be at end of file now.  */
-         if (__read_long (&temp, da_file, 8) == 0)
-           warning (".da file contents not exhausted\n");
-         fclose (da_file);
-       }
-    }
-
-  if (rtl_dump_file)
+  if (dump_file)
     {
-      fprintf (rtl_dump_file, "\n");
-      fprintf (rtl_dump_file, "Total number of blocks: %d\n",
+      fprintf (dump_file, "\n");
+      fprintf (dump_file, "Total number of blocks: %d\n",
               total_num_blocks);
-      fprintf (rtl_dump_file, "Total number of edges: %d\n", total_num_edges);
-      fprintf (rtl_dump_file, "Total number of instrumented edges: %d\n",
+      fprintf (dump_file, "Total number of edges: %d\n", total_num_edges);
+      fprintf (dump_file, "Total number of ignored edges: %d\n",
+              total_num_edges_ignored);
+      fprintf (dump_file, "Total number of instrumented edges: %d\n",
               total_num_edges_instrumented);
-      fprintf (rtl_dump_file, "Total number of blocks created: %d\n",
+      fprintf (dump_file, "Total number of blocks created: %d\n",
               total_num_blocks_created);
-      fprintf (rtl_dump_file, "Total number of graph solution passes: %d\n",
+      fprintf (dump_file, "Total number of graph solution passes: %d\n",
               total_num_passes);
       if (total_num_times_called != 0)
-       fprintf (rtl_dump_file, "Average number of graph solution passes: %d\n",
+       fprintf (dump_file, "Average number of graph solution passes: %d\n",
                 (total_num_passes + (total_num_times_called  >> 1))
                 / total_num_times_called);
-      fprintf (rtl_dump_file, "Total number of branches: %d\n",
+      fprintf (dump_file, "Total number of branches: %d\n",
               total_num_branches);
-      fprintf (rtl_dump_file, "Total number of branches never executed: %d\n",
+      fprintf (dump_file, "Total number of branches never executed: %d\n",
               total_num_never_executed);
       if (total_num_branches)
        {
          int i;
 
          for (i = 0; i < 10; i++)
-           fprintf (rtl_dump_file, "%d%% branches in range %d-%d%%\n",
+           fprintf (dump_file, "%d%% branches in range %d-%d%%\n",
                     (total_hist_br_prob[i] + total_hist_br_prob[19-i]) * 100
                     / total_num_branches, 5*i, 5*i+5);
        }
     }
 }
-\f
-/* The label used by the edge profiling code.  */
-
-static rtx profiler_label;
 
-/* Initialize the profiler_label.  */
+/* Set up hooks to enable tree-based profiling.  */
 
-static void
-init_edge_profiler ()
-{
-  /* Generate and save a copy of this so it can be shared.  */
-  char buf[20];
-  ASM_GENERATE_INTERNAL_LABEL (buf, "LPBX", 2);
-  profiler_label = gen_rtx_SYMBOL_REF (Pmode, ggc_strdup (buf));
-  ggc_add_rtx_root (&profiler_label, 1);
-}
-
-/* Output instructions as RTL to increment the edge execution count.  */
-
-static rtx
-gen_edge_profiler (edgeno)
-     int edgeno;
+void
+tree_register_profile_hooks (void)
 {
-  enum machine_mode mode = mode_for_size (LONG_TYPE_SIZE, MODE_INT, 0);
-  rtx mem_ref, tmp;
-  rtx sequence;
-
-  start_sequence ();
-
-  tmp = force_reg (Pmode, profiler_label);
-  tmp = plus_constant (tmp, LONG_TYPE_SIZE / BITS_PER_UNIT * edgeno);
-  mem_ref = validize_mem (gen_rtx_MEM (mode, tmp));
-
-  tmp = expand_binop (mode, add_optab, mem_ref, const1_rtx,
-                     mem_ref, 0, OPTAB_WIDEN);
-
-  if (tmp != mem_ref)
-    emit_move_insn (copy_rtx (mem_ref), tmp);
-
-  sequence = gen_sequence ();
-  end_sequence ();
-  return sequence;
+  profile_hooks = &tree_profile_hooks;
+  if (!ir_type ())
+    abort ();
 }
 
-/* Output code for a constructor that will invoke __bb_init_func, if
-   this has not already been done. */
+/* Set up hooks to enable RTL-based profiling.  */
 
 void
-output_func_start_profiler ()
+rtl_register_profile_hooks (void)
 {
-  tree fnname, fndecl;
-  char *name;
-  char buf[20];
-  const char *cfnname;
-  rtx table_address;
-  enum machine_mode mode = mode_for_size (LONG_TYPE_SIZE, MODE_INT, 0);
-  int save_flag_inline_functions = flag_inline_functions;
-  int save_flag_test_coverage = flag_test_coverage;
-  int save_profile_arc_flag = profile_arc_flag;
-  int save_flag_branch_probabilities = flag_branch_probabilities;
-
-  /* It's either already been output, or we don't need it because we're
-     not doing profile-edges. */
-  if (! need_func_profiler)
-    return;
-
-  need_func_profiler = 0;
-
-  /* Synthesize a constructor function to invoke __bb_init_func with a
-     pointer to this object file's profile block. */
-
-  /* Try and make a unique name given the "file function name".
-
-     And no, I don't like this either. */
-
-  fnname = get_file_function_name ('I');
-  cfnname = IDENTIFIER_POINTER (fnname);
-  name = xmalloc (strlen (cfnname) + 5);
-  sprintf (name, "%sGCOV",cfnname);
-  fnname = get_identifier (name);
-  free (name);
-
-  fndecl = build_decl (FUNCTION_DECL, fnname,
-                      build_function_type (void_type_node, NULL_TREE));
-  DECL_EXTERNAL (fndecl) = 0;
-
-#if defined(ASM_OUTPUT_CONSTRUCTOR) && defined(ASM_OUTPUT_DESTRUCTOR)
-  /* It can be a static function as long as collect2 does not have
-     to scan the object file to find its ctor/dtor routine.  */
-  TREE_PUBLIC (fndecl) = 0;
-#else
-  TREE_PUBLIC (fndecl) = 1;
-#endif
-
-  DECL_ASSEMBLER_NAME (fndecl) = fnname;
-  DECL_RESULT (fndecl) = build_decl (RESULT_DECL, NULL_TREE, void_type_node);
-
-  fndecl = pushdecl (fndecl);
-  rest_of_decl_compilation (fndecl, 0, 1, 0);
-  announce_function (fndecl);
-  current_function_decl = fndecl;
-  DECL_INITIAL (fndecl) = error_mark_node;
-  make_function_rtl (fndecl);
-  init_function_start (fndecl, input_filename, lineno);
-  pushlevel (0);
-  expand_function_start (fndecl, 0);
-
-  /* Actually generate the code to call __bb_init_func. */
-  ASM_GENERATE_INTERNAL_LABEL (buf, "LPBX", 0);
-  table_address = force_reg (Pmode,
-                            gen_rtx_SYMBOL_REF (Pmode, ggc_strdup (buf)));
-  emit_library_call (gen_rtx_SYMBOL_REF (Pmode, "__bb_init_func"), 0,
-                    mode, 1, table_address, Pmode);
-
-  expand_function_end (input_filename, lineno, 0);
-  poplevel (1, 0, 1);
-
-  /* Since fndecl isn't in the list of globals, it would never be emitted
-     when it's considered to be 'safe' for inlining, so turn off
-     flag_inline_functions.  */
-  flag_inline_functions = 0;
-
-  /* Don't instrument the function that turns on instrumentation.  Which
-     is also handy since we'd get silly warnings about not consuming all
-     of our da_file input.  */
-  flag_test_coverage = 0;
-  profile_arc_flag = 0;
-  flag_branch_probabilities = 0;
-
-  rest_of_compilation (fndecl);
-
-  /* Reset flag_inline_functions to its original value.  */
-  flag_inline_functions = save_flag_inline_functions;
-  flag_test_coverage = save_flag_test_coverage;
-  profile_arc_flag = save_profile_arc_flag;
-  flag_branch_probabilities = save_flag_branch_probabilities;
-
-  if (! quiet_flag)
-    fflush (asm_out_file);
-  current_function_decl = NULL_TREE;
-
-  assemble_constructor (IDENTIFIER_POINTER (DECL_NAME (fndecl)));
+  profile_hooks = &rtl_profile_hooks;
+  if (ir_type ())
+    abort ();
 }