std_code.cpp

Go to the documentation of this file.

/*******************************************************************\
 
Module: Data structures representing statements in a program
 
Author: Daniel Kroening, kroening@kroening.com
 
\*******************************************************************/
 
 
#include "std_code.h"
 
#include "arith_tools.h"
 
codet &codet::first_statement()
{
  const irep_idt &statement=get_statement();
 
  if(has_operands())
  {
    if(statement==ID_block)
      return to_code(op0()).first_statement();
    else if(statement==ID_label)
      return to_code(op0()).first_statement();
  }
 
  return *this;
}
 
const codet &codet::first_statement() const
{
  const irep_idt &statement=get_statement();
 
  if(has_operands())
  {
    if(statement==ID_block)
      return to_code(op0()).first_statement();
    else if(statement==ID_label)
      return to_code(op0()).first_statement();
  }
 
  return *this;
}
 
codet &codet::last_statement()
{
  const irep_idt &statement=get_statement();
 
  if(has_operands())
  {
    if(statement==ID_block)
      return to_code(operands().back()).last_statement();
    else if(statement==ID_label)
      return to_code(operands().back()).last_statement();
  }
 
  return *this;
}
 
const codet &codet::last_statement() const
{
  const irep_idt &statement=get_statement();
 
  if(has_operands())
  {
    if(statement==ID_block)
      return to_code(operands().back()).last_statement();
    else if(statement==ID_label)
      return to_code(operands().back()).last_statement();
  }
 
  return *this;
}
 
void code_blockt::append(const code_blockt &extra_block)
{
  statements().reserve(statements().size() + extra_block.statements().size());
 
  for(const auto &statement : extra_block.statements())
  {
    add(statement);
  }
}
 
codet &code_blockt::find_last_statement()
{
  codet *last=this;
 
  while(true)
  {
    const irep_idt &statement=last->get_statement();
 
    if(statement==ID_block &&
       !to_code_block(*last).statements().empty())
    {
      last=&to_code_block(*last).statements().back();
    }
    else if(statement==ID_label)
    {
      last = &(to_code_label(*last).code());
    }
    else
      break;
  }
 
  return *last;
}
 
code_blockt create_fatal_assertion(
  const exprt &condition, const source_locationt &loc)
{
  code_blockt result({code_assertt(condition), code_assumet(condition)});
 
  for(auto &op : result.statements())
    op.add_source_location() = loc;
 
  result.add_source_location() = loc;
 
  return result;
}
 
std::vector<irep_idt> code_function_bodyt::get_parameter_identifiers() const
{
  const auto &sub = find(ID_parameters).get_sub();
  std::vector<irep_idt> result;
  result.reserve(sub.size());
  for(const auto &s : sub)
    result.push_back(s.get(ID_identifier));
  return result;
}
 
void code_function_bodyt::set_parameter_identifiers(
  const std::vector<irep_idt> &parameter_identifiers)
{
  auto &sub = add(ID_parameters).get_sub();
  sub.reserve(parameter_identifiers.size());
  for(const auto &id : parameter_identifiers)
  {
    sub.push_back(irept(ID_parameter));
    sub.back().set(ID_identifier, id);
  }
}
 
code_fort code_fort::from_index_bounds(
  exprt start_index,
  exprt end_index,
  symbol_exprt loop_index,
  codet body,
  source_locationt location)
{
  PRECONDITION(start_index.type() == loop_index.type());
  PRECONDITION(end_index.type() == loop_index.type());
  side_effect_expr_assignt inc(
    loop_index,
    plus_exprt(loop_index, from_integer(1, loop_index.type())),
    location);
 
  return code_fort{
    code_frontend_assignt{loop_index, std::move(start_index)},
    binary_relation_exprt{loop_index, ID_lt, std::move(end_index)},
    std::move(inc),
    std::move(body)};
}