de.tuberlin.uebb.jdae.examples.BouncingBallArray.java Source code

Introduction

Here is the source code for de.tuberlin.uebb.jdae.examples.BouncingBallArray.java

Source

/*
 * Copyright (C) 2012 uebb.tu-berlin.de.
 *
 * This file is part of modim
 *
 * modim 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.
 *
 * modim 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 modim. If not, see <http://www.gnu.org/licenses/>.
 */

package de.tuberlin.uebb.jdae.examples;

import java.util.Collection;
import java.util.Map;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Iterables;

import de.tuberlin.uebb.jdae.dae.LoadableModel;
import de.tuberlin.uebb.jdae.hlmsl.Equation;
import de.tuberlin.uebb.jdae.hlmsl.Unknown;
import de.tuberlin.uebb.jdae.llmsl.GlobalVariable;
import de.tuberlin.uebb.jdae.llmsl.events.ContinuousEvent;
import de.tuberlin.uebb.jdae.simulation.SimulationRuntime;

public class BouncingBallArray implements LoadableModel {

    public final BouncingBallRadius[] balls = { null, null, null };
    public final double[] initial_h = { 5, 10, 20 };

    public BouncingBallArray(SimulationRuntime runtime) {
        for (int i = 1; i <= 3; ++i)
            balls[i - 1] = new BouncingBallRadius(i, runtime, initial_h[i - 1]);
    }

    @Override
    public Map<GlobalVariable, Double> initials(Map<Unknown, GlobalVariable> ctxt) {
        return ImmutableMap.of(ctxt.get(balls[0].h), initial_h[0], ctxt.get(balls[1].h), initial_h[1],
                ctxt.get(balls[2].h), initial_h[2]);
    }

    @Override
    public Collection<Equation> initialEquations() {
        return ImmutableList.copyOf(Iterables.concat(balls[0].initialEquations(), balls[1].initialEquations(),
                balls[2].initialEquations()));
    }

    @Override
    public Collection<Equation> equations() {
        return ImmutableList
                .copyOf(Iterables.concat(balls[2].equations(), balls[1].equations(), balls[0].equations()));
    }

    @Override
    public String name() {
        return "Array of bouncing balls";
    }

    @Override
    public Collection<ContinuousEvent> events(Map<Unknown, GlobalVariable> ctxt) {
        return ImmutableList
                .copyOf(Iterables.concat(balls[0].events(ctxt), balls[1].events(ctxt), balls[2].events(ctxt)));
    }

}