run_operation()

VisionExtractor::output_type MayaFlux::Yantra::VisionExtractor::run_operation ( const input_type &  input )

private

Definition at line 773 of file VisionExtractor.cpp.

{
    output_type output;
    output.metadata = input.metadata;
 
    const auto supplied = Kakshya::get_metadata_value<std::shared_ptr<Core::VKImage>>(input.metadata, "vision_image");
    const bool has_image = supplied && *supplied;
 
    if (!has_image && !input.data) {
        output.metadata["error"] = std::string("VisionExtractor: missing container");
        return output;
    }
 
    const auto analysis = Kakshya::get_metadata_value<Kinesis::Vision::VisionAnalysis>(
        input.metadata, "vision_analysis");
    if (!analysis) {
        output.metadata["error"] = std::string("VisionExtractor: missing vision_analysis metadata");
        return output;
    }
 
    const auto index = Kakshya::get_metadata_value<size_t>(input.metadata, "container_index").value_or(0);
    const auto image = has_image ? *supplied : resolve_image(input.data, index);
    if (!image) {
        output.metadata["error"] = std::string("VisionExtractor: could not resolve image from container");
        return output;
    }
 
    const uint32_t w = image->get_width();
    const uint32_t h = image->get_height();
 
    switch (m_mode) {
    case VisionExtractMode::Crop:
        output.metadata["vision_extraction"] = crop(image, *analysis, m_out_w, m_out_h);
        break;
    case VisionExtractMode::Sample:
        output.metadata["vision_extraction"] = sample(image, *analysis, w, h);
        break;
    case VisionExtractMode::Patches:
        output.metadata["vision_extraction"] = patches(image, *analysis, m_patch_w, m_patch_h, w, h);
        break;
    case VisionExtractMode::Crops:
        output.metadata["vision_extraction"] = crops(image, *analysis, m_tile_w, m_tile_h);
        break;
    case VisionExtractMode::SampleRegions:
        output.metadata["vision_extraction"] = sample_regions(image, *analysis, w, h);
        break;
    case VisionExtractMode::SamplePoints:
        output.metadata["vision_extraction"] = sample_points(image, *analysis, m_sample_radius, w, h);
        break;
    case VisionExtractMode::Silhouettes:
        output.metadata["vision_extraction"] = silhouettes(image, *analysis, m_tile_w, m_tile_h);
        break;
    case VisionExtractMode::Equalize:
        output.metadata["vision_extraction"] = equalize(image, *analysis, w, h);
        break;
    case VisionExtractMode::Expose:
        output.metadata["vision_extraction"] = expose(image, *analysis, m_exposure_target, w, h);
        break;
    case VisionExtractMode::Edges:
        output.metadata["vision_extraction"] = edges(image, *analysis, w, h);
        break;
    case VisionExtractMode::MotionMask:
        output.metadata["vision_extraction"] = motion_mask(image, *analysis, w, h);
        break;
    case VisionExtractMode::Annotate:
        output.metadata["vision_extraction"] = annotate(image, *analysis, w, h);
        break;
    case VisionExtractMode::SelectElement:
        if (const auto selected = select_element(*analysis, m_focus.value_or(ElementFocus {})))
            output.metadata["vision_extraction"] = *selected;
        break;
    }
 
    return output;
}

References MayaFlux::Yantra::Annotate, annotate(), MayaFlux::Yantra::Crop, crop(), MayaFlux::Yantra::Crops, crops(), MayaFlux::Yantra::Edges, edges(), MayaFlux::Yantra::Equalize, equalize(), MayaFlux::Yantra::Expose, expose(), h, image, index, input, m_exposure_target, m_focus, m_mode, m_out_h, m_out_w, m_patch_h, m_patch_w, m_sample_radius, m_tile_h, m_tile_w, MayaFlux::Yantra::Datum< T >::metadata, motion_mask(), MayaFlux::Yantra::MotionMask, MayaFlux::Yantra::Patches, patches(), resolve_image(), MayaFlux::Yantra::Sample, sample(), sample_points(), sample_regions(), MayaFlux::Yantra::SamplePoints, MayaFlux::Yantra::SampleRegions, select_element(), MayaFlux::Yantra::SelectElement, MayaFlux::Yantra::Silhouettes, and silhouettes().

Referenced by VisionExtractor().

Here is the call graph for this function:

Here is the caller graph for this function: