utils.hpp

#ifndef TURI_ANNOTATIONS_UTILS_HPP
#define TURI_ANNOTATIONS_UTILS_HPP

#include "build/format/cpp/annotate.pb.h"
#include "build/format/cpp/data.pb.h"
#include "build/format/cpp/message.pb.h"
#include "build/format/cpp/meta.pb.h"

#include <toolkits/image_deep_feature_extractor/image_deep_feature_extractor_toolkit.hpp>

#include <boost/regex.hpp>
#include <memory>
#include <core/data/sframe/gl_sarray.hpp>
#include <core/storage/sframe_interface/unity_sarray.hpp>
#include <vector>

namespace annotate_spec = TuriCreate::Annotation::Specification;

namespace turi {
namespace annotate {

template <typename T>
typename std::enable_if<
    std::is_same<T, annotate_spec::Annotations>::value>::type
populate_parcel(annotate_spec::Parcel &parcel, T message) {
  parcel.mutable_annotations()->CopyFrom(message);
}

template <typename T>
typename std::enable_if<std::is_same<T, annotate_spec::Data>::value>::type
populate_parcel(annotate_spec::Parcel &parcel, T message) {
  parcel.mutable_data()->CopyFrom(message);
}

template <typename T>
typename std::enable_if<std::is_same<T, annotate_spec::MetaData>::value>::type
populate_parcel(annotate_spec::Parcel &parcel, T message) {
  parcel.mutable_metadata()->CopyFrom(message);
}

template <typename T>
typename std::enable_if<
    std::is_same<T, annotate_spec::ProgressMeta>::value>::type
populate_parcel(annotate_spec::Parcel &parcel, T message) {
  parcel.mutable_progress()->CopyFrom(message);
}

template <typename T>
typename std::enable_if<
    std::is_same<T, annotate_spec::Similarity>::value>::type
populate_parcel(annotate_spec::Parcel &parcel, T message) {
  parcel.mutable_similarity()->CopyFrom(message);
}

float vectors_distance(const std::vector<double> &a,
                       const std::vector<double> &b);

bool is_integer(std::string s);

std::vector<flexible_type> similar_items(const gl_sarray &distances,
                                         size_t index, size_t k);

#ifdef __APPLE__

image_deep_feature_extractor::image_deep_feature_extractor_toolkit
create_feature_extractor(std::string base_directory = "./");

gl_sarray featurize_images(const gl_sarray &images,
                           std::string base_directory = "./");
#endif

} // namespace annotate
} // namespace turi

#endif