proc_test: do not overflow preempt_counter on AVR.
[bertos.git] / bertos / algo / rle.c
1 /**
2  * \file
3  * <!--
4  * This file is part of BeRTOS.
5  *
6  * Bertos is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19  *
20  * As a special exception, you may use this file as part of a free software
21  * library without restriction.  Specifically, if other files instantiate
22  * templates or use macros or inline functions from this file, or you compile
23  * this file and link it with other files to produce an executable, this
24  * file does not by itself cause the resulting executable to be covered by
25  * the GNU General Public License.  This exception does not however
26  * invalidate any other reasons why the executable file might be covered by
27  * the GNU General Public License.
28  *
29  * Copyright 2004 Develer S.r.l. (http://www.develer.com/)
30  * Copyright 1999, 2000, 2001 Bernie Innocenti <bernie@codewiz.org>
31  *
32  * -->
33  *
34  * \brief General-purpose run-length {en,de}coding algorithm (implementation)
35  *
36  * Original source code from http://www.compuphase.com/compress.htm
37  *
38  * \version $Id$
39  * \author Bernie Innocenti <bernie@codewiz.org>
40  */
41
42 #include "rle.h"
43
44
45 /**
46  * Run-length encode \a len bytes from the \a input buffer
47  * to the \a output buffer.
48  */
49 int rle(unsigned char *output, const unsigned char *input, int len)
50 {
51         int count, index, i;
52         unsigned char first;
53         unsigned char *out;
54
55
56         out = output;
57         count = 0;
58         while (count < len)
59         {
60                 index = count;
61                 first = input[index++];
62
63                 /* Scan for bytes identical to the first one */
64                 while ((index < len) && (index - count < 127) && (input[index] == first))
65                         index++;
66
67                 if (index - count == 1)
68                 {
69                         /* Failed to "replicate" the current byte. See how many to copy.
70                          */
71                         while ((index < len) && (index - count < 127))
72                         {
73                                 /* Avoid a replicate run of only 2-bytes after a literal run.
74                                  * There is no gain in this, and there is a risc of loss if the
75                                  * run after the two identical bytes is another literal run.
76                                  * So search for 3 identical bytes.
77                                  */
78                                 if ((input[index] == input[index - 1]) &&
79                                         ((index > 1) && (input[index] == input[index - 2])))
80                                 {
81                                         /* Reset the index so we can back up these three identical
82                                          * bytes in the next run.
83                                          */
84                                         index -= 2;
85                                         break;
86                                 }
87
88                                 index++;
89                         }
90
91                         /* Output a run of uncompressed bytes: write length and values */
92                         *out++ = (unsigned char)(count - index);
93                         for (i = count; i < index; i++)
94                                 *out++ = input[i];
95             }
96                 else
97                 {
98                         /* Output a compressed run: write length and value */
99                         *out++ = (unsigned char)(index - count);
100                         *out++ = first;
101             }
102
103                 count = index;
104         }
105
106         /* Output EOF marker */
107         *out++ = 0;
108
109         return (out - output);
110 }
111
112
113 /**
114  * Run-length decode from the \a input buffer to the \a output
115  * buffer.
116  *
117  * \note The output buffer must be large enough to accomodate
118  *       all decoded output.
119  */
120 int unrle(unsigned char *output, const unsigned char *input)
121 {
122         signed char count;
123         unsigned char *out;
124         unsigned char value;
125
126
127         out = output;
128
129         for (;;)
130         {
131                 count = (signed char)*input++;
132                 if (count > 0)
133                 {
134                         /* replicate run */
135                         value = *input++;
136                         while (count--)
137                                 *out++ = value;
138                 }
139                 else if (count < 0)
140                 {
141                         /* literal run */
142                         while (count++)
143                                 *out++ = *input++;
144                 }
145                 else
146                         /* EOF */
147                         break;
148         }
149
150         return (out - output);
151 }