org.btrplace.btrpsl.tree.ExportStatement Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2014 University Nice Sophia Antipolis
 *
 * This file is part of btrplace.
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */

package org.btrplace.btrpsl.tree;

import org.antlr.runtime.Token;
import org.btrplace.btrpsl.ANTLRBtrplaceSL2Parser;
import org.btrplace.btrpsl.ErrorReporter;
import org.btrplace.btrpsl.Script;
import org.btrplace.btrpsl.element.BtrpOperand;
import org.btrplace.btrpsl.element.BtrpSet;
import org.btrplace.btrpsl.element.IgnorableOperand;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * Statement to specify a list of variables to export.
 * This is only allowed if the script belong to a specified namespace.
 * When a variable is exported, it is exported using its current identifier
 * and its fully qualified name, with is the variable identifier prefixed by the namespace.
 *
 * @author Fabien Hermenier
 */
public class ExportStatement extends BtrPlaceTree {

    private Script script;

    /**
     * Make a new statement.
     *
     * @param t    the export token
     * @param scr  the script to alter with the variables to export
     * @param errs the list of errors
     */
    public ExportStatement(Token t, Script scr, ErrorReporter errs) {
        super(t, errs);
        this.script = scr;
    }

    @Override
    public BtrpOperand go(BtrPlaceTree parent) {

        Set scope = new HashSet<>();
        List toAdd = new ArrayList<>();

        for (int i = 0; i < getChildCount(); i++) {
            if (getChild(i).getType() == ANTLRBtrplaceSL2Parser.ENUM_VAR) {
                BtrpOperand r = getChild(i).go(this);
                if (r == IgnorableOperand.getInstance()) {
                    return r;
                }
                for (BtrpOperand o : ((BtrpSet) r).getValues()) {
                    toAdd.add(o);
                }
            } else if (getChild(i).getType() == ANTLRBtrplaceSL2Parser.VARIABLE) {
                toAdd.add(getChild(i).go(this));
            } else if (getChild(i).getType() == ANTLRBtrplaceSL2Parser.TIMES) {
                scope.add("*");
            } else if (getChild(i).getType() == ANTLRBtrplaceSL2Parser.IDENTIFIER) {
                scope.add(getChild(i).getText());
            } else {
                BtrpOperand e = getChild(i).go(this);
                if (e == IgnorableOperand.getInstance()) {
                    return e;
                }
                toAdd.addAll(flatten(e));
            }
        }
        try {
            for (BtrpOperand op : toAdd) {
                script.addExportable(op.label(), op, scope);
            }
        } catch (UnsupportedOperationException ex) {
            return ignoreError(ex.getMessage());
        }


        return IgnorableOperand.getInstance();
    }

    private static List flatten(BtrpOperand o) {
        List ret = new ArrayList<>();
        if (o.label() != null) {
            ret.add(o);
        } else {
            if (o.degree() == 0) {
                throw new UnsupportedOperationException(o + ": Variable expected");
            } else {
                BtrpSet s = (BtrpSet) o;
                for (BtrpOperand so : s.getValues()) {
                    ret.addAll(flatten(so));
                }
            }
        }
        return ret;
    }
}