102 lines
2.8 KiB
C++
102 lines
2.8 KiB
C++
#include "AvFormat.hpp"
|
|
|
|
#include <boost/foreach.hpp>
|
|
#include <boost/algorithm/string.hpp>
|
|
#include <boost/date_time/posix_time/posix_time.hpp>
|
|
|
|
#include "av/InputFormatContext.hpp"
|
|
|
|
#include "Utils.hpp"
|
|
|
|
namespace MetaData
|
|
{
|
|
|
|
void
|
|
AvFormat::parse(const boost::filesystem::path& p, Items& items)
|
|
{
|
|
|
|
try {
|
|
|
|
Av::InputFormatContext input(p);
|
|
input.findStreamInfo(); // needed by input.getDurationSecs
|
|
|
|
std::cout << "Processing file '" << p << "'" << std::endl;
|
|
|
|
std::map<std::string, std::string> metadata;
|
|
input.getMetadata().get(metadata);
|
|
|
|
// HACK or OGG files
|
|
// If we did not find tags, searched metadata in streams
|
|
if (metadata.empty())
|
|
{
|
|
// Get input streams
|
|
std::vector<Av::Stream> streams = input.getStreams();
|
|
|
|
BOOST_FOREACH(Av::Stream& stream, streams) {
|
|
stream.getMetadata().get(metadata);
|
|
|
|
if (!metadata.empty())
|
|
break;
|
|
}
|
|
}
|
|
|
|
std::cout << "Found " << metadata.size() << " tags!" << std::endl;
|
|
|
|
items.insert( std::make_pair(MetaData::Duration, boost::posix_time::time_duration( boost::posix_time::seconds( input.getDurationSecs() )) ));
|
|
|
|
// Make sure to convert strings into UTF-8
|
|
std::map<std::string, std::string>::const_iterator it;
|
|
for (it = metadata.begin(); it != metadata.end(); ++it)
|
|
{
|
|
if (boost::iequals(it->first, "artist"))
|
|
items.insert( std::make_pair(MetaData::Artist, string_to_utf8(it->second)) );
|
|
else if (boost::iequals(it->first, "album"))
|
|
items.insert( std::make_pair(MetaData::Album, string_to_utf8(it->second) ));
|
|
else if (boost::iequals(it->first, "title"))
|
|
items.insert( std::make_pair(MetaData::Title, string_to_utf8(it->second) ));
|
|
else if (boost::iequals(it->first, "track")) {
|
|
std::size_t number;
|
|
if (readAs<std::size_t>(it->second, number))
|
|
items.insert( std::make_pair(MetaData::TrackNumber, number ));
|
|
}
|
|
else if (boost::iequals(it->first, "disc"))
|
|
{
|
|
std::size_t number;
|
|
if (readAs<std::size_t>(it->second, number))
|
|
items.insert( std::make_pair(MetaData::DiscNumber, number ));
|
|
}
|
|
else if (boost::iequals(it->first, "date")
|
|
|| boost::iequals(it->first, "year")
|
|
|| boost::iequals(it->first, "WM/Year")
|
|
|| boost::iequals(it->first, "TDOR") // Original date fallback
|
|
|| boost::iequals(it->first, "TORY") // Original date fallback
|
|
)
|
|
{
|
|
boost::posix_time::ptime p;
|
|
if (readAsPosixTime(it->second, p))
|
|
items.insert( std::make_pair(MetaData::CreationTime, p));
|
|
}
|
|
else if (boost::iequals(it->first, "genre"))
|
|
{
|
|
std::list<std::string> genres;
|
|
if (readList(it->second, ";,", genres))
|
|
items.insert( std::make_pair(MetaData::Genre, genres));
|
|
|
|
}
|
|
/* else
|
|
std::cout << "key = " << it->first << ", value = " << it->second << std::endl;
|
|
*/
|
|
}
|
|
|
|
}
|
|
catch(std::exception &e)
|
|
{
|
|
std::cerr << "Parsing of '" << p << "' failed!" << std::endl;
|
|
}
|
|
|
|
|
|
}
|
|
|
|
} // namespace MetaData
|
|
|