setfontcandidates!

Author: guo-yong-zhi

src/artist.jl

@@ -39,7 +39,8 @@ end
 function fontsof(lang)
     union((listfonts(l) for l in expandlangcode(lang))...)
 end
-function getfonts(lang)
+function getfontcandidates(lang)
+    lang = StopWords.normcode(String(lang))
     if haskey(FontCandidates, lang)
         return FontCandidates[lang]
     else
@@ -50,6 +51,15 @@ function getfonts(lang)
     end
 end
 
+"""
+    setfontcandidates!(lang::AbstractString, str_set)  
+
+Customize font candidates for language `lang`
+"""
+function setfontcandidates!(lang::AbstractString, str_list)
+    FontCandidates[StopWords.normcode(String(lang))] = str_list
+end
+
 Schemes_colorbrewer = filter(s -> occursin("colorbrewer", colorschemes[s].category), collect(keys(colorschemes)))
 Schemes_colorbrewer =  filter(s -> (occursin("Accent", String(s)) 
         || occursin("Dark", String(s))
@@ -343,10 +353,10 @@ end
 randomoutline() = rand((0, 0, 0, rand(2:10)))
 function randomfonts(lang="")
     if rand() < 0.8
-        fonts = rand(getfonts(lang))
+        fonts = rand(getfontcandidates(lang))
         fonts = fonts * rand(WeightCandidates)
     else
-        fonts = rand(getfonts(lang), 2 + floor(Int, 2randexp()))
+        fonts = rand(getfontcandidates(lang), 2 + floor(Int, 2randexp()))
         fonts = [f * rand(WeightCandidates) for f in fonts]
         rand() > 0.5 && (fonts = tuple(fonts...))
     end

test/runtests.jl

@@ -160,4 +160,10 @@ include("test_textprocessing.jl")
     wordcloud(["the"=>1.0, "to"=>0.51, "and"=>0.50,
               "of"=>0.47, "a"=>0.44, "in"=>0.33]) |> generate!
     wordcloud("It's easy to generate word clouds", maxnum=10) |> generate!
+    
+    # font
+    WordCloud.setfontcandidates!("zh", ["CJK"])
+    @test WordCloud.getfontcandidates("zh") == WordCloud.getfontcandidates("zho") == ["CJK"]
+
+    @show keys(WordCloud.FontCandidates)
 end