mirror of https://github.com/saitohirga/WSJT-X.git
378 lines
12 KiB
C++
378 lines
12 KiB
C++
|
#include "WorkedBefore.hpp"
|
||
|
|
||
|
#include <boost/multi_index_container.hpp>
|
||
|
#include <boost/multi_index/ordered_index.hpp>
|
||
|
#include <boost/multi_index/key_extractors.hpp>
|
||
|
#include <QByteArray>
|
||
|
#include <QStandardPaths>
|
||
|
#include <QDir>
|
||
|
#include <QFile>
|
||
|
#include <QTextStream>
|
||
|
#include "countrydat.h"
|
||
|
|
||
|
#include "pimpl_impl.hpp"
|
||
|
|
||
|
// worked before set element
|
||
|
struct worked_entry
|
||
|
{
|
||
|
explicit worked_entry (std::string const& call
|
||
|
, std::string const& grid
|
||
|
, std::string const& band
|
||
|
, std::string const& mode
|
||
|
, std::string const& country)
|
||
|
: call {call}
|
||
|
, grid {grid}
|
||
|
, band {band}
|
||
|
, mode {mode}
|
||
|
, country {country}
|
||
|
{
|
||
|
}
|
||
|
|
||
|
std::string call;
|
||
|
std::string grid;
|
||
|
std::string band;
|
||
|
std::string mode;
|
||
|
std::string country;
|
||
|
};
|
||
|
|
||
|
// tags
|
||
|
struct call_mode_band {};
|
||
|
struct call_band {};
|
||
|
struct grid_mode_band {};
|
||
|
struct grid_band {};
|
||
|
struct entity_mode_band {};
|
||
|
struct entity_band {};
|
||
|
|
||
|
// set with multiple ordered unique indexes that allow for efficient
|
||
|
// determination of various categories of worked before status
|
||
|
typedef boost::multi_index::multi_index_container<
|
||
|
worked_entry,
|
||
|
boost::multi_index::indexed_by<
|
||
|
// call+mode+band
|
||
|
boost::multi_index::ordered_unique<
|
||
|
boost::multi_index::tag<call_mode_band>,
|
||
|
boost::multi_index::composite_key<
|
||
|
worked_entry,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::call>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::mode>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::band>
|
||
|
>
|
||
|
>,
|
||
|
// call+band
|
||
|
boost::multi_index::ordered_unique<
|
||
|
boost::multi_index::tag<call_band>,
|
||
|
boost::multi_index::composite_key<
|
||
|
worked_entry,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::call>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::band>
|
||
|
>
|
||
|
>,
|
||
|
// grid+mode+band
|
||
|
boost::multi_index::ordered_unique<
|
||
|
boost::multi_index::tag<grid_mode_band>,
|
||
|
boost::multi_index::composite_key<
|
||
|
worked_entry,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::grid>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::mode>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::band>
|
||
|
>
|
||
|
>,
|
||
|
// grid+band
|
||
|
boost::multi_index::ordered_unique<
|
||
|
boost::multi_index::tag<grid_band>,
|
||
|
boost::multi_index::composite_key<
|
||
|
worked_entry,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::grid>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::band>
|
||
|
>
|
||
|
>,
|
||
|
// country+mode+band
|
||
|
boost::multi_index::ordered_unique<
|
||
|
boost::multi_index::tag<entity_mode_band>,
|
||
|
boost::multi_index::composite_key<
|
||
|
worked_entry,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::country>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::mode>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::band>
|
||
|
>
|
||
|
>,
|
||
|
// country+band
|
||
|
boost::multi_index::ordered_unique<
|
||
|
boost::multi_index::tag<entity_band>,
|
||
|
boost::multi_index::composite_key<
|
||
|
worked_entry,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::country>,
|
||
|
boost::multi_index::member<worked_entry, std::string, &worked_entry::band>
|
||
|
>
|
||
|
>
|
||
|
>
|
||
|
> worked_type;
|
||
|
|
||
|
namespace
|
||
|
{
|
||
|
auto const logFileName = "wsjtx_log.adi";
|
||
|
|
||
|
std::string extractField (QString const& record, QString const& fieldName)
|
||
|
{
|
||
|
int fieldNameIndex = record.indexOf ('<' + fieldName + ':', 0, Qt::CaseInsensitive);
|
||
|
if (fieldNameIndex >=0)
|
||
|
{
|
||
|
int closingBracketIndex = record.indexOf('>',fieldNameIndex);
|
||
|
int fieldLengthIndex = record.indexOf(':',fieldNameIndex); // find the size delimiter
|
||
|
int dataTypeIndex = -1;
|
||
|
if (fieldLengthIndex >= 0)
|
||
|
{
|
||
|
dataTypeIndex = record.indexOf(':',fieldLengthIndex+1); // check for a second : indicating there is a data type
|
||
|
if (dataTypeIndex > closingBracketIndex)
|
||
|
dataTypeIndex = -1; // second : was found but it was beyond the closing >
|
||
|
}
|
||
|
|
||
|
if ((closingBracketIndex > fieldNameIndex) && (fieldLengthIndex > fieldNameIndex) && (fieldLengthIndex< closingBracketIndex))
|
||
|
{
|
||
|
int fieldLengthCharCount = closingBracketIndex - fieldLengthIndex -1;
|
||
|
if (dataTypeIndex >= 0)
|
||
|
fieldLengthCharCount -= 2; // data type indicator is always a colon followed by a single character
|
||
|
QString fieldLengthString = record.mid(fieldLengthIndex+1,fieldLengthCharCount);
|
||
|
int fieldLength = fieldLengthString.toInt();
|
||
|
if (fieldLength > 0)
|
||
|
{
|
||
|
QString field = record.mid(closingBracketIndex+1,fieldLength);
|
||
|
return field.toStdString ();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
return "";
|
||
|
}
|
||
|
}
|
||
|
|
||
|
class WorkedBefore::impl final
|
||
|
{
|
||
|
public:
|
||
|
impl ()
|
||
|
: path_ {QDir {QStandardPaths::writableLocation (QStandardPaths::DataLocation)}.absoluteFilePath (logFileName)}
|
||
|
{
|
||
|
}
|
||
|
|
||
|
QString path_;
|
||
|
CountryDat countries_;
|
||
|
worked_type worked_;
|
||
|
};
|
||
|
|
||
|
WorkedBefore::WorkedBefore ()
|
||
|
{
|
||
|
QFile inputFile {m_->path_};
|
||
|
if (inputFile.open (QFile::ReadOnly))
|
||
|
{
|
||
|
QTextStream in(&inputFile);
|
||
|
QString buffer;
|
||
|
bool pre_read {false};
|
||
|
int end_position {-1};
|
||
|
|
||
|
// skip optional header record
|
||
|
do
|
||
|
{
|
||
|
buffer += in.readLine () + '\n';
|
||
|
if (buffer.startsWith (QChar {'<'})) // denotes no header
|
||
|
{
|
||
|
pre_read = true;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
end_position = buffer.indexOf ("<EOH>", 0, Qt::CaseInsensitive);
|
||
|
}
|
||
|
}
|
||
|
while (!in.atEnd () && !pre_read && end_position < 0);
|
||
|
if (!pre_read) // found header
|
||
|
{
|
||
|
buffer.remove (0, end_position + 5);
|
||
|
}
|
||
|
while (buffer.size () || !in.atEnd ())
|
||
|
{
|
||
|
do
|
||
|
{
|
||
|
end_position = buffer.indexOf ("<EOR>", 0, Qt::CaseInsensitive);
|
||
|
if (!in.atEnd () && end_position < 0)
|
||
|
{
|
||
|
buffer += in.readLine () + '\n';
|
||
|
}
|
||
|
}
|
||
|
while (!in.atEnd () && end_position < 0);
|
||
|
int record_length {end_position >= 0 ? end_position + 5 : -1};
|
||
|
auto record = buffer.left (record_length).trimmed ();
|
||
|
auto next_record = buffer.indexOf (QChar {'<'}, record_length);
|
||
|
buffer.remove (0, next_record >=0 ? next_record : buffer.size ());
|
||
|
record = record.mid (record.indexOf (QChar {'<'}));
|
||
|
auto call = extractField (record, "CALL");
|
||
|
if (call.size ())
|
||
|
{
|
||
|
m_->worked_.emplace (call
|
||
|
, extractField (record, "GRIDSQUARE").substr (0, 4) // not interested in 6-digit grids
|
||
|
, extractField (record, "BAND")
|
||
|
, extractField (record, "MODE")
|
||
|
, m_->countries_.find (QString::fromStdString (call)).toStdString ());
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
WorkedBefore::~WorkedBefore ()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
QString const& WorkedBefore::path () const
|
||
|
{
|
||
|
return m_->path_;
|
||
|
}
|
||
|
|
||
|
CountryDat const& WorkedBefore::countries () const
|
||
|
{
|
||
|
return m_->countries_;
|
||
|
}
|
||
|
|
||
|
bool WorkedBefore::add (QString const& call
|
||
|
, QString const& grid
|
||
|
, QString const& band
|
||
|
, QString const& mode
|
||
|
, QByteArray const& ADIF_record)
|
||
|
{
|
||
|
if (call.size ())
|
||
|
{
|
||
|
QFile file {m_->path_};
|
||
|
if (!file.open(QIODevice::Text | QIODevice::Append))
|
||
|
{
|
||
|
return false;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
QTextStream out {&file};
|
||
|
if (!file.size ())
|
||
|
{
|
||
|
out << "WSJT-X ADIF Export<eoh>" << endl; // new file
|
||
|
}
|
||
|
out << ADIF_record << " <eor>" << endl;
|
||
|
}
|
||
|
m_->worked_.emplace (call.toStdString ()
|
||
|
, grid.toStdString ()
|
||
|
, band.toStdString ()
|
||
|
, mode.toStdString ()
|
||
|
, m_->countries_.find (call).toStdString ());
|
||
|
}
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
bool WorkedBefore::country_worked (QString const& country, QString const& mode, QString const& band) const
|
||
|
{
|
||
|
if (mode.size ())
|
||
|
{
|
||
|
if (band.size ())
|
||
|
{
|
||
|
return
|
||
|
country.size ()
|
||
|
&& m_->worked_.get<entity_mode_band> ().end ()
|
||
|
!= m_->worked_.get<entity_mode_band> ().find (std::make_tuple (country.toStdString ()
|
||
|
, mode.toStdString ()
|
||
|
, band.toStdString ()));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// partial key lookup
|
||
|
return
|
||
|
country.size ()
|
||
|
&& m_->worked_.get<entity_mode_band> ().end ()
|
||
|
!= m_->worked_.get<entity_mode_band> ().find (std::make_tuple (country.toStdString ()
|
||
|
, mode.toStdString ()));
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if (band.size ())
|
||
|
{
|
||
|
return
|
||
|
country.size ()
|
||
|
&& m_->worked_.get<entity_band> ().end ()
|
||
|
!= m_->worked_.get<entity_band> ().find (std::make_tuple (country.toStdString ()
|
||
|
, band.toStdString ()));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// partial key lookup
|
||
|
return
|
||
|
country.size ()
|
||
|
&& m_->worked_.get<entity_band> ().end ()
|
||
|
!= m_->worked_.get<entity_band> ().find (std::make_tuple (country.toStdString ()));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
bool WorkedBefore::grid_worked (QString const& grid, QString const& mode, QString const& band) const
|
||
|
{
|
||
|
if (mode.size ())
|
||
|
{
|
||
|
if (band.size ())
|
||
|
{
|
||
|
return m_->worked_.get<grid_mode_band> ().end ()
|
||
|
!= m_->worked_.get<grid_mode_band> ().find (std::make_tuple (grid.toStdString ()
|
||
|
, mode.toStdString ()
|
||
|
, band.toStdString ()));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// partial key lookup
|
||
|
return m_->worked_.get<grid_mode_band> ().end ()
|
||
|
!= m_->worked_.get<grid_mode_band> ().find (std::make_tuple (grid.toStdString ()
|
||
|
, mode.toStdString ()));
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if (band.size ())
|
||
|
{
|
||
|
return m_->worked_.get<grid_band> ().end ()
|
||
|
!= m_->worked_.get<grid_band> ().find (std::make_tuple (grid.toStdString ()
|
||
|
, band.toStdString ()));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// partial key lookup
|
||
|
return m_->worked_.get<grid_band> ().end ()
|
||
|
!= m_->worked_.get<grid_band> ().find (std::make_tuple (grid.toStdString ()));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
bool WorkedBefore::call_worked (QString const& call, QString const& mode, QString const& band) const
|
||
|
{
|
||
|
if (mode.size ())
|
||
|
{
|
||
|
if (band.size ())
|
||
|
{
|
||
|
return m_->worked_.get<call_mode_band> ().end ()
|
||
|
!= m_->worked_.get<call_mode_band> ().find (std::make_tuple (call.toStdString ()
|
||
|
, mode.toStdString ()
|
||
|
, band.toStdString ()));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// partial key lookup
|
||
|
return m_->worked_.get<call_mode_band> ().end ()
|
||
|
!= m_->worked_.get<call_mode_band> ().find (std::make_tuple (call.toStdString ()
|
||
|
, mode.toStdString ()));
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if (band.size ())
|
||
|
{
|
||
|
return m_->worked_.get<call_band> ().end ()
|
||
|
!= m_->worked_.get<call_band> ().find (std::make_tuple (call.toStdString ()
|
||
|
, band.toStdString ()));
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// partial key lookup
|
||
|
return m_->worked_.get<call_band> ().end ()
|
||
|
!= m_->worked_.get<call_band> ().find (std::make_tuple (call.toStdString ()));
|
||
|
}
|
||
|
}
|
||
|
}
|