xref: /vim-8.2.3635/src/spellfile.c (revision b85d3627)
1edf3f97aSBram Moolenaar /* vi:set ts=8 sts=4 sw=4 noet:
29ccfebddSBram Moolenaar  *
39ccfebddSBram Moolenaar  * VIM - Vi IMproved	by Bram Moolenaar
49ccfebddSBram Moolenaar  *
59ccfebddSBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
69ccfebddSBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
79ccfebddSBram Moolenaar  * See README.txt for an overview of the Vim source code.
89ccfebddSBram Moolenaar  */
99ccfebddSBram Moolenaar 
109ccfebddSBram Moolenaar /*
119ccfebddSBram Moolenaar  * spellfile.c: code for reading and writing spell files.
129ccfebddSBram Moolenaar  *
139ccfebddSBram Moolenaar  * See spell.c for information about spell checking.
149ccfebddSBram Moolenaar  */
159ccfebddSBram Moolenaar 
169ccfebddSBram Moolenaar /*
179ccfebddSBram Moolenaar  * Vim spell file format: <HEADER>
189ccfebddSBram Moolenaar  *			  <SECTIONS>
199ccfebddSBram Moolenaar  *			  <LWORDTREE>
209ccfebddSBram Moolenaar  *			  <KWORDTREE>
219ccfebddSBram Moolenaar  *			  <PREFIXTREE>
229ccfebddSBram Moolenaar  *
239ccfebddSBram Moolenaar  * <HEADER>: <fileID> <versionnr>
249ccfebddSBram Moolenaar  *
259ccfebddSBram Moolenaar  * <fileID>     8 bytes    "VIMspell"
269ccfebddSBram Moolenaar  * <versionnr>  1 byte	    VIMSPELLVERSION
279ccfebddSBram Moolenaar  *
289ccfebddSBram Moolenaar  *
299ccfebddSBram Moolenaar  * Sections make it possible to add information to the .spl file without
309ccfebddSBram Moolenaar  * making it incompatible with previous versions.  There are two kinds of
319ccfebddSBram Moolenaar  * sections:
329ccfebddSBram Moolenaar  * 1. Not essential for correct spell checking.  E.g. for making suggestions.
339ccfebddSBram Moolenaar  *    These are skipped when not supported.
349ccfebddSBram Moolenaar  * 2. Optional information, but essential for spell checking when present.
359ccfebddSBram Moolenaar  *    E.g. conditions for affixes.  When this section is present but not
369ccfebddSBram Moolenaar  *    supported an error message is given.
379ccfebddSBram Moolenaar  *
389ccfebddSBram Moolenaar  * <SECTIONS>: <section> ... <sectionend>
399ccfebddSBram Moolenaar  *
409ccfebddSBram Moolenaar  * <section>: <sectionID> <sectionflags> <sectionlen> (section contents)
419ccfebddSBram Moolenaar  *
429ccfebddSBram Moolenaar  * <sectionID>	  1 byte    number from 0 to 254 identifying the section
439ccfebddSBram Moolenaar  *
449ccfebddSBram Moolenaar  * <sectionflags> 1 byte    SNF_REQUIRED: this section is required for correct
459ccfebddSBram Moolenaar  *					    spell checking
469ccfebddSBram Moolenaar  *
479ccfebddSBram Moolenaar  * <sectionlen>   4 bytes   length of section contents, MSB first
489ccfebddSBram Moolenaar  *
499ccfebddSBram Moolenaar  * <sectionend>	  1 byte    SN_END
509ccfebddSBram Moolenaar  *
519ccfebddSBram Moolenaar  *
529ccfebddSBram Moolenaar  * sectionID == SN_INFO: <infotext>
539ccfebddSBram Moolenaar  * <infotext>	 N bytes    free format text with spell file info (version,
549ccfebddSBram Moolenaar  *			    website, etc)
559ccfebddSBram Moolenaar  *
569ccfebddSBram Moolenaar  * sectionID == SN_REGION: <regionname> ...
572993ac5fSBram Moolenaar  * <regionname>	 2 bytes    Up to MAXREGIONS region names: ca, au, etc.  Lower
582993ac5fSBram Moolenaar  *			    case.  First <regionname> is region 1.
599ccfebddSBram Moolenaar  *
609ccfebddSBram Moolenaar  * sectionID == SN_CHARFLAGS: <charflagslen> <charflags>
619ccfebddSBram Moolenaar  *				<folcharslen> <folchars>
629ccfebddSBram Moolenaar  * <charflagslen> 1 byte    Number of bytes in <charflags> (should be 128).
639ccfebddSBram Moolenaar  * <charflags>  N bytes     List of flags (first one is for character 128):
649ccfebddSBram Moolenaar  *			    0x01  word character	CF_WORD
659ccfebddSBram Moolenaar  *			    0x02  upper-case character	CF_UPPER
669ccfebddSBram Moolenaar  * <folcharslen>  2 bytes   Number of bytes in <folchars>.
679ccfebddSBram Moolenaar  * <folchars>     N bytes   Folded characters, first one is for character 128.
689ccfebddSBram Moolenaar  *
699ccfebddSBram Moolenaar  * sectionID == SN_MIDWORD: <midword>
709ccfebddSBram Moolenaar  * <midword>     N bytes    Characters that are word characters only when used
719ccfebddSBram Moolenaar  *			    in the middle of a word.
729ccfebddSBram Moolenaar  *
739ccfebddSBram Moolenaar  * sectionID == SN_PREFCOND: <prefcondcnt> <prefcond> ...
749ccfebddSBram Moolenaar  * <prefcondcnt> 2 bytes    Number of <prefcond> items following.
759ccfebddSBram Moolenaar  * <prefcond> : <condlen> <condstr>
769ccfebddSBram Moolenaar  * <condlen>	1 byte	    Length of <condstr>.
779ccfebddSBram Moolenaar  * <condstr>	N bytes	    Condition for the prefix.
789ccfebddSBram Moolenaar  *
799ccfebddSBram Moolenaar  * sectionID == SN_REP: <repcount> <rep> ...
809ccfebddSBram Moolenaar  * <repcount>	 2 bytes    number of <rep> items, MSB first.
819ccfebddSBram Moolenaar  * <rep> : <repfromlen> <repfrom> <reptolen> <repto>
829ccfebddSBram Moolenaar  * <repfromlen>	 1 byte	    length of <repfrom>
839ccfebddSBram Moolenaar  * <repfrom>	 N bytes    "from" part of replacement
849ccfebddSBram Moolenaar  * <reptolen>	 1 byte	    length of <repto>
859ccfebddSBram Moolenaar  * <repto>	 N bytes    "to" part of replacement
869ccfebddSBram Moolenaar  *
879ccfebddSBram Moolenaar  * sectionID == SN_REPSAL: <repcount> <rep> ...
889ccfebddSBram Moolenaar  *   just like SN_REP but for soundfolded words
899ccfebddSBram Moolenaar  *
909ccfebddSBram Moolenaar  * sectionID == SN_SAL: <salflags> <salcount> <sal> ...
919ccfebddSBram Moolenaar  * <salflags>	 1 byte	    flags for soundsalike conversion:
929ccfebddSBram Moolenaar  *			    SAL_F0LLOWUP
939ccfebddSBram Moolenaar  *			    SAL_COLLAPSE
949ccfebddSBram Moolenaar  *			    SAL_REM_ACCENTS
959ccfebddSBram Moolenaar  * <salcount>    2 bytes    number of <sal> items following
969ccfebddSBram Moolenaar  * <sal> : <salfromlen> <salfrom> <saltolen> <salto>
979ccfebddSBram Moolenaar  * <salfromlen>	 1 byte	    length of <salfrom>
989ccfebddSBram Moolenaar  * <salfrom>	 N bytes    "from" part of soundsalike
999ccfebddSBram Moolenaar  * <saltolen>	 1 byte	    length of <salto>
1009ccfebddSBram Moolenaar  * <salto>	 N bytes    "to" part of soundsalike
1019ccfebddSBram Moolenaar  *
1029ccfebddSBram Moolenaar  * sectionID == SN_SOFO: <sofofromlen> <sofofrom> <sofotolen> <sofoto>
1039ccfebddSBram Moolenaar  * <sofofromlen> 2 bytes    length of <sofofrom>
1049ccfebddSBram Moolenaar  * <sofofrom>	 N bytes    "from" part of soundfold
1059ccfebddSBram Moolenaar  * <sofotolen>	 2 bytes    length of <sofoto>
1069ccfebddSBram Moolenaar  * <sofoto>	 N bytes    "to" part of soundfold
1079ccfebddSBram Moolenaar  *
1089ccfebddSBram Moolenaar  * sectionID == SN_SUGFILE: <timestamp>
1099ccfebddSBram Moolenaar  * <timestamp>   8 bytes    time in seconds that must match with .sug file
1109ccfebddSBram Moolenaar  *
1119ccfebddSBram Moolenaar  * sectionID == SN_NOSPLITSUGS: nothing
1129ccfebddSBram Moolenaar 	 *
1139ccfebddSBram Moolenaar  * sectionID == SN_NOCOMPOUNDSUGS: nothing
1149ccfebddSBram Moolenaar  *
1159ccfebddSBram Moolenaar  * sectionID == SN_WORDS: <word> ...
1169ccfebddSBram Moolenaar  * <word>	 N bytes    NUL terminated common word
1179ccfebddSBram Moolenaar  *
1189ccfebddSBram Moolenaar  * sectionID == SN_MAP: <mapstr>
1199ccfebddSBram Moolenaar  * <mapstr>	 N bytes    String with sequences of similar characters,
1209ccfebddSBram Moolenaar  *			    separated by slashes.
1219ccfebddSBram Moolenaar  *
1229ccfebddSBram Moolenaar  * sectionID == SN_COMPOUND: <compmax> <compminlen> <compsylmax> <compoptions>
1239ccfebddSBram Moolenaar  *				<comppatcount> <comppattern> ... <compflags>
1249ccfebddSBram Moolenaar  * <compmax>     1 byte	    Maximum nr of words in compound word.
1259ccfebddSBram Moolenaar  * <compminlen>  1 byte	    Minimal word length for compounding.
1269ccfebddSBram Moolenaar  * <compsylmax>  1 byte	    Maximum nr of syllables in compound word.
1279ccfebddSBram Moolenaar  * <compoptions> 2 bytes    COMP_ flags.
1289ccfebddSBram Moolenaar  * <comppatcount> 2 bytes   number of <comppattern> following
1299ccfebddSBram Moolenaar  * <compflags>   N bytes    Flags from COMPOUNDRULE items, separated by
1309ccfebddSBram Moolenaar  *			    slashes.
1319ccfebddSBram Moolenaar  *
1329ccfebddSBram Moolenaar  * <comppattern>: <comppatlen> <comppattext>
1339ccfebddSBram Moolenaar  * <comppatlen>	 1 byte	    length of <comppattext>
1349ccfebddSBram Moolenaar  * <comppattext> N bytes    end or begin chars from CHECKCOMPOUNDPATTERN
1359ccfebddSBram Moolenaar  *
1369ccfebddSBram Moolenaar  * sectionID == SN_NOBREAK: (empty, its presence is what matters)
1379ccfebddSBram Moolenaar  *
1389ccfebddSBram Moolenaar  * sectionID == SN_SYLLABLE: <syllable>
1399ccfebddSBram Moolenaar  * <syllable>    N bytes    String from SYLLABLE item.
1409ccfebddSBram Moolenaar  *
1419ccfebddSBram Moolenaar  * <LWORDTREE>: <wordtree>
1429ccfebddSBram Moolenaar  *
1439ccfebddSBram Moolenaar  * <KWORDTREE>: <wordtree>
1449ccfebddSBram Moolenaar  *
1459ccfebddSBram Moolenaar  * <PREFIXTREE>: <wordtree>
1469ccfebddSBram Moolenaar  *
1479ccfebddSBram Moolenaar  *
1489ccfebddSBram Moolenaar  * <wordtree>: <nodecount> <nodedata> ...
1499ccfebddSBram Moolenaar  *
1509ccfebddSBram Moolenaar  * <nodecount>	4 bytes	    Number of nodes following.  MSB first.
1519ccfebddSBram Moolenaar  *
1529ccfebddSBram Moolenaar  * <nodedata>: <siblingcount> <sibling> ...
1539ccfebddSBram Moolenaar  *
1549ccfebddSBram Moolenaar  * <siblingcount> 1 byte    Number of siblings in this node.  The siblings
1559ccfebddSBram Moolenaar  *			    follow in sorted order.
1569ccfebddSBram Moolenaar  *
1579ccfebddSBram Moolenaar  * <sibling>: <byte> [ <nodeidx> <xbyte>
1589ccfebddSBram Moolenaar  *		      | <flags> [<flags2>] [<region>] [<affixID>]
1599ccfebddSBram Moolenaar  *		      | [<pflags>] <affixID> <prefcondnr> ]
1609ccfebddSBram Moolenaar  *
1619ccfebddSBram Moolenaar  * <byte>	1 byte	    Byte value of the sibling.  Special cases:
1629ccfebddSBram Moolenaar  *			    BY_NOFLAGS: End of word without flags and for all
1639ccfebddSBram Moolenaar  *					regions.
1649ccfebddSBram Moolenaar  *					For PREFIXTREE <affixID> and
1659ccfebddSBram Moolenaar  *					<prefcondnr> follow.
1669ccfebddSBram Moolenaar  *			    BY_FLAGS:   End of word, <flags> follow.
1679ccfebddSBram Moolenaar  *					For PREFIXTREE <pflags>, <affixID>
1689ccfebddSBram Moolenaar  *					and <prefcondnr> follow.
1699ccfebddSBram Moolenaar  *			    BY_FLAGS2:  End of word, <flags> and <flags2>
1709ccfebddSBram Moolenaar  *					follow.  Not used in PREFIXTREE.
1719ccfebddSBram Moolenaar  *			    BY_INDEX:   Child of sibling is shared, <nodeidx>
1729ccfebddSBram Moolenaar  *					and <xbyte> follow.
1739ccfebddSBram Moolenaar  *
1749ccfebddSBram Moolenaar  * <nodeidx>	3 bytes	    Index of child for this sibling, MSB first.
1759ccfebddSBram Moolenaar  *
1769ccfebddSBram Moolenaar  * <xbyte>	1 byte	    byte value of the sibling.
1779ccfebddSBram Moolenaar  *
1789ccfebddSBram Moolenaar  * <flags>	1 byte	    bitmask of:
1799ccfebddSBram Moolenaar  *			    WF_ALLCAP	word must have only capitals
1809ccfebddSBram Moolenaar  *			    WF_ONECAP   first char of word must be capital
1819ccfebddSBram Moolenaar  *			    WF_KEEPCAP	keep-case word
1829ccfebddSBram Moolenaar  *			    WF_FIXCAP   keep-case word, all caps not allowed
1839ccfebddSBram Moolenaar  *			    WF_RARE	rare word
1849ccfebddSBram Moolenaar  *			    WF_BANNED	bad word
1859ccfebddSBram Moolenaar  *			    WF_REGION	<region> follows
1869ccfebddSBram Moolenaar  *			    WF_AFX	<affixID> follows
1879ccfebddSBram Moolenaar  *
1889ccfebddSBram Moolenaar  * <flags2>	1 byte	    Bitmask of:
1899ccfebddSBram Moolenaar  *			    WF_HAS_AFF >> 8   word includes affix
1909ccfebddSBram Moolenaar  *			    WF_NEEDCOMP >> 8  word only valid in compound
1919ccfebddSBram Moolenaar  *			    WF_NOSUGGEST >> 8  word not used for suggestions
1929ccfebddSBram Moolenaar  *			    WF_COMPROOT >> 8  word already a compound
1939ccfebddSBram Moolenaar  *			    WF_NOCOMPBEF >> 8 no compounding before this word
1949ccfebddSBram Moolenaar  *			    WF_NOCOMPAFT >> 8 no compounding after this word
1959ccfebddSBram Moolenaar  *
1969ccfebddSBram Moolenaar  * <pflags>	1 byte	    bitmask of:
1979ccfebddSBram Moolenaar  *			    WFP_RARE	rare prefix
1989ccfebddSBram Moolenaar  *			    WFP_NC	non-combining prefix
1999ccfebddSBram Moolenaar  *			    WFP_UP	letter after prefix made upper case
2009ccfebddSBram Moolenaar  *
2019ccfebddSBram Moolenaar  * <region>	1 byte	    Bitmask for regions in which word is valid.  When
2029ccfebddSBram Moolenaar  *			    omitted it's valid in all regions.
2039ccfebddSBram Moolenaar  *			    Lowest bit is for region 1.
2049ccfebddSBram Moolenaar  *
2059ccfebddSBram Moolenaar  * <affixID>	1 byte	    ID of affix that can be used with this word.  In
2069ccfebddSBram Moolenaar  *			    PREFIXTREE used for the required prefix ID.
2079ccfebddSBram Moolenaar  *
2089ccfebddSBram Moolenaar  * <prefcondnr>	2 bytes	    Prefix condition number, index in <prefcond> list
2099ccfebddSBram Moolenaar  *			    from HEADER.
2109ccfebddSBram Moolenaar  *
2119ccfebddSBram Moolenaar  * All text characters are in 'encoding', but stored as single bytes.
2129ccfebddSBram Moolenaar  */
2139ccfebddSBram Moolenaar 
2149ccfebddSBram Moolenaar /*
2159ccfebddSBram Moolenaar  * Vim .sug file format:  <SUGHEADER>
2169ccfebddSBram Moolenaar  *			  <SUGWORDTREE>
2179ccfebddSBram Moolenaar  *			  <SUGTABLE>
2189ccfebddSBram Moolenaar  *
2199ccfebddSBram Moolenaar  * <SUGHEADER>: <fileID> <versionnr> <timestamp>
2209ccfebddSBram Moolenaar  *
2219ccfebddSBram Moolenaar  * <fileID>     6 bytes     "VIMsug"
2229ccfebddSBram Moolenaar  * <versionnr>  1 byte      VIMSUGVERSION
2239ccfebddSBram Moolenaar  * <timestamp>  8 bytes     timestamp that must match with .spl file
2249ccfebddSBram Moolenaar  *
2259ccfebddSBram Moolenaar  *
2269ccfebddSBram Moolenaar  * <SUGWORDTREE>: <wordtree>  (see above, no flags or region used)
2279ccfebddSBram Moolenaar  *
2289ccfebddSBram Moolenaar  *
2299ccfebddSBram Moolenaar  * <SUGTABLE>: <sugwcount> <sugline> ...
2309ccfebddSBram Moolenaar  *
2319ccfebddSBram Moolenaar  * <sugwcount>	4 bytes	    number of <sugline> following
2329ccfebddSBram Moolenaar  *
2339ccfebddSBram Moolenaar  * <sugline>: <sugnr> ... NUL
2349ccfebddSBram Moolenaar  *
2359ccfebddSBram Moolenaar  * <sugnr>:     X bytes     word number that results in this soundfolded word,
2369ccfebddSBram Moolenaar  *			    stored as an offset to the previous number in as
2379ccfebddSBram Moolenaar  *			    few bytes as possible, see offset2bytes())
2389ccfebddSBram Moolenaar  */
2399ccfebddSBram Moolenaar 
2409ccfebddSBram Moolenaar #include "vim.h"
2419ccfebddSBram Moolenaar 
2429ccfebddSBram Moolenaar #if defined(FEAT_SPELL) || defined(PROTO)
2439ccfebddSBram Moolenaar 
2440d6f5d97SBram Moolenaar #ifndef UNIX		// it's in os_unix.h for Unix
2450d6f5d97SBram Moolenaar # include <time.h>	// for time_t
2469ccfebddSBram Moolenaar #endif
2479ccfebddSBram Moolenaar 
2480d6f5d97SBram Moolenaar #ifndef UNIX		// it's in os_unix.h for Unix
2490d6f5d97SBram Moolenaar # include <time.h>	// for time_t
2509ccfebddSBram Moolenaar #endif
2519ccfebddSBram Moolenaar 
2520d6f5d97SBram Moolenaar // Special byte values for <byte>.  Some are only used in the tree for
2530d6f5d97SBram Moolenaar // postponed prefixes, some only in the other trees.  This is a bit messy...
2540d6f5d97SBram Moolenaar #define BY_NOFLAGS	0	// end of word without flags or region; for
2550d6f5d97SBram Moolenaar 				// postponed prefix: no <pflags>
2560d6f5d97SBram Moolenaar #define BY_INDEX	1	// child is shared, index follows
2570d6f5d97SBram Moolenaar #define BY_FLAGS	2	// end of word, <flags> byte follows; for
2580d6f5d97SBram Moolenaar 				// postponed prefix: <pflags> follows
2590d6f5d97SBram Moolenaar #define BY_FLAGS2	3	// end of word, <flags> and <flags2> bytes
2600d6f5d97SBram Moolenaar 				// follow; never used in prefix tree
2610d6f5d97SBram Moolenaar #define BY_SPECIAL  BY_FLAGS2	// highest special byte value
2629ccfebddSBram Moolenaar 
2633d2a47c7SBram Moolenaar #define ZERO_FLAG	65009	// used when flag is zero: "0"
2643d2a47c7SBram Moolenaar 
2650d6f5d97SBram Moolenaar // Flags used in .spl file for soundsalike flags.
2669ccfebddSBram Moolenaar #define SAL_F0LLOWUP		1
2679ccfebddSBram Moolenaar #define SAL_COLLAPSE		2
2689ccfebddSBram Moolenaar #define SAL_REM_ACCENTS		4
2699ccfebddSBram Moolenaar 
2700d6f5d97SBram Moolenaar #define VIMSPELLMAGIC "VIMspell"  // string at start of Vim spell file
2719ccfebddSBram Moolenaar #define VIMSPELLMAGICL 8
2729ccfebddSBram Moolenaar #define VIMSPELLVERSION 50
2739ccfebddSBram Moolenaar 
2740d6f5d97SBram Moolenaar // Section IDs.  Only renumber them when VIMSPELLVERSION changes!
2750d6f5d97SBram Moolenaar #define SN_REGION	0	// <regionname> section
2760d6f5d97SBram Moolenaar #define SN_CHARFLAGS	1	// charflags section
2770d6f5d97SBram Moolenaar #define SN_MIDWORD	2	// <midword> section
2780d6f5d97SBram Moolenaar #define SN_PREFCOND	3	// <prefcond> section
2790d6f5d97SBram Moolenaar #define SN_REP		4	// REP items section
2800d6f5d97SBram Moolenaar #define SN_SAL		5	// SAL items section
2810d6f5d97SBram Moolenaar #define SN_SOFO		6	// soundfolding section
2820d6f5d97SBram Moolenaar #define SN_MAP		7	// MAP items section
2830d6f5d97SBram Moolenaar #define SN_COMPOUND	8	// compound words section
2840d6f5d97SBram Moolenaar #define SN_SYLLABLE	9	// syllable section
2850d6f5d97SBram Moolenaar #define SN_NOBREAK	10	// NOBREAK section
2860d6f5d97SBram Moolenaar #define SN_SUGFILE	11	// timestamp for .sug file
2870d6f5d97SBram Moolenaar #define SN_REPSAL	12	// REPSAL items section
2880d6f5d97SBram Moolenaar #define SN_WORDS	13	// common words
2890d6f5d97SBram Moolenaar #define SN_NOSPLITSUGS	14	// don't split word for suggestions
2900d6f5d97SBram Moolenaar #define SN_INFO		15	// info section
2910d6f5d97SBram Moolenaar #define SN_NOCOMPOUNDSUGS 16	// don't compound for suggestions
2920d6f5d97SBram Moolenaar #define SN_END		255	// end of sections
2939ccfebddSBram Moolenaar 
2940d6f5d97SBram Moolenaar #define SNF_REQUIRED	1	// <sectionflags>: required section
2959ccfebddSBram Moolenaar 
2969ccfebddSBram Moolenaar #define CF_WORD		0x01
2979ccfebddSBram Moolenaar #define CF_UPPER	0x02
2989ccfebddSBram Moolenaar 
299aeea7215SBram Moolenaar /*
300aeea7215SBram Moolenaar  * Loop through all the siblings of a node (including the node)
301aeea7215SBram Moolenaar  */
302aeea7215SBram Moolenaar #define FOR_ALL_NODE_SIBLINGS(node, np) \
303aeea7215SBram Moolenaar     for ((np) = (node); (np) != NULL; (np) = (np)->wn_sibling)
304aeea7215SBram Moolenaar 
3059ccfebddSBram Moolenaar static int set_spell_finish(spelltab_T	*new_st);
3069ccfebddSBram Moolenaar static int write_spell_prefcond(FILE *fd, garray_T *gap);
3079ccfebddSBram Moolenaar static int read_region_section(FILE *fd, slang_T *slang, int len);
3089ccfebddSBram Moolenaar static int read_charflags_section(FILE *fd);
3099ccfebddSBram Moolenaar static int read_prefcond_section(FILE *fd, slang_T *lp);
3109ccfebddSBram Moolenaar static int read_rep_section(FILE *fd, garray_T *gap, short *first);
3119ccfebddSBram Moolenaar static int read_sal_section(FILE *fd, slang_T *slang);
3129ccfebddSBram Moolenaar static int read_words_section(FILE *fd, slang_T *lp, int len);
3139ccfebddSBram Moolenaar static int read_sofo_section(FILE *fd, slang_T *slang);
3149ccfebddSBram Moolenaar static int read_compound(FILE *fd, slang_T *slang, int len);
3159ccfebddSBram Moolenaar static int set_sofo(slang_T *lp, char_u *from, char_u *to);
3169ccfebddSBram Moolenaar static void set_sal_first(slang_T *lp);
3179ccfebddSBram Moolenaar static int *mb_str2wide(char_u *s);
31807399e7fSBram Moolenaar static int spell_read_tree(FILE *fd, char_u **bytsp, long *bytsp_len, idx_T **idxsp, int prefixtree, int prefixcnt);
3199ccfebddSBram Moolenaar static idx_T read_tree_node(FILE *fd, char_u *byts, idx_T *idxs, int maxidx, idx_T startidx, int prefixtree, int maxprefcondnr);
3209ccfebddSBram Moolenaar static void set_spell_charflags(char_u *flags, int cnt, char_u *upp);
3219ccfebddSBram Moolenaar static int set_spell_chartab(char_u *fol, char_u *low, char_u *upp);
3229ccfebddSBram Moolenaar static void set_map_str(slang_T *lp, char_u *map);
3239ccfebddSBram Moolenaar 
3249ccfebddSBram Moolenaar 
3259ccfebddSBram Moolenaar static char *e_spell_trunc = N_("E758: Truncated spell file");
3269ccfebddSBram Moolenaar static char *e_afftrailing = N_("Trailing text in %s line %d: %s");
3279ccfebddSBram Moolenaar static char *e_affname = N_("Affix name too long in %s line %d: %s");
3289ccfebddSBram Moolenaar static char *e_affform = N_("E761: Format error in affix file FOL, LOW or UPP");
3299ccfebddSBram Moolenaar static char *e_affrange = N_("E762: Character in FOL, LOW or UPP is out of range");
3309ccfebddSBram Moolenaar static char *msg_compressing = N_("Compressing word tree...");
3319ccfebddSBram Moolenaar 
3329ccfebddSBram Moolenaar /*
3339ccfebddSBram Moolenaar  * Load one spell file and store the info into a slang_T.
3349ccfebddSBram Moolenaar  *
3359ccfebddSBram Moolenaar  * This is invoked in three ways:
3369ccfebddSBram Moolenaar  * - From spell_load_cb() to load a spell file for the first time.  "lang" is
3379ccfebddSBram Moolenaar  *   the language name, "old_lp" is NULL.  Will allocate an slang_T.
3389ccfebddSBram Moolenaar  * - To reload a spell file that was changed.  "lang" is NULL and "old_lp"
3399ccfebddSBram Moolenaar  *   points to the existing slang_T.
3409ccfebddSBram Moolenaar  * - Just after writing a .spl file; it's read back to produce the .sug file.
3419ccfebddSBram Moolenaar  *   "old_lp" is NULL and "lang" is NULL.  Will allocate an slang_T.
3429ccfebddSBram Moolenaar  *
3439ccfebddSBram Moolenaar  * Returns the slang_T the spell file was loaded into.  NULL for error.
3449ccfebddSBram Moolenaar  */
3459ccfebddSBram Moolenaar     slang_T *
spell_load_file(char_u * fname,char_u * lang,slang_T * old_lp,int silent)3469ccfebddSBram Moolenaar spell_load_file(
3479ccfebddSBram Moolenaar     char_u	*fname,
3489ccfebddSBram Moolenaar     char_u	*lang,
3499ccfebddSBram Moolenaar     slang_T	*old_lp,
3500d6f5d97SBram Moolenaar     int		silent)		// no error if file doesn't exist
3519ccfebddSBram Moolenaar {
3529ccfebddSBram Moolenaar     FILE	*fd;
3539ccfebddSBram Moolenaar     char_u	buf[VIMSPELLMAGICL];
3549ccfebddSBram Moolenaar     char_u	*p;
3559ccfebddSBram Moolenaar     int		i;
3569ccfebddSBram Moolenaar     int		n;
3579ccfebddSBram Moolenaar     int		len;
3589ccfebddSBram Moolenaar     slang_T	*lp = NULL;
3599ccfebddSBram Moolenaar     int		c = 0;
3609ccfebddSBram Moolenaar     int		res;
361ce6db027SBram Moolenaar     int		did_estack_push = FALSE;
362e31ee868SBram Moolenaar     ESTACK_CHECK_DECLARATION
3639ccfebddSBram Moolenaar 
3649ccfebddSBram Moolenaar     fd = mch_fopen((char *)fname, "r");
3659ccfebddSBram Moolenaar     if (fd == NULL)
3669ccfebddSBram Moolenaar     {
3679ccfebddSBram Moolenaar 	if (!silent)
368f9e3e09fSBram Moolenaar 	    semsg(_(e_notopen), fname);
3699ccfebddSBram Moolenaar 	else if (p_verbose > 2)
3709ccfebddSBram Moolenaar 	{
3719ccfebddSBram Moolenaar 	    verbose_enter();
372f9e3e09fSBram Moolenaar 	    smsg((const char *)e_notopen, fname);
3739ccfebddSBram Moolenaar 	    verbose_leave();
3749ccfebddSBram Moolenaar 	}
3759ccfebddSBram Moolenaar 	goto endFAIL;
3769ccfebddSBram Moolenaar     }
3779ccfebddSBram Moolenaar     if (p_verbose > 2)
3789ccfebddSBram Moolenaar     {
3799ccfebddSBram Moolenaar 	verbose_enter();
380f9e3e09fSBram Moolenaar 	smsg(_("Reading spell file \"%s\""), fname);
3819ccfebddSBram Moolenaar 	verbose_leave();
3829ccfebddSBram Moolenaar     }
3839ccfebddSBram Moolenaar 
3849ccfebddSBram Moolenaar     if (old_lp == NULL)
3859ccfebddSBram Moolenaar     {
3869ccfebddSBram Moolenaar 	lp = slang_alloc(lang);
3879ccfebddSBram Moolenaar 	if (lp == NULL)
3889ccfebddSBram Moolenaar 	    goto endFAIL;
3899ccfebddSBram Moolenaar 
3900d6f5d97SBram Moolenaar 	// Remember the file name, used to reload the file when it's updated.
3919ccfebddSBram Moolenaar 	lp->sl_fname = vim_strsave(fname);
3929ccfebddSBram Moolenaar 	if (lp->sl_fname == NULL)
3939ccfebddSBram Moolenaar 	    goto endFAIL;
3949ccfebddSBram Moolenaar 
3950d6f5d97SBram Moolenaar 	// Check for .add.spl (_add.spl for VMS).
3969ccfebddSBram Moolenaar 	lp->sl_add = strstr((char *)gettail(fname), SPL_FNAME_ADD) != NULL;
3979ccfebddSBram Moolenaar     }
3989ccfebddSBram Moolenaar     else
3999ccfebddSBram Moolenaar 	lp = old_lp;
4009ccfebddSBram Moolenaar 
4010d6f5d97SBram Moolenaar     // Set sourcing_name, so that error messages mention the file name.
4021a47ae32SBram Moolenaar     estack_push(ETYPE_SPELL, fname, 0);
403e31ee868SBram Moolenaar     ESTACK_CHECK_SETUP
404ce6db027SBram Moolenaar     did_estack_push = TRUE;
4059ccfebddSBram Moolenaar 
4069ccfebddSBram Moolenaar     /*
4079ccfebddSBram Moolenaar      * <HEADER>: <fileID>
4089ccfebddSBram Moolenaar      */
4099ccfebddSBram Moolenaar     for (i = 0; i < VIMSPELLMAGICL; ++i)
4100d6f5d97SBram Moolenaar 	buf[i] = getc(fd);				// <fileID>
4119ccfebddSBram Moolenaar     if (STRNCMP(buf, VIMSPELLMAGIC, VIMSPELLMAGICL) != 0)
4129ccfebddSBram Moolenaar     {
413f9e3e09fSBram Moolenaar 	emsg(_("E757: This does not look like a spell file"));
4149ccfebddSBram Moolenaar 	goto endFAIL;
4159ccfebddSBram Moolenaar     }
4160d6f5d97SBram Moolenaar     c = getc(fd);					// <versionnr>
4179ccfebddSBram Moolenaar     if (c < VIMSPELLVERSION)
4189ccfebddSBram Moolenaar     {
419f9e3e09fSBram Moolenaar 	emsg(_("E771: Old spell file, needs to be updated"));
4209ccfebddSBram Moolenaar 	goto endFAIL;
4219ccfebddSBram Moolenaar     }
4229ccfebddSBram Moolenaar     else if (c > VIMSPELLVERSION)
4239ccfebddSBram Moolenaar     {
424f9e3e09fSBram Moolenaar 	emsg(_("E772: Spell file is for newer version of Vim"));
4259ccfebddSBram Moolenaar 	goto endFAIL;
4269ccfebddSBram Moolenaar     }
4279ccfebddSBram Moolenaar 
4289ccfebddSBram Moolenaar 
4299ccfebddSBram Moolenaar     /*
4309ccfebddSBram Moolenaar      * <SECTIONS>: <section> ... <sectionend>
4319ccfebddSBram Moolenaar      * <section>: <sectionID> <sectionflags> <sectionlen> (section contents)
4329ccfebddSBram Moolenaar      */
4339ccfebddSBram Moolenaar     for (;;)
4349ccfebddSBram Moolenaar     {
4350d6f5d97SBram Moolenaar 	n = getc(fd);			    // <sectionID> or <sectionend>
4369ccfebddSBram Moolenaar 	if (n == SN_END)
4379ccfebddSBram Moolenaar 	    break;
4380d6f5d97SBram Moolenaar 	c = getc(fd);					// <sectionflags>
4390d6f5d97SBram Moolenaar 	len = get4c(fd);				// <sectionlen>
4409ccfebddSBram Moolenaar 	if (len < 0)
4419ccfebddSBram Moolenaar 	    goto truncerr;
4429ccfebddSBram Moolenaar 
4439ccfebddSBram Moolenaar 	res = 0;
4449ccfebddSBram Moolenaar 	switch (n)
4459ccfebddSBram Moolenaar 	{
4469ccfebddSBram Moolenaar 	    case SN_INFO:
4470d6f5d97SBram Moolenaar 		lp->sl_info = read_string(fd, len);	// <infotext>
4489ccfebddSBram Moolenaar 		if (lp->sl_info == NULL)
4499ccfebddSBram Moolenaar 		    goto endFAIL;
4509ccfebddSBram Moolenaar 		break;
4519ccfebddSBram Moolenaar 
4529ccfebddSBram Moolenaar 	    case SN_REGION:
4539ccfebddSBram Moolenaar 		res = read_region_section(fd, lp, len);
4549ccfebddSBram Moolenaar 		break;
4559ccfebddSBram Moolenaar 
4569ccfebddSBram Moolenaar 	    case SN_CHARFLAGS:
4579ccfebddSBram Moolenaar 		res = read_charflags_section(fd);
4589ccfebddSBram Moolenaar 		break;
4599ccfebddSBram Moolenaar 
4609ccfebddSBram Moolenaar 	    case SN_MIDWORD:
4610d6f5d97SBram Moolenaar 		lp->sl_midword = read_string(fd, len);	// <midword>
4629ccfebddSBram Moolenaar 		if (lp->sl_midword == NULL)
4639ccfebddSBram Moolenaar 		    goto endFAIL;
4649ccfebddSBram Moolenaar 		break;
4659ccfebddSBram Moolenaar 
4669ccfebddSBram Moolenaar 	    case SN_PREFCOND:
4679ccfebddSBram Moolenaar 		res = read_prefcond_section(fd, lp);
4689ccfebddSBram Moolenaar 		break;
4699ccfebddSBram Moolenaar 
4709ccfebddSBram Moolenaar 	    case SN_REP:
4719ccfebddSBram Moolenaar 		res = read_rep_section(fd, &lp->sl_rep, lp->sl_rep_first);
4729ccfebddSBram Moolenaar 		break;
4739ccfebddSBram Moolenaar 
4749ccfebddSBram Moolenaar 	    case SN_REPSAL:
4759ccfebddSBram Moolenaar 		res = read_rep_section(fd, &lp->sl_repsal, lp->sl_repsal_first);
4769ccfebddSBram Moolenaar 		break;
4779ccfebddSBram Moolenaar 
4789ccfebddSBram Moolenaar 	    case SN_SAL:
4799ccfebddSBram Moolenaar 		res = read_sal_section(fd, lp);
4809ccfebddSBram Moolenaar 		break;
4819ccfebddSBram Moolenaar 
4829ccfebddSBram Moolenaar 	    case SN_SOFO:
4839ccfebddSBram Moolenaar 		res = read_sofo_section(fd, lp);
4849ccfebddSBram Moolenaar 		break;
4859ccfebddSBram Moolenaar 
4869ccfebddSBram Moolenaar 	    case SN_MAP:
4870d6f5d97SBram Moolenaar 		p = read_string(fd, len);		// <mapstr>
4889ccfebddSBram Moolenaar 		if (p == NULL)
4899ccfebddSBram Moolenaar 		    goto endFAIL;
4909ccfebddSBram Moolenaar 		set_map_str(lp, p);
4919ccfebddSBram Moolenaar 		vim_free(p);
4929ccfebddSBram Moolenaar 		break;
4939ccfebddSBram Moolenaar 
4949ccfebddSBram Moolenaar 	    case SN_WORDS:
4959ccfebddSBram Moolenaar 		res = read_words_section(fd, lp, len);
4969ccfebddSBram Moolenaar 		break;
4979ccfebddSBram Moolenaar 
4989ccfebddSBram Moolenaar 	    case SN_SUGFILE:
4990d6f5d97SBram Moolenaar 		lp->sl_sugtime = get8ctime(fd);		// <timestamp>
5009ccfebddSBram Moolenaar 		break;
5019ccfebddSBram Moolenaar 
5029ccfebddSBram Moolenaar 	    case SN_NOSPLITSUGS:
5039ccfebddSBram Moolenaar 		lp->sl_nosplitsugs = TRUE;
5049ccfebddSBram Moolenaar 		break;
5059ccfebddSBram Moolenaar 
5069ccfebddSBram Moolenaar 	    case SN_NOCOMPOUNDSUGS:
5079ccfebddSBram Moolenaar 		lp->sl_nocompoundsugs = TRUE;
5089ccfebddSBram Moolenaar 		break;
5099ccfebddSBram Moolenaar 
5109ccfebddSBram Moolenaar 	    case SN_COMPOUND:
5119ccfebddSBram Moolenaar 		res = read_compound(fd, lp, len);
5129ccfebddSBram Moolenaar 		break;
5139ccfebddSBram Moolenaar 
5149ccfebddSBram Moolenaar 	    case SN_NOBREAK:
5159ccfebddSBram Moolenaar 		lp->sl_nobreak = TRUE;
5169ccfebddSBram Moolenaar 		break;
5179ccfebddSBram Moolenaar 
5189ccfebddSBram Moolenaar 	    case SN_SYLLABLE:
5190d6f5d97SBram Moolenaar 		lp->sl_syllable = read_string(fd, len);	// <syllable>
5209ccfebddSBram Moolenaar 		if (lp->sl_syllable == NULL)
5219ccfebddSBram Moolenaar 		    goto endFAIL;
522fc2a47ffSBram Moolenaar 		if (init_syl_tab(lp) != OK)
5239ccfebddSBram Moolenaar 		    goto endFAIL;
5249ccfebddSBram Moolenaar 		break;
5259ccfebddSBram Moolenaar 
5269ccfebddSBram Moolenaar 	    default:
5270d6f5d97SBram Moolenaar 		// Unsupported section.  When it's required give an error
5280d6f5d97SBram Moolenaar 		// message.  When it's not required skip the contents.
5299ccfebddSBram Moolenaar 		if (c & SNF_REQUIRED)
5309ccfebddSBram Moolenaar 		{
531f9e3e09fSBram Moolenaar 		    emsg(_("E770: Unsupported section in spell file"));
5329ccfebddSBram Moolenaar 		    goto endFAIL;
5339ccfebddSBram Moolenaar 		}
5349ccfebddSBram Moolenaar 		while (--len >= 0)
5359ccfebddSBram Moolenaar 		    if (getc(fd) < 0)
5369ccfebddSBram Moolenaar 			goto truncerr;
5379ccfebddSBram Moolenaar 		break;
5389ccfebddSBram Moolenaar 	}
5399ccfebddSBram Moolenaar someerror:
5409ccfebddSBram Moolenaar 	if (res == SP_FORMERROR)
5419ccfebddSBram Moolenaar 	{
542f9e3e09fSBram Moolenaar 	    emsg(_(e_format));
5439ccfebddSBram Moolenaar 	    goto endFAIL;
5449ccfebddSBram Moolenaar 	}
5459ccfebddSBram Moolenaar 	if (res == SP_TRUNCERROR)
5469ccfebddSBram Moolenaar 	{
5479ccfebddSBram Moolenaar truncerr:
548f9e3e09fSBram Moolenaar 	    emsg(_(e_spell_trunc));
5499ccfebddSBram Moolenaar 	    goto endFAIL;
5509ccfebddSBram Moolenaar 	}
5519ccfebddSBram Moolenaar 	if (res == SP_OTHERERROR)
5529ccfebddSBram Moolenaar 	    goto endFAIL;
5539ccfebddSBram Moolenaar     }
5549ccfebddSBram Moolenaar 
5550d6f5d97SBram Moolenaar     // <LWORDTREE>
55607399e7fSBram Moolenaar     res = spell_read_tree(fd, &lp->sl_fbyts, &lp->sl_fbyts_len,
55707399e7fSBram Moolenaar 						      &lp->sl_fidxs, FALSE, 0);
5589ccfebddSBram Moolenaar     if (res != 0)
5599ccfebddSBram Moolenaar 	goto someerror;
5609ccfebddSBram Moolenaar 
5610d6f5d97SBram Moolenaar     // <KWORDTREE>
56207399e7fSBram Moolenaar     res = spell_read_tree(fd, &lp->sl_kbyts, NULL, &lp->sl_kidxs, FALSE, 0);
5639ccfebddSBram Moolenaar     if (res != 0)
5649ccfebddSBram Moolenaar 	goto someerror;
5659ccfebddSBram Moolenaar 
5660d6f5d97SBram Moolenaar     // <PREFIXTREE>
56707399e7fSBram Moolenaar     res = spell_read_tree(fd, &lp->sl_pbyts, NULL, &lp->sl_pidxs, TRUE,
5689ccfebddSBram Moolenaar 							    lp->sl_prefixcnt);
5699ccfebddSBram Moolenaar     if (res != 0)
5709ccfebddSBram Moolenaar 	goto someerror;
5719ccfebddSBram Moolenaar 
5720d6f5d97SBram Moolenaar     // For a new file link it in the list of spell files.
5739ccfebddSBram Moolenaar     if (old_lp == NULL && lang != NULL)
5749ccfebddSBram Moolenaar     {
5759ccfebddSBram Moolenaar 	lp->sl_next = first_lang;
5769ccfebddSBram Moolenaar 	first_lang = lp;
5779ccfebddSBram Moolenaar     }
5789ccfebddSBram Moolenaar 
5799ccfebddSBram Moolenaar     goto endOK;
5809ccfebddSBram Moolenaar 
5819ccfebddSBram Moolenaar endFAIL:
5829ccfebddSBram Moolenaar     if (lang != NULL)
5830d6f5d97SBram Moolenaar 	// truncating the name signals the error to spell_load_lang()
5849ccfebddSBram Moolenaar 	*lang = NUL;
5859ccfebddSBram Moolenaar     if (lp != NULL && old_lp == NULL)
5869ccfebddSBram Moolenaar 	slang_free(lp);
5879ccfebddSBram Moolenaar     lp = NULL;
5889ccfebddSBram Moolenaar 
5899ccfebddSBram Moolenaar endOK:
5909ccfebddSBram Moolenaar     if (fd != NULL)
5919ccfebddSBram Moolenaar 	fclose(fd);
592ce6db027SBram Moolenaar     if (did_estack_push)
593e31ee868SBram Moolenaar     {
594e31ee868SBram Moolenaar 	ESTACK_CHECK_NOW
5951a47ae32SBram Moolenaar 	estack_pop();
596e31ee868SBram Moolenaar     }
5979ccfebddSBram Moolenaar 
5989ccfebddSBram Moolenaar     return lp;
5999ccfebddSBram Moolenaar }
6009ccfebddSBram Moolenaar 
6019ccfebddSBram Moolenaar /*
6029ccfebddSBram Moolenaar  * Fill in the wordcount fields for a trie.
6039ccfebddSBram Moolenaar  * Returns the total number of words.
6049ccfebddSBram Moolenaar  */
6059ccfebddSBram Moolenaar     static void
tree_count_words(char_u * byts,idx_T * idxs)6069ccfebddSBram Moolenaar tree_count_words(char_u *byts, idx_T *idxs)
6079ccfebddSBram Moolenaar {
6089ccfebddSBram Moolenaar     int		depth;
6099ccfebddSBram Moolenaar     idx_T	arridx[MAXWLEN];
6109ccfebddSBram Moolenaar     int		curi[MAXWLEN];
6119ccfebddSBram Moolenaar     int		c;
6129ccfebddSBram Moolenaar     idx_T	n;
6139ccfebddSBram Moolenaar     int		wordcount[MAXWLEN];
6149ccfebddSBram Moolenaar 
6159ccfebddSBram Moolenaar     arridx[0] = 0;
6169ccfebddSBram Moolenaar     curi[0] = 1;
6179ccfebddSBram Moolenaar     wordcount[0] = 0;
6189ccfebddSBram Moolenaar     depth = 0;
6199ccfebddSBram Moolenaar     while (depth >= 0 && !got_int)
6209ccfebddSBram Moolenaar     {
6219ccfebddSBram Moolenaar 	if (curi[depth] > byts[arridx[depth]])
6229ccfebddSBram Moolenaar 	{
6230d6f5d97SBram Moolenaar 	    // Done all bytes at this node, go up one level.
6249ccfebddSBram Moolenaar 	    idxs[arridx[depth]] = wordcount[depth];
6259ccfebddSBram Moolenaar 	    if (depth > 0)
6269ccfebddSBram Moolenaar 		wordcount[depth - 1] += wordcount[depth];
6279ccfebddSBram Moolenaar 
6289ccfebddSBram Moolenaar 	    --depth;
6299ccfebddSBram Moolenaar 	    fast_breakcheck();
6309ccfebddSBram Moolenaar 	}
6319ccfebddSBram Moolenaar 	else
6329ccfebddSBram Moolenaar 	{
6330d6f5d97SBram Moolenaar 	    // Do one more byte at this node.
6349ccfebddSBram Moolenaar 	    n = arridx[depth] + curi[depth];
6359ccfebddSBram Moolenaar 	    ++curi[depth];
6369ccfebddSBram Moolenaar 
6379ccfebddSBram Moolenaar 	    c = byts[n];
6389ccfebddSBram Moolenaar 	    if (c == 0)
6399ccfebddSBram Moolenaar 	    {
6400d6f5d97SBram Moolenaar 		// End of word, count it.
6419ccfebddSBram Moolenaar 		++wordcount[depth];
6429ccfebddSBram Moolenaar 
6430d6f5d97SBram Moolenaar 		// Skip over any other NUL bytes (same word with different
6440d6f5d97SBram Moolenaar 		// flags).
6459ccfebddSBram Moolenaar 		while (byts[n + 1] == 0)
6469ccfebddSBram Moolenaar 		{
6479ccfebddSBram Moolenaar 		    ++n;
6489ccfebddSBram Moolenaar 		    ++curi[depth];
6499ccfebddSBram Moolenaar 		}
6509ccfebddSBram Moolenaar 	    }
6519ccfebddSBram Moolenaar 	    else
6529ccfebddSBram Moolenaar 	    {
6530d6f5d97SBram Moolenaar 		// Normal char, go one level deeper to count the words.
6549ccfebddSBram Moolenaar 		++depth;
6559ccfebddSBram Moolenaar 		arridx[depth] = idxs[n];
6569ccfebddSBram Moolenaar 		curi[depth] = 1;
6579ccfebddSBram Moolenaar 		wordcount[depth] = 0;
6589ccfebddSBram Moolenaar 	    }
6599ccfebddSBram Moolenaar 	}
6609ccfebddSBram Moolenaar     }
6619ccfebddSBram Moolenaar }
6629ccfebddSBram Moolenaar 
6639ccfebddSBram Moolenaar /*
6649ccfebddSBram Moolenaar  * Load the .sug files for languages that have one and weren't loaded yet.
6659ccfebddSBram Moolenaar  */
6669ccfebddSBram Moolenaar     void
suggest_load_files(void)6679ccfebddSBram Moolenaar suggest_load_files(void)
6689ccfebddSBram Moolenaar {
6699ccfebddSBram Moolenaar     langp_T	*lp;
6709ccfebddSBram Moolenaar     int		lpi;
6719ccfebddSBram Moolenaar     slang_T	*slang;
6729ccfebddSBram Moolenaar     char_u	*dotp;
6739ccfebddSBram Moolenaar     FILE	*fd;
6749ccfebddSBram Moolenaar     char_u	buf[MAXWLEN];
6759ccfebddSBram Moolenaar     int		i;
6769ccfebddSBram Moolenaar     time_t	timestamp;
6779ccfebddSBram Moolenaar     int		wcount;
6789ccfebddSBram Moolenaar     int		wordnr;
6799ccfebddSBram Moolenaar     garray_T	ga;
6809ccfebddSBram Moolenaar     int		c;
6819ccfebddSBram Moolenaar 
6820d6f5d97SBram Moolenaar     // Do this for all languages that support sound folding.
6839ccfebddSBram Moolenaar     for (lpi = 0; lpi < curwin->w_s->b_langp.ga_len; ++lpi)
6849ccfebddSBram Moolenaar     {
6859ccfebddSBram Moolenaar 	lp = LANGP_ENTRY(curwin->w_s->b_langp, lpi);
6869ccfebddSBram Moolenaar 	slang = lp->lp_slang;
6879ccfebddSBram Moolenaar 	if (slang->sl_sugtime != 0 && !slang->sl_sugloaded)
6889ccfebddSBram Moolenaar 	{
6890d6f5d97SBram Moolenaar 	    // Change ".spl" to ".sug" and open the file.  When the file isn't
6900d6f5d97SBram Moolenaar 	    // found silently skip it.  Do set "sl_sugloaded" so that we
6910d6f5d97SBram Moolenaar 	    // don't try again and again.
6929ccfebddSBram Moolenaar 	    slang->sl_sugloaded = TRUE;
6939ccfebddSBram Moolenaar 
6949ccfebddSBram Moolenaar 	    dotp = vim_strrchr(slang->sl_fname, '.');
6959ccfebddSBram Moolenaar 	    if (dotp == NULL || fnamecmp(dotp, ".spl") != 0)
6969ccfebddSBram Moolenaar 		continue;
6979ccfebddSBram Moolenaar 	    STRCPY(dotp, ".sug");
6989ccfebddSBram Moolenaar 	    fd = mch_fopen((char *)slang->sl_fname, "r");
6999ccfebddSBram Moolenaar 	    if (fd == NULL)
7009ccfebddSBram Moolenaar 		goto nextone;
7019ccfebddSBram Moolenaar 
7029ccfebddSBram Moolenaar 	    /*
7039ccfebddSBram Moolenaar 	     * <SUGHEADER>: <fileID> <versionnr> <timestamp>
7049ccfebddSBram Moolenaar 	     */
7059ccfebddSBram Moolenaar 	    for (i = 0; i < VIMSUGMAGICL; ++i)
7060d6f5d97SBram Moolenaar 		buf[i] = getc(fd);			// <fileID>
7079ccfebddSBram Moolenaar 	    if (STRNCMP(buf, VIMSUGMAGIC, VIMSUGMAGICL) != 0)
7089ccfebddSBram Moolenaar 	    {
709f9e3e09fSBram Moolenaar 		semsg(_("E778: This does not look like a .sug file: %s"),
7109ccfebddSBram Moolenaar 							     slang->sl_fname);
7119ccfebddSBram Moolenaar 		goto nextone;
7129ccfebddSBram Moolenaar 	    }
7130d6f5d97SBram Moolenaar 	    c = getc(fd);				// <versionnr>
7149ccfebddSBram Moolenaar 	    if (c < VIMSUGVERSION)
7159ccfebddSBram Moolenaar 	    {
716f9e3e09fSBram Moolenaar 		semsg(_("E779: Old .sug file, needs to be updated: %s"),
7179ccfebddSBram Moolenaar 							     slang->sl_fname);
7189ccfebddSBram Moolenaar 		goto nextone;
7199ccfebddSBram Moolenaar 	    }
7209ccfebddSBram Moolenaar 	    else if (c > VIMSUGVERSION)
7219ccfebddSBram Moolenaar 	    {
722f9e3e09fSBram Moolenaar 		semsg(_("E780: .sug file is for newer version of Vim: %s"),
7239ccfebddSBram Moolenaar 							     slang->sl_fname);
7249ccfebddSBram Moolenaar 		goto nextone;
7259ccfebddSBram Moolenaar 	    }
7269ccfebddSBram Moolenaar 
7270d6f5d97SBram Moolenaar 	    // Check the timestamp, it must be exactly the same as the one in
7280d6f5d97SBram Moolenaar 	    // the .spl file.  Otherwise the word numbers won't match.
7290d6f5d97SBram Moolenaar 	    timestamp = get8ctime(fd);			// <timestamp>
7309ccfebddSBram Moolenaar 	    if (timestamp != slang->sl_sugtime)
7319ccfebddSBram Moolenaar 	    {
732f9e3e09fSBram Moolenaar 		semsg(_("E781: .sug file doesn't match .spl file: %s"),
7339ccfebddSBram Moolenaar 							     slang->sl_fname);
7349ccfebddSBram Moolenaar 		goto nextone;
7359ccfebddSBram Moolenaar 	    }
7369ccfebddSBram Moolenaar 
7379ccfebddSBram Moolenaar 	    /*
7389ccfebddSBram Moolenaar 	     * <SUGWORDTREE>: <wordtree>
7399ccfebddSBram Moolenaar 	     * Read the trie with the soundfolded words.
7409ccfebddSBram Moolenaar 	     */
74107399e7fSBram Moolenaar 	    if (spell_read_tree(fd, &slang->sl_sbyts, NULL, &slang->sl_sidxs,
7429ccfebddSBram Moolenaar 							       FALSE, 0) != 0)
7439ccfebddSBram Moolenaar 	    {
7449ccfebddSBram Moolenaar someerror:
745f9e3e09fSBram Moolenaar 		semsg(_("E782: error while reading .sug file: %s"),
7469ccfebddSBram Moolenaar 							     slang->sl_fname);
7479ccfebddSBram Moolenaar 		slang_clear_sug(slang);
7489ccfebddSBram Moolenaar 		goto nextone;
7499ccfebddSBram Moolenaar 	    }
7509ccfebddSBram Moolenaar 
7519ccfebddSBram Moolenaar 	    /*
7529ccfebddSBram Moolenaar 	     * <SUGTABLE>: <sugwcount> <sugline> ...
7539ccfebddSBram Moolenaar 	     *
7549ccfebddSBram Moolenaar 	     * Read the table with word numbers.  We use a file buffer for
7559ccfebddSBram Moolenaar 	     * this, because it's so much like a file with lines.  Makes it
7569ccfebddSBram Moolenaar 	     * possible to swap the info and save on memory use.
7579ccfebddSBram Moolenaar 	     */
7589ccfebddSBram Moolenaar 	    slang->sl_sugbuf = open_spellbuf();
7599ccfebddSBram Moolenaar 	    if (slang->sl_sugbuf == NULL)
7609ccfebddSBram Moolenaar 		goto someerror;
7610d6f5d97SBram Moolenaar 							    // <sugwcount>
7629ccfebddSBram Moolenaar 	    wcount = get4c(fd);
7639ccfebddSBram Moolenaar 	    if (wcount < 0)
7649ccfebddSBram Moolenaar 		goto someerror;
7659ccfebddSBram Moolenaar 
7660d6f5d97SBram Moolenaar 	    // Read all the wordnr lists into the buffer, one NUL terminated
7670d6f5d97SBram Moolenaar 	    // list per line.
7689ccfebddSBram Moolenaar 	    ga_init2(&ga, 1, 100);
7699ccfebddSBram Moolenaar 	    for (wordnr = 0; wordnr < wcount; ++wordnr)
7709ccfebddSBram Moolenaar 	    {
7719ccfebddSBram Moolenaar 		ga.ga_len = 0;
7729ccfebddSBram Moolenaar 		for (;;)
7739ccfebddSBram Moolenaar 		{
7740d6f5d97SBram Moolenaar 		    c = getc(fd);			    // <sugline>
7759ccfebddSBram Moolenaar 		    if (c < 0 || ga_grow(&ga, 1) == FAIL)
7769ccfebddSBram Moolenaar 			goto someerror;
7779ccfebddSBram Moolenaar 		    ((char_u *)ga.ga_data)[ga.ga_len++] = c;
7789ccfebddSBram Moolenaar 		    if (c == NUL)
7799ccfebddSBram Moolenaar 			break;
7809ccfebddSBram Moolenaar 		}
7819ccfebddSBram Moolenaar 		if (ml_append_buf(slang->sl_sugbuf, (linenr_T)wordnr,
7829ccfebddSBram Moolenaar 					 ga.ga_data, ga.ga_len, TRUE) == FAIL)
7839ccfebddSBram Moolenaar 		    goto someerror;
7849ccfebddSBram Moolenaar 	    }
7859ccfebddSBram Moolenaar 	    ga_clear(&ga);
7869ccfebddSBram Moolenaar 
7879ccfebddSBram Moolenaar 	    /*
7889ccfebddSBram Moolenaar 	     * Need to put word counts in the word tries, so that we can find
7899ccfebddSBram Moolenaar 	     * a word by its number.
7909ccfebddSBram Moolenaar 	     */
7919ccfebddSBram Moolenaar 	    tree_count_words(slang->sl_fbyts, slang->sl_fidxs);
7929ccfebddSBram Moolenaar 	    tree_count_words(slang->sl_sbyts, slang->sl_sidxs);
7939ccfebddSBram Moolenaar 
7949ccfebddSBram Moolenaar nextone:
7959ccfebddSBram Moolenaar 	    if (fd != NULL)
7969ccfebddSBram Moolenaar 		fclose(fd);
7979ccfebddSBram Moolenaar 	    STRCPY(dotp, ".spl");
7989ccfebddSBram Moolenaar 	}
7999ccfebddSBram Moolenaar     }
8009ccfebddSBram Moolenaar }
8019ccfebddSBram Moolenaar 
8029ccfebddSBram Moolenaar 
8039ccfebddSBram Moolenaar /*
8049ccfebddSBram Moolenaar  * Read a length field from "fd" in "cnt_bytes" bytes.
8059ccfebddSBram Moolenaar  * Allocate memory, read the string into it and add a NUL at the end.
8069ccfebddSBram Moolenaar  * Returns NULL when the count is zero.
8079ccfebddSBram Moolenaar  * Sets "*cntp" to SP_*ERROR when there is an error, length of the result
8089ccfebddSBram Moolenaar  * otherwise.
8099ccfebddSBram Moolenaar  */
8109ccfebddSBram Moolenaar     static char_u *
read_cnt_string(FILE * fd,int cnt_bytes,int * cntp)8119ccfebddSBram Moolenaar read_cnt_string(FILE *fd, int cnt_bytes, int *cntp)
8129ccfebddSBram Moolenaar {
8139ccfebddSBram Moolenaar     int		cnt = 0;
8149ccfebddSBram Moolenaar     int		i;
8159ccfebddSBram Moolenaar     char_u	*str;
8169ccfebddSBram Moolenaar 
8170d6f5d97SBram Moolenaar     // read the length bytes, MSB first
8189ccfebddSBram Moolenaar     for (i = 0; i < cnt_bytes; ++i)
8194ad739fcSBram Moolenaar     {
8204ad739fcSBram Moolenaar 	int c = getc(fd);
8214ad739fcSBram Moolenaar 
8224ad739fcSBram Moolenaar 	if (c == EOF)
8239ccfebddSBram Moolenaar 	{
8249ccfebddSBram Moolenaar 	    *cntp = SP_TRUNCERROR;
8259ccfebddSBram Moolenaar 	    return NULL;
8269ccfebddSBram Moolenaar 	}
8274ad739fcSBram Moolenaar 	cnt = (cnt << 8) + (unsigned)c;
8284ad739fcSBram Moolenaar     }
8299ccfebddSBram Moolenaar     *cntp = cnt;
8309ccfebddSBram Moolenaar     if (cnt == 0)
8310d6f5d97SBram Moolenaar 	return NULL;	    // nothing to read, return NULL
8329ccfebddSBram Moolenaar 
8339ccfebddSBram Moolenaar     str = read_string(fd, cnt);
8349ccfebddSBram Moolenaar     if (str == NULL)
8359ccfebddSBram Moolenaar 	*cntp = SP_OTHERERROR;
8369ccfebddSBram Moolenaar     return str;
8379ccfebddSBram Moolenaar }
8389ccfebddSBram Moolenaar 
8399ccfebddSBram Moolenaar /*
8409ccfebddSBram Moolenaar  * Read SN_REGION: <regionname> ...
8419ccfebddSBram Moolenaar  * Return SP_*ERROR flags.
8429ccfebddSBram Moolenaar  */
8439ccfebddSBram Moolenaar     static int
read_region_section(FILE * fd,slang_T * lp,int len)8449ccfebddSBram Moolenaar read_region_section(FILE *fd, slang_T *lp, int len)
8459ccfebddSBram Moolenaar {
8469ccfebddSBram Moolenaar     int		i;
8479ccfebddSBram Moolenaar 
8482993ac5fSBram Moolenaar     if (len > MAXREGIONS * 2)
8499ccfebddSBram Moolenaar 	return SP_FORMERROR;
8509ccfebddSBram Moolenaar     for (i = 0; i < len; ++i)
8510d6f5d97SBram Moolenaar 	lp->sl_regions[i] = getc(fd);			// <regionname>
8529ccfebddSBram Moolenaar     lp->sl_regions[len] = NUL;
8539ccfebddSBram Moolenaar     return 0;
8549ccfebddSBram Moolenaar }
8559ccfebddSBram Moolenaar 
8569ccfebddSBram Moolenaar /*
8579ccfebddSBram Moolenaar  * Read SN_CHARFLAGS section: <charflagslen> <charflags>
8589ccfebddSBram Moolenaar  *				<folcharslen> <folchars>
8599ccfebddSBram Moolenaar  * Return SP_*ERROR flags.
8609ccfebddSBram Moolenaar  */
8619ccfebddSBram Moolenaar     static int
read_charflags_section(FILE * fd)8629ccfebddSBram Moolenaar read_charflags_section(FILE *fd)
8639ccfebddSBram Moolenaar {
8649ccfebddSBram Moolenaar     char_u	*flags;
8659ccfebddSBram Moolenaar     char_u	*fol;
8669ccfebddSBram Moolenaar     int		flagslen, follen;
8679ccfebddSBram Moolenaar 
8680d6f5d97SBram Moolenaar     // <charflagslen> <charflags>
8699ccfebddSBram Moolenaar     flags = read_cnt_string(fd, 1, &flagslen);
8709ccfebddSBram Moolenaar     if (flagslen < 0)
8719ccfebddSBram Moolenaar 	return flagslen;
8729ccfebddSBram Moolenaar 
8730d6f5d97SBram Moolenaar     // <folcharslen> <folchars>
8749ccfebddSBram Moolenaar     fol = read_cnt_string(fd, 2, &follen);
8759ccfebddSBram Moolenaar     if (follen < 0)
8769ccfebddSBram Moolenaar     {
8779ccfebddSBram Moolenaar 	vim_free(flags);
8789ccfebddSBram Moolenaar 	return follen;
8799ccfebddSBram Moolenaar     }
8809ccfebddSBram Moolenaar 
8810d6f5d97SBram Moolenaar     // Set the word-char flags and fill SPELL_ISUPPER() table.
8829ccfebddSBram Moolenaar     if (flags != NULL && fol != NULL)
8839ccfebddSBram Moolenaar 	set_spell_charflags(flags, flagslen, fol);
8849ccfebddSBram Moolenaar 
8859ccfebddSBram Moolenaar     vim_free(flags);
8869ccfebddSBram Moolenaar     vim_free(fol);
8879ccfebddSBram Moolenaar 
8880d6f5d97SBram Moolenaar     // When <charflagslen> is zero then <fcharlen> must also be zero.
8899ccfebddSBram Moolenaar     if ((flags == NULL) != (fol == NULL))
8909ccfebddSBram Moolenaar 	return SP_FORMERROR;
8919ccfebddSBram Moolenaar     return 0;
8929ccfebddSBram Moolenaar }
8939ccfebddSBram Moolenaar 
8949ccfebddSBram Moolenaar /*
8959ccfebddSBram Moolenaar  * Read SN_PREFCOND section.
8969ccfebddSBram Moolenaar  * Return SP_*ERROR flags.
8979ccfebddSBram Moolenaar  */
8989ccfebddSBram Moolenaar     static int
read_prefcond_section(FILE * fd,slang_T * lp)8999ccfebddSBram Moolenaar read_prefcond_section(FILE *fd, slang_T *lp)
9009ccfebddSBram Moolenaar {
9019ccfebddSBram Moolenaar     int		cnt;
9029ccfebddSBram Moolenaar     int		i;
9039ccfebddSBram Moolenaar     int		n;
9049ccfebddSBram Moolenaar     char_u	*p;
9059ccfebddSBram Moolenaar     char_u	buf[MAXWLEN + 1];
9069ccfebddSBram Moolenaar 
9070d6f5d97SBram Moolenaar     // <prefcondcnt> <prefcond> ...
9080d6f5d97SBram Moolenaar     cnt = get2c(fd);					// <prefcondcnt>
9099ccfebddSBram Moolenaar     if (cnt <= 0)
9109ccfebddSBram Moolenaar 	return SP_FORMERROR;
9119ccfebddSBram Moolenaar 
912c799fe20SBram Moolenaar     lp->sl_prefprog = ALLOC_CLEAR_MULT(regprog_T *, cnt);
9139ccfebddSBram Moolenaar     if (lp->sl_prefprog == NULL)
9149ccfebddSBram Moolenaar 	return SP_OTHERERROR;
9159ccfebddSBram Moolenaar     lp->sl_prefixcnt = cnt;
9169ccfebddSBram Moolenaar 
9179ccfebddSBram Moolenaar     for (i = 0; i < cnt; ++i)
9189ccfebddSBram Moolenaar     {
9190d6f5d97SBram Moolenaar 	// <prefcond> : <condlen> <condstr>
9200d6f5d97SBram Moolenaar 	n = getc(fd);					// <condlen>
9219ccfebddSBram Moolenaar 	if (n < 0 || n >= MAXWLEN)
9229ccfebddSBram Moolenaar 	    return SP_FORMERROR;
9239ccfebddSBram Moolenaar 
9240d6f5d97SBram Moolenaar 	// When <condlen> is zero we have an empty condition.  Otherwise
9250d6f5d97SBram Moolenaar 	// compile the regexp program used to check for the condition.
9269ccfebddSBram Moolenaar 	if (n > 0)
9279ccfebddSBram Moolenaar 	{
9280d6f5d97SBram Moolenaar 	    buf[0] = '^';	    // always match at one position only
9299ccfebddSBram Moolenaar 	    p = buf + 1;
9309ccfebddSBram Moolenaar 	    while (n-- > 0)
9310d6f5d97SBram Moolenaar 		*p++ = getc(fd);			// <condstr>
9329ccfebddSBram Moolenaar 	    *p = NUL;
9339ccfebddSBram Moolenaar 	    lp->sl_prefprog[i] = vim_regcomp(buf, RE_MAGIC + RE_STRING);
9349ccfebddSBram Moolenaar 	}
9359ccfebddSBram Moolenaar     }
9369ccfebddSBram Moolenaar     return 0;
9379ccfebddSBram Moolenaar }
9389ccfebddSBram Moolenaar 
9399ccfebddSBram Moolenaar /*
9409ccfebddSBram Moolenaar  * Read REP or REPSAL items section from "fd": <repcount> <rep> ...
9419ccfebddSBram Moolenaar  * Return SP_*ERROR flags.
9429ccfebddSBram Moolenaar  */
9439ccfebddSBram Moolenaar     static int
read_rep_section(FILE * fd,garray_T * gap,short * first)9449ccfebddSBram Moolenaar read_rep_section(FILE *fd, garray_T *gap, short *first)
9459ccfebddSBram Moolenaar {
9469ccfebddSBram Moolenaar     int		cnt;
9479ccfebddSBram Moolenaar     fromto_T	*ftp;
9489ccfebddSBram Moolenaar     int		i;
9499ccfebddSBram Moolenaar 
9500d6f5d97SBram Moolenaar     cnt = get2c(fd);					// <repcount>
9519ccfebddSBram Moolenaar     if (cnt < 0)
9529ccfebddSBram Moolenaar 	return SP_TRUNCERROR;
9539ccfebddSBram Moolenaar 
9549ccfebddSBram Moolenaar     if (ga_grow(gap, cnt) == FAIL)
9559ccfebddSBram Moolenaar 	return SP_OTHERERROR;
9569ccfebddSBram Moolenaar 
9570d6f5d97SBram Moolenaar     // <rep> : <repfromlen> <repfrom> <reptolen> <repto>
9589ccfebddSBram Moolenaar     for (; gap->ga_len < cnt; ++gap->ga_len)
9599ccfebddSBram Moolenaar     {
9609ccfebddSBram Moolenaar 	ftp = &((fromto_T *)gap->ga_data)[gap->ga_len];
9619ccfebddSBram Moolenaar 	ftp->ft_from = read_cnt_string(fd, 1, &i);
9629ccfebddSBram Moolenaar 	if (i < 0)
9639ccfebddSBram Moolenaar 	    return i;
9649ccfebddSBram Moolenaar 	if (i == 0)
9659ccfebddSBram Moolenaar 	    return SP_FORMERROR;
9669ccfebddSBram Moolenaar 	ftp->ft_to = read_cnt_string(fd, 1, &i);
9679ccfebddSBram Moolenaar 	if (i <= 0)
9689ccfebddSBram Moolenaar 	{
9699ccfebddSBram Moolenaar 	    vim_free(ftp->ft_from);
9709ccfebddSBram Moolenaar 	    if (i < 0)
9719ccfebddSBram Moolenaar 		return i;
9729ccfebddSBram Moolenaar 	    return SP_FORMERROR;
9739ccfebddSBram Moolenaar 	}
9749ccfebddSBram Moolenaar     }
9759ccfebddSBram Moolenaar 
9760d6f5d97SBram Moolenaar     // Fill the first-index table.
9779ccfebddSBram Moolenaar     for (i = 0; i < 256; ++i)
9789ccfebddSBram Moolenaar 	first[i] = -1;
9799ccfebddSBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
9809ccfebddSBram Moolenaar     {
9819ccfebddSBram Moolenaar 	ftp = &((fromto_T *)gap->ga_data)[i];
9829ccfebddSBram Moolenaar 	if (first[*ftp->ft_from] == -1)
9839ccfebddSBram Moolenaar 	    first[*ftp->ft_from] = i;
9849ccfebddSBram Moolenaar     }
9859ccfebddSBram Moolenaar     return 0;
9869ccfebddSBram Moolenaar }
9879ccfebddSBram Moolenaar 
9889ccfebddSBram Moolenaar /*
9899ccfebddSBram Moolenaar  * Read SN_SAL section: <salflags> <salcount> <sal> ...
9909ccfebddSBram Moolenaar  * Return SP_*ERROR flags.
9919ccfebddSBram Moolenaar  */
9929ccfebddSBram Moolenaar     static int
read_sal_section(FILE * fd,slang_T * slang)9939ccfebddSBram Moolenaar read_sal_section(FILE *fd, slang_T *slang)
9949ccfebddSBram Moolenaar {
9959ccfebddSBram Moolenaar     int		i;
9969ccfebddSBram Moolenaar     int		cnt;
9979ccfebddSBram Moolenaar     garray_T	*gap;
9989ccfebddSBram Moolenaar     salitem_T	*smp;
9999ccfebddSBram Moolenaar     int		ccnt;
10009ccfebddSBram Moolenaar     char_u	*p;
10019ccfebddSBram Moolenaar 
10029ccfebddSBram Moolenaar     slang->sl_sofo = FALSE;
10039ccfebddSBram Moolenaar 
10040d6f5d97SBram Moolenaar     i = getc(fd);				// <salflags>
10059ccfebddSBram Moolenaar     if (i & SAL_F0LLOWUP)
10069ccfebddSBram Moolenaar 	slang->sl_followup = TRUE;
10079ccfebddSBram Moolenaar     if (i & SAL_COLLAPSE)
10089ccfebddSBram Moolenaar 	slang->sl_collapse = TRUE;
10099ccfebddSBram Moolenaar     if (i & SAL_REM_ACCENTS)
10109ccfebddSBram Moolenaar 	slang->sl_rem_accents = TRUE;
10119ccfebddSBram Moolenaar 
10120d6f5d97SBram Moolenaar     cnt = get2c(fd);				// <salcount>
10139ccfebddSBram Moolenaar     if (cnt < 0)
10149ccfebddSBram Moolenaar 	return SP_TRUNCERROR;
10159ccfebddSBram Moolenaar 
10169ccfebddSBram Moolenaar     gap = &slang->sl_sal;
10179ccfebddSBram Moolenaar     ga_init2(gap, sizeof(salitem_T), 10);
10189ccfebddSBram Moolenaar     if (ga_grow(gap, cnt + 1) == FAIL)
10199ccfebddSBram Moolenaar 	return SP_OTHERERROR;
10209ccfebddSBram Moolenaar 
10210d6f5d97SBram Moolenaar     // <sal> : <salfromlen> <salfrom> <saltolen> <salto>
10229ccfebddSBram Moolenaar     for (; gap->ga_len < cnt; ++gap->ga_len)
10239ccfebddSBram Moolenaar     {
102497d2f34cSBram Moolenaar 	int	c = NUL;
102597d2f34cSBram Moolenaar 
10269ccfebddSBram Moolenaar 	smp = &((salitem_T *)gap->ga_data)[gap->ga_len];
10270d6f5d97SBram Moolenaar 	ccnt = getc(fd);			// <salfromlen>
10289ccfebddSBram Moolenaar 	if (ccnt < 0)
10299ccfebddSBram Moolenaar 	    return SP_TRUNCERROR;
10309ccfebddSBram Moolenaar 	if ((p = alloc(ccnt + 2)) == NULL)
10319ccfebddSBram Moolenaar 	    return SP_OTHERERROR;
10329ccfebddSBram Moolenaar 	smp->sm_lead = p;
10339ccfebddSBram Moolenaar 
10340d6f5d97SBram Moolenaar 	// Read up to the first special char into sm_lead.
10359ccfebddSBram Moolenaar 	for (i = 0; i < ccnt; ++i)
10369ccfebddSBram Moolenaar 	{
10370d6f5d97SBram Moolenaar 	    c = getc(fd);			// <salfrom>
10389ccfebddSBram Moolenaar 	    if (vim_strchr((char_u *)"0123456789(-<^$", c) != NULL)
10399ccfebddSBram Moolenaar 		break;
10409ccfebddSBram Moolenaar 	    *p++ = c;
10419ccfebddSBram Moolenaar 	}
10429ccfebddSBram Moolenaar 	smp->sm_leadlen = (int)(p - smp->sm_lead);
10439ccfebddSBram Moolenaar 	*p++ = NUL;
10449ccfebddSBram Moolenaar 
10450d6f5d97SBram Moolenaar 	// Put (abc) chars in sm_oneof, if any.
10469ccfebddSBram Moolenaar 	if (c == '(')
10479ccfebddSBram Moolenaar 	{
10489ccfebddSBram Moolenaar 	    smp->sm_oneof = p;
10499ccfebddSBram Moolenaar 	    for (++i; i < ccnt; ++i)
10509ccfebddSBram Moolenaar 	    {
10510d6f5d97SBram Moolenaar 		c = getc(fd);			// <salfrom>
10529ccfebddSBram Moolenaar 		if (c == ')')
10539ccfebddSBram Moolenaar 		    break;
10549ccfebddSBram Moolenaar 		*p++ = c;
10559ccfebddSBram Moolenaar 	    }
10569ccfebddSBram Moolenaar 	    *p++ = NUL;
10579ccfebddSBram Moolenaar 	    if (++i < ccnt)
10589ccfebddSBram Moolenaar 		c = getc(fd);
10599ccfebddSBram Moolenaar 	}
10609ccfebddSBram Moolenaar 	else
10619ccfebddSBram Moolenaar 	    smp->sm_oneof = NULL;
10629ccfebddSBram Moolenaar 
10630d6f5d97SBram Moolenaar 	// Any following chars go in sm_rules.
10649ccfebddSBram Moolenaar 	smp->sm_rules = p;
10659ccfebddSBram Moolenaar 	if (i < ccnt)
10660d6f5d97SBram Moolenaar 	    // store the char we got while checking for end of sm_lead
10679ccfebddSBram Moolenaar 	    *p++ = c;
10689ccfebddSBram Moolenaar 	for (++i; i < ccnt; ++i)
10690d6f5d97SBram Moolenaar 	    *p++ = getc(fd);			// <salfrom>
10709ccfebddSBram Moolenaar 	*p++ = NUL;
10719ccfebddSBram Moolenaar 
10720d6f5d97SBram Moolenaar 	// <saltolen> <salto>
10739ccfebddSBram Moolenaar 	smp->sm_to = read_cnt_string(fd, 1, &ccnt);
10749ccfebddSBram Moolenaar 	if (ccnt < 0)
10759ccfebddSBram Moolenaar 	{
10769ccfebddSBram Moolenaar 	    vim_free(smp->sm_lead);
10779ccfebddSBram Moolenaar 	    return ccnt;
10789ccfebddSBram Moolenaar 	}
10799ccfebddSBram Moolenaar 
10809ccfebddSBram Moolenaar 	if (has_mbyte)
10819ccfebddSBram Moolenaar 	{
10820d6f5d97SBram Moolenaar 	    // convert the multi-byte strings to wide char strings
10839ccfebddSBram Moolenaar 	    smp->sm_lead_w = mb_str2wide(smp->sm_lead);
10849ccfebddSBram Moolenaar 	    smp->sm_leadlen = mb_charlen(smp->sm_lead);
10859ccfebddSBram Moolenaar 	    if (smp->sm_oneof == NULL)
10869ccfebddSBram Moolenaar 		smp->sm_oneof_w = NULL;
10879ccfebddSBram Moolenaar 	    else
10889ccfebddSBram Moolenaar 		smp->sm_oneof_w = mb_str2wide(smp->sm_oneof);
10899ccfebddSBram Moolenaar 	    if (smp->sm_to == NULL)
10909ccfebddSBram Moolenaar 		smp->sm_to_w = NULL;
10919ccfebddSBram Moolenaar 	    else
10929ccfebddSBram Moolenaar 		smp->sm_to_w = mb_str2wide(smp->sm_to);
10939ccfebddSBram Moolenaar 	    if (smp->sm_lead_w == NULL
10949ccfebddSBram Moolenaar 		    || (smp->sm_oneof_w == NULL && smp->sm_oneof != NULL)
10959ccfebddSBram Moolenaar 		    || (smp->sm_to_w == NULL && smp->sm_to != NULL))
10969ccfebddSBram Moolenaar 	    {
10979ccfebddSBram Moolenaar 		vim_free(smp->sm_lead);
10989ccfebddSBram Moolenaar 		vim_free(smp->sm_to);
10999ccfebddSBram Moolenaar 		vim_free(smp->sm_lead_w);
11009ccfebddSBram Moolenaar 		vim_free(smp->sm_oneof_w);
11019ccfebddSBram Moolenaar 		vim_free(smp->sm_to_w);
11029ccfebddSBram Moolenaar 		return SP_OTHERERROR;
11039ccfebddSBram Moolenaar 	    }
11049ccfebddSBram Moolenaar 	}
11059ccfebddSBram Moolenaar     }
11069ccfebddSBram Moolenaar 
11079ccfebddSBram Moolenaar     if (gap->ga_len > 0)
11089ccfebddSBram Moolenaar     {
11090d6f5d97SBram Moolenaar 	// Add one extra entry to mark the end with an empty sm_lead.  Avoids
11100d6f5d97SBram Moolenaar 	// that we need to check the index every time.
11119ccfebddSBram Moolenaar 	smp = &((salitem_T *)gap->ga_data)[gap->ga_len];
11129ccfebddSBram Moolenaar 	if ((p = alloc(1)) == NULL)
11139ccfebddSBram Moolenaar 	    return SP_OTHERERROR;
11149ccfebddSBram Moolenaar 	p[0] = NUL;
11159ccfebddSBram Moolenaar 	smp->sm_lead = p;
11169ccfebddSBram Moolenaar 	smp->sm_leadlen = 0;
11179ccfebddSBram Moolenaar 	smp->sm_oneof = NULL;
11189ccfebddSBram Moolenaar 	smp->sm_rules = p;
11199ccfebddSBram Moolenaar 	smp->sm_to = NULL;
11209ccfebddSBram Moolenaar 	if (has_mbyte)
11219ccfebddSBram Moolenaar 	{
11229ccfebddSBram Moolenaar 	    smp->sm_lead_w = mb_str2wide(smp->sm_lead);
11239ccfebddSBram Moolenaar 	    smp->sm_leadlen = 0;
11249ccfebddSBram Moolenaar 	    smp->sm_oneof_w = NULL;
11259ccfebddSBram Moolenaar 	    smp->sm_to_w = NULL;
11269ccfebddSBram Moolenaar 	}
11279ccfebddSBram Moolenaar 	++gap->ga_len;
11289ccfebddSBram Moolenaar     }
11299ccfebddSBram Moolenaar 
11300d6f5d97SBram Moolenaar     // Fill the first-index table.
11319ccfebddSBram Moolenaar     set_sal_first(slang);
11329ccfebddSBram Moolenaar 
11339ccfebddSBram Moolenaar     return 0;
11349ccfebddSBram Moolenaar }
11359ccfebddSBram Moolenaar 
11369ccfebddSBram Moolenaar /*
11379ccfebddSBram Moolenaar  * Read SN_WORDS: <word> ...
11389ccfebddSBram Moolenaar  * Return SP_*ERROR flags.
11399ccfebddSBram Moolenaar  */
11409ccfebddSBram Moolenaar     static int
read_words_section(FILE * fd,slang_T * lp,int len)11419ccfebddSBram Moolenaar read_words_section(FILE *fd, slang_T *lp, int len)
11429ccfebddSBram Moolenaar {
11439ccfebddSBram Moolenaar     int		done = 0;
11449ccfebddSBram Moolenaar     int		i;
11459ccfebddSBram Moolenaar     int		c;
11469ccfebddSBram Moolenaar     char_u	word[MAXWLEN];
11479ccfebddSBram Moolenaar 
11489ccfebddSBram Moolenaar     while (done < len)
11499ccfebddSBram Moolenaar     {
11500d6f5d97SBram Moolenaar 	// Read one word at a time.
11519ccfebddSBram Moolenaar 	for (i = 0; ; ++i)
11529ccfebddSBram Moolenaar 	{
11539ccfebddSBram Moolenaar 	    c = getc(fd);
11549ccfebddSBram Moolenaar 	    if (c == EOF)
11559ccfebddSBram Moolenaar 		return SP_TRUNCERROR;
11569ccfebddSBram Moolenaar 	    word[i] = c;
11579ccfebddSBram Moolenaar 	    if (word[i] == NUL)
11589ccfebddSBram Moolenaar 		break;
11599ccfebddSBram Moolenaar 	    if (i == MAXWLEN - 1)
11609ccfebddSBram Moolenaar 		return SP_FORMERROR;
11619ccfebddSBram Moolenaar 	}
11629ccfebddSBram Moolenaar 
11630d6f5d97SBram Moolenaar 	// Init the count to 10.
11649ccfebddSBram Moolenaar 	count_common_word(lp, word, -1, 10);
11659ccfebddSBram Moolenaar 	done += i + 1;
11669ccfebddSBram Moolenaar     }
11679ccfebddSBram Moolenaar     return 0;
11689ccfebddSBram Moolenaar }
11699ccfebddSBram Moolenaar 
11709ccfebddSBram Moolenaar /*
11719ccfebddSBram Moolenaar  * SN_SOFO: <sofofromlen> <sofofrom> <sofotolen> <sofoto>
11729ccfebddSBram Moolenaar  * Return SP_*ERROR flags.
11739ccfebddSBram Moolenaar  */
11749ccfebddSBram Moolenaar     static int
read_sofo_section(FILE * fd,slang_T * slang)11759ccfebddSBram Moolenaar read_sofo_section(FILE *fd, slang_T *slang)
11769ccfebddSBram Moolenaar {
11779ccfebddSBram Moolenaar     int		cnt;
11789ccfebddSBram Moolenaar     char_u	*from, *to;
11799ccfebddSBram Moolenaar     int		res;
11809ccfebddSBram Moolenaar 
11819ccfebddSBram Moolenaar     slang->sl_sofo = TRUE;
11829ccfebddSBram Moolenaar 
11830d6f5d97SBram Moolenaar     // <sofofromlen> <sofofrom>
11849ccfebddSBram Moolenaar     from = read_cnt_string(fd, 2, &cnt);
11859ccfebddSBram Moolenaar     if (cnt < 0)
11869ccfebddSBram Moolenaar 	return cnt;
11879ccfebddSBram Moolenaar 
11880d6f5d97SBram Moolenaar     // <sofotolen> <sofoto>
11899ccfebddSBram Moolenaar     to = read_cnt_string(fd, 2, &cnt);
11909ccfebddSBram Moolenaar     if (cnt < 0)
11919ccfebddSBram Moolenaar     {
11929ccfebddSBram Moolenaar 	vim_free(from);
11939ccfebddSBram Moolenaar 	return cnt;
11949ccfebddSBram Moolenaar     }
11959ccfebddSBram Moolenaar 
11960d6f5d97SBram Moolenaar     // Store the info in slang->sl_sal and/or slang->sl_sal_first.
11979ccfebddSBram Moolenaar     if (from != NULL && to != NULL)
11989ccfebddSBram Moolenaar 	res = set_sofo(slang, from, to);
11999ccfebddSBram Moolenaar     else if (from != NULL || to != NULL)
12000d6f5d97SBram Moolenaar 	res = SP_FORMERROR;    // only one of two strings is an error
12019ccfebddSBram Moolenaar     else
12029ccfebddSBram Moolenaar 	res = 0;
12039ccfebddSBram Moolenaar 
12049ccfebddSBram Moolenaar     vim_free(from);
12059ccfebddSBram Moolenaar     vim_free(to);
12069ccfebddSBram Moolenaar     return res;
12079ccfebddSBram Moolenaar }
12089ccfebddSBram Moolenaar 
12099ccfebddSBram Moolenaar /*
12109ccfebddSBram Moolenaar  * Read the compound section from the .spl file:
12119ccfebddSBram Moolenaar  *	<compmax> <compminlen> <compsylmax> <compoptions> <compflags>
12129ccfebddSBram Moolenaar  * Returns SP_*ERROR flags.
12139ccfebddSBram Moolenaar  */
12149ccfebddSBram Moolenaar     static int
read_compound(FILE * fd,slang_T * slang,int len)12159ccfebddSBram Moolenaar read_compound(FILE *fd, slang_T *slang, int len)
12169ccfebddSBram Moolenaar {
12179ccfebddSBram Moolenaar     int		todo = len;
12189ccfebddSBram Moolenaar     int		c;
12199ccfebddSBram Moolenaar     int		atstart;
12209ccfebddSBram Moolenaar     char_u	*pat;
12219ccfebddSBram Moolenaar     char_u	*pp;
12229ccfebddSBram Moolenaar     char_u	*cp;
12239ccfebddSBram Moolenaar     char_u	*ap;
12249ccfebddSBram Moolenaar     char_u	*crp;
12259ccfebddSBram Moolenaar     int		cnt;
12269ccfebddSBram Moolenaar     garray_T	*gap;
12279ccfebddSBram Moolenaar 
12289ccfebddSBram Moolenaar     if (todo < 2)
12290d6f5d97SBram Moolenaar 	return SP_FORMERROR;	// need at least two bytes
12309ccfebddSBram Moolenaar 
12319ccfebddSBram Moolenaar     --todo;
12320d6f5d97SBram Moolenaar     c = getc(fd);					// <compmax>
12339ccfebddSBram Moolenaar     if (c < 2)
12349ccfebddSBram Moolenaar 	c = MAXWLEN;
12359ccfebddSBram Moolenaar     slang->sl_compmax = c;
12369ccfebddSBram Moolenaar 
12379ccfebddSBram Moolenaar     --todo;
12380d6f5d97SBram Moolenaar     c = getc(fd);					// <compminlen>
12399ccfebddSBram Moolenaar     if (c < 1)
12409ccfebddSBram Moolenaar 	c = 0;
12419ccfebddSBram Moolenaar     slang->sl_compminlen = c;
12429ccfebddSBram Moolenaar 
12439ccfebddSBram Moolenaar     --todo;
12440d6f5d97SBram Moolenaar     c = getc(fd);					// <compsylmax>
12459ccfebddSBram Moolenaar     if (c < 1)
12469ccfebddSBram Moolenaar 	c = MAXWLEN;
12479ccfebddSBram Moolenaar     slang->sl_compsylmax = c;
12489ccfebddSBram Moolenaar 
12490d6f5d97SBram Moolenaar     c = getc(fd);					// <compoptions>
12509ccfebddSBram Moolenaar     if (c != 0)
12510d6f5d97SBram Moolenaar 	ungetc(c, fd);	    // be backwards compatible with Vim 7.0b
12529ccfebddSBram Moolenaar     else
12539ccfebddSBram Moolenaar     {
12549ccfebddSBram Moolenaar 	--todo;
12550d6f5d97SBram Moolenaar 	c = getc(fd);	    // only use the lower byte for now
12569ccfebddSBram Moolenaar 	--todo;
12579ccfebddSBram Moolenaar 	slang->sl_compoptions = c;
12589ccfebddSBram Moolenaar 
12599ccfebddSBram Moolenaar 	gap = &slang->sl_comppat;
12600d6f5d97SBram Moolenaar 	c = get2c(fd);					// <comppatcount>
1261*b85d3627SBram Moolenaar 	if (c < 0)
1262*b85d3627SBram Moolenaar 	    return SP_TRUNCERROR;
12639ccfebddSBram Moolenaar 	todo -= 2;
12649ccfebddSBram Moolenaar 	ga_init2(gap, sizeof(char_u *), c);
12659ccfebddSBram Moolenaar 	if (ga_grow(gap, c) == OK)
12669ccfebddSBram Moolenaar 	    while (--c >= 0)
12679ccfebddSBram Moolenaar 	    {
12689ccfebddSBram Moolenaar 		((char_u **)(gap->ga_data))[gap->ga_len++] =
12699ccfebddSBram Moolenaar 						  read_cnt_string(fd, 1, &cnt);
12700d6f5d97SBram Moolenaar 					    // <comppatlen> <comppattext>
12719ccfebddSBram Moolenaar 		if (cnt < 0)
12729ccfebddSBram Moolenaar 		    return cnt;
12739ccfebddSBram Moolenaar 		todo -= cnt + 1;
12749ccfebddSBram Moolenaar 	    }
12759ccfebddSBram Moolenaar     }
12769ccfebddSBram Moolenaar     if (todo < 0)
12779ccfebddSBram Moolenaar 	return SP_FORMERROR;
12789ccfebddSBram Moolenaar 
12790d6f5d97SBram Moolenaar     // Turn the COMPOUNDRULE items into a regexp pattern:
12800d6f5d97SBram Moolenaar     // "a[bc]/a*b+" -> "^\(a[bc]\|a*b\+\)$".
12810d6f5d97SBram Moolenaar     // Inserting backslashes may double the length, "^\(\)$<Nul>" is 7 bytes.
12820d6f5d97SBram Moolenaar     // Conversion to utf-8 may double the size.
12839ccfebddSBram Moolenaar     c = todo * 2 + 7;
12849ccfebddSBram Moolenaar     if (enc_utf8)
12859ccfebddSBram Moolenaar 	c += todo * 2;
1286964b3746SBram Moolenaar     pat = alloc(c);
12879ccfebddSBram Moolenaar     if (pat == NULL)
12889ccfebddSBram Moolenaar 	return SP_OTHERERROR;
12899ccfebddSBram Moolenaar 
12900d6f5d97SBram Moolenaar     // We also need a list of all flags that can appear at the start and one
12910d6f5d97SBram Moolenaar     // for all flags.
12929ccfebddSBram Moolenaar     cp = alloc(todo + 1);
12939ccfebddSBram Moolenaar     if (cp == NULL)
12949ccfebddSBram Moolenaar     {
12959ccfebddSBram Moolenaar 	vim_free(pat);
12969ccfebddSBram Moolenaar 	return SP_OTHERERROR;
12979ccfebddSBram Moolenaar     }
12989ccfebddSBram Moolenaar     slang->sl_compstartflags = cp;
12999ccfebddSBram Moolenaar     *cp = NUL;
13009ccfebddSBram Moolenaar 
13019ccfebddSBram Moolenaar     ap = alloc(todo + 1);
13029ccfebddSBram Moolenaar     if (ap == NULL)
13039ccfebddSBram Moolenaar     {
13049ccfebddSBram Moolenaar 	vim_free(pat);
13059ccfebddSBram Moolenaar 	return SP_OTHERERROR;
13069ccfebddSBram Moolenaar     }
13079ccfebddSBram Moolenaar     slang->sl_compallflags = ap;
13089ccfebddSBram Moolenaar     *ap = NUL;
13099ccfebddSBram Moolenaar 
13100d6f5d97SBram Moolenaar     // And a list of all patterns in their original form, for checking whether
13110d6f5d97SBram Moolenaar     // compounding may work in match_compoundrule().  This is freed when we
13120d6f5d97SBram Moolenaar     // encounter a wildcard, the check doesn't work then.
13139ccfebddSBram Moolenaar     crp = alloc(todo + 1);
13149ccfebddSBram Moolenaar     slang->sl_comprules = crp;
13159ccfebddSBram Moolenaar 
13169ccfebddSBram Moolenaar     pp = pat;
13179ccfebddSBram Moolenaar     *pp++ = '^';
13189ccfebddSBram Moolenaar     *pp++ = '\\';
13199ccfebddSBram Moolenaar     *pp++ = '(';
13209ccfebddSBram Moolenaar 
13219ccfebddSBram Moolenaar     atstart = 1;
13229ccfebddSBram Moolenaar     while (todo-- > 0)
13239ccfebddSBram Moolenaar     {
13240d6f5d97SBram Moolenaar 	c = getc(fd);					// <compflags>
13259ccfebddSBram Moolenaar 	if (c == EOF)
13269ccfebddSBram Moolenaar 	{
13279ccfebddSBram Moolenaar 	    vim_free(pat);
13289ccfebddSBram Moolenaar 	    return SP_TRUNCERROR;
13299ccfebddSBram Moolenaar 	}
13309ccfebddSBram Moolenaar 
13310d6f5d97SBram Moolenaar 	// Add all flags to "sl_compallflags".
13329ccfebddSBram Moolenaar 	if (vim_strchr((char_u *)"?*+[]/", c) == NULL
13339ccfebddSBram Moolenaar 		&& !byte_in_str(slang->sl_compallflags, c))
13349ccfebddSBram Moolenaar 	{
13359ccfebddSBram Moolenaar 	    *ap++ = c;
13369ccfebddSBram Moolenaar 	    *ap = NUL;
13379ccfebddSBram Moolenaar 	}
13389ccfebddSBram Moolenaar 
13399ccfebddSBram Moolenaar 	if (atstart != 0)
13409ccfebddSBram Moolenaar 	{
13410d6f5d97SBram Moolenaar 	    // At start of item: copy flags to "sl_compstartflags".  For a
13420d6f5d97SBram Moolenaar 	    // [abc] item set "atstart" to 2 and copy up to the ']'.
13439ccfebddSBram Moolenaar 	    if (c == '[')
13449ccfebddSBram Moolenaar 		atstart = 2;
13459ccfebddSBram Moolenaar 	    else if (c == ']')
13469ccfebddSBram Moolenaar 		atstart = 0;
13479ccfebddSBram Moolenaar 	    else
13489ccfebddSBram Moolenaar 	    {
13499ccfebddSBram Moolenaar 		if (!byte_in_str(slang->sl_compstartflags, c))
13509ccfebddSBram Moolenaar 		{
13519ccfebddSBram Moolenaar 		    *cp++ = c;
13529ccfebddSBram Moolenaar 		    *cp = NUL;
13539ccfebddSBram Moolenaar 		}
13549ccfebddSBram Moolenaar 		if (atstart == 1)
13559ccfebddSBram Moolenaar 		    atstart = 0;
13569ccfebddSBram Moolenaar 	    }
13579ccfebddSBram Moolenaar 	}
13589ccfebddSBram Moolenaar 
13590d6f5d97SBram Moolenaar 	// Copy flag to "sl_comprules", unless we run into a wildcard.
13609ccfebddSBram Moolenaar 	if (crp != NULL)
13619ccfebddSBram Moolenaar 	{
13629ccfebddSBram Moolenaar 	    if (c == '?' || c == '+' || c == '*')
13639ccfebddSBram Moolenaar 	    {
1364d23a8236SBram Moolenaar 		VIM_CLEAR(slang->sl_comprules);
13659ccfebddSBram Moolenaar 		crp = NULL;
13669ccfebddSBram Moolenaar 	    }
13679ccfebddSBram Moolenaar 	    else
13689ccfebddSBram Moolenaar 		*crp++ = c;
13699ccfebddSBram Moolenaar 	}
13709ccfebddSBram Moolenaar 
13710d6f5d97SBram Moolenaar 	if (c == '/')	    // slash separates two items
13729ccfebddSBram Moolenaar 	{
13739ccfebddSBram Moolenaar 	    *pp++ = '\\';
13749ccfebddSBram Moolenaar 	    *pp++ = '|';
13759ccfebddSBram Moolenaar 	    atstart = 1;
13769ccfebddSBram Moolenaar 	}
13770d6f5d97SBram Moolenaar 	else		    // normal char, "[abc]" and '*' are copied as-is
13789ccfebddSBram Moolenaar 	{
13799ccfebddSBram Moolenaar 	    if (c == '?' || c == '+' || c == '~')
13800d6f5d97SBram Moolenaar 		*pp++ = '\\';	    // "a?" becomes "a\?", "a+" becomes "a\+"
13819ccfebddSBram Moolenaar 	    if (enc_utf8)
13829ccfebddSBram Moolenaar 		pp += mb_char2bytes(c, pp);
13839ccfebddSBram Moolenaar 	    else
13849ccfebddSBram Moolenaar 		*pp++ = c;
13859ccfebddSBram Moolenaar 	}
13869ccfebddSBram Moolenaar     }
13879ccfebddSBram Moolenaar 
13889ccfebddSBram Moolenaar     *pp++ = '\\';
13899ccfebddSBram Moolenaar     *pp++ = ')';
13909ccfebddSBram Moolenaar     *pp++ = '$';
13919ccfebddSBram Moolenaar     *pp = NUL;
13929ccfebddSBram Moolenaar 
13939ccfebddSBram Moolenaar     if (crp != NULL)
13949ccfebddSBram Moolenaar 	*crp = NUL;
13959ccfebddSBram Moolenaar 
13969ccfebddSBram Moolenaar     slang->sl_compprog = vim_regcomp(pat, RE_MAGIC + RE_STRING + RE_STRICT);
13979ccfebddSBram Moolenaar     vim_free(pat);
13989ccfebddSBram Moolenaar     if (slang->sl_compprog == NULL)
13999ccfebddSBram Moolenaar 	return SP_FORMERROR;
14009ccfebddSBram Moolenaar 
14019ccfebddSBram Moolenaar     return 0;
14029ccfebddSBram Moolenaar }
14039ccfebddSBram Moolenaar 
14049ccfebddSBram Moolenaar /*
14059ccfebddSBram Moolenaar  * Set the SOFOFROM and SOFOTO items in language "lp".
14069ccfebddSBram Moolenaar  * Returns SP_*ERROR flags when there is something wrong.
14079ccfebddSBram Moolenaar  */
14089ccfebddSBram Moolenaar     static int
set_sofo(slang_T * lp,char_u * from,char_u * to)14099ccfebddSBram Moolenaar set_sofo(slang_T *lp, char_u *from, char_u *to)
14109ccfebddSBram Moolenaar {
14119ccfebddSBram Moolenaar     int		i;
14129ccfebddSBram Moolenaar 
14139ccfebddSBram Moolenaar     garray_T	*gap;
14149ccfebddSBram Moolenaar     char_u	*s;
14159ccfebddSBram Moolenaar     char_u	*p;
14169ccfebddSBram Moolenaar     int		c;
14179ccfebddSBram Moolenaar     int		*inp;
14189ccfebddSBram Moolenaar 
14199ccfebddSBram Moolenaar     if (has_mbyte)
14209ccfebddSBram Moolenaar     {
14210d6f5d97SBram Moolenaar 	// Use "sl_sal" as an array with 256 pointers to a list of wide
14220d6f5d97SBram Moolenaar 	// characters.  The index is the low byte of the character.
14230d6f5d97SBram Moolenaar 	// The list contains from-to pairs with a terminating NUL.
14240d6f5d97SBram Moolenaar 	// sl_sal_first[] is used for latin1 "from" characters.
14259ccfebddSBram Moolenaar 	gap = &lp->sl_sal;
14269ccfebddSBram Moolenaar 	ga_init2(gap, sizeof(int *), 1);
14279ccfebddSBram Moolenaar 	if (ga_grow(gap, 256) == FAIL)
14289ccfebddSBram Moolenaar 	    return SP_OTHERERROR;
14299ccfebddSBram Moolenaar 	vim_memset(gap->ga_data, 0, sizeof(int *) * 256);
14309ccfebddSBram Moolenaar 	gap->ga_len = 256;
14319ccfebddSBram Moolenaar 
14320d6f5d97SBram Moolenaar 	// First count the number of items for each list.  Temporarily use
14330d6f5d97SBram Moolenaar 	// sl_sal_first[] for this.
14349ccfebddSBram Moolenaar 	for (p = from, s = to; *p != NUL && *s != NUL; )
14359ccfebddSBram Moolenaar 	{
14369ccfebddSBram Moolenaar 	    c = mb_cptr2char_adv(&p);
143791acfffcSBram Moolenaar 	    MB_CPTR_ADV(s);
14389ccfebddSBram Moolenaar 	    if (c >= 256)
14399ccfebddSBram Moolenaar 		++lp->sl_sal_first[c & 0xff];
14409ccfebddSBram Moolenaar 	}
14410d6f5d97SBram Moolenaar 	if (*p != NUL || *s != NUL)	    // lengths differ
14429ccfebddSBram Moolenaar 	    return SP_FORMERROR;
14439ccfebddSBram Moolenaar 
14440d6f5d97SBram Moolenaar 	// Allocate the lists.
14459ccfebddSBram Moolenaar 	for (i = 0; i < 256; ++i)
14469ccfebddSBram Moolenaar 	    if (lp->sl_sal_first[i] > 0)
14479ccfebddSBram Moolenaar 	    {
14489ccfebddSBram Moolenaar 		p = alloc(sizeof(int) * (lp->sl_sal_first[i] * 2 + 1));
14499ccfebddSBram Moolenaar 		if (p == NULL)
14509ccfebddSBram Moolenaar 		    return SP_OTHERERROR;
14519ccfebddSBram Moolenaar 		((int **)gap->ga_data)[i] = (int *)p;
14529ccfebddSBram Moolenaar 		*(int *)p = 0;
14539ccfebddSBram Moolenaar 	    }
14549ccfebddSBram Moolenaar 
14550d6f5d97SBram Moolenaar 	// Put the characters up to 255 in sl_sal_first[] the rest in a sl_sal
14560d6f5d97SBram Moolenaar 	// list.
14579ccfebddSBram Moolenaar 	vim_memset(lp->sl_sal_first, 0, sizeof(salfirst_T) * 256);
14589ccfebddSBram Moolenaar 	for (p = from, s = to; *p != NUL && *s != NUL; )
14599ccfebddSBram Moolenaar 	{
14609ccfebddSBram Moolenaar 	    c = mb_cptr2char_adv(&p);
14619ccfebddSBram Moolenaar 	    i = mb_cptr2char_adv(&s);
14629ccfebddSBram Moolenaar 	    if (c >= 256)
14639ccfebddSBram Moolenaar 	    {
14640d6f5d97SBram Moolenaar 		// Append the from-to chars at the end of the list with
14650d6f5d97SBram Moolenaar 		// the low byte.
14669ccfebddSBram Moolenaar 		inp = ((int **)gap->ga_data)[c & 0xff];
14679ccfebddSBram Moolenaar 		while (*inp != 0)
14689ccfebddSBram Moolenaar 		    ++inp;
14690d6f5d97SBram Moolenaar 		*inp++ = c;		// from char
14700d6f5d97SBram Moolenaar 		*inp++ = i;		// to char
14710d6f5d97SBram Moolenaar 		*inp++ = NUL;		// NUL at the end
14729ccfebddSBram Moolenaar 	    }
14739ccfebddSBram Moolenaar 	    else
14740d6f5d97SBram Moolenaar 		// mapping byte to char is done in sl_sal_first[]
14759ccfebddSBram Moolenaar 		lp->sl_sal_first[c] = i;
14769ccfebddSBram Moolenaar 	}
14779ccfebddSBram Moolenaar     }
14789ccfebddSBram Moolenaar     else
14799ccfebddSBram Moolenaar     {
14800d6f5d97SBram Moolenaar 	// mapping bytes to bytes is done in sl_sal_first[]
14819ccfebddSBram Moolenaar 	if (STRLEN(from) != STRLEN(to))
14829ccfebddSBram Moolenaar 	    return SP_FORMERROR;
14839ccfebddSBram Moolenaar 
14849ccfebddSBram Moolenaar 	for (i = 0; to[i] != NUL; ++i)
14859ccfebddSBram Moolenaar 	    lp->sl_sal_first[from[i]] = to[i];
14860d6f5d97SBram Moolenaar 	lp->sl_sal.ga_len = 1;		// indicates we have soundfolding
14879ccfebddSBram Moolenaar     }
14889ccfebddSBram Moolenaar 
14899ccfebddSBram Moolenaar     return 0;
14909ccfebddSBram Moolenaar }
14919ccfebddSBram Moolenaar 
14929ccfebddSBram Moolenaar /*
14939ccfebddSBram Moolenaar  * Fill the first-index table for "lp".
14949ccfebddSBram Moolenaar  */
14959ccfebddSBram Moolenaar     static void
set_sal_first(slang_T * lp)14969ccfebddSBram Moolenaar set_sal_first(slang_T *lp)
14979ccfebddSBram Moolenaar {
14989ccfebddSBram Moolenaar     salfirst_T	*sfirst;
14999ccfebddSBram Moolenaar     int		i;
15009ccfebddSBram Moolenaar     salitem_T	*smp;
15019ccfebddSBram Moolenaar     int		c;
15029ccfebddSBram Moolenaar     garray_T	*gap = &lp->sl_sal;
15039ccfebddSBram Moolenaar 
15049ccfebddSBram Moolenaar     sfirst = lp->sl_sal_first;
15059ccfebddSBram Moolenaar     for (i = 0; i < 256; ++i)
15069ccfebddSBram Moolenaar 	sfirst[i] = -1;
15079ccfebddSBram Moolenaar     smp = (salitem_T *)gap->ga_data;
15089ccfebddSBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
15099ccfebddSBram Moolenaar     {
15109ccfebddSBram Moolenaar 	if (has_mbyte)
15110d6f5d97SBram Moolenaar 	    // Use the lowest byte of the first character.  For latin1 it's
15120d6f5d97SBram Moolenaar 	    // the character, for other encodings it should differ for most
15130d6f5d97SBram Moolenaar 	    // characters.
15149ccfebddSBram Moolenaar 	    c = *smp[i].sm_lead_w & 0xff;
15159ccfebddSBram Moolenaar 	else
15169ccfebddSBram Moolenaar 	    c = *smp[i].sm_lead;
15179ccfebddSBram Moolenaar 	if (sfirst[c] == -1)
15189ccfebddSBram Moolenaar 	{
15199ccfebddSBram Moolenaar 	    sfirst[c] = i;
15209ccfebddSBram Moolenaar 	    if (has_mbyte)
15219ccfebddSBram Moolenaar 	    {
15229ccfebddSBram Moolenaar 		int		n;
15239ccfebddSBram Moolenaar 
15240d6f5d97SBram Moolenaar 		// Make sure all entries with this byte are following each
15250d6f5d97SBram Moolenaar 		// other.  Move the ones that are in the wrong position.  Do
15260d6f5d97SBram Moolenaar 		// keep the same ordering!
15279ccfebddSBram Moolenaar 		while (i + 1 < gap->ga_len
15289ccfebddSBram Moolenaar 				       && (*smp[i + 1].sm_lead_w & 0xff) == c)
15290d6f5d97SBram Moolenaar 		    // Skip over entry with same index byte.
15309ccfebddSBram Moolenaar 		    ++i;
15319ccfebddSBram Moolenaar 
15329ccfebddSBram Moolenaar 		for (n = 1; i + n < gap->ga_len; ++n)
15339ccfebddSBram Moolenaar 		    if ((*smp[i + n].sm_lead_w & 0xff) == c)
15349ccfebddSBram Moolenaar 		    {
15359ccfebddSBram Moolenaar 			salitem_T  tsal;
15369ccfebddSBram Moolenaar 
15370d6f5d97SBram Moolenaar 			// Move entry with same index byte after the entries
15380d6f5d97SBram Moolenaar 			// we already found.
15399ccfebddSBram Moolenaar 			++i;
15409ccfebddSBram Moolenaar 			--n;
15419ccfebddSBram Moolenaar 			tsal = smp[i + n];
15429ccfebddSBram Moolenaar 			mch_memmove(smp + i + 1, smp + i,
15439ccfebddSBram Moolenaar 						       sizeof(salitem_T) * n);
15449ccfebddSBram Moolenaar 			smp[i] = tsal;
15459ccfebddSBram Moolenaar 		    }
15469ccfebddSBram Moolenaar 	    }
15479ccfebddSBram Moolenaar 	}
15489ccfebddSBram Moolenaar     }
15499ccfebddSBram Moolenaar }
15509ccfebddSBram Moolenaar 
15519ccfebddSBram Moolenaar /*
15529ccfebddSBram Moolenaar  * Turn a multi-byte string into a wide character string.
15539ccfebddSBram Moolenaar  * Return it in allocated memory (NULL for out-of-memory)
15549ccfebddSBram Moolenaar  */
15559ccfebddSBram Moolenaar     static int *
mb_str2wide(char_u * s)15569ccfebddSBram Moolenaar mb_str2wide(char_u *s)
15579ccfebddSBram Moolenaar {
15589ccfebddSBram Moolenaar     int		*res;
15599ccfebddSBram Moolenaar     char_u	*p;
15609ccfebddSBram Moolenaar     int		i = 0;
15619ccfebddSBram Moolenaar 
1562c799fe20SBram Moolenaar     res = ALLOC_MULT(int, mb_charlen(s) + 1);
15639ccfebddSBram Moolenaar     if (res != NULL)
15649ccfebddSBram Moolenaar     {
15659ccfebddSBram Moolenaar 	for (p = s; *p != NUL; )
15669ccfebddSBram Moolenaar 	    res[i++] = mb_ptr2char_adv(&p);
15679ccfebddSBram Moolenaar 	res[i] = NUL;
15689ccfebddSBram Moolenaar     }
15699ccfebddSBram Moolenaar     return res;
15709ccfebddSBram Moolenaar }
15719ccfebddSBram Moolenaar 
15729ccfebddSBram Moolenaar /*
15739ccfebddSBram Moolenaar  * Read a tree from the .spl or .sug file.
15749ccfebddSBram Moolenaar  * Allocates the memory and stores pointers in "bytsp" and "idxsp".
15759ccfebddSBram Moolenaar  * This is skipped when the tree has zero length.
15769ccfebddSBram Moolenaar  * Returns zero when OK, SP_ value for an error.
15779ccfebddSBram Moolenaar  */
15789ccfebddSBram Moolenaar     static int
spell_read_tree(FILE * fd,char_u ** bytsp,long * bytsp_len,idx_T ** idxsp,int prefixtree,int prefixcnt)15799ccfebddSBram Moolenaar spell_read_tree(
15809ccfebddSBram Moolenaar     FILE	*fd,
15819ccfebddSBram Moolenaar     char_u	**bytsp,
158207399e7fSBram Moolenaar     long	*bytsp_len,
15839ccfebddSBram Moolenaar     idx_T	**idxsp,
15840d6f5d97SBram Moolenaar     int		prefixtree,	// TRUE for the prefix tree
15850d6f5d97SBram Moolenaar     int		prefixcnt)	// when "prefixtree" is TRUE: prefix count
15869ccfebddSBram Moolenaar {
15876d3c8586SBram Moolenaar     long	len;
15889ccfebddSBram Moolenaar     int		idx;
15899ccfebddSBram Moolenaar     char_u	*bp;
15909ccfebddSBram Moolenaar     idx_T	*ip;
15919ccfebddSBram Moolenaar 
15920d6f5d97SBram Moolenaar     // The tree size was computed when writing the file, so that we can
15930d6f5d97SBram Moolenaar     // allocate it as one long block. <nodecount>
15949ccfebddSBram Moolenaar     len = get4c(fd);
15959ccfebddSBram Moolenaar     if (len < 0)
15969ccfebddSBram Moolenaar 	return SP_TRUNCERROR;
15976d3c8586SBram Moolenaar     if (len >= LONG_MAX / (long)sizeof(int))
15980d6f5d97SBram Moolenaar 	// Invalid length, multiply with sizeof(int) would overflow.
1599399c297aSBram Moolenaar 	return SP_FORMERROR;
16009ccfebddSBram Moolenaar     if (len > 0)
16019ccfebddSBram Moolenaar     {
16020d6f5d97SBram Moolenaar 	// Allocate the byte array.
160318a4ba29SBram Moolenaar 	bp = alloc(len);
16049ccfebddSBram Moolenaar 	if (bp == NULL)
16059ccfebddSBram Moolenaar 	    return SP_OTHERERROR;
16069ccfebddSBram Moolenaar 	*bytsp = bp;
160707399e7fSBram Moolenaar 	if (bytsp_len != NULL)
160807399e7fSBram Moolenaar 	    *bytsp_len = len;
16099ccfebddSBram Moolenaar 
16100d6f5d97SBram Moolenaar 	// Allocate the index array.
1611c799fe20SBram Moolenaar 	ip = lalloc_clear(len * sizeof(int), TRUE);
16129ccfebddSBram Moolenaar 	if (ip == NULL)
16139ccfebddSBram Moolenaar 	    return SP_OTHERERROR;
16149ccfebddSBram Moolenaar 	*idxsp = ip;
16159ccfebddSBram Moolenaar 
16160d6f5d97SBram Moolenaar 	// Recursively read the tree and store it in the array.
16179ccfebddSBram Moolenaar 	idx = read_tree_node(fd, bp, ip, len, 0, prefixtree, prefixcnt);
16189ccfebddSBram Moolenaar 	if (idx < 0)
16199ccfebddSBram Moolenaar 	    return idx;
16209ccfebddSBram Moolenaar     }
16219ccfebddSBram Moolenaar     return 0;
16229ccfebddSBram Moolenaar }
16239ccfebddSBram Moolenaar 
16249ccfebddSBram Moolenaar /*
16259ccfebddSBram Moolenaar  * Read one row of siblings from the spell file and store it in the byte array
16269ccfebddSBram Moolenaar  * "byts" and index array "idxs".  Recursively read the children.
16279ccfebddSBram Moolenaar  *
16289ccfebddSBram Moolenaar  * NOTE: The code here must match put_node()!
16299ccfebddSBram Moolenaar  *
16309ccfebddSBram Moolenaar  * Returns the index (>= 0) following the siblings.
16319ccfebddSBram Moolenaar  * Returns SP_TRUNCERROR if the file is shorter than expected.
16329ccfebddSBram Moolenaar  * Returns SP_FORMERROR if there is a format error.
16339ccfebddSBram Moolenaar  */
16349ccfebddSBram Moolenaar     static idx_T
read_tree_node(FILE * fd,char_u * byts,idx_T * idxs,int maxidx,idx_T startidx,int prefixtree,int maxprefcondnr)16359ccfebddSBram Moolenaar read_tree_node(
16369ccfebddSBram Moolenaar     FILE	*fd,
16379ccfebddSBram Moolenaar     char_u	*byts,
16389ccfebddSBram Moolenaar     idx_T	*idxs,
16390d6f5d97SBram Moolenaar     int		maxidx,		    // size of arrays
16400d6f5d97SBram Moolenaar     idx_T	startidx,	    // current index in "byts" and "idxs"
16410d6f5d97SBram Moolenaar     int		prefixtree,	    // TRUE for reading PREFIXTREE
16420d6f5d97SBram Moolenaar     int		maxprefcondnr)	    // maximum for <prefcondnr>
16439ccfebddSBram Moolenaar {
16449ccfebddSBram Moolenaar     int		len;
16459ccfebddSBram Moolenaar     int		i;
16469ccfebddSBram Moolenaar     int		n;
16479ccfebddSBram Moolenaar     idx_T	idx = startidx;
16489ccfebddSBram Moolenaar     int		c;
16499ccfebddSBram Moolenaar     int		c2;
16509ccfebddSBram Moolenaar #define SHARED_MASK	0x8000000
16519ccfebddSBram Moolenaar 
16520d6f5d97SBram Moolenaar     len = getc(fd);					// <siblingcount>
16539ccfebddSBram Moolenaar     if (len <= 0)
16549ccfebddSBram Moolenaar 	return SP_TRUNCERROR;
16559ccfebddSBram Moolenaar 
16569ccfebddSBram Moolenaar     if (startidx + len >= maxidx)
16579ccfebddSBram Moolenaar 	return SP_FORMERROR;
16589ccfebddSBram Moolenaar     byts[idx++] = len;
16599ccfebddSBram Moolenaar 
16600d6f5d97SBram Moolenaar     // Read the byte values, flag/region bytes and shared indexes.
16619ccfebddSBram Moolenaar     for (i = 1; i <= len; ++i)
16629ccfebddSBram Moolenaar     {
16630d6f5d97SBram Moolenaar 	c = getc(fd);					// <byte>
16649ccfebddSBram Moolenaar 	if (c < 0)
16659ccfebddSBram Moolenaar 	    return SP_TRUNCERROR;
16669ccfebddSBram Moolenaar 	if (c <= BY_SPECIAL)
16679ccfebddSBram Moolenaar 	{
16689ccfebddSBram Moolenaar 	    if (c == BY_NOFLAGS && !prefixtree)
16699ccfebddSBram Moolenaar 	    {
16700d6f5d97SBram Moolenaar 		// No flags, all regions.
16719ccfebddSBram Moolenaar 		idxs[idx] = 0;
16729ccfebddSBram Moolenaar 		c = 0;
16739ccfebddSBram Moolenaar 	    }
16749ccfebddSBram Moolenaar 	    else if (c != BY_INDEX)
16759ccfebddSBram Moolenaar 	    {
16769ccfebddSBram Moolenaar 		if (prefixtree)
16779ccfebddSBram Moolenaar 		{
16780d6f5d97SBram Moolenaar 		    // Read the optional pflags byte, the prefix ID and the
16790d6f5d97SBram Moolenaar 		    // condition nr.  In idxs[] store the prefix ID in the low
16800d6f5d97SBram Moolenaar 		    // byte, the condition index shifted up 8 bits, the flags
16810d6f5d97SBram Moolenaar 		    // shifted up 24 bits.
16829ccfebddSBram Moolenaar 		    if (c == BY_FLAGS)
16830d6f5d97SBram Moolenaar 			c = getc(fd) << 24;		// <pflags>
16849ccfebddSBram Moolenaar 		    else
16859ccfebddSBram Moolenaar 			c = 0;
16869ccfebddSBram Moolenaar 
16870d6f5d97SBram Moolenaar 		    c |= getc(fd);			// <affixID>
16889ccfebddSBram Moolenaar 
16890d6f5d97SBram Moolenaar 		    n = get2c(fd);			// <prefcondnr>
16909ccfebddSBram Moolenaar 		    if (n >= maxprefcondnr)
16919ccfebddSBram Moolenaar 			return SP_FORMERROR;
16929ccfebddSBram Moolenaar 		    c |= (n << 8);
16939ccfebddSBram Moolenaar 		}
16940d6f5d97SBram Moolenaar 		else // c must be BY_FLAGS or BY_FLAGS2
16959ccfebddSBram Moolenaar 		{
16960d6f5d97SBram Moolenaar 		    // Read flags and optional region and prefix ID.  In
16970d6f5d97SBram Moolenaar 		    // idxs[] the flags go in the low two bytes, region above
16980d6f5d97SBram Moolenaar 		    // that and prefix ID above the region.
16999ccfebddSBram Moolenaar 		    c2 = c;
17000d6f5d97SBram Moolenaar 		    c = getc(fd);			// <flags>
17019ccfebddSBram Moolenaar 		    if (c2 == BY_FLAGS2)
17020d6f5d97SBram Moolenaar 			c = (getc(fd) << 8) + c;	// <flags2>
17039ccfebddSBram Moolenaar 		    if (c & WF_REGION)
17040d6f5d97SBram Moolenaar 			c = (getc(fd) << 16) + c;	// <region>
17059ccfebddSBram Moolenaar 		    if (c & WF_AFX)
17060d6f5d97SBram Moolenaar 			c = (getc(fd) << 24) + c;	// <affixID>
17079ccfebddSBram Moolenaar 		}
17089ccfebddSBram Moolenaar 
17099ccfebddSBram Moolenaar 		idxs[idx] = c;
17109ccfebddSBram Moolenaar 		c = 0;
17119ccfebddSBram Moolenaar 	    }
17120d6f5d97SBram Moolenaar 	    else // c == BY_INDEX
17139ccfebddSBram Moolenaar 	    {
17140d6f5d97SBram Moolenaar 							// <nodeidx>
17159ccfebddSBram Moolenaar 		n = get3c(fd);
17169ccfebddSBram Moolenaar 		if (n < 0 || n >= maxidx)
17179ccfebddSBram Moolenaar 		    return SP_FORMERROR;
17189ccfebddSBram Moolenaar 		idxs[idx] = n + SHARED_MASK;
17190d6f5d97SBram Moolenaar 		c = getc(fd);				// <xbyte>
17209ccfebddSBram Moolenaar 	    }
17219ccfebddSBram Moolenaar 	}
17229ccfebddSBram Moolenaar 	byts[idx++] = c;
17239ccfebddSBram Moolenaar     }
17249ccfebddSBram Moolenaar 
17250d6f5d97SBram Moolenaar     // Recursively read the children for non-shared siblings.
17260d6f5d97SBram Moolenaar     // Skip the end-of-word ones (zero byte value) and the shared ones (and
17270d6f5d97SBram Moolenaar     // remove SHARED_MASK)
17289ccfebddSBram Moolenaar     for (i = 1; i <= len; ++i)
17299ccfebddSBram Moolenaar 	if (byts[startidx + i] != 0)
17309ccfebddSBram Moolenaar 	{
17319ccfebddSBram Moolenaar 	    if (idxs[startidx + i] & SHARED_MASK)
17329ccfebddSBram Moolenaar 		idxs[startidx + i] &= ~SHARED_MASK;
17339ccfebddSBram Moolenaar 	    else
17349ccfebddSBram Moolenaar 	    {
17359ccfebddSBram Moolenaar 		idxs[startidx + i] = idx;
17369ccfebddSBram Moolenaar 		idx = read_tree_node(fd, byts, idxs, maxidx, idx,
17379ccfebddSBram Moolenaar 						     prefixtree, maxprefcondnr);
17389ccfebddSBram Moolenaar 		if (idx < 0)
17399ccfebddSBram Moolenaar 		    break;
17409ccfebddSBram Moolenaar 	    }
17419ccfebddSBram Moolenaar 	}
17429ccfebddSBram Moolenaar 
17439ccfebddSBram Moolenaar     return idx;
17449ccfebddSBram Moolenaar }
17459ccfebddSBram Moolenaar 
17469ccfebddSBram Moolenaar /*
17479ccfebddSBram Moolenaar  * Reload the spell file "fname" if it's loaded.
17489ccfebddSBram Moolenaar  */
17499ccfebddSBram Moolenaar     static void
spell_reload_one(char_u * fname,int added_word)17509ccfebddSBram Moolenaar spell_reload_one(
17519ccfebddSBram Moolenaar     char_u	*fname,
17520d6f5d97SBram Moolenaar     int		added_word)	// invoked through "zg"
17539ccfebddSBram Moolenaar {
17549ccfebddSBram Moolenaar     slang_T	*slang;
17559ccfebddSBram Moolenaar     int		didit = FALSE;
17569ccfebddSBram Moolenaar 
1757aeea7215SBram Moolenaar     FOR_ALL_SPELL_LANGS(slang)
17589ccfebddSBram Moolenaar     {
175999499b1cSBram Moolenaar 	if (fullpathcmp(fname, slang->sl_fname, FALSE, TRUE) == FPC_SAME)
17609ccfebddSBram Moolenaar 	{
17619ccfebddSBram Moolenaar 	    slang_clear(slang);
17629ccfebddSBram Moolenaar 	    if (spell_load_file(fname, NULL, slang, FALSE) == NULL)
17630d6f5d97SBram Moolenaar 		// reloading failed, clear the language
17649ccfebddSBram Moolenaar 		slang_clear(slang);
17659ccfebddSBram Moolenaar 	    redraw_all_later(SOME_VALID);
17669ccfebddSBram Moolenaar 	    didit = TRUE;
17679ccfebddSBram Moolenaar 	}
17689ccfebddSBram Moolenaar     }
17699ccfebddSBram Moolenaar 
17700d6f5d97SBram Moolenaar     // When "zg" was used and the file wasn't loaded yet, should redo
17710d6f5d97SBram Moolenaar     // 'spelllang' to load it now.
17729ccfebddSBram Moolenaar     if (added_word && !didit)
17739ccfebddSBram Moolenaar 	did_set_spelllang(curwin);
17749ccfebddSBram Moolenaar }
17759ccfebddSBram Moolenaar 
17769ccfebddSBram Moolenaar 
17779ccfebddSBram Moolenaar /*
17789ccfebddSBram Moolenaar  * Functions for ":mkspell".
17799ccfebddSBram Moolenaar  */
17809ccfebddSBram Moolenaar 
17810d6f5d97SBram Moolenaar #define MAXLINELEN  500		// Maximum length in bytes of a line in a .aff
17820d6f5d97SBram Moolenaar 				// and .dic file.
17839ccfebddSBram Moolenaar /*
17849ccfebddSBram Moolenaar  * Main structure to store the contents of a ".aff" file.
17859ccfebddSBram Moolenaar  */
17869ccfebddSBram Moolenaar typedef struct afffile_S
17879ccfebddSBram Moolenaar {
17880d6f5d97SBram Moolenaar     char_u	*af_enc;	// "SET", normalized, alloc'ed string or NULL
17890d6f5d97SBram Moolenaar     int		af_flagtype;	// AFT_CHAR, AFT_LONG, AFT_NUM or AFT_CAPLONG
17900d6f5d97SBram Moolenaar     unsigned	af_rare;	// RARE ID for rare word
17910d6f5d97SBram Moolenaar     unsigned	af_keepcase;	// KEEPCASE ID for keep-case word
17920d6f5d97SBram Moolenaar     unsigned	af_bad;		// BAD ID for banned word
17930d6f5d97SBram Moolenaar     unsigned	af_needaffix;	// NEEDAFFIX ID
17940d6f5d97SBram Moolenaar     unsigned	af_circumfix;	// CIRCUMFIX ID
17950d6f5d97SBram Moolenaar     unsigned	af_needcomp;	// NEEDCOMPOUND ID
17960d6f5d97SBram Moolenaar     unsigned	af_comproot;	// COMPOUNDROOT ID
17970d6f5d97SBram Moolenaar     unsigned	af_compforbid;	// COMPOUNDFORBIDFLAG ID
17980d6f5d97SBram Moolenaar     unsigned	af_comppermit;	// COMPOUNDPERMITFLAG ID
17990d6f5d97SBram Moolenaar     unsigned	af_nosuggest;	// NOSUGGEST ID
18000d6f5d97SBram Moolenaar     int		af_pfxpostpone;	// postpone prefixes without chop string and
18010d6f5d97SBram Moolenaar 				// without flags
18020d6f5d97SBram Moolenaar     int		af_ignoreextra;	// IGNOREEXTRA present
18030d6f5d97SBram Moolenaar     hashtab_T	af_pref;	// hashtable for prefixes, affheader_T
18040d6f5d97SBram Moolenaar     hashtab_T	af_suff;	// hashtable for suffixes, affheader_T
18050d6f5d97SBram Moolenaar     hashtab_T	af_comp;	// hashtable for compound flags, compitem_T
18069ccfebddSBram Moolenaar } afffile_T;
18079ccfebddSBram Moolenaar 
18080d6f5d97SBram Moolenaar #define AFT_CHAR	0	// flags are one character
18090d6f5d97SBram Moolenaar #define AFT_LONG	1	// flags are two characters
18100d6f5d97SBram Moolenaar #define AFT_CAPLONG	2	// flags are one or two characters
18110d6f5d97SBram Moolenaar #define AFT_NUM		3	// flags are numbers, comma separated
18129ccfebddSBram Moolenaar 
18139ccfebddSBram Moolenaar typedef struct affentry_S affentry_T;
18140d6f5d97SBram Moolenaar // Affix entry from ".aff" file.  Used for prefixes and suffixes.
18159ccfebddSBram Moolenaar struct affentry_S
18169ccfebddSBram Moolenaar {
18170d6f5d97SBram Moolenaar     affentry_T	*ae_next;	// next affix with same name/number
18180d6f5d97SBram Moolenaar     char_u	*ae_chop;	// text to chop off basic word (can be NULL)
18190d6f5d97SBram Moolenaar     char_u	*ae_add;	// text to add to basic word (can be NULL)
18200d6f5d97SBram Moolenaar     char_u	*ae_flags;	// flags on the affix (can be NULL)
18210d6f5d97SBram Moolenaar     char_u	*ae_cond;	// condition (NULL for ".")
18220d6f5d97SBram Moolenaar     regprog_T	*ae_prog;	// regexp program for ae_cond or NULL
18230d6f5d97SBram Moolenaar     char	ae_compforbid;	// COMPOUNDFORBIDFLAG found
18240d6f5d97SBram Moolenaar     char	ae_comppermit;	// COMPOUNDPERMITFLAG found
18259ccfebddSBram Moolenaar };
18269ccfebddSBram Moolenaar 
18270d6f5d97SBram Moolenaar #define AH_KEY_LEN 17		// 2 x 8 bytes + NUL
18289ccfebddSBram Moolenaar 
18290d6f5d97SBram Moolenaar // Affix header from ".aff" file.  Used for af_pref and af_suff.
18309ccfebddSBram Moolenaar typedef struct affheader_S
18319ccfebddSBram Moolenaar {
18320d6f5d97SBram Moolenaar     char_u	ah_key[AH_KEY_LEN]; // key for hashtab == name of affix
18330d6f5d97SBram Moolenaar     unsigned	ah_flag;	// affix name as number, uses "af_flagtype"
18340d6f5d97SBram Moolenaar     int		ah_newID;	// prefix ID after renumbering; 0 if not used
18350d6f5d97SBram Moolenaar     int		ah_combine;	// suffix may combine with prefix
18360d6f5d97SBram Moolenaar     int		ah_follows;	// another affix block should be following
18370d6f5d97SBram Moolenaar     affentry_T	*ah_first;	// first affix entry
18389ccfebddSBram Moolenaar } affheader_T;
18399ccfebddSBram Moolenaar 
18409ccfebddSBram Moolenaar #define HI2AH(hi)   ((affheader_T *)(hi)->hi_key)
18419ccfebddSBram Moolenaar 
18420d6f5d97SBram Moolenaar // Flag used in compound items.
18439ccfebddSBram Moolenaar typedef struct compitem_S
18449ccfebddSBram Moolenaar {
18450d6f5d97SBram Moolenaar     char_u	ci_key[AH_KEY_LEN]; // key for hashtab == name of compound
18460d6f5d97SBram Moolenaar     unsigned	ci_flag;	// affix name as number, uses "af_flagtype"
18470d6f5d97SBram Moolenaar     int		ci_newID;	// affix ID after renumbering.
18489ccfebddSBram Moolenaar } compitem_T;
18499ccfebddSBram Moolenaar 
18509ccfebddSBram Moolenaar #define HI2CI(hi)   ((compitem_T *)(hi)->hi_key)
18519ccfebddSBram Moolenaar 
18529ccfebddSBram Moolenaar /*
18539ccfebddSBram Moolenaar  * Structure that is used to store the items in the word tree.  This avoids
18549ccfebddSBram Moolenaar  * the need to keep track of each allocated thing, everything is freed all at
18559ccfebddSBram Moolenaar  * once after ":mkspell" is done.
18569ccfebddSBram Moolenaar  * Note: "sb_next" must be just before "sb_data" to make sure the alignment of
18579ccfebddSBram Moolenaar  * "sb_data" is correct for systems where pointers must be aligned on
18589ccfebddSBram Moolenaar  * pointer-size boundaries and sizeof(pointer) > sizeof(int) (e.g., Sparc).
18599ccfebddSBram Moolenaar  */
18600d6f5d97SBram Moolenaar #define  SBLOCKSIZE 16000	// size of sb_data
18619ccfebddSBram Moolenaar typedef struct sblock_S sblock_T;
18629ccfebddSBram Moolenaar struct sblock_S
18639ccfebddSBram Moolenaar {
18640d6f5d97SBram Moolenaar     int		sb_used;	// nr of bytes already in use
18650d6f5d97SBram Moolenaar     sblock_T	*sb_next;	// next block in list
18660d6f5d97SBram Moolenaar     char_u	sb_data[1];	// data, actually longer
18679ccfebddSBram Moolenaar };
18689ccfebddSBram Moolenaar 
18699ccfebddSBram Moolenaar /*
18709ccfebddSBram Moolenaar  * A node in the tree.
18719ccfebddSBram Moolenaar  */
18729ccfebddSBram Moolenaar typedef struct wordnode_S wordnode_T;
18739ccfebddSBram Moolenaar struct wordnode_S
18749ccfebddSBram Moolenaar {
18750d6f5d97SBram Moolenaar     union   // shared to save space
18769ccfebddSBram Moolenaar     {
18770d6f5d97SBram Moolenaar 	char_u	hashkey[6];	// the hash key, only used while compressing
18780d6f5d97SBram Moolenaar 	int	index;		// index in written nodes (valid after first
18790d6f5d97SBram Moolenaar 				// round)
18809ccfebddSBram Moolenaar     } wn_u1;
18810d6f5d97SBram Moolenaar     union   // shared to save space
18829ccfebddSBram Moolenaar     {
18830d6f5d97SBram Moolenaar 	wordnode_T *next;	// next node with same hash key
18840d6f5d97SBram Moolenaar 	wordnode_T *wnode;	// parent node that will write this node
18859ccfebddSBram Moolenaar     } wn_u2;
18860d6f5d97SBram Moolenaar     wordnode_T	*wn_child;	// child (next byte in word)
18870d6f5d97SBram Moolenaar     wordnode_T  *wn_sibling;	// next sibling (alternate byte in word,
18880d6f5d97SBram Moolenaar 				// always sorted)
18890d6f5d97SBram Moolenaar     int		wn_refs;	// Nr. of references to this node.  Only
18900d6f5d97SBram Moolenaar 				// relevant for first node in a list of
18910d6f5d97SBram Moolenaar 				// siblings, in following siblings it is
18920d6f5d97SBram Moolenaar 				// always one.
18930d6f5d97SBram Moolenaar     char_u	wn_byte;	// Byte for this node. NUL for word end
18949ccfebddSBram Moolenaar 
18950d6f5d97SBram Moolenaar     // Info for when "wn_byte" is NUL.
18960d6f5d97SBram Moolenaar     // In PREFIXTREE "wn_region" is used for the prefcondnr.
18970d6f5d97SBram Moolenaar     // In the soundfolded word tree "wn_flags" has the MSW of the wordnr and
18980d6f5d97SBram Moolenaar     // "wn_region" the LSW of the wordnr.
18990d6f5d97SBram Moolenaar     char_u	wn_affixID;	// supported/required prefix ID or 0
19000d6f5d97SBram Moolenaar     short_u	wn_flags;	// WF_ flags
19010d6f5d97SBram Moolenaar     short	wn_region;	// region mask
19029ccfebddSBram Moolenaar 
19039ccfebddSBram Moolenaar #ifdef SPELL_PRINTTREE
19040d6f5d97SBram Moolenaar     int		wn_nr;		// sequence nr for printing
19059ccfebddSBram Moolenaar #endif
19069ccfebddSBram Moolenaar };
19079ccfebddSBram Moolenaar 
19080d6f5d97SBram Moolenaar #define WN_MASK	 0xffff		// mask relevant bits of "wn_flags"
19099ccfebddSBram Moolenaar 
19109ccfebddSBram Moolenaar #define HI2WN(hi)    (wordnode_T *)((hi)->hi_key)
19119ccfebddSBram Moolenaar 
19129ccfebddSBram Moolenaar /*
19139ccfebddSBram Moolenaar  * Info used while reading the spell files.
19149ccfebddSBram Moolenaar  */
19159ccfebddSBram Moolenaar typedef struct spellinfo_S
19169ccfebddSBram Moolenaar {
19170d6f5d97SBram Moolenaar     wordnode_T	*si_foldroot;	// tree with case-folded words
19180d6f5d97SBram Moolenaar     long	si_foldwcount;	// nr of words in si_foldroot
19199ccfebddSBram Moolenaar 
19200d6f5d97SBram Moolenaar     wordnode_T	*si_keeproot;	// tree with keep-case words
19210d6f5d97SBram Moolenaar     long	si_keepwcount;	// nr of words in si_keeproot
19229ccfebddSBram Moolenaar 
19230d6f5d97SBram Moolenaar     wordnode_T	*si_prefroot;	// tree with postponed prefixes
19249ccfebddSBram Moolenaar 
19250d6f5d97SBram Moolenaar     long	si_sugtree;	// creating the soundfolding trie
19269ccfebddSBram Moolenaar 
19270d6f5d97SBram Moolenaar     sblock_T	*si_blocks;	// memory blocks used
19280d6f5d97SBram Moolenaar     long	si_blocks_cnt;	// memory blocks allocated
19290d6f5d97SBram Moolenaar     int		si_did_emsg;	// TRUE when ran out of memory
19309ccfebddSBram Moolenaar 
19310d6f5d97SBram Moolenaar     long	si_compress_cnt;    // words to add before lowering
19320d6f5d97SBram Moolenaar 				    // compression limit
19330d6f5d97SBram Moolenaar     wordnode_T	*si_first_free; // List of nodes that have been freed during
19340d6f5d97SBram Moolenaar 				// compression, linked by "wn_child" field.
19350d6f5d97SBram Moolenaar     long	si_free_count;	// number of nodes in si_first_free
19369ccfebddSBram Moolenaar #ifdef SPELL_PRINTTREE
19370d6f5d97SBram Moolenaar     int		si_wordnode_nr;	// sequence nr for nodes
19389ccfebddSBram Moolenaar #endif
19390d6f5d97SBram Moolenaar     buf_T	*si_spellbuf;	// buffer used to store soundfold word table
19409ccfebddSBram Moolenaar 
19410d6f5d97SBram Moolenaar     int		si_ascii;	// handling only ASCII words
19420d6f5d97SBram Moolenaar     int		si_add;		// addition file
19430d6f5d97SBram Moolenaar     int		si_clear_chartab;   // when TRUE clear char tables
19440d6f5d97SBram Moolenaar     int		si_region;	// region mask
19450d6f5d97SBram Moolenaar     vimconv_T	si_conv;	// for conversion to 'encoding'
19460d6f5d97SBram Moolenaar     int		si_memtot;	// runtime memory used
19470d6f5d97SBram Moolenaar     int		si_verbose;	// verbose messages
19480d6f5d97SBram Moolenaar     int		si_msg_count;	// number of words added since last message
19490d6f5d97SBram Moolenaar     char_u	*si_info;	// info text chars or NULL
19500d6f5d97SBram Moolenaar     int		si_region_count; // number of regions supported (1 when there
19510d6f5d97SBram Moolenaar 				 // are no regions)
19522993ac5fSBram Moolenaar     char_u	si_region_name[MAXREGIONS * 2 + 1];
19530d6f5d97SBram Moolenaar 				// region names; used only if
19540d6f5d97SBram Moolenaar 				// si_region_count > 1)
19559ccfebddSBram Moolenaar 
19560d6f5d97SBram Moolenaar     garray_T	si_rep;		// list of fromto_T entries from REP lines
19570d6f5d97SBram Moolenaar     garray_T	si_repsal;	// list of fromto_T entries from REPSAL lines
19580d6f5d97SBram Moolenaar     garray_T	si_sal;		// list of fromto_T entries from SAL lines
19590d6f5d97SBram Moolenaar     char_u	*si_sofofr;	// SOFOFROM text
19600d6f5d97SBram Moolenaar     char_u	*si_sofoto;	// SOFOTO text
19610d6f5d97SBram Moolenaar     int		si_nosugfile;	// NOSUGFILE item found
19620d6f5d97SBram Moolenaar     int		si_nosplitsugs;	// NOSPLITSUGS item found
19630d6f5d97SBram Moolenaar     int		si_nocompoundsugs; // NOCOMPOUNDSUGS item found
19640d6f5d97SBram Moolenaar     int		si_followup;	// soundsalike: ?
19650d6f5d97SBram Moolenaar     int		si_collapse;	// soundsalike: ?
19660d6f5d97SBram Moolenaar     hashtab_T	si_commonwords;	// hashtable for common words
19670d6f5d97SBram Moolenaar     time_t	si_sugtime;	// timestamp for .sug file
19680d6f5d97SBram Moolenaar     int		si_rem_accents;	// soundsalike: remove accents
19690d6f5d97SBram Moolenaar     garray_T	si_map;		// MAP info concatenated
19700d6f5d97SBram Moolenaar     char_u	*si_midword;	// MIDWORD chars or NULL
19710d6f5d97SBram Moolenaar     int		si_compmax;	// max nr of words for compounding
19720d6f5d97SBram Moolenaar     int		si_compminlen;	// minimal length for compounding
19730d6f5d97SBram Moolenaar     int		si_compsylmax;	// max nr of syllables for compounding
19740d6f5d97SBram Moolenaar     int		si_compoptions;	// COMP_ flags
19750d6f5d97SBram Moolenaar     garray_T	si_comppat;	// CHECKCOMPOUNDPATTERN items, each stored as
19760d6f5d97SBram Moolenaar 				// a string
19770d6f5d97SBram Moolenaar     char_u	*si_compflags;	// flags used for compounding
19780d6f5d97SBram Moolenaar     char_u	si_nobreak;	// NOBREAK
19790d6f5d97SBram Moolenaar     char_u	*si_syllable;	// syllable string
19800d6f5d97SBram Moolenaar     garray_T	si_prefcond;	// table with conditions for postponed
19810d6f5d97SBram Moolenaar 				// prefixes, each stored as a string
19820d6f5d97SBram Moolenaar     int		si_newprefID;	// current value for ah_newID
19830d6f5d97SBram Moolenaar     int		si_newcompID;	// current value for compound ID
19849ccfebddSBram Moolenaar } spellinfo_T;
19859ccfebddSBram Moolenaar 
19869ccfebddSBram Moolenaar static int is_aff_rule(char_u **items, int itemcnt, char *rulename, int	 mincount);
19879ccfebddSBram Moolenaar static void aff_process_flags(afffile_T *affile, affentry_T *entry);
19889ccfebddSBram Moolenaar static int spell_info_item(char_u *s);
19899ccfebddSBram Moolenaar static unsigned affitem2flag(int flagtype, char_u *item, char_u	*fname, int lnum);
19909ccfebddSBram Moolenaar static unsigned get_affitem(int flagtype, char_u **pp);
19919ccfebddSBram Moolenaar static void process_compflags(spellinfo_T *spin, afffile_T *aff, char_u *compflags);
19929ccfebddSBram Moolenaar static void check_renumber(spellinfo_T *spin);
19939ccfebddSBram Moolenaar static void aff_check_number(int spinval, int affval, char *name);
19949ccfebddSBram Moolenaar static void aff_check_string(char_u *spinval, char_u *affval, char *name);
19959ccfebddSBram Moolenaar static int str_equal(char_u *s1, char_u	*s2);
19969ccfebddSBram Moolenaar static void add_fromto(spellinfo_T *spin, garray_T *gap, char_u	*from, char_u *to);
19979ccfebddSBram Moolenaar static int sal_to_bool(char_u *s);
19989ccfebddSBram Moolenaar static int get_affix_flags(afffile_T *affile, char_u *afflist);
19999ccfebddSBram Moolenaar static int get_pfxlist(afffile_T *affile, char_u *afflist, char_u *store_afflist);
20009ccfebddSBram Moolenaar static void get_compflags(afffile_T *affile, char_u *afflist, char_u *store_afflist);
20019ccfebddSBram Moolenaar static int store_aff_word(spellinfo_T *spin, char_u *word, char_u *afflist, afffile_T *affile, hashtab_T *ht, hashtab_T *xht, int condit, int flags, char_u *pfxlist, int pfxlen);
20029ccfebddSBram Moolenaar static void *getroom(spellinfo_T *spin, size_t len, int align);
20039ccfebddSBram Moolenaar static char_u *getroom_save(spellinfo_T *spin, char_u *s);
20049ccfebddSBram Moolenaar static int store_word(spellinfo_T *spin, char_u *word, int flags, int region, char_u *pfxlist, int need_affix);
20059ccfebddSBram Moolenaar static int tree_add_word(spellinfo_T *spin, char_u *word, wordnode_T *tree, int flags, int region, int affixID);
20069ccfebddSBram Moolenaar static wordnode_T *get_wordnode(spellinfo_T *spin);
20079ccfebddSBram Moolenaar static void free_wordnode(spellinfo_T *spin, wordnode_T *n);
2008408c23b0SBram Moolenaar static void wordtree_compress(spellinfo_T *spin, wordnode_T *root, char *name);
200959f88fbfSBram Moolenaar static long node_compress(spellinfo_T *spin, wordnode_T *node, hashtab_T *ht, long *tot);
20109ccfebddSBram Moolenaar static int node_equal(wordnode_T *n1, wordnode_T *n2);
20119ccfebddSBram Moolenaar static void clear_node(wordnode_T *node);
20129ccfebddSBram Moolenaar static int put_node(FILE *fd, wordnode_T *node, int idx, int regionmask, int prefixtree);
20139ccfebddSBram Moolenaar static int sug_filltree(spellinfo_T *spin, slang_T *slang);
20149ccfebddSBram Moolenaar static int sug_maketable(spellinfo_T *spin);
20159ccfebddSBram Moolenaar static int sug_filltable(spellinfo_T *spin, wordnode_T *node, int startwordnr, garray_T *gap);
20169ccfebddSBram Moolenaar static int offset2bytes(int nr, char_u *buf);
20179ccfebddSBram Moolenaar static void sug_write(spellinfo_T *spin, char_u *fname);
20189ccfebddSBram Moolenaar static void spell_message(spellinfo_T *spin, char_u *str);
20199ccfebddSBram Moolenaar static void init_spellfile(void);
20209ccfebddSBram Moolenaar 
20210d6f5d97SBram Moolenaar // In the postponed prefixes tree wn_flags is used to store the WFP_ flags,
20220d6f5d97SBram Moolenaar // but it must be negative to indicate the prefix tree to tree_add_word().
20230d6f5d97SBram Moolenaar // Use a negative number with the lower 8 bits zero.
20249ccfebddSBram Moolenaar #define PFX_FLAGS	-256
20259ccfebddSBram Moolenaar 
20260d6f5d97SBram Moolenaar // flags for "condit" argument of store_aff_word()
20270d6f5d97SBram Moolenaar #define CONDIT_COMB	1	// affix must combine
20280d6f5d97SBram Moolenaar #define CONDIT_CFIX	2	// affix must have CIRCUMFIX flag
20290d6f5d97SBram Moolenaar #define CONDIT_SUF	4	// add a suffix for matching flags
20300d6f5d97SBram Moolenaar #define CONDIT_AFF	8	// word already has an affix
20319ccfebddSBram Moolenaar 
20329ccfebddSBram Moolenaar /*
203359f88fbfSBram Moolenaar  * Tunable parameters for when the tree is compressed.  Filled from the
203459f88fbfSBram Moolenaar  * 'mkspellmem' option.
20359ccfebddSBram Moolenaar  */
20360d6f5d97SBram Moolenaar static long compress_start = 30000;	// memory / SBLOCKSIZE
20370d6f5d97SBram Moolenaar static long compress_inc = 100;		// memory / SBLOCKSIZE
20380d6f5d97SBram Moolenaar static long compress_added = 500000;	// word count
20399ccfebddSBram Moolenaar 
20409ccfebddSBram Moolenaar /*
20419ccfebddSBram Moolenaar  * Check the 'mkspellmem' option.  Return FAIL if it's wrong.
20429ccfebddSBram Moolenaar  * Sets "sps_flags".
20439ccfebddSBram Moolenaar  */
20449ccfebddSBram Moolenaar     int
spell_check_msm(void)20459ccfebddSBram Moolenaar spell_check_msm(void)
20469ccfebddSBram Moolenaar {
20479ccfebddSBram Moolenaar     char_u	*p = p_msm;
20489ccfebddSBram Moolenaar     long	start = 0;
20499ccfebddSBram Moolenaar     long	incr = 0;
20509ccfebddSBram Moolenaar     long	added = 0;
20519ccfebddSBram Moolenaar 
20529ccfebddSBram Moolenaar     if (!VIM_ISDIGIT(*p))
20539ccfebddSBram Moolenaar 	return FAIL;
20540d6f5d97SBram Moolenaar     // block count = (value * 1024) / SBLOCKSIZE (but avoid overflow)
20559ccfebddSBram Moolenaar     start = (getdigits(&p) * 10) / (SBLOCKSIZE / 102);
20569ccfebddSBram Moolenaar     if (*p != ',')
20579ccfebddSBram Moolenaar 	return FAIL;
20589ccfebddSBram Moolenaar     ++p;
20599ccfebddSBram Moolenaar     if (!VIM_ISDIGIT(*p))
20609ccfebddSBram Moolenaar 	return FAIL;
20619ccfebddSBram Moolenaar     incr = (getdigits(&p) * 102) / (SBLOCKSIZE / 10);
20629ccfebddSBram Moolenaar     if (*p != ',')
20639ccfebddSBram Moolenaar 	return FAIL;
20649ccfebddSBram Moolenaar     ++p;
20659ccfebddSBram Moolenaar     if (!VIM_ISDIGIT(*p))
20669ccfebddSBram Moolenaar 	return FAIL;
20679ccfebddSBram Moolenaar     added = getdigits(&p) * 1024;
20689ccfebddSBram Moolenaar     if (*p != NUL)
20699ccfebddSBram Moolenaar 	return FAIL;
20709ccfebddSBram Moolenaar 
20719ccfebddSBram Moolenaar     if (start == 0 || incr == 0 || added == 0 || incr > start)
20729ccfebddSBram Moolenaar 	return FAIL;
20739ccfebddSBram Moolenaar 
20749ccfebddSBram Moolenaar     compress_start = start;
20759ccfebddSBram Moolenaar     compress_inc = incr;
20769ccfebddSBram Moolenaar     compress_added = added;
20779ccfebddSBram Moolenaar     return OK;
20789ccfebddSBram Moolenaar }
20799ccfebddSBram Moolenaar 
20809ccfebddSBram Moolenaar #ifdef SPELL_PRINTTREE
20819ccfebddSBram Moolenaar /*
20829ccfebddSBram Moolenaar  * For debugging the tree code: print the current tree in a (more or less)
20839ccfebddSBram Moolenaar  * readable format, so that we can see what happens when adding a word and/or
20849ccfebddSBram Moolenaar  * compressing the tree.
20859ccfebddSBram Moolenaar  * Based on code from Olaf Seibert.
20869ccfebddSBram Moolenaar  */
20879ccfebddSBram Moolenaar #define PRINTLINESIZE	1000
20889ccfebddSBram Moolenaar #define PRINTWIDTH	6
20899ccfebddSBram Moolenaar 
20909ccfebddSBram Moolenaar #define PRINTSOME(l, depth, fmt, a1, a2) vim_snprintf(l + depth * PRINTWIDTH, \
20919ccfebddSBram Moolenaar 	    PRINTLINESIZE - PRINTWIDTH * depth, fmt, a1, a2)
20929ccfebddSBram Moolenaar 
20939ccfebddSBram Moolenaar static char line1[PRINTLINESIZE];
20949ccfebddSBram Moolenaar static char line2[PRINTLINESIZE];
20959ccfebddSBram Moolenaar static char line3[PRINTLINESIZE];
20969ccfebddSBram Moolenaar 
20979ccfebddSBram Moolenaar     static void
spell_clear_flags(wordnode_T * node)20989ccfebddSBram Moolenaar spell_clear_flags(wordnode_T *node)
20999ccfebddSBram Moolenaar {
21009ccfebddSBram Moolenaar     wordnode_T	*np;
21019ccfebddSBram Moolenaar 
2102aeea7215SBram Moolenaar     FOR_ALL_NODE_SIBLINGS(node, np)
21039ccfebddSBram Moolenaar     {
21049ccfebddSBram Moolenaar 	np->wn_u1.index = FALSE;
21059ccfebddSBram Moolenaar 	spell_clear_flags(np->wn_child);
21069ccfebddSBram Moolenaar     }
21079ccfebddSBram Moolenaar }
21089ccfebddSBram Moolenaar 
21099ccfebddSBram Moolenaar     static void
spell_print_node(wordnode_T * node,int depth)21109ccfebddSBram Moolenaar spell_print_node(wordnode_T *node, int depth)
21119ccfebddSBram Moolenaar {
21129ccfebddSBram Moolenaar     if (node->wn_u1.index)
21139ccfebddSBram Moolenaar     {
21140d6f5d97SBram Moolenaar 	// Done this node before, print the reference.
21159ccfebddSBram Moolenaar 	PRINTSOME(line1, depth, "(%d)", node->wn_nr, 0);
21169ccfebddSBram Moolenaar 	PRINTSOME(line2, depth, "    ", 0, 0);
21179ccfebddSBram Moolenaar 	PRINTSOME(line3, depth, "    ", 0, 0);
211832526b3cSBram Moolenaar 	msg(line1);
211932526b3cSBram Moolenaar 	msg(line2);
212032526b3cSBram Moolenaar 	msg(line3);
21219ccfebddSBram Moolenaar     }
21229ccfebddSBram Moolenaar     else
21239ccfebddSBram Moolenaar     {
21249ccfebddSBram Moolenaar 	node->wn_u1.index = TRUE;
21259ccfebddSBram Moolenaar 
21269ccfebddSBram Moolenaar 	if (node->wn_byte != NUL)
21279ccfebddSBram Moolenaar 	{
21289ccfebddSBram Moolenaar 	    if (node->wn_child != NULL)
21299ccfebddSBram Moolenaar 		PRINTSOME(line1, depth, " %c -> ", node->wn_byte, 0);
21309ccfebddSBram Moolenaar 	    else
21310d6f5d97SBram Moolenaar 		// Cannot happen?
21329ccfebddSBram Moolenaar 		PRINTSOME(line1, depth, " %c ???", node->wn_byte, 0);
21339ccfebddSBram Moolenaar 	}
21349ccfebddSBram Moolenaar 	else
21359ccfebddSBram Moolenaar 	    PRINTSOME(line1, depth, " $    ", 0, 0);
21369ccfebddSBram Moolenaar 
21379ccfebddSBram Moolenaar 	PRINTSOME(line2, depth, "%d/%d    ", node->wn_nr, node->wn_refs);
21389ccfebddSBram Moolenaar 
21399ccfebddSBram Moolenaar 	if (node->wn_sibling != NULL)
21409ccfebddSBram Moolenaar 	    PRINTSOME(line3, depth, " |    ", 0, 0);
21419ccfebddSBram Moolenaar 	else
21429ccfebddSBram Moolenaar 	    PRINTSOME(line3, depth, "      ", 0, 0);
21439ccfebddSBram Moolenaar 
21449ccfebddSBram Moolenaar 	if (node->wn_byte == NUL)
21459ccfebddSBram Moolenaar 	{
214632526b3cSBram Moolenaar 	    msg(line1);
214732526b3cSBram Moolenaar 	    msg(line2);
214832526b3cSBram Moolenaar 	    msg(line3);
21499ccfebddSBram Moolenaar 	}
21509ccfebddSBram Moolenaar 
21510d6f5d97SBram Moolenaar 	// do the children
21529ccfebddSBram Moolenaar 	if (node->wn_byte != NUL && node->wn_child != NULL)
21539ccfebddSBram Moolenaar 	    spell_print_node(node->wn_child, depth + 1);
21549ccfebddSBram Moolenaar 
21550d6f5d97SBram Moolenaar 	// do the siblings
21569ccfebddSBram Moolenaar 	if (node->wn_sibling != NULL)
21579ccfebddSBram Moolenaar 	{
21580d6f5d97SBram Moolenaar 	    // get rid of all parent details except |
21599ccfebddSBram Moolenaar 	    STRCPY(line1, line3);
21609ccfebddSBram Moolenaar 	    STRCPY(line2, line3);
21619ccfebddSBram Moolenaar 	    spell_print_node(node->wn_sibling, depth);
21629ccfebddSBram Moolenaar 	}
21639ccfebddSBram Moolenaar     }
21649ccfebddSBram Moolenaar }
21659ccfebddSBram Moolenaar 
21669ccfebddSBram Moolenaar     static void
spell_print_tree(wordnode_T * root)21679ccfebddSBram Moolenaar spell_print_tree(wordnode_T *root)
21689ccfebddSBram Moolenaar {
21699ccfebddSBram Moolenaar     if (root != NULL)
21709ccfebddSBram Moolenaar     {
21710d6f5d97SBram Moolenaar 	// Clear the "wn_u1.index" fields, used to remember what has been
21720d6f5d97SBram Moolenaar 	// done.
21739ccfebddSBram Moolenaar 	spell_clear_flags(root);
21749ccfebddSBram Moolenaar 
21750d6f5d97SBram Moolenaar 	// Recursively print the tree.
21769ccfebddSBram Moolenaar 	spell_print_node(root, 0);
21779ccfebddSBram Moolenaar     }
21789ccfebddSBram Moolenaar }
21790d6f5d97SBram Moolenaar #endif // SPELL_PRINTTREE
21809ccfebddSBram Moolenaar 
21819ccfebddSBram Moolenaar /*
21829ccfebddSBram Moolenaar  * Read the affix file "fname".
21839ccfebddSBram Moolenaar  * Returns an afffile_T, NULL for complete failure.
21849ccfebddSBram Moolenaar  */
21859ccfebddSBram Moolenaar     static afffile_T *
spell_read_aff(spellinfo_T * spin,char_u * fname)21869ccfebddSBram Moolenaar spell_read_aff(spellinfo_T *spin, char_u *fname)
21879ccfebddSBram Moolenaar {
21889ccfebddSBram Moolenaar     FILE	*fd;
21899ccfebddSBram Moolenaar     afffile_T	*aff;
21909ccfebddSBram Moolenaar     char_u	rline[MAXLINELEN];
21919ccfebddSBram Moolenaar     char_u	*line;
21929ccfebddSBram Moolenaar     char_u	*pc = NULL;
21939ccfebddSBram Moolenaar #define MAXITEMCNT  30
21949ccfebddSBram Moolenaar     char_u	*(items[MAXITEMCNT]);
21959ccfebddSBram Moolenaar     int		itemcnt;
21969ccfebddSBram Moolenaar     char_u	*p;
21979ccfebddSBram Moolenaar     int		lnum = 0;
21989ccfebddSBram Moolenaar     affheader_T	*cur_aff = NULL;
21999ccfebddSBram Moolenaar     int		did_postpone_prefix = FALSE;
22009ccfebddSBram Moolenaar     int		aff_todo = 0;
22019ccfebddSBram Moolenaar     hashtab_T	*tp;
22029ccfebddSBram Moolenaar     char_u	*low = NULL;
22039ccfebddSBram Moolenaar     char_u	*fol = NULL;
22049ccfebddSBram Moolenaar     char_u	*upp = NULL;
22059ccfebddSBram Moolenaar     int		do_rep;
22069ccfebddSBram Moolenaar     int		do_repsal;
22079ccfebddSBram Moolenaar     int		do_sal;
22089ccfebddSBram Moolenaar     int		do_mapline;
22099ccfebddSBram Moolenaar     int		found_map = FALSE;
22109ccfebddSBram Moolenaar     hashitem_T	*hi;
22119ccfebddSBram Moolenaar     int		l;
22120d6f5d97SBram Moolenaar     int		compminlen = 0;		// COMPOUNDMIN value
22130d6f5d97SBram Moolenaar     int		compsylmax = 0;		// COMPOUNDSYLMAX value
22140d6f5d97SBram Moolenaar     int		compoptions = 0;	// COMP_ flags
22150d6f5d97SBram Moolenaar     int		compmax = 0;		// COMPOUNDWORDMAX value
22160d6f5d97SBram Moolenaar     char_u	*compflags = NULL;	// COMPOUNDFLAG and COMPOUNDRULE
22170d6f5d97SBram Moolenaar 					// concatenated
22180d6f5d97SBram Moolenaar     char_u	*midword = NULL;	// MIDWORD value
22190d6f5d97SBram Moolenaar     char_u	*syllable = NULL;	// SYLLABLE value
22200d6f5d97SBram Moolenaar     char_u	*sofofrom = NULL;	// SOFOFROM value
22210d6f5d97SBram Moolenaar     char_u	*sofoto = NULL;		// SOFOTO value
22229ccfebddSBram Moolenaar 
22239ccfebddSBram Moolenaar     /*
22249ccfebddSBram Moolenaar      * Open the file.
22259ccfebddSBram Moolenaar      */
22269ccfebddSBram Moolenaar     fd = mch_fopen((char *)fname, "r");
22279ccfebddSBram Moolenaar     if (fd == NULL)
22289ccfebddSBram Moolenaar     {
2229f9e3e09fSBram Moolenaar 	semsg(_(e_notopen), fname);
22309ccfebddSBram Moolenaar 	return NULL;
22319ccfebddSBram Moolenaar     }
22329ccfebddSBram Moolenaar 
22339ccfebddSBram Moolenaar     vim_snprintf((char *)IObuff, IOSIZE, _("Reading affix file %s..."), fname);
22349ccfebddSBram Moolenaar     spell_message(spin, IObuff);
22359ccfebddSBram Moolenaar 
22360d6f5d97SBram Moolenaar     // Only do REP lines when not done in another .aff file already.
22379ccfebddSBram Moolenaar     do_rep = spin->si_rep.ga_len == 0;
22389ccfebddSBram Moolenaar 
22390d6f5d97SBram Moolenaar     // Only do REPSAL lines when not done in another .aff file already.
22409ccfebddSBram Moolenaar     do_repsal = spin->si_repsal.ga_len == 0;
22419ccfebddSBram Moolenaar 
22420d6f5d97SBram Moolenaar     // Only do SAL lines when not done in another .aff file already.
22439ccfebddSBram Moolenaar     do_sal = spin->si_sal.ga_len == 0;
22449ccfebddSBram Moolenaar 
22450d6f5d97SBram Moolenaar     // Only do MAP lines when not done in another .aff file already.
22469ccfebddSBram Moolenaar     do_mapline = spin->si_map.ga_len == 0;
22479ccfebddSBram Moolenaar 
22489ccfebddSBram Moolenaar     /*
22499ccfebddSBram Moolenaar      * Allocate and init the afffile_T structure.
22509ccfebddSBram Moolenaar      */
22519ccfebddSBram Moolenaar     aff = (afffile_T *)getroom(spin, sizeof(afffile_T), TRUE);
22529ccfebddSBram Moolenaar     if (aff == NULL)
22539ccfebddSBram Moolenaar     {
22549ccfebddSBram Moolenaar 	fclose(fd);
22559ccfebddSBram Moolenaar 	return NULL;
22569ccfebddSBram Moolenaar     }
22579ccfebddSBram Moolenaar     hash_init(&aff->af_pref);
22589ccfebddSBram Moolenaar     hash_init(&aff->af_suff);
22599ccfebddSBram Moolenaar     hash_init(&aff->af_comp);
22609ccfebddSBram Moolenaar 
22619ccfebddSBram Moolenaar     /*
22629ccfebddSBram Moolenaar      * Read all the lines in the file one by one.
22639ccfebddSBram Moolenaar      */
22649ccfebddSBram Moolenaar     while (!vim_fgets(rline, MAXLINELEN, fd) && !got_int)
22659ccfebddSBram Moolenaar     {
22669ccfebddSBram Moolenaar 	line_breakcheck();
22679ccfebddSBram Moolenaar 	++lnum;
22689ccfebddSBram Moolenaar 
22690d6f5d97SBram Moolenaar 	// Skip comment lines.
22709ccfebddSBram Moolenaar 	if (*rline == '#')
22719ccfebddSBram Moolenaar 	    continue;
22729ccfebddSBram Moolenaar 
22730d6f5d97SBram Moolenaar 	// Convert from "SET" to 'encoding' when needed.
22749ccfebddSBram Moolenaar 	vim_free(pc);
22759ccfebddSBram Moolenaar 	if (spin->si_conv.vc_type != CONV_NONE)
22769ccfebddSBram Moolenaar 	{
22779ccfebddSBram Moolenaar 	    pc = string_convert(&spin->si_conv, rline, NULL);
22789ccfebddSBram Moolenaar 	    if (pc == NULL)
22799ccfebddSBram Moolenaar 	    {
2280f9e3e09fSBram Moolenaar 		smsg(_("Conversion failure for word in %s line %d: %s"),
22819ccfebddSBram Moolenaar 							   fname, lnum, rline);
22829ccfebddSBram Moolenaar 		continue;
22839ccfebddSBram Moolenaar 	    }
22849ccfebddSBram Moolenaar 	    line = pc;
22859ccfebddSBram Moolenaar 	}
22869ccfebddSBram Moolenaar 	else
22879ccfebddSBram Moolenaar 	{
22889ccfebddSBram Moolenaar 	    pc = NULL;
22899ccfebddSBram Moolenaar 	    line = rline;
22909ccfebddSBram Moolenaar 	}
22919ccfebddSBram Moolenaar 
22920d6f5d97SBram Moolenaar 	// Split the line up in white separated items.  Put a NUL after each
22930d6f5d97SBram Moolenaar 	// item.
22949ccfebddSBram Moolenaar 	itemcnt = 0;
22959ccfebddSBram Moolenaar 	for (p = line; ; )
22969ccfebddSBram Moolenaar 	{
22970d6f5d97SBram Moolenaar 	    while (*p != NUL && *p <= ' ')  // skip white space and CR/NL
22989ccfebddSBram Moolenaar 		++p;
22999ccfebddSBram Moolenaar 	    if (*p == NUL)
23009ccfebddSBram Moolenaar 		break;
23010d6f5d97SBram Moolenaar 	    if (itemcnt == MAXITEMCNT)	    // too many items
23029ccfebddSBram Moolenaar 		break;
23039ccfebddSBram Moolenaar 	    items[itemcnt++] = p;
23040d6f5d97SBram Moolenaar 	    // A few items have arbitrary text argument, don't split them.
23059ccfebddSBram Moolenaar 	    if (itemcnt == 2 && spell_info_item(items[0]))
23060d6f5d97SBram Moolenaar 		while (*p >= ' ' || *p == TAB)    // skip until CR/NL
23079ccfebddSBram Moolenaar 		    ++p;
23089ccfebddSBram Moolenaar 	    else
23090d6f5d97SBram Moolenaar 		while (*p > ' ')    // skip until white space or CR/NL
23109ccfebddSBram Moolenaar 		    ++p;
23119ccfebddSBram Moolenaar 	    if (*p == NUL)
23129ccfebddSBram Moolenaar 		break;
23139ccfebddSBram Moolenaar 	    *p++ = NUL;
23149ccfebddSBram Moolenaar 	}
23159ccfebddSBram Moolenaar 
23160d6f5d97SBram Moolenaar 	// Handle non-empty lines.
23179ccfebddSBram Moolenaar 	if (itemcnt > 0)
23189ccfebddSBram Moolenaar 	{
23199ccfebddSBram Moolenaar 	    if (is_aff_rule(items, itemcnt, "SET", 2) && aff->af_enc == NULL)
23209ccfebddSBram Moolenaar 	    {
23210d6f5d97SBram Moolenaar 		// Setup for conversion from "ENC" to 'encoding'.
23229ccfebddSBram Moolenaar 		aff->af_enc = enc_canonize(items[1]);
23239ccfebddSBram Moolenaar 		if (aff->af_enc != NULL && !spin->si_ascii
23249ccfebddSBram Moolenaar 			&& convert_setup(&spin->si_conv, aff->af_enc,
23259ccfebddSBram Moolenaar 							       p_enc) == FAIL)
2326f9e3e09fSBram Moolenaar 		    smsg(_("Conversion in %s not supported: from %s to %s"),
23279ccfebddSBram Moolenaar 					       fname, aff->af_enc, p_enc);
23289ccfebddSBram Moolenaar 		spin->si_conv.vc_fail = TRUE;
23299ccfebddSBram Moolenaar 	    }
23309ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "FLAG", 2)
23319ccfebddSBram Moolenaar 					      && aff->af_flagtype == AFT_CHAR)
23329ccfebddSBram Moolenaar 	    {
23339ccfebddSBram Moolenaar 		if (STRCMP(items[1], "long") == 0)
23349ccfebddSBram Moolenaar 		    aff->af_flagtype = AFT_LONG;
23359ccfebddSBram Moolenaar 		else if (STRCMP(items[1], "num") == 0)
23369ccfebddSBram Moolenaar 		    aff->af_flagtype = AFT_NUM;
23379ccfebddSBram Moolenaar 		else if (STRCMP(items[1], "caplong") == 0)
23389ccfebddSBram Moolenaar 		    aff->af_flagtype = AFT_CAPLONG;
23399ccfebddSBram Moolenaar 		else
2340f9e3e09fSBram Moolenaar 		    smsg(_("Invalid value for FLAG in %s line %d: %s"),
23419ccfebddSBram Moolenaar 			    fname, lnum, items[1]);
23429ccfebddSBram Moolenaar 		if (aff->af_rare != 0
23439ccfebddSBram Moolenaar 			|| aff->af_keepcase != 0
23449ccfebddSBram Moolenaar 			|| aff->af_bad != 0
23459ccfebddSBram Moolenaar 			|| aff->af_needaffix != 0
23469ccfebddSBram Moolenaar 			|| aff->af_circumfix != 0
23479ccfebddSBram Moolenaar 			|| aff->af_needcomp != 0
23489ccfebddSBram Moolenaar 			|| aff->af_comproot != 0
23499ccfebddSBram Moolenaar 			|| aff->af_nosuggest != 0
23509ccfebddSBram Moolenaar 			|| compflags != NULL
23519ccfebddSBram Moolenaar 			|| aff->af_suff.ht_used > 0
23529ccfebddSBram Moolenaar 			|| aff->af_pref.ht_used > 0)
2353f9e3e09fSBram Moolenaar 		    smsg(_("FLAG after using flags in %s line %d: %s"),
23549ccfebddSBram Moolenaar 			    fname, lnum, items[1]);
23559ccfebddSBram Moolenaar 	    }
23569ccfebddSBram Moolenaar 	    else if (spell_info_item(items[0]))
23579ccfebddSBram Moolenaar 	    {
23589ccfebddSBram Moolenaar 		    p = (char_u *)getroom(spin,
23599ccfebddSBram Moolenaar 			    (spin->si_info == NULL ? 0 : STRLEN(spin->si_info))
23609ccfebddSBram Moolenaar 			    + STRLEN(items[0])
23619ccfebddSBram Moolenaar 			    + STRLEN(items[1]) + 3, FALSE);
23629ccfebddSBram Moolenaar 		    if (p != NULL)
23639ccfebddSBram Moolenaar 		    {
23649ccfebddSBram Moolenaar 			if (spin->si_info != NULL)
23659ccfebddSBram Moolenaar 			{
23669ccfebddSBram Moolenaar 			    STRCPY(p, spin->si_info);
23679ccfebddSBram Moolenaar 			    STRCAT(p, "\n");
23689ccfebddSBram Moolenaar 			}
23699ccfebddSBram Moolenaar 			STRCAT(p, items[0]);
23709ccfebddSBram Moolenaar 			STRCAT(p, " ");
23719ccfebddSBram Moolenaar 			STRCAT(p, items[1]);
23729ccfebddSBram Moolenaar 			spin->si_info = p;
23739ccfebddSBram Moolenaar 		    }
23749ccfebddSBram Moolenaar 	    }
23759ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "MIDWORD", 2)
23769ccfebddSBram Moolenaar 							   && midword == NULL)
23779ccfebddSBram Moolenaar 	    {
23789ccfebddSBram Moolenaar 		midword = getroom_save(spin, items[1]);
23799ccfebddSBram Moolenaar 	    }
23809ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "TRY", 2))
23819ccfebddSBram Moolenaar 	    {
23820d6f5d97SBram Moolenaar 		// ignored, we look in the tree for what chars may appear
23839ccfebddSBram Moolenaar 	    }
23840d6f5d97SBram Moolenaar 	    // TODO: remove "RAR" later
23859ccfebddSBram Moolenaar 	    else if ((is_aff_rule(items, itemcnt, "RAR", 2)
23869ccfebddSBram Moolenaar 			|| is_aff_rule(items, itemcnt, "RARE", 2))
23879ccfebddSBram Moolenaar 							 && aff->af_rare == 0)
23889ccfebddSBram Moolenaar 	    {
23899ccfebddSBram Moolenaar 		aff->af_rare = affitem2flag(aff->af_flagtype, items[1],
23909ccfebddSBram Moolenaar 								 fname, lnum);
23919ccfebddSBram Moolenaar 	    }
23920d6f5d97SBram Moolenaar 	    // TODO: remove "KEP" later
23939ccfebddSBram Moolenaar 	    else if ((is_aff_rule(items, itemcnt, "KEP", 2)
23949ccfebddSBram Moolenaar 			|| is_aff_rule(items, itemcnt, "KEEPCASE", 2))
23959ccfebddSBram Moolenaar 						     && aff->af_keepcase == 0)
23969ccfebddSBram Moolenaar 	    {
23979ccfebddSBram Moolenaar 		aff->af_keepcase = affitem2flag(aff->af_flagtype, items[1],
23989ccfebddSBram Moolenaar 								 fname, lnum);
23999ccfebddSBram Moolenaar 	    }
24009ccfebddSBram Moolenaar 	    else if ((is_aff_rule(items, itemcnt, "BAD", 2)
24019ccfebddSBram Moolenaar 			|| is_aff_rule(items, itemcnt, "FORBIDDENWORD", 2))
24029ccfebddSBram Moolenaar 							  && aff->af_bad == 0)
24039ccfebddSBram Moolenaar 	    {
24049ccfebddSBram Moolenaar 		aff->af_bad = affitem2flag(aff->af_flagtype, items[1],
24059ccfebddSBram Moolenaar 								 fname, lnum);
24069ccfebddSBram Moolenaar 	    }
24079ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "NEEDAFFIX", 2)
24089ccfebddSBram Moolenaar 						    && aff->af_needaffix == 0)
24099ccfebddSBram Moolenaar 	    {
24109ccfebddSBram Moolenaar 		aff->af_needaffix = affitem2flag(aff->af_flagtype, items[1],
24119ccfebddSBram Moolenaar 								 fname, lnum);
24129ccfebddSBram Moolenaar 	    }
24139ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "CIRCUMFIX", 2)
24149ccfebddSBram Moolenaar 						    && aff->af_circumfix == 0)
24159ccfebddSBram Moolenaar 	    {
24169ccfebddSBram Moolenaar 		aff->af_circumfix = affitem2flag(aff->af_flagtype, items[1],
24179ccfebddSBram Moolenaar 								 fname, lnum);
24189ccfebddSBram Moolenaar 	    }
24199ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "NOSUGGEST", 2)
24209ccfebddSBram Moolenaar 						    && aff->af_nosuggest == 0)
24219ccfebddSBram Moolenaar 	    {
24229ccfebddSBram Moolenaar 		aff->af_nosuggest = affitem2flag(aff->af_flagtype, items[1],
24239ccfebddSBram Moolenaar 								 fname, lnum);
24249ccfebddSBram Moolenaar 	    }
24259ccfebddSBram Moolenaar 	    else if ((is_aff_rule(items, itemcnt, "NEEDCOMPOUND", 2)
24269ccfebddSBram Moolenaar 			|| is_aff_rule(items, itemcnt, "ONLYINCOMPOUND", 2))
24279ccfebddSBram Moolenaar 						     && aff->af_needcomp == 0)
24289ccfebddSBram Moolenaar 	    {
24299ccfebddSBram Moolenaar 		aff->af_needcomp = affitem2flag(aff->af_flagtype, items[1],
24309ccfebddSBram Moolenaar 								 fname, lnum);
24319ccfebddSBram Moolenaar 	    }
24329ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDROOT", 2)
24339ccfebddSBram Moolenaar 						     && aff->af_comproot == 0)
24349ccfebddSBram Moolenaar 	    {
24359ccfebddSBram Moolenaar 		aff->af_comproot = affitem2flag(aff->af_flagtype, items[1],
24369ccfebddSBram Moolenaar 								 fname, lnum);
24379ccfebddSBram Moolenaar 	    }
24389ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDFORBIDFLAG", 2)
24399ccfebddSBram Moolenaar 						   && aff->af_compforbid == 0)
24409ccfebddSBram Moolenaar 	    {
24419ccfebddSBram Moolenaar 		aff->af_compforbid = affitem2flag(aff->af_flagtype, items[1],
24429ccfebddSBram Moolenaar 								 fname, lnum);
24439ccfebddSBram Moolenaar 		if (aff->af_pref.ht_used > 0)
2444f9e3e09fSBram Moolenaar 		    smsg(_("Defining COMPOUNDFORBIDFLAG after PFX item may give wrong results in %s line %d"),
24459ccfebddSBram Moolenaar 			    fname, lnum);
24469ccfebddSBram Moolenaar 	    }
24479ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDPERMITFLAG", 2)
24489ccfebddSBram Moolenaar 						   && aff->af_comppermit == 0)
24499ccfebddSBram Moolenaar 	    {
24509ccfebddSBram Moolenaar 		aff->af_comppermit = affitem2flag(aff->af_flagtype, items[1],
24519ccfebddSBram Moolenaar 								 fname, lnum);
24529ccfebddSBram Moolenaar 		if (aff->af_pref.ht_used > 0)
2453f9e3e09fSBram Moolenaar 		    smsg(_("Defining COMPOUNDPERMITFLAG after PFX item may give wrong results in %s line %d"),
24549ccfebddSBram Moolenaar 			    fname, lnum);
24559ccfebddSBram Moolenaar 	    }
24569ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDFLAG", 2)
24579ccfebddSBram Moolenaar 							 && compflags == NULL)
24589ccfebddSBram Moolenaar 	    {
24590d6f5d97SBram Moolenaar 		// Turn flag "c" into COMPOUNDRULE compatible string "c+",
24600d6f5d97SBram Moolenaar 		// "Na" into "Na+", "1234" into "1234+".
24619ccfebddSBram Moolenaar 		p = getroom(spin, STRLEN(items[1]) + 2, FALSE);
24629ccfebddSBram Moolenaar 		if (p != NULL)
24639ccfebddSBram Moolenaar 		{
24649ccfebddSBram Moolenaar 		    STRCPY(p, items[1]);
24659ccfebddSBram Moolenaar 		    STRCAT(p, "+");
24669ccfebddSBram Moolenaar 		    compflags = p;
24679ccfebddSBram Moolenaar 		}
24689ccfebddSBram Moolenaar 	    }
24699ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDRULES", 2))
24709ccfebddSBram Moolenaar 	    {
24710d6f5d97SBram Moolenaar 		// We don't use the count, but do check that it's a number and
24720d6f5d97SBram Moolenaar 		// not COMPOUNDRULE mistyped.
24739ccfebddSBram Moolenaar 		if (atoi((char *)items[1]) == 0)
2474f9e3e09fSBram Moolenaar 		    smsg(_("Wrong COMPOUNDRULES value in %s line %d: %s"),
24759ccfebddSBram Moolenaar 						       fname, lnum, items[1]);
24769ccfebddSBram Moolenaar 	    }
24779ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDRULE", 2))
24789ccfebddSBram Moolenaar 	    {
24790d6f5d97SBram Moolenaar 		// Don't use the first rule if it is a number.
24809ccfebddSBram Moolenaar 		if (compflags != NULL || *skipdigits(items[1]) != NUL)
24819ccfebddSBram Moolenaar 		{
24820d6f5d97SBram Moolenaar 		    // Concatenate this string to previously defined ones,
24830d6f5d97SBram Moolenaar 		    // using a slash to separate them.
24849ccfebddSBram Moolenaar 		    l = (int)STRLEN(items[1]) + 1;
24859ccfebddSBram Moolenaar 		    if (compflags != NULL)
24869ccfebddSBram Moolenaar 			l += (int)STRLEN(compflags) + 1;
24879ccfebddSBram Moolenaar 		    p = getroom(spin, l, FALSE);
24889ccfebddSBram Moolenaar 		    if (p != NULL)
24899ccfebddSBram Moolenaar 		    {
24909ccfebddSBram Moolenaar 			if (compflags != NULL)
24919ccfebddSBram Moolenaar 			{
24929ccfebddSBram Moolenaar 			    STRCPY(p, compflags);
24939ccfebddSBram Moolenaar 			    STRCAT(p, "/");
24949ccfebddSBram Moolenaar 			}
24959ccfebddSBram Moolenaar 			STRCAT(p, items[1]);
24969ccfebddSBram Moolenaar 			compflags = p;
24979ccfebddSBram Moolenaar 		    }
24989ccfebddSBram Moolenaar 		}
24999ccfebddSBram Moolenaar 	    }
25009ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDWORDMAX", 2)
25019ccfebddSBram Moolenaar 							      && compmax == 0)
25029ccfebddSBram Moolenaar 	    {
25039ccfebddSBram Moolenaar 		compmax = atoi((char *)items[1]);
25049ccfebddSBram Moolenaar 		if (compmax == 0)
2505f9e3e09fSBram Moolenaar 		    smsg(_("Wrong COMPOUNDWORDMAX value in %s line %d: %s"),
25069ccfebddSBram Moolenaar 						       fname, lnum, items[1]);
25079ccfebddSBram Moolenaar 	    }
25089ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDMIN", 2)
25099ccfebddSBram Moolenaar 							   && compminlen == 0)
25109ccfebddSBram Moolenaar 	    {
25119ccfebddSBram Moolenaar 		compminlen = atoi((char *)items[1]);
25129ccfebddSBram Moolenaar 		if (compminlen == 0)
2513f9e3e09fSBram Moolenaar 		    smsg(_("Wrong COMPOUNDMIN value in %s line %d: %s"),
25149ccfebddSBram Moolenaar 						       fname, lnum, items[1]);
25159ccfebddSBram Moolenaar 	    }
25169ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "COMPOUNDSYLMAX", 2)
25179ccfebddSBram Moolenaar 							   && compsylmax == 0)
25189ccfebddSBram Moolenaar 	    {
25199ccfebddSBram Moolenaar 		compsylmax = atoi((char *)items[1]);
25209ccfebddSBram Moolenaar 		if (compsylmax == 0)
2521f9e3e09fSBram Moolenaar 		    smsg(_("Wrong COMPOUNDSYLMAX value in %s line %d: %s"),
25229ccfebddSBram Moolenaar 						       fname, lnum, items[1]);
25239ccfebddSBram Moolenaar 	    }
25249ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "CHECKCOMPOUNDDUP", 1))
25259ccfebddSBram Moolenaar 	    {
25269ccfebddSBram Moolenaar 		compoptions |= COMP_CHECKDUP;
25279ccfebddSBram Moolenaar 	    }
25289ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "CHECKCOMPOUNDREP", 1))
25299ccfebddSBram Moolenaar 	    {
25309ccfebddSBram Moolenaar 		compoptions |= COMP_CHECKREP;
25319ccfebddSBram Moolenaar 	    }
25329ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "CHECKCOMPOUNDCASE", 1))
25339ccfebddSBram Moolenaar 	    {
25349ccfebddSBram Moolenaar 		compoptions |= COMP_CHECKCASE;
25359ccfebddSBram Moolenaar 	    }
25369ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "CHECKCOMPOUNDTRIPLE", 1))
25379ccfebddSBram Moolenaar 	    {
25389ccfebddSBram Moolenaar 		compoptions |= COMP_CHECKTRIPLE;
25399ccfebddSBram Moolenaar 	    }
25409ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "CHECKCOMPOUNDPATTERN", 2))
25419ccfebddSBram Moolenaar 	    {
25429ccfebddSBram Moolenaar 		if (atoi((char *)items[1]) == 0)
2543f9e3e09fSBram Moolenaar 		    smsg(_("Wrong CHECKCOMPOUNDPATTERN value in %s line %d: %s"),
25449ccfebddSBram Moolenaar 						       fname, lnum, items[1]);
25459ccfebddSBram Moolenaar 	    }
25469ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "CHECKCOMPOUNDPATTERN", 3))
25479ccfebddSBram Moolenaar 	    {
25489ccfebddSBram Moolenaar 		garray_T    *gap = &spin->si_comppat;
25499ccfebddSBram Moolenaar 		int	    i;
25509ccfebddSBram Moolenaar 
25510d6f5d97SBram Moolenaar 		// Only add the couple if it isn't already there.
25529ccfebddSBram Moolenaar 		for (i = 0; i < gap->ga_len - 1; i += 2)
25539ccfebddSBram Moolenaar 		    if (STRCMP(((char_u **)(gap->ga_data))[i], items[1]) == 0
25549ccfebddSBram Moolenaar 			    && STRCMP(((char_u **)(gap->ga_data))[i + 1],
25559ccfebddSBram Moolenaar 							       items[2]) == 0)
25569ccfebddSBram Moolenaar 			break;
25579ccfebddSBram Moolenaar 		if (i >= gap->ga_len && ga_grow(gap, 2) == OK)
25589ccfebddSBram Moolenaar 		{
25599ccfebddSBram Moolenaar 		    ((char_u **)(gap->ga_data))[gap->ga_len++]
25609ccfebddSBram Moolenaar 					       = getroom_save(spin, items[1]);
25619ccfebddSBram Moolenaar 		    ((char_u **)(gap->ga_data))[gap->ga_len++]
25629ccfebddSBram Moolenaar 					       = getroom_save(spin, items[2]);
25639ccfebddSBram Moolenaar 		}
25649ccfebddSBram Moolenaar 	    }
25659ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "SYLLABLE", 2)
25669ccfebddSBram Moolenaar 							  && syllable == NULL)
25679ccfebddSBram Moolenaar 	    {
25689ccfebddSBram Moolenaar 		syllable = getroom_save(spin, items[1]);
25699ccfebddSBram Moolenaar 	    }
25709ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "NOBREAK", 1))
25719ccfebddSBram Moolenaar 	    {
25729ccfebddSBram Moolenaar 		spin->si_nobreak = TRUE;
25739ccfebddSBram Moolenaar 	    }
25749ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "NOSPLITSUGS", 1))
25759ccfebddSBram Moolenaar 	    {
25769ccfebddSBram Moolenaar 		spin->si_nosplitsugs = TRUE;
25779ccfebddSBram Moolenaar 	    }
25789ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "NOCOMPOUNDSUGS", 1))
25799ccfebddSBram Moolenaar 	    {
25809ccfebddSBram Moolenaar 		spin->si_nocompoundsugs = TRUE;
25819ccfebddSBram Moolenaar 	    }
25829ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "NOSUGFILE", 1))
25839ccfebddSBram Moolenaar 	    {
25849ccfebddSBram Moolenaar 		spin->si_nosugfile = TRUE;
25859ccfebddSBram Moolenaar 	    }
25869ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "PFXPOSTPONE", 1))
25879ccfebddSBram Moolenaar 	    {
25889ccfebddSBram Moolenaar 		aff->af_pfxpostpone = TRUE;
25899ccfebddSBram Moolenaar 	    }
25909ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "IGNOREEXTRA", 1))
25919ccfebddSBram Moolenaar 	    {
25929ccfebddSBram Moolenaar 		aff->af_ignoreextra = TRUE;
25939ccfebddSBram Moolenaar 	    }
25949ccfebddSBram Moolenaar 	    else if ((STRCMP(items[0], "PFX") == 0
25959ccfebddSBram Moolenaar 					      || STRCMP(items[0], "SFX") == 0)
25969ccfebddSBram Moolenaar 		    && aff_todo == 0
25979ccfebddSBram Moolenaar 		    && itemcnt >= 4)
25989ccfebddSBram Moolenaar 	    {
25999ccfebddSBram Moolenaar 		int	lasti = 4;
26009ccfebddSBram Moolenaar 		char_u	key[AH_KEY_LEN];
26019ccfebddSBram Moolenaar 
26029ccfebddSBram Moolenaar 		if (*items[0] == 'P')
26039ccfebddSBram Moolenaar 		    tp = &aff->af_pref;
26049ccfebddSBram Moolenaar 		else
26059ccfebddSBram Moolenaar 		    tp = &aff->af_suff;
26069ccfebddSBram Moolenaar 
26070d6f5d97SBram Moolenaar 		// Myspell allows the same affix name to be used multiple
26080d6f5d97SBram Moolenaar 		// times.  The affix files that do this have an undocumented
26090d6f5d97SBram Moolenaar 		// "S" flag on all but the last block, thus we check for that
26100d6f5d97SBram Moolenaar 		// and store it in ah_follows.
26119ccfebddSBram Moolenaar 		vim_strncpy(key, items[1], AH_KEY_LEN - 1);
26129ccfebddSBram Moolenaar 		hi = hash_find(tp, key);
26139ccfebddSBram Moolenaar 		if (!HASHITEM_EMPTY(hi))
26149ccfebddSBram Moolenaar 		{
26159ccfebddSBram Moolenaar 		    cur_aff = HI2AH(hi);
26169ccfebddSBram Moolenaar 		    if (cur_aff->ah_combine != (*items[2] == 'Y'))
2617f9e3e09fSBram Moolenaar 			smsg(_("Different combining flag in continued affix block in %s line %d: %s"),
26189ccfebddSBram Moolenaar 						   fname, lnum, items[1]);
26199ccfebddSBram Moolenaar 		    if (!cur_aff->ah_follows)
2620f9e3e09fSBram Moolenaar 			smsg(_("Duplicate affix in %s line %d: %s"),
26219ccfebddSBram Moolenaar 						       fname, lnum, items[1]);
26229ccfebddSBram Moolenaar 		}
26239ccfebddSBram Moolenaar 		else
26249ccfebddSBram Moolenaar 		{
26250d6f5d97SBram Moolenaar 		    // New affix letter.
26269ccfebddSBram Moolenaar 		    cur_aff = (affheader_T *)getroom(spin,
26279ccfebddSBram Moolenaar 						   sizeof(affheader_T), TRUE);
26289ccfebddSBram Moolenaar 		    if (cur_aff == NULL)
26299ccfebddSBram Moolenaar 			break;
26309ccfebddSBram Moolenaar 		    cur_aff->ah_flag = affitem2flag(aff->af_flagtype, items[1],
26319ccfebddSBram Moolenaar 								 fname, lnum);
26329ccfebddSBram Moolenaar 		    if (cur_aff->ah_flag == 0 || STRLEN(items[1]) >= AH_KEY_LEN)
26339ccfebddSBram Moolenaar 			break;
26349ccfebddSBram Moolenaar 		    if (cur_aff->ah_flag == aff->af_bad
26359ccfebddSBram Moolenaar 			    || cur_aff->ah_flag == aff->af_rare
26369ccfebddSBram Moolenaar 			    || cur_aff->ah_flag == aff->af_keepcase
26379ccfebddSBram Moolenaar 			    || cur_aff->ah_flag == aff->af_needaffix
26389ccfebddSBram Moolenaar 			    || cur_aff->ah_flag == aff->af_circumfix
26399ccfebddSBram Moolenaar 			    || cur_aff->ah_flag == aff->af_nosuggest
26409ccfebddSBram Moolenaar 			    || cur_aff->ah_flag == aff->af_needcomp
26419ccfebddSBram Moolenaar 			    || cur_aff->ah_flag == aff->af_comproot)
2642f9e3e09fSBram Moolenaar 			smsg(_("Affix also used for BAD/RARE/KEEPCASE/NEEDAFFIX/NEEDCOMPOUND/NOSUGGEST in %s line %d: %s"),
26439ccfebddSBram Moolenaar 						       fname, lnum, items[1]);
26449ccfebddSBram Moolenaar 		    STRCPY(cur_aff->ah_key, items[1]);
26459ccfebddSBram Moolenaar 		    hash_add(tp, cur_aff->ah_key);
26469ccfebddSBram Moolenaar 
26479ccfebddSBram Moolenaar 		    cur_aff->ah_combine = (*items[2] == 'Y');
26489ccfebddSBram Moolenaar 		}
26499ccfebddSBram Moolenaar 
26500d6f5d97SBram Moolenaar 		// Check for the "S" flag, which apparently means that another
26510d6f5d97SBram Moolenaar 		// block with the same affix name is following.
26529ccfebddSBram Moolenaar 		if (itemcnt > lasti && STRCMP(items[lasti], "S") == 0)
26539ccfebddSBram Moolenaar 		{
26549ccfebddSBram Moolenaar 		    ++lasti;
26559ccfebddSBram Moolenaar 		    cur_aff->ah_follows = TRUE;
26569ccfebddSBram Moolenaar 		}
26579ccfebddSBram Moolenaar 		else
26589ccfebddSBram Moolenaar 		    cur_aff->ah_follows = FALSE;
26599ccfebddSBram Moolenaar 
26600d6f5d97SBram Moolenaar 		// Myspell allows extra text after the item, but that might
26610d6f5d97SBram Moolenaar 		// mean mistakes go unnoticed.  Require a comment-starter.
26629ccfebddSBram Moolenaar 		if (itemcnt > lasti && *items[lasti] != '#')
2663f9e3e09fSBram Moolenaar 		    smsg(_(e_afftrailing), fname, lnum, items[lasti]);
26649ccfebddSBram Moolenaar 
26659ccfebddSBram Moolenaar 		if (STRCMP(items[2], "Y") != 0 && STRCMP(items[2], "N") != 0)
2666f9e3e09fSBram Moolenaar 		    smsg(_("Expected Y or N in %s line %d: %s"),
26679ccfebddSBram Moolenaar 						       fname, lnum, items[2]);
26689ccfebddSBram Moolenaar 
26699ccfebddSBram Moolenaar 		if (*items[0] == 'P' && aff->af_pfxpostpone)
26709ccfebddSBram Moolenaar 		{
26719ccfebddSBram Moolenaar 		    if (cur_aff->ah_newID == 0)
26729ccfebddSBram Moolenaar 		    {
26730d6f5d97SBram Moolenaar 			// Use a new number in the .spl file later, to be able
26740d6f5d97SBram Moolenaar 			// to handle multiple .aff files.
26759ccfebddSBram Moolenaar 			check_renumber(spin);
26769ccfebddSBram Moolenaar 			cur_aff->ah_newID = ++spin->si_newprefID;
26779ccfebddSBram Moolenaar 
26780d6f5d97SBram Moolenaar 			// We only really use ah_newID if the prefix is
26790d6f5d97SBram Moolenaar 			// postponed.  We know that only after handling all
26800d6f5d97SBram Moolenaar 			// the items.
26819ccfebddSBram Moolenaar 			did_postpone_prefix = FALSE;
26829ccfebddSBram Moolenaar 		    }
26839ccfebddSBram Moolenaar 		    else
26840d6f5d97SBram Moolenaar 			// Did use the ID in a previous block.
26859ccfebddSBram Moolenaar 			did_postpone_prefix = TRUE;
26869ccfebddSBram Moolenaar 		}
26879ccfebddSBram Moolenaar 
26889ccfebddSBram Moolenaar 		aff_todo = atoi((char *)items[3]);
26899ccfebddSBram Moolenaar 	    }
26909ccfebddSBram Moolenaar 	    else if ((STRCMP(items[0], "PFX") == 0
26919ccfebddSBram Moolenaar 					      || STRCMP(items[0], "SFX") == 0)
26929ccfebddSBram Moolenaar 		    && aff_todo > 0
26939ccfebddSBram Moolenaar 		    && STRCMP(cur_aff->ah_key, items[1]) == 0
26949ccfebddSBram Moolenaar 		    && itemcnt >= 5)
26959ccfebddSBram Moolenaar 	    {
26969ccfebddSBram Moolenaar 		affentry_T	*aff_entry;
26979ccfebddSBram Moolenaar 		int		upper = FALSE;
26989ccfebddSBram Moolenaar 		int		lasti = 5;
26999ccfebddSBram Moolenaar 
27000d6f5d97SBram Moolenaar 		// Myspell allows extra text after the item, but that might
27010d6f5d97SBram Moolenaar 		// mean mistakes go unnoticed.  Require a comment-starter,
27020d6f5d97SBram Moolenaar 		// unless IGNOREEXTRA is used.  Hunspell uses a "-" item.
27039ccfebddSBram Moolenaar 		if (itemcnt > lasti
27049ccfebddSBram Moolenaar 			&& !aff->af_ignoreextra
27059ccfebddSBram Moolenaar 			&& *items[lasti] != '#'
27069ccfebddSBram Moolenaar 			&& (STRCMP(items[lasti], "-") != 0
27079ccfebddSBram Moolenaar 						     || itemcnt != lasti + 1))
2708f9e3e09fSBram Moolenaar 		    smsg(_(e_afftrailing), fname, lnum, items[lasti]);
27099ccfebddSBram Moolenaar 
27100d6f5d97SBram Moolenaar 		// New item for an affix letter.
27119ccfebddSBram Moolenaar 		--aff_todo;
27129ccfebddSBram Moolenaar 		aff_entry = (affentry_T *)getroom(spin,
27139ccfebddSBram Moolenaar 						    sizeof(affentry_T), TRUE);
27149ccfebddSBram Moolenaar 		if (aff_entry == NULL)
27159ccfebddSBram Moolenaar 		    break;
27169ccfebddSBram Moolenaar 
27179ccfebddSBram Moolenaar 		if (STRCMP(items[2], "0") != 0)
27189ccfebddSBram Moolenaar 		    aff_entry->ae_chop = getroom_save(spin, items[2]);
27199ccfebddSBram Moolenaar 		if (STRCMP(items[3], "0") != 0)
27209ccfebddSBram Moolenaar 		{
27219ccfebddSBram Moolenaar 		    aff_entry->ae_add = getroom_save(spin, items[3]);
27229ccfebddSBram Moolenaar 
27230d6f5d97SBram Moolenaar 		    // Recognize flags on the affix: abcd/XYZ
27249ccfebddSBram Moolenaar 		    aff_entry->ae_flags = vim_strchr(aff_entry->ae_add, '/');
27259ccfebddSBram Moolenaar 		    if (aff_entry->ae_flags != NULL)
27269ccfebddSBram Moolenaar 		    {
27279ccfebddSBram Moolenaar 			*aff_entry->ae_flags++ = NUL;
27289ccfebddSBram Moolenaar 			aff_process_flags(aff, aff_entry);
27299ccfebddSBram Moolenaar 		    }
27309ccfebddSBram Moolenaar 		}
27319ccfebddSBram Moolenaar 
27320d6f5d97SBram Moolenaar 		// Don't use an affix entry with non-ASCII characters when
27330d6f5d97SBram Moolenaar 		// "spin->si_ascii" is TRUE.
27349ccfebddSBram Moolenaar 		if (!spin->si_ascii || !(has_non_ascii(aff_entry->ae_chop)
27359ccfebddSBram Moolenaar 					  || has_non_ascii(aff_entry->ae_add)))
27369ccfebddSBram Moolenaar 		{
27379ccfebddSBram Moolenaar 		    aff_entry->ae_next = cur_aff->ah_first;
27389ccfebddSBram Moolenaar 		    cur_aff->ah_first = aff_entry;
27399ccfebddSBram Moolenaar 
27409ccfebddSBram Moolenaar 		    if (STRCMP(items[4], ".") != 0)
27419ccfebddSBram Moolenaar 		    {
27429ccfebddSBram Moolenaar 			char_u	buf[MAXLINELEN];
27439ccfebddSBram Moolenaar 
27449ccfebddSBram Moolenaar 			aff_entry->ae_cond = getroom_save(spin, items[4]);
27459ccfebddSBram Moolenaar 			if (*items[0] == 'P')
27469ccfebddSBram Moolenaar 			    sprintf((char *)buf, "^%s", items[4]);
27479ccfebddSBram Moolenaar 			else
27489ccfebddSBram Moolenaar 			    sprintf((char *)buf, "%s$", items[4]);
27499ccfebddSBram Moolenaar 			aff_entry->ae_prog = vim_regcomp(buf,
27509ccfebddSBram Moolenaar 					    RE_MAGIC + RE_STRING + RE_STRICT);
27519ccfebddSBram Moolenaar 			if (aff_entry->ae_prog == NULL)
2752f9e3e09fSBram Moolenaar 			    smsg(_("Broken condition in %s line %d: %s"),
27539ccfebddSBram Moolenaar 						       fname, lnum, items[4]);
27549ccfebddSBram Moolenaar 		    }
27559ccfebddSBram Moolenaar 
27560d6f5d97SBram Moolenaar 		    // For postponed prefixes we need an entry in si_prefcond
27570d6f5d97SBram Moolenaar 		    // for the condition.  Use an existing one if possible.
27580d6f5d97SBram Moolenaar 		    // Can't be done for an affix with flags, ignoring
27590d6f5d97SBram Moolenaar 		    // COMPOUNDFORBIDFLAG and COMPOUNDPERMITFLAG.
27609ccfebddSBram Moolenaar 		    if (*items[0] == 'P' && aff->af_pfxpostpone
27619ccfebddSBram Moolenaar 					       && aff_entry->ae_flags == NULL)
27629ccfebddSBram Moolenaar 		    {
27630d6f5d97SBram Moolenaar 			// When the chop string is one lower-case letter and
27640d6f5d97SBram Moolenaar 			// the add string ends in the upper-case letter we set
27650d6f5d97SBram Moolenaar 			// the "upper" flag, clear "ae_chop" and remove the
27660d6f5d97SBram Moolenaar 			// letters from "ae_add".  The condition must either
27670d6f5d97SBram Moolenaar 			// be empty or start with the same letter.
27689ccfebddSBram Moolenaar 			if (aff_entry->ae_chop != NULL
27699ccfebddSBram Moolenaar 				&& aff_entry->ae_add != NULL
27709ccfebddSBram Moolenaar 				&& aff_entry->ae_chop[(*mb_ptr2len)(
2771264b74faSBram Moolenaar 						   aff_entry->ae_chop)] == NUL)
27729ccfebddSBram Moolenaar 			{
27739ccfebddSBram Moolenaar 			    int		c, c_up;
27749ccfebddSBram Moolenaar 
27759ccfebddSBram Moolenaar 			    c = PTR2CHAR(aff_entry->ae_chop);
27769ccfebddSBram Moolenaar 			    c_up = SPELL_TOUPPER(c);
27779ccfebddSBram Moolenaar 			    if (c_up != c
27789ccfebddSBram Moolenaar 				    && (aff_entry->ae_cond == NULL
27799ccfebddSBram Moolenaar 					|| PTR2CHAR(aff_entry->ae_cond) == c))
27809ccfebddSBram Moolenaar 			    {
27819ccfebddSBram Moolenaar 				p = aff_entry->ae_add
27829ccfebddSBram Moolenaar 						  + STRLEN(aff_entry->ae_add);
278391acfffcSBram Moolenaar 				MB_PTR_BACK(aff_entry->ae_add, p);
27849ccfebddSBram Moolenaar 				if (PTR2CHAR(p) == c_up)
27859ccfebddSBram Moolenaar 				{
27869ccfebddSBram Moolenaar 				    upper = TRUE;
27879ccfebddSBram Moolenaar 				    aff_entry->ae_chop = NULL;
27889ccfebddSBram Moolenaar 				    *p = NUL;
27899ccfebddSBram Moolenaar 
27900d6f5d97SBram Moolenaar 				    // The condition is matched with the
27910d6f5d97SBram Moolenaar 				    // actual word, thus must check for the
27920d6f5d97SBram Moolenaar 				    // upper-case letter.
27939ccfebddSBram Moolenaar 				    if (aff_entry->ae_cond != NULL)
27949ccfebddSBram Moolenaar 				    {
27959ccfebddSBram Moolenaar 					char_u	buf[MAXLINELEN];
2796264b74faSBram Moolenaar 
27979ccfebddSBram Moolenaar 					if (has_mbyte)
27989ccfebddSBram Moolenaar 					{
27999ccfebddSBram Moolenaar 					    onecap_copy(items[4], buf, TRUE);
28009ccfebddSBram Moolenaar 					    aff_entry->ae_cond = getroom_save(
28019ccfebddSBram Moolenaar 								   spin, buf);
28029ccfebddSBram Moolenaar 					}
28039ccfebddSBram Moolenaar 					else
28049ccfebddSBram Moolenaar 					    *aff_entry->ae_cond = c_up;
28059ccfebddSBram Moolenaar 					if (aff_entry->ae_cond != NULL)
28069ccfebddSBram Moolenaar 					{
28079ccfebddSBram Moolenaar 					    sprintf((char *)buf, "^%s",
28089ccfebddSBram Moolenaar 							  aff_entry->ae_cond);
28099ccfebddSBram Moolenaar 					    vim_regfree(aff_entry->ae_prog);
28109ccfebddSBram Moolenaar 					    aff_entry->ae_prog = vim_regcomp(
28119ccfebddSBram Moolenaar 						    buf, RE_MAGIC + RE_STRING);
28129ccfebddSBram Moolenaar 					}
28139ccfebddSBram Moolenaar 				    }
28149ccfebddSBram Moolenaar 				}
28159ccfebddSBram Moolenaar 			    }
28169ccfebddSBram Moolenaar 			}
28179ccfebddSBram Moolenaar 
28189ccfebddSBram Moolenaar 			if (aff_entry->ae_chop == NULL
28199ccfebddSBram Moolenaar 					       && aff_entry->ae_flags == NULL)
28209ccfebddSBram Moolenaar 			{
28219ccfebddSBram Moolenaar 			    int		idx;
28229ccfebddSBram Moolenaar 			    char_u	**pp;
28239ccfebddSBram Moolenaar 			    int		n;
28249ccfebddSBram Moolenaar 
28250d6f5d97SBram Moolenaar 			    // Find a previously used condition.
28269ccfebddSBram Moolenaar 			    for (idx = spin->si_prefcond.ga_len - 1; idx >= 0;
28279ccfebddSBram Moolenaar 									--idx)
28289ccfebddSBram Moolenaar 			    {
28299ccfebddSBram Moolenaar 				p = ((char_u **)spin->si_prefcond.ga_data)[idx];
28309ccfebddSBram Moolenaar 				if (str_equal(p, aff_entry->ae_cond))
28319ccfebddSBram Moolenaar 				    break;
28329ccfebddSBram Moolenaar 			    }
28339ccfebddSBram Moolenaar 			    if (idx < 0 && ga_grow(&spin->si_prefcond, 1) == OK)
28349ccfebddSBram Moolenaar 			    {
28350d6f5d97SBram Moolenaar 				// Not found, add a new condition.
28369ccfebddSBram Moolenaar 				idx = spin->si_prefcond.ga_len++;
28379ccfebddSBram Moolenaar 				pp = ((char_u **)spin->si_prefcond.ga_data)
28389ccfebddSBram Moolenaar 									+ idx;
28399ccfebddSBram Moolenaar 				if (aff_entry->ae_cond == NULL)
28409ccfebddSBram Moolenaar 				    *pp = NULL;
28419ccfebddSBram Moolenaar 				else
28429ccfebddSBram Moolenaar 				    *pp = getroom_save(spin,
28439ccfebddSBram Moolenaar 							  aff_entry->ae_cond);
28449ccfebddSBram Moolenaar 			    }
28459ccfebddSBram Moolenaar 
28460d6f5d97SBram Moolenaar 			    // Add the prefix to the prefix tree.
28479ccfebddSBram Moolenaar 			    if (aff_entry->ae_add == NULL)
28489ccfebddSBram Moolenaar 				p = (char_u *)"";
28499ccfebddSBram Moolenaar 			    else
28509ccfebddSBram Moolenaar 				p = aff_entry->ae_add;
28519ccfebddSBram Moolenaar 
28520d6f5d97SBram Moolenaar 			    // PFX_FLAGS is a negative number, so that
28530d6f5d97SBram Moolenaar 			    // tree_add_word() knows this is the prefix tree.
28549ccfebddSBram Moolenaar 			    n = PFX_FLAGS;
28559ccfebddSBram Moolenaar 			    if (!cur_aff->ah_combine)
28569ccfebddSBram Moolenaar 				n |= WFP_NC;
28579ccfebddSBram Moolenaar 			    if (upper)
28589ccfebddSBram Moolenaar 				n |= WFP_UP;
28599ccfebddSBram Moolenaar 			    if (aff_entry->ae_comppermit)
28609ccfebddSBram Moolenaar 				n |= WFP_COMPPERMIT;
28619ccfebddSBram Moolenaar 			    if (aff_entry->ae_compforbid)
28629ccfebddSBram Moolenaar 				n |= WFP_COMPFORBID;
28639ccfebddSBram Moolenaar 			    tree_add_word(spin, p, spin->si_prefroot, n,
28649ccfebddSBram Moolenaar 						      idx, cur_aff->ah_newID);
28659ccfebddSBram Moolenaar 			    did_postpone_prefix = TRUE;
28669ccfebddSBram Moolenaar 			}
28679ccfebddSBram Moolenaar 
28680d6f5d97SBram Moolenaar 			// Didn't actually use ah_newID, backup si_newprefID.
28699ccfebddSBram Moolenaar 			if (aff_todo == 0 && !did_postpone_prefix)
28709ccfebddSBram Moolenaar 			{
28719ccfebddSBram Moolenaar 			    --spin->si_newprefID;
28729ccfebddSBram Moolenaar 			    cur_aff->ah_newID = 0;
28739ccfebddSBram Moolenaar 			}
28749ccfebddSBram Moolenaar 		    }
28759ccfebddSBram Moolenaar 		}
28769ccfebddSBram Moolenaar 	    }
28779ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "FOL", 2) && fol == NULL)
28789ccfebddSBram Moolenaar 	    {
28799ccfebddSBram Moolenaar 		fol = vim_strsave(items[1]);
28809ccfebddSBram Moolenaar 	    }
28819ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "LOW", 2) && low == NULL)
28829ccfebddSBram Moolenaar 	    {
28839ccfebddSBram Moolenaar 		low = vim_strsave(items[1]);
28849ccfebddSBram Moolenaar 	    }
28859ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "UPP", 2) && upp == NULL)
28869ccfebddSBram Moolenaar 	    {
28879ccfebddSBram Moolenaar 		upp = vim_strsave(items[1]);
28889ccfebddSBram Moolenaar 	    }
28899ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "REP", 2)
28909ccfebddSBram Moolenaar 		     || is_aff_rule(items, itemcnt, "REPSAL", 2))
28919ccfebddSBram Moolenaar 	    {
28920d6f5d97SBram Moolenaar 		// Ignore REP/REPSAL count
28939ccfebddSBram Moolenaar 		if (!isdigit(*items[1]))
2894f9e3e09fSBram Moolenaar 		    smsg(_("Expected REP(SAL) count in %s line %d"),
28959ccfebddSBram Moolenaar 								 fname, lnum);
28969ccfebddSBram Moolenaar 	    }
28979ccfebddSBram Moolenaar 	    else if ((STRCMP(items[0], "REP") == 0
28989ccfebddSBram Moolenaar 			|| STRCMP(items[0], "REPSAL") == 0)
28999ccfebddSBram Moolenaar 		    && itemcnt >= 3)
29009ccfebddSBram Moolenaar 	    {
29010d6f5d97SBram Moolenaar 		// REP/REPSAL item
29020d6f5d97SBram Moolenaar 		// Myspell ignores extra arguments, we require it starts with
29030d6f5d97SBram Moolenaar 		// # to detect mistakes.
29049ccfebddSBram Moolenaar 		if (itemcnt > 3 && items[3][0] != '#')
2905f9e3e09fSBram Moolenaar 		    smsg(_(e_afftrailing), fname, lnum, items[3]);
29069ccfebddSBram Moolenaar 		if (items[0][3] == 'S' ? do_repsal : do_rep)
29079ccfebddSBram Moolenaar 		{
29080d6f5d97SBram Moolenaar 		    // Replace underscore with space (can't include a space
29090d6f5d97SBram Moolenaar 		    // directly).
291091acfffcSBram Moolenaar 		    for (p = items[1]; *p != NUL; MB_PTR_ADV(p))
29119ccfebddSBram Moolenaar 			if (*p == '_')
29129ccfebddSBram Moolenaar 			    *p = ' ';
291391acfffcSBram Moolenaar 		    for (p = items[2]; *p != NUL; MB_PTR_ADV(p))
29149ccfebddSBram Moolenaar 			if (*p == '_')
29159ccfebddSBram Moolenaar 			    *p = ' ';
29169ccfebddSBram Moolenaar 		    add_fromto(spin, items[0][3] == 'S'
29179ccfebddSBram Moolenaar 					 ? &spin->si_repsal
29189ccfebddSBram Moolenaar 					 : &spin->si_rep, items[1], items[2]);
29199ccfebddSBram Moolenaar 		}
29209ccfebddSBram Moolenaar 	    }
29219ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "MAP", 2))
29229ccfebddSBram Moolenaar 	    {
29230d6f5d97SBram Moolenaar 		// MAP item or count
29249ccfebddSBram Moolenaar 		if (!found_map)
29259ccfebddSBram Moolenaar 		{
29260d6f5d97SBram Moolenaar 		    // First line contains the count.
29279ccfebddSBram Moolenaar 		    found_map = TRUE;
29289ccfebddSBram Moolenaar 		    if (!isdigit(*items[1]))
2929f9e3e09fSBram Moolenaar 			smsg(_("Expected MAP count in %s line %d"),
29309ccfebddSBram Moolenaar 								 fname, lnum);
29319ccfebddSBram Moolenaar 		}
29329ccfebddSBram Moolenaar 		else if (do_mapline)
29339ccfebddSBram Moolenaar 		{
29349ccfebddSBram Moolenaar 		    int		c;
29359ccfebddSBram Moolenaar 
29360d6f5d97SBram Moolenaar 		    // Check that every character appears only once.
29379ccfebddSBram Moolenaar 		    for (p = items[1]; *p != NUL; )
29389ccfebddSBram Moolenaar 		    {
29399ccfebddSBram Moolenaar 			c = mb_ptr2char_adv(&p);
29409ccfebddSBram Moolenaar 			if ((spin->si_map.ga_len > 0
29419ccfebddSBram Moolenaar 				    && vim_strchr(spin->si_map.ga_data, c)
29429ccfebddSBram Moolenaar 								      != NULL)
29439ccfebddSBram Moolenaar 				|| vim_strchr(p, c) != NULL)
2944f9e3e09fSBram Moolenaar 			    smsg(_("Duplicate character in MAP in %s line %d"),
29459ccfebddSBram Moolenaar 								 fname, lnum);
29469ccfebddSBram Moolenaar 		    }
29479ccfebddSBram Moolenaar 
29480d6f5d97SBram Moolenaar 		    // We simply concatenate all the MAP strings, separated by
29490d6f5d97SBram Moolenaar 		    // slashes.
29509ccfebddSBram Moolenaar 		    ga_concat(&spin->si_map, items[1]);
29519ccfebddSBram Moolenaar 		    ga_append(&spin->si_map, '/');
29529ccfebddSBram Moolenaar 		}
29539ccfebddSBram Moolenaar 	    }
29540d6f5d97SBram Moolenaar 	    // Accept "SAL from to" and "SAL from to  #comment".
29559ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "SAL", 3))
29569ccfebddSBram Moolenaar 	    {
29579ccfebddSBram Moolenaar 		if (do_sal)
29589ccfebddSBram Moolenaar 		{
29590d6f5d97SBram Moolenaar 		    // SAL item (sounds-a-like)
29600d6f5d97SBram Moolenaar 		    // Either one of the known keys or a from-to pair.
29619ccfebddSBram Moolenaar 		    if (STRCMP(items[1], "followup") == 0)
29629ccfebddSBram Moolenaar 			spin->si_followup = sal_to_bool(items[2]);
29639ccfebddSBram Moolenaar 		    else if (STRCMP(items[1], "collapse_result") == 0)
29649ccfebddSBram Moolenaar 			spin->si_collapse = sal_to_bool(items[2]);
29659ccfebddSBram Moolenaar 		    else if (STRCMP(items[1], "remove_accents") == 0)
29669ccfebddSBram Moolenaar 			spin->si_rem_accents = sal_to_bool(items[2]);
29679ccfebddSBram Moolenaar 		    else
29680d6f5d97SBram Moolenaar 			// when "to" is "_" it means empty
29699ccfebddSBram Moolenaar 			add_fromto(spin, &spin->si_sal, items[1],
29709ccfebddSBram Moolenaar 				     STRCMP(items[2], "_") == 0 ? (char_u *)""
29719ccfebddSBram Moolenaar 								: items[2]);
29729ccfebddSBram Moolenaar 		}
29739ccfebddSBram Moolenaar 	    }
29749ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "SOFOFROM", 2)
29759ccfebddSBram Moolenaar 							  && sofofrom == NULL)
29769ccfebddSBram Moolenaar 	    {
29779ccfebddSBram Moolenaar 		sofofrom = getroom_save(spin, items[1]);
29789ccfebddSBram Moolenaar 	    }
29799ccfebddSBram Moolenaar 	    else if (is_aff_rule(items, itemcnt, "SOFOTO", 2)
29809ccfebddSBram Moolenaar 							    && sofoto == NULL)
29819ccfebddSBram Moolenaar 	    {
29829ccfebddSBram Moolenaar 		sofoto = getroom_save(spin, items[1]);
29839ccfebddSBram Moolenaar 	    }
29849ccfebddSBram Moolenaar 	    else if (STRCMP(items[0], "COMMON") == 0)
29859ccfebddSBram Moolenaar 	    {
29869ccfebddSBram Moolenaar 		int	i;
29879ccfebddSBram Moolenaar 
29889ccfebddSBram Moolenaar 		for (i = 1; i < itemcnt; ++i)
29899ccfebddSBram Moolenaar 		{
29909ccfebddSBram Moolenaar 		    if (HASHITEM_EMPTY(hash_find(&spin->si_commonwords,
29919ccfebddSBram Moolenaar 								   items[i])))
29929ccfebddSBram Moolenaar 		    {
29939ccfebddSBram Moolenaar 			p = vim_strsave(items[i]);
29949ccfebddSBram Moolenaar 			if (p == NULL)
29959ccfebddSBram Moolenaar 			    break;
29969ccfebddSBram Moolenaar 			hash_add(&spin->si_commonwords, p);
29979ccfebddSBram Moolenaar 		    }
29989ccfebddSBram Moolenaar 		}
29999ccfebddSBram Moolenaar 	    }
30009ccfebddSBram Moolenaar 	    else
3001f9e3e09fSBram Moolenaar 		smsg(_("Unrecognized or duplicate item in %s line %d: %s"),
30029ccfebddSBram Moolenaar 						       fname, lnum, items[0]);
30039ccfebddSBram Moolenaar 	}
30049ccfebddSBram Moolenaar     }
30059ccfebddSBram Moolenaar 
30069ccfebddSBram Moolenaar     if (fol != NULL || low != NULL || upp != NULL)
30079ccfebddSBram Moolenaar     {
30089ccfebddSBram Moolenaar 	if (spin->si_clear_chartab)
30099ccfebddSBram Moolenaar 	{
30100d6f5d97SBram Moolenaar 	    // Clear the char type tables, don't want to use any of the
30110d6f5d97SBram Moolenaar 	    // currently used spell properties.
30129ccfebddSBram Moolenaar 	    init_spell_chartab();
30139ccfebddSBram Moolenaar 	    spin->si_clear_chartab = FALSE;
30149ccfebddSBram Moolenaar 	}
30159ccfebddSBram Moolenaar 
30169ccfebddSBram Moolenaar 	/*
30179ccfebddSBram Moolenaar 	 * Don't write a word table for an ASCII file, so that we don't check
30189ccfebddSBram Moolenaar 	 * for conflicts with a word table that matches 'encoding'.
30199ccfebddSBram Moolenaar 	 * Don't write one for utf-8 either, we use utf_*() and
30209ccfebddSBram Moolenaar 	 * mb_get_class(), the list of chars in the file will be incomplete.
30219ccfebddSBram Moolenaar 	 */
3022264b74faSBram Moolenaar 	if (!spin->si_ascii && !enc_utf8)
30239ccfebddSBram Moolenaar 	{
30249ccfebddSBram Moolenaar 	    if (fol == NULL || low == NULL || upp == NULL)
3025f9e3e09fSBram Moolenaar 		smsg(_("Missing FOL/LOW/UPP line in %s"), fname);
30269ccfebddSBram Moolenaar 	    else
30279ccfebddSBram Moolenaar 		(void)set_spell_chartab(fol, low, upp);
30289ccfebddSBram Moolenaar 	}
30299ccfebddSBram Moolenaar 
30309ccfebddSBram Moolenaar 	vim_free(fol);
30319ccfebddSBram Moolenaar 	vim_free(low);
30329ccfebddSBram Moolenaar 	vim_free(upp);
30339ccfebddSBram Moolenaar     }
30349ccfebddSBram Moolenaar 
30350d6f5d97SBram Moolenaar     // Use compound specifications of the .aff file for the spell info.
30369ccfebddSBram Moolenaar     if (compmax != 0)
30379ccfebddSBram Moolenaar     {
30389ccfebddSBram Moolenaar 	aff_check_number(spin->si_compmax, compmax, "COMPOUNDWORDMAX");
30399ccfebddSBram Moolenaar 	spin->si_compmax = compmax;
30409ccfebddSBram Moolenaar     }
30419ccfebddSBram Moolenaar 
30429ccfebddSBram Moolenaar     if (compminlen != 0)
30439ccfebddSBram Moolenaar     {
30449ccfebddSBram Moolenaar 	aff_check_number(spin->si_compminlen, compminlen, "COMPOUNDMIN");
30459ccfebddSBram Moolenaar 	spin->si_compminlen = compminlen;
30469ccfebddSBram Moolenaar     }
30479ccfebddSBram Moolenaar 
30489ccfebddSBram Moolenaar     if (compsylmax != 0)
30499ccfebddSBram Moolenaar     {
30509ccfebddSBram Moolenaar 	if (syllable == NULL)
3051f9e3e09fSBram Moolenaar 	    smsg(_("COMPOUNDSYLMAX used without SYLLABLE"));
30529ccfebddSBram Moolenaar 	aff_check_number(spin->si_compsylmax, compsylmax, "COMPOUNDSYLMAX");
30539ccfebddSBram Moolenaar 	spin->si_compsylmax = compsylmax;
30549ccfebddSBram Moolenaar     }
30559ccfebddSBram Moolenaar 
30569ccfebddSBram Moolenaar     if (compoptions != 0)
30579ccfebddSBram Moolenaar     {
30589ccfebddSBram Moolenaar 	aff_check_number(spin->si_compoptions, compoptions, "COMPOUND options");
30599ccfebddSBram Moolenaar 	spin->si_compoptions |= compoptions;
30609ccfebddSBram Moolenaar     }
30619ccfebddSBram Moolenaar 
30629ccfebddSBram Moolenaar     if (compflags != NULL)
30639ccfebddSBram Moolenaar 	process_compflags(spin, aff, compflags);
30649ccfebddSBram Moolenaar 
30650d6f5d97SBram Moolenaar     // Check that we didn't use too many renumbered flags.
30669ccfebddSBram Moolenaar     if (spin->si_newcompID < spin->si_newprefID)
30679ccfebddSBram Moolenaar     {
30689ccfebddSBram Moolenaar 	if (spin->si_newcompID == 127 || spin->si_newcompID == 255)
306932526b3cSBram Moolenaar 	    msg(_("Too many postponed prefixes"));
30709ccfebddSBram Moolenaar 	else if (spin->si_newprefID == 0 || spin->si_newprefID == 127)
307132526b3cSBram Moolenaar 	    msg(_("Too many compound flags"));
30729ccfebddSBram Moolenaar 	else
307332526b3cSBram Moolenaar 	    msg(_("Too many postponed prefixes and/or compound flags"));
30749ccfebddSBram Moolenaar     }
30759ccfebddSBram Moolenaar 
30769ccfebddSBram Moolenaar     if (syllable != NULL)
30779ccfebddSBram Moolenaar     {
30789ccfebddSBram Moolenaar 	aff_check_string(spin->si_syllable, syllable, "SYLLABLE");
30799ccfebddSBram Moolenaar 	spin->si_syllable = syllable;
30809ccfebddSBram Moolenaar     }
30819ccfebddSBram Moolenaar 
30829ccfebddSBram Moolenaar     if (sofofrom != NULL || sofoto != NULL)
30839ccfebddSBram Moolenaar     {
30849ccfebddSBram Moolenaar 	if (sofofrom == NULL || sofoto == NULL)
3085f9e3e09fSBram Moolenaar 	    smsg(_("Missing SOFO%s line in %s"),
30869ccfebddSBram Moolenaar 				     sofofrom == NULL ? "FROM" : "TO", fname);
30879ccfebddSBram Moolenaar 	else if (spin->si_sal.ga_len > 0)
3088f9e3e09fSBram Moolenaar 	    smsg(_("Both SAL and SOFO lines in %s"), fname);
30899ccfebddSBram Moolenaar 	else
30909ccfebddSBram Moolenaar 	{
30919ccfebddSBram Moolenaar 	    aff_check_string(spin->si_sofofr, sofofrom, "SOFOFROM");
30929ccfebddSBram Moolenaar 	    aff_check_string(spin->si_sofoto, sofoto, "SOFOTO");
30939ccfebddSBram Moolenaar 	    spin->si_sofofr = sofofrom;
30949ccfebddSBram Moolenaar 	    spin->si_sofoto = sofoto;
30959ccfebddSBram Moolenaar 	}
30969ccfebddSBram Moolenaar     }
30979ccfebddSBram Moolenaar 
30989ccfebddSBram Moolenaar     if (midword != NULL)
30999ccfebddSBram Moolenaar     {
31009ccfebddSBram Moolenaar 	aff_check_string(spin->si_midword, midword, "MIDWORD");
31019ccfebddSBram Moolenaar 	spin->si_midword = midword;
31029ccfebddSBram Moolenaar     }
31039ccfebddSBram Moolenaar 
31049ccfebddSBram Moolenaar     vim_free(pc);
31059ccfebddSBram Moolenaar     fclose(fd);
31069ccfebddSBram Moolenaar     return aff;
31079ccfebddSBram Moolenaar }
31089ccfebddSBram Moolenaar 
31099ccfebddSBram Moolenaar /*
31109ccfebddSBram Moolenaar  * Return TRUE when items[0] equals "rulename", there are "mincount" items or
31119ccfebddSBram Moolenaar  * a comment is following after item "mincount".
31129ccfebddSBram Moolenaar  */
31139ccfebddSBram Moolenaar     static int
is_aff_rule(char_u ** items,int itemcnt,char * rulename,int mincount)31149ccfebddSBram Moolenaar is_aff_rule(
31159ccfebddSBram Moolenaar     char_u	**items,
31169ccfebddSBram Moolenaar     int		itemcnt,
31179ccfebddSBram Moolenaar     char	*rulename,
31189ccfebddSBram Moolenaar     int		mincount)
31199ccfebddSBram Moolenaar {
31209ccfebddSBram Moolenaar     return (STRCMP(items[0], rulename) == 0
31219ccfebddSBram Moolenaar 	    && (itemcnt == mincount
31229ccfebddSBram Moolenaar 		|| (itemcnt > mincount && items[mincount][0] == '#')));
31239ccfebddSBram Moolenaar }
31249ccfebddSBram Moolenaar 
31259ccfebddSBram Moolenaar /*
31269ccfebddSBram Moolenaar  * For affix "entry" move COMPOUNDFORBIDFLAG and COMPOUNDPERMITFLAG from
31279ccfebddSBram Moolenaar  * ae_flags to ae_comppermit and ae_compforbid.
31289ccfebddSBram Moolenaar  */
31299ccfebddSBram Moolenaar     static void
aff_process_flags(afffile_T * affile,affentry_T * entry)31309ccfebddSBram Moolenaar aff_process_flags(afffile_T *affile, affentry_T *entry)
31319ccfebddSBram Moolenaar {
31329ccfebddSBram Moolenaar     char_u	*p;
31339ccfebddSBram Moolenaar     char_u	*prevp;
31349ccfebddSBram Moolenaar     unsigned	flag;
31359ccfebddSBram Moolenaar 
31369ccfebddSBram Moolenaar     if (entry->ae_flags != NULL
31379ccfebddSBram Moolenaar 		&& (affile->af_compforbid != 0 || affile->af_comppermit != 0))
31389ccfebddSBram Moolenaar     {
31399ccfebddSBram Moolenaar 	for (p = entry->ae_flags; *p != NUL; )
31409ccfebddSBram Moolenaar 	{
31419ccfebddSBram Moolenaar 	    prevp = p;
31429ccfebddSBram Moolenaar 	    flag = get_affitem(affile->af_flagtype, &p);
31439ccfebddSBram Moolenaar 	    if (flag == affile->af_comppermit || flag == affile->af_compforbid)
31449ccfebddSBram Moolenaar 	    {
31459ccfebddSBram Moolenaar 		STRMOVE(prevp, p);
31469ccfebddSBram Moolenaar 		p = prevp;
31479ccfebddSBram Moolenaar 		if (flag == affile->af_comppermit)
31489ccfebddSBram Moolenaar 		    entry->ae_comppermit = TRUE;
31499ccfebddSBram Moolenaar 		else
31509ccfebddSBram Moolenaar 		    entry->ae_compforbid = TRUE;
31519ccfebddSBram Moolenaar 	    }
31529ccfebddSBram Moolenaar 	    if (affile->af_flagtype == AFT_NUM && *p == ',')
31539ccfebddSBram Moolenaar 		++p;
31549ccfebddSBram Moolenaar 	}
31559ccfebddSBram Moolenaar 	if (*entry->ae_flags == NUL)
31560d6f5d97SBram Moolenaar 	    entry->ae_flags = NULL;	// nothing left
31579ccfebddSBram Moolenaar     }
31589ccfebddSBram Moolenaar }
31599ccfebddSBram Moolenaar 
31609ccfebddSBram Moolenaar /*
31619ccfebddSBram Moolenaar  * Return TRUE if "s" is the name of an info item in the affix file.
31629ccfebddSBram Moolenaar  */
31639ccfebddSBram Moolenaar     static int
spell_info_item(char_u * s)31649ccfebddSBram Moolenaar spell_info_item(char_u *s)
31659ccfebddSBram Moolenaar {
31669ccfebddSBram Moolenaar     return STRCMP(s, "NAME") == 0
31679ccfebddSBram Moolenaar 	|| STRCMP(s, "HOME") == 0
31689ccfebddSBram Moolenaar 	|| STRCMP(s, "VERSION") == 0
31699ccfebddSBram Moolenaar 	|| STRCMP(s, "AUTHOR") == 0
31709ccfebddSBram Moolenaar 	|| STRCMP(s, "EMAIL") == 0
31719ccfebddSBram Moolenaar 	|| STRCMP(s, "COPYRIGHT") == 0;
31729ccfebddSBram Moolenaar }
31739ccfebddSBram Moolenaar 
31749ccfebddSBram Moolenaar /*
31759ccfebddSBram Moolenaar  * Turn an affix flag name into a number, according to the FLAG type.
31769ccfebddSBram Moolenaar  * returns zero for failure.
31779ccfebddSBram Moolenaar  */
31789ccfebddSBram Moolenaar     static unsigned
affitem2flag(int flagtype,char_u * item,char_u * fname,int lnum)31799ccfebddSBram Moolenaar affitem2flag(
31809ccfebddSBram Moolenaar     int		flagtype,
31819ccfebddSBram Moolenaar     char_u	*item,
31829ccfebddSBram Moolenaar     char_u	*fname,
31839ccfebddSBram Moolenaar     int		lnum)
31849ccfebddSBram Moolenaar {
31859ccfebddSBram Moolenaar     unsigned	res;
31869ccfebddSBram Moolenaar     char_u	*p = item;
31879ccfebddSBram Moolenaar 
31889ccfebddSBram Moolenaar     res = get_affitem(flagtype, &p);
31899ccfebddSBram Moolenaar     if (res == 0)
31909ccfebddSBram Moolenaar     {
31919ccfebddSBram Moolenaar 	if (flagtype == AFT_NUM)
3192f9e3e09fSBram Moolenaar 	    smsg(_("Flag is not a number in %s line %d: %s"),
31939ccfebddSBram Moolenaar 							   fname, lnum, item);
31949ccfebddSBram Moolenaar 	else
3195f9e3e09fSBram Moolenaar 	    smsg(_("Illegal flag in %s line %d: %s"),
31969ccfebddSBram Moolenaar 							   fname, lnum, item);
31979ccfebddSBram Moolenaar     }
31989ccfebddSBram Moolenaar     if (*p != NUL)
31999ccfebddSBram Moolenaar     {
3200f9e3e09fSBram Moolenaar 	smsg(_(e_affname), fname, lnum, item);
32019ccfebddSBram Moolenaar 	return 0;
32029ccfebddSBram Moolenaar     }
32039ccfebddSBram Moolenaar 
32049ccfebddSBram Moolenaar     return res;
32059ccfebddSBram Moolenaar }
32069ccfebddSBram Moolenaar 
32079ccfebddSBram Moolenaar /*
32089ccfebddSBram Moolenaar  * Get one affix name from "*pp" and advance the pointer.
32093d2a47c7SBram Moolenaar  * Returns ZERO_FLAG for "0".
32109ccfebddSBram Moolenaar  * Returns zero for an error, still advances the pointer then.
32119ccfebddSBram Moolenaar  */
32129ccfebddSBram Moolenaar     static unsigned
get_affitem(int flagtype,char_u ** pp)32139ccfebddSBram Moolenaar get_affitem(int flagtype, char_u **pp)
32149ccfebddSBram Moolenaar {
32159ccfebddSBram Moolenaar     int		res;
32169ccfebddSBram Moolenaar 
32179ccfebddSBram Moolenaar     if (flagtype == AFT_NUM)
32189ccfebddSBram Moolenaar     {
32199ccfebddSBram Moolenaar 	if (!VIM_ISDIGIT(**pp))
32209ccfebddSBram Moolenaar 	{
32210d6f5d97SBram Moolenaar 	    ++*pp;	// always advance, avoid getting stuck
32229ccfebddSBram Moolenaar 	    return 0;
32239ccfebddSBram Moolenaar 	}
32249ccfebddSBram Moolenaar 	res = getdigits(pp);
32253d2a47c7SBram Moolenaar 	if (res == 0)
32263d2a47c7SBram Moolenaar 	    res = ZERO_FLAG;
32279ccfebddSBram Moolenaar     }
32289ccfebddSBram Moolenaar     else
32299ccfebddSBram Moolenaar     {
32309ccfebddSBram Moolenaar 	res = mb_ptr2char_adv(pp);
32319ccfebddSBram Moolenaar 	if (flagtype == AFT_LONG || (flagtype == AFT_CAPLONG
32329ccfebddSBram Moolenaar 						 && res >= 'A' && res <= 'Z'))
32339ccfebddSBram Moolenaar 	{
32349ccfebddSBram Moolenaar 	    if (**pp == NUL)
32359ccfebddSBram Moolenaar 		return 0;
32369ccfebddSBram Moolenaar 	    res = mb_ptr2char_adv(pp) + (res << 16);
32379ccfebddSBram Moolenaar 	}
32389ccfebddSBram Moolenaar     }
32399ccfebddSBram Moolenaar     return res;
32409ccfebddSBram Moolenaar }
32419ccfebddSBram Moolenaar 
32429ccfebddSBram Moolenaar /*
32439ccfebddSBram Moolenaar  * Process the "compflags" string used in an affix file and append it to
32449ccfebddSBram Moolenaar  * spin->si_compflags.
32459ccfebddSBram Moolenaar  * The processing involves changing the affix names to ID numbers, so that
32469ccfebddSBram Moolenaar  * they fit in one byte.
32479ccfebddSBram Moolenaar  */
32489ccfebddSBram Moolenaar     static void
process_compflags(spellinfo_T * spin,afffile_T * aff,char_u * compflags)32499ccfebddSBram Moolenaar process_compflags(
32509ccfebddSBram Moolenaar     spellinfo_T	*spin,
32519ccfebddSBram Moolenaar     afffile_T	*aff,
32529ccfebddSBram Moolenaar     char_u	*compflags)
32539ccfebddSBram Moolenaar {
32549ccfebddSBram Moolenaar     char_u	*p;
32559ccfebddSBram Moolenaar     char_u	*prevp;
32569ccfebddSBram Moolenaar     unsigned	flag;
32579ccfebddSBram Moolenaar     compitem_T	*ci;
32589ccfebddSBram Moolenaar     int		id;
32599ccfebddSBram Moolenaar     int		len;
32609ccfebddSBram Moolenaar     char_u	*tp;
32619ccfebddSBram Moolenaar     char_u	key[AH_KEY_LEN];
32629ccfebddSBram Moolenaar     hashitem_T	*hi;
32639ccfebddSBram Moolenaar 
32640d6f5d97SBram Moolenaar     // Make room for the old and the new compflags, concatenated with a / in
32650d6f5d97SBram Moolenaar     // between.  Processing it makes it shorter, but we don't know by how
32660d6f5d97SBram Moolenaar     // much, thus allocate the maximum.
32679ccfebddSBram Moolenaar     len = (int)STRLEN(compflags) + 1;
32689ccfebddSBram Moolenaar     if (spin->si_compflags != NULL)
32699ccfebddSBram Moolenaar 	len += (int)STRLEN(spin->si_compflags) + 1;
32709ccfebddSBram Moolenaar     p = getroom(spin, len, FALSE);
32719ccfebddSBram Moolenaar     if (p == NULL)
32729ccfebddSBram Moolenaar 	return;
32739ccfebddSBram Moolenaar     if (spin->si_compflags != NULL)
32749ccfebddSBram Moolenaar     {
32759ccfebddSBram Moolenaar 	STRCPY(p, spin->si_compflags);
32769ccfebddSBram Moolenaar 	STRCAT(p, "/");
32779ccfebddSBram Moolenaar     }
32789ccfebddSBram Moolenaar     spin->si_compflags = p;
32799ccfebddSBram Moolenaar     tp = p + STRLEN(p);
32809ccfebddSBram Moolenaar 
32819ccfebddSBram Moolenaar     for (p = compflags; *p != NUL; )
32829ccfebddSBram Moolenaar     {
32839ccfebddSBram Moolenaar 	if (vim_strchr((char_u *)"/?*+[]", *p) != NULL)
32840d6f5d97SBram Moolenaar 	    // Copy non-flag characters directly.
32859ccfebddSBram Moolenaar 	    *tp++ = *p++;
32869ccfebddSBram Moolenaar 	else
32879ccfebddSBram Moolenaar 	{
32880d6f5d97SBram Moolenaar 	    // First get the flag number, also checks validity.
32899ccfebddSBram Moolenaar 	    prevp = p;
32909ccfebddSBram Moolenaar 	    flag = get_affitem(aff->af_flagtype, &p);
32919ccfebddSBram Moolenaar 	    if (flag != 0)
32929ccfebddSBram Moolenaar 	    {
32930d6f5d97SBram Moolenaar 		// Find the flag in the hashtable.  If it was used before, use
32940d6f5d97SBram Moolenaar 		// the existing ID.  Otherwise add a new entry.
32959ccfebddSBram Moolenaar 		vim_strncpy(key, prevp, p - prevp);
32969ccfebddSBram Moolenaar 		hi = hash_find(&aff->af_comp, key);
32979ccfebddSBram Moolenaar 		if (!HASHITEM_EMPTY(hi))
32989ccfebddSBram Moolenaar 		    id = HI2CI(hi)->ci_newID;
32999ccfebddSBram Moolenaar 		else
33009ccfebddSBram Moolenaar 		{
33019ccfebddSBram Moolenaar 		    ci = (compitem_T *)getroom(spin, sizeof(compitem_T), TRUE);
33029ccfebddSBram Moolenaar 		    if (ci == NULL)
33039ccfebddSBram Moolenaar 			break;
33049ccfebddSBram Moolenaar 		    STRCPY(ci->ci_key, key);
33059ccfebddSBram Moolenaar 		    ci->ci_flag = flag;
33060d6f5d97SBram Moolenaar 		    // Avoid using a flag ID that has a special meaning in a
33070d6f5d97SBram Moolenaar 		    // regexp (also inside []).
33089ccfebddSBram Moolenaar 		    do
33099ccfebddSBram Moolenaar 		    {
33109ccfebddSBram Moolenaar 			check_renumber(spin);
33119ccfebddSBram Moolenaar 			id = spin->si_newcompID--;
33129ccfebddSBram Moolenaar 		    } while (vim_strchr((char_u *)"/?*+[]\\-^", id) != NULL);
33139ccfebddSBram Moolenaar 		    ci->ci_newID = id;
33149ccfebddSBram Moolenaar 		    hash_add(&aff->af_comp, ci->ci_key);
33159ccfebddSBram Moolenaar 		}
33169ccfebddSBram Moolenaar 		*tp++ = id;
33179ccfebddSBram Moolenaar 	    }
33189ccfebddSBram Moolenaar 	    if (aff->af_flagtype == AFT_NUM && *p == ',')
33199ccfebddSBram Moolenaar 		++p;
33209ccfebddSBram Moolenaar 	}
33219ccfebddSBram Moolenaar     }
33229ccfebddSBram Moolenaar 
33239ccfebddSBram Moolenaar     *tp = NUL;
33249ccfebddSBram Moolenaar }
33259ccfebddSBram Moolenaar 
33269ccfebddSBram Moolenaar /*
33279ccfebddSBram Moolenaar  * Check that the new IDs for postponed affixes and compounding don't overrun
33289ccfebddSBram Moolenaar  * each other.  We have almost 255 available, but start at 0-127 to avoid
33299ccfebddSBram Moolenaar  * using two bytes for utf-8.  When the 0-127 range is used up go to 128-255.
33309ccfebddSBram Moolenaar  * When that is used up an error message is given.
33319ccfebddSBram Moolenaar  */
33329ccfebddSBram Moolenaar     static void
check_renumber(spellinfo_T * spin)33339ccfebddSBram Moolenaar check_renumber(spellinfo_T *spin)
33349ccfebddSBram Moolenaar {
33359ccfebddSBram Moolenaar     if (spin->si_newprefID == spin->si_newcompID && spin->si_newcompID < 128)
33369ccfebddSBram Moolenaar     {
33379ccfebddSBram Moolenaar 	spin->si_newprefID = 127;
33389ccfebddSBram Moolenaar 	spin->si_newcompID = 255;
33399ccfebddSBram Moolenaar     }
33409ccfebddSBram Moolenaar }
33419ccfebddSBram Moolenaar 
33429ccfebddSBram Moolenaar /*
33439ccfebddSBram Moolenaar  * Return TRUE if flag "flag" appears in affix list "afflist".
33449ccfebddSBram Moolenaar  */
33459ccfebddSBram Moolenaar     static int
flag_in_afflist(int flagtype,char_u * afflist,unsigned flag)33469ccfebddSBram Moolenaar flag_in_afflist(int flagtype, char_u *afflist, unsigned flag)
33479ccfebddSBram Moolenaar {
33489ccfebddSBram Moolenaar     char_u	*p;
33499ccfebddSBram Moolenaar     unsigned	n;
33509ccfebddSBram Moolenaar 
33519ccfebddSBram Moolenaar     switch (flagtype)
33529ccfebddSBram Moolenaar     {
33539ccfebddSBram Moolenaar 	case AFT_CHAR:
33549ccfebddSBram Moolenaar 	    return vim_strchr(afflist, flag) != NULL;
33559ccfebddSBram Moolenaar 
33569ccfebddSBram Moolenaar 	case AFT_CAPLONG:
33579ccfebddSBram Moolenaar 	case AFT_LONG:
33589ccfebddSBram Moolenaar 	    for (p = afflist; *p != NUL; )
33599ccfebddSBram Moolenaar 	    {
33609ccfebddSBram Moolenaar 		n = mb_ptr2char_adv(&p);
33619ccfebddSBram Moolenaar 		if ((flagtype == AFT_LONG || (n >= 'A' && n <= 'Z'))
33629ccfebddSBram Moolenaar 								 && *p != NUL)
33639ccfebddSBram Moolenaar 		    n = mb_ptr2char_adv(&p) + (n << 16);
33649ccfebddSBram Moolenaar 		if (n == flag)
33659ccfebddSBram Moolenaar 		    return TRUE;
33669ccfebddSBram Moolenaar 	    }
33679ccfebddSBram Moolenaar 	    break;
33689ccfebddSBram Moolenaar 
33699ccfebddSBram Moolenaar 	case AFT_NUM:
33709ccfebddSBram Moolenaar 	    for (p = afflist; *p != NUL; )
33719ccfebddSBram Moolenaar 	    {
33729ccfebddSBram Moolenaar 		n = getdigits(&p);
33733d2a47c7SBram Moolenaar 		if (n == 0)
33743d2a47c7SBram Moolenaar 		    n = ZERO_FLAG;
33759ccfebddSBram Moolenaar 		if (n == flag)
33769ccfebddSBram Moolenaar 		    return TRUE;
33770d6f5d97SBram Moolenaar 		if (*p != NUL)	// skip over comma
33789ccfebddSBram Moolenaar 		    ++p;
33799ccfebddSBram Moolenaar 	    }
33809ccfebddSBram Moolenaar 	    break;
33819ccfebddSBram Moolenaar     }
33829ccfebddSBram Moolenaar     return FALSE;
33839ccfebddSBram Moolenaar }
33849ccfebddSBram Moolenaar 
33859ccfebddSBram Moolenaar /*
33869ccfebddSBram Moolenaar  * Give a warning when "spinval" and "affval" numbers are set and not the same.
33879ccfebddSBram Moolenaar  */
33889ccfebddSBram Moolenaar     static void
aff_check_number(int spinval,int affval,char * name)33899ccfebddSBram Moolenaar aff_check_number(int spinval, int affval, char *name)
33909ccfebddSBram Moolenaar {
33919ccfebddSBram Moolenaar     if (spinval != 0 && spinval != affval)
3392f9e3e09fSBram Moolenaar 	smsg(_("%s value differs from what is used in another .aff file"), name);
33939ccfebddSBram Moolenaar }
33949ccfebddSBram Moolenaar 
33959ccfebddSBram Moolenaar /*
33969ccfebddSBram Moolenaar  * Give a warning when "spinval" and "affval" strings are set and not the same.
33979ccfebddSBram Moolenaar  */
33989ccfebddSBram Moolenaar     static void
aff_check_string(char_u * spinval,char_u * affval,char * name)33999ccfebddSBram Moolenaar aff_check_string(char_u *spinval, char_u *affval, char *name)
34009ccfebddSBram Moolenaar {
34019ccfebddSBram Moolenaar     if (spinval != NULL && STRCMP(spinval, affval) != 0)
3402f9e3e09fSBram Moolenaar 	smsg(_("%s value differs from what is used in another .aff file"), name);
34039ccfebddSBram Moolenaar }
34049ccfebddSBram Moolenaar 
34059ccfebddSBram Moolenaar /*
34069ccfebddSBram Moolenaar  * Return TRUE if strings "s1" and "s2" are equal.  Also consider both being
34079ccfebddSBram Moolenaar  * NULL as equal.
34089ccfebddSBram Moolenaar  */
34099ccfebddSBram Moolenaar     static int
str_equal(char_u * s1,char_u * s2)34109ccfebddSBram Moolenaar str_equal(char_u *s1, char_u *s2)
34119ccfebddSBram Moolenaar {
34129ccfebddSBram Moolenaar     if (s1 == NULL || s2 == NULL)
34139ccfebddSBram Moolenaar 	return s1 == s2;
34149ccfebddSBram Moolenaar     return STRCMP(s1, s2) == 0;
34159ccfebddSBram Moolenaar }
34169ccfebddSBram Moolenaar 
34179ccfebddSBram Moolenaar /*
34189ccfebddSBram Moolenaar  * Add a from-to item to "gap".  Used for REP and SAL items.
34199ccfebddSBram Moolenaar  * They are stored case-folded.
34209ccfebddSBram Moolenaar  */
34219ccfebddSBram Moolenaar     static void
add_fromto(spellinfo_T * spin,garray_T * gap,char_u * from,char_u * to)34229ccfebddSBram Moolenaar add_fromto(
34239ccfebddSBram Moolenaar     spellinfo_T	*spin,
34249ccfebddSBram Moolenaar     garray_T	*gap,
34259ccfebddSBram Moolenaar     char_u	*from,
34269ccfebddSBram Moolenaar     char_u	*to)
34279ccfebddSBram Moolenaar {
34289ccfebddSBram Moolenaar     fromto_T	*ftp;
34299ccfebddSBram Moolenaar     char_u	word[MAXWLEN];
34309ccfebddSBram Moolenaar 
34319ccfebddSBram Moolenaar     if (ga_grow(gap, 1) == OK)
34329ccfebddSBram Moolenaar     {
34339ccfebddSBram Moolenaar 	ftp = ((fromto_T *)gap->ga_data) + gap->ga_len;
34344f135275SBram Moolenaar 	(void)spell_casefold(curwin, from, (int)STRLEN(from), word, MAXWLEN);
34359ccfebddSBram Moolenaar 	ftp->ft_from = getroom_save(spin, word);
34364f135275SBram Moolenaar 	(void)spell_casefold(curwin, to, (int)STRLEN(to), word, MAXWLEN);
34379ccfebddSBram Moolenaar 	ftp->ft_to = getroom_save(spin, word);
34389ccfebddSBram Moolenaar 	++gap->ga_len;
34399ccfebddSBram Moolenaar     }
34409ccfebddSBram Moolenaar }
34419ccfebddSBram Moolenaar 
34429ccfebddSBram Moolenaar /*
34439ccfebddSBram Moolenaar  * Convert a boolean argument in a SAL line to TRUE or FALSE;
34449ccfebddSBram Moolenaar  */
34459ccfebddSBram Moolenaar     static int
sal_to_bool(char_u * s)34469ccfebddSBram Moolenaar sal_to_bool(char_u *s)
34479ccfebddSBram Moolenaar {
34489ccfebddSBram Moolenaar     return STRCMP(s, "1") == 0 || STRCMP(s, "true") == 0;
34499ccfebddSBram Moolenaar }
34509ccfebddSBram Moolenaar 
34519ccfebddSBram Moolenaar /*
34529ccfebddSBram Moolenaar  * Free the structure filled by spell_read_aff().
34539ccfebddSBram Moolenaar  */
34549ccfebddSBram Moolenaar     static void
spell_free_aff(afffile_T * aff)34559ccfebddSBram Moolenaar spell_free_aff(afffile_T *aff)
34569ccfebddSBram Moolenaar {
34579ccfebddSBram Moolenaar     hashtab_T	*ht;
34589ccfebddSBram Moolenaar     hashitem_T	*hi;
34599ccfebddSBram Moolenaar     int		todo;
34609ccfebddSBram Moolenaar     affheader_T	*ah;
34619ccfebddSBram Moolenaar     affentry_T	*ae;
34629ccfebddSBram Moolenaar 
34639ccfebddSBram Moolenaar     vim_free(aff->af_enc);
34649ccfebddSBram Moolenaar 
34650d6f5d97SBram Moolenaar     // All this trouble to free the "ae_prog" items...
34669ccfebddSBram Moolenaar     for (ht = &aff->af_pref; ; ht = &aff->af_suff)
34679ccfebddSBram Moolenaar     {
34689ccfebddSBram Moolenaar 	todo = (int)ht->ht_used;
34699ccfebddSBram Moolenaar 	for (hi = ht->ht_array; todo > 0; ++hi)
34709ccfebddSBram Moolenaar 	{
34719ccfebddSBram Moolenaar 	    if (!HASHITEM_EMPTY(hi))
34729ccfebddSBram Moolenaar 	    {
34739ccfebddSBram Moolenaar 		--todo;
34749ccfebddSBram Moolenaar 		ah = HI2AH(hi);
34759ccfebddSBram Moolenaar 		for (ae = ah->ah_first; ae != NULL; ae = ae->ae_next)
34769ccfebddSBram Moolenaar 		    vim_regfree(ae->ae_prog);
34779ccfebddSBram Moolenaar 	    }
34789ccfebddSBram Moolenaar 	}
34799ccfebddSBram Moolenaar 	if (ht == &aff->af_suff)
34809ccfebddSBram Moolenaar 	    break;
34819ccfebddSBram Moolenaar     }
34829ccfebddSBram Moolenaar 
34839ccfebddSBram Moolenaar     hash_clear(&aff->af_pref);
34849ccfebddSBram Moolenaar     hash_clear(&aff->af_suff);
34859ccfebddSBram Moolenaar     hash_clear(&aff->af_comp);
34869ccfebddSBram Moolenaar }
34879ccfebddSBram Moolenaar 
34889ccfebddSBram Moolenaar /*
34899ccfebddSBram Moolenaar  * Read dictionary file "fname".
34909ccfebddSBram Moolenaar  * Returns OK or FAIL;
34919ccfebddSBram Moolenaar  */
34929ccfebddSBram Moolenaar     static int
spell_read_dic(spellinfo_T * spin,char_u * fname,afffile_T * affile)34939ccfebddSBram Moolenaar spell_read_dic(spellinfo_T *spin, char_u *fname, afffile_T *affile)
34949ccfebddSBram Moolenaar {
34959ccfebddSBram Moolenaar     hashtab_T	ht;
34969ccfebddSBram Moolenaar     char_u	line[MAXLINELEN];
34979ccfebddSBram Moolenaar     char_u	*p;
34989ccfebddSBram Moolenaar     char_u	*afflist;
34999ccfebddSBram Moolenaar     char_u	store_afflist[MAXWLEN];
35009ccfebddSBram Moolenaar     int		pfxlen;
35019ccfebddSBram Moolenaar     int		need_affix;
35029ccfebddSBram Moolenaar     char_u	*dw;
35039ccfebddSBram Moolenaar     char_u	*pc;
35049ccfebddSBram Moolenaar     char_u	*w;
35059ccfebddSBram Moolenaar     int		l;
35069ccfebddSBram Moolenaar     hash_T	hash;
35079ccfebddSBram Moolenaar     hashitem_T	*hi;
35089ccfebddSBram Moolenaar     FILE	*fd;
35099ccfebddSBram Moolenaar     int		lnum = 1;
35109ccfebddSBram Moolenaar     int		non_ascii = 0;
35119ccfebddSBram Moolenaar     int		retval = OK;
35129ccfebddSBram Moolenaar     char_u	message[MAXLINELEN + MAXWLEN];
35139ccfebddSBram Moolenaar     int		flags;
35149ccfebddSBram Moolenaar     int		duplicate = 0;
3515408c23b0SBram Moolenaar     time_T	last_msg_time = 0;
35169ccfebddSBram Moolenaar 
35179ccfebddSBram Moolenaar     /*
35189ccfebddSBram Moolenaar      * Open the file.
35199ccfebddSBram Moolenaar      */
35209ccfebddSBram Moolenaar     fd = mch_fopen((char *)fname, "r");
35219ccfebddSBram Moolenaar     if (fd == NULL)
35229ccfebddSBram Moolenaar     {
3523f9e3e09fSBram Moolenaar 	semsg(_(e_notopen), fname);
35249ccfebddSBram Moolenaar 	return FAIL;
35259ccfebddSBram Moolenaar     }
35269ccfebddSBram Moolenaar 
35270d6f5d97SBram Moolenaar     // The hashtable is only used to detect duplicated words.
35289ccfebddSBram Moolenaar     hash_init(&ht);
35299ccfebddSBram Moolenaar 
35309ccfebddSBram Moolenaar     vim_snprintf((char *)IObuff, IOSIZE,
35319ccfebddSBram Moolenaar 				  _("Reading dictionary file %s..."), fname);
35329ccfebddSBram Moolenaar     spell_message(spin, IObuff);
35339ccfebddSBram Moolenaar 
35340d6f5d97SBram Moolenaar     // start with a message for the first line
35359ccfebddSBram Moolenaar     spin->si_msg_count = 999999;
35369ccfebddSBram Moolenaar 
35370d6f5d97SBram Moolenaar     // Read and ignore the first line: word count.
3538e90d63eaSBram Moolenaar     if (vim_fgets(line, MAXLINELEN, fd) || !vim_isdigit(*skipwhite(line)))
3539f9e3e09fSBram Moolenaar 	semsg(_("E760: No word count in %s"), fname);
35409ccfebddSBram Moolenaar 
35419ccfebddSBram Moolenaar     /*
35429ccfebddSBram Moolenaar      * Read all the lines in the file one by one.
35439ccfebddSBram Moolenaar      * The words are converted to 'encoding' here, before being added to
35449ccfebddSBram Moolenaar      * the hashtable.
35459ccfebddSBram Moolenaar      */
35469ccfebddSBram Moolenaar     while (!vim_fgets(line, MAXLINELEN, fd) && !got_int)
35479ccfebddSBram Moolenaar     {
35489ccfebddSBram Moolenaar 	line_breakcheck();
35499ccfebddSBram Moolenaar 	++lnum;
35509ccfebddSBram Moolenaar 	if (line[0] == '#' || line[0] == '/')
35510d6f5d97SBram Moolenaar 	    continue;	// comment line
35529ccfebddSBram Moolenaar 
35530d6f5d97SBram Moolenaar 	// Remove CR, LF and white space from the end.  White space halfway
35540d6f5d97SBram Moolenaar 	// the word is kept to allow e.g., "et al.".
35559ccfebddSBram Moolenaar 	l = (int)STRLEN(line);
35569ccfebddSBram Moolenaar 	while (l > 0 && line[l - 1] <= ' ')
35579ccfebddSBram Moolenaar 	    --l;
35589ccfebddSBram Moolenaar 	if (l == 0)
35590d6f5d97SBram Moolenaar 	    continue;	// empty line
35609ccfebddSBram Moolenaar 	line[l] = NUL;
35619ccfebddSBram Moolenaar 
35620d6f5d97SBram Moolenaar 	// Convert from "SET" to 'encoding' when needed.
35639ccfebddSBram Moolenaar 	if (spin->si_conv.vc_type != CONV_NONE)
35649ccfebddSBram Moolenaar 	{
35659ccfebddSBram Moolenaar 	    pc = string_convert(&spin->si_conv, line, NULL);
35669ccfebddSBram Moolenaar 	    if (pc == NULL)
35679ccfebddSBram Moolenaar 	    {
3568f9e3e09fSBram Moolenaar 		smsg(_("Conversion failure for word in %s line %d: %s"),
35699ccfebddSBram Moolenaar 						       fname, lnum, line);
35709ccfebddSBram Moolenaar 		continue;
35719ccfebddSBram Moolenaar 	    }
35729ccfebddSBram Moolenaar 	    w = pc;
35739ccfebddSBram Moolenaar 	}
35749ccfebddSBram Moolenaar 	else
35759ccfebddSBram Moolenaar 	{
35769ccfebddSBram Moolenaar 	    pc = NULL;
35779ccfebddSBram Moolenaar 	    w = line;
35789ccfebddSBram Moolenaar 	}
35799ccfebddSBram Moolenaar 
35800d6f5d97SBram Moolenaar 	// Truncate the word at the "/", set "afflist" to what follows.
35810d6f5d97SBram Moolenaar 	// Replace "\/" by "/" and "\\" by "\".
35829ccfebddSBram Moolenaar 	afflist = NULL;
358391acfffcSBram Moolenaar 	for (p = w; *p != NUL; MB_PTR_ADV(p))
35849ccfebddSBram Moolenaar 	{
35859ccfebddSBram Moolenaar 	    if (*p == '\\' && (p[1] == '\\' || p[1] == '/'))
35869ccfebddSBram Moolenaar 		STRMOVE(p, p + 1);
35879ccfebddSBram Moolenaar 	    else if (*p == '/')
35889ccfebddSBram Moolenaar 	    {
35899ccfebddSBram Moolenaar 		*p = NUL;
35909ccfebddSBram Moolenaar 		afflist = p + 1;
35919ccfebddSBram Moolenaar 		break;
35929ccfebddSBram Moolenaar 	    }
35939ccfebddSBram Moolenaar 	}
35949ccfebddSBram Moolenaar 
35950d6f5d97SBram Moolenaar 	// Skip non-ASCII words when "spin->si_ascii" is TRUE.
35969ccfebddSBram Moolenaar 	if (spin->si_ascii && has_non_ascii(w))
35979ccfebddSBram Moolenaar 	{
35989ccfebddSBram Moolenaar 	    ++non_ascii;
35999ccfebddSBram Moolenaar 	    vim_free(pc);
36009ccfebddSBram Moolenaar 	    continue;
36019ccfebddSBram Moolenaar 	}
36029ccfebddSBram Moolenaar 
3603408c23b0SBram Moolenaar 	// This takes time, print a message every 10000 words, but not more
3604408c23b0SBram Moolenaar 	// often than once per second.
36059ccfebddSBram Moolenaar 	if (spin->si_verbose && spin->si_msg_count > 10000)
36069ccfebddSBram Moolenaar 	{
36079ccfebddSBram Moolenaar 	    spin->si_msg_count = 0;
3608408c23b0SBram Moolenaar 	    if (vim_time() > last_msg_time)
3609408c23b0SBram Moolenaar 	    {
3610408c23b0SBram Moolenaar 		last_msg_time = vim_time();
36119ccfebddSBram Moolenaar 		vim_snprintf((char *)message, sizeof(message),
3612ea39176bSBram Moolenaar 			_("line %6d, word %6ld - %s"),
36139ccfebddSBram Moolenaar 			   lnum, spin->si_foldwcount + spin->si_keepwcount, w);
36149ccfebddSBram Moolenaar 		msg_start();
361532526b3cSBram Moolenaar 		msg_outtrans_long_attr(message, 0);
36169ccfebddSBram Moolenaar 		msg_clr_eos();
36179ccfebddSBram Moolenaar 		msg_didout = FALSE;
36189ccfebddSBram Moolenaar 		msg_col = 0;
36199ccfebddSBram Moolenaar 		out_flush();
36209ccfebddSBram Moolenaar 	    }
3621408c23b0SBram Moolenaar 	}
36229ccfebddSBram Moolenaar 
36230d6f5d97SBram Moolenaar 	// Store the word in the hashtable to be able to find duplicates.
36249ccfebddSBram Moolenaar 	dw = (char_u *)getroom_save(spin, w);
36259ccfebddSBram Moolenaar 	if (dw == NULL)
36269ccfebddSBram Moolenaar 	{
36279ccfebddSBram Moolenaar 	    retval = FAIL;
36289ccfebddSBram Moolenaar 	    vim_free(pc);
36299ccfebddSBram Moolenaar 	    break;
36309ccfebddSBram Moolenaar 	}
36319ccfebddSBram Moolenaar 
36329ccfebddSBram Moolenaar 	hash = hash_hash(dw);
36339ccfebddSBram Moolenaar 	hi = hash_lookup(&ht, dw, hash);
36349ccfebddSBram Moolenaar 	if (!HASHITEM_EMPTY(hi))
36359ccfebddSBram Moolenaar 	{
36369ccfebddSBram Moolenaar 	    if (p_verbose > 0)
3637f9e3e09fSBram Moolenaar 		smsg(_("Duplicate word in %s line %d: %s"),
36389ccfebddSBram Moolenaar 							     fname, lnum, dw);
36399ccfebddSBram Moolenaar 	    else if (duplicate == 0)
3640f9e3e09fSBram Moolenaar 		smsg(_("First duplicate word in %s line %d: %s"),
36419ccfebddSBram Moolenaar 							     fname, lnum, dw);
36429ccfebddSBram Moolenaar 	    ++duplicate;
36439ccfebddSBram Moolenaar 	}
36449ccfebddSBram Moolenaar 	else
36459ccfebddSBram Moolenaar 	    hash_add_item(&ht, hi, dw, hash);
36469ccfebddSBram Moolenaar 
36479ccfebddSBram Moolenaar 	flags = 0;
36489ccfebddSBram Moolenaar 	store_afflist[0] = NUL;
36499ccfebddSBram Moolenaar 	pfxlen = 0;
36509ccfebddSBram Moolenaar 	need_affix = FALSE;
36519ccfebddSBram Moolenaar 	if (afflist != NULL)
36529ccfebddSBram Moolenaar 	{
36530d6f5d97SBram Moolenaar 	    // Extract flags from the affix list.
36549ccfebddSBram Moolenaar 	    flags |= get_affix_flags(affile, afflist);
36559ccfebddSBram Moolenaar 
36569ccfebddSBram Moolenaar 	    if (affile->af_needaffix != 0 && flag_in_afflist(
36579ccfebddSBram Moolenaar 			  affile->af_flagtype, afflist, affile->af_needaffix))
36589ccfebddSBram Moolenaar 		need_affix = TRUE;
36599ccfebddSBram Moolenaar 
36609ccfebddSBram Moolenaar 	    if (affile->af_pfxpostpone)
36610d6f5d97SBram Moolenaar 		// Need to store the list of prefix IDs with the word.
36629ccfebddSBram Moolenaar 		pfxlen = get_pfxlist(affile, afflist, store_afflist);
36639ccfebddSBram Moolenaar 
36649ccfebddSBram Moolenaar 	    if (spin->si_compflags != NULL)
36650d6f5d97SBram Moolenaar 		// Need to store the list of compound flags with the word.
36660d6f5d97SBram Moolenaar 		// Concatenate them to the list of prefix IDs.
36679ccfebddSBram Moolenaar 		get_compflags(affile, afflist, store_afflist + pfxlen);
36689ccfebddSBram Moolenaar 	}
36699ccfebddSBram Moolenaar 
36700d6f5d97SBram Moolenaar 	// Add the word to the word tree(s).
36719ccfebddSBram Moolenaar 	if (store_word(spin, dw, flags, spin->si_region,
36729ccfebddSBram Moolenaar 					   store_afflist, need_affix) == FAIL)
36739ccfebddSBram Moolenaar 	    retval = FAIL;
36749ccfebddSBram Moolenaar 
36759ccfebddSBram Moolenaar 	if (afflist != NULL)
36769ccfebddSBram Moolenaar 	{
36770d6f5d97SBram Moolenaar 	    // Find all matching suffixes and add the resulting words.
36780d6f5d97SBram Moolenaar 	    // Additionally do matching prefixes that combine.
36799ccfebddSBram Moolenaar 	    if (store_aff_word(spin, dw, afflist, affile,
36809ccfebddSBram Moolenaar 			   &affile->af_suff, &affile->af_pref,
36819ccfebddSBram Moolenaar 			    CONDIT_SUF, flags, store_afflist, pfxlen) == FAIL)
36829ccfebddSBram Moolenaar 		retval = FAIL;
36839ccfebddSBram Moolenaar 
36840d6f5d97SBram Moolenaar 	    // Find all matching prefixes and add the resulting words.
36859ccfebddSBram Moolenaar 	    if (store_aff_word(spin, dw, afflist, affile,
36869ccfebddSBram Moolenaar 			  &affile->af_pref, NULL,
36879ccfebddSBram Moolenaar 			    CONDIT_SUF, flags, store_afflist, pfxlen) == FAIL)
36889ccfebddSBram Moolenaar 		retval = FAIL;
36899ccfebddSBram Moolenaar 	}
36909ccfebddSBram Moolenaar 
36919ccfebddSBram Moolenaar 	vim_free(pc);
36929ccfebddSBram Moolenaar     }
36939ccfebddSBram Moolenaar 
36949ccfebddSBram Moolenaar     if (duplicate > 0)
3695f9e3e09fSBram Moolenaar 	smsg(_("%d duplicate word(s) in %s"), duplicate, fname);
36969ccfebddSBram Moolenaar     if (spin->si_ascii && non_ascii > 0)
3697f9e3e09fSBram Moolenaar 	smsg(_("Ignored %d word(s) with non-ASCII characters in %s"),
36989ccfebddSBram Moolenaar 							    non_ascii, fname);
36999ccfebddSBram Moolenaar     hash_clear(&ht);
37009ccfebddSBram Moolenaar 
37019ccfebddSBram Moolenaar     fclose(fd);
37029ccfebddSBram Moolenaar     return retval;
37039ccfebddSBram Moolenaar }
37049ccfebddSBram Moolenaar 
37059ccfebddSBram Moolenaar /*
37069ccfebddSBram Moolenaar  * Check for affix flags in "afflist" that are turned into word flags.
37079ccfebddSBram Moolenaar  * Return WF_ flags.
37089ccfebddSBram Moolenaar  */
37099ccfebddSBram Moolenaar     static int
get_affix_flags(afffile_T * affile,char_u * afflist)37109ccfebddSBram Moolenaar get_affix_flags(afffile_T *affile, char_u *afflist)
37119ccfebddSBram Moolenaar {
37129ccfebddSBram Moolenaar     int		flags = 0;
37139ccfebddSBram Moolenaar 
37149ccfebddSBram Moolenaar     if (affile->af_keepcase != 0 && flag_in_afflist(
37159ccfebddSBram Moolenaar 			   affile->af_flagtype, afflist, affile->af_keepcase))
37169ccfebddSBram Moolenaar 	flags |= WF_KEEPCAP | WF_FIXCAP;
37179ccfebddSBram Moolenaar     if (affile->af_rare != 0 && flag_in_afflist(
37189ccfebddSBram Moolenaar 			       affile->af_flagtype, afflist, affile->af_rare))
37199ccfebddSBram Moolenaar 	flags |= WF_RARE;
37209ccfebddSBram Moolenaar     if (affile->af_bad != 0 && flag_in_afflist(
37219ccfebddSBram Moolenaar 				affile->af_flagtype, afflist, affile->af_bad))
37229ccfebddSBram Moolenaar 	flags |= WF_BANNED;
37239ccfebddSBram Moolenaar     if (affile->af_needcomp != 0 && flag_in_afflist(
37249ccfebddSBram Moolenaar 			   affile->af_flagtype, afflist, affile->af_needcomp))
37259ccfebddSBram Moolenaar 	flags |= WF_NEEDCOMP;
37269ccfebddSBram Moolenaar     if (affile->af_comproot != 0 && flag_in_afflist(
37279ccfebddSBram Moolenaar 			   affile->af_flagtype, afflist, affile->af_comproot))
37289ccfebddSBram Moolenaar 	flags |= WF_COMPROOT;
37299ccfebddSBram Moolenaar     if (affile->af_nosuggest != 0 && flag_in_afflist(
37309ccfebddSBram Moolenaar 			  affile->af_flagtype, afflist, affile->af_nosuggest))
37319ccfebddSBram Moolenaar 	flags |= WF_NOSUGGEST;
37329ccfebddSBram Moolenaar     return flags;
37339ccfebddSBram Moolenaar }
37349ccfebddSBram Moolenaar 
37359ccfebddSBram Moolenaar /*
37369ccfebddSBram Moolenaar  * Get the list of prefix IDs from the affix list "afflist".
37379ccfebddSBram Moolenaar  * Used for PFXPOSTPONE.
37389ccfebddSBram Moolenaar  * Put the resulting flags in "store_afflist[MAXWLEN]" with a terminating NUL
37399ccfebddSBram Moolenaar  * and return the number of affixes.
37409ccfebddSBram Moolenaar  */
37419ccfebddSBram Moolenaar     static int
get_pfxlist(afffile_T * affile,char_u * afflist,char_u * store_afflist)37429ccfebddSBram Moolenaar get_pfxlist(
37439ccfebddSBram Moolenaar     afffile_T	*affile,
37449ccfebddSBram Moolenaar     char_u	*afflist,
37459ccfebddSBram Moolenaar     char_u	*store_afflist)
37469ccfebddSBram Moolenaar {
37479ccfebddSBram Moolenaar     char_u	*p;
37489ccfebddSBram Moolenaar     char_u	*prevp;
37499ccfebddSBram Moolenaar     int		cnt = 0;
37509ccfebddSBram Moolenaar     int		id;
37519ccfebddSBram Moolenaar     char_u	key[AH_KEY_LEN];
37529ccfebddSBram Moolenaar     hashitem_T	*hi;
37539ccfebddSBram Moolenaar 
37549ccfebddSBram Moolenaar     for (p = afflist; *p != NUL; )
37559ccfebddSBram Moolenaar     {
37569ccfebddSBram Moolenaar 	prevp = p;
37579ccfebddSBram Moolenaar 	if (get_affitem(affile->af_flagtype, &p) != 0)
37589ccfebddSBram Moolenaar 	{
37590d6f5d97SBram Moolenaar 	    // A flag is a postponed prefix flag if it appears in "af_pref"
37600d6f5d97SBram Moolenaar 	    // and its ID is not zero.
37619ccfebddSBram Moolenaar 	    vim_strncpy(key, prevp, p - prevp);
37629ccfebddSBram Moolenaar 	    hi = hash_find(&affile->af_pref, key);
37639ccfebddSBram Moolenaar 	    if (!HASHITEM_EMPTY(hi))
37649ccfebddSBram Moolenaar 	    {
37659ccfebddSBram Moolenaar 		id = HI2AH(hi)->ah_newID;
37669ccfebddSBram Moolenaar 		if (id != 0)
37679ccfebddSBram Moolenaar 		    store_afflist[cnt++] = id;
37689ccfebddSBram Moolenaar 	    }
37699ccfebddSBram Moolenaar 	}
37709ccfebddSBram Moolenaar 	if (affile->af_flagtype == AFT_NUM && *p == ',')
37719ccfebddSBram Moolenaar 	    ++p;
37729ccfebddSBram Moolenaar     }
37739ccfebddSBram Moolenaar 
37749ccfebddSBram Moolenaar     store_afflist[cnt] = NUL;
37759ccfebddSBram Moolenaar     return cnt;
37769ccfebddSBram Moolenaar }
37779ccfebddSBram Moolenaar 
37789ccfebddSBram Moolenaar /*
37799ccfebddSBram Moolenaar  * Get the list of compound IDs from the affix list "afflist" that are used
37809ccfebddSBram Moolenaar  * for compound words.
37819ccfebddSBram Moolenaar  * Puts the flags in "store_afflist[]".
37829ccfebddSBram Moolenaar  */
37839ccfebddSBram Moolenaar     static void
get_compflags(afffile_T * affile,char_u * afflist,char_u * store_afflist)37849ccfebddSBram Moolenaar get_compflags(
37859ccfebddSBram Moolenaar     afffile_T	*affile,
37869ccfebddSBram Moolenaar     char_u	*afflist,
37879ccfebddSBram Moolenaar     char_u	*store_afflist)
37889ccfebddSBram Moolenaar {
37899ccfebddSBram Moolenaar     char_u	*p;
37909ccfebddSBram Moolenaar     char_u	*prevp;
37919ccfebddSBram Moolenaar     int		cnt = 0;
37929ccfebddSBram Moolenaar     char_u	key[AH_KEY_LEN];
37939ccfebddSBram Moolenaar     hashitem_T	*hi;
37949ccfebddSBram Moolenaar 
37959ccfebddSBram Moolenaar     for (p = afflist; *p != NUL; )
37969ccfebddSBram Moolenaar     {
37979ccfebddSBram Moolenaar 	prevp = p;
37989ccfebddSBram Moolenaar 	if (get_affitem(affile->af_flagtype, &p) != 0)
37999ccfebddSBram Moolenaar 	{
38000d6f5d97SBram Moolenaar 	    // A flag is a compound flag if it appears in "af_comp".
38019ccfebddSBram Moolenaar 	    vim_strncpy(key, prevp, p - prevp);
38029ccfebddSBram Moolenaar 	    hi = hash_find(&affile->af_comp, key);
38039ccfebddSBram Moolenaar 	    if (!HASHITEM_EMPTY(hi))
38049ccfebddSBram Moolenaar 		store_afflist[cnt++] = HI2CI(hi)->ci_newID;
38059ccfebddSBram Moolenaar 	}
38069ccfebddSBram Moolenaar 	if (affile->af_flagtype == AFT_NUM && *p == ',')
38079ccfebddSBram Moolenaar 	    ++p;
38089ccfebddSBram Moolenaar     }
38099ccfebddSBram Moolenaar 
38109ccfebddSBram Moolenaar     store_afflist[cnt] = NUL;
38119ccfebddSBram Moolenaar }
38129ccfebddSBram Moolenaar 
38139ccfebddSBram Moolenaar /*
38149ccfebddSBram Moolenaar  * Apply affixes to a word and store the resulting words.
38159ccfebddSBram Moolenaar  * "ht" is the hashtable with affentry_T that need to be applied, either
38169ccfebddSBram Moolenaar  * prefixes or suffixes.
38179ccfebddSBram Moolenaar  * "xht", when not NULL, is the prefix hashtable, to be used additionally on
38189ccfebddSBram Moolenaar  * the resulting words for combining affixes.
38199ccfebddSBram Moolenaar  *
38209ccfebddSBram Moolenaar  * Returns FAIL when out of memory.
38219ccfebddSBram Moolenaar  */
38229ccfebddSBram Moolenaar     static int
store_aff_word(spellinfo_T * spin,char_u * word,char_u * afflist,afffile_T * affile,hashtab_T * ht,hashtab_T * xht,int condit,int flags,char_u * pfxlist,int pfxlen)38239ccfebddSBram Moolenaar store_aff_word(
38240d6f5d97SBram Moolenaar     spellinfo_T	*spin,		// spell info
38250d6f5d97SBram Moolenaar     char_u	*word,		// basic word start
38260d6f5d97SBram Moolenaar     char_u	*afflist,	// list of names of supported affixes
38279ccfebddSBram Moolenaar     afffile_T	*affile,
38289ccfebddSBram Moolenaar     hashtab_T	*ht,
38299ccfebddSBram Moolenaar     hashtab_T	*xht,
38300d6f5d97SBram Moolenaar     int		condit,		// CONDIT_SUF et al.
38310d6f5d97SBram Moolenaar     int		flags,		// flags for the word
38320d6f5d97SBram Moolenaar     char_u	*pfxlist,	// list of prefix IDs
38330d6f5d97SBram Moolenaar     int		pfxlen)		// nr of flags in "pfxlist" for prefixes, rest
38340d6f5d97SBram Moolenaar 				// is compound flags
38359ccfebddSBram Moolenaar {
38369ccfebddSBram Moolenaar     int		todo;
38379ccfebddSBram Moolenaar     hashitem_T	*hi;
38389ccfebddSBram Moolenaar     affheader_T	*ah;
38399ccfebddSBram Moolenaar     affentry_T	*ae;
38409ccfebddSBram Moolenaar     char_u	newword[MAXWLEN];
38419ccfebddSBram Moolenaar     int		retval = OK;
38429ccfebddSBram Moolenaar     int		i, j;
38439ccfebddSBram Moolenaar     char_u	*p;
38449ccfebddSBram Moolenaar     int		use_flags;
38459ccfebddSBram Moolenaar     char_u	*use_pfxlist;
38469ccfebddSBram Moolenaar     int		use_pfxlen;
38479ccfebddSBram Moolenaar     int		need_affix;
38489ccfebddSBram Moolenaar     char_u	store_afflist[MAXWLEN];
38499ccfebddSBram Moolenaar     char_u	pfx_pfxlist[MAXWLEN];
38509ccfebddSBram Moolenaar     size_t	wordlen = STRLEN(word);
38519ccfebddSBram Moolenaar     int		use_condit;
38529ccfebddSBram Moolenaar 
38539ccfebddSBram Moolenaar     todo = (int)ht->ht_used;
38549ccfebddSBram Moolenaar     for (hi = ht->ht_array; todo > 0 && retval == OK; ++hi)
38559ccfebddSBram Moolenaar     {
38569ccfebddSBram Moolenaar 	if (!HASHITEM_EMPTY(hi))
38579ccfebddSBram Moolenaar 	{
38589ccfebddSBram Moolenaar 	    --todo;
38599ccfebddSBram Moolenaar 	    ah = HI2AH(hi);
38609ccfebddSBram Moolenaar 
38610d6f5d97SBram Moolenaar 	    // Check that the affix combines, if required, and that the word
38620d6f5d97SBram Moolenaar 	    // supports this affix.
38639ccfebddSBram Moolenaar 	    if (((condit & CONDIT_COMB) == 0 || ah->ah_combine)
38649ccfebddSBram Moolenaar 		    && flag_in_afflist(affile->af_flagtype, afflist,
38659ccfebddSBram Moolenaar 								 ah->ah_flag))
38669ccfebddSBram Moolenaar 	    {
38670d6f5d97SBram Moolenaar 		// Loop over all affix entries with this name.
38689ccfebddSBram Moolenaar 		for (ae = ah->ah_first; ae != NULL; ae = ae->ae_next)
38699ccfebddSBram Moolenaar 		{
38700d6f5d97SBram Moolenaar 		    // Check the condition.  It's not logical to match case
38710d6f5d97SBram Moolenaar 		    // here, but it is required for compatibility with
38720d6f5d97SBram Moolenaar 		    // Myspell.
38730d6f5d97SBram Moolenaar 		    // Another requirement from Myspell is that the chop
38740d6f5d97SBram Moolenaar 		    // string is shorter than the word itself.
38750d6f5d97SBram Moolenaar 		    // For prefixes, when "PFXPOSTPONE" was used, only do
38760d6f5d97SBram Moolenaar 		    // prefixes with a chop string and/or flags.
38770d6f5d97SBram Moolenaar 		    // When a previously added affix had CIRCUMFIX this one
38780d6f5d97SBram Moolenaar 		    // must have it too, if it had not then this one must not
38790d6f5d97SBram Moolenaar 		    // have one either.
38809ccfebddSBram Moolenaar 		    if ((xht != NULL || !affile->af_pfxpostpone
38819ccfebddSBram Moolenaar 				|| ae->ae_chop != NULL
38829ccfebddSBram Moolenaar 				|| ae->ae_flags != NULL)
38839ccfebddSBram Moolenaar 			    && (ae->ae_chop == NULL
38849ccfebddSBram Moolenaar 				|| STRLEN(ae->ae_chop) < wordlen)
38859ccfebddSBram Moolenaar 			    && (ae->ae_prog == NULL
38869ccfebddSBram Moolenaar 				|| vim_regexec_prog(&ae->ae_prog, FALSE,
38879ccfebddSBram Moolenaar 							    word, (colnr_T)0))
38889ccfebddSBram Moolenaar 			    && (((condit & CONDIT_CFIX) == 0)
38899ccfebddSBram Moolenaar 				== ((condit & CONDIT_AFF) == 0
38909ccfebddSBram Moolenaar 				    || ae->ae_flags == NULL
38919ccfebddSBram Moolenaar 				    || !flag_in_afflist(affile->af_flagtype,
38929ccfebddSBram Moolenaar 					ae->ae_flags, affile->af_circumfix))))
38939ccfebddSBram Moolenaar 		    {
38940d6f5d97SBram Moolenaar 			// Match.  Remove the chop and add the affix.
38959ccfebddSBram Moolenaar 			if (xht == NULL)
38969ccfebddSBram Moolenaar 			{
38970d6f5d97SBram Moolenaar 			    // prefix: chop/add at the start of the word
38989ccfebddSBram Moolenaar 			    if (ae->ae_add == NULL)
38999ccfebddSBram Moolenaar 				*newword = NUL;
39009ccfebddSBram Moolenaar 			    else
39019ccfebddSBram Moolenaar 				vim_strncpy(newword, ae->ae_add, MAXWLEN - 1);
39029ccfebddSBram Moolenaar 			    p = word;
39039ccfebddSBram Moolenaar 			    if (ae->ae_chop != NULL)
39049ccfebddSBram Moolenaar 			    {
39050d6f5d97SBram Moolenaar 				// Skip chop string.
39069ccfebddSBram Moolenaar 				if (has_mbyte)
39079ccfebddSBram Moolenaar 				{
39089ccfebddSBram Moolenaar 				    i = mb_charlen(ae->ae_chop);
39099ccfebddSBram Moolenaar 				    for ( ; i > 0; --i)
391091acfffcSBram Moolenaar 					MB_PTR_ADV(p);
39119ccfebddSBram Moolenaar 				}
39129ccfebddSBram Moolenaar 				else
39139ccfebddSBram Moolenaar 				    p += STRLEN(ae->ae_chop);
39149ccfebddSBram Moolenaar 			    }
39159ccfebddSBram Moolenaar 			    STRCAT(newword, p);
39169ccfebddSBram Moolenaar 			}
39179ccfebddSBram Moolenaar 			else
39189ccfebddSBram Moolenaar 			{
39190d6f5d97SBram Moolenaar 			    // suffix: chop/add at the end of the word
39209ccfebddSBram Moolenaar 			    vim_strncpy(newword, word, MAXWLEN - 1);
39219ccfebddSBram Moolenaar 			    if (ae->ae_chop != NULL)
39229ccfebddSBram Moolenaar 			    {
39230d6f5d97SBram Moolenaar 				// Remove chop string.
39249ccfebddSBram Moolenaar 				p = newword + STRLEN(newword);
39259ccfebddSBram Moolenaar 				i = (int)MB_CHARLEN(ae->ae_chop);
39269ccfebddSBram Moolenaar 				for ( ; i > 0; --i)
392791acfffcSBram Moolenaar 				    MB_PTR_BACK(newword, p);
39289ccfebddSBram Moolenaar 				*p = NUL;
39299ccfebddSBram Moolenaar 			    }
39309ccfebddSBram Moolenaar 			    if (ae->ae_add != NULL)
39319ccfebddSBram Moolenaar 				STRCAT(newword, ae->ae_add);
39329ccfebddSBram Moolenaar 			}
39339ccfebddSBram Moolenaar 
39349ccfebddSBram Moolenaar 			use_flags = flags;
39359ccfebddSBram Moolenaar 			use_pfxlist = pfxlist;
39369ccfebddSBram Moolenaar 			use_pfxlen = pfxlen;
39379ccfebddSBram Moolenaar 			need_affix = FALSE;
39389ccfebddSBram Moolenaar 			use_condit = condit | CONDIT_COMB | CONDIT_AFF;
39399ccfebddSBram Moolenaar 			if (ae->ae_flags != NULL)
39409ccfebddSBram Moolenaar 			{
39410d6f5d97SBram Moolenaar 			    // Extract flags from the affix list.
39429ccfebddSBram Moolenaar 			    use_flags |= get_affix_flags(affile, ae->ae_flags);
39439ccfebddSBram Moolenaar 
39449ccfebddSBram Moolenaar 			    if (affile->af_needaffix != 0 && flag_in_afflist(
39459ccfebddSBram Moolenaar 					affile->af_flagtype, ae->ae_flags,
39469ccfebddSBram Moolenaar 							affile->af_needaffix))
39479ccfebddSBram Moolenaar 				need_affix = TRUE;
39489ccfebddSBram Moolenaar 
39490d6f5d97SBram Moolenaar 			    // When there is a CIRCUMFIX flag the other affix
39500d6f5d97SBram Moolenaar 			    // must also have it and we don't add the word
39510d6f5d97SBram Moolenaar 			    // with one affix.
39529ccfebddSBram Moolenaar 			    if (affile->af_circumfix != 0 && flag_in_afflist(
39539ccfebddSBram Moolenaar 					affile->af_flagtype, ae->ae_flags,
39549ccfebddSBram Moolenaar 							affile->af_circumfix))
39559ccfebddSBram Moolenaar 			    {
39569ccfebddSBram Moolenaar 				use_condit |= CONDIT_CFIX;
39579ccfebddSBram Moolenaar 				if ((condit & CONDIT_CFIX) == 0)
39589ccfebddSBram Moolenaar 				    need_affix = TRUE;
39599ccfebddSBram Moolenaar 			    }
39609ccfebddSBram Moolenaar 
39619ccfebddSBram Moolenaar 			    if (affile->af_pfxpostpone
39629ccfebddSBram Moolenaar 						|| spin->si_compflags != NULL)
39639ccfebddSBram Moolenaar 			    {
39649ccfebddSBram Moolenaar 				if (affile->af_pfxpostpone)
39650d6f5d97SBram Moolenaar 				    // Get prefix IDS from the affix list.
39669ccfebddSBram Moolenaar 				    use_pfxlen = get_pfxlist(affile,
39679ccfebddSBram Moolenaar 						 ae->ae_flags, store_afflist);
39689ccfebddSBram Moolenaar 				else
39699ccfebddSBram Moolenaar 				    use_pfxlen = 0;
39709ccfebddSBram Moolenaar 				use_pfxlist = store_afflist;
39719ccfebddSBram Moolenaar 
39720d6f5d97SBram Moolenaar 				// Combine the prefix IDs. Avoid adding the
39730d6f5d97SBram Moolenaar 				// same ID twice.
39749ccfebddSBram Moolenaar 				for (i = 0; i < pfxlen; ++i)
39759ccfebddSBram Moolenaar 				{
39769ccfebddSBram Moolenaar 				    for (j = 0; j < use_pfxlen; ++j)
39779ccfebddSBram Moolenaar 					if (pfxlist[i] == use_pfxlist[j])
39789ccfebddSBram Moolenaar 					    break;
39799ccfebddSBram Moolenaar 				    if (j == use_pfxlen)
39809ccfebddSBram Moolenaar 					use_pfxlist[use_pfxlen++] = pfxlist[i];
39819ccfebddSBram Moolenaar 				}
39829ccfebddSBram Moolenaar 
39839ccfebddSBram Moolenaar 				if (spin->si_compflags != NULL)
39840d6f5d97SBram Moolenaar 				    // Get compound IDS from the affix list.
39859ccfebddSBram Moolenaar 				    get_compflags(affile, ae->ae_flags,
39869ccfebddSBram Moolenaar 						  use_pfxlist + use_pfxlen);
39879ccfebddSBram Moolenaar 
39880d6f5d97SBram Moolenaar 				// Combine the list of compound flags.
39890d6f5d97SBram Moolenaar 				// Concatenate them to the prefix IDs list.
39900d6f5d97SBram Moolenaar 				// Avoid adding the same ID twice.
39919ccfebddSBram Moolenaar 				for (i = pfxlen; pfxlist[i] != NUL; ++i)
39929ccfebddSBram Moolenaar 				{
39939ccfebddSBram Moolenaar 				    for (j = use_pfxlen;
39949ccfebddSBram Moolenaar 						   use_pfxlist[j] != NUL; ++j)
39959ccfebddSBram Moolenaar 					if (pfxlist[i] == use_pfxlist[j])
39969ccfebddSBram Moolenaar 					    break;
39979ccfebddSBram Moolenaar 				    if (use_pfxlist[j] == NUL)
39989ccfebddSBram Moolenaar 				    {
39999ccfebddSBram Moolenaar 					use_pfxlist[j++] = pfxlist[i];
40009ccfebddSBram Moolenaar 					use_pfxlist[j] = NUL;
40019ccfebddSBram Moolenaar 				    }
40029ccfebddSBram Moolenaar 				}
40039ccfebddSBram Moolenaar 			    }
40049ccfebddSBram Moolenaar 			}
40059ccfebddSBram Moolenaar 
40060d6f5d97SBram Moolenaar 			// Obey a "COMPOUNDFORBIDFLAG" of the affix: don't
40070d6f5d97SBram Moolenaar 			// use the compound flags.
40089ccfebddSBram Moolenaar 			if (use_pfxlist != NULL && ae->ae_compforbid)
40099ccfebddSBram Moolenaar 			{
40109ccfebddSBram Moolenaar 			    vim_strncpy(pfx_pfxlist, use_pfxlist, use_pfxlen);
40119ccfebddSBram Moolenaar 			    use_pfxlist = pfx_pfxlist;
40129ccfebddSBram Moolenaar 			}
40139ccfebddSBram Moolenaar 
40140d6f5d97SBram Moolenaar 			// When there are postponed prefixes...
40159ccfebddSBram Moolenaar 			if (spin->si_prefroot != NULL
40169ccfebddSBram Moolenaar 				&& spin->si_prefroot->wn_sibling != NULL)
40179ccfebddSBram Moolenaar 			{
40180d6f5d97SBram Moolenaar 			    // ... add a flag to indicate an affix was used.
40199ccfebddSBram Moolenaar 			    use_flags |= WF_HAS_AFF;
40209ccfebddSBram Moolenaar 
40210d6f5d97SBram Moolenaar 			    // ... don't use a prefix list if combining
40220d6f5d97SBram Moolenaar 			    // affixes is not allowed.  But do use the
40230d6f5d97SBram Moolenaar 			    // compound flags after them.
40249ccfebddSBram Moolenaar 			    if (!ah->ah_combine && use_pfxlist != NULL)
40259ccfebddSBram Moolenaar 				use_pfxlist += use_pfxlen;
40269ccfebddSBram Moolenaar 			}
40279ccfebddSBram Moolenaar 
40280d6f5d97SBram Moolenaar 			// When compounding is supported and there is no
40290d6f5d97SBram Moolenaar 			// "COMPOUNDPERMITFLAG" then forbid compounding on the
40300d6f5d97SBram Moolenaar 			// side where the affix is applied.
40319ccfebddSBram Moolenaar 			if (spin->si_compflags != NULL && !ae->ae_comppermit)
40329ccfebddSBram Moolenaar 			{
40339ccfebddSBram Moolenaar 			    if (xht != NULL)
40349ccfebddSBram Moolenaar 				use_flags |= WF_NOCOMPAFT;
40359ccfebddSBram Moolenaar 			    else
40369ccfebddSBram Moolenaar 				use_flags |= WF_NOCOMPBEF;
40379ccfebddSBram Moolenaar 			}
40389ccfebddSBram Moolenaar 
40390d6f5d97SBram Moolenaar 			// Store the modified word.
40409ccfebddSBram Moolenaar 			if (store_word(spin, newword, use_flags,
40419ccfebddSBram Moolenaar 						 spin->si_region, use_pfxlist,
40429ccfebddSBram Moolenaar 							  need_affix) == FAIL)
40439ccfebddSBram Moolenaar 			    retval = FAIL;
40449ccfebddSBram Moolenaar 
40450d6f5d97SBram Moolenaar 			// When added a prefix or a first suffix and the affix
40460d6f5d97SBram Moolenaar 			// has flags may add a(nother) suffix.  RECURSIVE!
40479ccfebddSBram Moolenaar 			if ((condit & CONDIT_SUF) && ae->ae_flags != NULL)
40489ccfebddSBram Moolenaar 			    if (store_aff_word(spin, newword, ae->ae_flags,
40499ccfebddSBram Moolenaar 					affile, &affile->af_suff, xht,
40509ccfebddSBram Moolenaar 					   use_condit & (xht == NULL
40519ccfebddSBram Moolenaar 							? ~0 :  ~CONDIT_SUF),
40529ccfebddSBram Moolenaar 				      use_flags, use_pfxlist, pfxlen) == FAIL)
40539ccfebddSBram Moolenaar 				retval = FAIL;
40549ccfebddSBram Moolenaar 
40550d6f5d97SBram Moolenaar 			// When added a suffix and combining is allowed also
40560d6f5d97SBram Moolenaar 			// try adding a prefix additionally.  Both for the
40570d6f5d97SBram Moolenaar 			// word flags and for the affix flags.  RECURSIVE!
40589ccfebddSBram Moolenaar 			if (xht != NULL && ah->ah_combine)
40599ccfebddSBram Moolenaar 			{
40609ccfebddSBram Moolenaar 			    if (store_aff_word(spin, newword,
40619ccfebddSBram Moolenaar 					afflist, affile,
40629ccfebddSBram Moolenaar 					xht, NULL, use_condit,
40639ccfebddSBram Moolenaar 					use_flags, use_pfxlist,
40649ccfebddSBram Moolenaar 					pfxlen) == FAIL
40659ccfebddSBram Moolenaar 				    || (ae->ae_flags != NULL
40669ccfebddSBram Moolenaar 					&& store_aff_word(spin, newword,
40679ccfebddSBram Moolenaar 					    ae->ae_flags, affile,
40689ccfebddSBram Moolenaar 					    xht, NULL, use_condit,
40699ccfebddSBram Moolenaar 					    use_flags, use_pfxlist,
40709ccfebddSBram Moolenaar 					    pfxlen) == FAIL))
40719ccfebddSBram Moolenaar 				retval = FAIL;
40729ccfebddSBram Moolenaar 			}
40739ccfebddSBram Moolenaar 		    }
40749ccfebddSBram Moolenaar 		}
40759ccfebddSBram Moolenaar 	    }
40769ccfebddSBram Moolenaar 	}
40779ccfebddSBram Moolenaar     }
40789ccfebddSBram Moolenaar 
40799ccfebddSBram Moolenaar     return retval;
40809ccfebddSBram Moolenaar }
40819ccfebddSBram Moolenaar 
40829ccfebddSBram Moolenaar /*
40839ccfebddSBram Moolenaar  * Read a file with a list of words.
40849ccfebddSBram Moolenaar  */
40859ccfebddSBram Moolenaar     static int
spell_read_wordfile(spellinfo_T * spin,char_u * fname)40869ccfebddSBram Moolenaar spell_read_wordfile(spellinfo_T *spin, char_u *fname)
40879ccfebddSBram Moolenaar {
40889ccfebddSBram Moolenaar     FILE	*fd;
40899ccfebddSBram Moolenaar     long	lnum = 0;
40909ccfebddSBram Moolenaar     char_u	rline[MAXLINELEN];
40919ccfebddSBram Moolenaar     char_u	*line;
40929ccfebddSBram Moolenaar     char_u	*pc = NULL;
40939ccfebddSBram Moolenaar     char_u	*p;
40949ccfebddSBram Moolenaar     int		l;
40959ccfebddSBram Moolenaar     int		retval = OK;
40969ccfebddSBram Moolenaar     int		did_word = FALSE;
40979ccfebddSBram Moolenaar     int		non_ascii = 0;
40989ccfebddSBram Moolenaar     int		flags;
40999ccfebddSBram Moolenaar     int		regionmask;
41009ccfebddSBram Moolenaar 
41019ccfebddSBram Moolenaar     /*
41029ccfebddSBram Moolenaar      * Open the file.
41039ccfebddSBram Moolenaar      */
41049ccfebddSBram Moolenaar     fd = mch_fopen((char *)fname, "r");
41059ccfebddSBram Moolenaar     if (fd == NULL)
41069ccfebddSBram Moolenaar     {
4107f9e3e09fSBram Moolenaar 	semsg(_(e_notopen), fname);
41089ccfebddSBram Moolenaar 	return FAIL;
41099ccfebddSBram Moolenaar     }
41109ccfebddSBram Moolenaar 
41119ccfebddSBram Moolenaar     vim_snprintf((char *)IObuff, IOSIZE, _("Reading word file %s..."), fname);
41129ccfebddSBram Moolenaar     spell_message(spin, IObuff);
41139ccfebddSBram Moolenaar 
41149ccfebddSBram Moolenaar     /*
41159ccfebddSBram Moolenaar      * Read all the lines in the file one by one.
41169ccfebddSBram Moolenaar      */
41179ccfebddSBram Moolenaar     while (!vim_fgets(rline, MAXLINELEN, fd) && !got_int)
41189ccfebddSBram Moolenaar     {
41199ccfebddSBram Moolenaar 	line_breakcheck();
41209ccfebddSBram Moolenaar 	++lnum;
41219ccfebddSBram Moolenaar 
41220d6f5d97SBram Moolenaar 	// Skip comment lines.
41239ccfebddSBram Moolenaar 	if (*rline == '#')
41249ccfebddSBram Moolenaar 	    continue;
41259ccfebddSBram Moolenaar 
41260d6f5d97SBram Moolenaar 	// Remove CR, LF and white space from the end.
41279ccfebddSBram Moolenaar 	l = (int)STRLEN(rline);
41289ccfebddSBram Moolenaar 	while (l > 0 && rline[l - 1] <= ' ')
41299ccfebddSBram Moolenaar 	    --l;
41309ccfebddSBram Moolenaar 	if (l == 0)
41310d6f5d97SBram Moolenaar 	    continue;	// empty or blank line
41329ccfebddSBram Moolenaar 	rline[l] = NUL;
41339ccfebddSBram Moolenaar 
41340d6f5d97SBram Moolenaar 	// Convert from "/encoding={encoding}" to 'encoding' when needed.
41359ccfebddSBram Moolenaar 	vim_free(pc);
41369ccfebddSBram Moolenaar 	if (spin->si_conv.vc_type != CONV_NONE)
41379ccfebddSBram Moolenaar 	{
41389ccfebddSBram Moolenaar 	    pc = string_convert(&spin->si_conv, rline, NULL);
41399ccfebddSBram Moolenaar 	    if (pc == NULL)
41409ccfebddSBram Moolenaar 	    {
4141db99f9f2SBram Moolenaar 		smsg(_("Conversion failure for word in %s line %ld: %s"),
41429ccfebddSBram Moolenaar 							   fname, lnum, rline);
41439ccfebddSBram Moolenaar 		continue;
41449ccfebddSBram Moolenaar 	    }
41459ccfebddSBram Moolenaar 	    line = pc;
41469ccfebddSBram Moolenaar 	}
41479ccfebddSBram Moolenaar 	else
41489ccfebddSBram Moolenaar 	{
41499ccfebddSBram Moolenaar 	    pc = NULL;
41509ccfebddSBram Moolenaar 	    line = rline;
41519ccfebddSBram Moolenaar 	}
41529ccfebddSBram Moolenaar 
41539ccfebddSBram Moolenaar 	if (*line == '/')
41549ccfebddSBram Moolenaar 	{
41559ccfebddSBram Moolenaar 	    ++line;
41569ccfebddSBram Moolenaar 	    if (STRNCMP(line, "encoding=", 9) == 0)
41579ccfebddSBram Moolenaar 	    {
41589ccfebddSBram Moolenaar 		if (spin->si_conv.vc_type != CONV_NONE)
4159db99f9f2SBram Moolenaar 		    smsg(_("Duplicate /encoding= line ignored in %s line %ld: %s"),
41609ccfebddSBram Moolenaar 						       fname, lnum, line - 1);
41619ccfebddSBram Moolenaar 		else if (did_word)
4162db99f9f2SBram Moolenaar 		    smsg(_("/encoding= line after word ignored in %s line %ld: %s"),
41639ccfebddSBram Moolenaar 						       fname, lnum, line - 1);
41649ccfebddSBram Moolenaar 		else
41659ccfebddSBram Moolenaar 		{
41669ccfebddSBram Moolenaar 		    char_u	*enc;
41679ccfebddSBram Moolenaar 
41680d6f5d97SBram Moolenaar 		    // Setup for conversion to 'encoding'.
41699ccfebddSBram Moolenaar 		    line += 9;
41709ccfebddSBram Moolenaar 		    enc = enc_canonize(line);
41719ccfebddSBram Moolenaar 		    if (enc != NULL && !spin->si_ascii
41729ccfebddSBram Moolenaar 			    && convert_setup(&spin->si_conv, enc,
41739ccfebddSBram Moolenaar 							       p_enc) == FAIL)
4174f9e3e09fSBram Moolenaar 			smsg(_("Conversion in %s not supported: from %s to %s"),
41759ccfebddSBram Moolenaar 							  fname, line, p_enc);
41769ccfebddSBram Moolenaar 		    vim_free(enc);
41779ccfebddSBram Moolenaar 		    spin->si_conv.vc_fail = TRUE;
41789ccfebddSBram Moolenaar 		}
41799ccfebddSBram Moolenaar 		continue;
41809ccfebddSBram Moolenaar 	    }
41819ccfebddSBram Moolenaar 
41829ccfebddSBram Moolenaar 	    if (STRNCMP(line, "regions=", 8) == 0)
41839ccfebddSBram Moolenaar 	    {
41849ccfebddSBram Moolenaar 		if (spin->si_region_count > 1)
4185db99f9f2SBram Moolenaar 		    smsg(_("Duplicate /regions= line ignored in %s line %ld: %s"),
41869ccfebddSBram Moolenaar 						       fname, lnum, line);
41879ccfebddSBram Moolenaar 		else
41889ccfebddSBram Moolenaar 		{
41899ccfebddSBram Moolenaar 		    line += 8;
41902993ac5fSBram Moolenaar 		    if (STRLEN(line) > MAXREGIONS * 2)
4191db99f9f2SBram Moolenaar 			smsg(_("Too many regions in %s line %ld: %s"),
41929ccfebddSBram Moolenaar 						       fname, lnum, line);
41939ccfebddSBram Moolenaar 		    else
41949ccfebddSBram Moolenaar 		    {
41959ccfebddSBram Moolenaar 			spin->si_region_count = (int)STRLEN(line) / 2;
41969ccfebddSBram Moolenaar 			STRCPY(spin->si_region_name, line);
41979ccfebddSBram Moolenaar 
41980d6f5d97SBram Moolenaar 			// Adjust the mask for a word valid in all regions.
41999ccfebddSBram Moolenaar 			spin->si_region = (1 << spin->si_region_count) - 1;
42009ccfebddSBram Moolenaar 		    }
42019ccfebddSBram Moolenaar 		}
42029ccfebddSBram Moolenaar 		continue;
42039ccfebddSBram Moolenaar 	    }
42049ccfebddSBram Moolenaar 
4205db99f9f2SBram Moolenaar 	    smsg(_("/ line ignored in %s line %ld: %s"),
42069ccfebddSBram Moolenaar 						       fname, lnum, line - 1);
42079ccfebddSBram Moolenaar 	    continue;
42089ccfebddSBram Moolenaar 	}
42099ccfebddSBram Moolenaar 
42109ccfebddSBram Moolenaar 	flags = 0;
42119ccfebddSBram Moolenaar 	regionmask = spin->si_region;
42129ccfebddSBram Moolenaar 
42130d6f5d97SBram Moolenaar 	// Check for flags and region after a slash.
42149ccfebddSBram Moolenaar 	p = vim_strchr(line, '/');
42159ccfebddSBram Moolenaar 	if (p != NULL)
42169ccfebddSBram Moolenaar 	{
42179ccfebddSBram Moolenaar 	    *p++ = NUL;
42189ccfebddSBram Moolenaar 	    while (*p != NUL)
42199ccfebddSBram Moolenaar 	    {
42200d6f5d97SBram Moolenaar 		if (*p == '=')		// keep-case word
42219ccfebddSBram Moolenaar 		    flags |= WF_KEEPCAP | WF_FIXCAP;
42220d6f5d97SBram Moolenaar 		else if (*p == '!')	// Bad, bad, wicked word.
42239ccfebddSBram Moolenaar 		    flags |= WF_BANNED;
42240d6f5d97SBram Moolenaar 		else if (*p == '?')	// Rare word.
42259ccfebddSBram Moolenaar 		    flags |= WF_RARE;
42260d6f5d97SBram Moolenaar 		else if (VIM_ISDIGIT(*p)) // region number(s)
42279ccfebddSBram Moolenaar 		{
42280d6f5d97SBram Moolenaar 		    if ((flags & WF_REGION) == 0)   // first one
42299ccfebddSBram Moolenaar 			regionmask = 0;
42309ccfebddSBram Moolenaar 		    flags |= WF_REGION;
42319ccfebddSBram Moolenaar 
42329ccfebddSBram Moolenaar 		    l = *p - '0';
4233ee03b941SBram Moolenaar 		    if (l == 0 || l > spin->si_region_count)
42349ccfebddSBram Moolenaar 		    {
4235db99f9f2SBram Moolenaar 			smsg(_("Invalid region nr in %s line %ld: %s"),
42369ccfebddSBram Moolenaar 							  fname, lnum, p);
42379ccfebddSBram Moolenaar 			break;
42389ccfebddSBram Moolenaar 		    }
42399ccfebddSBram Moolenaar 		    regionmask |= 1 << (l - 1);
42409ccfebddSBram Moolenaar 		}
42419ccfebddSBram Moolenaar 		else
42429ccfebddSBram Moolenaar 		{
4243db99f9f2SBram Moolenaar 		    smsg(_("Unrecognized flags in %s line %ld: %s"),
42449ccfebddSBram Moolenaar 							      fname, lnum, p);
42459ccfebddSBram Moolenaar 		    break;
42469ccfebddSBram Moolenaar 		}
42479ccfebddSBram Moolenaar 		++p;
42489ccfebddSBram Moolenaar 	    }
42499ccfebddSBram Moolenaar 	}
42509ccfebddSBram Moolenaar 
42510d6f5d97SBram Moolenaar 	// Skip non-ASCII words when "spin->si_ascii" is TRUE.
42529ccfebddSBram Moolenaar 	if (spin->si_ascii && has_non_ascii(line))
42539ccfebddSBram Moolenaar 	{
42549ccfebddSBram Moolenaar 	    ++non_ascii;
42559ccfebddSBram Moolenaar 	    continue;
42569ccfebddSBram Moolenaar 	}
42579ccfebddSBram Moolenaar 
42580d6f5d97SBram Moolenaar 	// Normal word: store it.
42599ccfebddSBram Moolenaar 	if (store_word(spin, line, flags, regionmask, NULL, FALSE) == FAIL)
42609ccfebddSBram Moolenaar 	{
42619ccfebddSBram Moolenaar 	    retval = FAIL;
42629ccfebddSBram Moolenaar 	    break;
42639ccfebddSBram Moolenaar 	}
42649ccfebddSBram Moolenaar 	did_word = TRUE;
42659ccfebddSBram Moolenaar     }
42669ccfebddSBram Moolenaar 
42679ccfebddSBram Moolenaar     vim_free(pc);
42689ccfebddSBram Moolenaar     fclose(fd);
42699ccfebddSBram Moolenaar 
42709ccfebddSBram Moolenaar     if (spin->si_ascii && non_ascii > 0)
42719ccfebddSBram Moolenaar     {
42729ccfebddSBram Moolenaar 	vim_snprintf((char *)IObuff, IOSIZE,
42739ccfebddSBram Moolenaar 		  _("Ignored %d words with non-ASCII characters"), non_ascii);
42749ccfebddSBram Moolenaar 	spell_message(spin, IObuff);
42759ccfebddSBram Moolenaar     }
42769ccfebddSBram Moolenaar 
42779ccfebddSBram Moolenaar     return retval;
42789ccfebddSBram Moolenaar }
42799ccfebddSBram Moolenaar 
42809ccfebddSBram Moolenaar /*
42819ccfebddSBram Moolenaar  * Get part of an sblock_T, "len" bytes long.
42829ccfebddSBram Moolenaar  * This avoids calling free() for every little struct we use (and keeping
42839ccfebddSBram Moolenaar  * track of them).
42849ccfebddSBram Moolenaar  * The memory is cleared to all zeros.
42859ccfebddSBram Moolenaar  * Returns NULL when out of memory.
42869ccfebddSBram Moolenaar  */
42879ccfebddSBram Moolenaar     static void *
getroom(spellinfo_T * spin,size_t len,int align)42889ccfebddSBram Moolenaar getroom(
42899ccfebddSBram Moolenaar     spellinfo_T *spin,
42900d6f5d97SBram Moolenaar     size_t	len,		// length needed
42910d6f5d97SBram Moolenaar     int		align)		// align for pointer
42929ccfebddSBram Moolenaar {
42939ccfebddSBram Moolenaar     char_u	*p;
42949ccfebddSBram Moolenaar     sblock_T	*bl = spin->si_blocks;
42959ccfebddSBram Moolenaar 
42969ccfebddSBram Moolenaar     if (align && bl != NULL)
42970d6f5d97SBram Moolenaar 	// Round size up for alignment.  On some systems structures need to be
42980d6f5d97SBram Moolenaar 	// aligned to the size of a pointer (e.g., SPARC).
42999ccfebddSBram Moolenaar 	bl->sb_used = (bl->sb_used + sizeof(char *) - 1)
43009ccfebddSBram Moolenaar 						      & ~(sizeof(char *) - 1);
43019ccfebddSBram Moolenaar 
43029ccfebddSBram Moolenaar     if (bl == NULL || bl->sb_used + len > SBLOCKSIZE)
43039ccfebddSBram Moolenaar     {
43049ccfebddSBram Moolenaar 	if (len >= SBLOCKSIZE)
43059ccfebddSBram Moolenaar 	    bl = NULL;
43069ccfebddSBram Moolenaar 	else
43070d6f5d97SBram Moolenaar 	    // Allocate a block of memory. It is not freed until much later.
4308c799fe20SBram Moolenaar 	    bl = alloc_clear(sizeof(sblock_T) + SBLOCKSIZE);
43099ccfebddSBram Moolenaar 	if (bl == NULL)
43109ccfebddSBram Moolenaar 	{
43119ccfebddSBram Moolenaar 	    if (!spin->si_did_emsg)
43129ccfebddSBram Moolenaar 	    {
4313f9e3e09fSBram Moolenaar 		emsg(_("E845: Insufficient memory, word list will be incomplete"));
43149ccfebddSBram Moolenaar 		spin->si_did_emsg = TRUE;
43159ccfebddSBram Moolenaar 	    }
43169ccfebddSBram Moolenaar 	    return NULL;
43179ccfebddSBram Moolenaar 	}
43189ccfebddSBram Moolenaar 	bl->sb_next = spin->si_blocks;
43199ccfebddSBram Moolenaar 	spin->si_blocks = bl;
43209ccfebddSBram Moolenaar 	bl->sb_used = 0;
43219ccfebddSBram Moolenaar 	++spin->si_blocks_cnt;
43229ccfebddSBram Moolenaar     }
43239ccfebddSBram Moolenaar 
43249ccfebddSBram Moolenaar     p = bl->sb_data + bl->sb_used;
43259ccfebddSBram Moolenaar     bl->sb_used += (int)len;
43269ccfebddSBram Moolenaar 
43279ccfebddSBram Moolenaar     return p;
43289ccfebddSBram Moolenaar }
43299ccfebddSBram Moolenaar 
43309ccfebddSBram Moolenaar /*
43319ccfebddSBram Moolenaar  * Make a copy of a string into memory allocated with getroom().
43329ccfebddSBram Moolenaar  * Returns NULL when out of memory.
43339ccfebddSBram Moolenaar  */
43349ccfebddSBram Moolenaar     static char_u *
getroom_save(spellinfo_T * spin,char_u * s)43359ccfebddSBram Moolenaar getroom_save(spellinfo_T *spin, char_u *s)
43369ccfebddSBram Moolenaar {
43379ccfebddSBram Moolenaar     char_u	*sc;
43389ccfebddSBram Moolenaar 
43399ccfebddSBram Moolenaar     sc = (char_u *)getroom(spin, STRLEN(s) + 1, FALSE);
43409ccfebddSBram Moolenaar     if (sc != NULL)
43419ccfebddSBram Moolenaar 	STRCPY(sc, s);
43429ccfebddSBram Moolenaar     return sc;
43439ccfebddSBram Moolenaar }
43449ccfebddSBram Moolenaar 
43459ccfebddSBram Moolenaar 
43469ccfebddSBram Moolenaar /*
43479ccfebddSBram Moolenaar  * Free the list of allocated sblock_T.
43489ccfebddSBram Moolenaar  */
43499ccfebddSBram Moolenaar     static void
free_blocks(sblock_T * bl)43509ccfebddSBram Moolenaar free_blocks(sblock_T *bl)
43519ccfebddSBram Moolenaar {
43529ccfebddSBram Moolenaar     sblock_T	*next;
43539ccfebddSBram Moolenaar 
43549ccfebddSBram Moolenaar     while (bl != NULL)
43559ccfebddSBram Moolenaar     {
43569ccfebddSBram Moolenaar 	next = bl->sb_next;
43579ccfebddSBram Moolenaar 	vim_free(bl);
43589ccfebddSBram Moolenaar 	bl = next;
43599ccfebddSBram Moolenaar     }
43609ccfebddSBram Moolenaar }
43619ccfebddSBram Moolenaar 
43629ccfebddSBram Moolenaar /*
43639ccfebddSBram Moolenaar  * Allocate the root of a word tree.
43649ccfebddSBram Moolenaar  * Returns NULL when out of memory.
43659ccfebddSBram Moolenaar  */
43669ccfebddSBram Moolenaar     static wordnode_T *
wordtree_alloc(spellinfo_T * spin)43679ccfebddSBram Moolenaar wordtree_alloc(spellinfo_T *spin)
43689ccfebddSBram Moolenaar {
43699ccfebddSBram Moolenaar     return (wordnode_T *)getroom(spin, sizeof(wordnode_T), TRUE);
43709ccfebddSBram Moolenaar }
43719ccfebddSBram Moolenaar 
43729ccfebddSBram Moolenaar /*
43739ccfebddSBram Moolenaar  * Store a word in the tree(s).
43749ccfebddSBram Moolenaar  * Always store it in the case-folded tree.  For a keep-case word this is
43759ccfebddSBram Moolenaar  * useful when the word can also be used with all caps (no WF_FIXCAP flag) and
43769ccfebddSBram Moolenaar  * used to find suggestions.
43779ccfebddSBram Moolenaar  * For a keep-case word also store it in the keep-case tree.
43789ccfebddSBram Moolenaar  * When "pfxlist" is not NULL store the word for each postponed prefix ID and
43799ccfebddSBram Moolenaar  * compound flag.
43809ccfebddSBram Moolenaar  */
43819ccfebddSBram Moolenaar     static int
store_word(spellinfo_T * spin,char_u * word,int flags,int region,char_u * pfxlist,int need_affix)43829ccfebddSBram Moolenaar store_word(
43839ccfebddSBram Moolenaar     spellinfo_T	*spin,
43849ccfebddSBram Moolenaar     char_u	*word,
43850d6f5d97SBram Moolenaar     int		flags,		// extra flags, WF_BANNED
43860d6f5d97SBram Moolenaar     int		region,		// supported region(s)
43870d6f5d97SBram Moolenaar     char_u	*pfxlist,	// list of prefix IDs or NULL
43880d6f5d97SBram Moolenaar     int		need_affix)	// only store word with affix ID
43899ccfebddSBram Moolenaar {
43909ccfebddSBram Moolenaar     int		len = (int)STRLEN(word);
43919ccfebddSBram Moolenaar     int		ct = captype(word, word + len);
43929ccfebddSBram Moolenaar     char_u	foldword[MAXWLEN];
43939ccfebddSBram Moolenaar     int		res = OK;
43949ccfebddSBram Moolenaar     char_u	*p;
43959ccfebddSBram Moolenaar 
43964f135275SBram Moolenaar     (void)spell_casefold(curwin, word, len, foldword, MAXWLEN);
43979ccfebddSBram Moolenaar     for (p = pfxlist; res == OK; ++p)
43989ccfebddSBram Moolenaar     {
43999ccfebddSBram Moolenaar 	if (!need_affix || (p != NULL && *p != NUL))
44009ccfebddSBram Moolenaar 	    res = tree_add_word(spin, foldword, spin->si_foldroot, ct | flags,
44019ccfebddSBram Moolenaar 						  region, p == NULL ? 0 : *p);
44029ccfebddSBram Moolenaar 	if (p == NULL || *p == NUL)
44039ccfebddSBram Moolenaar 	    break;
44049ccfebddSBram Moolenaar     }
44059ccfebddSBram Moolenaar     ++spin->si_foldwcount;
44069ccfebddSBram Moolenaar 
44079ccfebddSBram Moolenaar     if (res == OK && (ct == WF_KEEPCAP || (flags & WF_KEEPCAP)))
44089ccfebddSBram Moolenaar     {
44099ccfebddSBram Moolenaar 	for (p = pfxlist; res == OK; ++p)
44109ccfebddSBram Moolenaar 	{
44119ccfebddSBram Moolenaar 	    if (!need_affix || (p != NULL && *p != NUL))
44129ccfebddSBram Moolenaar 		res = tree_add_word(spin, word, spin->si_keeproot, flags,
44139ccfebddSBram Moolenaar 						  region, p == NULL ? 0 : *p);
44149ccfebddSBram Moolenaar 	    if (p == NULL || *p == NUL)
44159ccfebddSBram Moolenaar 		break;
44169ccfebddSBram Moolenaar 	}
44179ccfebddSBram Moolenaar 	++spin->si_keepwcount;
44189ccfebddSBram Moolenaar     }
44199ccfebddSBram Moolenaar     return res;
44209ccfebddSBram Moolenaar }
44219ccfebddSBram Moolenaar 
44229ccfebddSBram Moolenaar /*
44239ccfebddSBram Moolenaar  * Add word "word" to a word tree at "root".
44249ccfebddSBram Moolenaar  * When "flags" < 0 we are adding to the prefix tree where "flags" is used for
44259ccfebddSBram Moolenaar  * "rare" and "region" is the condition nr.
44269ccfebddSBram Moolenaar  * Returns FAIL when out of memory.
44279ccfebddSBram Moolenaar  */
44289ccfebddSBram Moolenaar     static int
tree_add_word(spellinfo_T * spin,char_u * word,wordnode_T * root,int flags,int region,int affixID)44299ccfebddSBram Moolenaar tree_add_word(
44309ccfebddSBram Moolenaar     spellinfo_T	*spin,
44319ccfebddSBram Moolenaar     char_u	*word,
44329ccfebddSBram Moolenaar     wordnode_T	*root,
44339ccfebddSBram Moolenaar     int		flags,
44349ccfebddSBram Moolenaar     int		region,
44359ccfebddSBram Moolenaar     int		affixID)
44369ccfebddSBram Moolenaar {
44379ccfebddSBram Moolenaar     wordnode_T	*node = root;
44389ccfebddSBram Moolenaar     wordnode_T	*np;
44399ccfebddSBram Moolenaar     wordnode_T	*copyp, **copyprev;
44409ccfebddSBram Moolenaar     wordnode_T	**prev = NULL;
44419ccfebddSBram Moolenaar     int		i;
44429ccfebddSBram Moolenaar 
44430d6f5d97SBram Moolenaar     // Add each byte of the word to the tree, including the NUL at the end.
44449ccfebddSBram Moolenaar     for (i = 0; ; ++i)
44459ccfebddSBram Moolenaar     {
44460d6f5d97SBram Moolenaar 	// When there is more than one reference to this node we need to make
44470d6f5d97SBram Moolenaar 	// a copy, so that we can modify it.  Copy the whole list of siblings
44480d6f5d97SBram Moolenaar 	// (we don't optimize for a partly shared list of siblings).
44499ccfebddSBram Moolenaar 	if (node != NULL && node->wn_refs > 1)
44509ccfebddSBram Moolenaar 	{
44519ccfebddSBram Moolenaar 	    --node->wn_refs;
44529ccfebddSBram Moolenaar 	    copyprev = prev;
4453aeea7215SBram Moolenaar 	    FOR_ALL_NODE_SIBLINGS(node, copyp)
44549ccfebddSBram Moolenaar 	    {
44550d6f5d97SBram Moolenaar 		// Allocate a new node and copy the info.
44569ccfebddSBram Moolenaar 		np = get_wordnode(spin);
44579ccfebddSBram Moolenaar 		if (np == NULL)
44589ccfebddSBram Moolenaar 		    return FAIL;
44599ccfebddSBram Moolenaar 		np->wn_child = copyp->wn_child;
44609ccfebddSBram Moolenaar 		if (np->wn_child != NULL)
44610d6f5d97SBram Moolenaar 		    ++np->wn_child->wn_refs;	// child gets extra ref
44629ccfebddSBram Moolenaar 		np->wn_byte = copyp->wn_byte;
44639ccfebddSBram Moolenaar 		if (np->wn_byte == NUL)
44649ccfebddSBram Moolenaar 		{
44659ccfebddSBram Moolenaar 		    np->wn_flags = copyp->wn_flags;
44669ccfebddSBram Moolenaar 		    np->wn_region = copyp->wn_region;
44679ccfebddSBram Moolenaar 		    np->wn_affixID = copyp->wn_affixID;
44689ccfebddSBram Moolenaar 		}
44699ccfebddSBram Moolenaar 
44700d6f5d97SBram Moolenaar 		// Link the new node in the list, there will be one ref.
44719ccfebddSBram Moolenaar 		np->wn_refs = 1;
44729ccfebddSBram Moolenaar 		if (copyprev != NULL)
44739ccfebddSBram Moolenaar 		    *copyprev = np;
44749ccfebddSBram Moolenaar 		copyprev = &np->wn_sibling;
44759ccfebddSBram Moolenaar 
44760d6f5d97SBram Moolenaar 		// Let "node" point to the head of the copied list.
44779ccfebddSBram Moolenaar 		if (copyp == node)
44789ccfebddSBram Moolenaar 		    node = np;
44799ccfebddSBram Moolenaar 	    }
44809ccfebddSBram Moolenaar 	}
44819ccfebddSBram Moolenaar 
44820d6f5d97SBram Moolenaar 	// Look for the sibling that has the same character.  They are sorted
44830d6f5d97SBram Moolenaar 	// on byte value, thus stop searching when a sibling is found with a
44840d6f5d97SBram Moolenaar 	// higher byte value.  For zero bytes (end of word) the sorting is
44850d6f5d97SBram Moolenaar 	// done on flags and then on affixID.
44869ccfebddSBram Moolenaar 	while (node != NULL
44879ccfebddSBram Moolenaar 		&& (node->wn_byte < word[i]
44889ccfebddSBram Moolenaar 		    || (node->wn_byte == NUL
44899ccfebddSBram Moolenaar 			&& (flags < 0
44909ccfebddSBram Moolenaar 			    ? node->wn_affixID < (unsigned)affixID
44919ccfebddSBram Moolenaar 			    : (node->wn_flags < (unsigned)(flags & WN_MASK)
44929ccfebddSBram Moolenaar 				|| (node->wn_flags == (flags & WN_MASK)
44939ccfebddSBram Moolenaar 				    && (spin->si_sugtree
44949ccfebddSBram Moolenaar 					? (node->wn_region & 0xffff) < region
44959ccfebddSBram Moolenaar 					: node->wn_affixID
44969ccfebddSBram Moolenaar 						    < (unsigned)affixID)))))))
44979ccfebddSBram Moolenaar 	{
44989ccfebddSBram Moolenaar 	    prev = &node->wn_sibling;
44999ccfebddSBram Moolenaar 	    node = *prev;
45009ccfebddSBram Moolenaar 	}
45019ccfebddSBram Moolenaar 	if (node == NULL
45029ccfebddSBram Moolenaar 		|| node->wn_byte != word[i]
45039ccfebddSBram Moolenaar 		|| (word[i] == NUL
45049ccfebddSBram Moolenaar 		    && (flags < 0
45059ccfebddSBram Moolenaar 			|| spin->si_sugtree
45069ccfebddSBram Moolenaar 			|| node->wn_flags != (flags & WN_MASK)
45079ccfebddSBram Moolenaar 			|| node->wn_affixID != affixID)))
45089ccfebddSBram Moolenaar 	{
45090d6f5d97SBram Moolenaar 	    // Allocate a new node.
45109ccfebddSBram Moolenaar 	    np = get_wordnode(spin);
45119ccfebddSBram Moolenaar 	    if (np == NULL)
45129ccfebddSBram Moolenaar 		return FAIL;
45139ccfebddSBram Moolenaar 	    np->wn_byte = word[i];
45149ccfebddSBram Moolenaar 
45150d6f5d97SBram Moolenaar 	    // If "node" is NULL this is a new child or the end of the sibling
45160d6f5d97SBram Moolenaar 	    // list: ref count is one.  Otherwise use ref count of sibling and
45170d6f5d97SBram Moolenaar 	    // make ref count of sibling one (matters when inserting in front
45180d6f5d97SBram Moolenaar 	    // of the list of siblings).
45199ccfebddSBram Moolenaar 	    if (node == NULL)
45209ccfebddSBram Moolenaar 		np->wn_refs = 1;
45219ccfebddSBram Moolenaar 	    else
45229ccfebddSBram Moolenaar 	    {
45239ccfebddSBram Moolenaar 		np->wn_refs = node->wn_refs;
45249ccfebddSBram Moolenaar 		node->wn_refs = 1;
45259ccfebddSBram Moolenaar 	    }
45269ccfebddSBram Moolenaar 	    if (prev != NULL)
45279ccfebddSBram Moolenaar 		*prev = np;
45289ccfebddSBram Moolenaar 	    np->wn_sibling = node;
45299ccfebddSBram Moolenaar 	    node = np;
45309ccfebddSBram Moolenaar 	}
45319ccfebddSBram Moolenaar 
45329ccfebddSBram Moolenaar 	if (word[i] == NUL)
45339ccfebddSBram Moolenaar 	{
45349ccfebddSBram Moolenaar 	    node->wn_flags = flags;
45359ccfebddSBram Moolenaar 	    node->wn_region |= region;
45369ccfebddSBram Moolenaar 	    node->wn_affixID = affixID;
45379ccfebddSBram Moolenaar 	    break;
45389ccfebddSBram Moolenaar 	}
45399ccfebddSBram Moolenaar 	prev = &node->wn_child;
45409ccfebddSBram Moolenaar 	node = *prev;
45419ccfebddSBram Moolenaar     }
45429ccfebddSBram Moolenaar #ifdef SPELL_PRINTTREE
4543f9e3e09fSBram Moolenaar     smsg("Added \"%s\"", word);
45449ccfebddSBram Moolenaar     spell_print_tree(root->wn_sibling);
45459ccfebddSBram Moolenaar #endif
45469ccfebddSBram Moolenaar 
45470d6f5d97SBram Moolenaar     // count nr of words added since last message
45489ccfebddSBram Moolenaar     ++spin->si_msg_count;
45499ccfebddSBram Moolenaar 
45509ccfebddSBram Moolenaar     if (spin->si_compress_cnt > 1)
45519ccfebddSBram Moolenaar     {
45529ccfebddSBram Moolenaar 	if (--spin->si_compress_cnt == 1)
45530d6f5d97SBram Moolenaar 	    // Did enough words to lower the block count limit.
4554408c23b0SBram Moolenaar 	    spin->si_blocks_cnt += compress_inc;
45559ccfebddSBram Moolenaar     }
45569ccfebddSBram Moolenaar 
45579ccfebddSBram Moolenaar     /*
45589ccfebddSBram Moolenaar      * When we have allocated lots of memory we need to compress the word tree
45599ccfebddSBram Moolenaar      * to free up some room.  But compression is slow, and we might actually
45609ccfebddSBram Moolenaar      * need that room, thus only compress in the following situations:
45619ccfebddSBram Moolenaar      * 1. When not compressed before (si_compress_cnt == 0): when using
45629ccfebddSBram Moolenaar      *    "compress_start" blocks.
4563408c23b0SBram Moolenaar      * 2. When compressed before and used "compress_inc" blocks before
4564408c23b0SBram Moolenaar      *    adding "compress_added" words (si_compress_cnt > 1).
4565408c23b0SBram Moolenaar      * 3. When compressed before, added "compress_added" words
45669ccfebddSBram Moolenaar      *    (si_compress_cnt == 1) and the number of free nodes drops below the
45679ccfebddSBram Moolenaar      *    maximum word length.
45689ccfebddSBram Moolenaar      */
45699ccfebddSBram Moolenaar #ifndef SPELL_COMPRESS_ALLWAYS
45709ccfebddSBram Moolenaar     if (spin->si_compress_cnt == 1
45719ccfebddSBram Moolenaar 	    ? spin->si_free_count < MAXWLEN
45729ccfebddSBram Moolenaar 	    : spin->si_blocks_cnt >= compress_start)
45739ccfebddSBram Moolenaar #endif
45749ccfebddSBram Moolenaar     {
45750d6f5d97SBram Moolenaar 	// Decrement the block counter.  The effect is that we compress again
4576408c23b0SBram Moolenaar 	// when the freed up room has been used and another "compress_inc"
4577408c23b0SBram Moolenaar 	// blocks have been allocated.  Unless "compress_added" words have
45780d6f5d97SBram Moolenaar 	// been added, then the limit is put back again.
4579408c23b0SBram Moolenaar 	spin->si_blocks_cnt -= compress_inc;
4580408c23b0SBram Moolenaar 	spin->si_compress_cnt = compress_added;
45819ccfebddSBram Moolenaar 
45829ccfebddSBram Moolenaar 	if (spin->si_verbose)
45839ccfebddSBram Moolenaar 	{
45849ccfebddSBram Moolenaar 	    msg_start();
458532526b3cSBram Moolenaar 	    msg_puts(_(msg_compressing));
45869ccfebddSBram Moolenaar 	    msg_clr_eos();
45879ccfebddSBram Moolenaar 	    msg_didout = FALSE;
45889ccfebddSBram Moolenaar 	    msg_col = 0;
45899ccfebddSBram Moolenaar 	    out_flush();
45909ccfebddSBram Moolenaar 	}
45919ccfebddSBram Moolenaar 
45920d6f5d97SBram Moolenaar 	// Compress both trees.  Either they both have many nodes, which makes
45930d6f5d97SBram Moolenaar 	// compression useful, or one of them is small, which means
45940d6f5d97SBram Moolenaar 	// compression goes fast.  But when filling the soundfold word tree
45950d6f5d97SBram Moolenaar 	// there is no keep-case tree.
4596408c23b0SBram Moolenaar 	wordtree_compress(spin, spin->si_foldroot, "case-folded");
45979ccfebddSBram Moolenaar 	if (affixID >= 0)
4598408c23b0SBram Moolenaar 	    wordtree_compress(spin, spin->si_keeproot, "keep-case");
45999ccfebddSBram Moolenaar     }
46009ccfebddSBram Moolenaar 
46019ccfebddSBram Moolenaar     return OK;
46029ccfebddSBram Moolenaar }
46039ccfebddSBram Moolenaar 
46049ccfebddSBram Moolenaar /*
46059ccfebddSBram Moolenaar  * Get a wordnode_T, either from the list of previously freed nodes or
46069ccfebddSBram Moolenaar  * allocate a new one.
46079ccfebddSBram Moolenaar  * Returns NULL when out of memory.
46089ccfebddSBram Moolenaar  */
46099ccfebddSBram Moolenaar     static wordnode_T *
get_wordnode(spellinfo_T * spin)46109ccfebddSBram Moolenaar get_wordnode(spellinfo_T *spin)
46119ccfebddSBram Moolenaar {
46129ccfebddSBram Moolenaar     wordnode_T *n;
46139ccfebddSBram Moolenaar 
46149ccfebddSBram Moolenaar     if (spin->si_first_free == NULL)
46159ccfebddSBram Moolenaar 	n = (wordnode_T *)getroom(spin, sizeof(wordnode_T), TRUE);
46169ccfebddSBram Moolenaar     else
46179ccfebddSBram Moolenaar     {
46189ccfebddSBram Moolenaar 	n = spin->si_first_free;
46199ccfebddSBram Moolenaar 	spin->si_first_free = n->wn_child;
4620a80faa89SBram Moolenaar 	CLEAR_POINTER(n);
46219ccfebddSBram Moolenaar 	--spin->si_free_count;
46229ccfebddSBram Moolenaar     }
46239ccfebddSBram Moolenaar #ifdef SPELL_PRINTTREE
46249ccfebddSBram Moolenaar     if (n != NULL)
46259ccfebddSBram Moolenaar 	n->wn_nr = ++spin->si_wordnode_nr;
46269ccfebddSBram Moolenaar #endif
46279ccfebddSBram Moolenaar     return n;
46289ccfebddSBram Moolenaar }
46299ccfebddSBram Moolenaar 
46309ccfebddSBram Moolenaar /*
46319ccfebddSBram Moolenaar  * Decrement the reference count on a node (which is the head of a list of
46329ccfebddSBram Moolenaar  * siblings).  If the reference count becomes zero free the node and its
46339ccfebddSBram Moolenaar  * siblings.
46349ccfebddSBram Moolenaar  * Returns the number of nodes actually freed.
46359ccfebddSBram Moolenaar  */
46369ccfebddSBram Moolenaar     static int
deref_wordnode(spellinfo_T * spin,wordnode_T * node)46379ccfebddSBram Moolenaar deref_wordnode(spellinfo_T *spin, wordnode_T *node)
46389ccfebddSBram Moolenaar {
46399ccfebddSBram Moolenaar     wordnode_T	*np;
46409ccfebddSBram Moolenaar     int		cnt = 0;
46419ccfebddSBram Moolenaar 
46429ccfebddSBram Moolenaar     if (--node->wn_refs == 0)
46439ccfebddSBram Moolenaar     {
4644aeea7215SBram Moolenaar 	FOR_ALL_NODE_SIBLINGS(node, np)
46459ccfebddSBram Moolenaar 	{
46469ccfebddSBram Moolenaar 	    if (np->wn_child != NULL)
46479ccfebddSBram Moolenaar 		cnt += deref_wordnode(spin, np->wn_child);
46489ccfebddSBram Moolenaar 	    free_wordnode(spin, np);
46499ccfebddSBram Moolenaar 	    ++cnt;
46509ccfebddSBram Moolenaar 	}
46510d6f5d97SBram Moolenaar 	++cnt;	    // length field
46529ccfebddSBram Moolenaar     }
46539ccfebddSBram Moolenaar     return cnt;
46549ccfebddSBram Moolenaar }
46559ccfebddSBram Moolenaar 
46569ccfebddSBram Moolenaar /*
46579ccfebddSBram Moolenaar  * Free a wordnode_T for re-use later.
46589ccfebddSBram Moolenaar  * Only the "wn_child" field becomes invalid.
46599ccfebddSBram Moolenaar  */
46609ccfebddSBram Moolenaar     static void
free_wordnode(spellinfo_T * spin,wordnode_T * n)46619ccfebddSBram Moolenaar free_wordnode(spellinfo_T *spin, wordnode_T *n)
46629ccfebddSBram Moolenaar {
46639ccfebddSBram Moolenaar     n->wn_child = spin->si_first_free;
46649ccfebddSBram Moolenaar     spin->si_first_free = n;
46659ccfebddSBram Moolenaar     ++spin->si_free_count;
46669ccfebddSBram Moolenaar }
46679ccfebddSBram Moolenaar 
46689ccfebddSBram Moolenaar /*
46699ccfebddSBram Moolenaar  * Compress a tree: find tails that are identical and can be shared.
46709ccfebddSBram Moolenaar  */
46719ccfebddSBram Moolenaar     static void
wordtree_compress(spellinfo_T * spin,wordnode_T * root,char * name)4672408c23b0SBram Moolenaar wordtree_compress(spellinfo_T *spin, wordnode_T *root, char *name)
46739ccfebddSBram Moolenaar {
46749ccfebddSBram Moolenaar     hashtab_T	    ht;
467559f88fbfSBram Moolenaar     long	    n;
467659f88fbfSBram Moolenaar     long	    tot = 0;
467759f88fbfSBram Moolenaar     long	    perc;
46789ccfebddSBram Moolenaar 
46790d6f5d97SBram Moolenaar     // Skip the root itself, it's not actually used.  The first sibling is the
46800d6f5d97SBram Moolenaar     // start of the tree.
46819ccfebddSBram Moolenaar     if (root->wn_sibling != NULL)
46829ccfebddSBram Moolenaar     {
46839ccfebddSBram Moolenaar 	hash_init(&ht);
46849ccfebddSBram Moolenaar 	n = node_compress(spin, root->wn_sibling, &ht, &tot);
46859ccfebddSBram Moolenaar 
46869ccfebddSBram Moolenaar #ifndef SPELL_PRINTTREE
46879ccfebddSBram Moolenaar 	if (spin->si_verbose || p_verbose > 2)
46889ccfebddSBram Moolenaar #endif
46899ccfebddSBram Moolenaar 	{
46909ccfebddSBram Moolenaar 	    if (tot > 1000000)
46919ccfebddSBram Moolenaar 		perc = (tot - n) / (tot / 100);
46929ccfebddSBram Moolenaar 	    else if (tot == 0)
46939ccfebddSBram Moolenaar 		perc = 0;
46949ccfebddSBram Moolenaar 	    else
46959ccfebddSBram Moolenaar 		perc = (tot - n) * 100 / tot;
46969ccfebddSBram Moolenaar 	    vim_snprintf((char *)IObuff, IOSIZE,
4697408c23b0SBram Moolenaar 		       _("Compressed %s: %ld of %ld nodes; %ld (%ld%%) remaining"),
4698408c23b0SBram Moolenaar 						       name, n, tot, tot - n, perc);
46999ccfebddSBram Moolenaar 	    spell_message(spin, IObuff);
47009ccfebddSBram Moolenaar 	}
47019ccfebddSBram Moolenaar #ifdef SPELL_PRINTTREE
47029ccfebddSBram Moolenaar 	spell_print_tree(root->wn_sibling);
47039ccfebddSBram Moolenaar #endif
47049ccfebddSBram Moolenaar 	hash_clear(&ht);
47059ccfebddSBram Moolenaar     }
47069ccfebddSBram Moolenaar }
47079ccfebddSBram Moolenaar 
47089ccfebddSBram Moolenaar /*
47099ccfebddSBram Moolenaar  * Compress a node, its siblings and its children, depth first.
47109ccfebddSBram Moolenaar  * Returns the number of compressed nodes.
47119ccfebddSBram Moolenaar  */
471259f88fbfSBram Moolenaar     static long
node_compress(spellinfo_T * spin,wordnode_T * node,hashtab_T * ht,long * tot)47139ccfebddSBram Moolenaar node_compress(
47149ccfebddSBram Moolenaar     spellinfo_T	*spin,
47159ccfebddSBram Moolenaar     wordnode_T	*node,
47169ccfebddSBram Moolenaar     hashtab_T	*ht,
471759f88fbfSBram Moolenaar     long	*tot)	    // total count of nodes before compressing,
47180d6f5d97SBram Moolenaar 			    // incremented while going through the tree
47199ccfebddSBram Moolenaar {
47209ccfebddSBram Moolenaar     wordnode_T	*np;
47219ccfebddSBram Moolenaar     wordnode_T	*tp;
47229ccfebddSBram Moolenaar     wordnode_T	*child;
47239ccfebddSBram Moolenaar     hash_T	hash;
47249ccfebddSBram Moolenaar     hashitem_T	*hi;
472559f88fbfSBram Moolenaar     long	len = 0;
47269ccfebddSBram Moolenaar     unsigned	nr, n;
472759f88fbfSBram Moolenaar     long	compressed = 0;
47289ccfebddSBram Moolenaar 
47299ccfebddSBram Moolenaar     /*
47309ccfebddSBram Moolenaar      * Go through the list of siblings.  Compress each child and then try
47319ccfebddSBram Moolenaar      * finding an identical child to replace it.
47329ccfebddSBram Moolenaar      * Note that with "child" we mean not just the node that is pointed to,
47339ccfebddSBram Moolenaar      * but the whole list of siblings of which the child node is the first.
47349ccfebddSBram Moolenaar      */
47359ccfebddSBram Moolenaar     for (np = node; np != NULL && !got_int; np = np->wn_sibling)
47369ccfebddSBram Moolenaar     {
47379ccfebddSBram Moolenaar 	++len;
47389ccfebddSBram Moolenaar 	if ((child = np->wn_child) != NULL)
47399ccfebddSBram Moolenaar 	{
47400d6f5d97SBram Moolenaar 	    // Compress the child first.  This fills hashkey.
47419ccfebddSBram Moolenaar 	    compressed += node_compress(spin, child, ht, tot);
47429ccfebddSBram Moolenaar 
47430d6f5d97SBram Moolenaar 	    // Try to find an identical child.
47449ccfebddSBram Moolenaar 	    hash = hash_hash(child->wn_u1.hashkey);
47459ccfebddSBram Moolenaar 	    hi = hash_lookup(ht, child->wn_u1.hashkey, hash);
47469ccfebddSBram Moolenaar 	    if (!HASHITEM_EMPTY(hi))
47479ccfebddSBram Moolenaar 	    {
47480d6f5d97SBram Moolenaar 		// There are children we encountered before with a hash value
47490d6f5d97SBram Moolenaar 		// identical to the current child.  Now check if there is one
47500d6f5d97SBram Moolenaar 		// that is really identical.
47519ccfebddSBram Moolenaar 		for (tp = HI2WN(hi); tp != NULL; tp = tp->wn_u2.next)
47529ccfebddSBram Moolenaar 		    if (node_equal(child, tp))
47539ccfebddSBram Moolenaar 		    {
47540d6f5d97SBram Moolenaar 			// Found one!  Now use that child in place of the
47550d6f5d97SBram Moolenaar 			// current one.  This means the current child and all
47560d6f5d97SBram Moolenaar 			// its siblings is unlinked from the tree.
47579ccfebddSBram Moolenaar 			++tp->wn_refs;
47589ccfebddSBram Moolenaar 			compressed += deref_wordnode(spin, child);
47599ccfebddSBram Moolenaar 			np->wn_child = tp;
47609ccfebddSBram Moolenaar 			break;
47619ccfebddSBram Moolenaar 		    }
47629ccfebddSBram Moolenaar 		if (tp == NULL)
47639ccfebddSBram Moolenaar 		{
47640d6f5d97SBram Moolenaar 		    // No other child with this hash value equals the child of
47650d6f5d97SBram Moolenaar 		    // the node, add it to the linked list after the first
47660d6f5d97SBram Moolenaar 		    // item.
47679ccfebddSBram Moolenaar 		    tp = HI2WN(hi);
47689ccfebddSBram Moolenaar 		    child->wn_u2.next = tp->wn_u2.next;
47699ccfebddSBram Moolenaar 		    tp->wn_u2.next = child;
47709ccfebddSBram Moolenaar 		}
47719ccfebddSBram Moolenaar 	    }
47729ccfebddSBram Moolenaar 	    else
47730d6f5d97SBram Moolenaar 		// No other child has this hash value, add it to the
47740d6f5d97SBram Moolenaar 		// hashtable.
47759ccfebddSBram Moolenaar 		hash_add_item(ht, hi, child->wn_u1.hashkey, hash);
47769ccfebddSBram Moolenaar 	}
47779ccfebddSBram Moolenaar     }
47780d6f5d97SBram Moolenaar     *tot += len + 1;	// add one for the node that stores the length
47799ccfebddSBram Moolenaar 
47809ccfebddSBram Moolenaar     /*
47819ccfebddSBram Moolenaar      * Make a hash key for the node and its siblings, so that we can quickly
47829ccfebddSBram Moolenaar      * find a lookalike node.  This must be done after compressing the sibling
47839ccfebddSBram Moolenaar      * list, otherwise the hash key would become invalid by the compression.
47849ccfebddSBram Moolenaar      */
47859ccfebddSBram Moolenaar     node->wn_u1.hashkey[0] = len;
47869ccfebddSBram Moolenaar     nr = 0;
4787aeea7215SBram Moolenaar     FOR_ALL_NODE_SIBLINGS(node, np)
47889ccfebddSBram Moolenaar     {
47899ccfebddSBram Moolenaar 	if (np->wn_byte == NUL)
47900d6f5d97SBram Moolenaar 	    // end node: use wn_flags, wn_region and wn_affixID
47919ccfebddSBram Moolenaar 	    n = np->wn_flags + (np->wn_region << 8) + (np->wn_affixID << 16);
47929ccfebddSBram Moolenaar 	else
47930d6f5d97SBram Moolenaar 	    // byte node: use the byte value and the child pointer
47949ccfebddSBram Moolenaar 	    n = (unsigned)(np->wn_byte + ((long_u)np->wn_child << 8));
47959ccfebddSBram Moolenaar 	nr = nr * 101 + n;
47969ccfebddSBram Moolenaar     }
47979ccfebddSBram Moolenaar 
47980d6f5d97SBram Moolenaar     // Avoid NUL bytes, it terminates the hash key.
47999ccfebddSBram Moolenaar     n = nr & 0xff;
48009ccfebddSBram Moolenaar     node->wn_u1.hashkey[1] = n == 0 ? 1 : n;
48019ccfebddSBram Moolenaar     n = (nr >> 8) & 0xff;
48029ccfebddSBram Moolenaar     node->wn_u1.hashkey[2] = n == 0 ? 1 : n;
48039ccfebddSBram Moolenaar     n = (nr >> 16) & 0xff;
48049ccfebddSBram Moolenaar     node->wn_u1.hashkey[3] = n == 0 ? 1 : n;
48059ccfebddSBram Moolenaar     n = (nr >> 24) & 0xff;
48069ccfebddSBram Moolenaar     node->wn_u1.hashkey[4] = n == 0 ? 1 : n;
48079ccfebddSBram Moolenaar     node->wn_u1.hashkey[5] = NUL;
48089ccfebddSBram Moolenaar 
48090d6f5d97SBram Moolenaar     // Check for CTRL-C pressed now and then.
4810408c23b0SBram Moolenaar     veryfast_breakcheck();
48119ccfebddSBram Moolenaar 
48129ccfebddSBram Moolenaar     return compressed;
48139ccfebddSBram Moolenaar }
48149ccfebddSBram Moolenaar 
48159ccfebddSBram Moolenaar /*
48169ccfebddSBram Moolenaar  * Return TRUE when two nodes have identical siblings and children.
48179ccfebddSBram Moolenaar  */
48189ccfebddSBram Moolenaar     static int
node_equal(wordnode_T * n1,wordnode_T * n2)48199ccfebddSBram Moolenaar node_equal(wordnode_T *n1, wordnode_T *n2)
48209ccfebddSBram Moolenaar {
48219ccfebddSBram Moolenaar     wordnode_T	*p1;
48229ccfebddSBram Moolenaar     wordnode_T	*p2;
48239ccfebddSBram Moolenaar 
48249ccfebddSBram Moolenaar     for (p1 = n1, p2 = n2; p1 != NULL && p2 != NULL;
48259ccfebddSBram Moolenaar 				     p1 = p1->wn_sibling, p2 = p2->wn_sibling)
48269ccfebddSBram Moolenaar 	if (p1->wn_byte != p2->wn_byte
48279ccfebddSBram Moolenaar 		|| (p1->wn_byte == NUL
48289ccfebddSBram Moolenaar 		    ? (p1->wn_flags != p2->wn_flags
48299ccfebddSBram Moolenaar 			|| p1->wn_region != p2->wn_region
48309ccfebddSBram Moolenaar 			|| p1->wn_affixID != p2->wn_affixID)
48319ccfebddSBram Moolenaar 		    : (p1->wn_child != p2->wn_child)))
48329ccfebddSBram Moolenaar 	    break;
48339ccfebddSBram Moolenaar 
48349ccfebddSBram Moolenaar     return p1 == NULL && p2 == NULL;
48359ccfebddSBram Moolenaar }
48369ccfebddSBram Moolenaar 
4837eae1b91fSBram Moolenaar static int rep_compare(const void *s1, const void *s2);
48389ccfebddSBram Moolenaar 
48399ccfebddSBram Moolenaar /*
48409ccfebddSBram Moolenaar  * Function given to qsort() to sort the REP items on "from" string.
48419ccfebddSBram Moolenaar  */
48429ccfebddSBram Moolenaar     static int
rep_compare(const void * s1,const void * s2)48439ccfebddSBram Moolenaar rep_compare(const void *s1, const void *s2)
48449ccfebddSBram Moolenaar {
48459ccfebddSBram Moolenaar     fromto_T	*p1 = (fromto_T *)s1;
48469ccfebddSBram Moolenaar     fromto_T	*p2 = (fromto_T *)s2;
48479ccfebddSBram Moolenaar 
48489ccfebddSBram Moolenaar     return STRCMP(p1->ft_from, p2->ft_from);
48499ccfebddSBram Moolenaar }
48509ccfebddSBram Moolenaar 
48519ccfebddSBram Moolenaar /*
48529ccfebddSBram Moolenaar  * Write the Vim .spl file "fname".
48539ccfebddSBram Moolenaar  * Return FAIL or OK;
48549ccfebddSBram Moolenaar  */
48559ccfebddSBram Moolenaar     static int
write_vim_spell(spellinfo_T * spin,char_u * fname)48569ccfebddSBram Moolenaar write_vim_spell(spellinfo_T *spin, char_u *fname)
48579ccfebddSBram Moolenaar {
48589ccfebddSBram Moolenaar     FILE	*fd;
48599ccfebddSBram Moolenaar     int		regionmask;
48609ccfebddSBram Moolenaar     int		round;
48619ccfebddSBram Moolenaar     wordnode_T	*tree;
48629ccfebddSBram Moolenaar     int		nodecount;
48639ccfebddSBram Moolenaar     int		i;
48649ccfebddSBram Moolenaar     int		l;
48659ccfebddSBram Moolenaar     garray_T	*gap;
48669ccfebddSBram Moolenaar     fromto_T	*ftp;
48679ccfebddSBram Moolenaar     char_u	*p;
48689ccfebddSBram Moolenaar     int		rr;
48699ccfebddSBram Moolenaar     int		retval = OK;
48700d6f5d97SBram Moolenaar     size_t	fwv = 1;  // collect return value of fwrite() to avoid
48710d6f5d97SBram Moolenaar 			  // warnings from picky compiler
48729ccfebddSBram Moolenaar 
48739ccfebddSBram Moolenaar     fd = mch_fopen((char *)fname, "w");
48749ccfebddSBram Moolenaar     if (fd == NULL)
48759ccfebddSBram Moolenaar     {
4876f9e3e09fSBram Moolenaar 	semsg(_(e_notopen), fname);
48779ccfebddSBram Moolenaar 	return FAIL;
48789ccfebddSBram Moolenaar     }
48799ccfebddSBram Moolenaar 
48800d6f5d97SBram Moolenaar     // <HEADER>: <fileID> <versionnr>
48810d6f5d97SBram Moolenaar 							    // <fileID>
48829ccfebddSBram Moolenaar     fwv &= fwrite(VIMSPELLMAGIC, VIMSPELLMAGICL, (size_t)1, fd);
48839ccfebddSBram Moolenaar     if (fwv != (size_t)1)
48840d6f5d97SBram Moolenaar 	// Catch first write error, don't try writing more.
48859ccfebddSBram Moolenaar 	goto theend;
48869ccfebddSBram Moolenaar 
48870d6f5d97SBram Moolenaar     putc(VIMSPELLVERSION, fd);				    // <versionnr>
48889ccfebddSBram Moolenaar 
48899ccfebddSBram Moolenaar     /*
48909ccfebddSBram Moolenaar      * <SECTIONS>: <section> ... <sectionend>
48919ccfebddSBram Moolenaar      */
48929ccfebddSBram Moolenaar 
48930d6f5d97SBram Moolenaar     // SN_INFO: <infotext>
48949ccfebddSBram Moolenaar     if (spin->si_info != NULL)
48959ccfebddSBram Moolenaar     {
48960d6f5d97SBram Moolenaar 	putc(SN_INFO, fd);				// <sectionID>
48970d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
48989ccfebddSBram Moolenaar 
48999ccfebddSBram Moolenaar 	i = (int)STRLEN(spin->si_info);
49000d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)i, 4);			// <sectionlen>
49010d6f5d97SBram Moolenaar 	fwv &= fwrite(spin->si_info, (size_t)i, (size_t)1, fd); // <infotext>
49029ccfebddSBram Moolenaar     }
49039ccfebddSBram Moolenaar 
49040d6f5d97SBram Moolenaar     // SN_REGION: <regionname> ...
49050d6f5d97SBram Moolenaar     // Write the region names only if there is more than one.
49069ccfebddSBram Moolenaar     if (spin->si_region_count > 1)
49079ccfebddSBram Moolenaar     {
49080d6f5d97SBram Moolenaar 	putc(SN_REGION, fd);				// <sectionID>
49090d6f5d97SBram Moolenaar 	putc(SNF_REQUIRED, fd);				// <sectionflags>
49109ccfebddSBram Moolenaar 	l = spin->si_region_count * 2;
49110d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)l, 4);			// <sectionlen>
49129ccfebddSBram Moolenaar 	fwv &= fwrite(spin->si_region_name, (size_t)l, (size_t)1, fd);
49130d6f5d97SBram Moolenaar 							// <regionname> ...
49149ccfebddSBram Moolenaar 	regionmask = (1 << spin->si_region_count) - 1;
49159ccfebddSBram Moolenaar     }
49169ccfebddSBram Moolenaar     else
49179ccfebddSBram Moolenaar 	regionmask = 0;
49189ccfebddSBram Moolenaar 
49190d6f5d97SBram Moolenaar     // SN_CHARFLAGS: <charflagslen> <charflags> <folcharslen> <folchars>
49200d6f5d97SBram Moolenaar     //
49210d6f5d97SBram Moolenaar     // The table with character flags and the table for case folding.
49220d6f5d97SBram Moolenaar     // This makes sure the same characters are recognized as word characters
49230d6f5d97SBram Moolenaar     // when generating an when using a spell file.
49240d6f5d97SBram Moolenaar     // Skip this for ASCII, the table may conflict with the one used for
49250d6f5d97SBram Moolenaar     // 'encoding'.
49260d6f5d97SBram Moolenaar     // Also skip this for an .add.spl file, the main spell file must contain
49270d6f5d97SBram Moolenaar     // the table (avoids that it conflicts).  File is shorter too.
49289ccfebddSBram Moolenaar     if (!spin->si_ascii && !spin->si_add)
49299ccfebddSBram Moolenaar     {
49309ccfebddSBram Moolenaar 	char_u	folchars[128 * 8];
49319ccfebddSBram Moolenaar 	int	flags;
49329ccfebddSBram Moolenaar 
49330d6f5d97SBram Moolenaar 	putc(SN_CHARFLAGS, fd);				// <sectionID>
49340d6f5d97SBram Moolenaar 	putc(SNF_REQUIRED, fd);				// <sectionflags>
49359ccfebddSBram Moolenaar 
49360d6f5d97SBram Moolenaar 	// Form the <folchars> string first, we need to know its length.
49379ccfebddSBram Moolenaar 	l = 0;
49389ccfebddSBram Moolenaar 	for (i = 128; i < 256; ++i)
49399ccfebddSBram Moolenaar 	{
49409ccfebddSBram Moolenaar 	    if (has_mbyte)
49419ccfebddSBram Moolenaar 		l += mb_char2bytes(spelltab.st_fold[i], folchars + l);
49429ccfebddSBram Moolenaar 	    else
49439ccfebddSBram Moolenaar 		folchars[l++] = spelltab.st_fold[i];
49449ccfebddSBram Moolenaar 	}
49450d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)(1 + 128 + 2 + l), 4);	// <sectionlen>
49469ccfebddSBram Moolenaar 
49470d6f5d97SBram Moolenaar 	fputc(128, fd);					// <charflagslen>
49489ccfebddSBram Moolenaar 	for (i = 128; i < 256; ++i)
49499ccfebddSBram Moolenaar 	{
49509ccfebddSBram Moolenaar 	    flags = 0;
49519ccfebddSBram Moolenaar 	    if (spelltab.st_isw[i])
49529ccfebddSBram Moolenaar 		flags |= CF_WORD;
49539ccfebddSBram Moolenaar 	    if (spelltab.st_isu[i])
49549ccfebddSBram Moolenaar 		flags |= CF_UPPER;
49550d6f5d97SBram Moolenaar 	    fputc(flags, fd);				// <charflags>
49569ccfebddSBram Moolenaar 	}
49579ccfebddSBram Moolenaar 
49580d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)l, 2);			// <folcharslen>
49590d6f5d97SBram Moolenaar 	fwv &= fwrite(folchars, (size_t)l, (size_t)1, fd); // <folchars>
49609ccfebddSBram Moolenaar     }
49619ccfebddSBram Moolenaar 
49620d6f5d97SBram Moolenaar     // SN_MIDWORD: <midword>
49639ccfebddSBram Moolenaar     if (spin->si_midword != NULL)
49649ccfebddSBram Moolenaar     {
49650d6f5d97SBram Moolenaar 	putc(SN_MIDWORD, fd);				// <sectionID>
49660d6f5d97SBram Moolenaar 	putc(SNF_REQUIRED, fd);				// <sectionflags>
49679ccfebddSBram Moolenaar 
49689ccfebddSBram Moolenaar 	i = (int)STRLEN(spin->si_midword);
49690d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)i, 4);			// <sectionlen>
49709ccfebddSBram Moolenaar 	fwv &= fwrite(spin->si_midword, (size_t)i, (size_t)1, fd);
49710d6f5d97SBram Moolenaar 							// <midword>
49729ccfebddSBram Moolenaar     }
49739ccfebddSBram Moolenaar 
49740d6f5d97SBram Moolenaar     // SN_PREFCOND: <prefcondcnt> <prefcond> ...
49759ccfebddSBram Moolenaar     if (spin->si_prefcond.ga_len > 0)
49769ccfebddSBram Moolenaar     {
49770d6f5d97SBram Moolenaar 	putc(SN_PREFCOND, fd);				// <sectionID>
49780d6f5d97SBram Moolenaar 	putc(SNF_REQUIRED, fd);				// <sectionflags>
49799ccfebddSBram Moolenaar 
49809ccfebddSBram Moolenaar 	l = write_spell_prefcond(NULL, &spin->si_prefcond);
49810d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)l, 4);			// <sectionlen>
49829ccfebddSBram Moolenaar 
49839ccfebddSBram Moolenaar 	write_spell_prefcond(fd, &spin->si_prefcond);
49849ccfebddSBram Moolenaar     }
49859ccfebddSBram Moolenaar 
49860d6f5d97SBram Moolenaar     // SN_REP: <repcount> <rep> ...
49870d6f5d97SBram Moolenaar     // SN_SAL: <salflags> <salcount> <sal> ...
49880d6f5d97SBram Moolenaar     // SN_REPSAL: <repcount> <rep> ...
49899ccfebddSBram Moolenaar 
49900d6f5d97SBram Moolenaar     // round 1: SN_REP section
49910d6f5d97SBram Moolenaar     // round 2: SN_SAL section (unless SN_SOFO is used)
49920d6f5d97SBram Moolenaar     // round 3: SN_REPSAL section
49939ccfebddSBram Moolenaar     for (round = 1; round <= 3; ++round)
49949ccfebddSBram Moolenaar     {
49959ccfebddSBram Moolenaar 	if (round == 1)
49969ccfebddSBram Moolenaar 	    gap = &spin->si_rep;
49979ccfebddSBram Moolenaar 	else if (round == 2)
49989ccfebddSBram Moolenaar 	{
49990d6f5d97SBram Moolenaar 	    // Don't write SN_SAL when using a SN_SOFO section
50009ccfebddSBram Moolenaar 	    if (spin->si_sofofr != NULL && spin->si_sofoto != NULL)
50019ccfebddSBram Moolenaar 		continue;
50029ccfebddSBram Moolenaar 	    gap = &spin->si_sal;
50039ccfebddSBram Moolenaar 	}
50049ccfebddSBram Moolenaar 	else
50059ccfebddSBram Moolenaar 	    gap = &spin->si_repsal;
50069ccfebddSBram Moolenaar 
50070d6f5d97SBram Moolenaar 	// Don't write the section if there are no items.
50089ccfebddSBram Moolenaar 	if (gap->ga_len == 0)
50099ccfebddSBram Moolenaar 	    continue;
50109ccfebddSBram Moolenaar 
50110d6f5d97SBram Moolenaar 	// Sort the REP/REPSAL items.
50129ccfebddSBram Moolenaar 	if (round != 2)
50139ccfebddSBram Moolenaar 	    qsort(gap->ga_data, (size_t)gap->ga_len,
50149ccfebddSBram Moolenaar 					       sizeof(fromto_T), rep_compare);
50159ccfebddSBram Moolenaar 
50169ccfebddSBram Moolenaar 	i = round == 1 ? SN_REP : (round == 2 ? SN_SAL : SN_REPSAL);
50170d6f5d97SBram Moolenaar 	putc(i, fd);					// <sectionID>
50189ccfebddSBram Moolenaar 
50190d6f5d97SBram Moolenaar 	// This is for making suggestions, section is not required.
50200d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
50219ccfebddSBram Moolenaar 
50220d6f5d97SBram Moolenaar 	// Compute the length of what follows.
50230d6f5d97SBram Moolenaar 	l = 2;	    // count <repcount> or <salcount>
50249ccfebddSBram Moolenaar 	for (i = 0; i < gap->ga_len; ++i)
50259ccfebddSBram Moolenaar 	{
50269ccfebddSBram Moolenaar 	    ftp = &((fromto_T *)gap->ga_data)[i];
50270d6f5d97SBram Moolenaar 	    l += 1 + (int)STRLEN(ftp->ft_from);  // count <*fromlen> and <*from>
50280d6f5d97SBram Moolenaar 	    l += 1 + (int)STRLEN(ftp->ft_to);    // count <*tolen> and <*to>
50299ccfebddSBram Moolenaar 	}
50309ccfebddSBram Moolenaar 	if (round == 2)
50310d6f5d97SBram Moolenaar 	    ++l;	// count <salflags>
50320d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)l, 4);			// <sectionlen>
50339ccfebddSBram Moolenaar 
50349ccfebddSBram Moolenaar 	if (round == 2)
50359ccfebddSBram Moolenaar 	{
50369ccfebddSBram Moolenaar 	    i = 0;
50379ccfebddSBram Moolenaar 	    if (spin->si_followup)
50389ccfebddSBram Moolenaar 		i |= SAL_F0LLOWUP;
50399ccfebddSBram Moolenaar 	    if (spin->si_collapse)
50409ccfebddSBram Moolenaar 		i |= SAL_COLLAPSE;
50419ccfebddSBram Moolenaar 	    if (spin->si_rem_accents)
50429ccfebddSBram Moolenaar 		i |= SAL_REM_ACCENTS;
50430d6f5d97SBram Moolenaar 	    putc(i, fd);			// <salflags>
50449ccfebddSBram Moolenaar 	}
50459ccfebddSBram Moolenaar 
50460d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)gap->ga_len, 2);	// <repcount> or <salcount>
50479ccfebddSBram Moolenaar 	for (i = 0; i < gap->ga_len; ++i)
50489ccfebddSBram Moolenaar 	{
50490d6f5d97SBram Moolenaar 	    // <rep> : <repfromlen> <repfrom> <reptolen> <repto>
50500d6f5d97SBram Moolenaar 	    // <sal> : <salfromlen> <salfrom> <saltolen> <salto>
50519ccfebddSBram Moolenaar 	    ftp = &((fromto_T *)gap->ga_data)[i];
50529ccfebddSBram Moolenaar 	    for (rr = 1; rr <= 2; ++rr)
50539ccfebddSBram Moolenaar 	    {
50549ccfebddSBram Moolenaar 		p = rr == 1 ? ftp->ft_from : ftp->ft_to;
50559ccfebddSBram Moolenaar 		l = (int)STRLEN(p);
50569ccfebddSBram Moolenaar 		putc(l, fd);
50579ccfebddSBram Moolenaar 		if (l > 0)
50589ccfebddSBram Moolenaar 		    fwv &= fwrite(p, l, (size_t)1, fd);
50599ccfebddSBram Moolenaar 	    }
50609ccfebddSBram Moolenaar 	}
50619ccfebddSBram Moolenaar 
50629ccfebddSBram Moolenaar     }
50639ccfebddSBram Moolenaar 
50640d6f5d97SBram Moolenaar     // SN_SOFO: <sofofromlen> <sofofrom> <sofotolen> <sofoto>
50650d6f5d97SBram Moolenaar     // This is for making suggestions, section is not required.
50669ccfebddSBram Moolenaar     if (spin->si_sofofr != NULL && spin->si_sofoto != NULL)
50679ccfebddSBram Moolenaar     {
50680d6f5d97SBram Moolenaar 	putc(SN_SOFO, fd);				// <sectionID>
50690d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
50709ccfebddSBram Moolenaar 
50719ccfebddSBram Moolenaar 	l = (int)STRLEN(spin->si_sofofr);
50729ccfebddSBram Moolenaar 	put_bytes(fd, (long_u)(l + STRLEN(spin->si_sofoto) + 4), 4);
50730d6f5d97SBram Moolenaar 							// <sectionlen>
50749ccfebddSBram Moolenaar 
50750d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)l, 2);			// <sofofromlen>
50760d6f5d97SBram Moolenaar 	fwv &= fwrite(spin->si_sofofr, l, (size_t)1, fd); // <sofofrom>
50779ccfebddSBram Moolenaar 
50789ccfebddSBram Moolenaar 	l = (int)STRLEN(spin->si_sofoto);
50790d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)l, 2);			// <sofotolen>
50800d6f5d97SBram Moolenaar 	fwv &= fwrite(spin->si_sofoto, l, (size_t)1, fd); // <sofoto>
50819ccfebddSBram Moolenaar     }
50829ccfebddSBram Moolenaar 
50830d6f5d97SBram Moolenaar     // SN_WORDS: <word> ...
50840d6f5d97SBram Moolenaar     // This is for making suggestions, section is not required.
50859ccfebddSBram Moolenaar     if (spin->si_commonwords.ht_used > 0)
50869ccfebddSBram Moolenaar     {
50870d6f5d97SBram Moolenaar 	putc(SN_WORDS, fd);				// <sectionID>
50880d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
50899ccfebddSBram Moolenaar 
50900d6f5d97SBram Moolenaar 	// round 1: count the bytes
50910d6f5d97SBram Moolenaar 	// round 2: write the bytes
50929ccfebddSBram Moolenaar 	for (round = 1; round <= 2; ++round)
50939ccfebddSBram Moolenaar 	{
50949ccfebddSBram Moolenaar 	    int		todo;
50959ccfebddSBram Moolenaar 	    int		len = 0;
50969ccfebddSBram Moolenaar 	    hashitem_T	*hi;
50979ccfebddSBram Moolenaar 
50989ccfebddSBram Moolenaar 	    todo = (int)spin->si_commonwords.ht_used;
50999ccfebddSBram Moolenaar 	    for (hi = spin->si_commonwords.ht_array; todo > 0; ++hi)
51009ccfebddSBram Moolenaar 		if (!HASHITEM_EMPTY(hi))
51019ccfebddSBram Moolenaar 		{
51029ccfebddSBram Moolenaar 		    l = (int)STRLEN(hi->hi_key) + 1;
51039ccfebddSBram Moolenaar 		    len += l;
51040d6f5d97SBram Moolenaar 		    if (round == 2)			// <word>
51059ccfebddSBram Moolenaar 			fwv &= fwrite(hi->hi_key, (size_t)l, (size_t)1, fd);
51069ccfebddSBram Moolenaar 		    --todo;
51079ccfebddSBram Moolenaar 		}
51089ccfebddSBram Moolenaar 	    if (round == 1)
51090d6f5d97SBram Moolenaar 		put_bytes(fd, (long_u)len, 4);		// <sectionlen>
51109ccfebddSBram Moolenaar 	}
51119ccfebddSBram Moolenaar     }
51129ccfebddSBram Moolenaar 
51130d6f5d97SBram Moolenaar     // SN_MAP: <mapstr>
51140d6f5d97SBram Moolenaar     // This is for making suggestions, section is not required.
51159ccfebddSBram Moolenaar     if (spin->si_map.ga_len > 0)
51169ccfebddSBram Moolenaar     {
51170d6f5d97SBram Moolenaar 	putc(SN_MAP, fd);				// <sectionID>
51180d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
51199ccfebddSBram Moolenaar 	l = spin->si_map.ga_len;
51200d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)l, 4);			// <sectionlen>
51219ccfebddSBram Moolenaar 	fwv &= fwrite(spin->si_map.ga_data, (size_t)l, (size_t)1, fd);
51220d6f5d97SBram Moolenaar 							// <mapstr>
51239ccfebddSBram Moolenaar     }
51249ccfebddSBram Moolenaar 
51250d6f5d97SBram Moolenaar     // SN_SUGFILE: <timestamp>
51260d6f5d97SBram Moolenaar     // This is used to notify that a .sug file may be available and at the
51270d6f5d97SBram Moolenaar     // same time allows for checking that a .sug file that is found matches
51280d6f5d97SBram Moolenaar     // with this .spl file.  That's because the word numbers must be exactly
51290d6f5d97SBram Moolenaar     // right.
51309ccfebddSBram Moolenaar     if (!spin->si_nosugfile
51319ccfebddSBram Moolenaar 	    && (spin->si_sal.ga_len > 0
51329ccfebddSBram Moolenaar 		     || (spin->si_sofofr != NULL && spin->si_sofoto != NULL)))
51339ccfebddSBram Moolenaar     {
51340d6f5d97SBram Moolenaar 	putc(SN_SUGFILE, fd);				// <sectionID>
51350d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
51360d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)8, 4);			// <sectionlen>
51379ccfebddSBram Moolenaar 
51380d6f5d97SBram Moolenaar 	// Set si_sugtime and write it to the file.
51399ccfebddSBram Moolenaar 	spin->si_sugtime = time(NULL);
51400d6f5d97SBram Moolenaar 	put_time(fd, spin->si_sugtime);			// <timestamp>
51419ccfebddSBram Moolenaar     }
51429ccfebddSBram Moolenaar 
51430d6f5d97SBram Moolenaar     // SN_NOSPLITSUGS: nothing
51440d6f5d97SBram Moolenaar     // This is used to notify that no suggestions with word splits are to be
51450d6f5d97SBram Moolenaar     // made.
51469ccfebddSBram Moolenaar     if (spin->si_nosplitsugs)
51479ccfebddSBram Moolenaar     {
51480d6f5d97SBram Moolenaar 	putc(SN_NOSPLITSUGS, fd);			// <sectionID>
51490d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
51500d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)0, 4);			// <sectionlen>
51519ccfebddSBram Moolenaar     }
51529ccfebddSBram Moolenaar 
51530d6f5d97SBram Moolenaar     // SN_NOCOMPUNDSUGS: nothing
51540d6f5d97SBram Moolenaar     // This is used to notify that no suggestions with compounds are to be
51550d6f5d97SBram Moolenaar     // made.
51569ccfebddSBram Moolenaar     if (spin->si_nocompoundsugs)
51579ccfebddSBram Moolenaar     {
51580d6f5d97SBram Moolenaar 	putc(SN_NOCOMPOUNDSUGS, fd);			// <sectionID>
51590d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
51600d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)0, 4);			// <sectionlen>
51619ccfebddSBram Moolenaar     }
51629ccfebddSBram Moolenaar 
51630d6f5d97SBram Moolenaar     // SN_COMPOUND: compound info.
51640d6f5d97SBram Moolenaar     // We don't mark it required, when not supported all compound words will
51650d6f5d97SBram Moolenaar     // be bad words.
51669ccfebddSBram Moolenaar     if (spin->si_compflags != NULL)
51679ccfebddSBram Moolenaar     {
51680d6f5d97SBram Moolenaar 	putc(SN_COMPOUND, fd);				// <sectionID>
51690d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
51709ccfebddSBram Moolenaar 
51719ccfebddSBram Moolenaar 	l = (int)STRLEN(spin->si_compflags);
51729ccfebddSBram Moolenaar 	for (i = 0; i < spin->si_comppat.ga_len; ++i)
51739ccfebddSBram Moolenaar 	    l += (int)STRLEN(((char_u **)(spin->si_comppat.ga_data))[i]) + 1;
51740d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)(l + 7), 4);		// <sectionlen>
51759ccfebddSBram Moolenaar 
51760d6f5d97SBram Moolenaar 	putc(spin->si_compmax, fd);			// <compmax>
51770d6f5d97SBram Moolenaar 	putc(spin->si_compminlen, fd);			// <compminlen>
51780d6f5d97SBram Moolenaar 	putc(spin->si_compsylmax, fd);			// <compsylmax>
51790d6f5d97SBram Moolenaar 	putc(0, fd);		// for Vim 7.0b compatibility
51800d6f5d97SBram Moolenaar 	putc(spin->si_compoptions, fd);			// <compoptions>
51819ccfebddSBram Moolenaar 	put_bytes(fd, (long_u)spin->si_comppat.ga_len, 2);
51820d6f5d97SBram Moolenaar 							// <comppatcount>
51839ccfebddSBram Moolenaar 	for (i = 0; i < spin->si_comppat.ga_len; ++i)
51849ccfebddSBram Moolenaar 	{
51859ccfebddSBram Moolenaar 	    p = ((char_u **)(spin->si_comppat.ga_data))[i];
51860d6f5d97SBram Moolenaar 	    putc((int)STRLEN(p), fd);			// <comppatlen>
51879ccfebddSBram Moolenaar 	    fwv &= fwrite(p, (size_t)STRLEN(p), (size_t)1, fd);
51880d6f5d97SBram Moolenaar 							// <comppattext>
51899ccfebddSBram Moolenaar 	}
51900d6f5d97SBram Moolenaar 							// <compflags>
51919ccfebddSBram Moolenaar 	fwv &= fwrite(spin->si_compflags, (size_t)STRLEN(spin->si_compflags),
51929ccfebddSBram Moolenaar 							       (size_t)1, fd);
51939ccfebddSBram Moolenaar     }
51949ccfebddSBram Moolenaar 
51950d6f5d97SBram Moolenaar     // SN_NOBREAK: NOBREAK flag
51969ccfebddSBram Moolenaar     if (spin->si_nobreak)
51979ccfebddSBram Moolenaar     {
51980d6f5d97SBram Moolenaar 	putc(SN_NOBREAK, fd);				// <sectionID>
51990d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
52009ccfebddSBram Moolenaar 
52010d6f5d97SBram Moolenaar 	// It's empty, the presence of the section flags the feature.
52020d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)0, 4);			// <sectionlen>
52039ccfebddSBram Moolenaar     }
52049ccfebddSBram Moolenaar 
52050d6f5d97SBram Moolenaar     // SN_SYLLABLE: syllable info.
52060d6f5d97SBram Moolenaar     // We don't mark it required, when not supported syllables will not be
52070d6f5d97SBram Moolenaar     // counted.
52089ccfebddSBram Moolenaar     if (spin->si_syllable != NULL)
52099ccfebddSBram Moolenaar     {
52100d6f5d97SBram Moolenaar 	putc(SN_SYLLABLE, fd);				// <sectionID>
52110d6f5d97SBram Moolenaar 	putc(0, fd);					// <sectionflags>
52129ccfebddSBram Moolenaar 
52139ccfebddSBram Moolenaar 	l = (int)STRLEN(spin->si_syllable);
52140d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)l, 4);			// <sectionlen>
52159ccfebddSBram Moolenaar 	fwv &= fwrite(spin->si_syllable, (size_t)l, (size_t)1, fd);
52160d6f5d97SBram Moolenaar 							// <syllable>
52179ccfebddSBram Moolenaar     }
52189ccfebddSBram Moolenaar 
52190d6f5d97SBram Moolenaar     // end of <SECTIONS>
52200d6f5d97SBram Moolenaar     putc(SN_END, fd);					// <sectionend>
52219ccfebddSBram Moolenaar 
52229ccfebddSBram Moolenaar 
52239ccfebddSBram Moolenaar     /*
52249ccfebddSBram Moolenaar      * <LWORDTREE>  <KWORDTREE>  <PREFIXTREE>
52259ccfebddSBram Moolenaar      */
52269ccfebddSBram Moolenaar     spin->si_memtot = 0;
52279ccfebddSBram Moolenaar     for (round = 1; round <= 3; ++round)
52289ccfebddSBram Moolenaar     {
52299ccfebddSBram Moolenaar 	if (round == 1)
52309ccfebddSBram Moolenaar 	    tree = spin->si_foldroot->wn_sibling;
52319ccfebddSBram Moolenaar 	else if (round == 2)
52329ccfebddSBram Moolenaar 	    tree = spin->si_keeproot->wn_sibling;
52339ccfebddSBram Moolenaar 	else
52349ccfebddSBram Moolenaar 	    tree = spin->si_prefroot->wn_sibling;
52359ccfebddSBram Moolenaar 
52360d6f5d97SBram Moolenaar 	// Clear the index and wnode fields in the tree.
52379ccfebddSBram Moolenaar 	clear_node(tree);
52389ccfebddSBram Moolenaar 
52390d6f5d97SBram Moolenaar 	// Count the number of nodes.  Needed to be able to allocate the
52400d6f5d97SBram Moolenaar 	// memory when reading the nodes.  Also fills in index for shared
52410d6f5d97SBram Moolenaar 	// nodes.
52429ccfebddSBram Moolenaar 	nodecount = put_node(NULL, tree, 0, regionmask, round == 3);
52439ccfebddSBram Moolenaar 
52440d6f5d97SBram Moolenaar 	// number of nodes in 4 bytes
52450d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)nodecount, 4);	// <nodecount>
52469ccfebddSBram Moolenaar 	spin->si_memtot += nodecount + nodecount * sizeof(int);
52479ccfebddSBram Moolenaar 
52480d6f5d97SBram Moolenaar 	// Write the nodes.
52499ccfebddSBram Moolenaar 	(void)put_node(fd, tree, 0, regionmask, round == 3);
52509ccfebddSBram Moolenaar     }
52519ccfebddSBram Moolenaar 
52520d6f5d97SBram Moolenaar     // Write another byte to check for errors (file system full).
52539ccfebddSBram Moolenaar     if (putc(0, fd) == EOF)
52549ccfebddSBram Moolenaar 	retval = FAIL;
52559ccfebddSBram Moolenaar theend:
52569ccfebddSBram Moolenaar     if (fclose(fd) == EOF)
52579ccfebddSBram Moolenaar 	retval = FAIL;
52589ccfebddSBram Moolenaar 
52599ccfebddSBram Moolenaar     if (fwv != (size_t)1)
52609ccfebddSBram Moolenaar 	retval = FAIL;
52619ccfebddSBram Moolenaar     if (retval == FAIL)
5262f9e3e09fSBram Moolenaar 	emsg(_(e_write));
52639ccfebddSBram Moolenaar 
52649ccfebddSBram Moolenaar     return retval;
52659ccfebddSBram Moolenaar }
52669ccfebddSBram Moolenaar 
52679ccfebddSBram Moolenaar /*
52689ccfebddSBram Moolenaar  * Clear the index and wnode fields of "node", it siblings and its
52699ccfebddSBram Moolenaar  * children.  This is needed because they are a union with other items to save
52709ccfebddSBram Moolenaar  * space.
52719ccfebddSBram Moolenaar  */
52729ccfebddSBram Moolenaar     static void
clear_node(wordnode_T * node)52739ccfebddSBram Moolenaar clear_node(wordnode_T *node)
52749ccfebddSBram Moolenaar {
52759ccfebddSBram Moolenaar     wordnode_T	*np;
52769ccfebddSBram Moolenaar 
52779ccfebddSBram Moolenaar     if (node != NULL)
5278aeea7215SBram Moolenaar 	FOR_ALL_NODE_SIBLINGS(node, np)
52799ccfebddSBram Moolenaar 	{
52809ccfebddSBram Moolenaar 	    np->wn_u1.index = 0;
52819ccfebddSBram Moolenaar 	    np->wn_u2.wnode = NULL;
52829ccfebddSBram Moolenaar 
52839ccfebddSBram Moolenaar 	    if (np->wn_byte != NUL)
52849ccfebddSBram Moolenaar 		clear_node(np->wn_child);
52859ccfebddSBram Moolenaar 	}
52869ccfebddSBram Moolenaar }
52879ccfebddSBram Moolenaar 
52889ccfebddSBram Moolenaar 
52899ccfebddSBram Moolenaar /*
52909ccfebddSBram Moolenaar  * Dump a word tree at node "node".
52919ccfebddSBram Moolenaar  *
52929ccfebddSBram Moolenaar  * This first writes the list of possible bytes (siblings).  Then for each
52939ccfebddSBram Moolenaar  * byte recursively write the children.
52949ccfebddSBram Moolenaar  *
52959ccfebddSBram Moolenaar  * NOTE: The code here must match the code in read_tree_node(), since
52969ccfebddSBram Moolenaar  * assumptions are made about the indexes (so that we don't have to write them
52979ccfebddSBram Moolenaar  * in the file).
52989ccfebddSBram Moolenaar  *
52999ccfebddSBram Moolenaar  * Returns the number of nodes used.
53009ccfebddSBram Moolenaar  */
53019ccfebddSBram Moolenaar     static int
put_node(FILE * fd,wordnode_T * node,int idx,int regionmask,int prefixtree)53029ccfebddSBram Moolenaar put_node(
53030d6f5d97SBram Moolenaar     FILE	*fd,		// NULL when only counting
53049ccfebddSBram Moolenaar     wordnode_T	*node,
53059ccfebddSBram Moolenaar     int		idx,
53069ccfebddSBram Moolenaar     int		regionmask,
53070d6f5d97SBram Moolenaar     int		prefixtree)	// TRUE for PREFIXTREE
53089ccfebddSBram Moolenaar {
53099ccfebddSBram Moolenaar     int		newindex = idx;
53109ccfebddSBram Moolenaar     int		siblingcount = 0;
53119ccfebddSBram Moolenaar     wordnode_T	*np;
53129ccfebddSBram Moolenaar     int		flags;
53139ccfebddSBram Moolenaar 
53140d6f5d97SBram Moolenaar     // If "node" is zero the tree is empty.
53159ccfebddSBram Moolenaar     if (node == NULL)
53169ccfebddSBram Moolenaar 	return 0;
53179ccfebddSBram Moolenaar 
53180d6f5d97SBram Moolenaar     // Store the index where this node is written.
53199ccfebddSBram Moolenaar     node->wn_u1.index = idx;
53209ccfebddSBram Moolenaar 
53210d6f5d97SBram Moolenaar     // Count the number of siblings.
5322aeea7215SBram Moolenaar     FOR_ALL_NODE_SIBLINGS(node, np)
53239ccfebddSBram Moolenaar 	++siblingcount;
53249ccfebddSBram Moolenaar 
53250d6f5d97SBram Moolenaar     // Write the sibling count.
53269ccfebddSBram Moolenaar     if (fd != NULL)
53270d6f5d97SBram Moolenaar 	putc(siblingcount, fd);				// <siblingcount>
53289ccfebddSBram Moolenaar 
53290d6f5d97SBram Moolenaar     // Write each sibling byte and optionally extra info.
5330aeea7215SBram Moolenaar     FOR_ALL_NODE_SIBLINGS(node, np)
53319ccfebddSBram Moolenaar     {
53329ccfebddSBram Moolenaar 	if (np->wn_byte == 0)
53339ccfebddSBram Moolenaar 	{
53349ccfebddSBram Moolenaar 	    if (fd != NULL)
53359ccfebddSBram Moolenaar 	    {
53360d6f5d97SBram Moolenaar 		// For a NUL byte (end of word) write the flags etc.
53379ccfebddSBram Moolenaar 		if (prefixtree)
53389ccfebddSBram Moolenaar 		{
53390d6f5d97SBram Moolenaar 		    // In PREFIXTREE write the required affixID and the
53400d6f5d97SBram Moolenaar 		    // associated condition nr (stored in wn_region).  The
53410d6f5d97SBram Moolenaar 		    // byte value is misused to store the "rare" and "not
53420d6f5d97SBram Moolenaar 		    // combining" flags
53439ccfebddSBram Moolenaar 		    if (np->wn_flags == (short_u)PFX_FLAGS)
53440d6f5d97SBram Moolenaar 			putc(BY_NOFLAGS, fd);		// <byte>
53459ccfebddSBram Moolenaar 		    else
53469ccfebddSBram Moolenaar 		    {
53470d6f5d97SBram Moolenaar 			putc(BY_FLAGS, fd);		// <byte>
53480d6f5d97SBram Moolenaar 			putc(np->wn_flags, fd);		// <pflags>
53499ccfebddSBram Moolenaar 		    }
53500d6f5d97SBram Moolenaar 		    putc(np->wn_affixID, fd);		// <affixID>
53510d6f5d97SBram Moolenaar 		    put_bytes(fd, (long_u)np->wn_region, 2); // <prefcondnr>
53529ccfebddSBram Moolenaar 		}
53539ccfebddSBram Moolenaar 		else
53549ccfebddSBram Moolenaar 		{
53550d6f5d97SBram Moolenaar 		    // For word trees we write the flag/region items.
53569ccfebddSBram Moolenaar 		    flags = np->wn_flags;
53579ccfebddSBram Moolenaar 		    if (regionmask != 0 && np->wn_region != regionmask)
53589ccfebddSBram Moolenaar 			flags |= WF_REGION;
53599ccfebddSBram Moolenaar 		    if (np->wn_affixID != 0)
53609ccfebddSBram Moolenaar 			flags |= WF_AFX;
53619ccfebddSBram Moolenaar 		    if (flags == 0)
53629ccfebddSBram Moolenaar 		    {
53630d6f5d97SBram Moolenaar 			// word without flags or region
53640d6f5d97SBram Moolenaar 			putc(BY_NOFLAGS, fd);			// <byte>
53659ccfebddSBram Moolenaar 		    }
53669ccfebddSBram Moolenaar 		    else
53679ccfebddSBram Moolenaar 		    {
53689ccfebddSBram Moolenaar 			if (np->wn_flags >= 0x100)
53699ccfebddSBram Moolenaar 			{
53700d6f5d97SBram Moolenaar 			    putc(BY_FLAGS2, fd);		// <byte>
53710d6f5d97SBram Moolenaar 			    putc(flags, fd);			// <flags>
53720d6f5d97SBram Moolenaar 			    putc((unsigned)flags >> 8, fd);	// <flags2>
53739ccfebddSBram Moolenaar 			}
53749ccfebddSBram Moolenaar 			else
53759ccfebddSBram Moolenaar 			{
53760d6f5d97SBram Moolenaar 			    putc(BY_FLAGS, fd);			// <byte>
53770d6f5d97SBram Moolenaar 			    putc(flags, fd);			// <flags>
53789ccfebddSBram Moolenaar 			}
53799ccfebddSBram Moolenaar 			if (flags & WF_REGION)
53800d6f5d97SBram Moolenaar 			    putc(np->wn_region, fd);		// <region>
53819ccfebddSBram Moolenaar 			if (flags & WF_AFX)
53820d6f5d97SBram Moolenaar 			    putc(np->wn_affixID, fd);		// <affixID>
53839ccfebddSBram Moolenaar 		    }
53849ccfebddSBram Moolenaar 		}
53859ccfebddSBram Moolenaar 	    }
53869ccfebddSBram Moolenaar 	}
53879ccfebddSBram Moolenaar 	else
53889ccfebddSBram Moolenaar 	{
53899ccfebddSBram Moolenaar 	    if (np->wn_child->wn_u1.index != 0
53909ccfebddSBram Moolenaar 					 && np->wn_child->wn_u2.wnode != node)
53919ccfebddSBram Moolenaar 	    {
53920d6f5d97SBram Moolenaar 		// The child is written elsewhere, write the reference.
53939ccfebddSBram Moolenaar 		if (fd != NULL)
53949ccfebddSBram Moolenaar 		{
53950d6f5d97SBram Moolenaar 		    putc(BY_INDEX, fd);			// <byte>
53960d6f5d97SBram Moolenaar 							// <nodeidx>
53979ccfebddSBram Moolenaar 		    put_bytes(fd, (long_u)np->wn_child->wn_u1.index, 3);
53989ccfebddSBram Moolenaar 		}
53999ccfebddSBram Moolenaar 	    }
54009ccfebddSBram Moolenaar 	    else if (np->wn_child->wn_u2.wnode == NULL)
54010d6f5d97SBram Moolenaar 		// We will write the child below and give it an index.
54029ccfebddSBram Moolenaar 		np->wn_child->wn_u2.wnode = node;
54039ccfebddSBram Moolenaar 
54049ccfebddSBram Moolenaar 	    if (fd != NULL)
54050d6f5d97SBram Moolenaar 		if (putc(np->wn_byte, fd) == EOF) // <byte> or <xbyte>
54069ccfebddSBram Moolenaar 		{
5407f9e3e09fSBram Moolenaar 		    emsg(_(e_write));
54089ccfebddSBram Moolenaar 		    return 0;
54099ccfebddSBram Moolenaar 		}
54109ccfebddSBram Moolenaar 	}
54119ccfebddSBram Moolenaar     }
54129ccfebddSBram Moolenaar 
54130d6f5d97SBram Moolenaar     // Space used in the array when reading: one for each sibling and one for
54140d6f5d97SBram Moolenaar     // the count.
54159ccfebddSBram Moolenaar     newindex += siblingcount + 1;
54169ccfebddSBram Moolenaar 
54170d6f5d97SBram Moolenaar     // Recursively dump the children of each sibling.
5418aeea7215SBram Moolenaar     FOR_ALL_NODE_SIBLINGS(node, np)
54199ccfebddSBram Moolenaar 	if (np->wn_byte != 0 && np->wn_child->wn_u2.wnode == node)
54209ccfebddSBram Moolenaar 	    newindex = put_node(fd, np->wn_child, newindex, regionmask,
54219ccfebddSBram Moolenaar 								  prefixtree);
54229ccfebddSBram Moolenaar 
54239ccfebddSBram Moolenaar     return newindex;
54249ccfebddSBram Moolenaar }
54259ccfebddSBram Moolenaar 
54269ccfebddSBram Moolenaar 
54279ccfebddSBram Moolenaar /*
54289ccfebddSBram Moolenaar  * ":mkspell [-ascii] outfile  infile ..."
54299ccfebddSBram Moolenaar  * ":mkspell [-ascii] addfile"
54309ccfebddSBram Moolenaar  */
54319ccfebddSBram Moolenaar     void
ex_mkspell(exarg_T * eap)54329ccfebddSBram Moolenaar ex_mkspell(exarg_T *eap)
54339ccfebddSBram Moolenaar {
54349ccfebddSBram Moolenaar     int		fcount;
54359ccfebddSBram Moolenaar     char_u	**fnames;
54369ccfebddSBram Moolenaar     char_u	*arg = eap->arg;
54379ccfebddSBram Moolenaar     int		ascii = FALSE;
54389ccfebddSBram Moolenaar 
54399ccfebddSBram Moolenaar     if (STRNCMP(arg, "-ascii", 6) == 0)
54409ccfebddSBram Moolenaar     {
54419ccfebddSBram Moolenaar 	ascii = TRUE;
54429ccfebddSBram Moolenaar 	arg = skipwhite(arg + 6);
54439ccfebddSBram Moolenaar     }
54449ccfebddSBram Moolenaar 
54450d6f5d97SBram Moolenaar     // Expand all the remaining arguments (e.g., $VIMRUNTIME).
54469ccfebddSBram Moolenaar     if (get_arglist_exp(arg, &fcount, &fnames, FALSE) == OK)
54479ccfebddSBram Moolenaar     {
54489ccfebddSBram Moolenaar 	mkspell(fcount, fnames, ascii, eap->forceit, FALSE);
54499ccfebddSBram Moolenaar 	FreeWild(fcount, fnames);
54509ccfebddSBram Moolenaar     }
54519ccfebddSBram Moolenaar }
54529ccfebddSBram Moolenaar 
54539ccfebddSBram Moolenaar /*
54549ccfebddSBram Moolenaar  * Create the .sug file.
54559ccfebddSBram Moolenaar  * Uses the soundfold info in "spin".
54569ccfebddSBram Moolenaar  * Writes the file with the name "wfname", with ".spl" changed to ".sug".
54579ccfebddSBram Moolenaar  */
54589ccfebddSBram Moolenaar     static void
spell_make_sugfile(spellinfo_T * spin,char_u * wfname)54599ccfebddSBram Moolenaar spell_make_sugfile(spellinfo_T *spin, char_u *wfname)
54609ccfebddSBram Moolenaar {
54619ccfebddSBram Moolenaar     char_u	*fname = NULL;
54629ccfebddSBram Moolenaar     int		len;
54639ccfebddSBram Moolenaar     slang_T	*slang;
54649ccfebddSBram Moolenaar     int		free_slang = FALSE;
54659ccfebddSBram Moolenaar 
54669ccfebddSBram Moolenaar     /*
54679ccfebddSBram Moolenaar      * Read back the .spl file that was written.  This fills the required
54689ccfebddSBram Moolenaar      * info for soundfolding.  This also uses less memory than the
54699ccfebddSBram Moolenaar      * pointer-linked version of the trie.  And it avoids having two versions
54709ccfebddSBram Moolenaar      * of the code for the soundfolding stuff.
54719ccfebddSBram Moolenaar      * It might have been done already by spell_reload_one().
54729ccfebddSBram Moolenaar      */
5473aeea7215SBram Moolenaar     FOR_ALL_SPELL_LANGS(slang)
547499499b1cSBram Moolenaar 	if (fullpathcmp(wfname, slang->sl_fname, FALSE, TRUE) == FPC_SAME)
54759ccfebddSBram Moolenaar 	    break;
54769ccfebddSBram Moolenaar     if (slang == NULL)
54779ccfebddSBram Moolenaar     {
54789ccfebddSBram Moolenaar 	spell_message(spin, (char_u *)_("Reading back spell file..."));
54799ccfebddSBram Moolenaar 	slang = spell_load_file(wfname, NULL, NULL, FALSE);
54809ccfebddSBram Moolenaar 	if (slang == NULL)
54819ccfebddSBram Moolenaar 	    return;
54829ccfebddSBram Moolenaar 	free_slang = TRUE;
54839ccfebddSBram Moolenaar     }
54849ccfebddSBram Moolenaar 
54859ccfebddSBram Moolenaar     /*
54869ccfebddSBram Moolenaar      * Clear the info in "spin" that is used.
54879ccfebddSBram Moolenaar      */
54889ccfebddSBram Moolenaar     spin->si_blocks = NULL;
54899ccfebddSBram Moolenaar     spin->si_blocks_cnt = 0;
54900d6f5d97SBram Moolenaar     spin->si_compress_cnt = 0;	    // will stay at 0 all the time
54919ccfebddSBram Moolenaar     spin->si_free_count = 0;
54929ccfebddSBram Moolenaar     spin->si_first_free = NULL;
54939ccfebddSBram Moolenaar     spin->si_foldwcount = 0;
54949ccfebddSBram Moolenaar 
54959ccfebddSBram Moolenaar     /*
54969ccfebddSBram Moolenaar      * Go through the trie of good words, soundfold each word and add it to
54979ccfebddSBram Moolenaar      * the soundfold trie.
54989ccfebddSBram Moolenaar      */
54999ccfebddSBram Moolenaar     spell_message(spin, (char_u *)_("Performing soundfolding..."));
55009ccfebddSBram Moolenaar     if (sug_filltree(spin, slang) == FAIL)
55019ccfebddSBram Moolenaar 	goto theend;
55029ccfebddSBram Moolenaar 
55039ccfebddSBram Moolenaar     /*
55049ccfebddSBram Moolenaar      * Create the table which links each soundfold word with a list of the
55059ccfebddSBram Moolenaar      * good words it may come from.  Creates buffer "spin->si_spellbuf".
55069ccfebddSBram Moolenaar      * This also removes the wordnr from the NUL byte entries to make
55079ccfebddSBram Moolenaar      * compression possible.
55089ccfebddSBram Moolenaar      */
55099ccfebddSBram Moolenaar     if (sug_maketable(spin) == FAIL)
55109ccfebddSBram Moolenaar 	goto theend;
55119ccfebddSBram Moolenaar 
5512f9e3e09fSBram Moolenaar     smsg(_("Number of words after soundfolding: %ld"),
55139ccfebddSBram Moolenaar 				 (long)spin->si_spellbuf->b_ml.ml_line_count);
55149ccfebddSBram Moolenaar 
55159ccfebddSBram Moolenaar     /*
55169ccfebddSBram Moolenaar      * Compress the soundfold trie.
55179ccfebddSBram Moolenaar      */
55189ccfebddSBram Moolenaar     spell_message(spin, (char_u *)_(msg_compressing));
5519408c23b0SBram Moolenaar     wordtree_compress(spin, spin->si_foldroot, "case-folded");
55209ccfebddSBram Moolenaar 
55219ccfebddSBram Moolenaar     /*
55229ccfebddSBram Moolenaar      * Write the .sug file.
55239ccfebddSBram Moolenaar      * Make the file name by changing ".spl" to ".sug".
55249ccfebddSBram Moolenaar      */
55259ccfebddSBram Moolenaar     fname = alloc(MAXPATHL);
55269ccfebddSBram Moolenaar     if (fname == NULL)
55279ccfebddSBram Moolenaar 	goto theend;
55289ccfebddSBram Moolenaar     vim_strncpy(fname, wfname, MAXPATHL - 1);
55299ccfebddSBram Moolenaar     len = (int)STRLEN(fname);
55309ccfebddSBram Moolenaar     fname[len - 2] = 'u';
55319ccfebddSBram Moolenaar     fname[len - 1] = 'g';
55329ccfebddSBram Moolenaar     sug_write(spin, fname);
55339ccfebddSBram Moolenaar 
55349ccfebddSBram Moolenaar theend:
55359ccfebddSBram Moolenaar     vim_free(fname);
55369ccfebddSBram Moolenaar     if (free_slang)
55379ccfebddSBram Moolenaar 	slang_free(slang);
55389ccfebddSBram Moolenaar     free_blocks(spin->si_blocks);
55399ccfebddSBram Moolenaar     close_spellbuf(spin->si_spellbuf);
55409ccfebddSBram Moolenaar }
55419ccfebddSBram Moolenaar 
55429ccfebddSBram Moolenaar /*
55439ccfebddSBram Moolenaar  * Build the soundfold trie for language "slang".
55449ccfebddSBram Moolenaar  */
55459ccfebddSBram Moolenaar     static int
sug_filltree(spellinfo_T * spin,slang_T * slang)55469ccfebddSBram Moolenaar sug_filltree(spellinfo_T *spin, slang_T *slang)
55479ccfebddSBram Moolenaar {
55489ccfebddSBram Moolenaar     char_u	*byts;
55499ccfebddSBram Moolenaar     idx_T	*idxs;
55509ccfebddSBram Moolenaar     int		depth;
55519ccfebddSBram Moolenaar     idx_T	arridx[MAXWLEN];
55529ccfebddSBram Moolenaar     int		curi[MAXWLEN];
55539ccfebddSBram Moolenaar     char_u	tword[MAXWLEN];
55549ccfebddSBram Moolenaar     char_u	tsalword[MAXWLEN];
55559ccfebddSBram Moolenaar     int		c;
55569ccfebddSBram Moolenaar     idx_T	n;
55579ccfebddSBram Moolenaar     unsigned	words_done = 0;
55589ccfebddSBram Moolenaar     int		wordcount[MAXWLEN];
55599ccfebddSBram Moolenaar 
55600d6f5d97SBram Moolenaar     // We use si_foldroot for the soundfolded trie.
55619ccfebddSBram Moolenaar     spin->si_foldroot = wordtree_alloc(spin);
55629ccfebddSBram Moolenaar     if (spin->si_foldroot == NULL)
55639ccfebddSBram Moolenaar 	return FAIL;
55649ccfebddSBram Moolenaar 
55650d6f5d97SBram Moolenaar     // let tree_add_word() know we're adding to the soundfolded tree
55669ccfebddSBram Moolenaar     spin->si_sugtree = TRUE;
55679ccfebddSBram Moolenaar 
55689ccfebddSBram Moolenaar     /*
55699ccfebddSBram Moolenaar      * Go through the whole case-folded tree, soundfold each word and put it
55709ccfebddSBram Moolenaar      * in the trie.
55719ccfebddSBram Moolenaar      */
55729ccfebddSBram Moolenaar     byts = slang->sl_fbyts;
55739ccfebddSBram Moolenaar     idxs = slang->sl_fidxs;
55749ccfebddSBram Moolenaar 
55759ccfebddSBram Moolenaar     arridx[0] = 0;
55769ccfebddSBram Moolenaar     curi[0] = 1;
55779ccfebddSBram Moolenaar     wordcount[0] = 0;
55789ccfebddSBram Moolenaar 
55799ccfebddSBram Moolenaar     depth = 0;
55809ccfebddSBram Moolenaar     while (depth >= 0 && !got_int)
55819ccfebddSBram Moolenaar     {
55829ccfebddSBram Moolenaar 	if (curi[depth] > byts[arridx[depth]])
55839ccfebddSBram Moolenaar 	{
55840d6f5d97SBram Moolenaar 	    // Done all bytes at this node, go up one level.
55859ccfebddSBram Moolenaar 	    idxs[arridx[depth]] = wordcount[depth];
55869ccfebddSBram Moolenaar 	    if (depth > 0)
55879ccfebddSBram Moolenaar 		wordcount[depth - 1] += wordcount[depth];
55889ccfebddSBram Moolenaar 
55899ccfebddSBram Moolenaar 	    --depth;
55909ccfebddSBram Moolenaar 	    line_breakcheck();
55919ccfebddSBram Moolenaar 	}
55929ccfebddSBram Moolenaar 	else
55939ccfebddSBram Moolenaar 	{
55949ccfebddSBram Moolenaar 
55950d6f5d97SBram Moolenaar 	    // Do one more byte at this node.
55969ccfebddSBram Moolenaar 	    n = arridx[depth] + curi[depth];
55979ccfebddSBram Moolenaar 	    ++curi[depth];
55989ccfebddSBram Moolenaar 
55999ccfebddSBram Moolenaar 	    c = byts[n];
56009ccfebddSBram Moolenaar 	    if (c == 0)
56019ccfebddSBram Moolenaar 	    {
56020d6f5d97SBram Moolenaar 		// Sound-fold the word.
56039ccfebddSBram Moolenaar 		tword[depth] = NUL;
56049ccfebddSBram Moolenaar 		spell_soundfold(slang, tword, TRUE, tsalword);
56059ccfebddSBram Moolenaar 
56060d6f5d97SBram Moolenaar 		// We use the "flags" field for the MSB of the wordnr,
56070d6f5d97SBram Moolenaar 		// "region" for the LSB of the wordnr.
56089ccfebddSBram Moolenaar 		if (tree_add_word(spin, tsalword, spin->si_foldroot,
56099ccfebddSBram Moolenaar 				words_done >> 16, words_done & 0xffff,
56109ccfebddSBram Moolenaar 							   0) == FAIL)
56119ccfebddSBram Moolenaar 		    return FAIL;
56129ccfebddSBram Moolenaar 
56139ccfebddSBram Moolenaar 		++words_done;
56149ccfebddSBram Moolenaar 		++wordcount[depth];
56159ccfebddSBram Moolenaar 
56160d6f5d97SBram Moolenaar 		// Reset the block count each time to avoid compression
56170d6f5d97SBram Moolenaar 		// kicking in.
56189ccfebddSBram Moolenaar 		spin->si_blocks_cnt = 0;
56199ccfebddSBram Moolenaar 
56200d6f5d97SBram Moolenaar 		// Skip over any other NUL bytes (same word with different
562107399e7fSBram Moolenaar 		// flags).  But don't go over the end.
562207399e7fSBram Moolenaar 		while (n + 1 < slang->sl_fbyts_len && byts[n + 1] == 0)
56239ccfebddSBram Moolenaar 		{
56249ccfebddSBram Moolenaar 		    ++n;
56259ccfebddSBram Moolenaar 		    ++curi[depth];
56269ccfebddSBram Moolenaar 		}
56279ccfebddSBram Moolenaar 	    }
56289ccfebddSBram Moolenaar 	    else
56299ccfebddSBram Moolenaar 	    {
56300d6f5d97SBram Moolenaar 		// Normal char, go one level deeper.
56319ccfebddSBram Moolenaar 		tword[depth++] = c;
56329ccfebddSBram Moolenaar 		arridx[depth] = idxs[n];
56339ccfebddSBram Moolenaar 		curi[depth] = 1;
56349ccfebddSBram Moolenaar 		wordcount[depth] = 0;
56359ccfebddSBram Moolenaar 	    }
56369ccfebddSBram Moolenaar 	}
56379ccfebddSBram Moolenaar     }
56389ccfebddSBram Moolenaar 
5639f9e3e09fSBram Moolenaar     smsg(_("Total number of words: %d"), words_done);
56409ccfebddSBram Moolenaar 
56419ccfebddSBram Moolenaar     return OK;
56429ccfebddSBram Moolenaar }
56439ccfebddSBram Moolenaar 
56449ccfebddSBram Moolenaar /*
56459ccfebddSBram Moolenaar  * Make the table that links each word in the soundfold trie to the words it
56469ccfebddSBram Moolenaar  * can be produced from.
56479ccfebddSBram Moolenaar  * This is not unlike lines in a file, thus use a memfile to be able to access
56489ccfebddSBram Moolenaar  * the table efficiently.
56499ccfebddSBram Moolenaar  * Returns FAIL when out of memory.
56509ccfebddSBram Moolenaar  */
56519ccfebddSBram Moolenaar     static int
sug_maketable(spellinfo_T * spin)56529ccfebddSBram Moolenaar sug_maketable(spellinfo_T *spin)
56539ccfebddSBram Moolenaar {
56549ccfebddSBram Moolenaar     garray_T	ga;
56559ccfebddSBram Moolenaar     int		res = OK;
56569ccfebddSBram Moolenaar 
56570d6f5d97SBram Moolenaar     // Allocate a buffer, open a memline for it and create the swap file
56580d6f5d97SBram Moolenaar     // (uses a temp file, not a .swp file).
56599ccfebddSBram Moolenaar     spin->si_spellbuf = open_spellbuf();
56609ccfebddSBram Moolenaar     if (spin->si_spellbuf == NULL)
56619ccfebddSBram Moolenaar 	return FAIL;
56629ccfebddSBram Moolenaar 
56630d6f5d97SBram Moolenaar     // Use a buffer to store the line info, avoids allocating many small
56640d6f5d97SBram Moolenaar     // pieces of memory.
56659ccfebddSBram Moolenaar     ga_init2(&ga, 1, 100);
56669ccfebddSBram Moolenaar 
56670d6f5d97SBram Moolenaar     // recursively go through the tree
56689ccfebddSBram Moolenaar     if (sug_filltable(spin, spin->si_foldroot->wn_sibling, 0, &ga) == -1)
56699ccfebddSBram Moolenaar 	res = FAIL;
56709ccfebddSBram Moolenaar 
56719ccfebddSBram Moolenaar     ga_clear(&ga);
56729ccfebddSBram Moolenaar     return res;
56739ccfebddSBram Moolenaar }
56749ccfebddSBram Moolenaar 
56759ccfebddSBram Moolenaar /*
56769ccfebddSBram Moolenaar  * Fill the table for one node and its children.
56779ccfebddSBram Moolenaar  * Returns the wordnr at the start of the node.
56789ccfebddSBram Moolenaar  * Returns -1 when out of memory.
56799ccfebddSBram Moolenaar  */
56809ccfebddSBram Moolenaar     static int
sug_filltable(spellinfo_T * spin,wordnode_T * node,int startwordnr,garray_T * gap)56819ccfebddSBram Moolenaar sug_filltable(
56829ccfebddSBram Moolenaar     spellinfo_T	*spin,
56839ccfebddSBram Moolenaar     wordnode_T	*node,
56849ccfebddSBram Moolenaar     int		startwordnr,
56850d6f5d97SBram Moolenaar     garray_T	*gap)	    // place to store line of numbers
56869ccfebddSBram Moolenaar {
56879ccfebddSBram Moolenaar     wordnode_T	*p, *np;
56889ccfebddSBram Moolenaar     int		wordnr = startwordnr;
56899ccfebddSBram Moolenaar     int		nr;
56909ccfebddSBram Moolenaar     int		prev_nr;
56919ccfebddSBram Moolenaar 
5692aeea7215SBram Moolenaar     FOR_ALL_NODE_SIBLINGS(node, p)
56939ccfebddSBram Moolenaar     {
56949ccfebddSBram Moolenaar 	if (p->wn_byte == NUL)
56959ccfebddSBram Moolenaar 	{
56969ccfebddSBram Moolenaar 	    gap->ga_len = 0;
56979ccfebddSBram Moolenaar 	    prev_nr = 0;
56989ccfebddSBram Moolenaar 	    for (np = p; np != NULL && np->wn_byte == NUL; np = np->wn_sibling)
56999ccfebddSBram Moolenaar 	    {
57009ccfebddSBram Moolenaar 		if (ga_grow(gap, 10) == FAIL)
57019ccfebddSBram Moolenaar 		    return -1;
57029ccfebddSBram Moolenaar 
57039ccfebddSBram Moolenaar 		nr = (np->wn_flags << 16) + (np->wn_region & 0xffff);
57040d6f5d97SBram Moolenaar 		// Compute the offset from the previous nr and store the
57050d6f5d97SBram Moolenaar 		// offset in a way that it takes a minimum number of bytes.
57060d6f5d97SBram Moolenaar 		// It's a bit like utf-8, but without the need to mark
57070d6f5d97SBram Moolenaar 		// following bytes.
57089ccfebddSBram Moolenaar 		nr -= prev_nr;
57099ccfebddSBram Moolenaar 		prev_nr += nr;
57109ccfebddSBram Moolenaar 		gap->ga_len += offset2bytes(nr,
57119ccfebddSBram Moolenaar 					 (char_u *)gap->ga_data + gap->ga_len);
57129ccfebddSBram Moolenaar 	    }
57139ccfebddSBram Moolenaar 
57140d6f5d97SBram Moolenaar 	    // add the NUL byte
57159ccfebddSBram Moolenaar 	    ((char_u *)gap->ga_data)[gap->ga_len++] = NUL;
57169ccfebddSBram Moolenaar 
57179ccfebddSBram Moolenaar 	    if (ml_append_buf(spin->si_spellbuf, (linenr_T)wordnr,
57189ccfebddSBram Moolenaar 				     gap->ga_data, gap->ga_len, TRUE) == FAIL)
57199ccfebddSBram Moolenaar 		return -1;
57209ccfebddSBram Moolenaar 	    ++wordnr;
57219ccfebddSBram Moolenaar 
57220d6f5d97SBram Moolenaar 	    // Remove extra NUL entries, we no longer need them. We don't
57230d6f5d97SBram Moolenaar 	    // bother freeing the nodes, the won't be reused anyway.
57249ccfebddSBram Moolenaar 	    while (p->wn_sibling != NULL && p->wn_sibling->wn_byte == NUL)
57259ccfebddSBram Moolenaar 		p->wn_sibling = p->wn_sibling->wn_sibling;
57269ccfebddSBram Moolenaar 
57270d6f5d97SBram Moolenaar 	    // Clear the flags on the remaining NUL node, so that compression
57280d6f5d97SBram Moolenaar 	    // works a lot better.
57299ccfebddSBram Moolenaar 	    p->wn_flags = 0;
57309ccfebddSBram Moolenaar 	    p->wn_region = 0;
57319ccfebddSBram Moolenaar 	}
57329ccfebddSBram Moolenaar 	else
57339ccfebddSBram Moolenaar 	{
57349ccfebddSBram Moolenaar 	    wordnr = sug_filltable(spin, p->wn_child, wordnr, gap);
57359ccfebddSBram Moolenaar 	    if (wordnr == -1)
57369ccfebddSBram Moolenaar 		return -1;
57379ccfebddSBram Moolenaar 	}
57389ccfebddSBram Moolenaar     }
57399ccfebddSBram Moolenaar     return wordnr;
57409ccfebddSBram Moolenaar }
57419ccfebddSBram Moolenaar 
57429ccfebddSBram Moolenaar /*
57439ccfebddSBram Moolenaar  * Convert an offset into a minimal number of bytes.
57449ccfebddSBram Moolenaar  * Similar to utf_char2byters, but use 8 bits in followup bytes and avoid NUL
57459ccfebddSBram Moolenaar  * bytes.
57469ccfebddSBram Moolenaar  */
57479ccfebddSBram Moolenaar     static int
offset2bytes(int nr,char_u * buf)57489ccfebddSBram Moolenaar offset2bytes(int nr, char_u *buf)
57499ccfebddSBram Moolenaar {
57509ccfebddSBram Moolenaar     int	    rem;
57519ccfebddSBram Moolenaar     int	    b1, b2, b3, b4;
57529ccfebddSBram Moolenaar 
57530d6f5d97SBram Moolenaar     // Split the number in parts of base 255.  We need to avoid NUL bytes.
57549ccfebddSBram Moolenaar     b1 = nr % 255 + 1;
57559ccfebddSBram Moolenaar     rem = nr / 255;
57569ccfebddSBram Moolenaar     b2 = rem % 255 + 1;
57579ccfebddSBram Moolenaar     rem = rem / 255;
57589ccfebddSBram Moolenaar     b3 = rem % 255 + 1;
57599ccfebddSBram Moolenaar     b4 = rem / 255 + 1;
57609ccfebddSBram Moolenaar 
57610d6f5d97SBram Moolenaar     if (b4 > 1 || b3 > 0x1f)	// 4 bytes
57629ccfebddSBram Moolenaar     {
57639ccfebddSBram Moolenaar 	buf[0] = 0xe0 + b4;
57649ccfebddSBram Moolenaar 	buf[1] = b3;
57659ccfebddSBram Moolenaar 	buf[2] = b2;
57669ccfebddSBram Moolenaar 	buf[3] = b1;
57679ccfebddSBram Moolenaar 	return 4;
57689ccfebddSBram Moolenaar     }
57690d6f5d97SBram Moolenaar     if (b3 > 1 || b2 > 0x3f )	// 3 bytes
57709ccfebddSBram Moolenaar     {
57719ccfebddSBram Moolenaar 	buf[0] = 0xc0 + b3;
57729ccfebddSBram Moolenaar 	buf[1] = b2;
57739ccfebddSBram Moolenaar 	buf[2] = b1;
57749ccfebddSBram Moolenaar 	return 3;
57759ccfebddSBram Moolenaar     }
57760d6f5d97SBram Moolenaar     if (b2 > 1 || b1 > 0x7f )	// 2 bytes
57779ccfebddSBram Moolenaar     {
57789ccfebddSBram Moolenaar 	buf[0] = 0x80 + b2;
57799ccfebddSBram Moolenaar 	buf[1] = b1;
57809ccfebddSBram Moolenaar 	return 2;
57819ccfebddSBram Moolenaar     }
57820d6f5d97SBram Moolenaar 				// 1 byte
57839ccfebddSBram Moolenaar     buf[0] = b1;
57849ccfebddSBram Moolenaar     return 1;
57859ccfebddSBram Moolenaar }
57869ccfebddSBram Moolenaar 
57879ccfebddSBram Moolenaar /*
57889ccfebddSBram Moolenaar  * Write the .sug file in "fname".
57899ccfebddSBram Moolenaar  */
57909ccfebddSBram Moolenaar     static void
sug_write(spellinfo_T * spin,char_u * fname)57919ccfebddSBram Moolenaar sug_write(spellinfo_T *spin, char_u *fname)
57929ccfebddSBram Moolenaar {
57939ccfebddSBram Moolenaar     FILE	*fd;
57949ccfebddSBram Moolenaar     wordnode_T	*tree;
57959ccfebddSBram Moolenaar     int		nodecount;
57969ccfebddSBram Moolenaar     int		wcount;
57979ccfebddSBram Moolenaar     char_u	*line;
57989ccfebddSBram Moolenaar     linenr_T	lnum;
57999ccfebddSBram Moolenaar     int		len;
58009ccfebddSBram Moolenaar 
58010d6f5d97SBram Moolenaar     // Create the file.  Note that an existing file is silently overwritten!
58029ccfebddSBram Moolenaar     fd = mch_fopen((char *)fname, "w");
58039ccfebddSBram Moolenaar     if (fd == NULL)
58049ccfebddSBram Moolenaar     {
5805f9e3e09fSBram Moolenaar 	semsg(_(e_notopen), fname);
58069ccfebddSBram Moolenaar 	return;
58079ccfebddSBram Moolenaar     }
58089ccfebddSBram Moolenaar 
58099ccfebddSBram Moolenaar     vim_snprintf((char *)IObuff, IOSIZE,
58109ccfebddSBram Moolenaar 				  _("Writing suggestion file %s..."), fname);
58119ccfebddSBram Moolenaar     spell_message(spin, IObuff);
58129ccfebddSBram Moolenaar 
58139ccfebddSBram Moolenaar     /*
58149ccfebddSBram Moolenaar      * <SUGHEADER>: <fileID> <versionnr> <timestamp>
58159ccfebddSBram Moolenaar      */
58160d6f5d97SBram Moolenaar     if (fwrite(VIMSUGMAGIC, VIMSUGMAGICL, (size_t)1, fd) != 1) // <fileID>
58179ccfebddSBram Moolenaar     {
5818f9e3e09fSBram Moolenaar 	emsg(_(e_write));
58199ccfebddSBram Moolenaar 	goto theend;
58209ccfebddSBram Moolenaar     }
58210d6f5d97SBram Moolenaar     putc(VIMSUGVERSION, fd);				// <versionnr>
58229ccfebddSBram Moolenaar 
58230d6f5d97SBram Moolenaar     // Write si_sugtime to the file.
58240d6f5d97SBram Moolenaar     put_time(fd, spin->si_sugtime);			// <timestamp>
58259ccfebddSBram Moolenaar 
58269ccfebddSBram Moolenaar     /*
58279ccfebddSBram Moolenaar      * <SUGWORDTREE>
58289ccfebddSBram Moolenaar      */
58299ccfebddSBram Moolenaar     spin->si_memtot = 0;
58309ccfebddSBram Moolenaar     tree = spin->si_foldroot->wn_sibling;
58319ccfebddSBram Moolenaar 
58320d6f5d97SBram Moolenaar     // Clear the index and wnode fields in the tree.
58339ccfebddSBram Moolenaar     clear_node(tree);
58349ccfebddSBram Moolenaar 
58350d6f5d97SBram Moolenaar     // Count the number of nodes.  Needed to be able to allocate the
58360d6f5d97SBram Moolenaar     // memory when reading the nodes.  Also fills in index for shared
58370d6f5d97SBram Moolenaar     // nodes.
58389ccfebddSBram Moolenaar     nodecount = put_node(NULL, tree, 0, 0, FALSE);
58399ccfebddSBram Moolenaar 
58400d6f5d97SBram Moolenaar     // number of nodes in 4 bytes
58410d6f5d97SBram Moolenaar     put_bytes(fd, (long_u)nodecount, 4);	// <nodecount>
58429ccfebddSBram Moolenaar     spin->si_memtot += nodecount + nodecount * sizeof(int);
58439ccfebddSBram Moolenaar 
58440d6f5d97SBram Moolenaar     // Write the nodes.
58459ccfebddSBram Moolenaar     (void)put_node(fd, tree, 0, 0, FALSE);
58469ccfebddSBram Moolenaar 
58479ccfebddSBram Moolenaar     /*
58489ccfebddSBram Moolenaar      * <SUGTABLE>: <sugwcount> <sugline> ...
58499ccfebddSBram Moolenaar      */
58509ccfebddSBram Moolenaar     wcount = spin->si_spellbuf->b_ml.ml_line_count;
58510d6f5d97SBram Moolenaar     put_bytes(fd, (long_u)wcount, 4);	// <sugwcount>
58529ccfebddSBram Moolenaar 
58539ccfebddSBram Moolenaar     for (lnum = 1; lnum <= (linenr_T)wcount; ++lnum)
58549ccfebddSBram Moolenaar     {
58550d6f5d97SBram Moolenaar 	// <sugline>: <sugnr> ... NUL
58569ccfebddSBram Moolenaar 	line = ml_get_buf(spin->si_spellbuf, lnum, FALSE);
58579ccfebddSBram Moolenaar 	len = (int)STRLEN(line) + 1;
58589ccfebddSBram Moolenaar 	if (fwrite(line, (size_t)len, (size_t)1, fd) == 0)
58599ccfebddSBram Moolenaar 	{
5860f9e3e09fSBram Moolenaar 	    emsg(_(e_write));
58619ccfebddSBram Moolenaar 	    goto theend;
58629ccfebddSBram Moolenaar 	}
58639ccfebddSBram Moolenaar 	spin->si_memtot += len;
58649ccfebddSBram Moolenaar     }
58659ccfebddSBram Moolenaar 
58660d6f5d97SBram Moolenaar     // Write another byte to check for errors.
58679ccfebddSBram Moolenaar     if (putc(0, fd) == EOF)
5868f9e3e09fSBram Moolenaar 	emsg(_(e_write));
58699ccfebddSBram Moolenaar 
58709ccfebddSBram Moolenaar     vim_snprintf((char *)IObuff, IOSIZE,
58719ccfebddSBram Moolenaar 		 _("Estimated runtime memory use: %d bytes"), spin->si_memtot);
58729ccfebddSBram Moolenaar     spell_message(spin, IObuff);
58739ccfebddSBram Moolenaar 
58749ccfebddSBram Moolenaar theend:
58750d6f5d97SBram Moolenaar     // close the file
58769ccfebddSBram Moolenaar     fclose(fd);
58779ccfebddSBram Moolenaar }
58789ccfebddSBram Moolenaar 
58799ccfebddSBram Moolenaar 
58809ccfebddSBram Moolenaar /*
58819ccfebddSBram Moolenaar  * Create a Vim spell file from one or more word lists.
58829ccfebddSBram Moolenaar  * "fnames[0]" is the output file name.
58839ccfebddSBram Moolenaar  * "fnames[fcount - 1]" is the last input file name.
58849ccfebddSBram Moolenaar  * Exception: when "fnames[0]" ends in ".add" it's used as the input file name
58859ccfebddSBram Moolenaar  * and ".spl" is appended to make the output file name.
58869ccfebddSBram Moolenaar  */
58879ccfebddSBram Moolenaar     void
mkspell(int fcount,char_u ** fnames,int ascii,int over_write,int added_word)58889ccfebddSBram Moolenaar mkspell(
58899ccfebddSBram Moolenaar     int		fcount,
58909ccfebddSBram Moolenaar     char_u	**fnames,
58910d6f5d97SBram Moolenaar     int		ascii,		    // -ascii argument given
58920d6f5d97SBram Moolenaar     int		over_write,	    // overwrite existing output file
58930d6f5d97SBram Moolenaar     int		added_word)	    // invoked through "zg"
58949ccfebddSBram Moolenaar {
58959ccfebddSBram Moolenaar     char_u	*fname = NULL;
58969ccfebddSBram Moolenaar     char_u	*wfname;
58979ccfebddSBram Moolenaar     char_u	**innames;
58989ccfebddSBram Moolenaar     int		incount;
58992993ac5fSBram Moolenaar     afffile_T	*(afile[MAXREGIONS]);
59009ccfebddSBram Moolenaar     int		i;
59019ccfebddSBram Moolenaar     int		len;
59029ccfebddSBram Moolenaar     stat_T	st;
59039ccfebddSBram Moolenaar     int		error = FALSE;
59049ccfebddSBram Moolenaar     spellinfo_T spin;
59059ccfebddSBram Moolenaar 
5906a80faa89SBram Moolenaar     CLEAR_FIELD(spin);
59079ccfebddSBram Moolenaar     spin.si_verbose = !added_word;
59089ccfebddSBram Moolenaar     spin.si_ascii = ascii;
59099ccfebddSBram Moolenaar     spin.si_followup = TRUE;
59109ccfebddSBram Moolenaar     spin.si_rem_accents = TRUE;
59119ccfebddSBram Moolenaar     ga_init2(&spin.si_rep, (int)sizeof(fromto_T), 20);
59129ccfebddSBram Moolenaar     ga_init2(&spin.si_repsal, (int)sizeof(fromto_T), 20);
59139ccfebddSBram Moolenaar     ga_init2(&spin.si_sal, (int)sizeof(fromto_T), 20);
59149ccfebddSBram Moolenaar     ga_init2(&spin.si_map, (int)sizeof(char_u), 100);
59159ccfebddSBram Moolenaar     ga_init2(&spin.si_comppat, (int)sizeof(char_u *), 20);
59169ccfebddSBram Moolenaar     ga_init2(&spin.si_prefcond, (int)sizeof(char_u *), 50);
59179ccfebddSBram Moolenaar     hash_init(&spin.si_commonwords);
59180d6f5d97SBram Moolenaar     spin.si_newcompID = 127;	// start compound ID at first maximum
59199ccfebddSBram Moolenaar 
59200d6f5d97SBram Moolenaar     // default: fnames[0] is output file, following are input files
5921927b7dd0SBram Moolenaar     // When "fcount" is 1 there is only one file.
5922927b7dd0SBram Moolenaar     innames = &fnames[fcount == 1 ? 0 : 1];
59239ccfebddSBram Moolenaar     incount = fcount - 1;
59249ccfebddSBram Moolenaar 
59259ccfebddSBram Moolenaar     wfname = alloc(MAXPATHL);
59269ccfebddSBram Moolenaar     if (wfname == NULL)
59279ccfebddSBram Moolenaar 	return;
59289ccfebddSBram Moolenaar 
59299ccfebddSBram Moolenaar     if (fcount >= 1)
59309ccfebddSBram Moolenaar     {
59319ccfebddSBram Moolenaar 	len = (int)STRLEN(fnames[0]);
59329ccfebddSBram Moolenaar 	if (fcount == 1 && len > 4 && STRCMP(fnames[0] + len - 4, ".add") == 0)
59339ccfebddSBram Moolenaar 	{
59340d6f5d97SBram Moolenaar 	    // For ":mkspell path/en.latin1.add" output file is
59350d6f5d97SBram Moolenaar 	    // "path/en.latin1.add.spl".
59369ccfebddSBram Moolenaar 	    incount = 1;
59379ccfebddSBram Moolenaar 	    vim_snprintf((char *)wfname, MAXPATHL, "%s.spl", fnames[0]);
59389ccfebddSBram Moolenaar 	}
59399ccfebddSBram Moolenaar 	else if (fcount == 1)
59409ccfebddSBram Moolenaar 	{
59410d6f5d97SBram Moolenaar 	    // For ":mkspell path/vim" output file is "path/vim.latin1.spl".
59429ccfebddSBram Moolenaar 	    incount = 1;
59439ccfebddSBram Moolenaar 	    vim_snprintf((char *)wfname, MAXPATHL, SPL_FNAME_TMPL,
59449ccfebddSBram Moolenaar 		  fnames[0], spin.si_ascii ? (char_u *)"ascii" : spell_enc());
59459ccfebddSBram Moolenaar 	}
59469ccfebddSBram Moolenaar 	else if (len > 4 && STRCMP(fnames[0] + len - 4, ".spl") == 0)
59479ccfebddSBram Moolenaar 	{
59480d6f5d97SBram Moolenaar 	    // Name ends in ".spl", use as the file name.
59499ccfebddSBram Moolenaar 	    vim_strncpy(wfname, fnames[0], MAXPATHL - 1);
59509ccfebddSBram Moolenaar 	}
59519ccfebddSBram Moolenaar 	else
59520d6f5d97SBram Moolenaar 	    // Name should be language, make the file name from it.
59539ccfebddSBram Moolenaar 	    vim_snprintf((char *)wfname, MAXPATHL, SPL_FNAME_TMPL,
59549ccfebddSBram Moolenaar 		  fnames[0], spin.si_ascii ? (char_u *)"ascii" : spell_enc());
59559ccfebddSBram Moolenaar 
59560d6f5d97SBram Moolenaar 	// Check for .ascii.spl.
59579ccfebddSBram Moolenaar 	if (strstr((char *)gettail(wfname), SPL_FNAME_ASCII) != NULL)
59589ccfebddSBram Moolenaar 	    spin.si_ascii = TRUE;
59599ccfebddSBram Moolenaar 
59600d6f5d97SBram Moolenaar 	// Check for .add.spl.
59619ccfebddSBram Moolenaar 	if (strstr((char *)gettail(wfname), SPL_FNAME_ADD) != NULL)
59629ccfebddSBram Moolenaar 	    spin.si_add = TRUE;
59639ccfebddSBram Moolenaar     }
59649ccfebddSBram Moolenaar 
59659ccfebddSBram Moolenaar     if (incount <= 0)
59660d6f5d97SBram Moolenaar 	emsg(_(e_invarg));	// need at least output and input names
59679ccfebddSBram Moolenaar     else if (vim_strchr(gettail(wfname), '_') != NULL)
5968f9e3e09fSBram Moolenaar 	emsg(_("E751: Output file name must not have region name"));
59692993ac5fSBram Moolenaar     else if (incount > MAXREGIONS)
5970b5443cc4SBram Moolenaar 	semsg(_("E754: Only up to %d regions supported"), MAXREGIONS);
59719ccfebddSBram Moolenaar     else
59729ccfebddSBram Moolenaar     {
59730d6f5d97SBram Moolenaar 	// Check for overwriting before doing things that may take a lot of
59740d6f5d97SBram Moolenaar 	// time.
59759ccfebddSBram Moolenaar 	if (!over_write && mch_stat((char *)wfname, &st) >= 0)
59769ccfebddSBram Moolenaar 	{
5977108010aaSBram Moolenaar 	    emsg(_(e_file_exists));
59789ccfebddSBram Moolenaar 	    goto theend;
59799ccfebddSBram Moolenaar 	}
59809ccfebddSBram Moolenaar 	if (mch_isdir(wfname))
59819ccfebddSBram Moolenaar 	{
5982108010aaSBram Moolenaar 	    semsg(_(e_src_is_directory), wfname);
59839ccfebddSBram Moolenaar 	    goto theend;
59849ccfebddSBram Moolenaar 	}
59859ccfebddSBram Moolenaar 
59869ccfebddSBram Moolenaar 	fname = alloc(MAXPATHL);
59879ccfebddSBram Moolenaar 	if (fname == NULL)
59889ccfebddSBram Moolenaar 	    goto theend;
59899ccfebddSBram Moolenaar 
59909ccfebddSBram Moolenaar 	/*
59919ccfebddSBram Moolenaar 	 * Init the aff and dic pointers.
59929ccfebddSBram Moolenaar 	 * Get the region names if there are more than 2 arguments.
59939ccfebddSBram Moolenaar 	 */
59949ccfebddSBram Moolenaar 	for (i = 0; i < incount; ++i)
59959ccfebddSBram Moolenaar 	{
59969ccfebddSBram Moolenaar 	    afile[i] = NULL;
59979ccfebddSBram Moolenaar 
59989ccfebddSBram Moolenaar 	    if (incount > 1)
59999ccfebddSBram Moolenaar 	    {
60009ccfebddSBram Moolenaar 		len = (int)STRLEN(innames[i]);
60019ccfebddSBram Moolenaar 		if (STRLEN(gettail(innames[i])) < 5
60029ccfebddSBram Moolenaar 						|| innames[i][len - 3] != '_')
60039ccfebddSBram Moolenaar 		{
6004f9e3e09fSBram Moolenaar 		    semsg(_("E755: Invalid region in %s"), innames[i]);
60059ccfebddSBram Moolenaar 		    goto theend;
60069ccfebddSBram Moolenaar 		}
60079ccfebddSBram Moolenaar 		spin.si_region_name[i * 2] = TOLOWER_ASC(innames[i][len - 2]);
60089ccfebddSBram Moolenaar 		spin.si_region_name[i * 2 + 1] =
60099ccfebddSBram Moolenaar 					     TOLOWER_ASC(innames[i][len - 1]);
60109ccfebddSBram Moolenaar 	    }
60119ccfebddSBram Moolenaar 	}
60129ccfebddSBram Moolenaar 	spin.si_region_count = incount;
60139ccfebddSBram Moolenaar 
60149ccfebddSBram Moolenaar 	spin.si_foldroot = wordtree_alloc(&spin);
60159ccfebddSBram Moolenaar 	spin.si_keeproot = wordtree_alloc(&spin);
60169ccfebddSBram Moolenaar 	spin.si_prefroot = wordtree_alloc(&spin);
60179ccfebddSBram Moolenaar 	if (spin.si_foldroot == NULL
60189ccfebddSBram Moolenaar 		|| spin.si_keeproot == NULL
60199ccfebddSBram Moolenaar 		|| spin.si_prefroot == NULL)
60209ccfebddSBram Moolenaar 	{
60219ccfebddSBram Moolenaar 	    free_blocks(spin.si_blocks);
60229ccfebddSBram Moolenaar 	    goto theend;
60239ccfebddSBram Moolenaar 	}
60249ccfebddSBram Moolenaar 
60250d6f5d97SBram Moolenaar 	// When not producing a .add.spl file clear the character table when
60260d6f5d97SBram Moolenaar 	// we encounter one in the .aff file.  This means we dump the current
60270d6f5d97SBram Moolenaar 	// one in the .spl file if the .aff file doesn't define one.  That's
60280d6f5d97SBram Moolenaar 	// better than guessing the contents, the table will match a
60290d6f5d97SBram Moolenaar 	// previously loaded spell file.
60309ccfebddSBram Moolenaar 	if (!spin.si_add)
60319ccfebddSBram Moolenaar 	    spin.si_clear_chartab = TRUE;
60329ccfebddSBram Moolenaar 
60339ccfebddSBram Moolenaar 	/*
60349ccfebddSBram Moolenaar 	 * Read all the .aff and .dic files.
60359ccfebddSBram Moolenaar 	 * Text is converted to 'encoding'.
60369ccfebddSBram Moolenaar 	 * Words are stored in the case-folded and keep-case trees.
60379ccfebddSBram Moolenaar 	 */
60389ccfebddSBram Moolenaar 	for (i = 0; i < incount && !error; ++i)
60399ccfebddSBram Moolenaar 	{
60409ccfebddSBram Moolenaar 	    spin.si_conv.vc_type = CONV_NONE;
60419ccfebddSBram Moolenaar 	    spin.si_region = 1 << i;
60429ccfebddSBram Moolenaar 
60439ccfebddSBram Moolenaar 	    vim_snprintf((char *)fname, MAXPATHL, "%s.aff", innames[i]);
60449ccfebddSBram Moolenaar 	    if (mch_stat((char *)fname, &st) >= 0)
60459ccfebddSBram Moolenaar 	    {
60460d6f5d97SBram Moolenaar 		// Read the .aff file.  Will init "spin->si_conv" based on the
60470d6f5d97SBram Moolenaar 		// "SET" line.
60489ccfebddSBram Moolenaar 		afile[i] = spell_read_aff(&spin, fname);
60499ccfebddSBram Moolenaar 		if (afile[i] == NULL)
60509ccfebddSBram Moolenaar 		    error = TRUE;
60519ccfebddSBram Moolenaar 		else
60529ccfebddSBram Moolenaar 		{
60530d6f5d97SBram Moolenaar 		    // Read the .dic file and store the words in the trees.
60549ccfebddSBram Moolenaar 		    vim_snprintf((char *)fname, MAXPATHL, "%s.dic",
60559ccfebddSBram Moolenaar 								  innames[i]);
60569ccfebddSBram Moolenaar 		    if (spell_read_dic(&spin, fname, afile[i]) == FAIL)
60579ccfebddSBram Moolenaar 			error = TRUE;
60589ccfebddSBram Moolenaar 		}
60599ccfebddSBram Moolenaar 	    }
60609ccfebddSBram Moolenaar 	    else
60619ccfebddSBram Moolenaar 	    {
60620d6f5d97SBram Moolenaar 		// No .aff file, try reading the file as a word list.  Store
60630d6f5d97SBram Moolenaar 		// the words in the trees.
60649ccfebddSBram Moolenaar 		if (spell_read_wordfile(&spin, innames[i]) == FAIL)
60659ccfebddSBram Moolenaar 		    error = TRUE;
60669ccfebddSBram Moolenaar 	    }
60679ccfebddSBram Moolenaar 
60680d6f5d97SBram Moolenaar 	    // Free any conversion stuff.
60699ccfebddSBram Moolenaar 	    convert_setup(&spin.si_conv, NULL, NULL);
60709ccfebddSBram Moolenaar 	}
60719ccfebddSBram Moolenaar 
60729ccfebddSBram Moolenaar 	if (spin.si_compflags != NULL && spin.si_nobreak)
607332526b3cSBram Moolenaar 	    msg(_("Warning: both compounding and NOBREAK specified"));
60749ccfebddSBram Moolenaar 
60759ccfebddSBram Moolenaar 	if (!error && !got_int)
60769ccfebddSBram Moolenaar 	{
60779ccfebddSBram Moolenaar 	    /*
60789ccfebddSBram Moolenaar 	     * Combine tails in the tree.
60799ccfebddSBram Moolenaar 	     */
60809ccfebddSBram Moolenaar 	    spell_message(&spin, (char_u *)_(msg_compressing));
6081408c23b0SBram Moolenaar 	    wordtree_compress(&spin, spin.si_foldroot, "case-folded");
6082408c23b0SBram Moolenaar 	    wordtree_compress(&spin, spin.si_keeproot, "keep-case");
6083408c23b0SBram Moolenaar 	    wordtree_compress(&spin, spin.si_prefroot, "prefixes");
60849ccfebddSBram Moolenaar 	}
60859ccfebddSBram Moolenaar 
60869ccfebddSBram Moolenaar 	if (!error && !got_int)
60879ccfebddSBram Moolenaar 	{
60889ccfebddSBram Moolenaar 	    /*
60899ccfebddSBram Moolenaar 	     * Write the info in the spell file.
60909ccfebddSBram Moolenaar 	     */
60919ccfebddSBram Moolenaar 	    vim_snprintf((char *)IObuff, IOSIZE,
60929ccfebddSBram Moolenaar 				      _("Writing spell file %s..."), wfname);
60939ccfebddSBram Moolenaar 	    spell_message(&spin, IObuff);
60949ccfebddSBram Moolenaar 
60959ccfebddSBram Moolenaar 	    error = write_vim_spell(&spin, wfname) == FAIL;
60969ccfebddSBram Moolenaar 
60979ccfebddSBram Moolenaar 	    spell_message(&spin, (char_u *)_("Done!"));
60989ccfebddSBram Moolenaar 	    vim_snprintf((char *)IObuff, IOSIZE,
60999ccfebddSBram Moolenaar 		 _("Estimated runtime memory use: %d bytes"), spin.si_memtot);
61009ccfebddSBram Moolenaar 	    spell_message(&spin, IObuff);
61019ccfebddSBram Moolenaar 
61029ccfebddSBram Moolenaar 	    /*
61039ccfebddSBram Moolenaar 	     * If the file is loaded need to reload it.
61049ccfebddSBram Moolenaar 	     */
61059ccfebddSBram Moolenaar 	    if (!error)
61069ccfebddSBram Moolenaar 		spell_reload_one(wfname, added_word);
61079ccfebddSBram Moolenaar 	}
61089ccfebddSBram Moolenaar 
61090d6f5d97SBram Moolenaar 	// Free the allocated memory.
61109ccfebddSBram Moolenaar 	ga_clear(&spin.si_rep);
61119ccfebddSBram Moolenaar 	ga_clear(&spin.si_repsal);
61129ccfebddSBram Moolenaar 	ga_clear(&spin.si_sal);
61139ccfebddSBram Moolenaar 	ga_clear(&spin.si_map);
61149ccfebddSBram Moolenaar 	ga_clear(&spin.si_comppat);
61159ccfebddSBram Moolenaar 	ga_clear(&spin.si_prefcond);
61169ccfebddSBram Moolenaar 	hash_clear_all(&spin.si_commonwords, 0);
61179ccfebddSBram Moolenaar 
61180d6f5d97SBram Moolenaar 	// Free the .aff file structures.
61199ccfebddSBram Moolenaar 	for (i = 0; i < incount; ++i)
61209ccfebddSBram Moolenaar 	    if (afile[i] != NULL)
61219ccfebddSBram Moolenaar 		spell_free_aff(afile[i]);
61229ccfebddSBram Moolenaar 
61230d6f5d97SBram Moolenaar 	// Free all the bits and pieces at once.
61249ccfebddSBram Moolenaar 	free_blocks(spin.si_blocks);
61259ccfebddSBram Moolenaar 
61269ccfebddSBram Moolenaar 	/*
61279ccfebddSBram Moolenaar 	 * If there is soundfolding info and no NOSUGFILE item create the
61289ccfebddSBram Moolenaar 	 * .sug file with the soundfolded word trie.
61299ccfebddSBram Moolenaar 	 */
61309ccfebddSBram Moolenaar 	if (spin.si_sugtime != 0 && !error && !got_int)
61319ccfebddSBram Moolenaar 	    spell_make_sugfile(&spin, wfname);
61329ccfebddSBram Moolenaar 
61339ccfebddSBram Moolenaar     }
61349ccfebddSBram Moolenaar 
61359ccfebddSBram Moolenaar theend:
61369ccfebddSBram Moolenaar     vim_free(fname);
61379ccfebddSBram Moolenaar     vim_free(wfname);
61389ccfebddSBram Moolenaar }
61399ccfebddSBram Moolenaar 
61409ccfebddSBram Moolenaar /*
61419ccfebddSBram Moolenaar  * Display a message for spell file processing when 'verbose' is set or using
61429ccfebddSBram Moolenaar  * ":mkspell".  "str" can be IObuff.
61439ccfebddSBram Moolenaar  */
61449ccfebddSBram Moolenaar     static void
spell_message(spellinfo_T * spin,char_u * str)61459ccfebddSBram Moolenaar spell_message(spellinfo_T *spin, char_u *str)
61469ccfebddSBram Moolenaar {
61479ccfebddSBram Moolenaar     if (spin->si_verbose || p_verbose > 2)
61489ccfebddSBram Moolenaar     {
61499ccfebddSBram Moolenaar 	if (!spin->si_verbose)
61509ccfebddSBram Moolenaar 	    verbose_enter();
615132526b3cSBram Moolenaar 	msg((char *)str);
61529ccfebddSBram Moolenaar 	out_flush();
61539ccfebddSBram Moolenaar 	if (!spin->si_verbose)
61549ccfebddSBram Moolenaar 	    verbose_leave();
61559ccfebddSBram Moolenaar     }
61569ccfebddSBram Moolenaar }
61579ccfebddSBram Moolenaar 
61589ccfebddSBram Moolenaar /*
61599ccfebddSBram Moolenaar  * ":[count]spellgood  {word}"
61609ccfebddSBram Moolenaar  * ":[count]spellwrong {word}"
61619ccfebddSBram Moolenaar  * ":[count]spellundo  {word}"
616208cc374dSBram Moolenaar  * ":[count]spellrare  {word}"
61639ccfebddSBram Moolenaar  */
61649ccfebddSBram Moolenaar     void
ex_spell(exarg_T * eap)61659ccfebddSBram Moolenaar ex_spell(exarg_T *eap)
61669ccfebddSBram Moolenaar {
616708cc374dSBram Moolenaar     spell_add_word(eap->arg, (int)STRLEN(eap->arg),
616808cc374dSBram Moolenaar 		eap->cmdidx == CMD_spellwrong ? SPELL_ADD_BAD :
616908cc374dSBram Moolenaar 		eap->cmdidx == CMD_spellrare ? SPELL_ADD_RARE : SPELL_ADD_GOOD,
61709ccfebddSBram Moolenaar 				   eap->forceit ? 0 : (int)eap->line2,
61719ccfebddSBram Moolenaar 				   eap->cmdidx == CMD_spellundo);
61729ccfebddSBram Moolenaar }
61739ccfebddSBram Moolenaar 
61749ccfebddSBram Moolenaar /*
617508cc374dSBram Moolenaar  * Add "word[len]" to 'spellfile' as a good, rare or bad word.
61769ccfebddSBram Moolenaar  */
61779ccfebddSBram Moolenaar     void
spell_add_word(char_u * word,int len,int what,int idx,int undo)61789ccfebddSBram Moolenaar spell_add_word(
61799ccfebddSBram Moolenaar     char_u	*word,
61809ccfebddSBram Moolenaar     int		len,
618108cc374dSBram Moolenaar     int		what,	    // SPELL_ADD_ values
618208cc374dSBram Moolenaar     int		idx,	    // "zG" and "zW": zero, otherwise index in
618308cc374dSBram Moolenaar 			    // 'spellfile'
618408cc374dSBram Moolenaar     int		undo)	    // TRUE for "zug", "zuG", "zuw" and "zuW"
61859ccfebddSBram Moolenaar {
61869ccfebddSBram Moolenaar     FILE	*fd = NULL;
61879ccfebddSBram Moolenaar     buf_T	*buf = NULL;
61889ccfebddSBram Moolenaar     int		new_spf = FALSE;
61899ccfebddSBram Moolenaar     char_u	*fname;
61909ccfebddSBram Moolenaar     char_u	*fnamebuf = NULL;
61919ccfebddSBram Moolenaar     char_u	line[MAXWLEN * 2];
61929ccfebddSBram Moolenaar     long	fpos, fpos_next = 0;
61939ccfebddSBram Moolenaar     int		i;
61949ccfebddSBram Moolenaar     char_u	*spf;
61959ccfebddSBram Moolenaar 
61960d6f5d97SBram Moolenaar     if (idx == 0)	    // use internal wordlist
61979ccfebddSBram Moolenaar     {
61989ccfebddSBram Moolenaar 	if (int_wordlist == NULL)
61999ccfebddSBram Moolenaar 	{
62009ccfebddSBram Moolenaar 	    int_wordlist = vim_tempname('s', FALSE);
62019ccfebddSBram Moolenaar 	    if (int_wordlist == NULL)
62029ccfebddSBram Moolenaar 		return;
62039ccfebddSBram Moolenaar 	}
62049ccfebddSBram Moolenaar 	fname = int_wordlist;
62059ccfebddSBram Moolenaar     }
62069ccfebddSBram Moolenaar     else
62079ccfebddSBram Moolenaar     {
62080d6f5d97SBram Moolenaar 	// If 'spellfile' isn't set figure out a good default value.
62099ccfebddSBram Moolenaar 	if (*curwin->w_s->b_p_spf == NUL)
62109ccfebddSBram Moolenaar 	{
62119ccfebddSBram Moolenaar 	    init_spellfile();
62129ccfebddSBram Moolenaar 	    new_spf = TRUE;
62139ccfebddSBram Moolenaar 	}
62149ccfebddSBram Moolenaar 
62159ccfebddSBram Moolenaar 	if (*curwin->w_s->b_p_spf == NUL)
62169ccfebddSBram Moolenaar 	{
6217f9e3e09fSBram Moolenaar 	    semsg(_(e_notset), "spellfile");
62189ccfebddSBram Moolenaar 	    return;
62199ccfebddSBram Moolenaar 	}
62209ccfebddSBram Moolenaar 	fnamebuf = alloc(MAXPATHL);
62219ccfebddSBram Moolenaar 	if (fnamebuf == NULL)
62229ccfebddSBram Moolenaar 	    return;
62239ccfebddSBram Moolenaar 
62249ccfebddSBram Moolenaar 	for (spf = curwin->w_s->b_p_spf, i = 1; *spf != NUL; ++i)
62259ccfebddSBram Moolenaar 	{
62269ccfebddSBram Moolenaar 	    copy_option_part(&spf, fnamebuf, MAXPATHL, ",");
62279ccfebddSBram Moolenaar 	    if (i == idx)
62289ccfebddSBram Moolenaar 		break;
62299ccfebddSBram Moolenaar 	    if (*spf == NUL)
62309ccfebddSBram Moolenaar 	    {
6231b5443cc4SBram Moolenaar 		semsg(_("E765: 'spellfile' does not have %d entries"), idx);
62329ccfebddSBram Moolenaar 		vim_free(fnamebuf);
62339ccfebddSBram Moolenaar 		return;
62349ccfebddSBram Moolenaar 	    }
62359ccfebddSBram Moolenaar 	}
62369ccfebddSBram Moolenaar 
62370d6f5d97SBram Moolenaar 	// Check that the user isn't editing the .add file somewhere.
62389ccfebddSBram Moolenaar 	buf = buflist_findname_exp(fnamebuf);
62399ccfebddSBram Moolenaar 	if (buf != NULL && buf->b_ml.ml_mfp == NULL)
62409ccfebddSBram Moolenaar 	    buf = NULL;
62419ccfebddSBram Moolenaar 	if (buf != NULL && bufIsChanged(buf))
62429ccfebddSBram Moolenaar 	{
6243f9e3e09fSBram Moolenaar 	    emsg(_(e_bufloaded));
62449ccfebddSBram Moolenaar 	    vim_free(fnamebuf);
62459ccfebddSBram Moolenaar 	    return;
62469ccfebddSBram Moolenaar 	}
62479ccfebddSBram Moolenaar 
62489ccfebddSBram Moolenaar 	fname = fnamebuf;
62499ccfebddSBram Moolenaar     }
62509ccfebddSBram Moolenaar 
625108cc374dSBram Moolenaar     if (what == SPELL_ADD_BAD || undo)
62529ccfebddSBram Moolenaar     {
62530d6f5d97SBram Moolenaar 	// When the word appears as good word we need to remove that one,
62540d6f5d97SBram Moolenaar 	// since its flags sort before the one with WF_BANNED.
62559ccfebddSBram Moolenaar 	fd = mch_fopen((char *)fname, "r");
62569ccfebddSBram Moolenaar 	if (fd != NULL)
62579ccfebddSBram Moolenaar 	{
62589ccfebddSBram Moolenaar 	    while (!vim_fgets(line, MAXWLEN * 2, fd))
62599ccfebddSBram Moolenaar 	    {
62609ccfebddSBram Moolenaar 		fpos = fpos_next;
62619ccfebddSBram Moolenaar 		fpos_next = ftell(fd);
62629ccfebddSBram Moolenaar 		if (STRNCMP(word, line, len) == 0
62639ccfebddSBram Moolenaar 			&& (line[len] == '/' || line[len] < ' '))
62649ccfebddSBram Moolenaar 		{
62650d6f5d97SBram Moolenaar 		    // Found duplicate word.  Remove it by writing a '#' at
62660d6f5d97SBram Moolenaar 		    // the start of the line.  Mixing reading and writing
62670d6f5d97SBram Moolenaar 		    // doesn't work for all systems, close the file first.
62689ccfebddSBram Moolenaar 		    fclose(fd);
62699ccfebddSBram Moolenaar 		    fd = mch_fopen((char *)fname, "r+");
62709ccfebddSBram Moolenaar 		    if (fd == NULL)
62719ccfebddSBram Moolenaar 			break;
62729ccfebddSBram Moolenaar 		    if (fseek(fd, fpos, SEEK_SET) == 0)
62739ccfebddSBram Moolenaar 		    {
62749ccfebddSBram Moolenaar 			fputc('#', fd);
62759ccfebddSBram Moolenaar 			if (undo)
62769ccfebddSBram Moolenaar 			{
62779ccfebddSBram Moolenaar 			    home_replace(NULL, fname, NameBuff, MAXPATHL, TRUE);
6278f9e3e09fSBram Moolenaar 			    smsg(_("Word '%.*s' removed from %s"),
62799ccfebddSBram Moolenaar 							 len, word, NameBuff);
62809ccfebddSBram Moolenaar 			}
62819ccfebddSBram Moolenaar 		    }
62822c363a2eSBram Moolenaar 		    if (fseek(fd, fpos_next, SEEK_SET) != 0)
62832c363a2eSBram Moolenaar 		    {
62842c363a2eSBram Moolenaar 			PERROR(_("Seek error in spellfile"));
62852c363a2eSBram Moolenaar 			break;
62862c363a2eSBram Moolenaar 		    }
62879ccfebddSBram Moolenaar 		}
62889ccfebddSBram Moolenaar 	    }
62899ccfebddSBram Moolenaar 	    if (fd != NULL)
62909ccfebddSBram Moolenaar 		fclose(fd);
62919ccfebddSBram Moolenaar 	}
62929ccfebddSBram Moolenaar     }
62939ccfebddSBram Moolenaar 
62949ccfebddSBram Moolenaar     if (!undo)
62959ccfebddSBram Moolenaar     {
62969ccfebddSBram Moolenaar 	fd = mch_fopen((char *)fname, "a");
62979ccfebddSBram Moolenaar 	if (fd == NULL && new_spf)
62989ccfebddSBram Moolenaar 	{
62999ccfebddSBram Moolenaar 	    char_u *p;
63009ccfebddSBram Moolenaar 
63010d6f5d97SBram Moolenaar 	    // We just initialized the 'spellfile' option and can't open the
63020d6f5d97SBram Moolenaar 	    // file.  We may need to create the "spell" directory first.  We
63030d6f5d97SBram Moolenaar 	    // already checked the runtime directory is writable in
63040d6f5d97SBram Moolenaar 	    // init_spellfile().
63059ccfebddSBram Moolenaar 	    if (!dir_of_file_exists(fname) && (p = gettail_sep(fname)) != fname)
63069ccfebddSBram Moolenaar 	    {
63079ccfebddSBram Moolenaar 		int c = *p;
63089ccfebddSBram Moolenaar 
63090d6f5d97SBram Moolenaar 		// The directory doesn't exist.  Try creating it and opening
63100d6f5d97SBram Moolenaar 		// the file again.
63119ccfebddSBram Moolenaar 		*p = NUL;
63129ccfebddSBram Moolenaar 		vim_mkdir(fname, 0755);
63139ccfebddSBram Moolenaar 		*p = c;
63149ccfebddSBram Moolenaar 		fd = mch_fopen((char *)fname, "a");
63159ccfebddSBram Moolenaar 	    }
63169ccfebddSBram Moolenaar 	}
63179ccfebddSBram Moolenaar 
63189ccfebddSBram Moolenaar 	if (fd == NULL)
6319f9e3e09fSBram Moolenaar 	    semsg(_(e_notopen), fname);
63209ccfebddSBram Moolenaar 	else
63219ccfebddSBram Moolenaar 	{
632208cc374dSBram Moolenaar 	    if (what == SPELL_ADD_BAD)
63239ccfebddSBram Moolenaar 		fprintf(fd, "%.*s/!\n", len, word);
632408cc374dSBram Moolenaar 	    else if (what == SPELL_ADD_RARE)
632508cc374dSBram Moolenaar 		fprintf(fd, "%.*s/?\n", len, word);
63269ccfebddSBram Moolenaar 	    else
63279ccfebddSBram Moolenaar 		fprintf(fd, "%.*s\n", len, word);
63289ccfebddSBram Moolenaar 	    fclose(fd);
63299ccfebddSBram Moolenaar 
63309ccfebddSBram Moolenaar 	    home_replace(NULL, fname, NameBuff, MAXPATHL, TRUE);
6331f9e3e09fSBram Moolenaar 	    smsg(_("Word '%.*s' added to %s"), len, word, NameBuff);
63329ccfebddSBram Moolenaar 	}
63339ccfebddSBram Moolenaar     }
63349ccfebddSBram Moolenaar 
63359ccfebddSBram Moolenaar     if (fd != NULL)
63369ccfebddSBram Moolenaar     {
63370d6f5d97SBram Moolenaar 	// Update the .add.spl file.
63389ccfebddSBram Moolenaar 	mkspell(1, &fname, FALSE, TRUE, TRUE);
63399ccfebddSBram Moolenaar 
63400d6f5d97SBram Moolenaar 	// If the .add file is edited somewhere, reload it.
63419ccfebddSBram Moolenaar 	if (buf != NULL)
63429ccfebddSBram Moolenaar 	    buf_reload(buf, buf->b_orig_mode);
63439ccfebddSBram Moolenaar 
63449ccfebddSBram Moolenaar 	redraw_all_later(SOME_VALID);
63459ccfebddSBram Moolenaar     }
63469ccfebddSBram Moolenaar     vim_free(fnamebuf);
63479ccfebddSBram Moolenaar }
63489ccfebddSBram Moolenaar 
63499ccfebddSBram Moolenaar /*
63509ccfebddSBram Moolenaar  * Initialize 'spellfile' for the current buffer.
63519ccfebddSBram Moolenaar  */
63529ccfebddSBram Moolenaar     static void
init_spellfile(void)63539ccfebddSBram Moolenaar init_spellfile(void)
63549ccfebddSBram Moolenaar {
63559ccfebddSBram Moolenaar     char_u	*buf;
63569ccfebddSBram Moolenaar     int		l;
63579ccfebddSBram Moolenaar     char_u	*fname;
63589ccfebddSBram Moolenaar     char_u	*rtp;
63599ccfebddSBram Moolenaar     char_u	*lend;
63609ccfebddSBram Moolenaar     int		aspath = FALSE;
63619ccfebddSBram Moolenaar     char_u	*lstart = curbuf->b_s.b_p_spl;
63629ccfebddSBram Moolenaar 
63639ccfebddSBram Moolenaar     if (*curwin->w_s->b_p_spl != NUL && curwin->w_s->b_langp.ga_len > 0)
63649ccfebddSBram Moolenaar     {
63659ccfebddSBram Moolenaar 	buf = alloc(MAXPATHL);
63669ccfebddSBram Moolenaar 	if (buf == NULL)
63679ccfebddSBram Moolenaar 	    return;
63689ccfebddSBram Moolenaar 
63690d6f5d97SBram Moolenaar 	// Find the end of the language name.  Exclude the region.  If there
63700d6f5d97SBram Moolenaar 	// is a path separator remember the start of the tail.
63719ccfebddSBram Moolenaar 	for (lend = curwin->w_s->b_p_spl; *lend != NUL
63729ccfebddSBram Moolenaar 			&& vim_strchr((char_u *)",._", *lend) == NULL; ++lend)
63739ccfebddSBram Moolenaar 	    if (vim_ispathsep(*lend))
63749ccfebddSBram Moolenaar 	    {
63759ccfebddSBram Moolenaar 		aspath = TRUE;
63769ccfebddSBram Moolenaar 		lstart = lend + 1;
63779ccfebddSBram Moolenaar 	    }
63789ccfebddSBram Moolenaar 
63790d6f5d97SBram Moolenaar 	// Loop over all entries in 'runtimepath'.  Use the first one where we
63800d6f5d97SBram Moolenaar 	// are allowed to write.
63819ccfebddSBram Moolenaar 	rtp = p_rtp;
63829ccfebddSBram Moolenaar 	while (*rtp != NUL)
63839ccfebddSBram Moolenaar 	{
63849ccfebddSBram Moolenaar 	    if (aspath)
63850d6f5d97SBram Moolenaar 		// Use directory of an entry with path, e.g., for
63860d6f5d97SBram Moolenaar 		// "/dir/lg.utf-8.spl" use "/dir".
63879ccfebddSBram Moolenaar 		vim_strncpy(buf, curbuf->b_s.b_p_spl,
63889ccfebddSBram Moolenaar 					    lstart - curbuf->b_s.b_p_spl - 1);
63899ccfebddSBram Moolenaar 	    else
63900d6f5d97SBram Moolenaar 		// Copy the path from 'runtimepath' to buf[].
63919ccfebddSBram Moolenaar 		copy_option_part(&rtp, buf, MAXPATHL, ",");
63929ccfebddSBram Moolenaar 	    if (filewritable(buf) == 2)
63939ccfebddSBram Moolenaar 	    {
63940d6f5d97SBram Moolenaar 		// Use the first language name from 'spelllang' and the
63950d6f5d97SBram Moolenaar 		// encoding used in the first loaded .spl file.
63969ccfebddSBram Moolenaar 		if (aspath)
63979ccfebddSBram Moolenaar 		    vim_strncpy(buf, curbuf->b_s.b_p_spl,
63989ccfebddSBram Moolenaar 						  lend - curbuf->b_s.b_p_spl);
63999ccfebddSBram Moolenaar 		else
64009ccfebddSBram Moolenaar 		{
64010d6f5d97SBram Moolenaar 		    // Create the "spell" directory if it doesn't exist yet.
64029ccfebddSBram Moolenaar 		    l = (int)STRLEN(buf);
64039ccfebddSBram Moolenaar 		    vim_snprintf((char *)buf + l, MAXPATHL - l, "/spell");
64049ccfebddSBram Moolenaar 		    if (filewritable(buf) != 2)
64059ccfebddSBram Moolenaar 			vim_mkdir(buf, 0755);
64069ccfebddSBram Moolenaar 
64079ccfebddSBram Moolenaar 		    l = (int)STRLEN(buf);
64089ccfebddSBram Moolenaar 		    vim_snprintf((char *)buf + l, MAXPATHL - l,
64099ccfebddSBram Moolenaar 				 "/%.*s", (int)(lend - lstart), lstart);
64109ccfebddSBram Moolenaar 		}
64119ccfebddSBram Moolenaar 		l = (int)STRLEN(buf);
64129ccfebddSBram Moolenaar 		fname = LANGP_ENTRY(curwin->w_s->b_langp, 0)
64139ccfebddSBram Moolenaar 							 ->lp_slang->sl_fname;
64149ccfebddSBram Moolenaar 		vim_snprintf((char *)buf + l, MAXPATHL - l, ".%s.add",
64159ccfebddSBram Moolenaar 			fname != NULL
64169ccfebddSBram Moolenaar 			  && strstr((char *)gettail(fname), ".ascii.") != NULL
64179ccfebddSBram Moolenaar 				       ? (char_u *)"ascii" : spell_enc());
64189ccfebddSBram Moolenaar 		set_option_value((char_u *)"spellfile", 0L, buf, OPT_LOCAL);
64199ccfebddSBram Moolenaar 		break;
64209ccfebddSBram Moolenaar 	    }
64219ccfebddSBram Moolenaar 	    aspath = FALSE;
64229ccfebddSBram Moolenaar 	}
64239ccfebddSBram Moolenaar 
64249ccfebddSBram Moolenaar 	vim_free(buf);
64259ccfebddSBram Moolenaar     }
64269ccfebddSBram Moolenaar }
64279ccfebddSBram Moolenaar 
64289ccfebddSBram Moolenaar 
64299ccfebddSBram Moolenaar 
64309ccfebddSBram Moolenaar /*
64319ccfebddSBram Moolenaar  * Set the spell character tables from strings in the affix file.
64329ccfebddSBram Moolenaar  */
64339ccfebddSBram Moolenaar     static int
set_spell_chartab(char_u * fol,char_u * low,char_u * upp)64349ccfebddSBram Moolenaar set_spell_chartab(char_u *fol, char_u *low, char_u *upp)
64359ccfebddSBram Moolenaar {
64360d6f5d97SBram Moolenaar     // We build the new tables here first, so that we can compare with the
64370d6f5d97SBram Moolenaar     // previous one.
64389ccfebddSBram Moolenaar     spelltab_T	new_st;
64399ccfebddSBram Moolenaar     char_u	*pf = fol, *pl = low, *pu = upp;
64409ccfebddSBram Moolenaar     int		f, l, u;
64419ccfebddSBram Moolenaar 
64429ccfebddSBram Moolenaar     clear_spell_chartab(&new_st);
64439ccfebddSBram Moolenaar 
64449ccfebddSBram Moolenaar     while (*pf != NUL)
64459ccfebddSBram Moolenaar     {
64469ccfebddSBram Moolenaar 	if (*pl == NUL || *pu == NUL)
64479ccfebddSBram Moolenaar 	{
6448f9e3e09fSBram Moolenaar 	    emsg(_(e_affform));
64499ccfebddSBram Moolenaar 	    return FAIL;
64509ccfebddSBram Moolenaar 	}
64519ccfebddSBram Moolenaar 	f = mb_ptr2char_adv(&pf);
64529ccfebddSBram Moolenaar 	l = mb_ptr2char_adv(&pl);
64539ccfebddSBram Moolenaar 	u = mb_ptr2char_adv(&pu);
6454264b74faSBram Moolenaar 
64550d6f5d97SBram Moolenaar 	// Every character that appears is a word character.
64569ccfebddSBram Moolenaar 	if (f < 256)
64579ccfebddSBram Moolenaar 	    new_st.st_isw[f] = TRUE;
64589ccfebddSBram Moolenaar 	if (l < 256)
64599ccfebddSBram Moolenaar 	    new_st.st_isw[l] = TRUE;
64609ccfebddSBram Moolenaar 	if (u < 256)
64619ccfebddSBram Moolenaar 	    new_st.st_isw[u] = TRUE;
64629ccfebddSBram Moolenaar 
64630d6f5d97SBram Moolenaar 	// if "LOW" and "FOL" are not the same the "LOW" char needs
64640d6f5d97SBram Moolenaar 	// case-folding
64659ccfebddSBram Moolenaar 	if (l < 256 && l != f)
64669ccfebddSBram Moolenaar 	{
64679ccfebddSBram Moolenaar 	    if (f >= 256)
64689ccfebddSBram Moolenaar 	    {
6469f9e3e09fSBram Moolenaar 		emsg(_(e_affrange));
64709ccfebddSBram Moolenaar 		return FAIL;
64719ccfebddSBram Moolenaar 	    }
64729ccfebddSBram Moolenaar 	    new_st.st_fold[l] = f;
64739ccfebddSBram Moolenaar 	}
64749ccfebddSBram Moolenaar 
64750d6f5d97SBram Moolenaar 	// if "UPP" and "FOL" are not the same the "UPP" char needs
64760d6f5d97SBram Moolenaar 	// case-folding, it's upper case and the "UPP" is the upper case of
64770d6f5d97SBram Moolenaar 	// "FOL" .
64789ccfebddSBram Moolenaar 	if (u < 256 && u != f)
64799ccfebddSBram Moolenaar 	{
64809ccfebddSBram Moolenaar 	    if (f >= 256)
64819ccfebddSBram Moolenaar 	    {
6482f9e3e09fSBram Moolenaar 		emsg(_(e_affrange));
64839ccfebddSBram Moolenaar 		return FAIL;
64849ccfebddSBram Moolenaar 	    }
64859ccfebddSBram Moolenaar 	    new_st.st_fold[u] = f;
64869ccfebddSBram Moolenaar 	    new_st.st_isu[u] = TRUE;
64879ccfebddSBram Moolenaar 	    new_st.st_upper[f] = u;
64889ccfebddSBram Moolenaar 	}
64899ccfebddSBram Moolenaar     }
64909ccfebddSBram Moolenaar 
64919ccfebddSBram Moolenaar     if (*pl != NUL || *pu != NUL)
64929ccfebddSBram Moolenaar     {
6493f9e3e09fSBram Moolenaar 	emsg(_(e_affform));
64949ccfebddSBram Moolenaar 	return FAIL;
64959ccfebddSBram Moolenaar     }
64969ccfebddSBram Moolenaar 
64979ccfebddSBram Moolenaar     return set_spell_finish(&new_st);
64989ccfebddSBram Moolenaar }
64999ccfebddSBram Moolenaar 
65009ccfebddSBram Moolenaar /*
65019ccfebddSBram Moolenaar  * Set the spell character tables from strings in the .spl file.
65029ccfebddSBram Moolenaar  */
65039ccfebddSBram Moolenaar     static void
set_spell_charflags(char_u * flags,int cnt,char_u * fol)65049ccfebddSBram Moolenaar set_spell_charflags(
65059ccfebddSBram Moolenaar     char_u	*flags,
65060d6f5d97SBram Moolenaar     int		cnt,	    // length of "flags"
65079ccfebddSBram Moolenaar     char_u	*fol)
65089ccfebddSBram Moolenaar {
65090d6f5d97SBram Moolenaar     // We build the new tables here first, so that we can compare with the
65100d6f5d97SBram Moolenaar     // previous one.
65119ccfebddSBram Moolenaar     spelltab_T	new_st;
65129ccfebddSBram Moolenaar     int		i;
65139ccfebddSBram Moolenaar     char_u	*p = fol;
65149ccfebddSBram Moolenaar     int		c;
65159ccfebddSBram Moolenaar 
65169ccfebddSBram Moolenaar     clear_spell_chartab(&new_st);
65179ccfebddSBram Moolenaar 
65189ccfebddSBram Moolenaar     for (i = 0; i < 128; ++i)
65199ccfebddSBram Moolenaar     {
65209ccfebddSBram Moolenaar 	if (i < cnt)
65219ccfebddSBram Moolenaar 	{
65229ccfebddSBram Moolenaar 	    new_st.st_isw[i + 128] = (flags[i] & CF_WORD) != 0;
65239ccfebddSBram Moolenaar 	    new_st.st_isu[i + 128] = (flags[i] & CF_UPPER) != 0;
65249ccfebddSBram Moolenaar 	}
65259ccfebddSBram Moolenaar 
65269ccfebddSBram Moolenaar 	if (*p != NUL)
65279ccfebddSBram Moolenaar 	{
65289ccfebddSBram Moolenaar 	    c = mb_ptr2char_adv(&p);
65299ccfebddSBram Moolenaar 	    new_st.st_fold[i + 128] = c;
65309ccfebddSBram Moolenaar 	    if (i + 128 != c && new_st.st_isu[i + 128] && c < 256)
65319ccfebddSBram Moolenaar 		new_st.st_upper[c] = i + 128;
65329ccfebddSBram Moolenaar 	}
65339ccfebddSBram Moolenaar     }
65349ccfebddSBram Moolenaar 
65359ccfebddSBram Moolenaar     (void)set_spell_finish(&new_st);
65369ccfebddSBram Moolenaar }
65379ccfebddSBram Moolenaar 
65389ccfebddSBram Moolenaar     static int
set_spell_finish(spelltab_T * new_st)65399ccfebddSBram Moolenaar set_spell_finish(spelltab_T *new_st)
65409ccfebddSBram Moolenaar {
65419ccfebddSBram Moolenaar     int		i;
65429ccfebddSBram Moolenaar 
65439ccfebddSBram Moolenaar     if (did_set_spelltab)
65449ccfebddSBram Moolenaar     {
65450d6f5d97SBram Moolenaar 	// check that it's the same table
65469ccfebddSBram Moolenaar 	for (i = 0; i < 256; ++i)
65479ccfebddSBram Moolenaar 	{
65489ccfebddSBram Moolenaar 	    if (spelltab.st_isw[i] != new_st->st_isw[i]
65499ccfebddSBram Moolenaar 		    || spelltab.st_isu[i] != new_st->st_isu[i]
65509ccfebddSBram Moolenaar 		    || spelltab.st_fold[i] != new_st->st_fold[i]
65519ccfebddSBram Moolenaar 		    || spelltab.st_upper[i] != new_st->st_upper[i])
65529ccfebddSBram Moolenaar 	    {
6553f9e3e09fSBram Moolenaar 		emsg(_("E763: Word characters differ between spell files"));
65549ccfebddSBram Moolenaar 		return FAIL;
65559ccfebddSBram Moolenaar 	    }
65569ccfebddSBram Moolenaar 	}
65579ccfebddSBram Moolenaar     }
65589ccfebddSBram Moolenaar     else
65599ccfebddSBram Moolenaar     {
65600d6f5d97SBram Moolenaar 	// copy the new spelltab into the one being used
65619ccfebddSBram Moolenaar 	spelltab = *new_st;
65629ccfebddSBram Moolenaar 	did_set_spelltab = TRUE;
65639ccfebddSBram Moolenaar     }
65649ccfebddSBram Moolenaar 
65659ccfebddSBram Moolenaar     return OK;
65669ccfebddSBram Moolenaar }
65679ccfebddSBram Moolenaar 
65689ccfebddSBram Moolenaar /*
65699ccfebddSBram Moolenaar  * Write the table with prefix conditions to the .spl file.
65709ccfebddSBram Moolenaar  * When "fd" is NULL only count the length of what is written.
65719ccfebddSBram Moolenaar  */
65729ccfebddSBram Moolenaar     static int
write_spell_prefcond(FILE * fd,garray_T * gap)65739ccfebddSBram Moolenaar write_spell_prefcond(FILE *fd, garray_T *gap)
65749ccfebddSBram Moolenaar {
65759ccfebddSBram Moolenaar     int		i;
65769ccfebddSBram Moolenaar     char_u	*p;
65779ccfebddSBram Moolenaar     int		len;
65789ccfebddSBram Moolenaar     int		totlen;
65790d6f5d97SBram Moolenaar     size_t	x = 1;  // collect return value of fwrite()
65809ccfebddSBram Moolenaar 
65819ccfebddSBram Moolenaar     if (fd != NULL)
65820d6f5d97SBram Moolenaar 	put_bytes(fd, (long_u)gap->ga_len, 2);	    // <prefcondcnt>
65839ccfebddSBram Moolenaar 
65840d6f5d97SBram Moolenaar     totlen = 2 + gap->ga_len; // length of <prefcondcnt> and <condlen> bytes
65859ccfebddSBram Moolenaar 
65869ccfebddSBram Moolenaar     for (i = 0; i < gap->ga_len; ++i)
65879ccfebddSBram Moolenaar     {
65880d6f5d97SBram Moolenaar 	// <prefcond> : <condlen> <condstr>
65899ccfebddSBram Moolenaar 	p = ((char_u **)gap->ga_data)[i];
65909ccfebddSBram Moolenaar 	if (p != NULL)
65919ccfebddSBram Moolenaar 	{
65929ccfebddSBram Moolenaar 	    len = (int)STRLEN(p);
65939ccfebddSBram Moolenaar 	    if (fd != NULL)
65949ccfebddSBram Moolenaar 	    {
65959ccfebddSBram Moolenaar 		fputc(len, fd);
65969ccfebddSBram Moolenaar 		x &= fwrite(p, (size_t)len, (size_t)1, fd);
65979ccfebddSBram Moolenaar 	    }
65989ccfebddSBram Moolenaar 	    totlen += len;
65999ccfebddSBram Moolenaar 	}
66009ccfebddSBram Moolenaar 	else if (fd != NULL)
66019ccfebddSBram Moolenaar 	    fputc(0, fd);
66029ccfebddSBram Moolenaar     }
66039ccfebddSBram Moolenaar 
66049ccfebddSBram Moolenaar     return totlen;
66059ccfebddSBram Moolenaar }
66069ccfebddSBram Moolenaar 
66079ccfebddSBram Moolenaar 
66089ccfebddSBram Moolenaar /*
66099ccfebddSBram Moolenaar  * Use map string "map" for languages "lp".
66109ccfebddSBram Moolenaar  */
66119ccfebddSBram Moolenaar     static void
set_map_str(slang_T * lp,char_u * map)66129ccfebddSBram Moolenaar set_map_str(slang_T *lp, char_u *map)
66139ccfebddSBram Moolenaar {
66149ccfebddSBram Moolenaar     char_u	*p;
66159ccfebddSBram Moolenaar     int		headc = 0;
66169ccfebddSBram Moolenaar     int		c;
66179ccfebddSBram Moolenaar     int		i;
66189ccfebddSBram Moolenaar 
66199ccfebddSBram Moolenaar     if (*map == NUL)
66209ccfebddSBram Moolenaar     {
66219ccfebddSBram Moolenaar 	lp->sl_has_map = FALSE;
66229ccfebddSBram Moolenaar 	return;
66239ccfebddSBram Moolenaar     }
66249ccfebddSBram Moolenaar     lp->sl_has_map = TRUE;
66259ccfebddSBram Moolenaar 
66260d6f5d97SBram Moolenaar     // Init the array and hash tables empty.
66279ccfebddSBram Moolenaar     for (i = 0; i < 256; ++i)
66289ccfebddSBram Moolenaar 	lp->sl_map_array[i] = 0;
66299ccfebddSBram Moolenaar     hash_init(&lp->sl_map_hash);
66309ccfebddSBram Moolenaar 
66319ccfebddSBram Moolenaar     /*
66329ccfebddSBram Moolenaar      * The similar characters are stored separated with slashes:
66339ccfebddSBram Moolenaar      * "aaa/bbb/ccc/".  Fill sl_map_array[c] with the character before c and
66349ccfebddSBram Moolenaar      * before the same slash.  For characters above 255 sl_map_hash is used.
66359ccfebddSBram Moolenaar      */
66369ccfebddSBram Moolenaar     for (p = map; *p != NUL; )
66379ccfebddSBram Moolenaar     {
66389ccfebddSBram Moolenaar 	c = mb_cptr2char_adv(&p);
66399ccfebddSBram Moolenaar 	if (c == '/')
66409ccfebddSBram Moolenaar 	    headc = 0;
66419ccfebddSBram Moolenaar 	else
66429ccfebddSBram Moolenaar 	{
66439ccfebddSBram Moolenaar 	    if (headc == 0)
66449ccfebddSBram Moolenaar 		 headc = c;
66459ccfebddSBram Moolenaar 
66460d6f5d97SBram Moolenaar 	    // Characters above 255 don't fit in sl_map_array[], put them in
66470d6f5d97SBram Moolenaar 	    // the hash table.  Each entry is the char, a NUL the headchar and
66480d6f5d97SBram Moolenaar 	    // a NUL.
66499ccfebddSBram Moolenaar 	    if (c >= 256)
66509ccfebddSBram Moolenaar 	    {
66519ccfebddSBram Moolenaar 		int	    cl = mb_char2len(c);
66529ccfebddSBram Moolenaar 		int	    headcl = mb_char2len(headc);
66539ccfebddSBram Moolenaar 		char_u	    *b;
66549ccfebddSBram Moolenaar 		hash_T	    hash;
66559ccfebddSBram Moolenaar 		hashitem_T  *hi;
66569ccfebddSBram Moolenaar 
6657964b3746SBram Moolenaar 		b = alloc(cl + headcl + 2);
66589ccfebddSBram Moolenaar 		if (b == NULL)
66599ccfebddSBram Moolenaar 		    return;
66609ccfebddSBram Moolenaar 		mb_char2bytes(c, b);
66619ccfebddSBram Moolenaar 		b[cl] = NUL;
66629ccfebddSBram Moolenaar 		mb_char2bytes(headc, b + cl + 1);
66639ccfebddSBram Moolenaar 		b[cl + 1 + headcl] = NUL;
66649ccfebddSBram Moolenaar 		hash = hash_hash(b);
66659ccfebddSBram Moolenaar 		hi = hash_lookup(&lp->sl_map_hash, b, hash);
66669ccfebddSBram Moolenaar 		if (HASHITEM_EMPTY(hi))
66679ccfebddSBram Moolenaar 		    hash_add_item(&lp->sl_map_hash, hi, b, hash);
66689ccfebddSBram Moolenaar 		else
66699ccfebddSBram Moolenaar 		{
66700d6f5d97SBram Moolenaar 		    // This should have been checked when generating the .spl
66710d6f5d97SBram Moolenaar 		    // file.
6672f9e3e09fSBram Moolenaar 		    emsg(_("E783: duplicate char in MAP entry"));
66739ccfebddSBram Moolenaar 		    vim_free(b);
66749ccfebddSBram Moolenaar 		}
66759ccfebddSBram Moolenaar 	    }
66769ccfebddSBram Moolenaar 	    else
66779ccfebddSBram Moolenaar 		lp->sl_map_array[c] = headc;
66789ccfebddSBram Moolenaar 	}
66799ccfebddSBram Moolenaar     }
66809ccfebddSBram Moolenaar }
66819ccfebddSBram Moolenaar 
66820d6f5d97SBram Moolenaar #endif  // FEAT_SPELL
6683