molframe-query 0.2.1

Typed and textual structural selection plans for molframe
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

//! Geometric selection productions: `within`, `around`, zones and layers.

use super::helpers::geometric_keyword;
use super::{GeometricKeyword, Parser};
use crate::ast::GeometricExpr;
use molframe_core::diagnostic::Diagnostic;

impl Parser {
    /// Parses a geometric prefix if the next value is a geometric keyword.
    ///
    /// Keyword recognition is ASCII case-insensitive and allocation-free.
    pub(super) fn parse_geometric(&mut self) -> Result<Option<GeometricExpr>, Diagnostic> {
        let Some(kind) = self.peek_value().and_then(geometric_keyword) else {
            return Ok(None);
        };

        let _ = self.take_value();

        let geometric = match kind {
            GeometricKeyword::Within => {
                let radius = self.number()?;
                self.require_keyword("of")?;

                GeometricExpr::Within {
                    radius,
                    target: Box::new(self.parse_modifier()?),
                }
            }
            GeometricKeyword::Beyond => {
                let radius = self.number()?;
                self.require_keyword("of")?;

                GeometricExpr::Beyond {
                    radius,
                    target: Box::new(self.parse_modifier()?),
                }
            }
            GeometricKeyword::Around => {
                let radius = self.number()?;

                GeometricExpr::Around {
                    radius,
                    target: Box::new(self.parse_modifier()?),
                }
            }
            GeometricKeyword::SphereZone => {
                let radius = self.number()?;

                GeometricExpr::SphereZone {
                    radius,
                    target: Box::new(self.parse_modifier()?),
                }
            }
            GeometricKeyword::SphereLayer => {
                let inner = self.number()?;
                let outer = self.number()?;

                GeometricExpr::SphereLayer {
                    inner,
                    outer,
                    target: Box::new(self.parse_modifier()?),
                }
            }
            GeometricKeyword::IsoLayer => {
                let inner = self.number()?;
                let outer = self.number()?;

                GeometricExpr::IsoLayer {
                    inner,
                    outer,
                    target: Box::new(self.parse_modifier()?),
                }
            }
            GeometricKeyword::CylinderZone => {
                let radius = self.number()?;
                let z_max = self.number()?;
                let z_min = self.number()?;

                GeometricExpr::CylinderZone {
                    radius,
                    z_max,
                    z_min,
                    target: Box::new(self.parse_modifier()?),
                }
            }
            GeometricKeyword::CylinderLayer => {
                let inner = self.number()?;
                let outer = self.number()?;
                let z_max = self.number()?;
                let z_min = self.number()?;

                GeometricExpr::CylinderLayer {
                    inner,
                    outer,
                    z_max,
                    z_min,
                    target: Box::new(self.parse_modifier()?),
                }
            }
            GeometricKeyword::Point => GeometricExpr::Point {
                point: [self.number()?, self.number()?, self.number()?],
                radius: self.number()?,
            },
        };

        Ok(Some(geometric))
    }
}