--- deliantra/server/common/treasure.C	2007/04/19 19:24:25	1.52
+++ deliantra/server/common/treasure.C	2007/04/29 00:19:16	1.59
@@ -1,5 +1,5 @@
 /*
- * CrossFire, A Multiplayer game for X-windows
+ * CrossFire, A Multiplayer game
  * 
  * Copyright (C) 2005, 2006, 2007 Marc Lehmann & Crossfire+ Development Team
  * Copyright (C) 2002 Mark Wedel & Crossfire Development Team
@@ -199,8 +199,6 @@
    * Add up the chance total, and check to make sure the yes & no
    * fields of the treasures are not being used.
    */
-  tl->total_chance = 0;
-
   if (one)
     {
       for (treasure *t = tl->items; t; t = t->next)
@@ -232,13 +230,22 @@
 static void
 put_treasure (object *op, object *creator, int flags)
 {
-  /* Bit of a hack - spells should never be put onto the map.  The entire
-   * treasure stuff is a problem - there is no clear idea of knowing
-   * this is the original object, or if this is an object that should be created
-   * by another object.
-   */
-  if (flags & GT_ENVIRONMENT && op->type != SPELL)
+  if (flags & GT_ENVIRONMENT)
     {
+      /* Bit of a hack - spells should never be put onto the map.  The entire
+       * treasure stuff is a problem - there is no clear idea of knowing
+       * this is the original object, or if this is an object that should be created
+       * by another object.
+       */
+      //TODO: flag such as objects... as such (no drop, anybody?)
+      if (op->type == SPELL)
+        {
+          op->destroy ();
+          return;
+        }
+
+      op->expand_tail ();
+
       if (ob_blocked (op, creator->map, creator->x, creator->y))
         op->destroy ();
       else
@@ -366,6 +373,12 @@
     }
 }
 
+void
+object::create_treasure (treasurelist *tl, int flags)
+{
+  ::create_treasure (tl, this, flags, map ? map->difficulty : 0);
+}
+
 /* This calls the appropriate treasure creation function.  tries is passed
  * to determine how many list transitions or attempts to create treasure
  * have been made.  It is really in place to prevent infinite loops with
@@ -386,6 +399,15 @@
       return;
     }
 
+  if (op->flag [FLAG_TREASURE_ENV])
+    {
+      // do not generate items when there already is something above the object
+      if (op->flag [FLAG_IS_FLOOR] && op->above)
+        return;
+
+      flag |= GT_ENVIRONMENT;
+    }
+
   if (tl->total_chance)
     create_one_treasure (tl, op, flag, difficulty, tries);
   else