fix log; add vad rtf

pull/3026/head
Hui Zhang 3 years ago
parent 29fdbb124b
commit 5f8d11034a

@ -29,9 +29,9 @@ LogMessage::LogMessage(const char* file,
bool out_to_file /* = false */)
: level_(level), verbose_(verbose), out_to_file_(out_to_file) {
if (FLAGS_logtostderr == 0) {
stream_ = std::shared_ptr<std::ostream>(&std::cout);
stream_ = static_cast<std::ostream*>(&std::cout);
} else if (FLAGS_logtostderr == 1) {
stream_ = std::shared_ptr<std::ostream>(&std::cerr);
stream_ = static_cast<std::ostream*>(&std::cerr);
} else if (out_to_file_) {
// logfile
lock_.lock();
@ -46,7 +46,7 @@ LogMessage::~LogMessage() {
lock_.unlock();
}
if (level_ == FATAL) {
if (verbose_ && level_ == FATAL) {
std::abort();
}
}
@ -73,30 +73,20 @@ void LogMessage::init(const char* file, int line) {
std::string("log." + proc_name + ".log.FATAL." + tmp + "." + pid);
}
std::ofstream ofs;
thread_local static std::ofstream ofs;
if (level_ == DEBUG) {
stream_ = std::make_shared<std::ofstream>(
s_debug_logfile_.c_str(), std::ios::out | std::ios::app);
// ofs.open(s_debug_logfile_.c_str(), std::ios::out | std::ios::app);
ofs.open(s_debug_logfile_.c_str(), std::ios::out | std::ios::app);
} else if (level_ == INFO) {
// ofs.open(s_warning_logfile_.c_str(), std::ios::out | std::ios::app);
stream_ = std::make_shared<std::ofstream>(
s_warning_logfile_.c_str(), std::ios::out | std::ios::app);
ofs.open(s_info_logfile_.c_str(), std::ios::out | std::ios::app);
} else if (level_ == WARNING) {
// ofs.open(s_warning_logfile_.c_str(), std::ios::out | std::ios::app);
stream_ = std::make_shared<std::ofstream>(
s_warning_logfile_.c_str(), std::ios::out | std::ios::app);
ofs.open(s_warning_logfile_.c_str(), std::ios::out | std::ios::app);
} else if (level_ == ERROR) {
// ofs.open(s_error_logfile_.c_str(), std::ios::out | std::ios::app);
stream_ = std::make_shared<std::ofstream>(
s_error_logfile_.c_str(), std::ios::out | std::ios::app);
ofs.open(s_error_logfile_.c_str(), std::ios::out | std::ios::app);
} else {
// ofs.open(s_fatal_logfile_.c_str(), std::ios::out | std::ios::app);
stream_ = std::make_shared<std::ofstream>(
s_fatal_logfile_.c_str(), std::ios::out | std::ios::app);
ofs.open(s_fatal_logfile_.c_str(), std::ios::out | std::ios::app);
}
// stream_ = &ofs;
stream_ = &ofs;
stream() << tmp << " " << file << " line " << line << "; ";
stream() << std::flush;

@ -67,7 +67,7 @@ class LogMessage {
void init(const char* file, int line);
private:
std::shared_ptr<std::ostream> stream_;
std::ostream* stream_;
Severity level_;
bool verbose_;
bool out_to_file_;

@ -16,6 +16,7 @@
#include <iostream>
#include <vector>
#include "common/base/common.h"
#include "vad/frontend/wav.h"
#include "vad/interface/vad_interface.h"
@ -35,6 +36,8 @@ int main(int argc, char* argv[]) {
std::vector<float> inputWav; // [0, 1]
wav::WavReader wav_reader = wav::WavReader(audio_file);
auto sr = wav_reader.sample_rate();
CHECK(sr == 16000) << " sr is " << sr << " expect 16000";
auto num_samples = wav_reader.num_samples();
inputWav.resize(num_samples);
@ -42,6 +45,7 @@ int main(int argc, char* argv[]) {
inputWav[i] = wav_reader.data()[i] / 32768;
}
ppspeech::Timer timer;
int window_size_samples = PPSVadChunkSizeSamples(handle);
for (int64_t j = 0; j < num_samples; j += window_size_samples) {
auto start = j;
@ -58,6 +62,9 @@ int main(int argc, char* argv[]) {
}
std::cout << std::endl;
std::cout << "RTF=" << timer.Elapsed() / double(num_samples / sr)
<< std::endl;
PPSVadReset(handle);
return 0;

Loading…
Cancel
Save