You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

322 lines
5.3 KiB
C++

/*!
\file clocation.cpp
*/
#include "common.h"
#include "clocation.h"
#include <QSqlQuery>
#include <QSqlError>
#include <QApplication>
cLocation::cLocation(qint32 iID, QObject *parent) :
QObject(parent),
m_iID(iID),
m_szName("")
{
}
bool cLocation::toDB()
{
QSqlQuery query;
if(m_iID != -1)
{
query.prepare("SELECT id FROM location WHERE id=:id;");
query.bindValue(":id", m_iID);
if(!query.exec())
{
myDebug << query.lastError().text();
return(false);
}
if(!query.next())
query.prepare("INSERT INTO location (name) VALUES (:name);");
else
query.prepare("UPDATE location SET name=:name WHERE id=:id;");
}
else
query.prepare("INSERT INTO location (name) VALUES (:name);");
query.bindValue(":id", m_iID);
query.bindValue(":name", m_szName);
if(!query.exec())
{
myDebug << query.lastError().text();
return(false);
}
if(m_iID == -1)
{
query.prepare("SELECT id FROM location WHERE _rowid_=(SELECT MAX(_rowid_) FROM location);");
if(!query.exec())
{
myDebug << query.lastError().text();
return(false);
}
query.next();
m_iID = query.value("id").toInt();
}
return(true);
}
void cLocation::setID(const qint32& id)
{
m_iID = id;
}
qint32 cLocation::id()
{
return(m_iID);
}
void cLocation::setName(const QString &name)
{
m_szName = name;
}
QString cLocation::name()
{
return(m_szName);
}
bool cLocation::operator==(const cLocation& other) const
{
if(this->m_szName != other.m_szName)
return(false);
return(true);
}
bool cLocation::operator==(const cLocation*& other) const
{
if(this->m_szName != other->m_szName)
return(false);
return(true);
}
bool cLocation::operator==(const cLocation* other) const
{
if(this->m_szName != other->m_szName)
return(false);
return(true);
}
bool cLocation::operator==(cLocation* other)
{
if(this->m_szName != other->m_szName)
return(false);
return(true);
}
cLocationList::cLocationList(QObject *parent) :
QObject(parent)
{
}
bool cLocationList::load(cSplashScreen *lpSplashScreen, QProgressBar *lpProgressBar)
{
QSqlQuery query;
query.prepare("SELECT COUNT(1) cnt FROM location;");
if(!query.exec())
{
myDebug << query.lastError().text();
return(false);
}
query.next();
qint32 max = static_cast<qint32>(query.value("cnt").toLongLong());
if(lpProgressBar)
lpProgressBar->setMaximum(max);
else if(lpSplashScreen)
lpSplashScreen->setMax(max);
query.prepare("SELECT id, "
" name "
"FROM location "
"ORDER BY UPPER(name);");
if(!query.exec())
{
myDebug << query.lastError().text();
return(false);
}
qint32 count = 0;
qint32 step = max/200;
if(!step)
step = 1;
while(query.next())
{
cLocation* lpLocation = add(query.value("id").toInt(), true);
lpLocation->setName(query.value("name").toString());
count++;
if(!(count % step))
{
if(lpProgressBar)
lpProgressBar->setValue(count);
else if(lpSplashScreen)
lpSplashScreen->setProgress(count);
qApp->processEvents();
}
}
return(true);
}
cLocation* cLocationList::add(qint32 iID, bool bNoCheck)
{
if(iID != -1 && !bNoCheck)
{
if(find(iID))
return(nullptr);
}
cLocation* lpNew = new cLocation(iID);
append(lpNew);
return(lpNew);
}
bool cLocationList::add(cLocation* lpLocation, bool bNoCheck)
{
if(bNoCheck)
{
append(lpLocation);
return(true);
}
if(contains(lpLocation))
return(false);
append(lpLocation);
return(true);
}
cLocation* cLocationList::find(qint32 iID)
{
for(cLocationList::iterator i = begin(); i != end(); i++)
{
if((*i)->id() == iID)
return(*i);
}
return(nullptr);
}
cLocation* cLocationList::find(cLocation* lpLocation)
{
for(cLocationList::iterator i = begin(); i != end(); i++)
{
if(*lpLocation == (**i))
return(*i);
}
return(nullptr);
}
cLocation* cLocationList::find(const QString& szName)
{
for(cLocationList::iterator i = begin(); i != end(); i++)
{
if(szName == (*i)->name())
return(*i);
}
return(nullptr);
}
bool cLocationList::remove(cLocation* lpLocation)
{
if(!find(lpLocation))
return(true);
QSqlQuery query;
query.prepare("SELECT COUNT(1) cnt FROM picture_location WHERE locationID=:id;");
query.bindValue(":id", lpLocation->id());
if(!query.exec())
return(false);
if(!query.next())
return(false);
if(query.value("cnt").toInt() != 0)
return(false);
query.prepare("DELETE FROM location WHERE id=:id;");
query.bindValue(":id", lpLocation->id());
if(!query.exec())
return(false);
this->removeOne(lpLocation);
delete lpLocation;
return(true);
}
QStringList cLocationList::locationList()
{
QStringList szLocationList;
for(cLocationList::iterator i = begin(); i != end();i++)
szLocationList.append((*i)->name());
szLocationList.removeDuplicates();
szLocationList.sort(Qt::CaseInsensitive);
return(szLocationList);
}
bool cLocationList::contains(cLocation* const& lpLocation)
{
return(QList::contains(lpLocation));
}
bool cLocationList::contains(QList<qint32> idList, bool bAnd)
{
if(bAnd)
{
int found = 0;
for(QList<qint32>::iterator z = idList.begin();z != idList.end();z++)
{
for(cLocationList::iterator i = begin();i != end();i++)
{
if((*i)->id() == (*z))
{
found++;
break;
}
}
}
if(found == idList.count())
return(true);
return(false);
}
else
{
for(cLocationList::iterator i = begin();i != end();i++)
{
if(idList.contains((*i)->id()))
return(true);
}
return(false);
}
}