提交 d53fc743 编写于 作者: C Calvin Miao

Added evaluator factory implementation

上级 e87f0985
......@@ -3,10 +3,13 @@ package(default_visibility = ["//visibility:public"])
cc_library(
name = "evaluator_factory",
hdrs = ["evaluator_factory.h"],
srcs = ["evaluator_factory.cc"],
deps = [
"//modules/prediction/evaluator:evaluator",
"//modules/prediction/proto:prediction_conf_proto",
"//modules/common/util:factory",
"//modules/common:macro",
"@glog//:glog",
]
)
......
/******************************************************************************
* Copyright 2017 The Apollo Authors. All Rights Reserved.
*
* 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 "modules/prediction/evaluator/evaluator_factory.h"
#include "modules/common/log.h"
namespace apollo {
namespace prediction {
EvaluatorFactory::EvaluatorFactory() {}
void EvaluatorFactory::RegisterEvaluator() {
Register(PredictionConf::DEF_EVAL,
[]() -> Evaluator* { return nullptr; });
}
std::unique_ptr<Evaluator> EvaluatorFactory::CreateEvaluator(
const PredictionConf::Eval& eval) {
auto evaluator = CreateObject(eval);
if (!evaluator) {
AERROR << "Failed to create an evaluator with " << eval;
} else {
ADEBUG << "Succeeded in creating an evaluator with " << eval;
}
return evaluator;
}
}
}
\ No newline at end of file
......@@ -24,6 +24,8 @@
#include <memory>
#include "modules/prediction/evaluator/evaluator.h"
#include "modules/prediction/proto/prediction_conf.pb.h"
#include "modules/common/util/factory.h"
#include "modules/common/macro.h"
......@@ -32,6 +34,9 @@
* @brief apollo::prediction
*/
namespace apollo {
namespace prediction {
class EvaluatorFactory
: public apollo::common::util::Factory<PredictionConf::Eval,
Evaluator> {
......@@ -50,6 +55,9 @@ class EvaluatorFactory
private:
DECLARE_SINGLETON(EvaluatorFactory);
}
};
} // namespace prediction
} // namespace apollo
#endif // MODULES_PREDICTION_EVALUATOR_EVALUATOR_FACTORY_H_
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册