ksuserdb.h

Go to the documentation of this file.

/***************************************************************************
                          ksuserdb.h  -  K Desktop Planetarium
                             -------------------
    begin                : Wed May 2 2012
    copyright            : (C) 2012 by Rishab Arora
    email                : ra.rishab@gmail.com
 ***************************************************************************/


/***************************************************************************
 *                                                                         *
 *   This program 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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

#ifndef KSTARS_KSUSERDB_H
#define KSTARS_KSUSERDB_H
#define KSTARS_USERDB "data/userdb.sqlite"
#include <kstandarddirs.h>
#include <klocale.h>
#include <QSqlDatabase>
#include <QDebug>
#include <QSqlQuery>
#include <QSqlRecord>
#include <QSqlTableModel>
#include <QString>
#include <QHash>
#include <QSqlError>
#include <QVariant>
#include <QFile>
#include <QXmlStreamReader>
#include "skyobjects/skyobject.h"
#include "oal/oal.h"


class KSUserDB {
 public:
    bool Initialize();
    ~KSUserDB();

    void AddObserver(const QString &name, const QString &surname, const QString &contact);
    int FindObserver(const QString &name, const QString &surname);
    bool DeleteObserver(const QString &id);
    void GetAllObservers(QList<OAL::Observer *> &observer_list);

    void EraseAllFlags();
    void AddFlag(const QString &ra, const QString &dec, const QString &epoch,
                 const QString &image_name, const QString &label, const QString &labelColor);
    QList<QStringList> ReturnAllFlags();

    void EraseEquipment(const QString &type, const int &id);
    void EraseAllEquipment(const QString &type);

    void AddScope(const QString &model, const QString &vendor, const QString &driver,
                  const QString &type, const double &focalLength, const double &aperture);
    void AddScope(const QString &model, const QString &vendor, const QString &driver,
                  const QString &type, const double &focalLength, const double &aperture,
                  const QString &id);
    void GetAllScopes(QList<OAL::Scope *> &m_scopeList);

    void AddEyepiece(const QString &vendor, const QString &model, const double &focalLength,
                     const double &fov, const QString &fovunit);
    void AddEyepiece(const QString &vendor, const QString &model, const double &focalLength,
                           const double &fov, const QString &fovunit, const QString &id);
    void GetAllEyepieces(QList<OAL::Eyepiece *> &m_eyepieceList);

    void AddLens(const QString &vendor, const QString &model, const double &factor);
    void AddLens(const QString &vendor, const QString &model, const double &factor, const QString &id);
    void GetAllLenses(QList<OAL::Lens *>& m_lensList);

    void AddFilter(const QString &vendor, const QString &model, const QString &type, const QString &color);
    void AddFilter(const QString &vendor, const QString &model, const QString &type, const QString &color,
                   const QString &id);
    void GetAllFilters(QList<OAL::Filter *>& m_filterList);

 private:
    bool RebuildDB();
    bool FirstRun();
    bool ImportFlags();
    bool ImportUsers();
    bool ImportEquipment();
    void readScopes();
    void readScope(QString id);
    void readEyepieces();
    void readEyepiece(QString id);
    void readLenses();
    void readLens(QString id);
    void readFilters();
    void readFilter(QString id);
    QSqlDatabase userdb_;
    QXmlStreamReader *reader_;
    inline QSqlError LastError();
};

#endif  // KSTARS_KSUSERDB_H