lbessard@16: /*
lbessard@16:  * (c) 2003 Mario de Sousa
lbessard@16:  *
lbessard@16:  * Offered to the public under the terms of the GNU General Public License
lbessard@16:  * as published by the Free Software Foundation; either version 2 of the
lbessard@16:  * License, or (at your option) any later version.
lbessard@16:  *
lbessard@16:  * This program is distributed in the hope that it will be useful, but
lbessard@16:  * WITHOUT ANY WARRANTY; without even the implied warranty of
lbessard@16:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
lbessard@16:  * Public License for more details.
lbessard@16:  *
lbessard@16:  * This code is made available on the understanding that it will not be
lbessard@16:  * used in safety-critical situations without a full and competent review.
lbessard@16:  */
lbessard@16: 
lbessard@16: /*
lbessard@16:  * An IEC 61131-3 IL and ST compiler.
lbessard@16:  *
lbessard@16:  * Based on the
lbessard@16:  * FINAL DRAFT - IEC 61131-3, 2nd Ed. (2001-12-10)
lbessard@16:  *
lbessard@16:  */
lbessard@16: 
lbessard@16: 
lbessard@16: /* Determine the data type of an ST expression.
lbessard@16:  * A reference to the relevant type definition is returned.
lbessard@16:  *
lbessard@16:  * For example:
lbessard@16:  *       2 + 3       -> returns reference to a int_type_name_c object.
lbessard@16:  *       22.2 - 5    -> returns reference to a real_type_name_c object.
lbessard@16:  *       etc...
lbessard@16:  */
lbessard@16: class search_expression_type_c: public search_constant_type_c {
lbessard@16:   private:
lbessard@16:     search_varfb_instance_type_c *search_varfb_instance_type;
lbessard@16:     search_base_type_c search_base_type;
lbessard@16: 
lbessard@16:   public:
lbessard@16:     search_expression_type_c(symbol_c *search_scope) {
lbessard@16:       search_varfb_instance_type = new search_varfb_instance_type_c(search_scope);
lbessard@16:     }
lbessard@16: 
lbessard@16:     virtual ~search_expression_type_c(void) {
lbessard@16:       delete search_varfb_instance_type;
lbessard@16:     }
lbessard@16: 
lbessard@16:     /* A helper function... */
lbessard@16:     bool is_bool_type(symbol_c *type_symbol) {
lbessard@16:       return (typeid(*type_symbol) == typeid(bool_type_name_c));
lbessard@16:     }
lbessard@16: 
lbessard@16:     /* A helper function... */
lbessard@16:     bool is_time_compatible(symbol_c *type_symbol) {
lbessard@16:       if (typeid(*type_symbol) == typeid(time_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(date_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(tod_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(dt_type_name_c)) {return true;}
lbessard@16:       return false;
lbessard@16:     }
lbessard@16: 
lbessard@16:     /* A helper function... */
lbessard@16:     bool is_numeric_compatible(symbol_c *type_symbol) {
lbessard@16:       if (typeid(*type_symbol) == typeid(sint_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(int_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(dint_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(lint_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(usint_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(uint_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(udint_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(ulint_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(real_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(lreal_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(byte_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(word_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(dword_type_name_c)) {return true;}
lbessard@16:       if (typeid(*type_symbol) == typeid(lword_type_name_c)) {return true;}
lbessard@16:       return false;
lbessard@16:     }
lbessard@16:     
lbessard@16:   private:
lbessard@16: 
lbessard@16:   static bool_type_name_c bool_type_name;
lbessard@16: 
lbessard@16:   /* A helper function... */
lbessard@16:   void *compute_boolean_expression(symbol_c *left_type, symbol_c *right_type) {
lbessard@16:     if (typeid(*left_type) == typeid(bool_type_name_c) and typeid(*right_type) == typeid(bool_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (is_numeric_compatible(left_type) && is_numeric_compatible(right_type)) {return (void *)left_type;}
lbessard@16:     ERROR;
lbessard@16:     return NULL;
lbessard@16:   }
lbessard@16: 
lbessard@16:   /* A helper function... */
lbessard@16:   void *compute_numeric_expression(symbol_c *left_type, symbol_c *right_type) {
lbessard@16:     if (is_numeric_compatible(left_type) && is_numeric_compatible(right_type)) {return (void *)left_type;}
lbessard@16:     ERROR;
lbessard@16:     return NULL;
lbessard@16:   }
lbessard@16: 
lbessard@16:   /* a helper function... */
lbessard@16:   symbol_c *base_type(symbol_c *symbol) {
lbessard@16:     return (symbol_c *)symbol->accept(search_base_type);
lbessard@16:   }
lbessard@16: 
lbessard@16: /*********************/
lbessard@16: /* B 1.4 - Variables */
lbessard@16: /*********************/
lbessard@16: 
lbessard@16:   void *visit(symbolic_variable_c *symbol) {
lbessard@16:     symbol_c *res;
lbessard@16:     
lbessard@16:     /* Nope, now we assume it is a variable, and determine its type... */
lbessard@16:     res = search_varfb_instance_type->get_type(symbol);
lbessard@16:     if (NULL != res) return res;
lbessard@16:     
lbessard@16:     return NULL;
lbessard@16:   }
lbessard@16: 
lbessard@16: 
lbessard@16: /***************************************/
lbessard@16: /* B.3 - Language ST (Structured Text) */
lbessard@16: /***************************************/
lbessard@16: /***********************/
lbessard@16: /* B 3.1 - Expressions */
lbessard@16: /***********************/
lbessard@16:   void *visit(or_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     return compute_boolean_expression(left_type, right_type);
lbessard@16:   }
lbessard@16: 
lbessard@16:   void *visit(xor_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     return compute_boolean_expression(left_type, right_type);
lbessard@16:   }
lbessard@16: 
lbessard@16:   void *visit(and_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     return compute_boolean_expression(left_type, right_type); 
lbessard@16:   }
lbessard@16: 
lbessard@16:   void *visit(equ_expression_c *symbol) {return (void *)&bool_type_name;}
lbessard@16:   void *visit(notequ_expression_c *symbol) {return (void *)&bool_type_name;}
lbessard@16:   void *visit(lt_expression_c *symbol) {return (void *)&bool_type_name;}
lbessard@16:   void *visit(gt_expression_c *symbol) {return (void *)&bool_type_name;}
lbessard@16:   void *visit(le_expression_c *symbol) {return (void *)&bool_type_name;}
lbessard@16:   void *visit(ge_expression_c *symbol) {return (void *)&bool_type_name;}
lbessard@16: 
lbessard@16:   void *visit(add_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     if (is_time_compatible(left_type) && is_time_compatible(right_type)) {return left_type;}
lbessard@16:     return compute_numeric_expression(left_type, right_type);
lbessard@16:   }
lbessard@16: 
lbessard@16:   void *visit(sub_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     if (is_time_compatible(left_type) && is_time_compatible(right_type)) {return left_type;}
lbessard@16:     return compute_numeric_expression(left_type, right_type);
lbessard@16:   }
lbessard@16:   
lbessard@16:   void *visit(mul_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     if (is_time_compatible(left_type) && is_time_compatible(right_type)) {return left_type;}
lbessard@16:     return compute_numeric_expression(left_type, right_type);
lbessard@16:   }
lbessard@16:   
lbessard@16:   void *visit(div_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     return compute_numeric_expression(left_type, right_type);
lbessard@16:   }
lbessard@16: 
lbessard@16:   void *visit(mod_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     return compute_numeric_expression(left_type, right_type);
lbessard@16:   }
lbessard@16: 
lbessard@16:   void *visit(power_expression_c *symbol) {
lbessard@16:     symbol_c *left_type = base_type((symbol_c *)symbol->l_exp->accept(*this));
lbessard@16:     symbol_c *right_type = base_type((symbol_c *)symbol->r_exp->accept(*this));
lbessard@16:     if (typeid(*left_type) == typeid(real_type_name_c) and typeid(*right_type) == typeid(sint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(real_type_name_c) and typeid(*right_type) == typeid(int_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(real_type_name_c) and typeid(*right_type) == typeid(dint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(real_type_name_c) and typeid(*right_type) == typeid(lint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(real_type_name_c) and typeid(*right_type) == typeid(usint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(real_type_name_c) and typeid(*right_type) == typeid(uint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(real_type_name_c) and typeid(*right_type) == typeid(udint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(real_type_name_c) and typeid(*right_type) == typeid(ulint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(lreal_type_name_c) and typeid(*right_type) == typeid(sint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(lreal_type_name_c) and typeid(*right_type) == typeid(int_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(lreal_type_name_c) and typeid(*right_type) == typeid(dint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(lreal_type_name_c) and typeid(*right_type) == typeid(lint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(lreal_type_name_c) and typeid(*right_type) == typeid(usint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(lreal_type_name_c) and typeid(*right_type) == typeid(uint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(lreal_type_name_c) and typeid(*right_type) == typeid(udint_type_name_c)) {return (void *)left_type;}
lbessard@16:     if (typeid(*left_type) == typeid(lreal_type_name_c) and typeid(*right_type) == typeid(ulint_type_name_c)) {return (void *)left_type;}
lbessard@16:     ERROR;
lbessard@16:     return NULL;
lbessard@16:   }
lbessard@16:   
lbessard@16:   void *visit(neg_expression_c *symbol) {
lbessard@16:     symbol_c *exp_type = base_type((symbol_c *)symbol->exp->accept(*this));
lbessard@16:     if (typeid(*exp_type) == typeid(sint_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(int_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(dint_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(lint_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(usint_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(uint_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(udint_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(ulint_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(real_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(lreal_type_name_c)) {return (void *)exp_type;}
lbessard@16:     if (typeid(*exp_type) == typeid(time_type_name_c)) {return (void *)exp_type;}
lbessard@16:     ERROR;
lbessard@16:     return NULL;
lbessard@16:   }
lbessard@16:   
lbessard@16:   void *visit(not_expression_c *symbol) {
lbessard@16:     symbol_c *exp_type = base_type((symbol_c *)symbol->exp->accept(*this));
lbessard@16:     if (typeid(*exp_type) == typeid(bool_type_name_c)) {return (void *)exp_type;}
lbessard@16:     ERROR;
lbessard@16:     return NULL;
lbessard@16:   }
lbessard@16:   
lbessard@16:   void *visit(function_invocation_c *symbol) {
lbessard@16: 	  function_declaration_c *f_decl = function_symtable.find_value(symbol->function_name);
lbessard@16: 	
lbessard@16: 	  if (f_decl == function_symtable.end_value())
lbessard@16: 	    /* should never occur. The function being called MUST be in the symtable... */
lbessard@16: 	    ERROR;
lbessard@16: 
lbessard@16:     return base_type(f_decl->type_name);
lbessard@16:   }
lbessard@16: 
lbessard@16: };
lbessard@16: 
lbessard@16: bool_type_name_c     search_expression_type_c::bool_type_name;
lbessard@16: