Restored similarity dumps in the similarity tool

2019-11-20 13:47:11 +01:00
parent 6d69fc043b
commit c2ee50ab70
6 changed files with 100 additions and 2 deletions
@@ -3,8 +3,11 @@
 #include <stdexcept>
 #include <stdlib.h>

+#include "database/Artist.hpp"
 #include "database/Db.hpp"
+#include "database/Release.hpp"
 #include "database/Session.hpp"
+#include "database/Track.hpp"
 #include "utils/Config.hpp"
 #include "utils/Service.hpp"
 #include "utils/StreamLogger.hpp"
@@ -48,6 +51,81 @@ int main(int argc, char *argv[])
 		FeaturesSearcher searcher {session, featuresSettings};

 		std::cout << "Classifying tracks DONE" << std::endl;
+
+		const std::vector<Database::IdType> trackIds = std::invoke([&]()
+					{
+						auto transaction {session.createSharedTransaction()};
+						return Database::Track::getAllIds(session);
+					});
+
+		std::cout << "*** Tracks ***" << std::endl;
+		for (Database::IdType trackId : trackIds)
+		{
+			auto trackToString = [&](Database::IdType trackId)
+			{
+				std::string res;
+				auto transaction {session.createSharedTransaction()};
+				Database::Track::pointer track {Database::Track::getById(session, trackId)};
+
+				res += track->getName();
+				auto artists {track->getArtists()};
+				for (auto artist : artists )
+					res += " - " + artist->getName();
+				if (track->getRelease())
+					res += " [" + track->getRelease()->getName() + "]";
+
+				return res;
+			};
+
+			std::cout << "Processing track '" << trackToString(trackId) << std::endl;
+			for (Database::IdType similarTrackId : searcher.getSimilarTracks({trackId}, 3))
+				std::cout << "\t- Similar track '" << trackToString(similarTrackId) << std::endl;
+		}
+
+		const std::vector<Database::IdType> releaseIds = std::invoke([&]()
+					{
+						auto transaction {session.createSharedTransaction()};
+						return Database::Release::getAllIds(session);
+					});
+
+		std::cout << "*** Releases ***" << std::endl;
+		for (Database::IdType releaseId : releaseIds)
+		{
+			auto releaseToString = [&](Database::IdType releaseId)
+			{
+				auto transaction {session.createSharedTransaction()};
+
+				Database::Release::pointer release {Database::Release::getById(session, releaseId)};
+				return release->getName();
+			};
+
+			std::cout << "Processing release '" << releaseToString(releaseId) << "'" << std::endl;
+			for (Database::IdType similarReleaseId : searcher.getSimilarReleases({releaseId}, 3))
+				std::cout << "\t- Similar release '" << releaseToString(similarReleaseId) << "'" << std::endl;
+		}
+
+		const std::vector<Database::IdType> artistIds = std::invoke([&]()
+					{
+						auto transaction {session.createSharedTransaction()};
+						return Database::Artist::getAllIds(session);
+					});
+
+		std::cout << "*** Artists ***" << std::endl;
+		for (Database::IdType artistId : artistIds)
+		{
+			auto artistToString = [&](Database::IdType artistId)
+			{
+				auto transaction {session.createSharedTransaction()};
+
+				Database::Artist::pointer artist {Database::Artist::getById(session, artistId)};
+				return artist->getName();
+			};
+
+			std::cout << "Processing artist '" << artistToString(artistId) << "'" << std::endl;
+			for (Database::IdType similarArtistId : searcher.getSimilarArtists({artistId}, 3))
+				std::cout << "\t- Similar artist '" << artistToString(similarArtistId) << "'" << std::endl;
+		}
+
 	}
 	catch( std::exception& e)
 	{