85 lines
2.5 KiB
C++
85 lines
2.5 KiB
C++
/*
|
|
* Copyright (C) 2018 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include "lang_id/common/lite_base/compact-logging.h"
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <iostream>
|
|
|
|
#include "lang_id/common/lite_base/compact-logging-raw.h"
|
|
|
|
#ifndef SAFTM_LOGGING_TAG
|
|
|
|
// Tag inserted in the prefix of the generated log messages. The user can
|
|
// override this by defining this macro on the blaze build command-line.
|
|
#define SAFTM_LOGGING_TAG "saftm"
|
|
#endif
|
|
|
|
namespace libtextclassifier3 {
|
|
namespace mobile {
|
|
namespace internal_logging {
|
|
|
|
namespace {
|
|
// Returns pointer to beginning of last /-separated token from file_name.
|
|
// file_name should be a pointer to a zero-terminated array of chars.
|
|
// E.g., "foo/bar.cc" -> "bar.cc", "foo/" -> "", "foo" -> "foo".
|
|
const char *JumpToBasename(const char *file_name) {
|
|
if (file_name == nullptr) {
|
|
return nullptr;
|
|
}
|
|
|
|
// Points to the beginning of the last encountered token.
|
|
const char *last_token_start = file_name;
|
|
while (*file_name != '\0') {
|
|
if (*file_name == '/') {
|
|
// Found token separator. A new (potentially empty) token starts after
|
|
// this position. Notice that if file_name is a valid zero-terminated
|
|
// string, file_name + 1 is a valid pointer (there is at least one char
|
|
// after address file_name, the zero terminator).
|
|
last_token_start = file_name + 1;
|
|
}
|
|
file_name++;
|
|
}
|
|
return last_token_start;
|
|
}
|
|
} // namespace
|
|
|
|
LogMessage::LogMessage(LogSeverity severity, const char *file_name,
|
|
int line_number)
|
|
: severity_(severity) {
|
|
stream_ << JumpToBasename(file_name) << ":" << line_number << ": ";
|
|
}
|
|
|
|
LogMessage::~LogMessage() {
|
|
LogSeverity level = severity_;
|
|
if (level == DFATAL) {
|
|
#ifdef NDEBUG
|
|
level = ERROR;
|
|
#else
|
|
level = FATAL;
|
|
#endif
|
|
}
|
|
LowLevelLogging(level, /* tag = */ SAFTM_LOGGING_TAG, stream_.message);
|
|
if (level == FATAL) {
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
} // namespace internal_logging
|
|
} // namespace mobile
|
|
} // namespace nlp_saft
|