1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
/* Copyright (C) 2012 Olga Yakovleva <yakovleva.o.v@gmail.com> */
/* This program is free software: you can redistribute it and/or modify */
/* it under the terms of the GNU General Public License as published by */
/* the Free Software Foundation, either version 2 of the License, or */
/* (at your option) any later version. */
/* This program is distributed in the hope that it will be useful, */
/* but WITHOUT ANY WARRANTY; without even the implied warranty of */
/* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the */
/* GNU General Public License for more details. */
/* You should have received a copy of the GNU General Public License */
/* along with this program. If not, see <https://www.gnu.org/licenses/>. */
#include <stdexcept>
#include <string>
#include <iostream>
#include <fstream>
#include <iterator>
#include "tclap/CmdLine.h"
#include "core/engine.hpp"
#include "core/document.hpp"
using namespace RHVoice;
namespace
{
void output_flags(const item& seg, std::ostream& out)
{
for(const auto& flag: seg.get_relation().get_utterance().get_language().get_ph_flags())
{
if(seg.eval("ph_flag_"+flag, std::string("0")).as<std::string>()=="1")
out << "_" << flag;
}
}
}
int main(int argc,const char* argv[])
{
try
{
TCLAP::CmdLine cmd("Transcribe sentences in SSML document");
TCLAP::UnlabeledValueArg<std::string> inpath_arg("input","input file",true,"text.ssml","infile",cmd);
TCLAP::UnlabeledValueArg<std::string> outpath_arg("output","output file",true,"transcription.txt","outfile",cmd);
TCLAP::ValueArg<std::string> boundary_arg("b","boundary","word boundary marker",false,"","string",cmd);
TCLAP::SwitchArg verbose_switch("v","verbose","Output stress and syllable and word boundaries",cmd,false);
TCLAP::SwitchArg stress_switch("s","stress","Output stress",cmd,false);
TCLAP::SwitchArg flags_switch("f","flags","Output flags",cmd,false);
cmd.parse(argc,argv);
std::shared_ptr<engine> eng(new engine);
std::ifstream f_in(inpath_arg.getValue().c_str());
if(!f_in.is_open())
throw std::runtime_error("Cannot open the input file");
std::istreambuf_iterator<char> text_start(f_in);
std::istreambuf_iterator<char> text_end;
std::unique_ptr<document> doc=document::create_from_ssml(eng,text_start,text_end);
std::ofstream f_out(outpath_arg.getValue().c_str());
if(!f_out.is_open())
throw std::runtime_error("Cannot open the output file");
for(document::iterator it(doc->begin());it!=doc->end();++it)
{
std::unique_ptr<utterance> utt=it->create_utterance(sentence_position_single);
const relation& seg_rel=utt->get_relation("Segment");
for(relation::const_iterator seg_iter(seg_rel.begin());seg_iter!=seg_rel.end();++seg_iter)
{
f_out << seg_iter->get("name");
if((verbose_switch.getValue() || stress_switch.getValue()) && seg_iter->in("SylStructure") && seg_iter->eval("ph_vc").as<std::string>()=="+" && seg_iter->eval("R:SylStructure.parent.stress", std::string("0")).as<std::string>()=="1")
f_out << "1";
if(verbose_switch.getValue() || flags_switch.getValue())
output_flags(*seg_iter, f_out);
f_out << " ";
if(verbose_switch.getValue() && seg_iter->in("SylStructure") && !seg_iter->as("SylStructure").has_next() && seg_iter->as("SylStructure").parent().has_next())
f_out << ". ";
if(verbose_switch.getValue()&&
(seg_iter->in("Transcription"))&&
(!seg_iter->as("Transcription").has_next()))
{
auto pos=seg_iter->as("Transcription").parent().eval("gpos").as<std::string>();
if(pos!="content")
f_out << "(" << pos << ") ";
}
if((!boundary_arg.getValue().empty() || verbose_switch.getValue())&&
(seg_iter->in("Transcription"))&&
(!seg_iter->as("Transcription").has_next())&&
(seg_iter->eval("n.name").as<std::string>()!="pau"))
f_out << (boundary_arg.getValue().empty()?"#":boundary_arg.getValue()) << " ";
}
f_out << std::endl;
}
return 0;
}
catch(const std::exception& e)
{
std::cerr << e.what() << std::endl;
return -1;
}
}
|
