Skip to main content

cosmic_text/font/
system.rs

1use crate::{Attrs, Font, FontMatchAttrs, HashMap, HashSet, ShapeBuffer};
2use alloc::boxed::Box;
3use alloc::collections::BTreeSet;
4use alloc::string::String;
5use alloc::sync::Arc;
6use alloc::vec::Vec;
7use core::fmt;
8use core::ops::{Deref, DerefMut};
9use fontdb::{FaceInfo, Query, Style};
10use skrifa::raw::{ReadError, TableProvider as _};
11use skrifa::MetadataProvider;
12
13// re-export fontdb and harfrust
14pub use fontdb;
15pub use harfrust;
16
17use super::fallback::{Fallback, Fallbacks, MonospaceFallbackInfo, PlatformFallback};
18
19// The fields are used in the derived Ord implementation for sorting fallback candidates.
20#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
21pub struct FontMatchKey {
22    pub(crate) not_emoji: bool,
23    pub(crate) font_weight_diff: u16,
24    pub(crate) font_stretch_diff: u16,
25    pub(crate) font_style_diff: u8,
26    pub(crate) font_weight: u16,
27    pub(crate) font_stretch: u16,
28    pub(crate) id: fontdb::ID,
29    pub(crate) variable_weight_match: bool,
30}
31
32impl FontMatchKey {
33    fn new(attrs: &Attrs, face: &FaceInfo, db: &fontdb::Database) -> FontMatchKey {
34        // TODO: smarter way of detecting emoji
35        let not_emoji = !face.post_script_name.contains("Emoji");
36        let font_weight_diff = attrs.weight.0.abs_diff(face.weight.0);
37
38        let variable_weight_match = font_weight_diff != 0
39            && db.with_face_data(face.id, |font_data, face_index| {
40                let font_ref = skrifa::FontRef::from_index(font_data, face_index).ok()?;
41                let axis = font_ref.axes().get_by_tag(skrifa::Tag::new(b"wght"))?;
42                let w = attrs.weight.0 as f32;
43                Some(w >= axis.min_value() && w <= axis.max_value())
44            }) == Some(Some(true));
45        let font_weight = face.weight.0;
46        let font_stretch_diff = attrs.stretch.to_number().abs_diff(face.stretch.to_number());
47        let font_stretch = face.stretch.to_number();
48        let font_style_diff = match (attrs.style, face.style) {
49            (Style::Normal, Style::Normal)
50            | (Style::Italic, Style::Italic)
51            | (Style::Oblique, Style::Oblique) => 0,
52            (Style::Italic, Style::Oblique) | (Style::Oblique, Style::Italic) => 1,
53            (Style::Normal, Style::Italic)
54            | (Style::Normal, Style::Oblique)
55            | (Style::Italic, Style::Normal)
56            | (Style::Oblique, Style::Normal) => 2,
57        };
58        let id = face.id;
59        FontMatchKey {
60            not_emoji,
61            font_weight_diff,
62            font_stretch_diff,
63            font_style_diff,
64            font_weight,
65            font_stretch,
66            id,
67            variable_weight_match,
68        }
69    }
70}
71
72struct FontCachedCodepointSupportInfo {
73    supported: Vec<u32>,
74    not_supported: Vec<u32>,
75}
76
77impl FontCachedCodepointSupportInfo {
78    const SUPPORTED_MAX_SZ: usize = 512;
79    const NOT_SUPPORTED_MAX_SZ: usize = 1024;
80
81    fn new() -> Self {
82        Self {
83            supported: Vec::with_capacity(Self::SUPPORTED_MAX_SZ),
84            not_supported: Vec::with_capacity(Self::NOT_SUPPORTED_MAX_SZ),
85        }
86    }
87
88    #[inline(always)]
89    fn unknown_has_codepoint(
90        &mut self,
91        font_codepoints: &[u32],
92        codepoint: u32,
93        supported_insert_pos: usize,
94        not_supported_insert_pos: usize,
95    ) -> bool {
96        let ret = font_codepoints.contains(&codepoint);
97        if ret {
98            // don't bother inserting if we are going to truncate the entry away
99            if supported_insert_pos != Self::SUPPORTED_MAX_SZ {
100                self.supported.insert(supported_insert_pos, codepoint);
101                self.supported.truncate(Self::SUPPORTED_MAX_SZ);
102            }
103        } else {
104            // don't bother inserting if we are going to truncate the entry away
105            if not_supported_insert_pos != Self::NOT_SUPPORTED_MAX_SZ {
106                self.not_supported
107                    .insert(not_supported_insert_pos, codepoint);
108                self.not_supported.truncate(Self::NOT_SUPPORTED_MAX_SZ);
109            }
110        }
111        ret
112    }
113
114    #[inline(always)]
115    fn has_codepoint(&mut self, font_codepoints: &[u32], codepoint: u32) -> bool {
116        match self.supported.binary_search(&codepoint) {
117            Ok(_) => true,
118            Err(supported_insert_pos) => match self.not_supported.binary_search(&codepoint) {
119                Ok(_) => false,
120                Err(not_supported_insert_pos) => self.unknown_has_codepoint(
121                    font_codepoints,
122                    codepoint,
123                    supported_insert_pos,
124                    not_supported_insert_pos,
125                ),
126            },
127        }
128    }
129}
130
131/// Access to the system fonts.
132pub struct FontSystem {
133    /// The locale of the system.
134    locale: String,
135
136    /// The underlying font database.
137    db: fontdb::Database,
138
139    /// Cache for loaded fonts from the database.
140    font_cache: HashMap<(fontdb::ID, fontdb::Weight), Option<Arc<Font>>>,
141
142    /// Sorted unique ID's of all Monospace fonts in DB
143    monospace_font_ids: HashSet<fontdb::ID>,
144
145    /// Sorted unique ID's of all Monospace fonts in DB per script.
146    /// A font may support multiple scripts of course, so the same ID
147    /// may appear in multiple map value vecs.
148    per_script_monospace_font_ids: HashMap<[u8; 4], Vec<fontdb::ID>>,
149
150    /// Cache for font codepoint support info
151    font_codepoint_support_info_cache: HashMap<fontdb::ID, FontCachedCodepointSupportInfo>,
152
153    /// Cache for font matches.
154    font_matches_cache: HashMap<FontMatchAttrs, Arc<Vec<FontMatchKey>>>,
155
156    /// Scratch buffer for shaping and laying out.
157    pub(crate) shape_buffer: ShapeBuffer,
158
159    /// Buffer for use in `FontFallbackIter`.
160    pub(crate) monospace_fallbacks_buffer: BTreeSet<MonospaceFallbackInfo>,
161
162    /// Cache for shaped runs
163    #[cfg(feature = "shape-run-cache")]
164    pub shape_run_cache: crate::ShapeRunCache,
165
166    /// List of fallbacks
167    pub(crate) dyn_fallback: Box<dyn Fallback>,
168
169    /// List of fallbacks
170    pub(crate) fallbacks: Fallbacks,
171}
172
173impl fmt::Debug for FontSystem {
174    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
175        f.debug_struct("FontSystem")
176            .field("locale", &self.locale)
177            .field("db", &self.db)
178            .finish_non_exhaustive()
179    }
180}
181
182impl FontSystem {
183    const FONT_MATCHES_CACHE_SIZE_LIMIT: usize = 256;
184    /// Create a new [`FontSystem`], that allows access to any installed system fonts
185    ///
186    /// # Timing
187    ///
188    /// This function takes some time to run. On the release build, it can take up to a second,
189    /// while debug builds can take up to ten times longer. For this reason, it should only be
190    /// called once, and the resulting [`FontSystem`] should be shared.
191    pub fn new() -> Self {
192        Self::new_with_fonts(core::iter::empty())
193    }
194
195    /// Create a new [`FontSystem`] with a pre-specified set of fonts.
196    pub fn new_with_fonts(fonts: impl IntoIterator<Item = fontdb::Source>) -> Self {
197        let mut db = fontdb::Database::new();
198        Self::load_fonts(&mut db, fonts.into_iter());
199        Self::finish_with_db(db)
200    }
201
202    /// Create a new [`FontSystem`] backed by a persistent on-disk system-font index cache
203    /// at the platform's conventional cache location
204    /// (e.g. `$XDG_CACHE_HOME/cosmic-text/fonts.cache`).
205    ///
206    /// Falls back to a normal, uncached scan if no cache directory can be determined.
207    #[cfg(all(feature = "std", not(target_arch = "wasm32")))]
208    pub fn new_cached() -> Self {
209        Self::new_with_fonts_and_cache(core::iter::empty())
210    }
211
212    /// Returns the default system-font cache file path used by [`FontSystem::new_cached`]
213    /// and [`FontSystem::new_with_fonts_and_cache`]
214    /// (`<cache-dir>/cosmic-text/fonts.cache`, following the platform's conventional cache
215    /// directory).
216    ///
217    /// Returns `None` if no cache directory can be determined from the environment.
218    #[cfg(all(feature = "std", not(target_arch = "wasm32")))]
219    pub fn default_cache_path() -> Option<std::path::PathBuf> {
220        super::cache::default_cache_path()
221    }
222
223    /// Like [`FontSystem::new_with_fonts`], but reads from and writes to a persistent
224    /// system-font index cache at the default platform cache location (see
225    /// [`FontSystem::new_cached`]).
226    ///
227    /// The cache path is resolved automatically; if no cache directory can be determined
228    /// this falls back to a normal, uncached scan.
229    /// The user-provided `fonts` are always loaded fresh and
230    /// are never persisted to the cache
231    ///
232    /// To cache at an explicit location instead, use
233    /// [`FontSystem::new_with_fonts_and_cache_path`].
234    #[cfg(all(feature = "std", not(target_arch = "wasm32")))]
235    pub fn new_with_fonts_and_cache(fonts: impl IntoIterator<Item = fontdb::Source>) -> Self {
236        Self::new_with_fonts_and_cache_inner(fonts, Self::default_cache_path())
237    }
238
239    /// Like [`FontSystem::new_with_fonts_and_cache`], but uses the persistent system-font
240    /// index cache at the explicitly provided `cache_path` rather than the default
241    /// location.
242    #[cfg(all(feature = "std", not(target_arch = "wasm32")))]
243    pub fn new_with_fonts_and_cache_path(
244        fonts: impl IntoIterator<Item = fontdb::Source>,
245        cache_path: std::path::PathBuf,
246    ) -> Self {
247        Self::new_with_fonts_and_cache_inner(fonts, Some(cache_path))
248    }
249
250    #[cfg(all(feature = "std", not(target_arch = "wasm32")))]
251    fn new_with_fonts_and_cache_inner(
252        fonts: impl IntoIterator<Item = fontdb::Source>,
253        cache_path: Option<std::path::PathBuf>,
254    ) -> Self {
255        let mut db = fontdb::Database::new();
256
257        let now = std::time::Instant::now();
258
259        match cache_path {
260            Some(cache_path) => super::cache::load_system_fonts_cached(&mut db, &cache_path),
261            None => db.load_system_fonts(),
262        }
263        for source in fonts {
264            db.load_font_source(source);
265        }
266
267        log::debug!(
268            "Loaded {} font faces in {}ms.",
269            db.len(),
270            now.elapsed().as_millis()
271        );
272
273        Self::finish_with_db(db)
274    }
275
276    /// Apply the default font families and finish constructing the [`FontSystem`] from a
277    /// loaded font database.
278    fn finish_with_db(mut db: fontdb::Database) -> Self {
279        let locale = Self::get_locale();
280        log::debug!("Locale: {locale}");
281
282        //TODO: configurable default fonts
283        db.set_monospace_family("Noto Sans Mono");
284        db.set_sans_serif_family("Open Sans");
285        db.set_serif_family("DejaVu Serif");
286
287        Self::new_with_locale_and_db_and_fallback(locale, db, PlatformFallback)
288    }
289
290    /// Create a new [`FontSystem`] with a pre-specified locale, font database and font fallback list.
291    pub fn new_with_locale_and_db_and_fallback(
292        locale: String,
293        db: fontdb::Database,
294        impl_fallback: impl Fallback + 'static,
295    ) -> Self {
296        let monospace_font_ids = db
297            .faces()
298            .filter(|face_info| {
299                face_info.monospaced && !face_info.post_script_name.contains("Emoji")
300            })
301            .map(|face_info| face_info.id)
302            .collect::<HashSet<_>>();
303
304        let mut per_script_monospace_font_ids: HashMap<[u8; 4], BTreeSet<fontdb::ID>> =
305            HashMap::default();
306
307        if cfg!(feature = "monospace_fallback") {
308            for &id in &monospace_font_ids {
309                db.with_face_data(id, |font_data, face_index| {
310                    let face = skrifa::FontRef::from_index(font_data, face_index)?;
311                    for script in face
312                        .gpos()?
313                        .script_list()?
314                        .script_records()
315                        .iter()
316                        .chain(face.gsub()?.script_list()?.script_records().iter())
317                    {
318                        per_script_monospace_font_ids
319                            .entry(script.script_tag().into_bytes())
320                            .or_default()
321                            .insert(id);
322                    }
323                    Ok::<_, ReadError>(())
324                });
325            }
326        }
327
328        let per_script_monospace_font_ids = per_script_monospace_font_ids
329            .into_iter()
330            .map(|(k, v)| (k, Vec::from_iter(v)))
331            .collect();
332
333        let fallbacks = Fallbacks::new(&impl_fallback, &[], &locale);
334
335        Self {
336            locale,
337            db,
338            monospace_font_ids,
339            per_script_monospace_font_ids,
340            font_cache: HashMap::default(),
341            font_matches_cache: HashMap::default(),
342            font_codepoint_support_info_cache: HashMap::default(),
343            monospace_fallbacks_buffer: BTreeSet::default(),
344            #[cfg(feature = "shape-run-cache")]
345            shape_run_cache: crate::ShapeRunCache::default(),
346            shape_buffer: ShapeBuffer::default(),
347            dyn_fallback: Box::new(impl_fallback),
348            fallbacks,
349        }
350    }
351
352    /// Create a new [`FontSystem`] with a pre-specified locale and font database.
353    pub fn new_with_locale_and_db(locale: String, db: fontdb::Database) -> Self {
354        Self::new_with_locale_and_db_and_fallback(locale, db, PlatformFallback)
355    }
356
357    /// Get the locale.
358    pub fn locale(&self) -> &str {
359        &self.locale
360    }
361
362    /// Get the database.
363    pub const fn db(&self) -> &fontdb::Database {
364        &self.db
365    }
366
367    /// Get a mutable reference to the database.
368    pub fn db_mut(&mut self) -> &mut fontdb::Database {
369        self.font_matches_cache.clear();
370        &mut self.db
371    }
372
373    /// Consume this [`FontSystem`] and return the locale and database.
374    pub fn into_locale_and_db(self) -> (String, fontdb::Database) {
375        (self.locale, self.db)
376    }
377
378    /// Get a font by its ID and weight.
379    pub fn get_font(&mut self, id: fontdb::ID, weight: fontdb::Weight) -> Option<Arc<Font>> {
380        self.font_cache
381            .entry((id, weight))
382            .or_insert_with(|| {
383                #[cfg(feature = "std")]
384                unsafe {
385                    self.db.make_shared_face_data(id);
386                }
387                if let Some(font) = Font::new(&self.db, id, weight) {
388                    Some(Arc::new(font))
389                } else {
390                    log::warn!(
391                        "failed to load font '{}'",
392                        self.db.face(id)?.post_script_name
393                    );
394                    None
395                }
396            })
397            .clone()
398    }
399
400    pub fn is_monospace(&self, id: fontdb::ID) -> bool {
401        self.monospace_font_ids.contains(&id)
402    }
403
404    pub fn get_monospace_ids_for_scripts(
405        &self,
406        scripts: impl Iterator<Item = [u8; 4]>,
407    ) -> Vec<fontdb::ID> {
408        let mut ret = scripts
409            .filter_map(|script| self.per_script_monospace_font_ids.get(&script))
410            .flat_map(|ids| ids.iter().copied())
411            .collect::<Vec<_>>();
412        ret.sort();
413        ret.dedup();
414        ret
415    }
416
417    #[inline(always)]
418    pub fn get_font_supported_codepoints_in_word(
419        &mut self,
420        id: fontdb::ID,
421        weight: fontdb::Weight,
422        word: &str,
423    ) -> Option<usize> {
424        self.get_font(id, weight).map(|font| {
425            let code_points = font.unicode_codepoints();
426            let cache = self
427                .font_codepoint_support_info_cache
428                .entry(id)
429                .or_insert_with(FontCachedCodepointSupportInfo::new);
430            word.chars()
431                .filter(|ch| cache.has_codepoint(code_points, u32::from(*ch)))
432                .count()
433        })
434    }
435
436    pub fn get_font_matches(&mut self, attrs: &Attrs<'_>) -> Arc<Vec<FontMatchKey>> {
437        // Clear the cache first if it reached the size limit
438        if self.font_matches_cache.len() >= Self::FONT_MATCHES_CACHE_SIZE_LIMIT {
439            log::trace!("clear font mache cache");
440            self.font_matches_cache.clear();
441        }
442
443        self.font_matches_cache
444            //TODO: do not create AttrsOwned unless entry does not already exist
445            .entry(attrs.into())
446            .or_insert_with(|| {
447                #[cfg(all(feature = "std", not(target_arch = "wasm32")))]
448                let now = std::time::Instant::now();
449
450                let mut font_match_keys = self
451                    .db
452                    .faces()
453                    .map(|face| FontMatchKey::new(attrs, face, &self.db))
454                    .collect::<Vec<_>>();
455
456                // Sort so we get the keys with weight_offset=0 first
457                font_match_keys.sort();
458
459                // db.query is better than above, but returns just one font
460                let query = Query {
461                    families: &[attrs.family],
462                    weight: attrs.weight,
463                    stretch: attrs.stretch,
464                    style: attrs.style,
465                };
466
467                if let Some(id) = self.db.query(&query) {
468                    if let Some(i) = font_match_keys
469                        .iter()
470                        .enumerate()
471                        .find(|(_i, key)| key.id == id)
472                        .map(|(i, _)| i)
473                    {
474                        // if exists move to front
475                        let match_key = font_match_keys.remove(i);
476                        font_match_keys.insert(0, match_key);
477                    } else if let Some(face) = self.db.face(id) {
478                        // else insert in front
479                        let match_key = FontMatchKey::new(attrs, face, &self.db);
480                        font_match_keys.insert(0, match_key);
481                    } else {
482                        log::error!("Could not get face from db, that should've been there.");
483                    }
484                }
485
486                #[cfg(all(feature = "std", not(target_arch = "wasm32")))]
487                {
488                    let elapsed = now.elapsed();
489                    log::debug!("font matches for {attrs:?} in {elapsed:?}");
490                }
491
492                Arc::new(font_match_keys)
493            })
494            .clone()
495    }
496
497    #[cfg(feature = "std")]
498    fn get_locale() -> String {
499        sys_locale::get_locale().unwrap_or_else(|| {
500            log::warn!("failed to get system locale, falling back to en-US");
501            String::from("en-US")
502        })
503    }
504
505    #[cfg(not(feature = "std"))]
506    fn get_locale() -> String {
507        String::from("en-US")
508    }
509
510    #[cfg(feature = "std")]
511    fn load_fonts(db: &mut fontdb::Database, fonts: impl Iterator<Item = fontdb::Source>) {
512        #[cfg(not(target_arch = "wasm32"))]
513        let now = std::time::Instant::now();
514
515        db.load_system_fonts();
516
517        for source in fonts {
518            db.load_font_source(source);
519        }
520
521        #[cfg(not(target_arch = "wasm32"))]
522        log::debug!(
523            "Parsed {} font faces in {}ms.",
524            db.len(),
525            now.elapsed().as_millis()
526        );
527    }
528
529    #[cfg(not(feature = "std"))]
530    fn load_fonts(db: &mut fontdb::Database, fonts: impl Iterator<Item = fontdb::Source>) {
531        for source in fonts {
532            db.load_font_source(source);
533        }
534    }
535}
536
537/// A value borrowed together with an [`FontSystem`]
538#[derive(Debug)]
539pub struct BorrowedWithFontSystem<'a, T> {
540    pub(crate) inner: &'a mut T,
541    pub(crate) font_system: &'a mut FontSystem,
542}
543
544impl<T> Deref for BorrowedWithFontSystem<'_, T> {
545    type Target = T;
546
547    fn deref(&self) -> &Self::Target {
548        self.inner
549    }
550}
551
552impl<T> DerefMut for BorrowedWithFontSystem<'_, T> {
553    fn deref_mut(&mut self) -> &mut Self::Target {
554        self.inner
555    }
556}