//----------------------------------------------------------------------------
// Copyright (C) 2011 Ingrid Nunes
//
// This library 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 2.1 of the License, or (at your option) any later version.
//
// This library 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 this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//
// To contact the authors:
// http://inf.ufrgs.br/prosoft/bdi4jade/
//
//----------------------------------------------------------------------------
package br.ufrgs.inf.bdinetr.agent;
import jade.content.ContentElement;
import jade.content.lang.sl.SLCodec;
import jade.core.AID;
import jade.domain.DFService;
import jade.domain.FIPAException;
import jade.domain.FIPANames.ContentLanguage;
import jade.domain.FIPAAgentManagement.DFAgentDescription;
import jade.domain.FIPAAgentManagement.ServiceDescription;
import jade.lang.acl.ACLMessage;
import jade.lang.acl.MessageTemplate;
import java.util.HashSet;
import java.util.Set;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import bdi4jade.belief.Belief;
import bdi4jade.goal.BeliefGoal;
import bdi4jade.plan.Plan.EndState;
import bdi4jade.plan.planbody.BeliefGoalPlanBody;
import br.ufrgs.inf.bdinetr.domain.Role;
import br.ufrgs.inf.bdinetr.domain.ontology.BDINetROntology;
/**
* @author Ingrid Nunes
*/
public class RequestBeliefGoalPlanBody extends BeliefGoalPlanBody {
public static final int MSG_TIME_OUT = 100;
public static final int ANSWER_TIME_OUT = 1000;
public static final String LANGUAGE = new SLCodec().getName();
private enum State {
Resquesting, ReceivingResponses, Selecting, AchievingGoal, Ended;
}
private static final Log log = LogFactory
.getLog(RequestBeliefGoalPlanBody.class);
private static final long serialVersionUID = -1833810388789537049L;
private MessageTemplate mt;
private State state;
private long requestTime;
private Set<ACLMessage> positiveAnswers;
private Set<AID> receivers;
private int answers;
@Override
protected void execute() {
try {
switch (state) {
case Resquesting:
ACLMessage msg = new ACLMessage(ACLMessage.CFP);
msg.setLanguage(LANGUAGE);
msg.setOntology(BDINetROntology.ONTOLOGY_NAME);
myAgent.getContentManager().fillContent(msg, getGoal());
// FIXME send request to specific role
receivers = new HashSet<>();
for (Role role : Role.values()) {
for (DFAgentDescription agentDesc : getReceivers(role)) {
if (receivers.add(agentDesc.getName())) {
msg.addReceiver(agentDesc.getName());
}
}
}
log.info(receivers);
this.requestTime = System.currentTimeMillis();
msg.setConversationId("cin" + requestTime);
msg.setReplyWith("cfp" + requestTime);
myAgent.send(msg);
this.mt = MessageTemplate.and(MessageTemplate
.MatchConversationId(msg.getConversationId()),
MessageTemplate.MatchInReplyTo(msg.getReplyWith()));
this.answers = 0;
this.positiveAnswers = new HashSet<>();
this.state = State.ReceivingResponses;
break;
case ReceivingResponses:
ACLMessage reply = myAgent.receive(mt);
if (reply != null) {
this.answers++;
if (ACLMessage.PROPOSE == reply.getPerformative()) {
log.info("Agent " + reply.getSender()
+ " sent a proposal");
positiveAnswers.add(reply);
} else {
log.info("Agent " + reply.getSender()
+ " refused the request");
}
} else {
block(MSG_TIME_OUT);
}
if (answers >= receivers.size()
|| (System.currentTimeMillis() - requestTime) >= ANSWER_TIME_OUT) {
this.state = State.Selecting;
} else {
log.info("Waiting for more answers...");
}
break;
case Selecting:
if (positiveAnswers.isEmpty()) {
log.info("No positive answers.");
setEndState(EndState.FAILED);
this.state = State.Ended;
return;
} else {
// TODO Better selection process.
ACLMessage chosenMsg = positiveAnswers.iterator().next();
for (ACLMessage answer : positiveAnswers) {
reply = answer.createReply();
if (answer == chosenMsg) {
reply.setPerformative(ACLMessage.ACCEPT_PROPOSAL);
reply.setReplyWith("cfp"
+ System.currentTimeMillis());
this.mt = MessageTemplate.and(MessageTemplate
.MatchConversationId(reply
.getConversationId()),
MessageTemplate.MatchInReplyTo(reply
.getReplyWith()));
log.info("Accepted proposal of agent: "
+ answer.getSender());
} else {
reply.setPerformative(ACLMessage.REJECT_PROPOSAL);
log.info("Rejected proposal of agent: "
+ answer.getSender());
}
this.myAgent.send(reply);
}
this.state = State.AchievingGoal;
}
break;
case AchievingGoal:
reply = myAgent.receive(mt);
if (reply != null) {
if (ACLMessage.INFORM == reply.getPerformative()) {
ContentElement content = myAgent.getContentManager()
.extractContent(reply);
getBeliefBase().addBelief((Belief<?, ?>) content);
assert ((BeliefGoal<?>) getGoal())
.isAchieved(getBeliefBase());
log.info("Goal " + getGoal() + " achieved.");
} else {
setEndState(EndState.FAILED);
}
this.state = State.Ended;
} else {
block();
}
break;
case Ended:
break;
}
} catch (Exception exc) {
log.error(exc);
exc.printStackTrace();
setEndState(EndState.FAILED);
}
}
@Override
public void onStart() {
this.state = State.Resquesting;
}
private DFAgentDescription[] getReceivers(Role role) {
DFAgentDescription dfd = new DFAgentDescription();
ServiceDescription sd = new ServiceDescription();
sd.setType(role.name());
sd.addLanguages(ContentLanguage.FIPA_SL);
sd.addOntologies(BDINetROntology.ONTOLOGY_NAME);
dfd.addServices(sd);
try {
return DFService.search(myAgent, dfd);
} catch (FIPAException fe) {
log.error(myAgent.getLocalName()
+ " search with DF unsucceeded. Reason: " + fe.getMessage());
log.error(fe);
fe.printStackTrace();
return new DFAgentDescription[0];
}
}
}