re_space_view_text_document 0.15.0-alpha.3

A simple Space View that shows a single text box.
Documentation
use re_data_store::LatestAtQuery;
use re_query::{query_archetype, QueryError};
use re_types::{
    archetypes::{self, TextDocument},
    components,
};
use re_viewer_context::{
    IdentifiedViewSystem, SpaceViewSystemExecutionError, ViewContextCollection, ViewQuery,
    ViewerContext, VisualizerQueryInfo, VisualizerSystem,
};

// ---

#[derive(Debug, Clone)]
pub struct TextDocumentEntry {
    pub body: components::Text,
    pub media_type: components::MediaType,
}

/// A text scene, with everything needed to render it.
#[derive(Default)]
pub struct TextDocumentSystem {
    pub text_entries: Vec<TextDocumentEntry>,
}

impl IdentifiedViewSystem for TextDocumentSystem {
    fn identifier() -> re_viewer_context::ViewSystemIdentifier {
        "TextDocument".into()
    }
}

impl VisualizerSystem for TextDocumentSystem {
    fn visualizer_query_info(&self) -> VisualizerQueryInfo {
        VisualizerQueryInfo::from_archetype::<TextDocument>()
    }

    fn execute(
        &mut self,
        ctx: &ViewerContext<'_>,
        query: &ViewQuery<'_>,
        _view_ctx: &ViewContextCollection,
    ) -> Result<Vec<re_renderer::QueueableDrawData>, SpaceViewSystemExecutionError> {
        let store = ctx.entity_db.store();

        let timeline_query = LatestAtQuery::new(query.timeline, query.latest_at);

        for data_result in query.iter_visible_data_results(ctx, Self::identifier()) {
            // TODO(#3320): this match can go away once the issue is resolved
            match query_archetype::<archetypes::TextDocument>(
                store,
                &timeline_query,
                &data_result.entity_path,
            ) {
                Ok(arch_view) => {
                    let bodies = arch_view.iter_required_component::<components::Text>()?;
                    let media_types =
                        arch_view.iter_optional_component::<components::MediaType>()?;

                    for (body, media_type) in itertools::izip!(bodies, media_types) {
                        let media_type = media_type.unwrap_or(components::MediaType::plain_text());
                        self.text_entries
                            .push(TextDocumentEntry { body, media_type });
                    }
                }
                Err(QueryError::PrimaryNotFound(_)) => {}
                Err(err) => {
                    re_log::error_once!(
                        "Unexpected error querying {:?}: {err}",
                        &data_result.entity_path
                    );
                }
            };
        }

        Ok(Vec::new())
    }

    fn as_any(&self) -> &dyn std::any::Any {
        self
    }
}