TemplatesIndex.h

Go to the documentation of this file.

#ifndef __PBIO_API__PBIO__TEMPLATES_INDEX_H_
#define __PBIO_API__PBIO__TEMPLATES_INDEX_H_


#include <ostream>
#include <sstream>
#include <vector>
#include <string>


#include "ComplexObject.h"
#include "Error.h"
#include "SmartPtr.h"
#include "stl_wraps_impls/WrapOStreamImpl.h"




namespace pbio
{

class Recognizer;

class TemplatesIndex : public ComplexObject
{
public:

    typedef LightSmartPtr<TemplatesIndex>::tPtr Ptr;

    std::string getMethodName() const;

    size_t size() const;

    pbio::Template::Ptr at(size_t i) const;


    void reserveSearchMemory(const int queries_count) const;

    void add(pbio::Template::Ptr templ, const std::string& uuid);

    void remove(const std::string& uuid);

    void remove(const std::vector<std::string>& uuids);

    void concat(TemplatesIndex::Ptr otherIndex);

private:

    TemplatesIndex(
        const DHPtr &dll_handle,
        void* impl);

    friend class Recognizer;
    friend class object_with_ref_counter<TemplatesIndex>;
};

}  // pbio namespace





namespace pbio
{

inline
TemplatesIndex::TemplatesIndex(
    const DHPtr &dll_handle,
    void* impl):
ComplexObject(dll_handle, impl)
{
    // nothing else
}



inline
std::string TemplatesIndex::getMethodName() const
{
    std::ostringstream name_stream;
    pbio::stl_wraps::WrapOStreamImpl name_stream_wrap(name_stream);

    void* exception = NULL;

    _dll_handle->TemplatesIndex_getMethodName(
        _impl,
        &name_stream_wrap,
        pbio::stl_wraps::WrapOStream::write_func,
        &exception);

    checkException(exception, *_dll_handle);

    return name_stream.str();
}


inline
size_t TemplatesIndex::size() const
{
    void* exception = NULL;

    const size_t size = _dll_handle->TemplatesIndex_size(
        _impl,
        &exception);

    checkException(exception, *_dll_handle);

    return size;
}


inline
Template::Ptr TemplatesIndex::at(size_t i) const
{
    void* exception = NULL;

    pbio::facerec::TemplateImpl* const result_impl = _dll_handle->TemplatesIndex_at(
        _impl,
        i,
        &exception);

    checkException(exception, *_dll_handle);

    return Template::Ptr::make(_dll_handle, result_impl);
}


inline
void TemplatesIndex::reserveSearchMemory(const int queries_count) const
{
    void* exception = NULL;

    _dll_handle->TemplatesIndex_reserveSearchMemory(
        _impl,
        queries_count,
        &exception);

    checkException(exception, *_dll_handle);
}

inline void TemplatesIndex::add(pbio::Template::Ptr templ, const std::string& uuid)
{
    void* exception = NULL;

    _dll_handle->ResizableTemplatesIndex_add_1(_impl, static_cast<pbio::facerec::TemplateImpl*>(templ->_impl), uuid.data(), &exception);

    checkException(exception, *_dll_handle);
}

inline void TemplatesIndex::remove(const std::string& uuid)
{
    void* exception = NULL;

    _dll_handle->ResizableTemplatesIndex_remove_1(_impl, uuid.data(), &exception);

    checkException(exception, *_dll_handle);
}

inline void TemplatesIndex::remove(const std::vector<std::string>& uuids)
{
    void* exception = NULL;

    const char** data = new const char*[uuids.size()];

    for (size_t i = 0; i < uuids.size(); i++)
    {
        data[i] = uuids[i].data();
    }

    _dll_handle->ResizableTemplatesIndex_remove_2(_impl, data, uuids.size(), &exception);

    delete[] data;

    checkException(exception, *_dll_handle);
}

inline void TemplatesIndex::concat(TemplatesIndex::Ptr otherIndex)
{
    void* exception = NULL;

    _dll_handle->ResizableTemplatesIndex_concatenate(_impl, otherIndex->_impl, &exception);

    checkException(exception, *_dll_handle);
}

}  // pbio namespace
#endif  // __PBIO_API__PBIO__TEMPLATES_INDEX_H_