Merge commit 'a39d4083cab589d7cd6a13e8a4b8db8875261d75'
[gnulib.git] / lib / wcsspn-impl.h
1 /* Advance in a wide string, skipping any of a set of wide characters.
2    Copyright (C) 1999, 2011-2014 Free Software Foundation, Inc.
3    Written by Bruno Haible <bruno@clisp.org>, 1999.
4
5    This program is free software: you can redistribute it and/or modify
6    it under the terms of the GNU General Public License as published by
7    the Free Software Foundation; either version 3 of the License, or
8    (at your option) any later version.
9
10    This program is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13    GNU General Public License for more details.
14
15    You should have received a copy of the GNU General Public License
16    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
17
18 size_t
19 wcsspn (const wchar_t *wcs, const wchar_t *accept)
20 {
21   /* Optimize two cases.  */
22   if (accept[0] == (wchar_t)'\0')
23     return 0;
24
25   if (accept[1] == (wchar_t)'\0')
26     {
27       wchar_t wc = accept[0];
28       const wchar_t *ptr = wcs;
29       for (; *ptr != (wchar_t)'\0'; ptr++)
30         {
31           if (*ptr != wc)
32             break;
33         }
34       return ptr - wcs;
35     }
36
37   /* General case.  */
38   {
39     const wchar_t *ptr = wcs;
40     for (; *ptr != (wchar_t)'\0'; ptr++)
41       {
42         if (!wcschr (accept, *ptr))
43           break;
44       }
45     return ptr - wcs;
46   }
47 }