OSDN Git Service

gcc/fortran/:
[pf3gnuchains/gcc-fork.git] / gcc / fortran / data.c
index 10fcd48..c217e1c 100644 (file)
@@ -1,5 +1,5 @@
 /* Supporting functions for resolving DATA statement.
-   Copyright (C) 2002, 2003, 2004, 2005, 2006, 2007
+   Copyright (C) 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010
    Free Software Foundation, Inc.
    Contributed by Lifang Zeng <zlf605@hotmail.com>
 
@@ -23,7 +23,7 @@ along with GCC; see the file COPYING3.  If not see
 /* Notes for DATA statement implementation:
                                                                               
    We first assign initial value to each symbol by gfc_assign_data_value
-   during resolveing DATA statement. Refer to check_data_variable and
+   during resolving DATA statement. Refer to check_data_variable and
    traverse_data_list in resolve.c.
                                                                               
    The complexity exists in the handling of array section, implied do
@@ -36,6 +36,7 @@ along with GCC; see the file COPYING3.  If not see
 #include "config.h"
 #include "gfortran.h"
 #include "data.h"
+#include "constructor.h"
 
 static void formalize_init_expr (gfc_expr *);
 
@@ -46,7 +47,6 @@ get_array_index (gfc_array_ref *ar, mpz_t *offset)
 {
   gfc_expr *e;
   int i;
-  try re;
   mpz_t delta;
   mpz_t tmp;
 
@@ -56,7 +56,7 @@ get_array_index (gfc_array_ref *ar, mpz_t *offset)
   for (i = 0; i < ar->dimen; i++)
     {
       e = gfc_copy_expr (ar->start[i]);
-      re = gfc_simplify_expr (e, 1);
+      gfc_simplify_expr (e, 1);
 
       if ((gfc_is_constant_expr (ar->as->lower[i]) == 0)
          || (gfc_is_constant_expr (ar->as->upper[i]) == 0)
@@ -77,104 +77,44 @@ get_array_index (gfc_array_ref *ar, mpz_t *offset)
   mpz_clear (tmp);
 }
 
-
-/* Find if there is a constructor which offset is equal to OFFSET.  */
+/* Find if there is a constructor which component is equal to COM.
+   TODO: remove this, use symbol.c(gfc_find_component) instead.  */
 
 static gfc_constructor *
-find_con_by_offset (splay_tree spt, mpz_t offset)
+find_con_by_component (gfc_component *com, gfc_constructor_base base)
 {
-  mpz_t tmp;
-  gfc_constructor *ret = NULL;
-  gfc_constructor *con;
-  splay_tree_node sptn;
-
-  /* The complexity is due to needing quick access to the linked list of
-     constructors.  Both a linked list and a splay tree are used, and both
-     are kept up to date if they are array elements (which is the only time
-     that a specific constructor has to be found).  */  
-
-  gcc_assert (spt != NULL);
-  mpz_init (tmp);
-
-  sptn = splay_tree_lookup (spt, (splay_tree_key) mpz_get_si (offset));
-
-  if (sptn)
-    ret = (gfc_constructor*) sptn->value;  
-  else
-    {
-       /* Need to check and see if we match a range, so we will pull
-         the next lowest index and see if the range matches.  */
-       sptn = splay_tree_predecessor (spt,
-                                     (splay_tree_key) mpz_get_si (offset));
-       if (sptn)
-        {
-           con = (gfc_constructor*) sptn->value;
-           if (mpz_cmp_ui (con->repeat, 1) > 0)
-             {
-                mpz_init (tmp);
-                mpz_add (tmp, con->n.offset, con->repeat);
-                if (mpz_cmp (offset, tmp) < 0)
-                  ret = con;
-                mpz_clear (tmp);
-             }
-           else 
-             ret = NULL; /* The range did not match.  */
-        }
-      else
-       ret = NULL; /* No pred, so no match.  */
-    }
-
-  return ret;
-}
-
+  gfc_constructor *c;
 
-/* Find if there is a constructor which component is equal to COM.  */
+  for (c = gfc_constructor_first (base); c; c = gfc_constructor_next (c))
+    if (com == c->n.component)
+      return c;
 
-static gfc_constructor *
-find_con_by_component (gfc_component *com, gfc_constructor *con)
-{
-  for (; con; con = con->next)
-    {
-      if (com == con->n.component)
-       return con;
-    }
   return NULL;
 }
 
 
 /* Create a character type initialization expression from RVALUE.
    TS [and REF] describe [the substring of] the variable being initialized.
-   INIT is thh existing initializer, not NULL.  Initialization is performed
+   INIT is the existing initializer, not NULL.  Initialization is performed
    according to normal assignment rules.  */
 
 static gfc_expr *
 create_character_intializer (gfc_expr *init, gfc_typespec *ts,
                             gfc_ref *ref, gfc_expr *rvalue)
 {
-  int len;
-  int start;
-  int end;
-  char *dest, *rvalue_string;
+  int len, start, end;
+  gfc_char_t *dest;
            
-  gfc_extract_int (ts->cl->length, &len);
+  gfc_extract_int (ts->u.cl->length, &len);
 
   if (init == NULL)
     {
       /* Create a new initializer.  */
-      init = gfc_get_expr ();
-      init->expr_type = EXPR_CONSTANT;
+      init = gfc_get_character_expr (ts->kind, NULL, NULL, len);
       init->ts = *ts;
-      
-      dest = gfc_getmem (len + 1);
-      dest[len] = '\0';
-      init->value.character.length = len;
-      init->value.character.string = dest;
-      /* Blank the string if we're only setting a substring.  */
-      if (ref != NULL)
-       memset (dest, ' ', len);
     }
-  else
-    dest = init->value.character.string;
+
+  dest = init->value.character.string;
 
   if (ref)
     {
@@ -208,15 +148,9 @@ create_character_intializer (gfc_expr *init, gfc_typespec *ts,
 
   /* Copy the initial value.  */
   if (rvalue->ts.type == BT_HOLLERITH)
-    {
-      len = rvalue->representation.length;
-      rvalue_string = rvalue->representation.string;
-    }
+    len = rvalue->representation.length;
   else
-    {
-      len = rvalue->value.character.length;
-      rvalue_string = rvalue->value.character.string;
-    }
+    len = rvalue->value.character.length;
 
   if (len > end - start)
     {
@@ -225,16 +159,26 @@ create_character_intializer (gfc_expr *init, gfc_typespec *ts,
                       "at %L", &rvalue->where);
     }
 
-  memcpy (&dest[start], rvalue_string, len);
+  if (rvalue->ts.type == BT_HOLLERITH)
+    {
+      int i;
+      for (i = 0; i < len; i++)
+       dest[start+i] = rvalue->representation.string[i];
+    }
+  else
+    memcpy (&dest[start], rvalue->value.character.string,
+           len * sizeof (gfc_char_t));
 
   /* Pad with spaces.  Substrings will already be blanked.  */
   if (len < end - start && ref == NULL)
-    memset (&dest[start + len], ' ', end - (start + len));
+    gfc_wide_memset (&dest[start + len], ' ', end - (start + len));
 
   if (rvalue->ts.type == BT_HOLLERITH)
     {
       init->representation.length = init->value.character.length;
-      init->representation.string = init->value.character.string;
+      init->representation.string
+       = gfc_widechar_to_char (init->value.character.string,
+                               init->value.character.length);
     }
 
   return init;
@@ -245,7 +189,7 @@ create_character_intializer (gfc_expr *init, gfc_typespec *ts,
    LVALUE already has an initialization, we extend this, otherwise we
    create a new one.  */
 
-try
+gfc_try
 gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index)
 {
   gfc_ref *ref;
@@ -253,12 +197,9 @@ gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index)
   gfc_expr *expr;
   gfc_constructor *con;
   gfc_constructor *last_con;
-  gfc_constructor *pred;
   gfc_symbol *symbol;
   gfc_typespec *last_ts;
   mpz_t offset;
-  splay_tree spt;
-  splay_tree_node sptn;
 
   symbol = lvalue->symtree->n.sym;
   init = symbol->value;
@@ -288,6 +229,14 @@ gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index)
       switch (ref->type)
        {
        case REF_ARRAY:
+         if (ref->u.ar.as->rank == 0)
+           {
+             gcc_assert (ref->u.ar.as->corank > 0);
+             if (init == NULL)
+               gfc_free (expr);
+             continue;
+           }
+
          if (init && expr->expr_type != EXPR_ARRAY)
            {
              gfc_error ("'%s' at %L already is initialized at %L",
@@ -321,51 +270,26 @@ gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index)
          else
            {
              mpz_t size;
-             if (spec_size (ref->u.ar.as, &size) == SUCCESS
-                  && mpz_cmp (offset, size) >= 0)
+             if (spec_size (ref->u.ar.as, &size) == SUCCESS)
                {
+                 if (mpz_cmp (offset, size) >= 0)
+                 {
+                   mpz_clear (size);
+                   gfc_error ("Data element above array upper bound at %L",
+                              &lvalue->where);
+                   return FAILURE;
+                 }
                  mpz_clear (size);
-                 gfc_error ("Data element above array upper bound at %L",
-                            &lvalue->where);
-                 return FAILURE;
                }
-             mpz_clear (size);
            }
 
-         /* Splay tree containing offset and gfc_constructor.  */
-         spt = expr->con_by_offset;
-
-         if (spt == NULL)
+         con = gfc_constructor_lookup (expr->value.constructor,
+                                       mpz_get_si (offset));
+         if (!con)
            {
-              spt = splay_tree_new (splay_tree_compare_ints, NULL, NULL);
-              expr->con_by_offset = spt; 
-              con = NULL;
-           }
-        else
-         con = find_con_by_offset (spt, offset);
-
-         if (con == NULL)
-           {
-             splay_tree_key j;
-
-             /* Create a new constructor.  */
-             con = gfc_get_constructor ();
-             mpz_set (con->n.offset, offset);
-             j = (splay_tree_key) mpz_get_si (offset);
-             sptn = splay_tree_insert (spt, j, (splay_tree_value) con);
-             /* Fix up the linked list.  */
-             sptn = splay_tree_predecessor (spt, j);
-             if (sptn == NULL)
-               {  /* Insert at the head.  */
-                  con->next = expr->value.constructor;
-                  expr->value.constructor = con;
-               }
-             else
-               {  /* Insert in the chain.  */
-                  pred = (gfc_constructor*) sptn->value;
-                  con->next = pred->next;
-                  pred->next = con;
-               }
+             con = gfc_constructor_insert_expr (&expr->value.constructor,
+                                                NULL, &rvalue->where,
+                                                mpz_get_si (offset));
            }
          break;
 
@@ -375,23 +299,22 @@ gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index)
              /* Setup the expression to hold the constructor.  */
              expr->expr_type = EXPR_STRUCTURE;
              expr->ts.type = BT_DERIVED;
-             expr->ts.derived = ref->u.c.sym;
+             expr->ts.u.derived = ref->u.c.sym;
            }
          else
            gcc_assert (expr->expr_type == EXPR_STRUCTURE);
          last_ts = &ref->u.c.component->ts;
 
          /* Find the same element in the existing constructor.  */
-         con = expr->value.constructor;
-         con = find_con_by_component (ref->u.c.component, con);
+         con = find_con_by_component (ref->u.c.component,
+                                      expr->value.constructor);
 
          if (con == NULL)
            {
              /* Create a new constructor.  */
-             con = gfc_get_constructor ();
+             con = gfc_constructor_append_expr (&expr->value.constructor,
+                                                NULL, NULL);
              con->n.component = ref->u.c.component;
-             con->next = expr->value.constructor;
-             expr->value.constructor = con;
            }
          break;
 
@@ -412,7 +335,11 @@ gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index)
     }
 
   if (ref || last_ts->type == BT_CHARACTER)
-    expr = create_character_intializer (init, last_ts, ref, rvalue);
+    {
+      if (lvalue->ts.u.cl->length == NULL && !(ref && ref->u.ss.length != NULL))
+       return FAILURE;
+      expr = create_character_intializer (init, last_ts, ref, rvalue);
+    }
   else
     {
       /* Overwriting an existing initializer is non-standard but usually only
@@ -422,16 +349,13 @@ gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index)
          /* Order in which the expressions arrive here depends on whether
             they are from data statements or F95 style declarations.
             Therefore, check which is the most recent.  */
-#ifdef USE_MAPPED_LOCATION
          expr = (LOCATION_LINE (init->where.lb->location)
                  > LOCATION_LINE (rvalue->where.lb->location))
               ? init : rvalue;
-#else
-         expr = (init->where.lb->linenum > rvalue->where.lb->linenum)
-              ? init : rvalue;
-#endif
-         gfc_notify_std (GFC_STD_GNU, "Extension: re-initialization "
-                         "of '%s' at %L", symbol->name, &expr->where);
+         if (gfc_notify_std (GFC_STD_GNU,"Extension: "
+                             "re-initialization of '%s' at %L",
+                             symbol->name, &expr->where) == FAILURE)
+           return FAILURE;
        }
 
       expr = gfc_copy_expr (rvalue);
@@ -449,182 +373,35 @@ gfc_assign_data_value (gfc_expr *lvalue, gfc_expr *rvalue, mpz_t index)
 
 
 /* Similarly, but initialize REPEAT consecutive values in LVALUE the same
-   value in RVALUE.  For the nonce, LVALUE must refer to a full array, not
-   an array section.  */
+   value in RVALUE.  */
 
-void
+gfc_try
 gfc_assign_data_value_range (gfc_expr *lvalue, gfc_expr *rvalue,
                             mpz_t index, mpz_t repeat)
 {
-  gfc_ref *ref;
-  gfc_expr *init, *expr;
-  gfc_constructor *con, *last_con;
-  gfc_constructor *pred;
-  gfc_symbol *symbol;
-  gfc_typespec *last_ts;
-  mpz_t offset;
-  splay_tree spt;
-  splay_tree_node sptn;
-
-  symbol = lvalue->symtree->n.sym;
-  init = symbol->value;
-  last_ts = &symbol->ts;
-  last_con = NULL;
-  mpz_init_set_si (offset, 0);
-
-  /* Find/create the parent expressions for subobject references.  */
-  for (ref = lvalue->ref; ref; ref = ref->next)
-    {
-      /* Use the existing initializer expression if it exists.
-        Otherwise create a new one.  */
-      if (init == NULL)
-       expr = gfc_get_expr ();
-      else
-       expr = init;
-
-      /* Find or create this element.  */
-      switch (ref->type)
-       {
-       case REF_ARRAY:
-         if (init == NULL)
-           {
-             /* The element typespec will be the same as the array
-                typespec.  */
-             expr->ts = *last_ts;
-             /* Setup the expression to hold the constructor.  */
-             expr->expr_type = EXPR_ARRAY;
-             expr->rank = ref->u.ar.as->rank;
-           }
-         else
-           gcc_assert (expr->expr_type == EXPR_ARRAY);
-
-         if (ref->u.ar.type == AR_ELEMENT)
-           {
-             get_array_index (&ref->u.ar, &offset);
-
-             /* This had better not be the bottom of the reference.
-                We can still get to a full array via a component.  */
-             gcc_assert (ref->next != NULL);
-           }
-         else
-           {
-             mpz_set (offset, index);
-
-             /* We're at a full array or an array section.  This means
-                that we've better have found a full array, and that we're
-                at the bottom of the reference.  */
-             gcc_assert (ref->u.ar.type == AR_FULL);
-             gcc_assert (ref->next == NULL);
-           }
-
-         /* Find the same element in the existing constructor.  */
-
-         /* Splay tree containing offset and gfc_constructor.  */
-         spt = expr->con_by_offset;
-
-         if (spt == NULL)
-           {
-              spt = splay_tree_new (splay_tree_compare_ints, NULL, NULL);
-              expr->con_by_offset = spt;
-              con = NULL;
-           }
-         else 
-           con = find_con_by_offset (spt, offset);
-
-         if (con == NULL)
-           {
-             splay_tree_key j;
-             /* Create a new constructor.  */
-             con = gfc_get_constructor ();
-             mpz_set (con->n.offset, offset);
-             j = (splay_tree_key) mpz_get_si (offset);
-         
-             if (ref->next == NULL)
-               mpz_set (con->repeat, repeat);
-             sptn = splay_tree_insert (spt, j, (splay_tree_value) con);
-             /* Fix up the linked list.  */
-             sptn = splay_tree_predecessor (spt, j);
-             if (sptn == NULL)
-               {  /* Insert at the head.  */
-                  con->next = expr->value.constructor;
-                  expr->value.constructor = con;
-               }
-             else
-               {  /* Insert in the chain.  */
-                  pred = (gfc_constructor*) sptn->value;
-                  con->next = pred->next;
-                  pred->next = con;
-               }
-           }
-         else
-           gcc_assert (ref->next != NULL);
-         break;
-
-       case REF_COMPONENT:
-         if (init == NULL)
-           {
-             /* Setup the expression to hold the constructor.  */
-             expr->expr_type = EXPR_STRUCTURE;
-             expr->ts.type = BT_DERIVED;
-             expr->ts.derived = ref->u.c.sym;
-           }
-         else
-           gcc_assert (expr->expr_type == EXPR_STRUCTURE);
-         last_ts = &ref->u.c.component->ts;
-
-         /* Find the same element in the existing constructor.  */
-         con = expr->value.constructor;
-         con = find_con_by_component (ref->u.c.component, con);
-
-         if (con == NULL)
-           {
-             /* Create a new constructor.  */
-             con = gfc_get_constructor ();
-             con->n.component = ref->u.c.component;
-             con->next = expr->value.constructor;
-             expr->value.constructor = con;
-           }
-
-         /* Since we're only intending to initialize arrays here,
-            there better be an inner reference.  */
-         gcc_assert (ref->next != NULL);
-         break;
-
-       case REF_SUBSTRING:
-       default:
-         gcc_unreachable ();
-       }
-
-      if (init == NULL)
-       {
-         /* Point the container at the new expression.  */
-         if (last_con == NULL)
-           symbol->value = expr;
-         else
-           last_con->expr = expr;
-       }
-      init = con->expr;
-      last_con = con;
-    }
+  mpz_t offset, last_offset;
+  gfc_try t;
+
+  mpz_init (offset);
+  mpz_init (last_offset);
+  mpz_add (last_offset, index, repeat);
+
+  t = SUCCESS;
+  for (mpz_set(offset, index) ; mpz_cmp(offset, last_offset) < 0;
+                  mpz_add_ui (offset, offset, 1))
+    if (gfc_assign_data_value (lvalue, rvalue, offset) == FAILURE)
+      {
+       t = FAILURE;
+       break;
+      }
 
-  if (last_ts->type == BT_CHARACTER)
-    expr = create_character_intializer (init, last_ts, NULL, rvalue);
-  else
-    {
-      /* We should never be overwriting an existing initializer.  */
-      gcc_assert (!init);
+  mpz_clear (offset);
+  mpz_clear (last_offset);
 
-      expr = gfc_copy_expr (rvalue);
-      if (!gfc_compare_types (&lvalue->ts, &expr->ts))
-       gfc_convert_type (expr, &lvalue->ts, 0);
-    }
-
-  if (last_con == NULL)
-    symbol->value = expr;
-  else
-    last_con->expr = expr;
+  return t;
 }
 
+
 /* Modify the index of array section and re-calculate the array offset.  */
 
 void 
@@ -699,63 +476,34 @@ gfc_advance_section (mpz_t *section_index, gfc_array_ref *ar,
 static void
 formalize_structure_cons (gfc_expr *expr)
 {
-  gfc_constructor *head;
-  gfc_constructor *tail;
+  gfc_constructor_base base = NULL;
   gfc_constructor *cur;
-  gfc_constructor *last;
-  gfc_constructor *c;
   gfc_component *order;
 
-  c = expr->value.constructor;
-
   /* Constructor is already formalized.  */
-  if (!c || c->n.component == NULL)
+  cur = gfc_constructor_first (expr->value.constructor);
+  if (!cur || cur->n.component == NULL)
     return;
 
-  head = tail = NULL;
-  for (order = expr->ts.derived->components; order; order = order->next)
+  for (order = expr->ts.u.derived->components; order; order = order->next)
     {
-      /* Find the next component.  */
-      last = NULL;
-      cur = c;
-      while (cur != NULL && cur->n.component != order)
-       {
-         last = cur;
-         cur = cur->next;
-       }
-
-      if (cur == NULL)
-       {
-         /* Create a new one.  */
-         cur = gfc_get_constructor ();
-       }
+      cur = find_con_by_component (order, expr->value.constructor);
+      if (cur)
+       gfc_constructor_append_expr (&base, cur->expr, &cur->expr->where);
       else
-       {
-         /* Remove it from the chain.  */
-         if (last == NULL)
-           c = cur->next;
-         else
-           last->next = cur->next;
-         cur->next = NULL;
+       gfc_constructor_append_expr (&base, NULL, NULL);
+    }
 
-         formalize_init_expr (cur->expr);
-       }
+  /* For all what it's worth, one would expect
+       gfc_constructor_free (expr->value.constructor);
+     here. However, if the constructor is actually free'd,
+     hell breaks loose in the testsuite?!  */
 
-      /* Add it to the new constructor.  */
-      if (head == NULL)
-       head = tail = cur;
-      else
-       {
-         tail->next = cur;
-         tail = tail->next;
-       }
-    }
-  gcc_assert (c == NULL);
-  expr->value.constructor = head;
+  expr->value.constructor = base;
 }
 
 
-/* Make sure an initialization expression is in normalized form.  Ie. all
+/* Make sure an initialization expression is in normalized form, i.e., all
    elements of the constructors are in the correct order.  */
 
 static void
@@ -771,13 +519,11 @@ formalize_init_expr (gfc_expr *expr)
   switch (type)
     {
     case EXPR_ARRAY:
-      c = expr->value.constructor;
-      while (c)
-       {
-         formalize_init_expr (c->expr);
-         c = c->next;
-       }
-      break;
+      for (c = gfc_constructor_first (expr->value.constructor);
+          c; c = gfc_constructor_next (c))
+       formalize_init_expr (c->expr);
+
+    break;
 
     case EXPR_STRUCTURE:
       formalize_structure_cons (expr);