Minor cleanup
This commit is contained in:
@@ -35,7 +35,7 @@ static Cluster::pointer getCluster(std::string type, std::string value)
|
||||
return cluster;
|
||||
}
|
||||
|
||||
static std::list<std::string> getClustersFromFeature(Feature::Type& feature, double minProb)
|
||||
static std::list<std::string> getClustersFromFeature(boost::property_tree::ptree& feature, double minProb)
|
||||
{
|
||||
struct HighLevelNodeDesc
|
||||
{
|
||||
@@ -181,7 +181,7 @@ HighLevelCluster::handleFilesUpdated(void)
|
||||
std::list<std::string> newClusterNames;
|
||||
if (createTags)
|
||||
{
|
||||
Feature::Type feature;
|
||||
boost::property_tree::ptree feature;
|
||||
if (!Feature::Store::instance().get(UpdaterDboSession(), trackId, "high_level", feature))
|
||||
continue;
|
||||
|
||||
|
||||
@@ -112,7 +112,8 @@ Extractor::getLowLevel(boost::property_tree::ptree& pt, std::string mbid)
|
||||
return false;
|
||||
}
|
||||
|
||||
res.erase("metadata");
|
||||
// Keep metadata to ease debugging
|
||||
// res.erase("metadata");
|
||||
|
||||
pt = res;
|
||||
|
||||
|
||||
@@ -25,8 +25,6 @@
|
||||
|
||||
namespace Feature {
|
||||
|
||||
typedef boost::property_tree::ptree Type;
|
||||
|
||||
class Extractor
|
||||
{
|
||||
public:
|
||||
|
||||
@@ -62,7 +62,7 @@ Store::exists(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::
|
||||
}
|
||||
|
||||
bool
|
||||
Store::get(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type, Type& feature)
|
||||
Store::get(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type, boost::property_tree::ptree& feature)
|
||||
{
|
||||
Wt::Dbo::Transaction transaction(session);
|
||||
|
||||
@@ -94,7 +94,7 @@ Store::get(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::str
|
||||
|
||||
|
||||
bool
|
||||
Store::set(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type, const Type& feature)
|
||||
Store::set(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type, const boost::property_tree::ptree& feature)
|
||||
{
|
||||
Wt::Dbo::Transaction transaction(session);
|
||||
|
||||
|
||||
@@ -20,9 +20,9 @@
|
||||
#pragma once
|
||||
|
||||
#include <boost/filesystem/path.hpp>
|
||||
#include <boost/property_tree/ptree.hpp>
|
||||
|
||||
#include "database/Types.hpp"
|
||||
#include "FeatureExtractor.hpp"
|
||||
|
||||
namespace Feature {
|
||||
|
||||
@@ -35,9 +35,9 @@ class Store
|
||||
static Store& instance();
|
||||
|
||||
bool exists(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type);
|
||||
bool get(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type, Type& feature);
|
||||
bool get(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type, boost::property_tree::ptree& feature);
|
||||
|
||||
bool set(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type, const Type& feature);
|
||||
bool set(Wt::Dbo::Session& session, Database::Track::id_type trackId, std::string type, const boost::property_tree::ptree& feature);
|
||||
|
||||
private:
|
||||
Store();
|
||||
|
||||
+1
-1
@@ -23,9 +23,9 @@
|
||||
#include <Wt/WServer>
|
||||
|
||||
#include "utils/Config.hpp"
|
||||
#include "utils/Logger.hpp"
|
||||
#include "av/AvInfo.hpp"
|
||||
#include "av/AvTranscoder.hpp"
|
||||
#include "utils/Logger.hpp"
|
||||
#include "image/Image.hpp"
|
||||
#include "feature/FeatureExtractor.hpp"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user