"""
********************************************************************************
* Copyright (c) 2025 the Qrisp authors
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* This Source Code may also be made available under the following Secondary
* Licenses when the conditions for such availability set forth in the Eclipse
* Public License, v. 2.0 are satisfied: GNU General Public License, version 2
* with the GNU Classpath Exception which is
* available at https://www.gnu.org/software/classpath/license.html.
*
* SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
********************************************************************************
"""

from jax.extend.core import Primitive

class StimPrimitive(Primitive):
    """
    Base class for Stim-related JAX primitives.
    
    This class extends the JAX Primitive class to support Stim specific operations
    within the JASP / JAX tracing framework.
    """
    def __init__(self, name):
        Primitive.__init__(self, "stim." + name)
        self.stim_name = name
