TestRBSolve.java

Go to the documentation of this file.

package rb.test;

import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import jarmos.SimulationResult;
import jarmos.geometry.GeometryData;
import jarmos.io.AModelManager.ModelManagerException;
import jarmos.io.FileModelManager;
import jarmos.visual.ColorGenerator;
import jarmos.visual.VisualizationData;

import org.junit.Test;

import rb.RBContainer;
import rb.RBSystem;

public class TestRBSolve {

    // @Test
    public void testRBSolveJRB() {
        FileModelManager f = new FileModelManager("models");
        try {
            f.useModel("rbm_advec");
        } catch (ModelManagerException e) {
            e.printStackTrace();
            fail(e.getMessage());
        }

        RBContainer rb = new RBContainer();
        assertTrue(rb.loadModel(f));

        // Perform the solve
        RBSystem s = rb.mRbSystem;
        // double[] par = s.getParams().getRandomParam();
        double[] par = new double[] { .5, .5 };
        s.getParams().setCurrent(par);
        s.computeRBSolution(4);
    }

    // @Test
    public void testRBSolveJRB_TimeConst() {
        FileModelManager f = new FileModelManager("models");
        try {
            f.useModel("rbm_advec");
        } catch (ModelManagerException e) {
            e.printStackTrace();
            fail(e.getMessage());
        }

        RBContainer rb = new RBContainer();
        assertTrue(rb.loadModel(f));

        // Perform the solve
        RBSystem s = rb.mRbSystem;
        // double[] par = s.getParams().getRandomParam();
        double[] par = new double[] { .5, .5, .5 };
        s.getParams().setCurrent(par);
        s.computeRBSolution(4);

        SimulationResult sol = s.getSimulationResults();

        GeometryData g = new GeometryData();
        g.loadModelGeometry(f);
        VisualizationData d = new VisualizationData(g);
        d.useResult(sol);
        d.computeVisualFeatures(new ColorGenerator());
    }

    @Test
    public void testRBSolverbappmit() {
        FileModelManager f = new FileModelManager("models");
        try {
            f.useModel("demo3");
        } catch (ModelManagerException e) {
            e.printStackTrace();
            fail(e.getMessage());
        }

        RBContainer rb = new RBContainer();
        assertTrue(rb.loadModel(f));

        // Perform the solve
        RBSystem s = rb.mRbSystem;
        double[] par = s.getParams().getRandomParam();
        // double[] par = new double[]{.5, .5};
        s.getParams().setCurrent(par);
        s.computeRBSolution(s.getNBF() / 2);

        SimulationResult res = s.getSimulationResults();
        GeometryData g = rb.mRbSystem.getGeometry();
        VisualizationData v = new VisualizationData(g);
        v.useResult(res);

        v.computeVisualFeatures(new ColorGenerator());
    }

}