ObjectFinder.h

Go to the documentation of this file.

#pragma once
 
#include <filesystem>
#include <unordered_map>
 
#include <VirtualRobot/VirtualRobot.h>
 
#include <ArmarXCore/core/logging/Logging.h>
 
#include <RobotAPI/libraries/armem_objects/types.h>
 
#include "ObjectInfo.h"
#include "ObjectPose.h"
 
namespace armarx
{
    /**
     * @brief Used to find objects in the ArmarX objects repository [1] (formerly [2]).
     *
     * @see [1] https://git.h2t.iar.kit.edu/sw/armarx/prior-knowledge-data
     */
    class ObjectFinder : Logging
    {
    public:
        using path = std::filesystem::path;
        inline static const std::string DefaultObjectsPackageName = "PriorKnowledgeData";
        inline static const std::string DefaultObjectsDirectory = "objects";
 
 
    public:
        ObjectFinder(const std::string& objectsPackageName = DefaultObjectsPackageName,
                     const path& relObjectsDir = DefaultObjectsDirectory);
 
        ObjectFinder(ObjectFinder&&) = default;
        ObjectFinder(const ObjectFinder&) = default;
        ObjectFinder& operator=(ObjectFinder&&) = default;
        ObjectFinder& operator=(const ObjectFinder&) = default;
 
 
        void setPath(const std::string& path);
 
        std::string getPackageName() const;
 
        std::optional<ObjectInfo> findObject(const std::string& dataset,
                                             const std::string& name) const;
        std::optional<ObjectInfo> findObject(const std::string& nameOrID) const;
        std::optional<ObjectInfo> findObject(const ObjectID& id) const;
        std::optional<ObjectInfo> findObject(const objpose::ObjectPose& obj) const;
 
        std::vector<std::string> getDatasets() const;
        std::vector<path> getDatasetDirectories() const;
 
        std::vector<ObjectInfo> findAllObjects(bool checkPaths = true) const;
        std::map<std::string, std::vector<ObjectInfo>>
        findAllObjectsByDataset(bool checkPaths = true) const;
        std::vector<ObjectInfo> findAllObjectsOfDataset(const std::string& dataset,
                                                        bool checkPaths = true) const;
 
        std::vector<armem::articulated_object::ArticulatedObjectDescription>
        findAllArticulatedObjects(bool checkPaths) const;
        std::vector<armem::articulated_object::ArticulatedObjectDescription>
        findAllArticulatedObjectsOfDataset(const std::string& dataset, bool checkPaths) const;
        std::unordered_map<std::string,
                           std::vector<armem::articulated_object::ArticulatedObjectDescription>>
        findAllArticulatedObjectsByDataset(bool checkPaths = true) const;
 
        VirtualRobot::ManipulationObjectPtr static loadManipulationObject(
            const std::optional<ObjectInfo>& ts);
        VirtualRobot::ManipulationObjectPtr
        loadManipulationObject(const objpose::ObjectPose& obj) const;
 
        VirtualRobot::ObstaclePtr static loadObstacle(const std::optional<ObjectInfo>& ts);
        VirtualRobot::ObstaclePtr loadObstacle(const objpose::ObjectPose& obj) const;
 
 
        /**
         * @brief Load names to use when matched when recognizing an object by name.
         *
         * If the object's names JSON file does not exist, no names will be added from a file.
         * If you would like to detect this case, first `findObject()`, then use
         * `ObjectInfo::loadRecognizedNames()`, which returns a `std::optional`.
         *
         * @param includeClassName If true, include the raw class name in the result.
         * @see `ObjectInfo::loadRecognizedNames()`
         */
        std::vector<std::string> loadRecognizedNames(const ObjectID& objectID,
                                                     bool includeClassName = false) const;
        /**
         * @brief Load names to use when verbalizing an object name.
         *
         * If the object's names JSON file does not exist, no names will be added from a file.
         * If you would like to detect this case, first `findObject()`, then use
         * `ObjectInfo::loadSpokenNames()`, which returns a `std::optional`.
         *
         * @param includeClassName If true, include the raw class name in the result.
         * @see `ObjectInfo::loadSpokenNames()`
         */
        std::vector<std::string> loadSpokenNames(const ObjectID& objectID,
                                                 bool includeClassName = false) const;
 
        void setLogObjectDiscoveryError(bool logEnabled);
 
 
    private:
        void init() const;
        bool isDatasetDirValid(const std::filesystem::path& path) const;
 
        path _rootDirAbs() const;
        path _rootDirRel() const;
 
        bool _ready() const;
 
 
    private:
        /// Name of package containing the object models (ArmarXObjects by default).
        mutable std::string packageName;
 
        /**
         * @brief Absolute path to data directory (e.g. "/.../repos/ArmarXObjects/data").
         * Empty if package could not be found.
         */
        mutable path absPackageDataDir;
 
        /// Path to the directory containing objects in the package's data directory.
        path relObjectsDir;
 
 
        // see: ObjectInfo::checkPaths()
        bool logObjectDiscoveryError = true;
    };
} // namespace armarx