PortugueseG2p.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.ML.OnnxRuntime;
using OpenUtau.Api;

namespace OpenUtau.Core.G2p {
    public class PortugueseG2p : G2pPack {
        private static readonly string[] graphemes = new string[] {
            "", "", "", "", "-", "a", "b", "c", "d", "e", "f", "g", "h",
            "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t",
            "u", "v", "w", "x", "y", "z", "à", "á", "â", "ã", "ç",
            "è", "é", "ê", "í", "î", "ó", "ô", "õ", "ú", "û", "ü",
        };

        private static readonly string[] phonemes = new string[] {
            "", "", "", "", "E", "J", "L", "O", "R", "S", "X", "Z",
            "a", "a~", "b", "d", "dZ", "e", "e~", "f", "g",
            "i", "i~", "j", "j~", "k", "l", "m", "n", "o", "o~",
            "p", "r", "s", "t", "tS", "u", "u~", "v", "w", "w~", "z",
        };

        private static object lockObj = new object();
        private static Dictionary<string, int> graphemeIndexes;
        private static IG2p dict;
        private static InferenceSession session;
        private static Dictionary<string, string[]> predCache = new Dictionary<string, string[]>();

        public PortugueseG2p() {
            lock (lockObj) {
                if (graphemeIndexes == null) {
                    graphemeIndexes = graphemes
                    .Skip(4)
                    .Select((g, i) => Tuple.Create(g, i))
                    .ToDictionary(t => t.Item1, t => t.Item2 + 4);
                    var tuple = LoadPack(Data.Resources.g2p_pt);
                    dict = tuple.Item1;
                    session = tuple.Item2;
                }
            }
            GraphemeIndexes = graphemeIndexes;
            Phonemes = phonemes;
            Dict = dict;
            Session = session;
            PredCache = predCache;
        }
    }
}