Added scorekeeper class to manage scores/weights for platecorners

This commit is contained in:
Matt Hill
2014-10-26 16:43:27 -04:00
parent 41072737c4
commit b448fbfe04
2 changed files with 123 additions and 0 deletions

View File

@@ -0,0 +1,75 @@
/*
* Copyright (c) 2014 New Designs Unlimited, LLC
* Opensource Automated License Plate Recognition [http://www.openalpr.com]
*
* This file is part of OpenAlpr.
*
* OpenAlpr is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License
* version 3 as published by the Free Software Foundation
*
* 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <vector>
#include "scorekeeper.h"
ScoreKeeper::ScoreKeeper() {
}
ScoreKeeper::~ScoreKeeper() {
}
void ScoreKeeper::setScore(std::string weight_id, float score, float weight) {
// Assume that we never set this value twice
weight_ids.push_back(weight_id);
scores.push_back(score);
weights.push_back(weight);
}
float ScoreKeeper::getTotal() {
float score = 0;
for (unsigned int i = 0; i < weights.size(); i++)
{
score += scores[i] * weights[i];
}
return score;
}
void ScoreKeeper::printDebugScores() {
int longest_weight_id = 0;
for (unsigned int i = 0; i < weight_ids.size(); i++)
{
if (weight_ids[i].length() > longest_weight_id)
longest_weight_id = weight_ids[i].length();
}
float total = getTotal();
for (unsigned int i = 0; i < weight_ids.size(); i++)
{
float percent_of_total = (scores[i] * weights[i]) / total * 100;
std::cout << " - " << std::setw(longest_weight_id + 1) << std::left << weight_ids[i] <<
" Weighted Score: " << std::setw(10) << std::left << (scores[i] * weights[i]) <<
" Orig Score: " << std::setw(10) << std::left << scores[i] <<
" (" << percent_of_total << "% of total)" << std::endl;
}
std::cout << "Total: " << total << std::endl;
}