[DB] Experimental debug version of track clustering

This commit is contained in:
emeric
2016-04-28 12:38:30 +02:00
parent 795ce08043
commit 5b00229978
17 changed files with 1038 additions and 68 deletions
+5
View File
@@ -89,6 +89,11 @@ AC_CHECK_LIB( [boost_iostreams],
,
[AC_MSG_ERROR([libboost_iostreams not found!])])
AC_CHECK_LIB( [boost_thread],
[main],
,
[AC_MSG_ERROR([libboost_thread not found!])])
AC_CONFIG_FILES([Makefile
src/Makefile
test/Makefile])
+6 -2
View File
@@ -6,6 +6,8 @@ lms_SOURCES = \
$(srcdir)/av/AvTranscoder.cpp \
$(srcdir)/cover/CoverArtGrabber.cpp \
$(srcdir)/database/Artist.cpp \
$(srcdir)/database/Classification.cpp \
$(srcdir)/database/DatabaseClassifier.cpp \
$(srcdir)/database/DatabaseHandler.cpp \
$(srcdir)/database/DatabaseUpdater.cpp \
$(srcdir)/database/MediaDirectory.cpp \
@@ -16,6 +18,7 @@ lms_SOURCES = \
$(srcdir)/database/Track.cpp \
$(srcdir)/database/User.cpp \
$(srcdir)/database/Video.cpp \
$(srcdir)/feature/FeatureExtractor.cpp \
$(srcdir)/image/Image.cpp \
$(srcdir)/logger/Logger.cpp \
$(srcdir)/metadata/AvFormat.cpp \
@@ -54,7 +57,7 @@ lms_SOURCES = \
$(srcdir)/ui/settings/SettingsMediaDirectoryFormView.cpp \
$(srcdir)/ui/settings/SettingsUserFormView.cpp \
$(srcdir)/ui/settings/SettingsUsers.cpp \
$(srcdir)/utils/Checksum.cpp \
$(srcdir)/utils/Path.cpp \
$(srcdir)/utils/Utils.cpp
if VIDEO
@@ -65,5 +68,6 @@ lms_SOURCES += \
$(srcdir)/ui/video/VideoParametersDialog.cpp
endif
lms_CXXFLAGS=-std=c++11 -Wall -I$(srcdir)/third-party -I$(srcdir)/ui $(MAGICKXX_CFLAGS)
lms_CXXFLAGS=-std=c++11 -Wall -I$(srcdir)/third-party -I$(srcdir)/ui $(MAGICKXX_CFLAGS) -D_REENTRANT -DBOOST_SPIRIT_THREADSAFE
lms_LDADD=$(MAGICKXX_LIBS)
+2 -27
View File
@@ -19,7 +19,7 @@
#include <atomic>
#include <mutex>
#include <boost/tokenizer.hpp>
#include "utils/Path.hpp"
#include "logger/Logger.hpp"
@@ -91,37 +91,12 @@ static std::mutex transcoderMutex;
static boost::filesystem::path avConvPath = boost::filesystem::path();
static std::atomic<size_t> globalId = {0};
static std::string searchPath(std::string filename)
{
std::string path;
path = ::getenv("PATH");
if (path.empty())
throw std::runtime_error("Environment variable PATH not found");
std::string result;
typedef boost::tokenizer<boost::char_separator<char> > tokenizer;
boost::char_separator<char> sep(":");
tokenizer tok(path, sep);
for (tokenizer::iterator it = tok.begin(); it != tok.end(); ++it)
{
boost::filesystem::path p = *it;
p /= filename;
if (!::access(p.c_str(), X_OK))
{
result = p.string();
break;
}
}
return result;
}
void
Transcoder::init()
{
for (std::string execName : execNames)
{
boost::filesystem::path p = searchPath(execName);
boost::filesystem::path p = searchExecPath(execName);
if (!p.empty())
{
avConvPath = p;
+1 -1
View File
@@ -22,7 +22,7 @@
#include <map>
#include <set>
#include "pstreams/pstream.h"
#include <pstreams/pstream.h>
#include <boost/date_time/posix_time/posix_time_types.hpp> //no i/o just types
#include <boost/filesystem/path.hpp>
+11 -3
View File
@@ -25,12 +25,20 @@ namespace Database {
class Track;
class ClassificationData
class Feature
{
public:
typedef Wt::Dbo::ptr<ClassificationData> pointer;
typedef Wt::Dbo::ptr<Feature> pointer;
ClassificationData() {}
Feature() {}
Feature(Wt::Dbo::ptr<Track> track, const std::string& type, const std::string& value);
static pointer create(Wt::Dbo::Session& session, Wt::Dbo::ptr<Track> track, const std::string& type, const std::string& value);
static std::vector<pointer> getByTrack(Wt::Dbo::Session& session, Track::id_type trackId, const std::string& type);
std::string getType(void) const { return _type; }
std::string getValue(void) const { return _value; }
template<class Action>
void persist(Action& a)
+852
View File
@@ -0,0 +1,852 @@
/*
* Copyright (C) 2016 Emeric Poupon
*
* This file is part of LMS.
*
* LMS is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* LMS 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with LMS. If not, see <http://www.gnu.org/licenses/>.
*/
//#define BOOST_SPIRIT_THREADSAFE
#include <boost/property_tree/json_parser.hpp>
#include "logger/Logger.hpp"
#include "feature/FeatureExtractor.hpp"
#include "knnl/neural_net_headers.hpp"
#include "DatabaseClassifier.hpp"
namespace Database {
Classifier::Classifier(Wt::Dbo::SqlConnectionPool& connectionPool)
: _db(connectionPool)
{}
void
Classifier::processTrackUpdate(bool added, Track::id_type trackId)
{
if (!added)
return;
LMS_LOG(DBUPDATER, DEBUG) << "Processing track id " << trackId;
boost::filesystem::path path;
{
Wt::Dbo::Transaction transaction(_db.getSession());
Track::pointer track = Database::Track::getById(_db.getSession(), trackId);
if (!track)
return;
path = track->getPath();
// Remove outdated features
for (auto feature : track->getFeatures())
feature.remove();
}
boost::property_tree::ptree pt;
if (!::Feature::Extractor::getLowLevel(pt, path))
return;
std::ostringstream oss;
boost::property_tree::write_json(oss, pt);
{
Wt::Dbo::Transaction transaction(_db.getSession());
Track::pointer track = Database::Track::getById(_db.getSession(), trackId);
Feature::pointer feature = Feature::create( _db.getSession(), track, "low_level", oss.str());
}
}
typedef std::vector<double> entry_t;
typedef std::vector<entry_t> Entries;
static bool entryAddData(entry_t& entry, const boost::property_tree::ptree& pt, std::size_t nbDimensions)
{
if (pt.empty() && nbDimensions == 1)
{
double value = std::stod(pt.data());
entry.push_back(value);
return true;
}
if (pt.size() == nbDimensions)
{
for (auto it : pt)
{
double value = std::stod(it.second.data());
entry.push_back(value);
}
return true;
}
LMS_LOG(DBUPDATER, DEBUG) << "Bad entry";
return false;
}
struct FeatureDesc
{
std::string lowLevelName;
std::size_t nbDimensions;
double coeff;
};
static std::vector<FeatureDesc> features =
{
// { "lowlevel.average_loudness", 1, 1.0 },
// { "lowlevel.barkbands.dmean", 27, 1.0 },
// { "lowlevel.barkbands.dmean2", 27, 1.0 },
// { "lowlevel.barkbands.dvar", 27, 1.0 },
// { "lowlevel.barkbands.dvar2", 27, 1.0 },
// { "lowlevel.barkbands.max", 27, 1.0 },
// { "lowlevel.barkbands.mean", 27, 1.0 },
// { "lowlevel.barkbands.median", 27, 1.0 },
// { "lowlevel.barkbands.min", 27, 1.0 },
// { "lowlevel.barkbands.var", 27, 1.0 },
// { "lowlevel.barkbands_crest.dmean", 1, 1.0 },
// { "lowlevel.barkbands_crest.dmean2", 1, 1.0 },
// { "lowlevel.barkbands_crest.dvar", 1, 1.0 },
// { "lowlevel.barkbands_crest.dvar2", 1, 1.0 },
// { "lowlevel.barkbands_crest.max", 1, 1.0 },
// { "lowlevel.barkbands_crest.mean", 1, 1.0 },
// { "lowlevel.barkbands_crest.median", 1, 1.0 },
// { "lowlevel.barkbands_crest.min", 1, 1.0 },
// { "lowlevel.barkbands_crest.var", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.dmean", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.dmean2", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.dvar", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.dvar2", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.max", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.mean", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.median", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.min", 1, 1.0 },
// { "lowlevel.barkbands_flatness_db.var", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.dmean", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.dmean2", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.dvar", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.dvar2", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.max", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.mean", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.median", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.min", 1, 1.0 },
// { "lowlevel.barkbands_kurtosis.var", 1, 1.0 },
// { "lowlevel.barkbands_skewness.dmean", 1, 1.0 },
// { "lowlevel.barkbands_skewness.dmean2", 1, 1.0 },
// { "lowlevel.barkbands_skewness.dvar", 1, 1.0 },
// { "lowlevel.barkbands_skewness.dvar2", 1, 1.0 },
// { "lowlevel.barkbands_skewness.max", 1, 1.0 },
// { "lowlevel.barkbands_skewness.mean", 1, 1.0 },
// { "lowlevel.barkbands_skewness.median", 1, 1.0 },
// { "lowlevel.barkbands_skewness.min", 1, 1.0 },
// { "lowlevel.barkbands_skewness.var", 1, 1.0 },
// { "lowlevel.barkbands_spread.dmean", 1, 1.0 },
// { "lowlevel.barkbands_spread.dmean2", 1, 1.0 },
// { "lowlevel.barkbands_spread.dvar", 1, 1.0 },
// { "lowlevel.barkbands_spread.dvar2", 1, 1.0 },
// { "lowlevel.barkbands_spread.max", 1, 1.0 },
// { "lowlevel.barkbands_spread.mean", 1, 1.0 },
// { "lowlevel.barkbands_spread.median", 1, 1.0 },
// { "lowlevel.barkbands_spread.min", 1, 1.0 },
// { "lowlevel.barkbands_spread.var", 1, 1.0 },
// { "lowlevel.dissonance.dmean", 1, 1.0 },
// { "lowlevel.dissonance.dmean2", 1, 1.0 },
// { "lowlevel.dissonance.dvar", 1, 1.0 },
// { "lowlevel.dissonance.dvar2", 1, 1.0 },
// { "lowlevel.dissonance.max", 1, 1.0 },
// { "lowlevel.dissonance.mean", 1, 1.0 },
// { "lowlevel.dissonance.median", 1, 1.0 },
// { "lowlevel.dissonance.min", 1, 1.0 },
// { "lowlevel.dissonance.var", 1, 1.0 },
// { "lowlevel.dynamic_complexity", 1, 1.0 },
// { "lowlevel.erbbands.dmean", 40, 1.0 },
// { "lowlevel.erbbands.dmean2", 40, 1.0 },
// { "lowlevel.erbbands.dvar", 40, 1.0 },
// { "lowlevel.erbbands.dvar2", 40, 1.0 },
// { "lowlevel.erbbands.max", 40, 1.0 },
// { "lowlevel.erbbands.mean", 40, 1.0 },
// { "lowlevel.erbbands.median", 40, 1.0 },
// { "lowlevel.erbbands.min", 40, 1.0 },
// { "lowlevel.erbbands.var", 40, 1.0 },
// { "lowlevel.erbbands_crest.dmean", 1, 1.0 },
// { "lowlevel.erbbands_crest.dmean2", 1, 1.0 },
// { "lowlevel.erbbands_crest.dvar", 1, 1.0 },
// { "lowlevel.erbbands_crest.dvar2", 1, 1.0 },
// { "lowlevel.erbbands_crest.max", 1, 1.0 },
// { "lowlevel.erbbands_crest.mean", 1, 1.0 },
// { "lowlevel.erbbands_crest.median", 1, 1.0 },
// { "lowlevel.erbbands_crest.min", 1, 1.0 },
// { "lowlevel.erbbands_crest.var", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.dmean", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.dmean2", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.dvar", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.dvar2", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.max", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.mean", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.median", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.min", 1, 1.0 },
// { "lowlevel.erbbands_flatness_db.var", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.dmean", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.dmean2", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.dvar", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.dvar2", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.max", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.mean", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.median", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.min", 1, 1.0 },
// { "lowlevel.erbbands_kurtosis.var", 1, 1.0 },
// { "lowlevel.erbbands_skewness.dmean", 1, 1.0 },
// { "lowlevel.erbbands_skewness.dmean2", 1, 1.0 },
// { "lowlevel.erbbands_skewness.dvar", 1, 1.0 },
// { "lowlevel.erbbands_skewness.dvar2", 1, 1.0 },
// { "lowlevel.erbbands_skewness.max", 1, 1.0 },
// { "lowlevel.erbbands_skewness.mean", 1, 1.0 },
// { "lowlevel.erbbands_skewness.median", 1, 1.0 },
// { "lowlevel.erbbands_skewness.min", 1, 1.0 },
// { "lowlevel.erbbands_skewness.var", 1, 1.0 },
// { "lowlevel.erbbands_spread.dmean", 1, 1.0 },
// { "lowlevel.erbbands_spread.dmean2", 1, 1.0 },
// { "lowlevel.erbbands_spread.dvar", 1, 1.0 },
// { "lowlevel.erbbands_spread.dvar2", 1, 1.0 },
// { "lowlevel.erbbands_spread.max", 1, 1.0 },
// { "lowlevel.erbbands_spread.mean", 1, 1.0 },
// { "lowlevel.erbbands_spread.median", 1, 1.0 },
// { "lowlevel.erbbands_spread.min", 1, 1.0 },
// { "lowlevel.erbbands_spread.var", 1, 1.0 },
// { "lowlevel.gfcc.mean", 13, 1.0 },
// { "lowlevel.hfc.dmean", 1, 1.0 },
// { "lowlevel.hfc.dmean2", 1, 1.0 },
// { "lowlevel.hfc.dvar", 1, 1.0 },
// { "lowlevel.hfc.dvar2", 1, 1.0 },
// { "lowlevel.hfc.max", 1, 1.0 },
// { "lowlevel.hfc.mean", 1, 1.0 },
// { "lowlevel.hfc.median", 1, 1.0 },
// { "lowlevel.hfc.min", 1, 1.0 },
// { "lowlevel.hfc.var", 1, 1.0 },
// { "lowlevel.melbands.dmean", 40, 1.0 },
// { "lowlevel.melbands.dmean2", 40, 1.0 },
// { "lowlevel.melbands.dvar", 40, 1.0 },
// { "lowlevel.melbands.dvar2", 40, 1.0 },
// { "lowlevel.melbands.max", 40, 1.0 },
// { "lowlevel.melbands.mean", 40, 1.0 },
// { "lowlevel.melbands.median", 40, 1.0 },
// { "lowlevel.melbands.min", 40, 1.0 },
// { "lowlevel.melbands.var", 40, 1.0 },
// { "lowlevel.melbands_crest.dmean", 1, 1.0 },
// { "lowlevel.melbands_crest.dmean2", 1, 1.0 },
// { "lowlevel.melbands_crest.dvar", 1, 1.0 },
// { "lowlevel.melbands_crest.dvar2", 1, 1.0 },
// { "lowlevel.melbands_crest.max", 1, 1.0 },
// { "lowlevel.melbands_crest.mean", 1, 1.0 },
// { "lowlevel.melbands_crest.median", 1, 1.0 },
// { "lowlevel.melbands_crest.min", 1, 1.0 },
// { "lowlevel.melbands_crest.var", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.dmean", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.dmean2", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.dvar", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.dvar2", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.max", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.mean", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.median", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.min", 1, 1.0 },
// { "lowlevel.melbands_flatness_db.var", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.dmean", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.dmean2", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.dvar", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.dvar2", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.max", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.mean", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.median", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.min", 1, 1.0 },
// { "lowlevel.melbands_kurtosis.var", 1, 1.0 },
// { "lowlevel.melbands_spread.dmean", 1, 1.0 },
// { "lowlevel.melbands_spread.dmean2", 1, 1.0 },
// { "lowlevel.melbands_spread.dvar", 1, 1.0 },
// { "lowlevel.melbands_spread.dvar2", 1, 1.0 },
// { "lowlevel.melbands_spread.max", 1, 1.0 },
// { "lowlevel.melbands_spread.mean", 1, 1.0 },
// { "lowlevel.melbands_spread.median", 1, 1.0 },
// { "lowlevel.melbands_spread.min", 1, 1.0 },
// { "lowlevel.melbands_spread.var", 1, 1.0 },
// { "lowlevel.mfcc.mean", 13, 1.0 },
// { "lowlevel.pitch_salience.dmean", 1, 1.0 },
// { "lowlevel.pitch_salience.dmean2", 1, 1.0 },
// { "lowlevel.pitch_salience.dvar", 1, 1.0 },
// { "lowlevel.pitch_salience.dvar2", 1, 1.0 },
// { "lowlevel.pitch_salience.max", 1, 1.0 },
// { "lowlevel.pitch_salience.mean", 1, 1.0 },
// { "lowlevel.pitch_salience.median", 1, 1.0 },
// { "lowlevel.pitch_salience.min", 1, 1.0 },
// { "lowlevel.pitch_salience.var", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.dmean", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.dmean2", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.dvar", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.dvar2", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.max", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.mean", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.median", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.min", 1, 1.0 },
// { "lowlevel.silence_rate_20dB.var", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.dmean", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.dmean2", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.dvar", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.dvar2", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.max", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.mean", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.median", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.min", 1, 1.0 },
// { "lowlevel.silence_rate_30dB.var", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.dmean", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.dmean2", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.dvar", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.dvar2", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.max", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.mean", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.median", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.min", 1, 1.0 },
// { "lowlevel.silence_rate_60dB.var", 1, 1.0 },
// { "lowlevel.spectral_centroid.dmean", 1, 1.0 },
// { "lowlevel.spectral_centroid.dmean2", 1, 1.0 },
// { "lowlevel.spectral_centroid.dvar", 1, 1.0 },
// { "lowlevel.spectral_centroid.dvar2", 1, 1.0 },
// { "lowlevel.spectral_centroid.max", 1, 1.0 },
{ "lowlevel.spectral_centroid.mean", 1, 1.0 },
// { "lowlevel.spectral_centroid.median", 1, 1.0 },
// { "lowlevel.spectral_centroid.min", 1, 1.0 },
// { "lowlevel.spectral_centroid.var", 1, 1.0 },
// { "lowlevel.spectral_complexity.dmean", 1, 1.0 },
// { "lowlevel.spectral_complexity.dmean2", 1, 1.0 },
// { "lowlevel.spectral_complexity.dvar", 1, 1.0 },
// { "lowlevel.spectral_complexity.dvar2", 1, 1.0 },
// { "lowlevel.spectral_complexity.max", 1, 1.0 },
{ "lowlevel.spectral_complexity.mean", 1, 1.0 },
// { "lowlevel.spectral_complexity.median", 1, 1.0 },
// { "lowlevel.spectral_complexity.min", 1, 1.0 },
// { "lowlevel.spectral_complexity.var", 1, 1.0 },
// { "lowlevel.spectral_decrease.dmean", 1, 1.0 },
// { "lowlevel.spectral_decrease.dmean2", 1, 1.0 },
// { "lowlevel.spectral_decrease.dvar", 1, 1.0 },
// { "lowlevel.spectral_decrease.dvar2", 1, 1.0 },
// { "lowlevel.spectral_decrease.max", 1, 1.0 },
{ "lowlevel.spectral_decrease.mean", 1, 1.0 },
// { "lowlevel.spectral_decrease.median", 1, 1.0 },
// { "lowlevel.spectral_decrease.min", 1, 1.0 },
// { "lowlevel.spectral_decrease.var", 1, 1.0 },
// { "lowlevel.spectral_energy.dmean", 1, 1.0 },
// { "lowlevel.spectral_energy.dmean2", 1, 1.0 },
// { "lowlevel.spectral_energy.dvar", 1, 1.0 },
// { "lowlevel.spectral_energy.dvar2", 1, 1.0 },
// { "lowlevel.spectral_energy.max", 1, 1.0 },
{ "lowlevel.spectral_energy.mean", 1, 1.0 },
// { "lowlevel.spectral_energy.median", 1, 1.0 },
// { "lowlevel.spectral_energy.min", 1, 1.0 },
// { "lowlevel.spectral_energy.var", 1, 1.0 },
// { "lowlevel.spectral_energyband_low.dmean", 1, 1.0 },
// { "lowlevel.spectral_energyband_low.dmean2", 1, 1.0 },
// { "lowlevel.spectral_energyband_low.dvar", 1, 1.0 },
// { "lowlevel.spectral_energyband_low.dvar2", 1, 1.0 },
// { "lowlevel.spectral_energyband_low.max", 1, 1.0 },
{ "lowlevel.spectral_energyband_low.mean", 1, 1.0 },
// { "lowlevel.spectral_energyband_low.median", 1, 1.0 },
// { "lowlevel.spectral_energyband_low.min", 1, 1.0 },
// { "lowlevel.spectral_energyband_low.var", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_high.dmean", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_high.dmean2", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_high.dvar", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_high.dvar2", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_high.max", 1, 1.0 },
{ "lowlevel.spectral_energyband_middle_high.mean", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_high.median", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_high.min", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_high.var", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_low.dmean", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_low.dmean2", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_low.dvar", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_low.dvar2", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_low.max", 1, 1.0 },
{ "lowlevel.spectral_energyband_middle_low.mean", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_low.median", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_low.min", 1, 1.0 },
// { "lowlevel.spectral_energyband_middle_low.var", 1, 1.0 },
// { "lowlevel.spectral_entropy.dmean", 1, 1.0 },
// { "lowlevel.spectral_entropy.dmean2", 1, 1.0 },
// { "lowlevel.spectral_entropy.dvar", 1, 1.0 },
// { "lowlevel.spectral_entropy.dvar2", 1, 1.0 },
// { "lowlevel.spectral_entropy.max", 1, 1.0 },
{ "lowlevel.spectral_entropy.mean", 1, 1.0 },
// { "lowlevel.spectral_entropy.median", 1, 1.0 },
// { "lowlevel.spectral_entropy.min", 1, 1.0 },
// { "lowlevel.spectral_entropy.var", 1, 1.0 },
// { "lowlevel.spectral_flux.dmean", 1, 1.0 },
// { "lowlevel.spectral_flux.dmean2", 1, 1.0 },
// { "lowlevel.spectral_flux.dvar", 1, 1.0 },
// { "lowlevel.spectral_flux.dvar2", 1, 1.0 },
// { "lowlevel.spectral_flux.max", 1, 1.0 },
// { "lowlevel.spectral_flux.mean", 1, 1.0 },
// { "lowlevel.spectral_flux.median", 1, 1.0 },
// { "lowlevel.spectral_flux.min", 1, 1.0 },
// { "lowlevel.spectral_flux.var", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.dmean", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.dmean2", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.dvar", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.dvar2", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.max", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.mean", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.median", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.min", 1, 1.0 },
// { "lowlevel.spectral_kurtosis.var", 1, 1.0 },
// { "lowlevel.spectral_rms.dmean", 1, 1.0 },
// { "lowlevel.spectral_rms.dmean2", 1, 1.0 },
// { "lowlevel.spectral_rms.dvar", 1, 1.0 },
// { "lowlevel.spectral_rms.dvar2", 1, 1.0 },
// { "lowlevel.spectral_rms.max", 1, 1.0 },
{ "lowlevel.spectral_rms.mean", 1, 1.0 },
// { "lowlevel.spectral_rms.median", 1, 1.0 },
// { "lowlevel.spectral_rms.min", 1, 1.0 },
// { "lowlevel.spectral_rms.var", 1, 1.0 },
// { "lowlevel.spectral_rolloff.dmean", 1, 1.0 },
// { "lowlevel.spectral_rolloff.dmean2", 1, 1.0 },
// { "lowlevel.spectral_rolloff.dvar", 1, 1.0 },
// { "lowlevel.spectral_rolloff.dvar2", 1, 1.0 },
// { "lowlevel.spectral_rolloff.max", 1, 1.0 },
// { "lowlevel.spectral_rolloff.mean", 1, 1.0 },
// { "lowlevel.spectral_rolloff.median", 1, 1.0 },
// { "lowlevel.spectral_rolloff.min", 1, 1.0 },
// { "lowlevel.spectral_rolloff.var", 1, 1.0 },
// { "lowlevel.spectral_skewness.dmean", 1, 1.0 },
// { "lowlevel.spectral_skewness.dmean2", 1, 1.0 },
// { "lowlevel.spectral_skewness.dvar", 1, 1.0 },
// { "lowlevel.spectral_skewness.dvar2", 1, 1.0 },
// { "lowlevel.spectral_skewness.max", 1, 1.0 },
// { "lowlevel.spectral_skewness.mean", 1, 1.0 },
// { "lowlevel.spectral_skewness.median", 1, 1.0 },
// { "lowlevel.spectral_skewness.min", 1, 1.0 },
// { "lowlevel.spectral_skewness.var", 1, 1.0 },
// { "lowlevel.spectral_spread.dmean", 1, 1.0 },
// { "lowlevel.spectral_spread.dmean2", 1, 1.0 },
// { "lowlevel.spectral_spread.dvar", 1, 1.0 },
// { "lowlevel.spectral_spread.dvar2", 1, 1.0 },
// { "lowlevel.spectral_spread.max", 1, 1.0 },
{ "lowlevel.spectral_spread.mean", 1, 1.0 },
// { "lowlevel.spectral_spread.median", 1, 1.0 },
// { "lowlevel.spectral_spread.min", 1, 1.0 },
// { "lowlevel.spectral_spread.var", 1, 1.0 },
// { "lowlevel.spectral_strongpeak.dmean", 1, 1.0 },
// { "lowlevel.spectral_strongpeak.dmean2", 1, 1.0 },
// { "lowlevel.spectral_strongpeak.dvar", 1, 1.0 },
// { "lowlevel.spectral_strongpeak.dvar2", 1, 1.0 },
// { "lowlevel.spectral_strongpeak.max", 1, 1.0 },
{ "lowlevel.spectral_strongpeak.mean", 1, 1.0 },
// { "lowlevel.spectral_strongpeak.median", 1, 1.0 },
// { "lowlevel.spectral_strongpeak.min", 1, 1.0 },
// { "lowlevel.spectral_strongpeak.var", 1, 1.0 },
// { "lowlevel.zerocrossingrate.dmean", 1, 1.0 },
// { "lowlevel.zerocrossingrate.dmean2", 1, 1.0 },
// { "lowlevel.zerocrossingrate.dvar", 1, 1.0 },
// { "lowlevel.zerocrossingrate.dvar2", 1, 1.0 },
// { "lowlevel.zerocrossingrate.max", 1, 1.0 },
{ "lowlevel.zerocrossingrate.mean", 1, 1.0 },
// { "lowlevel.zerocrossingrate.median", 1, 1.0 },
// { "lowlevel.zerocrossingrate.min", 1, 1.0 },
// { "lowlevel.zerocrossingrate.var", 1, 1.0 },
// { "rhythm.beats_count", 1, 1.0 },
// { "rhythm.beats_loudness.dmean", 1, 1.0 },
// { "rhythm.beats_loudness.dmean2", 1, 1.0 },
// { "rhythm.beats_loudness.dvar", 1, 1.0 },
// { "rhythm.beats_loudness.dvar2", 1, 1.0 },
// { "rhythm.beats_loudness.max", 1, 1.0 },
{ "rhythm.beats_loudness.mean", 1, 1.0 },
// { "rhythm.beats_loudness.median", 1, 1.0 },
// { "rhythm.beats_loudness.min", 1, 1.0 },
// { "rhythm.beats_loudness.var", 1, 1.0 },
// { "rhythm.bpm", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.dmean", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.dmean2", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.dvar", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.dvar2", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.max", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.mean", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.median", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.min", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_bpm.var", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.dmean", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.dmean2", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.dvar", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.dvar2", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.max", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.mean", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.median", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.min", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_spread.var", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.dmean", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.dmean2", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.dvar", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.dvar2", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.max", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.mean", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.median", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.min", 1, 1.0 },
// { "rhythm.bpm_histogram_first_peak_weight.var", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.dmean", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.dmean2", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.dvar", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.dvar2", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.max", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.mean", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.median", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.min", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_bpm.var", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.dmean", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.dmean2", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.dvar", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.dvar2", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.max", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.mean", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.median", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.min", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_spread.var", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.dmean", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.dmean2", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.dvar", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.dvar2", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.max", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.mean", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.median", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.min", 1, 1.0 },
// { "rhythm.bpm_histogram_second_peak_weight.var", 1, 1.0 },
// { "rhythm.danceability", 1, 1.0},
// { "rhythm.onset_rate", 1, 1.0},
//
// { "tonal.chords_changes_rate", 1, 1.0 },
// { "tonal.chords_histogram", 24, 1.0 },
// { "tonal.chords_number_rate", 1, 1.0 },
// { "tonal.hpcp_entropy.dmean", 1, 1.0 },
// { "tonal.hpcp_entropy.dmean2", 1, 1.0 },
// { "tonal.hpcp_entropy.dvar", 1, 1.0 },
// { "tonal.hpcp_entropy.dvar2", 1, 1.0 },
// { "tonal.hpcp_entropy.max", 1, 1.0 },
// { "tonal.hpcp_entropy.mean", 1, 1.0 },
// { "tonal.hpcp_entropy.median", 1, 1.0 },
// { "tonal.hpcp_entropy.min", 1, 1.0 },
// { "tonal.hpcp_entropy.var", 1, 1.0 },
// { "tonal.tuning_frequency", 1, 1.0 },
};
static bool entryConstructFromJSON(entry_t& entry, const std::string& jsonData)
{
std::istringstream iss(jsonData);
boost::property_tree::ptree pt;
boost::property_tree::json_parser::read_json(iss, pt);
for (auto& elem : features)
{
auto child = pt.get_child_optional(elem.lowLevelName);
if (!child)
{
LMS_LOG(DBUPDATER, DEBUG) << "Cannot get '" << elem.lowLevelName << "'";
return false;
}
if (!entryAddData(entry, *child, elem.nbDimensions))
return false;
}
return true;
}
static entry_t computeWeightCoeffs(const Entries& entries)
{
std::vector<FeatureDesc> expandedFeatures;
std::vector<double> userCoeffs;
for (auto& feature : features)
{
for (std::size_t i = 0; i < feature.nbDimensions; ++i)
{
expandedFeatures.push_back( { feature.lowLevelName + std::to_string(i), 1, feature.coeff / (double)feature.nbDimensions} );
}
}
entry_t coeffs;
::neural_net::Ranges<Entries> ranges (*entries.begin());
ranges( entries );
entry_t::const_iterator pos_max = ranges.get_max().begin();
entry_t::const_iterator pos_min = ranges.get_min().begin();
for (std::size_t i = 0; pos_max != ranges.get_max().end(); ++pos_max, ++pos_min, ++i)
{
coeffs.push_back( expandedFeatures.at(i).coeff *
::operators::inverse (
( *pos_max - *pos_min ) * ( *pos_max - *pos_min )
) ); // weight for i-th axis
std::cout << "Feature = " << expandedFeatures.at(i).lowLevelName << std::endl;
std::cout << "min = " << *pos_min << ", max = " << *pos_max << std::endl;
std::cout << "Coeff = " << coeffs.back() << std::endl;
}
return coeffs;
}
template <class T, size_t ROW, size_t COL>
using Matrix = std::array<std::array<T, COL>, ROW>;
void
Classifier::processDatabaseUpdate(Updater::Stats stats)
{
LMS_LOG(DBUPDATER, DEBUG) << "Database complete Called!";
// typedef ::neural_net::Cauchy_function < entry_t::value_type, entry_t::value_type, ::boost::int32_t> CauchyFunction;
typedef ::neural_net::Gauss_function < entry_t::value_type, entry_t::value_type, ::boost::int32_t > GaussFunction;
// typedef ::distance::Euclidean_distance_function < entry_t > EuclideanDistFunction;
typedef ::distance::Weighted_euclidean_distance_function < entry_t, entry_t > WeightedEuclideanDistFunction;
typedef ::neural_net::Basic_neuron < GaussFunction, WeightedEuclideanDistFunction > KohonenNeuron;
typedef ::neural_net::Rectangular_container < KohonenNeuron > KohonenNetwork;
// typedef ::neural_net::Hexagonal_topology < ::boost::int32_t > HexagonalTopology;
typedef ::neural_net::Max_topology < ::boost::int32_t > MaxTopology;
typedef GaussFunction GaussFunctionSpace;
typedef ::neural_net::Gauss_function < ::boost::int32_t, entry_t::value_type, ::boost::int32_t > GaussFunctionNet;
// typedef ::neural_net::Constant_function < entry_t::value_type, entry_t::value_type > ConstFuncSpace;
typedef ::neural_net::Classic_training_weight
<
entry_t,
::boost::int32_t,
GaussFunctionNet,
GaussFunctionSpace,
MaxTopology,
WeightedEuclideanDistFunction,
::boost::int32_t
> ClassicWeight;
typedef ::neural_net::Wtm_classical_training_functional
<
entry_t,
double,
::boost::int32_t,
::boost::int32_t,
ClassicWeight
> WtmTrainingFunc;
typedef ::neural_net::Wtm_training_algorithm
<
KohonenNetwork,
entry_t,
Entries::iterator,
WtmTrainingFunc,
::boost::int32_t
> WtmTrainingAlg;
Entries entries;
LMS_LOG(DBUPDATER, DEBUG) << "Getting track ids";
std::vector<Track::id_type> trackIdsAll;
{
Wt::Dbo::Transaction transaction(_db.getSession());
trackIdsAll = Database::Track::getAllIds(_db.getSession());
}
::std::random_shuffle ( trackIdsAll.begin(), trackIdsAll.end() );
LMS_LOG(DBUPDATER, DEBUG) << "Getting JSON data";
std::vector<Track::id_type> trackIds;
for (Track::id_type trackId : trackIdsAll)
{
std::string jsonData;
{
Wt::Dbo::Transaction transaction(_db.getSession());
std::vector<Feature::pointer> features = Feature::getByTrack(_db.getSession(), trackId, "low_level");
if (features.empty())
{
LMS_LOG(DBUPDATER, DEBUG) << "No JSON data for track " << trackId;;
continue;
}
jsonData = features.front()->getValue();
}
entry_t entry;
if (entryConstructFromJSON(entry, jsonData))
{
trackIds.push_back(trackId);
entries.push_back(entry);
}
else
{
LMS_LOG(DBUPDATER, ERROR) << "Skipping track " << trackId;
continue;
}
// if (entries.size() > 100)
// break;
}
std::cout << "Feature vector dimension = " << entries.front().size() << std::endl;
if (entries.empty())
return;
// TODO compute rows / columns from the DBB
const std::size_t nbRows = 32;
const std::size_t nbColumns = 32;
// CauchyFunction cauchyFunc(2.0, 1);
GaussFunction gaussFunc(2.0, 1 );
LMS_LOG(DBUPDATER, DEBUG) << "Computing coeffs";
entry_t coeffs = computeWeightCoeffs(entries);
LMS_LOG(DBUPDATER, DEBUG) << "Initializing distfunc";
WeightedEuclideanDistFunction weightedEuclideanDistFunc (&coeffs);
// prepare randomization policy
::neural_net::Internal_randomize internalRandomize;
KohonenNetwork network;
// generate networks initialized by data
LMS_LOG(DBUPDATER, DEBUG) << "Generating network...";
::neural_net::generate_kohonen_network(nbRows, nbColumns, gaussFunc, weightedEuclideanDistFunc, entries, network, internalRandomize);
::std::cout << "Network weights:" << ::std::endl;
::neural_net::print_network_weights ( ::std::cout, network );
::std::cout << ::std::endl;
GaussFunctionNet gaussFuncNetwork(10, 1);
GaussFunctionSpace gaussFuncSpace(10, 1);
MaxTopology maxTopology;
// EuclideanDistFunction euclideanDistFunc;
ClassicWeight classicWeight(gaussFuncNetwork, gaussFuncSpace, maxTopology, weightedEuclideanDistFunc);
WtmTrainingFunc trainingFunc(classicWeight, 0.3);
WtmTrainingAlg wtmTrainAlg( trainingFunc );
LMS_LOG(DBUPDATER, DEBUG) << "Training...";
// tricky training
std::size_t nbPass = 20;
for (std::size_t i = 0; i < nbPass; ++i )
{
LMS_LOG(DBUPDATER, DEBUG) << "Training pass " << i << " / " << nbPass;
// train network using data
auto entries_copy = entries;
::std::random_shuffle ( entries_copy.begin(), entries_copy.end() );
wtmTrainAlg(entries_copy.begin(), entries_copy.end(), &network);
// decrease sigma parameter in network will make training proces more sharpen with each epoch,
// but it have to be done slowly :-)
wtmTrainAlg.training_functional.generalized_training_weight.network_function.sigma *= 2.0/3.0;
// wtmTrainAlg.training_functional.generalized_training_weight.network_function.sigma *= 9.0/10.0;
// shuffle data
// ::std::random_shuffle ( entries.begin(), entries.end() );
}
LMS_LOG(DBUPDATER, DEBUG) << "Training DONE...";
::std::cout << "Network weights:" << ::std::endl;
::neural_net::print_network_weights ( ::std::cout, network );
::std::cout << ::std::endl;
struct ClusterEntry
{
Track::id_type trackId;
entry_t entry;
double distance;
};
Matrix<std::vector<ClusterEntry>, nbRows, nbColumns> trackClusters;
for (std::size_t id = 0; id < entries.size(); ++id)
{
std::cout << "Entry " << id << " : " << std::endl;
// Display input vector
for (auto& value : entries[id])
{
std::cout << value << " ";
}
std::cout << std::endl;
auto entry = entries[id];
std::pair<int, int> coordinates = {0,0};
double maxValue = 0.0;
for (std::size_t i = 0; i < network.objects.size(); ++i)
{
for (std::size_t j = 0; j < network.objects[0].size(); ++j)
{
double value = network.objects[i][j]( entry);
if (value > maxValue)
{
coordinates = {i, j};
maxValue = value;
}
}
}
/* ::std::cout << "Network for entry " << id << std::endl;
::neural_net::print_network ( ::std::cout, network, entry );
::std::cout << ::std::endl;
std::cout << "Max is in {" << coordinates.first << ", " << coordinates.second << "}" << std::endl;
*/
trackClusters[coordinates.first][coordinates.second].push_back({trackIds[id], entry, maxValue});
}
for (std::size_t i = 0; i < nbRows; i++)
{
for (std::size_t j = 0; j < nbColumns; j++)
{
std::cout << "Cluster [" << i << "," << j << "] - ";
// Display the neuron for this cluster
for (auto& value : network.objects[i][j].weights)
std::cout << value << " ";
std::cout << std::endl;
for (auto track : trackClusters[i][j])
{
Wt::Dbo::Transaction transaction(_db.getSession());
std::cout << "- " << track.distance;
Track::pointer t = Database::Track::getById(_db.getSession(), track.trackId);
std::cout << " - " << track.trackId << " - " << t->getArtist()->getName() << " - " << t->getName() << " - (" ;
// Display input vector
for (auto& value : track.entry)
std::cout << value << " ";
std::cout << ")" << std::endl;
}
}
}
}
} // namespace Database
+55
View File
@@ -0,0 +1,55 @@
/*
* Copyright (C) 2016 Emeric Poupon
*
* This file is part of LMS.
*
* LMS is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* LMS 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with LMS. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
#include <Wt/Dbo/Dbo>
#include "DatabaseHandler.hpp"
#include "DatabaseUpdater.hpp"
namespace Database {
class ClusterClassifier
{
public:
ClusterClassifier();
private:
};
class Classifier
{
public:
Classifier(Wt::Dbo::SqlConnectionPool& connectionPool);
void processTrackUpdate(bool added, Track::id_type trackId);
void processDatabaseUpdate(Updater::Stats stats);
private:
Database::Handler _db;
};
} // namespace Database
+2 -1
View File
@@ -84,7 +84,7 @@ Handler::Handler(Wt::Dbo::SqlConnectionPool& connectionPool)
_session.mapClass<Database::Genre>("genre");
_session.mapClass<Database::Track>("track");
_session.mapClass<Database::Classification>("classification");
_session.mapClass<Database::ClassificationData>("classification_data");
_session.mapClass<Database::Feature>("feature");
_session.mapClass<Database::Playlist>("playlist");
_session.mapClass<Database::PlaylistEntry>("playlist_entry");
_session.mapClass<Database::Release>("release");
@@ -105,6 +105,7 @@ Handler::Handler(Wt::Dbo::SqlConnectionPool& connectionPool)
_session.execute("CREATE INDEX genre_name_idx ON genre(name)");
_session.execute("CREATE INDEX release_name_idx ON release(name)");
_session.execute("CREATE INDEX track_name_idx ON track(name)");
_session.execute("CREATE INDEX feature_type ON feature(type)");
}
catch(std::exception& e) {
LMS_LOG(DB, ERROR) << "Cannot create tables: " << e.what();
+7 -1
View File
@@ -26,7 +26,7 @@
#include "logger/Logger.hpp"
#include "cover/CoverArtGrabber.hpp"
#include "utils/Utils.hpp"
#include "utils/Checksum.hpp"
#include "utils/Path.hpp"
#include "Types.hpp"
#include "DatabaseUpdater.hpp"
@@ -288,6 +288,8 @@ Updater::process(boost::system::error_code err)
}
scanComplete().emit(stats);
if (_running)
processNextJob();
}
@@ -597,7 +599,11 @@ Updater::processAudioFile( const boost::filesystem::path& file, Stats& stats)
track.modify()->setCoverType( hasCover ? Track::CoverType::Embedded : Track::CoverType::None );
}
Track::id_type trackId = track.id();
transaction.commit();
_sigTrackChanged.emit(true, trackId);
}
+22 -15
View File
@@ -17,8 +17,7 @@
* along with LMS. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef DB_UPDATER_HPP
#define DB_UPDATER_HPP
#pragma once
#include <Wt/WIOService>
#include <Wt/WSignal>
@@ -36,6 +35,18 @@ namespace Database {
class Updater
{
public:
static Updater& instance();
void setConnectionPool(Wt::Dbo::SqlConnectionPool& connectionPool);
void setAudioExtensions(const std::vector<std::string>& extensions);
void setVideoExtensions(const std::vector<std::string>& extensions);
void start();
void stop();
void restart();
struct Stats
{
std::size_t nbSkipped = 0; // no change since last scan
@@ -49,18 +60,12 @@ class Updater
std::size_t nbChanges() const { return nbAdded + nbRemoved + nbModified;}
};
static Updater& instance();
// Emitted when the whole database has been scanned
Wt::Signal<Stats>& scanComplete() { return _sigScanComplete; }
void setConnectionPool(Wt::Dbo::SqlConnectionPool& connectionPool);
void setAudioExtensions(const std::vector<std::string>& extensions);
void setVideoExtensions(const std::vector<std::string>& extensions);
void start();
void stop();
void restart();
Wt::Signal<Stats>& changed() { return _sigChanged; }
// Emitted when a track changed
// true -> added or modified, false -> to be deleted
Wt::Signal<bool, Track::id_type>& trackChanged() { return _sigTrackChanged; }
std::mutex& getMutex(void) { return _mutex; }
@@ -109,7 +114,10 @@ class Updater
bool _running;
Wt::WIOService _ioService;
Wt::Signal<Stats> _sigChanged;
Wt::Signal<Stats> _sigScanComplete;
Wt::Signal<bool, Artist::id_type> _sigArtistChanged;
Wt::Signal<bool, Release::id_type> _sigReleaseChanged;
Wt::Signal<bool, Track::id_type> _sigTrackChanged;
std::mutex _mutex;
boost::asio::deadline_timer _scheduleTimer;
@@ -126,4 +134,3 @@ class Updater
} // Database
#endif
+15
View File
@@ -44,6 +44,13 @@ Track::getAll(Wt::Dbo::Session& session)
return session.find<Track>();
}
std::vector<Track::id_type>
Track::getAllIds(Wt::Dbo::Session& session)
{
Wt::Dbo::collection<Track::id_type> res = session.query<Track::id_type>("SELECT id from track");
return std::vector<Track::id_type>(res.begin(), res.end());
}
void
Track::setGenres(std::vector<Genre::pointer> genres)
{
@@ -109,6 +116,14 @@ Track::getGenres(void) const
return genres;
}
std::vector< Wt::Dbo::ptr<Feature> >
Track::getFeatures(void) const
{
std::vector< Wt::Dbo::ptr<Feature> > features;
std::copy(_features.begin(), _features.end(), std::back_inserter(features));
return features;
}
Wt::Dbo::Query< Track::pointer >
Track::getQuery(Wt::Dbo::Session& session, SearchFilter filter)
{
+9 -6
View File
@@ -30,6 +30,8 @@
#include <Wt/WDateTime>
#include "SearchFilter.hpp"
namespace Database {
@@ -38,7 +40,7 @@ class Release;
class Track;
class PlaylistEntry;
class Classification;
class ClassificationData;
class Feature;
class Genre
{
@@ -109,6 +111,7 @@ class Track
static std::vector<pointer> getByFilter(Wt::Dbo::Session& session, SearchFilter filter, int offset = -1, int size = -1);
static std::vector<pointer> getByFilter(Wt::Dbo::Session& session, SearchFilter filter, int offset, int size, bool &moreResults);
static Wt::Dbo::collection< pointer > getAll(Wt::Dbo::Session& session);
static std::vector<id_type> getAllIds(Wt::Dbo::Session& session);
static std::vector<boost::filesystem::path> getAllPaths(Wt::Dbo::Session& session);
static std::vector<pointer> getMBIDDuplicates(Wt::Dbo::Session& session);
static std::vector<pointer> getChecksumDuplicates(Wt::Dbo::Session& session);
@@ -158,8 +161,8 @@ class Track
void setArtist(Wt::Dbo::ptr<Artist> artist) { _artist = artist; }
void setRelease(Wt::Dbo::ptr<Release> release) { _release = release; }
void setGenres(std::vector<Genre::pointer> genres);
void addClassifification(Wt::Dbo::ptr<Classification> classification);
void addClassifificationData(Wt::Dbo::ptr<ClassificationData> classificationData);
void addClassification(Wt::Dbo::ptr<Classification> classification);
void addFeature(Wt::Dbo::ptr<Feature> feature);
int getTrackNumber(void) const { return _trackNumber; }
int getTotalTrackNumber(void) const { return _totalTrackNumber; }
@@ -180,7 +183,7 @@ class Track
std::vector< Genre::pointer > getGenres(void) const;
bool hasGenre(Genre::pointer genre) const { return _genres.count(genre); }
std::vector< Wt::Dbo::ptr<Classification> > getClassifications(void) const;
std::vector< Wt::Dbo::ptr<ClassificationData> > getClassificationData(void) const;
std::vector< Wt::Dbo::ptr<Feature> > getFeatures(void) const;
template<class Action>
void persist(Action& a)
@@ -205,7 +208,7 @@ class Track
Wt::Dbo::hasMany(a, _genres, Wt::Dbo::ManyToMany, "track_genre", "", Wt::Dbo::OnDeleteCascade);
Wt::Dbo::hasMany(a, _playlistEntries, Wt::Dbo::ManyToOne, "track");
Wt::Dbo::hasMany(a, _classifications, Wt::Dbo::ManyToOne, "track");
Wt::Dbo::hasMany(a, _classificationData, Wt::Dbo::ManyToOne, "track");
Wt::Dbo::hasMany(a, _features, Wt::Dbo::ManyToOne, "track");
}
private:
@@ -237,7 +240,7 @@ class Track
Wt::Dbo::collection< Genre::pointer > _genres; // Genres that are related to this track
Wt::Dbo::collection< Wt::Dbo::ptr<PlaylistEntry> > _playlistEntries;
Wt::Dbo::collection< Wt::Dbo::ptr<Classification> > _classifications;
Wt::Dbo::collection< Wt::Dbo::ptr<ClassificationData> > _classificationData;
Wt::Dbo::collection< Wt::Dbo::ptr<Feature> > _features;
};
+1
View File
@@ -26,6 +26,7 @@ std::string getModuleName(Module mod)
case Module::AV: return "AV";
case Module::COVER: return "COVER";
case Module::DB: return "DB";
case Module::CLASSIFICATION: return "CLASSIFICATION";
case Module::DBUPDATER: return "DB UPDATER";
case Module::MAIN: return "MAIN";
case Module::METADATA: return "METADATA";
+1
View File
@@ -38,6 +38,7 @@ enum class Severity
enum class Module
{
AV,
CLASSIFICATION,
COVER,
DB,
DBUPDATER,
+14 -5
View File
@@ -28,6 +28,8 @@
#include "image/Image.hpp"
#include "database/DatabaseUpdater.hpp"
#include "database/DatabaseClassifier.hpp"
#include "feature/FeatureExtractor.hpp"
#include "ui/LmsApplication.hpp"
@@ -53,18 +55,26 @@ int main(int argc, char* argv[])
Av::AvInit();
Av::Transcoder::init();
Database::Handler::configureAuth();
Feature::Extractor::init();
// Initializing a connection pool to the database that will be shared along services
std::unique_ptr<Wt::Dbo::SqlConnectionPool> connectionPool( Database::Handler::createConnectionPool("/var/lms/lms.db")); // TODO use $datadir from autotools
Database::Updater::instance().setConnectionPool(*connectionPool);
Database::Updater& dbUpdater = Database::Updater::instance();
dbUpdater.setConnectionPool(*connectionPool);
Database::Classifier dbClassifier(*connectionPool);
// Connect the classifier to the update events
dbUpdater.trackChanged().connect(std::bind(&Database::Classifier::processTrackUpdate, &dbClassifier, std::placeholders::_1, std::placeholders::_2));
dbUpdater.scanComplete().connect(std::bind(&Database::Classifier::processDatabaseUpdate, &dbClassifier, std::placeholders::_1));
// bind entry point
server.addEntryPoint(Wt::Application, boost::bind(UserInterface::LmsApplication::create, _1, boost::ref(*connectionPool)));
// Start
LMS_LOG(MAIN, INFO) << "Starting database updater...";
Database::Updater::instance().start();
dbUpdater.start();
LMS_LOG(MAIN, INFO) << "Starting server...";
server.start();
@@ -75,11 +85,10 @@ int main(int argc, char* argv[])
// Stop
LMS_LOG(MAIN, INFO) << "Stopping server...";
Database::Updater::instance().stop();
server.stop();
LMS_LOG(MAIN, INFO) << "Stopping database updater...";
Database::Updater::instance().stop();
dbUpdater.stop();
res = EXIT_SUCCESS;
}
+28 -4
View File
@@ -1,5 +1,5 @@
/*
* Copyright (C) 2013 Emeric Poupon
* Copyright (C) 2016 Emeric Poupon
*
* This file is part of LMS.
*
@@ -21,10 +21,36 @@
#include <stdexcept>
#include <boost/crc.hpp> // for boost::crc_32_type
#include <boost/tokenizer.hpp>
#include "logger/Logger.hpp"
#include "Checksum.hpp"
#include "Path.hpp"
boost::filesystem::path searchExecPath(std::string filename)
{
std::string path;
path = ::getenv("PATH");
if (path.empty())
throw std::runtime_error("Environment variable PATH not found");
std::string result;
typedef boost::tokenizer<boost::char_separator<char> > tokenizer;
boost::char_separator<char> sep(":");
tokenizer tok(path, sep);
for (tokenizer::iterator it = tok.begin(); it != tok.end(); ++it)
{
boost::filesystem::path p = *it;
p /= filename;
if (!::access(p.c_str(), X_OK))
{
result = p.string();
break;
}
}
return result;
}
typedef boost::crc_32_type crc_type;
@@ -60,6 +86,4 @@ void computeCrc(const boost::filesystem::path& p, std::vector<unsigned char>& cr
const unsigned char* data = reinterpret_cast<const unsigned char*>( &checksum );
crc.push_back(data[i]);
}
}
@@ -1,5 +1,6 @@
/*
* Copyright (C) 2013 Emeric Poupon
* Copyright (C) 2016 Emeric Poupon
*
* This file is part of LMS.
*
@@ -17,9 +18,12 @@
* along with LMS. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
#include <string>
#include <vector>
#include <boost/filesystem.hpp>
// TODO move to utils
void computeCrc(const boost::filesystem::path& p, std::vector<unsigned char>& checksum);
boost::filesystem::path searchExecPath(std::string filename);
void computeCrc(const boost::filesystem::path& p, std::vector<unsigned char>& checksum);