io.github.agentsoz.jill.example.greeter.Greeter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jill-examples Show documentation
Jill cognitive agents (examples)
There is a newer version: 0.4.0
package io.github.agentsoz.jill.example.greeter;

/*
 * #%L
 * Jill Cognitive Agents Platform
 * %%
 * Copyright (C) 2014 - 2015 by its authors. See AUTHORS file.
 * %%
 * This program 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 program 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 General Lesser Public License for more details.
 * 
 * You should have received a copy of the GNU General Lesser Public
 * License along with this program.  If not, see
 * .
 * #L%
 */

import io.github.agentsoz.jill.core.beliefbase.BeliefBaseException;
import io.github.agentsoz.jill.core.beliefbase.BeliefSetField;
import io.github.agentsoz.jill.lang.Agent;
import io.github.agentsoz.jill.lang.AgentInfo;
import io.github.agentsoz.jill.util.Log;

import java.io.PrintStream;
import java.util.Random;

@AgentInfo(hasGoals={"io.github.agentsoz.jill.example.greeter.BeFriendly"})
public class Greeter extends Agent {

	// Defaults
	private static Random rand = new Random();
	private static int numNeighbours = 1;
	private static final String beliefset = "neighbour";
	public Greeter(String name) {
		super(name);
	}
	
	@Override
	public void start(PrintStream writer, String[] params) {
		// Parse the arguments
		parse(params);
		
		// Create a new belief set about neighbours
		BeliefSetField[] fields = {
				new BeliefSetField("name", String.class, true),
				new BeliefSetField("gender", String.class, false),
		};

		try {
			// Attach this belief set to this agent
			this.createBeliefSet(beliefset, fields);
		
			// Add beliefs about neighbours
			registerNeighbours(rand, numNeighbours);
			Log.debug("Agent " + getName() + " is initialising with " + numNeighbours + " neighbours");

			// Post the goal to be friendly
			post(new BeFriendly("BeFriendly"));
		} catch (BeliefBaseException e) {
			Log.error(e.getMessage());
		}
	}

	/**
	 * Helper function to add beliefs about neighbours
	 * @param rand random number generator to use
	 * @param count number of beliefs to add
	 * @throws BeliefBaseException 
	 */
	private void registerNeighbours(Random rand, int count) throws BeliefBaseException {
		final String[] MALES = {
				"Alex", "Daniel", "John",
				"Lionel", "Nick", "Oscar", "Paul",
				"Rod", "Sam", "Tom"
		};
		final String[] FEMALES = {
				"Alice", "Elisa", "Fiona", 
				"Julia", "Kate", "Laura", "Margaret", 
				"Nancy", "Pam", "Rachael"
		};
		final String[] MIDDLE = {
				"A.", "B.", "C.", "D.", "E.", "F.", "G.", "H.", "I.", "J.", 
				"K.", "L.", "M.", "N.", "O.", "P.", "Q.", "R.", "S.", "T.", 
				"U.", "V.", "W.", "X.", "Y.", "Z." 
		};
		final String[] SURNAMES = {
				"Anderson", "Brown", "Jones", "Martin", "Morton", 
				"Smith", "Taylor", "White", "Williams", "Wilson",
		};
		int size = (count < 0) ? 0 : count;
		for (int i = 0; i < size; i++) {
			boolean male = (rand.nextDouble() < 0.5) ? true : false;
			String name = male ? MALES[rand.nextInt(MALES.length)] : FEMALES[rand.nextInt(FEMALES.length)];
			name += " " + MIDDLE[rand.nextInt(MIDDLE.length)] + " ";
			name += SURNAMES[rand.nextInt(SURNAMES.length)]; 
			String gender = male ? "male" : "female";
			this.addBelief(beliefset, name, gender);
		}
	}
	
    public static void parse(String[] args) {
        for (int i = 0; i < args.length; i++) {
                switch (args[i]) {
                case "-seed":
                	if (i + 1 < args.length) {
                		i++;
                		int seed = 0;
                		try {
                			seed = Integer.parseInt(args[i]);
                    		rand = new Random(seed);
                		} catch (Exception e) {
                			Log.warn("Seed value '" + args[i] + "' is not a number");
                		}
                	}
                	break;
                case "-neighbourhoodSize":
                	if (i + 1 < args.length) {
                		i++;
                      	try {
                       		numNeighbours = Integer.parseInt(args[i]);
                       	} catch (Exception e) {
                       		Log.warn("Neighbourhood size value '" + args[i] + "' is not a number");
                       	}
                	}
                	break;
                }
        }
    }
}