Merge pull request #171 from leod/clear_trace_cache

Clear `Trace` cache after transpilation

Author: leod

src/sl/dag/trace.rs

@@ -22,10 +22,6 @@ impl Trace {
         Trace::Id(id)
     }
 
-    pub(crate) const fn c(f: fn() -> Rc<Expr>) -> Self {
-        Trace::Const(f)
-    }
-
     pub fn expr(&self) -> Rc<Expr> {
         use Trace::*;
 
@@ -34,6 +30,14 @@ impl Trace {
             Const(f) => f(),
         }
     }
+
+    pub(crate) const fn c(f: fn() -> Rc<Expr>) -> Self {
+        Trace::Const(f)
+    }
+
+    pub(crate) fn clear_cache() {
+        REGISTRY.with(|reg| *reg.borrow_mut() = Registry::default());
+    }
 }
 
 #[derive(Default)]

src/sl/transpile.rs

@@ -12,7 +12,7 @@ use crate::{
 
 use super::{
     codegen,
-    dag::{Expr, SamplerType, Type},
+    dag::{Expr, SamplerType, Trace, Type},
     primitives::value_arg,
     program_def::{ProgramDef, UniformBlockDef, UniformSamplerDef, VertexBlockDef},
     sig::{FromFsInput, FromVsInput, VsFunc, VsSig},
@@ -190,6 +190,8 @@ where
         source
     };
 
+    Trace::clear_cache();
+
     ProgramDef {
         uniform_block_defs,
         uniform_sampler_defs,