EntityRecognizerInputDataConfig.h

 
#pragma once
#include <aws/comprehend/Comprehend_EXPORTS.h>
#include <aws/comprehend/model/AugmentedManifestsListItem.h>
#include <aws/comprehend/model/EntityRecognizerAnnotations.h>
#include <aws/comprehend/model/EntityRecognizerDataFormat.h>
#include <aws/comprehend/model/EntityRecognizerDocuments.h>
#include <aws/comprehend/model/EntityRecognizerEntityList.h>
#include <aws/comprehend/model/EntityTypesListItem.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace Comprehend {
namespace Model {
 
class EntityRecognizerInputDataConfig {
 public:
  AWS_COMPREHEND_API EntityRecognizerInputDataConfig() = default;
  AWS_COMPREHEND_API EntityRecognizerInputDataConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_COMPREHEND_API EntityRecognizerInputDataConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_COMPREHEND_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline EntityRecognizerDataFormat GetDataFormat() const { return m_dataFormat; }
  inline bool DataFormatHasBeenSet() const { return m_dataFormatHasBeenSet; }
  inline void SetDataFormat(EntityRecognizerDataFormat value) {
    m_dataFormatHasBeenSet = true;
    m_dataFormat = value;
  }
  inline EntityRecognizerInputDataConfig& WithDataFormat(EntityRecognizerDataFormat value) {
    SetDataFormat(value);
    return *this;
  }
 
 
  inline const Aws::Vector<EntityTypesListItem>& GetEntityTypes() const { return m_entityTypes; }
  inline bool EntityTypesHasBeenSet() const { return m_entityTypesHasBeenSet; }
  template <typename EntityTypesT = Aws::Vector<EntityTypesListItem>>
  void SetEntityTypes(EntityTypesT&& value) {
    m_entityTypesHasBeenSet = true;
    m_entityTypes = std::forward<EntityTypesT>(value);
  }
  template <typename EntityTypesT = Aws::Vector<EntityTypesListItem>>
  EntityRecognizerInputDataConfig& WithEntityTypes(EntityTypesT&& value) {
    SetEntityTypes(std::forward<EntityTypesT>(value));
    return *this;
  }
  template <typename EntityTypesT = EntityTypesListItem>
  EntityRecognizerInputDataConfig& AddEntityTypes(EntityTypesT&& value) {
    m_entityTypesHasBeenSet = true;
    m_entityTypes.emplace_back(std::forward<EntityTypesT>(value));
    return *this;
  }
 
 
  inline const EntityRecognizerDocuments& GetDocuments() const { return m_documents; }
  inline bool DocumentsHasBeenSet() const { return m_documentsHasBeenSet; }
  template <typename DocumentsT = EntityRecognizerDocuments>
  void SetDocuments(DocumentsT&& value) {
    m_documentsHasBeenSet = true;
    m_documents = std::forward<DocumentsT>(value);
  }
  template <typename DocumentsT = EntityRecognizerDocuments>
  EntityRecognizerInputDataConfig& WithDocuments(DocumentsT&& value) {
    SetDocuments(std::forward<DocumentsT>(value));
    return *this;
  }
 
 
  inline const EntityRecognizerAnnotations& GetAnnotations() const { return m_annotations; }
  inline bool AnnotationsHasBeenSet() const { return m_annotationsHasBeenSet; }
  template <typename AnnotationsT = EntityRecognizerAnnotations>
  void SetAnnotations(AnnotationsT&& value) {
    m_annotationsHasBeenSet = true;
    m_annotations = std::forward<AnnotationsT>(value);
  }
  template <typename AnnotationsT = EntityRecognizerAnnotations>
  EntityRecognizerInputDataConfig& WithAnnotations(AnnotationsT&& value) {
    SetAnnotations(std::forward<AnnotationsT>(value));
    return *this;
  }
 
 
  inline const EntityRecognizerEntityList& GetEntityList() const { return m_entityList; }
  inline bool EntityListHasBeenSet() const { return m_entityListHasBeenSet; }
  template <typename EntityListT = EntityRecognizerEntityList>
  void SetEntityList(EntityListT&& value) {
    m_entityListHasBeenSet = true;
    m_entityList = std::forward<EntityListT>(value);
  }
  template <typename EntityListT = EntityRecognizerEntityList>
  EntityRecognizerInputDataConfig& WithEntityList(EntityListT&& value) {
    SetEntityList(std::forward<EntityListT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<AugmentedManifestsListItem>& GetAugmentedManifests() const { return m_augmentedManifests; }
  inline bool AugmentedManifestsHasBeenSet() const { return m_augmentedManifestsHasBeenSet; }
  template <typename AugmentedManifestsT = Aws::Vector<AugmentedManifestsListItem>>
  void SetAugmentedManifests(AugmentedManifestsT&& value) {
    m_augmentedManifestsHasBeenSet = true;
    m_augmentedManifests = std::forward<AugmentedManifestsT>(value);
  }
  template <typename AugmentedManifestsT = Aws::Vector<AugmentedManifestsListItem>>
  EntityRecognizerInputDataConfig& WithAugmentedManifests(AugmentedManifestsT&& value) {
    SetAugmentedManifests(std::forward<AugmentedManifestsT>(value));
    return *this;
  }
  template <typename AugmentedManifestsT = AugmentedManifestsListItem>
  EntityRecognizerInputDataConfig& AddAugmentedManifests(AugmentedManifestsT&& value) {
    m_augmentedManifestsHasBeenSet = true;
    m_augmentedManifests.emplace_back(std::forward<AugmentedManifestsT>(value));
    return *this;
  }
 private:
  EntityRecognizerDataFormat m_dataFormat{EntityRecognizerDataFormat::NOT_SET};
 
  Aws::Vector<EntityTypesListItem> m_entityTypes;
 
  EntityRecognizerDocuments m_documents;
 
  EntityRecognizerAnnotations m_annotations;
 
  EntityRecognizerEntityList m_entityList;
 
  Aws::Vector<AugmentedManifestsListItem> m_augmentedManifests;
  bool m_dataFormatHasBeenSet = false;
  bool m_entityTypesHasBeenSet = false;
  bool m_documentsHasBeenSet = false;
  bool m_annotationsHasBeenSet = false;
  bool m_entityListHasBeenSet = false;
  bool m_augmentedManifestsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace Comprehend
}  // namespace Aws