Se corrige un bug.

[z.facultad/75.42/plaqui.git] / Model / src / union.cpp

diff --git a/Model/src/union.cpp b/Model/src/union.cpp
index 331e9536245f099f0ce5f54dc2d9572783db5ead..443cb2bfbf0f78bdd2c8d09fbb9c805d273ba2b2 100644 (file)
--- a/Model/src/union.cpp
+++ b/Model/src/union.cpp
@@ -9,6 +9,8 @@ Union::Union(const std::string &_name):Transport(_name)
        in_slots = 2;
        out_slots = 1;
        max_flow = actual_flow = 0.0f;
+       in_on_zero = 0;
+       in_ready = 0;
 }
 
 Union::~Union()
@@ -19,34 +21,62 @@ void Union::recieve_msg(int msg, IConector *who, void *data)
 {
        int pos = OUT;
 
-       // Verifico si esta conectado a mi entrada
+       // Verifico si esta conectado a mi entrada o a mi salida
        std::list<IConector *>::iterator i;
        for(i=in_list.begin(); i!=in_list.end(); i++) {
                if ((*i) == who) pos = IN;
        }
        
        switch (msg) {
-               case MSG_QUERY_MAX_FLOW: {
+               case MSG_QUERY_MAX_FLOW_OUT: {
                        // Me preguntan por el flujo máximo.
                        // Primero me actualizo, y luego respondo
-                       update();
+                       float m_data =  *((float *)data)*2;
                        float tmp;
 
-                       tmp = (actual_flow<max_flow)?actual_flow:max_flow;
-                       if (pos == IN) {
-                                       // Si esta conectado a mi entrada, le puedo aceptar
-                                       // solo la mitad del flujo maximo
-                                       tmp /= 2;
+                       if (m_data == 0) {
+                               in_on_zero++;
+                               tmp = 0.0f;
+                               who->recieve_msg(MSG_RESPONSE_MAX_FLOW, this, &tmp);
+                               if (in_on_zero == 1) {
+                                       if (in_ready == 0)
+                                               tmp = max_flow;
+                                       else
+                                               tmp = max_flow/2.0f;
+                               }
+                               send_msg(OUT, MSG_QUERY_MAX_FLOW_OUT, &tmp);
+                       } else {
+                               switch (in_on_zero) {
+                                       case 0:
+                                               actual_flow = (m_data<actual_flow)?m_data:actual_flow;
+                                               actual_flow = (actual_flow<max_flow)?actual_flow:max_flow;
+                                       break;
+                                       case 1:
+                                               if (in_ready == 1)
+                                                       actual_flow = actual_flow/2.0f;
+                                               else
+                                                       actual_flow = max_flow/2.0f;
+                                       break;
+                                       case 2:
+                                               actual_flow = 0;
+                               }
+                               send_msg(OUT, MSG_QUERY_MAX_FLOW_OUT, &actual_flow);
+                               std::cout << in_on_zero << " " << actual_flow << std::endl;
+                               tmp = (in_on_zero==0)?actual_flow/2.0f:actual_flow;
+                               who->recieve_msg(MSG_RESPONSE_MAX_FLOW, this, &tmp);
                        }
-                       who->recieve_msg(MSG_RESPONSE_MAX_FLOW, this, &tmp);
+                       updated = true;
+                       if (pos == IN) in_ready++;
                }
                break;
                case MSG_RESPONSE_MAX_FLOW: {
                        float max = *((float *)data);
                        if (pos == OUT) {
                                if (max < actual_flow)  actual_flow = max;
+                               if (in_on_zero == 2) actual_flow = 0;
                        } else {
-                               if ((2*max) < actual_flow) actual_flow = 2*max;
+                                       if (((2*max) < actual_flow) && (max != 0))
+                                               actual_flow = 2*max;
                        }
                }
                break;
@@ -55,15 +85,21 @@ void Union::recieve_msg(int msg, IConector *who, void *data)
        }
 }
 
-void Union::update()
+void Union::update(int dir)
 {
        // Si ya me actualice, no lo tengo que hacer de nuevo
        if (updated) return;
        // Seteo mi actualizar en true para evitar entrar de nuevo
+       // FIXME : 99999 == INFINITO!!
        actual_flow = 99999;
        updated = true;
-       send_msg(OUT, MSG_QUERY_MAX_FLOW);
-       send_msg(IN, MSG_QUERY_MAX_FLOW);
+       switch (dir) {
+               case IN:
+                       send_msg(IN, MSG_QUERY_MAX_FLOW_IN, (void *)&max_flow);
+               break;
+               case OUT:
+                       send_msg(OUT, MSG_QUERY_MAX_FLOW_OUT, (void *)&max_flow);
+       }
 }
 
 void Union::simulate()
@@ -74,5 +110,8 @@ void Union::simulate()
 
        std::cout << name << "::Flujo actual = " << actual_flow << std::endl;
        updated = false;
+       actual_flow = 99999;
+       in_on_zero = 0;
+       in_ready = 0;
 }