Minor clean-up.
[neverball] / ball / set.c
index 368bdf3..89c558e 100644 (file)
@@ -1,4 +1,4 @@
-/*   
+/*
  * Copyright (C) 2003 Robert Kooima
  *
  * NEVERBALL is  free software; you can redistribute  it and/or modify
 
 /*---------------------------------------------------------------------------*/
 
-static int count;                    /* number of sets */
+static int set;
+static int count;
 
-static struct set set_v[MAXSET];     /* array of sets */
-
-static struct set *current_set;      /* currently selected set */
-
-static struct level level_v[MAXLVL]; /* levels of the current set  */
+static struct set set_v[MAXSET];
+static struct level level_v[MAXLVL];
 
 /*---------------------------------------------------------------------------*/
 
 static void put_score(FILE *fp, const struct score *s)
 {
     int j;
+
     for (j = 0; j < NSCORE; j++)
        fprintf(fp, "%d %d %s\n", s->timer[j], s->coins[j], s->player[j]);
 }
 
+/* Store the score of the set. */
 static void set_store_hs(void)
-/* Store the score of the set */
 {
-    const struct set *s = current_set;
+    const struct set *s = &set_v[set];
     FILE *fout;
     int i;
     const struct level *l;
-    char states[MAXLVL+1];
-    
+    char states[MAXLVL + 1];
+
     if ((fout = fopen(config_user(s->user_scores), "w")))
     {
-       for (i = 0 ; i < s->count ; i++)
-       {
-           if (level_v[i].is_locked)
-               states[i] = 'L';
-           else if (level_v[i].is_completed)
-               states[i] = 'C';
-           else
-               states[i] = 'O';
-       }
-       states[s->count] = '\0';
-       fprintf(fout, "%s\n",states);
-       
-       put_score(fout, &s->time_score);
-       put_score(fout, &s->coin_score);
-
-       for (i = 0 ; i < s->count ; i++)
-       {
-           l = &level_v[i];
-           put_score(fout, &l->time_score);
-           put_score(fout, &l->goal_score);
-           put_score(fout, &l->coin_score);
-       }
-
-       fclose(fout);
+        for (i = 0; i < s->count; i++)
+        {
+            if (level_v[i].is_locked)
+                states[i] = 'L';
+            else if (level_v[i].is_completed)
+                states[i] = 'C';
+            else
+                states[i] = 'O';
+        }
+        states[s->count] = '\0';
+        fprintf(fout, "%s\n",states);
+
+        put_score(fout, &s->time_score);
+        put_score(fout, &s->coin_score);
+
+        for (i = 0; i < s->count; i++)
+        {
+            l = &level_v[i];
+
+            put_score(fout, &l->score.best_times);
+            put_score(fout, &l->score.unlock_goal);
+            put_score(fout, &l->score.most_coins);
+        }
+
+        fclose(fout);
     }
 }
 
@@ -84,208 +84,194 @@ static int get_score(FILE *fp, struct score *s)
 {
     int j;
     int res = 1;
+
     for (j = 0; j < NSCORE && res; j++)
     {
-       res = (fscanf(fp, "%d %d %s\n",
-              &s->timer[j], &s->coins[j], s->player[j])) == 3;
+        res = fscanf(fp, "%d %d %s\n",
+                     &s->timer[j],
+                     &s->coins[j],
+                     s->player[j]) == 3;
     }
     return res;
 }
 
+/* Get the score of the set. */
 static void set_load_hs(void)
-/* Get the score of the set */
 {
-    struct set *s = current_set;
+    struct set *s = &set_v[set];
     FILE *fin;
     int i;
     int res = 0;
     struct level *l;
     const char *fn = config_user(s->user_scores);
-    char states[MAXLVL+1];
+    char states[MAXLVL + 1];
 
     if ((fin = fopen(fn, "r")))
     {
-        res = ((fscanf(fin, "%s\n", states) == 1) && 
-               (strlen(states) == s->count));
-       for (i = 0 ; i < s->count && res ; i++)
-       {
-           if (states[i] == 'L')
-           {
-               level_v[i].is_locked = 1;
-               level_v[i].is_completed = 0;
-           }
-           else if (states[i] == 'C')
-           {
-               level_v[i].is_locked = 0;
-               level_v[i].is_completed = 1;
-           } 
-           else if (states[i] == 'O')
-           {
-               level_v[i].is_locked = 0;
-               level_v[i].is_completed = 0;
-           }
-           else
-               res = 0;
-       }
-       
+        res = fscanf(fin, "%s\n", states) == 1 && strlen(states) == s->count;
+
+        for (i = 0; i < s->count && res; i++)
+        {
+            switch (states[i])
+            {
+            case 'L':
+                level_v[i].is_locked = 1;
+                level_v[i].is_completed = 0;
+                break;
+
+            case 'C':
+                level_v[i].is_locked = 0;
+                level_v[i].is_completed = 1;
+                break;
+
+            case 'O':
+                level_v[i].is_locked = 0;
+                level_v[i].is_completed = 0;
+                break;
+
+            default:
+                res = 0;
+            }
+        }
+
         res = res &&
-           get_score(fin, &s->time_score) &&
-           get_score(fin, &s->coin_score);
-       
-       for (i = 0; i < s->count && res; i++)
-       {
-           l = &level_v[i];
-           res = get_score(fin, &l->time_score) &&
-                 get_score(fin, &l->goal_score) &&
-                 get_score(fin, &l->coin_score);
-       }
-
-       fclose(fin);
+            get_score(fin, &s->time_score) &&
+            get_score(fin, &s->coin_score);
+
+        for (i = 0; i < s->count && res; i++)
+        {
+            l = &level_v[i];
+            res = get_score(fin, &l->score.best_times) &&
+                  get_score(fin, &l->score.unlock_goal) &&
+                  get_score(fin, &l->score.most_coins);
+        }
+
+        fclose(fin);
     }
-    
+
     if (!res && errno != ENOENT)
     {
-       fprintf(stderr, _("Error while loading user high-score file '%s': "), fn);
-       if (errno)
-           perror(NULL);
-       else
-           fprintf(stderr, _("Incorrect format\n"));
+        fprintf(stderr,
+                _("Error while loading user high-score file '%s': %s\n"),
+                fn, errno ? strerror(errno) : _("Incorrect format"));
     }
 }
 
-static char* chomp(char *str)
-/* Remove trailing \n if any */
+static char *strip_eol(char *str)
 {
-    char *p = str + strlen(str) - 1;
-    if (p >= str && *p == '\n') *p = 0;
+    char *c = str + strlen(str) - 1;
+
+    while (c >= str && (*c == '\n' || *c =='\r'))
+        *c-- = '\0';
+
     return str;
 }
 
 static int set_load(struct set *s, const char *filename)
-/* Count levels */
 {
     FILE *fin;
     char buf[MAXSTR];
-    int res = 0;
-    
-    /* Open the datafile */
+    int res;
+
+    fin = fopen(config_data(filename), "r");
 
-    fin = fopen(filename, "r");
-    if (fin == NULL)
+    if (!fin)
     {
-       fprintf(stderr, _("Cannot load the set file '%s':"), filename);
-       perror(NULL);
-       return 0;
+        fprintf(stderr, _("Cannot load the set file '%s': %s\n"),
+                filename, strerror(errno));
+        return 0;
     }
 
-    /* Raz the set structure */
-
-    memset(s, 0, sizeof(struct set));
+    memset(s, 0, sizeof (struct set));
 
     /* Set some sane values in case the scores hs is missing. */
-    
+
     score_init_hs(&s->time_score, 359999, 0);
     score_init_hs(&s->coin_score, 359999, 0);
-    
-    /* Load set metadata */
-    
+
+    /* Load set metadata. */
+
     strcpy(s->file, filename);
+
     if ((res = fgets(buf, MAXSTR, fin) != NULL))
-       strcpy(s->name, chomp(buf));
+        strcpy(s->name, strip_eol(buf));
     if (res && (res = fgets(buf, MAXSTR, fin) != NULL))
-       strcpy(s->desc, chomp(buf));
+        strcpy(s->desc, strip_eol(buf));
     if (res && (res = fgets(buf, MAXSTR, fin) != NULL))
-       strcpy(s->setname, chomp(buf));
+        strcpy(s->id, strip_eol(buf));
     if (res && (res = fgets(buf, MAXSTR, fin) != NULL))
-       strcpy(s->shot, chomp(buf));
+        strcpy(s->shot, strip_eol(buf));
     if (res && (res = fgets(buf, MAXSTR, fin) != NULL))
-       sscanf(buf, "%d %d %d %d %d %d", 
-               &s->time_score.timer[0],
-               &s->time_score.timer[1],
-               &s->time_score.timer[2],
-               &s->coin_score.coins[0],
-               &s->coin_score.coins[1],
-               &s->coin_score.coins[2]);
+        sscanf(buf, "%d %d %d %d %d %d",
+                &s->time_score.timer[0],
+                &s->time_score.timer[1],
+                &s->time_score.timer[2],
+                &s->coin_score.coins[0],
+                &s->coin_score.coins[1],
+                &s->coin_score.coins[2]);
+
     strcpy(s->user_scores, "neverballhs-");
-    strcat(s->user_scores, s->setname);
+    strcat(s->user_scores, s->id);
+
+    /* Count levels. */
 
-    /* Count levels levels. */
-    
     s->count = 0;
 
     while (s->count < MAXLVL && (fscanf(fin, "%s", buf) == 1))
-       s->count++;
-    
-    /* Close the file, since it's no more needed */
-    
+        s->count++;
+
     fclose(fin);
 
-    /* Load the levels states (stored in the user highscore file) */
+    /* Load the levels states (stored in the user high score file) */
+
     s->locked = s->count;
     s->completed = 0;
+
     if ((fin = fopen(config_user(s->user_scores), "r")))
     {
-       char states[MAXLVL + 1];
-       int i;
-       if ((fscanf(fin, "%s\n", states) == 1) && (strlen(states) == s->count))
-       {
-           for (i=0; i<s->count; i++)
-           {
-               if (states[i] == 'O')
-                   s->locked -= 1;
-               else if (states[i] == 'C')
-               {
-                   s->completed += 1;
-                   s->locked -= 1;
-               }
-           }
-       }
-       fclose(fin);
+        char states[MAXLVL + 1];
+        int i;
+        if ((fscanf(fin, "%s\n", states) == 1) && (strlen(states) == s->count))
+        {
+            for (i = 0; i < s->count; i++)
+            {
+                if (states[i] == 'O')
+                    s->locked -= 1;
+                else if (states[i] == 'C')
+                {
+                    s->completed += 1;
+                    s->locked -= 1;
+                }
+            }
+        }
+        fclose(fin);
     }
     if (s->locked == s->count)
-       s->locked = s->count-1;
+        s->locked = s->count-1;
 
     return 1;
 }
 
 /*---------------------------------------------------------------------------*/
 
-void set_init()
+int set_init()
 {
     FILE *fin;
-    struct set * set;
-    char filename[MAXSTR];
-    int res;
+    char  name[MAXSTR];
 
-    current_set = NULL;
-    
+    set   = 0;
     count = 0;
 
     if ((fin = fopen(config_data(SET_FILE), "r")))
     {
-       res = 1;
-       while (count < MAXSET && res)
-       {
-           set = &(set_v[count]);
-
-           /* clean the set data */
-           
-           res = (fgets(filename, MAXSTR, fin) != NULL);
-           if (res)
-           {
-               chomp(filename);
-
-               res = set_load(set, config_data(filename));
-               if (res)
-               {
-                   set->number = count;
-                   count++;
-               }
-           }
-        }
+        while (count < MAXSET && fgets(name, MAXSTR, fin))
+            if (set_load(&set_v[count], strip_eol(name)))
+                count++;
 
         fclose(fin);
     }
+
+    return count;
 }
 
 /*---------------------------------------------------------------------------*/
@@ -302,20 +288,17 @@ const struct set *get_set(int i)
 
 /*---------------------------------------------------------------------------*/
 
-int  set_unlocked(const struct set *s)
-/* Are all levels (even extra bonus) unlocked? */
+int set_unlocked(const struct set *s)
 {
     return s->locked == 0;
 }
 
-int  set_completed(const struct set *s)
-/* Are all levels (even extra bonus) completed? */
+int set_completed(const struct set *s)
 {
     return s->completed == s->count;
 }
 
-int  set_level_exists(const struct set *s, int i)
-/* Is the level i of the set exists */
+int set_level_exists(const struct set *s, int i)
 {
     return (i >= 0) && (i < s->count);
 }
@@ -323,284 +306,241 @@ int  set_level_exists(const struct set *s, int i)
 /*---------------------------------------------------------------------------*/
 
 static void set_load_levels(void)
-/* Load more the levels of the current set */
 {
     FILE *fin;
+    struct level *l;
+
     char buf[MAXSTR];
     char name[MAXSTR];
-    struct level * l;
-
-    int i=0, res;
-    int nb=1, bnb=1;
-    
-    fin = fopen(current_set->file, "r");
-    assert(fin != NULL);
-    
-    res = 1;
-    /* Skip the five first lines */
-    for(i=0; i<5; i++)
-       fgets(buf, MAXSTR, fin);
-    for(i=0; i<current_set->count && res; i++)
-    {
-       l = &level_v[i];
-       res = (fscanf(fin, "%s", name) == 1);
-       assert(res);
-
-       level_load(config_data(name), l);
-
-       /* Initialize set related info */
-       l->set        = current_set;
-       l->number     = i;
-       if (l->is_bonus)
-           sprintf(l->numbername, _("B%d"), bnb++);
-       else
-           sprintf(l->numbername, "%02d", nb++);
-       l->is_locked    = 1;
-       l->is_completed = 0;
-    }
-    level_v[0].is_locked = 0; /* unlock the first level */
-    fclose(fin);
-    assert(i == current_set->count);
-}
 
-void set_goto(int i)
-{
-    assert(set_exists(i));
-    current_set = &set_v[i];
-    set_load_levels();
-    set_load_hs();
-}
+    int i = 0, res;
+    int nb = 1, bnb = 1;
 
-const struct set *curr_set(void)
-{
-    return current_set;
-}
+    const char *roman[] = {
+        "",
+        "I",   "II",   "III",   "IV",   "V",
+        "VI",  "VII",  "VIII",  "IX",   "X",
+        "XI",  "XII",  "XIII",  "XIV",  "XV",
+        "XVI", "XVII", "XVIII", "XIX",  "XX",
+        "XXI", "XXII", "XXIII", "XXIV", "XXV"
+    };
 
-const struct level *get_level(int i)
-{
-    return (i>=0 && i<current_set->count) ? &level_v[i] : NULL;
-}
+    if ((fin = fopen(config_data(set_v[set].file), "r")))
+    {
+        res = 1;
 
-/*---------------------------------------------------------------------------*/
+        /* Skip the five first lines */
+        for (i = 0; i < 5; i++)
+            fgets(buf, MAXSTR, fin);
 
-static int score_time_comp(const struct score *S, int i, int j)
-{
-    if (S->timer[i] <  S->timer[j])
-        return 1;
+        for (i = 0; i < set_v[set].count && res; i++)
+        {
+            l = &level_v[i];
 
-    if (S->timer[i] == S->timer[j] &&
-        S->coins[i] >  S->coins[j])
-        return 1;
+            res = (fscanf(fin, "%s", name) == 1);
+            assert(res);
 
-    return 0;
-}
+            level_load(name, l);
 
-static int score_coin_comp(const struct score *S, int i, int j)
-{
-    if (S->coins[i] >  S->coins[j])
-        return 1;
+            /* Initialize set related info */
+            l->set    = &set_v[set];
+            l->number = i;
+
+            if (l->is_bonus)
+                sprintf(l->repr, "%s", roman[bnb++]);
+            else
+                sprintf(l->repr, "%02d", nb++);
 
-    if (S->coins[i] == S->coins[j] &&
-        S->timer[i] <  S->timer[j])
-        return 1;
+            l->is_locked    = 1;
+            l->is_completed = 0;
+        }
+        level_v[0].is_locked = 0; /* unlock the first level */
+        fclose(fin);
+    }
 
-    return 0;
+    assert(i == set_v[set].count);
 }
 
-static void score_swap(struct score *S, int i, int j)
+void set_goto(int i)
 {
-    char player[MAXNAM];
-    int  tmp;
+    set = i;
 
-    strncpy(player,       S->player[i], MAXNAM);
-    strncpy(S->player[i], S->player[j], MAXNAM);
-    strncpy(S->player[j], player,       MAXNAM);
-
-    tmp         = S->timer[i];
-    S->timer[i] = S->timer[j];
-    S->timer[j] = tmp;
-
-    tmp         = S->coins[i];
-    S->coins[i] = S->coins[j];
-    S->coins[j] = tmp;
+    set_load_levels();
+    set_load_hs();
 }
 
-static int score_time_insert(struct score *s, const char* player, int timer, int coins)
+const struct set *curr_set(void)
 {
-    int i;
-    
-    strncpy(s->player[3], player, MAXNAM);
-    s->timer[3] = timer;
-    s->coins[3] = coins;
-
-    for (i = 2; i >= 0 && score_time_comp(s, i + 1, i); i--)
-        score_swap(s, i + 1, i);
-    return i+1;
+    return &set_v[set];
 }
 
-static int score_coin_insert(struct score *s, const char* player, int timer, int coins)
+const struct level *get_level(int i)
 {
-    int i;
-    
-    strncpy(s->player[3], player, MAXNAM);
-    s->timer[3] = timer;
-    s->coins[3] = coins;
-
-    for (i = 2; i >= 0 && score_coin_comp(s, i + 1, i); i--)
-        score_swap(s, i + 1, i);
-    return i+1;
+    return (i >= 0 && i < set_v[set].count) ? &level_v[i] : NULL;
 }
 
+/*---------------------------------------------------------------------------*/
+
+/* Update the level score rank according to coins and timer. */
 static int level_score_update(struct level_game *lg, const char *player)
-/* Update the level score rank according to coins and timer */
 {
     int timer = lg->timer;
     int coins = lg->coins;
-    struct level * l = &level_v[lg->level->number];
+    struct level *l = &level_v[lg->level->number];
+
+    lg->time_rank = score_time_insert(&l->score.best_times,
+                                      player, timer, coins);
 
-    lg->time_rank = score_time_insert(&l->time_score, player, timer, coins);
-       
     if (lg->mode == MODE_CHALLENGE || lg->mode == MODE_NORMAL)
-       lg->goal_rank = score_time_insert(&l->goal_score, player, timer, coins);
+        lg->goal_rank = score_time_insert(&l->score.unlock_goal,
+                                          player, timer, coins);
     else
-       lg->goal_rank = 3;
+        lg->goal_rank = 3;
 
-    lg->coin_rank = score_coin_insert(&l->coin_score, player, timer, coins);
+    lg->coin_rank = score_coin_insert(&l->score.most_coins,
+                                      player, timer, coins);
 
     return (lg->time_rank < 3 || lg->goal_rank < 3 || lg->coin_rank < 3);
 }
 
+/* Update the set score rank according to score and times. */
 static int set_score_update(struct level_game *lg, const char *player)
-/* Update the set score rank according to score and times */
 {
     int timer = lg->times;
     int coins = lg->score;
-    struct set * s = current_set;
+    struct set *s = &set_v[set];
 
     lg->score_rank = score_time_insert(&s->time_score, player, timer, coins);
     lg->times_rank = score_time_insert(&s->coin_score, player, timer, coins);
+
     return (lg->score_rank < 3 || lg->times_rank < 3);
 }
 
-
+/* Update the player name for set and level high-score. */
 void score_change_name(struct level_game *lg, const char *player)
-/* Update the player name for set and level high-score */
 {
-#define UPDATE(i, x) (strncpy((x).player[(i)], player, MAXNAM))
-    struct set * s = current_set;
+    struct set   *s = &set_v[set];
     struct level *l = &level_v[lg->level->number];
-    UPDATE(lg->time_rank, l->time_score);
-    UPDATE(lg->goal_rank, l->goal_score);
-    UPDATE(lg->coin_rank, l->coin_score);
-    UPDATE(lg->score_rank, s->coin_score);
-    UPDATE(lg->times_rank, s->time_score);
+
+    strncpy(l->score.best_times.player [lg->time_rank], player, MAXNAM);
+    strncpy(l->score.unlock_goal.player[lg->goal_rank], player, MAXNAM);
+    strncpy(l->score.most_coins.player [lg->coin_rank], player, MAXNAM);
+
+    strncpy(s->coin_score.player[lg->score_rank], player, MAXNAM);
+    strncpy(s->time_score.player[lg->times_rank], player, MAXNAM);
+
     set_store_hs();
 }
 
 static struct level *next_level(int i)
 {
-/* Return the ith level, or NULL */
-    return set_level_exists(current_set, i+1) ? &level_v[i+1] : NULL;
+    return set_level_exists(&set_v[set], i + 1) ? &level_v[i + 1] : NULL;
 }
 
 static struct level *next_normal_level(int i)
-/* Return the next notmal level (starting for i) 
- * Return NULL if there is not a such level */
 {
-    for (i++; i < current_set->count; i++)
-       if (!level_v[i].is_bonus)
-           return &level_v[i];
+    for (i++; i < set_v[set].count; i++)
+        if (!level_v[i].is_bonus)
+            return &level_v[i];
+
     return NULL;
 }
 
+/*---------------------------------------------------------------------------*/
+
 void set_finish_level(struct level_game *lg, const char *player)
-/* Inform the set that a level is finished. 
- * Update next_level and score rank fields */
 {
-    struct set *s = current_set;
-    int ln = lg->level->number;     /* curent level number */
-    struct level *cl = &level_v[ln]; /* current level */
-    struct level *nl = NULL;        /* next level*/
-    int dirty = 0;                  /* HS should be saved? */
+    struct set *s = &set_v[set];
+    int ln = lg->level->number;      /* Current level number       */
+    struct level *cl = &level_v[ln]; /* Current level              */
+    struct level *nl = NULL;         /* Next level                 */
+    int dirty = 0;                   /* Should the score be saved? */
 
     assert(s == cl->set);
 
-    /* On success */
-    if (lg->state == GAME_GOAL)
-    {      
-       /* Update scores */
-       dirty = level_score_update(lg, player);
-       dirty = set_score_update(lg, player) || dirty;
-       
-       /* if no set, no next level */    
-       if (s == NULL)
-       {
-           /* if no set, return */
-           lg->next_level = NULL;
-           return;
-       }
-
-       /* Complete the level */
-       if (lg->mode == MODE_CHALLENGE || lg->mode == MODE_NORMAL)
-       {
-           /* Complete the level */
-           if (!cl->is_completed)
-           {
-               cl->is_completed = 1;
-               s->completed += 1;
-               dirty = 1;
-           }
-       }
-       
-       /* Identify the follwing level */
-       nl = next_level(ln + lg->state_value);
-       if (nl != NULL)
-       {
-           /* skip bonuses if unlocked in non challenge mode*/
-           if(nl->is_bonus && nl->is_locked && lg->mode != MODE_CHALLENGE)
-               nl = next_normal_level(nl->number);
-       }
-       else if (lg->mode == MODE_CHALLENGE)
-           lg->win = 1;
+    /* if no set, no next level */
+    if (s == NULL)
+    {
+        /* if no set, return */
+        lg->next_level = NULL;
+        return;
+    }
+
+    /* On level completed */
+    if (lg->status == GAME_GOAL)
+    {
+        /* Update level scores */
+        dirty = level_score_update(lg, player);
+
+        /* Complete the level */
+        if (lg->mode == MODE_CHALLENGE || lg->mode == MODE_NORMAL)
+        {
+            /* Complete the level */
+            if (!cl->is_completed)
+            {
+                cl->is_completed = 1;
+                s->completed += 1;
+                dirty = 1;
+            }
+        }
+    }
+
+    /* On goal reached */
+    if (lg->status == GAME_GOAL)
+    {
+        /* Identify the following level */
+        nl = next_level(ln);
+        if (nl != NULL)
+        {
+            /* skip bonuses if unlocked in non challenge mode */
+            if (nl->is_bonus && nl->is_locked && lg->mode != MODE_CHALLENGE)
+                nl = next_normal_level(nl->number);
+        }
+        else if (lg->mode == MODE_CHALLENGE)
+            lg->win = 1;
     }
     else if (cl->is_bonus || lg->mode != MODE_CHALLENGE)
     {
-       /* On fail, identify the next level (only in bonus for challenge) */
-       nl = next_normal_level(ln);
-       /* Next level may be unavailable */
-       if (!cl->is_bonus && nl != NULL && nl->is_locked)
-           nl = NULL;
-       /* Fail a bonus level but win the set! */
-       else if (nl == NULL && lg->mode == MODE_CHALLENGE)
-           lg->win = 1;
+        /* On fail, identify the next level (only in bonus for challenge) */
+        nl = next_normal_level(ln);
+        /* Next level may be unavailable */
+        if (!cl->is_bonus && nl != NULL && nl->is_locked)
+            nl = NULL;
+        /* Fail a bonus level but win the set! */
+        else if (nl == NULL && lg->mode == MODE_CHALLENGE)
+            lg->win = 1;
     }
 
-    /* Win ! => unlock all levels */
+    /* Win ! */
     if (lg->win)
     {
-       set_cheat();
-       dirty = 1;
+        /* update set score */
+        set_score_update(lg, player);
+        /* unlock all levels */
+        set_cheat();
+        dirty = 1;
     }
-   
+
     /* unlock the next level if needed */
-    if(nl != NULL && nl->is_locked)
+    if (nl != NULL && nl->is_locked)
     {
-       if (lg->mode == MODE_CHALLENGE || lg->mode == MODE_NORMAL)
-       {
-           nl->is_locked = 0;
-           s->locked -= 1;
-           dirty = 1;
-       }
-       else
-           nl = NULL;
+        if (lg->mode == MODE_CHALLENGE || lg->mode == MODE_NORMAL)
+        {
+            lg->unlock = 1;
+            nl->is_locked = 0;
+            s->locked -= 1;
+            dirty = 1;
+        }
+        else
+            nl = NULL;
     }
-    
-    /* got the next level */ 
+
+    /* got the next level */
     lg->next_level = nl;
 
     /* Update file */
     if (dirty)
-       set_store_hs();
+        set_store_hs();
 }
 
 /*---------------------------------------------------------------------------*/
@@ -608,17 +548,26 @@ void set_finish_level(struct level_game *lg, const char *player)
 void level_snap(int i)
 {
     char filename[MAXSTR];
+    char *ext;
 
-    /* Convert the level name to a BMP filename. */
+    /* Convert the level name to a PNG filename. */
 
     memset(filename, 0, MAXSTR);
-    strncpy(filename, level_v[i].file, strcspn(level_v[i].file, "."));
-    strcat(filename, ".bmp");
+
+    ext = strrchr(level_v[i].file, '.');
+    strncpy(filename, level_v[i].file,
+            ext ? ext - level_v[i].file : strlen(level_v[i].file));
+    strcat(filename, ".png");
 
     /* Initialize the game for a snapshot. */
 
     if (game_init(&level_v[i], 0, 0))
     {
+        int shadow;
+
+        if ((shadow = config_get_d(CONFIG_SHADOW)))
+            config_set_d(CONFIG_SHADOW, 0);
+
         /* Render the level and grab the screen. */
 
         config_clear();
@@ -627,7 +576,10 @@ void level_snap(int i)
         game_draw(1, 0);
         SDL_GL_SwapBuffers();
 
-        image_snap(filename, config_get_d(CONFIG_WIDTH), config_get_d(CONFIG_HEIGHT));
+        image_snap(filename);
+
+        if (shadow)
+            config_set_d(CONFIG_SHADOW, 1);
     }
 }
 
@@ -635,11 +587,12 @@ void set_cheat(void)
 /* Open each level of the current set */
 {
     int i;
-    current_set->locked = 0;
-    for (i=0; i < current_set->count; i++)
-       level_v[i].is_locked = 0;
+
+    set_v[set].locked = 0;
+
+    for (i = 0; i < set_v[set].count; i++)
+        level_v[i].is_locked = 0;
 }
 
 
 /*---------------------------------------------------------------------------*/
-