Model/src/tank.cpp

   1
   2 #include "tank.h"
   3 #include <iostream>
   4 #include "condition.h"
   5
   6 using namespace PlaQui::Model;
   7
   8 Tank::Tank(const std::string &_name):Source(_name),Drain(_name),Control(_name)
   9 {
  10         litros = 0.0f;
  11
  12         /* Utilizo en input como el flotante de arriba y el output como el de abajo */
  13         input = new Condition(Condition::GT, 0.9, this);
  14         output = new Condition(Condition::LT, 0.1, this);
  15 }
  16
  17 Tank::~Tank()
  18 {
  19         delete input;
  20         delete output;
  21 }
  22
  23 bool Tank::get_output()
  24 {
  25         return litros > 0;
  26 }
  27
  28 void Tank::update(int dir)
  29 {
  30         updated = true;
  31 }
  32
  33 void Tank::simulate()
  34 {
  35         if (!updated) return;
  36
  37         // le sumo lo que recibo
  38         litros += actual_in_flow;
  39         // calculo el nuevo color
  40
  41         // le resto lo que entrego
  42         litros -= actual_out_flow;
  43
  44 #ifdef DEBUG
  45         std::cout << name << "Capacidad: " << capacity;
  46         std::cout << "  Litros : " << litros << std::endl;
  47 #endif
  48         updated = false;
  49         color_updated = false;
  50 }
  51
  52 void Tank::recieve_msg(int msg, IConector *who, void *data)
  53 {
  54         int pos = OUT;
  55
  56         // Verifico si esta conectado a mi entrada
  57         std::list<IConector *>::iterator i;
  58         for(i=in_list.begin(); i!=in_list.end(); i++) {
  59                 if ((*i) == who) pos = IN;
  60         }
  61
  62         switch (msg) {
  63                 case MSG_QUERY_MAX_FLOW_OUT:
  64                         if (updated) {
  65                                 who->recieve_msg(MSG_RESPONSE_MAX_FLOW, this, &actual_in_flow);
  66                         }
  67                         actual_in_flow = capacity - litros;
  68                         if (*((float *)data) < actual_in_flow)
  69                                 actual_in_flow = *((float *)data);
  70                         actual_flow = actual_out_flow = litros;
  71                         updated = true;
  72                         send_msg(OUT, MSG_QUERY_MAX_FLOW_OUT, &actual_out_flow);
  73
  74                         who->recieve_msg(MSG_RESPONSE_MAX_FLOW, this, &actual_in_flow);
  75                 break;
  76                 case MSG_RESPONSE_MAX_FLOW:
  77                         if (pos == OUT) {
  78                                 if (*((float *)data) < actual_out_flow)
  79                                         actual_flow = actual_out_flow = *((float *)data);
  80                         }
  81                 break;
  82                 case MSG_RESPONSE_COLOR:
  83                 {
  84                         RGB c = *((RGB *)data);
  85                         PlantItem *ic = static_cast<PlantItem *>(*(in_list.begin()));
  86                         int r,g,b;
  87                         float total = litros + ic->get_actual_flow();
  88
  89                         r = (int)(get_color().r()*litros/total + c.r()*ic->get_actual_flow()/total);
  90                         g = (int)(get_color().g()*litros/total + c.g()*ic->get_actual_flow()/total);
  91                         b = (int)(get_color().b()*litros/total + c.b()*ic->get_actual_flow()/total);
  92                         r %= 256;
  93                         g %= 256;
  94                         b %= 256;
  95                         set_color(RGB(r,g,b));
  96                         color_updated = true;
  97                 }
  98                 break;
  99                 default:
 100                         Control::recieve_msg(msg, who, data);
 101         }
 102 }
 103
 104 void Tank::get_state_as_xml(std::stringstream &out)
 105 {
 106         PlantItem::get_state_as_xml(out);
 107
 108         out << "\t<tank name=\"" << name << "\">" << std::endl;
 109         out << "\t\t<capacity>" << capacity << "</capacity>" << std::endl;
 110         out << "\t\t<litros>" << litros << "</litros>" << std::endl;
 111         out << "\t</tank>" << std::endl;
 112 }