9a81ab4222
This change moves prune-trace's --variants-exclude / --benchmarks-exclude capabilities to Database::get_variants() to make it available to all users. Change-Id: Icbc6bb1a3ae7c846d2de40b881f47a9cc1ed7bbf
90 lines
2.8 KiB
C++
90 lines
2.8 KiB
C++
#include <sstream>
|
|
#include <iostream>
|
|
#include "util/Logger.hpp"
|
|
|
|
using namespace fail;
|
|
static Logger LOG ("Pruner");
|
|
|
|
#include "Pruner.hpp"
|
|
|
|
|
|
bool Pruner::init(fail::Database *db,
|
|
const std::vector<std::string>& variants,
|
|
const std::vector<std::string>& variants_exclude,
|
|
const std::vector<std::string>& benchmarks,
|
|
const std::vector<std::string>& benchmarks_exclude)
|
|
{
|
|
this->db = db;
|
|
|
|
m_variants = db->get_variants(
|
|
variants, variants_exclude,
|
|
benchmarks, benchmarks_exclude);
|
|
if (m_variants.size() == 0) {
|
|
LOG << "no variants found, nothing to do" << std::endl;
|
|
return false;
|
|
}
|
|
|
|
std::stringstream ss;
|
|
for (std::vector<fail::Database::Variant>::const_iterator it = m_variants.begin();
|
|
it != m_variants.end(); ++it) {
|
|
if (it != m_variants.begin()) {
|
|
ss << ",";
|
|
}
|
|
ss << it->id;
|
|
}
|
|
m_variants_sql = ss.str();
|
|
|
|
if (!(m_method_id = db->get_fspmethod_id(method_name()))) {
|
|
return false;
|
|
}
|
|
|
|
LOG << "Pruning with method " << method_name() << " (ID: " << m_method_id << ")"
|
|
<< std::endl;
|
|
return true;
|
|
}
|
|
|
|
bool Pruner::create_database() {
|
|
std::string create_statement = "CREATE TABLE IF NOT EXISTS fsppilot ("
|
|
" id int(11) NOT NULL AUTO_INCREMENT,"
|
|
" known_outcome tinyint(4) NOT NULL,"
|
|
" variant_id int(11) NOT NULL,"
|
|
" instr2 int(10) unsigned NOT NULL,"
|
|
" injection_instr int(10) unsigned NOT NULL,"
|
|
" injection_instr_absolute int(10) unsigned,"
|
|
" data_address int(10) unsigned NOT NULL,"
|
|
" data_width int(10) unsigned NOT NULL,"
|
|
" fspmethod_id int(11) NOT NULL,"
|
|
" PRIMARY KEY (id),"
|
|
" KEY fspmethod_id (fspmethod_id,variant_id,data_address,instr2)"
|
|
") engine=MyISAM ";
|
|
bool success = (bool) db->query(create_statement.c_str());
|
|
if (!success) return false;
|
|
|
|
create_statement = "CREATE TABLE IF NOT EXISTS fspgroup ("
|
|
" variant_id int(11) NOT NULL,"
|
|
" instr2 int(11) unsigned NOT NULL,"
|
|
" data_address int(10) unsigned NOT NULL,"
|
|
" fspmethod_id int(11) NOT NULL,"
|
|
" pilot_id int(11) NOT NULL,"
|
|
" PRIMARY KEY (variant_id, data_address, instr2, fspmethod_id),"
|
|
" KEY joinresults (pilot_id,fspmethod_id)) engine=MyISAM";
|
|
|
|
return db->query(create_statement.c_str());
|
|
}
|
|
|
|
bool Pruner::clear_database() {
|
|
std::stringstream ss;
|
|
ss << "DELETE FROM fsppilot WHERE variant_id IN (" << m_variants_sql
|
|
<< ") AND fspmethod_id = " << m_method_id;
|
|
bool ret = (bool) db->query(ss.str().c_str());
|
|
LOG << "deleted " << db->affected_rows() << " rows from fsppilot table" << std::endl;
|
|
ss.str("");
|
|
|
|
ss << "DELETE FROM fspgroup WHERE variant_id IN (" << m_variants_sql
|
|
<< ") AND fspmethod_id = " << m_method_id;
|
|
ret = ret && (bool) db->query(ss.str().c_str());
|
|
LOG << "deleted " << db->affected_rows() << " rows from fspgroup table" << std::endl;
|
|
|
|
return ret;
|
|
}
|