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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of btrpsl Show documentation

The btrplace specification language (btrpsl) allows to express constraints related to the placement of virtual machines in a datacenters. This language is dedicated to datacenters administrators and applications administrators that use Btrplace(http://btrp.inria.fr) to manage their nodes and virtual machines.

There is a newer version: 1.12.10

Show newest version

/*
 * Copyright (c) 2016 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.ErrorReporter;
import org.btrplace.btrpsl.Script;
import org.btrplace.btrpsl.antlr.ANTLRBtrplaceSL2Parser;
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);
        }


        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");
            }
            BtrpSet s = (BtrpSet) o;
            for (BtrpOperand so : s.getValues()) {
                ret.addAll(flatten(so));
            }
        }
        return ret;
    }
}