CMain.java

Go to the documentation of this file.

package org.lightvoting;

import com.google.common.collect.Sets;
import org.lightvoting.simulation.action.message.CSend;
import org.lightvoting.simulation.agent.CChairAgent;
import org.lightvoting.simulation.agent.CVotingAgent;
import org.lightvoting.simulation.environment.CEnvironment;
import org.yaml.snakeyaml.Yaml;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.Collection;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public final class CMain
{
    private static CEnvironment s_environment;
    private static int s_altnum;
    private static String s_grouping;
    private static String s_protocol;

    private CMain()
    {
    }

    public static void main( final String[] p_args ) throws Exception
    {
        // Example code taken from
        // https://lightjason.github.io/tutorial/tutorial-agentspeak-in-fifteen-minutes/
        //
        // parameter of the command-line arguments:
        // 1. ASL file
        // 2. number of agents
        // 3. number of iterations (if not set maximum)

        readYaml();

        final Set<CVotingAgent> l_agents;
        final CVotingAgent.CVotingAgentGenerator l_votingagentgenerator;

       // final CDataWriter l_writer = new CDataWriter();
        final String l_name = "results.h5";
        CDataWriter.createHDF5( l_name );
        CDataWriter.test( l_name );

        try
        {
            final FileInputStream l_stream = new FileInputStream( p_args[0] );
            final FileInputStream l_chairstream = new FileInputStream( p_args[1] );

            s_environment = new CEnvironment( Integer.parseInt( p_args[2] ), l_name );

            l_votingagentgenerator = new CVotingAgent.CVotingAgentGenerator( new CSend(), l_stream, s_environment, s_altnum, s_grouping, l_name );
            l_agents = l_votingagentgenerator
                    .generatemultiple( Integer.parseInt( p_args[2] ), new CChairAgent.CChairAgentGenerator( l_chairstream, s_environment, s_grouping, s_protocol, l_name )  )
                    .collect( Collectors.toSet() );


            System.out.println( " Numbers of agents: " + l_agents.size() );


        }
        catch ( final Exception l_exception )
        {
            l_exception.printStackTrace();
            throw new RuntimeException();
        }

        // generate empty set of active agents

        final Set<CVotingAgent> l_activeAgents = Sets.newConcurrentHashSet();

        System.out.println( " Numbers of active agents: " + l_activeAgents.size() );

        System.out.println( " Numbers of active agents: " + l_activeAgents.size() );
        System.out.println( " Will run " + p_args[3] + " cycles." );

        IntStream
            // define cycle range, i.e. number of cycles to run sequentially
            .range( 0,
                    p_args.length < 4
                    ? Integer.MAX_VALUE
                    : Integer.parseInt( p_args[3] ) )
            .forEach( j ->
            {
                System.out.println( "Global cycle: " + j );
                l_agents.parallelStream().forEach( i ->
                {
                    try
                    {
                        // check if the conditions for triggering a new cycle are fulfilled in the environment
                        // call each agent, i.e. trigger a new agent cycle
                        i.call();
                        //   i.getChair().sleep( 0 );
                        i.getChair().call();
                    }
                    catch ( final Exception l_exception )
                    {
                        l_exception.printStackTrace();
                        throw new RuntimeException();
                    }
                } );
            } );
    }

    @SuppressWarnings( "unchecked" )
    private static void readYaml() throws FileNotFoundException
    {
        final Yaml l_yaml = new Yaml();

        System.out.println( l_yaml.dump( l_yaml.load( new FileInputStream( "src/main/resources/org/lightvoting/configuration.yaml" ) ) ) );

        final Map<String, Map<String, String>> l_values = (Map<String, Map<String, String>>) l_yaml
            .load( new FileInputStream( "src/main/resources/org/lightvoting/configuration.yaml" ) );

        for ( final String l_key : l_values.keySet() )
        {
            final Map<String, String> l_subValues = l_values.get( l_key );
            System.out.println( l_key );

            for ( final String l_subValueKey : l_subValues.keySet() )
            {
                System.out.println( String.format( "\t%s = %s",
                                                   l_subValueKey, l_subValues.get( l_subValueKey ) ) );

                // parse input
                if ( "grouping".equals( l_subValueKey ) )
                    s_grouping = l_subValues.get( l_subValueKey );
                if ( "protocol".equals( l_subValueKey ) )
                    s_protocol = l_subValues.get( l_subValueKey );
                if ( "altnum".equals( l_subValueKey ) )
                    s_altnum = Integer.parseInt( l_subValues.get( l_subValueKey ) );
            }
        }
    }

    private static void addAgents( final Collection<CVotingAgent> p_activeAgents, final int p_newAgNum, final Iterator<CVotingAgent> p_agentIterator  )
    {

        for ( int i = 0; i < p_newAgNum; i++ )
        {
            if ( p_agentIterator.hasNext() )
            {
                final CVotingAgent l_curAg = p_agentIterator.next();
                p_activeAgents.add( l_curAg );
                System.out.println( "added Agent " + l_curAg.name() );
                p_agentIterator.remove();

            }
        }

    }


}