attributes.h

/* Copyright 2021 ID R&D Inc. All Rights Reserved. */
 
#pragma once
 
#include <voicesdk/core/config.h>
 
#include <memory>
#include <string>
#include <iostream>
 
namespace voicesdk {
namespace attr {
 
struct Attributes {
 
    enum class Gender : uint8_t { MALE = 0, FEMALE = 1 };
 
    enum class PhoneCallParticipant : uint8_t {
        OPERATOR = 0, 
        CUSTOMER = 1, 
        OTHER = 2,    
        UNDEFINED = 3 
    };
 
    Attributes(const Attributes& other) = default;
 
    Attributes(const float& gender_score_, const Gender& gender_, const size_t& age_,
               const PhoneCallParticipant& phone_call_participant_ = PhoneCallParticipant::UNDEFINED)
        : gender_score(gender_score_),
          gender(gender_),
          age(age_),
          phone_call_participant(phone_call_participant_) {}
 
    bool operator==(const Attributes& other) const {
        return gender_score == other.gender_score && gender == other.gender && age == other.age;
    }
 
    friend std::ostream& operator<<(std::ostream& os, const Attributes& obj) {
        os << "Attributes["
           << "genderScore: " << obj.gender_score << ", "
           << "gender: " << (obj.gender == Gender::MALE ? "MALE" : "FEMALE") << ", "
           << "age: " << obj.age << ", "
           << "phoneCallParticipant: "
           << (obj.phone_call_participant == PhoneCallParticipant::OPERATOR
                       ? "OPERATOR"
                       : (obj.phone_call_participant == PhoneCallParticipant::CUSTOMER
                                  ? "CUSTOMER"
                                  : (obj.phone_call_participant == PhoneCallParticipant::OTHER ? "OTHER"
                                                                                               : "UNDEFINED")))
           << "]";
        return os;
    }
 
    const float gender_score;
 
    const Gender gender;
 
    const size_t age;
 
    const PhoneCallParticipant phone_call_participant;
};
 
class VOICE_SDK_API AttributesEstimator {
public:
    using Ptr = std::shared_ptr<AttributesEstimator>;
 
    static AttributesEstimator::Ptr Create(const std::string& init_path);
 
    virtual Attributes Estimate(const uint8_t* bytes, size_t bytes_num, size_t sample_rate) = 0;
 
    virtual Attributes Estimate(const int16_t* pcm16_samples, size_t samples_num, size_t sample_rate) = 0;
 
    virtual Attributes Estimate(const float* float_samples, size_t samples_num, size_t sample_rate) = 0;
 
    virtual Attributes Estimate(const std::string& audio_file) = 0;
 
    virtual ~AttributesEstimator() = default;
};
 
}  // namespace attr
}  // namespace voicesdk