..

Author: dweindl

python/sdist/amici/de_model.py

@@ -9,6 +9,7 @@
 import re
 from collections.abc import Callable, Sequence
 from itertools import chain
+from operator import itemgetter
 from typing import TYPE_CHECKING
 
 import numpy as np
@@ -798,12 +799,16 @@ def num_events(self) -> int:
 
     def num_events_solver(self) -> int:
         """
-        Number of Events.
+        Number of Events that rely on numerical root-finding.
 
         :return:
             number of event symbols (length of the root vector in AMICI)
         """
-        constant_syms = set(self.sym("k")) | set(self.sym("p"))
+        constant_syms = (
+            set(self.sym("k"))
+            | set(self.sym("p"))
+            | set(itemgetter(*self.static_indices("w"))(self.sym("w")))
+        )
         return sum(
             not event.has_explicit_trigger_times(constant_syms)
             for event in self.events()
@@ -1307,13 +1312,12 @@ def parse_events(self) -> None:
             # add roots of heaviside functions
             self.add_component(root)
 
-        # # Substitute 'w' expressions into root expressions, to avoid rewriting
-        # # 'root.cpp' and 'stau.cpp' headers to include 'w.h'.
-        # for event in self.events():
-        #     event.set_val(event.get_val().subs(w_toposorted))
-
         # re-order events - first those that require root tracking, then the others
-        constant_syms = set(self.sym("k")) | set(self.sym("p"))
+        constant_syms = (
+            set(self.sym("k"))
+            | set(self.sym("p"))
+            | set(itemgetter(*self.static_indices("w"))(self.sym("w")))
+        )
         self._events = list(
             chain(
                 itertools.filterfalse(
@@ -2506,11 +2510,6 @@ def _process_heavisides(
         heavisides = []
         # run through the expression tree and get the roots
         tmp_roots_old = self._collect_heaviside_roots((dxdt,))
-        # TODO remove: substitute 'w' symbols in the root expression by their equations,
-        #  because currently,
-        #    1) root functions must not depend on 'w'
-        # FIXME 2) the check for time-dependence currently assumes only state
-        #     variables are implicitly time-dependent
         for tmp_root_old, tmp_x0_old in unique_preserve_order(tmp_roots_old):
             # we want unique identifiers for the roots
             tmp_root_new = self._get_unique_root(tmp_root_old, roots)