58 lines
1.5 KiB
Rust
58 lines
1.5 KiB
Rust
|
use std::str::FromStr;
|
||
|
|
use crate::{PreprocessError, SourceOutput};
|
||
|
|
|
||
|
|
enum ActiveStage {
|
||
|
|
Both,
|
||
|
|
Fragment,
|
||
|
|
Vertex
|
||
|
|
}
|
||
|
|
|
||
|
|
impl FromStr for ActiveStage {
|
||
|
|
type Err = PreprocessError;
|
||
|
|
|
||
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
||
|
|
match s {
|
||
|
|
"vertex" => Ok(ActiveStage::Vertex),
|
||
|
|
"fragment" => Ok(ActiveStage::Fragment),
|
||
|
|
_ => Err(PreprocessError::InvalidStage)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Default)]
|
||
|
|
pub(crate) struct ShaderOutput {
|
||
|
|
pub(crate) fragment: String,
|
||
|
|
pub(crate) vertex: String
|
||
|
|
}
|
||
|
|
|
||
|
|
pub(crate) fn process_stages(source: &str) -> Result<ShaderOutput, PreprocessError> {
|
||
|
|
let mut active_stage = ActiveStage::Both;
|
||
|
|
let mut output = ShaderOutput::default();
|
||
|
|
|
||
|
|
for line in source.lines() {
|
||
|
|
if line.starts_with("#pragma stage ") {
|
||
|
|
let stage = line["#pragma stage ".len()..].trim();
|
||
|
|
active_stage = ActiveStage::from_str(stage)?;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
if line.starts_with("#pragma name ") || line.starts_with("#pragma format ") {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
match active_stage {
|
||
|
|
ActiveStage::Both => {
|
||
|
|
output.fragment.push_line(line);
|
||
|
|
output.vertex.push_line(line);
|
||
|
|
}
|
||
|
|
ActiveStage::Fragment => {
|
||
|
|
output.fragment.push_line(line);
|
||
|
|
}
|
||
|
|
ActiveStage::Vertex => {
|
||
|
|
output.vertex.push_line(line)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(output)
|
||
|
|
}
|