New modules 'unistr/u8-strspn', 'unistr/u16-strspn', 'unistr/u32-strspn'.
[gnulib.git] / lib / unistr / u-strspn.h
1 /* Search for some characters in UTF-8/UTF-16/UTF-32 string.
2    Copyright (C) 1999, 2002, 2006 Free Software Foundation, Inc.
3    Written by Bruno Haible <bruno@clisp.org>, 2002.
4
5    This program is free software; you can redistribute it and/or modify it
6    under the terms of the GNU Library General Public License as published
7    by the Free Software Foundation; either version 2, or (at your option)
8    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 GNU
13    Library General Public License for more details.
14
15    You should have received a copy of the GNU Library General Public
16    License along with this program; if not, write to the Free Software
17    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
18    USA.  */
19
20 size_t
21 FUNC (const UNIT *str, const UNIT *accept)
22 {
23   /* Optimize two cases.  */
24   if (accept[0] == 0)
25     return 0;
26   {
27     ucs4_t uc;
28     int count = U_STRMBTOUC (&uc, accept);
29     if (count >= 0 && accept[count] == 0)
30       {
31         const UNIT *ptr = str;
32         for (; *ptr != 0; ptr += count)
33           if (U_CMP (ptr, accept, count) != 0)
34             break;
35         return ptr - str;
36       }
37   }
38   /* General case.  */
39   {
40     const UNIT *ptr = str;
41
42     for (;;)
43       {
44         ucs4_t uc;
45         int count = U_STRMBTOUC (&uc, ptr);
46         if (count == 0)
47           return ptr - str;
48         if (count < 0)
49           break;
50         if (!U_STRCHR (accept, uc))
51           return ptr - str;
52         ptr += count;
53       }
54     return U_STRLEN (str);
55   }
56 }