workspacelibrarydb.h

Go to the documentation of this file.

/*
 * LibrePCB - Professional EDA for everyone!
 * Copyright (C) 2013 LibrePCB Developers, see AUTHORS.md for contributors.
 * https://librepcb.org/
 *
 * 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 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef LIBREPCB_CORE_WORKSPACELIBRARYDB_H
#define LIBREPCB_CORE_WORKSPACELIBRARYDB_H
 
/*******************************************************************************
 *  Includes
 ******************************************************************************/
#include "../attribute/attribute.h"
#include "../attribute/attributetype.h"
#include "../fileio/filepath.h"
#include "../library/resource.h"
#include "../types/uuid.h"
#include "../types/version.h"
 
#include <QtCore>
 
/*******************************************************************************
 *  Namespace / Forward Declarations
 ******************************************************************************/
class QSqlQuery;
 
namespace librepcb {
 
class Component;
class ComponentCategory;
class Device;
class Organization;
class Package;
class PackageCategory;
class SQLiteDatabase;
class Symbol;
class WorkspaceLibraryScanner;
 
/*******************************************************************************
 *  Class WorkspaceLibraryDb
 ******************************************************************************/
 
class WorkspaceLibraryDb final : public QObject {
  Q_OBJECT
 
public:
  struct Part {
    QString mpn;
    QString manufacturer;
    AttributeList attributes;
 
    bool operator==(const Part& rhs) const noexcept {
      return (mpn == rhs.mpn) && (manufacturer == rhs.manufacturer) &&
          (attributes == rhs.attributes);
    }
    bool operator<(const Part& rhs) const noexcept {
      if (mpn.isEmpty() != rhs.mpn.isEmpty()) {
        return mpn.count() < rhs.mpn.count();
      }
      if (mpn != rhs.mpn) {
        return mpn < rhs.mpn;
      }
      if (manufacturer != rhs.manufacturer) {
        return manufacturer < rhs.manufacturer;
      }
      QCollator collator;
      collator.setNumericMode(true);
      collator.setCaseSensitivity(Qt::CaseInsensitive);
      collator.setIgnorePunctuation(false);
      for (int i = 0; i < std::max(attributes.count(), rhs.attributes.count());
           ++i) {
        auto a = attributes.value(i);
        auto b = rhs.attributes.value(i);
        if (a && (!b)) {
          return false;
        } else if ((!a) && b) {
          return true;
        } else if (a->getKey() != b->getKey()) {
          return a->getKey() < b->getKey();
        } else if (&a->getType() != &b->getType()) {
          return a->getType().getName() < b->getType().getName();
        } else if (a->getValueTr(true) != b->getValueTr(true)) {
          return collator(a->getValueTr(true), b->getValueTr(true));
        }
      }
      return false;
    }
  };
  struct PcbDesignRules {
    Uuid uuid;
    QString name;
    QString description;
    QUrl url;
    int maxLayerCount;
  };
  enum class OutputJobKind : int {
    Pcb = 1,
    Assembly = 2,
    User = 99,
  };
  struct OutputJob {
    OutputJobKind kind;
    Uuid uuid;
    QString type;
    QString name;
  };
  struct Organization {
    int id;
    FilePath elemDir;
    Uuid uuid;
    QString name;
    QString description;
    Version version;
    QPixmap logo;
    QUrl url;
    QString country;
    QStringList fabs;
    QStringList shipping;
    bool isSponsor;
    int priority;
    QList<PcbDesignRules> pcbDesignRules;
    QList<OutputJob> outputJobs;
  };
 
  // Constructors / Destructor
  WorkspaceLibraryDb() = delete;
  WorkspaceLibraryDb(const WorkspaceLibraryDb& other) = delete;
 
  explicit WorkspaceLibraryDb(const FilePath& librariesPath);
  ~WorkspaceLibraryDb() noexcept;
 
  // Getters
 
  const FilePath& getFilePath() const noexcept { return mFilePath; }
 
  bool isScanInProgress() const noexcept {
    return getScanProgressPercent() < 100;
  }
 
  int getScanProgressPercent() const noexcept;
 
  template <typename ElementType>
  QMultiMap<Version, FilePath> getAll(
      const std::optional<Uuid>& uuid = std::nullopt,
      const FilePath& lib = FilePath()) const {
    return getAll(getTable<ElementType>(), uuid, lib);
  }
 
  template <typename ElementType>
  QHash<FilePath, Uuid> getAll(const FilePath& lib) const {
    return getAll(getTable<ElementType>(), lib);
  }
 
  template <typename ElementType>
  FilePath getLatest(const Uuid& uuid) const {
    return getLatestVersionFilePath(getAll<ElementType>(uuid));
  }
 
  template <typename ElementType>
  QList<Uuid> find(const QString& keyword) const;
 
  QList<Uuid> findDevicesOfParts(const QString& keyword) const;
 
  QList<Part> findPartsOfDevice(const Uuid& device,
                                const QString& keyword) const;
 
  template <typename ElementType>
  bool getTranslations(const FilePath& elemDir, const QStringList& localeOrder,
                       QString* name = nullptr, QString* description = nullptr,
                       QString* keywords = nullptr) const {
    return getTranslations(getTable<ElementType>(), elemDir, localeOrder, name,
                           description, keywords);
  }
 
  template <typename ElementType>
  bool getMetadata(const FilePath elemDir, Uuid* uuid = nullptr,
                   Version* version = nullptr,
                   bool* deprecated = nullptr) const {
    return getMetadata(getTable<ElementType>(), elemDir, uuid, version,
                       deprecated);
  }
 
  bool getLibraryMetadata(const FilePath libDir, QPixmap* icon = nullptr,
                          QString* manufacturer = nullptr) const;
 
  template <typename ElementType>
  bool getCategoryMetadata(const FilePath catDir,
                           std::optional<Uuid>* parent = nullptr) const {
    static_assert(std::is_same<ElementType, ComponentCategory>::value ||
                      std::is_same<ElementType, PackageCategory>::value,
                  "Unsupported ElementType");
    return getCategoryMetadata(getTable<ElementType>(), catDir, parent);
  }
 
  bool getDeviceMetadata(const FilePath& devDir, Uuid* cmpUuid = nullptr,
                         Uuid* pkgUuid = nullptr) const;
 
  template <typename ElementType>
  QSet<Uuid> getChilds(const std::optional<Uuid>& parent) const {
    static_assert(std::is_same<ElementType, ComponentCategory>::value ||
                      std::is_same<ElementType, PackageCategory>::value,
                  "Unsupported ElementType");
    return getChilds(getTable<ElementType>(), parent);
  }
 
  template <typename ElementType>
  QSet<Uuid> getByCategory(const std::optional<Uuid>& category,
                           int limit = -1) const {
    static_assert(std::is_same<ElementType, Symbol>::value ||
                      std::is_same<ElementType, Package>::value ||
                      std::is_same<ElementType, Component>::value ||
                      std::is_same<ElementType, Device>::value,
                  "Unsupported ElementType");
    return getByCategory(getTable<ElementType>(),
                         getCategoryTable<ElementType>(), category, limit);
  }
 
  template <typename ElementType>
  QSet<Uuid> getCategoriesOf(const FilePath& elemDir) const {
    static_assert(std::is_same<ElementType, Symbol>::value ||
                      std::is_same<ElementType, Package>::value ||
                      std::is_same<ElementType, Component>::value ||
                      std::is_same<ElementType, Device>::value,
                  "Unsupported ElementType");
    return getCategoriesOf(getTable<ElementType>(),
                           getCategoryTable<ElementType>(), elemDir);
  }
 
  template <typename ElementType>
  QSet<Uuid> getGenerated(const QString& generatedBy) const {
    static_assert(std::is_same<ElementType, Symbol>::value ||
                      std::is_same<ElementType, Package>::value ||
                      std::is_same<ElementType, Component>::value ||
                      std::is_same<ElementType, Device>::value,
                  "Unsupported ElementType");
    return getGenerated(getTable<ElementType>(), generatedBy);
  }
 
  template <typename ElementType>
  ResourceList getResources(const FilePath elemDir) const {
    static_assert(std::is_same<ElementType, Component>::value ||
                      std::is_same<ElementType, Device>::value,
                  "Unsupported ElementType");
    return getResources(getTable<ElementType>(), elemDir);
  }
 
  QSet<Uuid> getComponentDevices(const Uuid& component) const;
 
  QList<Part> getDeviceParts(const Uuid& device) const;
 
  QList<Organization> getAllLatestOrganizations(const QStringList& localeOrder,
                                                bool pcbDesignRules,
                                                bool outputJobs) const;
 
  // General Methods
 
  void resetAndRescan() noexcept;
 
  void startLibraryRescan() noexcept;
 
  bool cancelLibraryRescan() noexcept;
 
  // Operator Overloadings
  WorkspaceLibraryDb& operator=(const WorkspaceLibraryDb& rhs) = delete;
 
signals:
  void scanStarted();
  void scanLibraryListUpdated(int libraryCount);
  void scanProgressUpdate(int percent);
  void scanSucceeded(int elementCount);
  void scanFailed(QString errorMsg);
  void scanFinished();
  void scanInProgressChanged(bool inProgress);
 
private:
  // Private Methods
  void reset();
  QMultiMap<Version, FilePath> getAll(const QString& elementsTable,
                                      const std::optional<Uuid>& uuid,
                                      const FilePath& lib) const;
  QHash<FilePath, Uuid> getAll(const QString& elementsTable,
                               const FilePath& lib) const;
  FilePath getLatestVersionFilePath(
      const QMultiMap<Version, FilePath>& list) const noexcept;
  QList<Uuid> find(const QString& elementsTable, const QString& keyword) const;
  bool getTranslations(const QString& elementsTable, const FilePath& elemDir,
                       const QStringList& localeOrder, QString* name,
                       QString* description, QString* keywords) const;
  bool getMetadata(const QString& elementsTable, const FilePath elemDir,
                   Uuid* uuid, Version* version, bool* deprecated) const;
  bool getCategoryMetadata(const QString& categoriesTable,
                           const FilePath catDir,
                           std::optional<Uuid>* parent) const;
  AttributeList getPartAttributes(int partId) const;
  QSet<Uuid> getChilds(const QString& categoriesTable,
                       const std::optional<Uuid>& categoryUuid) const;
  QSet<Uuid> getByCategory(const QString& elementsTable,
                           const QString& categoryTable,
                           const std::optional<Uuid>& category,
                           int limit) const;
  QSet<Uuid> getCategoriesOf(const QString& elementsTable,
                             const QString& categoryTable,
                             const FilePath& elemDir) const;
  QSet<Uuid> getGenerated(const QString& elementsTable,
                          const QString& generatedBy) const;
  ResourceList getResources(const QString& elementsTable,
                            const FilePath& elemDir) const;
  static QSet<Uuid> getUuidSet(QSqlQuery& query);
  int getDbVersion() const noexcept;
  template <typename ElementType>
  static QString getTable() noexcept;
  template <typename ElementType>
  static QString getCategoryTable() noexcept;
 
  // Attributes
  const FilePath mLibrariesPath;  
  const FilePath mFilePath;  
  QScopedPointer<SQLiteDatabase> mDb;  
  QScopedPointer<WorkspaceLibraryScanner> mLibraryScanner;
 
  // Constants
  static const int sCurrentDbVersion = 8;
};
 
/*******************************************************************************
 *  End of File
 ******************************************************************************/
 
}  // namespace librepcb
 
#endif