SGC_ToneProt/ToneRecognition.praat

   1 #! praat\r
   2 #\r
   3 # Construct all tone patterns and look for the one closest to the given one\r
   4 #\r
   5 \r
   6 procedure FreeToneRecognition pinyin$ test_word$ exclude$ upperRegister freqRange durScale\r
   7     # Clean up input\r
   8     if pinyin$ <> ""\r
   9         pinyin$ = replace_regex$(pinyin$, "^\s*(.+)\s*$", "\1", 1)\r
  10         pinyin$ = replace_regex$(pinyin$, "5", "0", 0)\r
  11     endif\r
  12 \r
  13     referenceFrequency = 300\r
  14     frequencyFactor = referenceFrequency / upperRegister\r
  15 \r
  16     referenceExt$ = "pitch"\r
  17     \r
  18     # Bias Z-normalized value of the distance difference between smallest and correct\r
  19     biasDistance = 0.6\r
  20     \r
  21     # Debugging\r
  22     keepIntermediates = 0\r
  23     debug = 0\r
  24 \r
  25     # Generate reference tones\r
  26     execute ToneScript.praat 'pinyin$' 'upperRegister' 'freqRange' 'durScale' Pitch\r
  27 \r
  28     # Convert input to Pitch\r
  29     if test_word$ <> "" and test_word$ <> "REUSEPITCH"\r
  30         Read from file... 'test_word$'\r
  31         Rename... Source\r
  32     endif\r
  33     if test_word$ <> "REUSEPITCH"\r
  34         select Sound Source\r
  35         noprogress To Pitch (ac)... 0 60 15 yes 0.1 0.45 0.01 0.5 0.3 600\r
  36         # To Pitch... 0.0 60.0 600.0\r
  37         Formula... self*'frequencyFactor'; Normalize Pitch\r
  38         Rename... SourcePitch\r
  39     endif\r
  40     select Pitch SourcePitch\r
  41 \r
  42     countDistance = 0\r
  43     sumDistance = 0\r
  44     sumSqrDistance = 0\r
  45     correctDistance = -1\r
  46 \r
  47     smallestDistance=999999\r
  48     choiceReference$ = "empty"\r
  49     select Table ToneList\r
  50     listLength = Get number of rows\r
  51     for i from 1 to listLength\r
  52         select Table ToneList\r
  53         inFile$ = Get value... 'i' Word\r
  54 \r
  55         if (exclude$ = "" or rindex_regex(inFile$, exclude$) <= 0) and rindex_regex(inFile$, "[\d]") > 0 \r
  56             referenceName$ = inFile$\r
  57             select Pitch 'inFile$'\r
  58             plus Pitch SourcePitch\r
  59             noprogress To DTW... 24 10 yes yes no restriction\r
  60             Rename... DTW'inFile$'\r
  61             distance = Get distance (weighted)\r
  62             \r
  63             countDistance = countDistance + 1\r
  64             sumDistance = sumDistance + distance\r
  65             sumSqrDistance = sumSqrDistance + distance^2\r
  66             \r
  67             if pinyin$ = inFile$\r
  68                 correctDistance = distance\r
  69             endif\r
  70 \r
  71             if debug > 0\r
  72                 # printline 'distance' - 'inFile$'\r
  73             endif\r
  74 \r
  75             if distance < smallestDistance\r
  76                 smallestDistance = distance\r
  77                 choiceReference$ = "'inFile$'"\r
  78             endif\r
  79 \r
  80             # Clean up\r
  81             select DTW DTW'inFile$'\r
  82 \r
  83             if keepIntermediates = 0\r
  84                 Remove\r
  85             endif\r
  86         endif\r
  87     endfor\r
  88     \r
  89     if countDistance > 1\r
  90         meanDistance = sumDistance / countDistance\r
  91         varDistance = (sumSqrDistance - sumDistance^2/countDistance)/(countDistance - 1)\r
  92         stdDistance = sqrt(varDistance)\r
  93         diffDistance = correctDistance - smallestDistance\r
  94         zDistance = diffDistance/stdDistance\r
  95 \r
  96         if debug > 0\r
  97             printline Match: 'pinyin$' <== 'choiceReference$' small='smallestDistance' Z='zDistance'\r
  98         endif\r
  99 \r
 100         if zDistance < biasDistance\r
 101             choiceReference$ = pinyin$\r
 102             smallestDistance = correctDistance\r
 103         endif\r
 104     endif\r
 105     \r
 106 \r
 107     # Clean up\r
 108     for i from 1 to listLength\r
 109         select Table ToneList\r
 110         inFile$ = Get value... 'i' Word\r
 111 \r
 112         if (exclude$ = "" or rindex_regex(inFile$, exclude$) <= 0) and rindex_regex(inFile$, "[\d]") > 0 \r
 113 \r
 114             # Clean up\r
 115             select Pitch 'inFile$'\r
 116             if keepIntermediates = 0\r
 117                 Remove\r
 118             endif\r
 119         endif\r
 120     endfor\r
 121 \r
 122     select Table ToneList\r
 123     if test_word$ <> "" and test_word$ <> "REUSEPITCH"\r
 124         plus Sound Source\r
 125     endif\r
 126     if test_word$ <> "" and test_word$ <> "REUSEPITCH"\r
 127         plus Pitch SourcePitch\r
 128     endif\r
 129     if keepIntermediates = 0\r
 130         Remove\r
 131     endif\r
 132 \r
 133 endproc\r