bool SqlDatabaseManager::GetNeighbors(const Model& model,
const string& distance_function_name,
const int num_neighbors,
vector<pair<Model*, double> >* neighbors) const {
if (neighbors == NULL) return false;
Table results;
stringstream query_text;
query_text << "SELECT * FROM get_saved_neighbors('" << model.ModelName()
<< "','" << distance_function_name
<< "'," << num_neighbors << ");";
if (!database_.Query(query_text.str(), &results)) return false;
int distance_column;
if (!results.GetColumnIndex("distance", &distance_column)) return false;
// Turn each database row into a Model.
LoadModelFunctor load_model_functor(results, model_root_);
double distance;
const int num_rows = results.NumRows();
neighbors->reserve(num_rows);
for (int row = 0; row < num_rows; ++row) {
Model* neighbor_model = model_allocator_->Get();
if (!load_model_functor(neighbor_model, row) ||
!results.GetField(distance_column, row, &distance)) return false;
neighbors->push_back(make_pair(neighbor_model, distance));
}
return true;
}
Table* Update::Execute()
{
Table* source = &db->GetTable(tableName);
/*hash_map<string, Field> fields = source->GetFields();
for (hash_map<string, Expression*>::iterator colIter = columns.begin(); colIter != columns.end(); colIter++)
{
Field suitableField = colIter->second->GetSuitableField(colIter->first, fields);
fields[suitableField.name] = suitableField;
}*/
TableDataIterator iter = source->GetIterator(*predicate);
while (iter.NextRow())
{
TableDataRow row = iter.GetFullDataRow();
hash_map<string, Value> fieldsData;
int numFields = source->GetNumFields();
for (int i = 0; i < numFields; i++)
{
fieldsData[source->GetField(i).name] = row[i];
}
for (hash_map<string, Expression*>::iterator colIter = columns.begin(); colIter != columns.end(); colIter++)
{
Value val = colIter->second->Compute(fieldsData);
row[source->GetColumnIndex(colIter->first.c_str())].Set(val);
}
}
return NULL;
}
/*! This is a helper to filter out duplicate code from the GetXXXList functions.
The name_type is assumed to be the name of the table, and the column in the table
is assumed to be "<name_type>_name".*/
static bool NameList(const string& name_type,
const DatabaseConnection& database,
vector<string>* list) {
if (list == NULL) return false;
Table results;
if (!database.Query("SELECT * FROM get_" + name_type + "_names();",
&results)) return false;
const int num_rows = results.NumRows();
list->reserve(num_rows);
int name_column;
if (!results.GetColumnIndex(name_type + "_name", &name_column)) return false;
string name;
for (int row = 0; row < num_rows; ++row) {
if (!results.GetField(name_column, row, &name)) return false;
list->push_back(name);
}
return true;
}
bool SqlDatabaseManager::GetGrasps(const Model& model,
const string& hand_name,
vector<Grasp*>* grasp_list) const {
if (grasp_list == NULL) return false;
Table results;
PROF_START_TIMER(GET_GRASPS_SQL);
if (!database_.Query("SELECT * FROM get_grasps('" + model.ModelName() +
"','" + hand_name + "');", &results))
return false;
PROF_STOP_TIMER(GET_GRASPS_SQL);
// Get the column indices for the columns we care about.
int pregrasp_joints_column, grasp_joints_column,
pregrasp_position_column, grasp_position_column,
grasp_id_column, epsilon_quality_column, volume_quality_column,
grasp_contacts_column, grasp_source_name_column;
PROF_START_TIMER(GET_GRASPS_GETCOLUMN);
if (!results.GetColumnIndex("grasp_pregrasp_joints", &pregrasp_joints_column) ||
!results.GetColumnIndex("grasp_grasp_joints", &grasp_joints_column) ||
!results.GetColumnIndex("grasp_pregrasp_position",
&pregrasp_position_column) ||
!results.GetColumnIndex("grasp_grasp_position", &grasp_position_column) ||
!results.GetColumnIndex("grasp_id", &grasp_id_column) ||
!results.GetColumnIndex("grasp_epsilon_quality", &epsilon_quality_column) ||
!results.GetColumnIndex("grasp_volume_quality", &volume_quality_column) ||
!results.GetColumnIndex("grasp_contacts", &grasp_contacts_column) ||
!results.GetColumnIndex("grasp_source_name", &grasp_source_name_column))
return false;
PROF_STOP_TIMER(GET_GRASPS_GETCOLUMN);
// Turn each database row into a Grasp.
vector<double> pregrasp_joints, grasp_joints,
pregrasp_position, grasp_position, grasp_contacts;
int grasp_id;
string grasp_source_name;
double epsilon_quality, volume_quality;
const int num_rows = results.NumRows();
grasp_list->reserve(num_rows);
for (int row = 0; row < num_rows; ++row) {
grasp_list->push_back(grasp_allocator_->Get());
Grasp& grasp = *(grasp_list->back());
pregrasp_joints.clear();
pregrasp_position.clear();
grasp_joints.clear();
grasp_position.clear();
PROF_START_TIMER(GET_GRASPS_GETFIELD);
if (!results.GetField(pregrasp_joints_column, row, &pregrasp_joints) ||
!results.GetField(grasp_joints_column, row, &grasp_joints) ||
!results.GetField(pregrasp_position_column, row, &pregrasp_position) ||
!results.GetField(grasp_position_column, row, &grasp_position) ||
!results.GetField(grasp_id_column, row, &grasp_id) ||
!results.GetField(epsilon_quality_column, row, &epsilon_quality) ||
!results.GetField(volume_quality_column, row, &volume_quality) ||
!results.GetField(grasp_contacts_column, row, &grasp_contacts) ||
!results.GetField(grasp_source_name_column, row, &grasp_source_name))
return false;
PROF_STOP_TIMER(GET_GRASPS_GETFIELD);
grasp.SetSourceModel(model);
grasp.SetHandName(hand_name);
grasp.SetEpsilonQuality(epsilon_quality);
grasp.SetVolumeQuality(volume_quality);
grasp.SetGraspId(grasp_id);
pregrasp_joints.erase(pregrasp_joints.begin());
pregrasp_position.erase(pregrasp_position.begin());
grasp_joints.erase(grasp_joints.begin());
grasp_position.erase(grasp_position.begin());
grasp.SetGraspParameters(pregrasp_joints,
pregrasp_position,
grasp_joints,
grasp_position);
grasp.SetContacts(grasp_contacts);
grasp.SetPregraspJoints(pregrasp_joints);
grasp.SetPregraspPosition(pregrasp_position);
grasp.SetFinalgraspJoints(grasp_joints);
grasp.SetFinalgraspPosition(grasp_position);
grasp.SetSource(grasp_source_name);
}
return true;
}