Package com.asakusafw.compiler.flow.stage.StageModel

Examples of com.asakusafw.compiler.flow.stage.StageModel.Fragment


    public void op2() {
        List<StageModel> stages = compile(CoGroupFlowOp2.class);
        StageModel stage = stages.get(0);
        Assume.assumeThat(stage.getReduceUnits().size(), is(1));
        ReduceUnit reduce = stage.getReduceUnits().get(0);
        Fragment fragment = reduce.getFragments().get(0);
        Name name = fragment.getCompiled().getQualifiedName();

        ClassLoader loader = start();
        PortMapper mapper = new PortMapper(fragment);
        MockResult<Ex1> r1 = mapper.create("r1");
        MockResult<Ex2> r2 = mapper.create("r2");

        @SuppressWarnings("unchecked")
        Rendezvous<Writable> f = (Rendezvous<Writable>) create(loader, name, mapper.toArguments());

        Segment s1 = stage.getShuffleModel().findSegment(fragment.getInputPorts().get(0));
        Segment s2 = stage.getShuffleModel().findSegment(fragment.getInputPorts().get(1));
        SegmentedWritable value = createShuffleValue(loader, stage);

        Ex1 ex1 = new Ex1();
        Ex2 ex2 = new Ex2();
        ex1.setStringAsString("string");
View Full Code Here


    public void op3() {
        List<StageModel> stages = compile(CoGroupFlowOp3.class);
        StageModel stage = stages.get(0);
        Assume.assumeThat(stage.getReduceUnits().size(), is(1));
        ReduceUnit reduce = stage.getReduceUnits().get(0);
        Fragment fragment = reduce.getFragments().get(0);
        Name name = fragment.getCompiled().getQualifiedName();

        ClassLoader loader = start();
        PortMapper mapper = new PortMapper(fragment);
        MockResult<Ex1> r1 = mapper.create("r1");
        MockResult<Ex1> r2 = mapper.create("r2");
        MockResult<Ex1> r3 = mapper.create("r3");

        @SuppressWarnings("unchecked")
        Rendezvous<Writable> f = (Rendezvous<Writable>) create(loader, name, mapper.toArguments());

        Segment s1 = stage.getShuffleModel().findSegment(fragment.getInputPorts().get(0));
        Segment s2 = stage.getShuffleModel().findSegment(fragment.getInputPorts().get(1));
        Segment s3 = stage.getShuffleModel().findSegment(fragment.getInputPorts().get(2));
        SegmentedWritable value = createShuffleValue(loader, stage);

        Ex1 ex1 = new Ex1();
        ex1.setStringAsString("string");
View Full Code Here

    public void withParameter() {
        List<StageModel> stages = compile(CoGroupFlowWithParameter.class);
        StageModel stage = stages.get(0);
        Assume.assumeThat(stage.getReduceUnits().size(), is(1));
        ReduceUnit reduce = stage.getReduceUnits().get(0);
        Fragment fragment = reduce.getFragments().get(0);
        Name name = fragment.getCompiled().getQualifiedName();

        ClassLoader loader = start();
        PortMapper mapper = new PortMapper(fragment);
        MockResult<Ex1> result = mapper.create("r1");

        @SuppressWarnings("unchecked")
        Rendezvous<Writable> f = (Rendezvous<Writable>) create(loader, name, mapper.toArguments());

        Segment segment = stage.getShuffleModel().findSegment(fragment.getInputPorts().get(0));
        SegmentedWritable value = createShuffleValue(loader, stage);

        Ex1 ex1 = new Ex1();
        ex1.setStringAsString("string");
View Full Code Here

    public void swap() {
        List<StageModel> stages = compile(CoGroupFlowSwap.class);
        StageModel stage = stages.get(0);
        Assume.assumeThat(stage.getReduceUnits().size(), is(1));
        ReduceUnit reduce = stage.getReduceUnits().get(0);
        Fragment fragment = reduce.getFragments().get(0);
        Name name = fragment.getCompiled().getQualifiedName();

        ClassLoader loader = start();
        PortMapper mapper = new PortMapper(fragment);
        MockResult<Ex1> result = mapper.create("r1");

        @SuppressWarnings("unchecked")
        Rendezvous<Writable> f = (Rendezvous<Writable>) create(loader, name, mapper.toArguments());

        Segment segment = stage.getShuffleModel().findSegment(fragment.getInputPorts().get(0));
        SegmentedWritable value = createShuffleValue(loader, stage);

        Ex1 ex1 = new Ex1();
        ex1.setStringAsString("string");
View Full Code Here

        assertThat(analyzed.getReduceUnits().size(), is(0));

        MapUnit map = analyzed.getMapUnits().get(0);
        assertThat(map.getFragments().size(), is(1));

        Fragment fragment = map.getFragments().get(0);
        assertThat(fragment.isRendezvous(), is(false));
        assertThat(fragment.getInputPorts().size(), is(1));
        assertThat(fragment.getOutputPorts().size(), is(1));
        assertThat(fragment.getFactors().size(), is(3));
    }
View Full Code Here

     * () -> (in) -> (r1)
     */
    @Test
    public void op1() {
        List<StageModel> stages = compile(ExtractFlowOp1.class);
        Fragment fragment = stages.get(0).getMapUnits().get(0).getFragments().get(0);
        Name name = fragment.getCompiled().getQualifiedName();

        ClassLoader loader = start();
        PortMapper mapper = new PortMapper(fragment);
        MockResult<Ex1> out1 = mapper.create("r1");

View Full Code Here

     * () -> (in) -> (r1, r2)
     */
    @Test
    public void op2() {
        List<StageModel> stages = compile(ExtractFlowOp2.class);
        Fragment fragment = stages.get(0).getMapUnits().get(0).getFragments().get(0);
        Name name = fragment.getCompiled().getQualifiedName();

        ClassLoader loader = start();
        PortMapper mapper = new PortMapper(fragment);
        MockResult<Ex1> out1 = mapper.create("r1");
        MockResult<Ex2> out2 = mapper.create("r2");
View Full Code Here

     * () -> (in) -> (r1, r2, r3)
     */
    @Test
    public void op3() {
        List<StageModel> stages = compile(ExtractFlowOp3.class);
        Fragment fragment = stages.get(0).getMapUnits().get(0).getFragments().get(0);
        Name name = fragment.getCompiled().getQualifiedName();

        ClassLoader loader = start();
        PortMapper mapper = new PortMapper(fragment);
        MockResult<Ex1> out1 = mapper.create("r1");
        MockResult<Ex2> out2 = mapper.create("r2");
View Full Code Here

     * (100) -> (in) -> (r1)
     */
    @Test
    public void withParameter() {
        List<StageModel> stages = compile(ExtractFlowWithParameter.class);
        Fragment fragment = stages.get(0).getMapUnits().get(0).getFragments().get(0);
        Name name = fragment.getCompiled().getQualifiedName();

        ClassLoader loader = start();
        PortMapper mapper = new PortMapper(fragment);
        MockResult<Ex2> out2 = mapper.create("r1");

View Full Code Here

    @Test
    public void simple() throws Exception {
        StageModel analyzed = mr(SimpleShuffleStage.class);
        ReduceUnit red = analyzed.getReduceUnits().get(0);
        assertThat(red.getFragments().size(), is(1));
        Fragment fragment = red.getFragments().get(0);

        ReduceFragmentEmitter emitter = new ReduceFragmentEmitter(environment);
        ShuffleModel shuffle = analyzed.getShuffleModel();
        CompiledType name = emitter.emit(fragment, shuffle, analyzed.getStageBlock());
View Full Code Here

TOP

Related Classes of com.asakusafw.compiler.flow.stage.StageModel.Fragment

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.