hu.bme.mit.theta.xta.dsl.XtaParameterSymbol Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of theta-xta Show documentation
Xta subproject in the Theta model checking framework
The newest version!
/*
 *  Copyright 2024 Budapest University of Technology and Economics
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package hu.bme.mit.theta.xta.dsl;

import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Preconditions.checkState;
import static java.util.stream.Collectors.toCollection;

import java.util.LinkedHashSet;
import java.util.Set;

import hu.bme.mit.theta.common.dsl.Env;
import hu.bme.mit.theta.common.dsl.Scope;
import hu.bme.mit.theta.common.dsl.Symbol;
import hu.bme.mit.theta.core.type.Expr;
import hu.bme.mit.theta.core.type.Type;
import hu.bme.mit.theta.xta.dsl.gen.XtaDslParser.ParameterIdContext;
import hu.bme.mit.theta.xta.dsl.gen.XtaDslParser.TypeContext;
import hu.bme.mit.theta.xta.utils.RangeType;

final class XtaParameterSymbol implements Symbol {

    private final String name;
    private final boolean reference;
    private final boolean constant;

    private final XtaType type;

    public XtaParameterSymbol(final Scope scope, final TypeContext typeContext,
                              final ParameterIdContext parameterIdContext) {
        checkNotNull(typeContext);
        checkNotNull(parameterIdContext);
        name = parameterIdContext.fArrayId.fId.getText();
        reference = (parameterIdContext.fRef != null);
        constant = (typeContext.fTypePrefix.fConst != null);
        type = new XtaType(scope, typeContext, parameterIdContext.fArrayId.fArrayIndexes);

        checkArgument(constant && !reference);
    }

    @Override
    public String getName() {
        return name;
    }

    public boolean isReference() {
        return reference;
    }

    public boolean isConstant() {
        return constant;
    }

    public Type instantiateType(final Env env) {
        checkNotNull(env);
        final Type paramType = type.instantiate(env);
        return paramType;
    }

    public Set> instantiateValues(final Env env) {
        checkState(!reference);
        final Type paramType = instantiateType(env);
        final RangeType rangeType = (RangeType) paramType;
        final Set> values = rangeType.values().collect(toCollection(LinkedHashSet::new));
        return values;
    }

}