static void write_flt(double flt)
{
int icnt = 0 ;
int tmpint = 0 ;
tmpint = (int)flt ;
write_dec(tmpint) ;
write_ch('.') ;
tmpint = (int)(flt * 10000000) ;
write_dec(tmpint) ;
}
static void write_fmt(const char* fmt, va_list vp)
{
double vargflt = 0 ;
int vargint = 0 ;
char* vargpch = NULL ;
char vargch = 0 ;
const char* pfmt = NULL ;
pfmt = fmt ;
while(*pfmt)
{
if(*pfmt == '%')
{
switch(*(++pfmt))
{
case 'c' :
vargch = va_arg(vp, int) ;
write_ch(vargch) ;
break ;
case 'd' :
case 'i' :
vargint = va_arg(vp, int) ;
write_dec(vargint) ;
break ;
case 'f':
vargflt = va_arg(vp, double) ;
write_flt(vargflt) ;
break ;
case 's' :
vargpch = va_arg(vp, char*);
write_str(vargpch) ;
break ;
case 'b':
case 'B':
vargint = va_arg(vp, int);
write_bin(vargint) ;
break ;
case 'x' :
case 'X':
vargint = va_arg(vp, int) ;
write_hex(vargint) ;
break ;
case '%' :
write_ch('%') ;
break ;
default :
break ;
}
pfmt++ ;
}
else
{
write_ch(*pfmt++) ;
}
}
static int myprintf(const char *format, ...){
va_list ap;
const char *s;
const char *p;
union {
int intv;
char *charv;
unsigned int uintv;
uint64_t uint64v;
} param;
va_start(ap, format);
p=s=format;
while(*p != 0){
while(*p && *p != '%')
++p;
if(p!=s){
write(1, s, p-s);
s=p;
}
if(*p == '%'){
switch(p[1]){
case 's':
param.charv = va_arg(ap, char *);
write(1, param.charv, strlen(param.charv));
break;
case 'd':
param.intv = va_arg(ap, int);
write_dec(param.intv);
break;
case 'x':
param.uintv = va_arg(ap, unsigned int);
write_hex(param.uintv);
break;
case 'p':
#if defined(ARCH_X86) || defined(ARCH_ARM)
param.uintv = va_arg(ap, unsigned int);
write_hex(param.uintv);
break;
#else
param.uint64v = va_arg(ap, uint64_t);
write_hex64(param.uint64v);
break;
#endif
case '\0':
goto end;
}
p += 2;
s=p;
}
}
end:
va_end(ap);
return 0;
}
static void write_dec(int dec)
{
if(dec == 0)
{
write_ch('0') ;
return ;
}
if(dec/10 != 0)
{
write_dec(dec/10);
}
write_ch( (char) (dec%10 + '0') ) ;
}
int lame_vsprintf(char *str, const char *format, va_list ap)
{
bool in_control_code = false;
char *current = str;
for(; *format; format++) {
if(!in_control_code) {
if(*format == '%')
in_control_code = true;
else
*current++ = *format;
} else {
switch(*format) {
case 'x':
case 'X': {
unsigned val = va_arg(ap, unsigned);
current = write_hex(current, val);
break;
}
case 'd': {
unsigned val = va_arg(ap, unsigned);
current = write_dec(current, val);
break;
}
case 'p': {
void *val = va_arg(ap, void *);
current = write_ptr(current, val);
break;
}
case 's': {
const char *val = va_arg(ap, const char *);
current = write_str(current, val);
break;
}
default:
*current++ = *format;
break;
}
in_control_code = false;
}
}
*current++ = '\0';
return current - str;
}
void sccwriter::write_code( Expr* code, std::ostream& os, int ind, const char* retModStr, int opts )
{
std::string retModString;
std::string incString;
if ( retModStr )
{
retModString = std::string( retModStr );
retModString.append( " = " );
incString = std::string( retModStr );
incString.append( "->inc();" );
}
switch( code->getclass() )
{
case INT_EXPR:
{
indent( os, ind );
os << retModString.c_str();
os << "new IntExpr( " << mpz_get_si( ((IntExpr*)code)->n ) << " );" << std::endl;
indent( os, ind );
os << incString.c_str() << std::endl;
}
break;
case RAT_EXPR:
{
mpz_t num, den;
mpz_init(num);
mpz_init(den);
mpq_get_num( num, ((RatExpr*)code)->n );
mpq_get_den( den, ((RatExpr*)code)->n );
indent( os, ind );
os << retModString.c_str();
os << "new RatExpr( " << mpz_get_si( num ) << ", " << mpz_get_si( den ) << " );" << std::endl;
indent( os, ind );
os << incString.c_str() << std::endl;
}
break;
case SYMS_EXPR:
{
//if it is a variable, simply write it to buffer
if( is_var( ((SymSExpr*)code)->s ) )
{
indent( os, ind );
os << retModString.c_str();
write_variable( ((SymSExpr*)code)->s.c_str(), os );
os << ";" << std::endl;
}
else //else must look at symbol lookup table
{
std::string var;
get_var_name( ((SymSExpr*)code)->s, var );
indent( os, ind );
os << retModString.c_str() << "e_" << var.c_str() << ";" << std::endl;
add_global_sym( var );
}
indent( os, ind );
os << incString.c_str() << std::endl;
}
break;
default:
switch( code->getop() )
{
case APP:
{
//collect the arguments
std::vector< Expr* > argVector;
code->collect_args( argVector );
//write the arguments
std::vector< std::string > args;
for( int a=0; a<(int)argVector.size(); a++ )
{
std::string expr;
write_expr( argVector[a], os, ind, expr );
args.push_back( expr );
}
//write_args( (CExpr*)code, os, ind, 1, args );
Expr* hd = code->get_head();
//map to a program in the case that it is a program
if( hd->getop()==PROG && get_prog_index_by_expr( hd )!=-1 )
{
indent( os, ind );
os << retModString << "f_" << progNames[ get_prog_index_by_expr( hd ) ].c_str() << "( ";
for( int a=0; a<(int)args.size(); a++ )
{
os << args[a].c_str();
if( a!=(int)( args.size()-1 ) ){
os << ", ";
}
}
os << " );" << std::endl;
for( int a=0; a<(int)args.size(); a++ )
{
write_dec( args[a], os, ind );
}
}
else
{
#ifdef USE_FLAT_APP
std::string expr;
write_expr( hd, os, ind, expr );
indent( os, ind );
os << retModString << "new CExpr( APP, ";
os << expr.c_str() << ", ";
for( int a=0; a<(int)args.size(); a++ )
{
os << args[a].c_str();
if( a!=(int)( args.size()-1 ) ){
os << ", ";
}
}
os << " );" << std::endl;
#else
std::string expr;
write_expr( hd, os, ind, expr );
indent( os, ind );
os << retModString;
for( int a=0; a<(int)args.size(); a++ )
{
os << "new CExpr( APP, ";
}
os << expr.c_str() << ", ";
for( int a=0; a<(int)args.size(); a++ )
{
os << args[a].c_str();
os << " )";
if( a!=(int)( args.size()-1 ) ){
os << ", ";
}
}
os << ";" << std::endl;
#endif
//indent( os, ind );
//os << expr.c_str() << "->dec();" << std::endl;
}
}
break;
case MATCH:
{
//calculate the value for the expression
std::string expr;
write_expr( ((CExpr*)code)->kids[0], os, ind, expr );
//get the head
std::ostringstream sshd;
sshd << "e" << exprCount;
exprCount++;
indent( os, ind );
os << "Expr* " << sshd.str().c_str() << " = " << expr.c_str() << "->followDefs()->get_head();" << std::endl;
//write the arguments
std::vector< std::string > args;
write_args( (CExpr*)code, os, ind, 1, args );
bool encounterDefault = false;
//now make an if statement corresponding to the match
int a = 0;
while( ((CExpr*)code)->kids[a+1] )
{
indent( os, ind );
if( a!=0 ){
os << "}else";
}
if( ((CExpr*)code)->kids[a+1]->getop()!=CASE ){
encounterDefault = true;
os << "{" << std::endl;
//write the body of the case
write_code( ((CExpr*)code)->kids[a+1], os, ind+1, retModStr );
indent( os, ind );
os << "}" << std::endl;
}else{
if( a!=0 )
os << " ";
os << "if( " << sshd.str().c_str() << "==" << args[a].c_str() << " ){" << std::endl;
//collect args from the variable in the code
std::ostringstream ssargs;
ssargs << "args" << argsCount;
argsCount++;
#ifndef USE_FLAT_APP
indent( os, ind+1 );
os << "std::vector< Expr* > " << ssargs.str().c_str() << ";" << std::endl;
indent( os, ind+1 );
os << expr.c_str() << "->followDefs()->collect_args( " << ssargs.str().c_str() << " );" << std::endl;
#endif
//set the variables defined in the pattern equal to the arguments
std::vector< Expr* > caseArgs;
((CExpr*)((CExpr*)code)->kids[a+1])->kids[0]->collect_args( caseArgs );
for( int b=0; b<(int)caseArgs.size(); b++ )
{
indent( os, ind+1 );
os << "Expr* ";
write_variable( ((SymSExpr*)caseArgs[b])->s.c_str(), os );
#ifdef USE_FLAT_APP
os << " = ((CExpr*)" << expr.c_str() << "->followDefs())->kids[" << b+1 << "];" << std::endl;
#else
os << " = " << ssargs.str().c_str() << "[" << b << "];" << std::endl;
#endif
vars.push_back( ((SymSExpr*)caseArgs[b])->s );
}
//write the body of the case
write_code( ((CExpr*)code)->kids[a+1], os, ind+1, retModStr, opt_write_case_body );
}
a++;
}
if( !encounterDefault )
{
indent( os, ind );
os << "}else{" << std::endl;
indent( os, ind + 1 );
os << "std::cout << \"Could not find match for expression in function f_";
os << progNames[currProgram].c_str() << " \";" << std::endl;
indent( os, ind + 1 );
os << sshd.str().c_str() << "->print( std::cout );" << std::endl;
indent( os, ind + 1 );
os << "std::cout << std::endl;" << std::endl;
indent( os, ind + 1 );
os << "exit( 1 );" << std::endl;
indent( os, ind );
os << "}" << std::endl;
}
write_dec( expr, os, ind );
for( int a=0; a<(int)args.size(); a++ )
{
write_dec( args[a], os, ind );
}
}
break;
case CASE:
if( opts&opt_write_case_body )
{
write_code( ((CExpr*)code)->kids[1], os, ind, retModStr );
}
else
{
write_code( ((CExpr*)code)->kids[0]->get_head(), os, ind, retModStr );
}
break;
case DO:
{
//write each of the children in sequence
int counter = 0;
while( ((CExpr*)code)->kids[counter] )
{
if( ((CExpr*)code)->kids[counter+1]==NULL )
{
write_code( ((CExpr*)code)->kids[counter], os, ind, retModStr );
}
else
{
std::string expr;
write_expr( ((CExpr*)code)->kids[counter], os, ind, expr );
//clean up memory
write_dec( expr, os, ind );
}
counter++;
}
}
break;
case LET:
{
indent( os, ind );
os << "Expr* ";
write_variable( ((SymSExpr*)((CExpr*)code)->kids[0])->s, os );
os << ";" << std::endl;
std::ostringstream ss;
write_variable( ((SymSExpr*)((CExpr*)code)->kids[0])->s, ss );
write_code( ((CExpr*)code)->kids[1], os, ind, ss.str().c_str() );
//add it to the variables
vars.push_back( ((SymSExpr*)((CExpr*)code)->kids[0])->s );
write_code( ((CExpr*)code)->kids[2], os, ind, retModStr );
//clean up memory
indent( os, ind );
write_variable( ((SymSExpr*)((CExpr*)code)->kids[0])->s, os );
os << "->dec();" << std::endl;
}
break;
case FAIL:
{
indent( os, ind );
os << retModString.c_str() << "NULL;" << std::endl;
}
break;
#ifndef MARKVAR_32
case MARKVAR:
{
//calculate the value for the expression
std::string expr;
write_expr( ((CExpr*)code)->kids[0], os, ind, expr, opt_write_check_sym_expr );
//set the mark on the expression
indent( os, ind );
os << "if (" << expr.c_str() << "->followDefs()->getmark())" << std::endl;
indent( os, ind+1 );
os << expr.c_str() << "->followDefs()->clearmark();" << std::endl;
indent( os, ind );
os << "else" << std::endl;
indent( os, ind+1 );
os << expr.c_str() << "->followDefs()->setmark();" << std::endl;
//write the return if necessary
if( retModStr!=NULL ){
indent( os, ind );
os << retModString.c_str() << expr.c_str() << ";" << std::endl;
indent( os, ind );
os << incString.c_str() << std::endl;
}
write_dec( expr, os, ind );
}
break;
case IFMARKED:
{
//calculate the value for the expression
std::string expr;
write_expr( ((CExpr*)code)->kids[0], os, ind, expr, opt_write_check_sym_expr );
//if mark is set, write code for kids[1]
indent( os, ind );
os << "if (" << expr.c_str() << "->followDefs()->getmark()){" << std::endl;
write_code( ((CExpr*)code)->kids[1], os, ind+1, retModStr );
//else write code for kids[2]
indent( os, ind );
os << "}else{" << std::endl;
write_code( ((CExpr*)code)->kids[2], os, ind+1, retModStr );
indent( os, ind );
os << "}" << std::endl;
//clean up memory
write_dec( expr, os, ind );
}
break;
#else
case MARKVAR:
{
//calculate the value for the expression
std::string expr;
write_expr( ((CExpr*)code)->kids[1], os, ind, expr, opt_write_check_sym_expr );
//set the mark on the expression
indent( os, ind );
os << "if ( ((SymExpr*)" << expr.c_str() << "->followDefs())->getmark(";
os << ((IntExpr*)((CExpr*)code)->kids[0])->get_num() << "))" << std::endl;
indent( os, ind+1 );
os << "((SymExpr*)" << expr.c_str() << "->followDefs())->clearmark(";
os << ((IntExpr*)((CExpr*)code)->kids[0])->get_num() << ");" << std::endl;
indent( os, ind );
os << "else" << std::endl;
indent( os, ind+1 );
os << "((SymExpr*)" << expr.c_str() << "->followDefs())->setmark(";
os << ((IntExpr*)((CExpr*)code)->kids[0])->get_num() << ");" << std::endl;
//write the return if necessary
if( retModStr!=NULL ){
indent( os, ind );
os << retModString.c_str() << expr.c_str() << ";" << std::endl;
indent( os, ind );
os << incString.c_str() << std::endl;
}
write_dec( expr, os, ind );
}
break;
case COMPARE:
{
std::string expr1, expr2;
write_expr( ((CExpr*)code)->kids[0], os, ind, expr1, opt_write_check_sym_expr );
write_expr( ((CExpr*)code)->kids[1], os, ind, expr2, opt_write_check_sym_expr );
indent( os, ind );
os << "if( ((SymExpr*)" << expr1.c_str() << ")->followDefs() < ((SymExpr*)" << expr2.c_str() << ")->followDefs() ){" << std::endl;
write_code( ((CExpr*)code)->kids[2], os, ind+1, retModStr );
indent( os, ind );
os << "}else{" << std::endl;
write_code( ((CExpr*)code)->kids[3], os, ind+1, retModStr );
indent( os, ind );
os << "}" << std::endl;
//clean up memory
write_dec( expr1, os, ind );
write_dec( expr2, os, ind );
}
break;
case IFMARKED:
{
//calculate the value for the expression
std::string expr;
write_expr( ((CExpr*)code)->kids[1], os, ind, expr, opt_write_check_sym_expr );
//if mark is set, write code for kids[1]
indent( os, ind );
os << "if ( ((SymExpr*)" << expr.c_str() << "->followDefs())->getmark(";
os << ((IntExpr*)((CExpr*)code)->kids[0])->get_num() << ")){" << std::endl;
write_code( ((CExpr*)code)->kids[2], os, ind+1, retModStr );
//else write code for kids[2]
indent( os, ind );
os << "}else{" << std::endl;
write_code( ((CExpr*)code)->kids[3], os, ind+1, retModStr );
indent( os, ind );
os << "}" << std::endl;
//clean up memory
write_dec( expr, os, ind );
}
break;
#endif
case ADD:
case MUL:
case DIV:
{
//calculate the value for the first expression
std::string expr1;
write_expr( ((CExpr*)code)->kids[0], os, ind, expr1 );
//calculate the value for the second expression
std::string expr2;
write_expr( ((CExpr*)code)->kids[1], os, ind, expr2 );
std::ostringstream ss;
ss << "rnum" << rnumCount;
rnumCount++;
indent( os, ind );
os << "if( " << expr1.c_str() << "->followDefs()->getclass()==INT_EXPR ){" << std::endl;
indent( os, ind+1 );
os << "mpz_t " << ss.str().c_str() << ";" << std::endl;
indent( os, ind+1 );
os << "mpz_init(" << ss.str().c_str() << ");" << std::endl;
indent( os, ind+1 );
os << "mpz_";
if( code->getop()==ADD )
os << "add";
else
os << "mul";
os << "( " << ss.str().c_str() << ", ((IntExpr*)" << expr1.c_str() << "->followDefs())->n, ((IntExpr*)" << expr2.c_str() << "->followDefs())->n);" << std::endl;
indent( os, ind+1 );
os << retModString.c_str() << "new IntExpr(" << ss.str().c_str() << ");" << std::endl;
indent( os, ind );
os << "}else if( " << expr1.c_str() << "->followDefs()->getclass()==RAT_EXPR ){" << std::endl;
indent( os, ind+1 );
os << "mpq_t " << ss.str().c_str() << ";" << std::endl;
indent( os, ind+1 );
os << "mpq_init(" << ss.str().c_str() << ");" << std::endl;
indent( os, ind+1 );
os << "mpq_";
if( code->getop()==ADD )
os << "add";
else if( code->getop()==MUL )
os << "mul";
else
os << "div";
os << "( " << ss.str().c_str() << ", ((RatExpr*)" << expr1.c_str() << "->followDefs())->n, ((RatExpr*)" << expr2.c_str() << "->followDefs())->n);" << std::endl;
indent( os, ind+1 );
os << retModString.c_str() << "new RatExpr(" << ss.str().c_str() << ");" << std::endl;
indent( os, ind );
os << "}" << std::endl;
//clean up memory
write_dec( expr1, os, ind );
write_dec( expr2, os, ind );
}
break;
case NEG:
{
//calculate the value for the first expression
std::string expr1;
write_expr( ((CExpr*)code)->kids[0], os, ind, expr1 );
std::ostringstream ss;
ss << "rnum" << rnumCount;
rnumCount++;
indent( os, ind );
os << "if( " << expr1.c_str() << "->followDefs()->getclass()==INT_EXPR ){" << std::endl;
indent( os, ind+1 );
os << "mpz_t " << ss.str().c_str() << ";" << std::endl;
indent( os, ind+1 );
os << "mpz_init(" << ss.str().c_str() << ");" << std::endl;
indent( os, ind+1 );
os << "mpz_neg( " << ss.str().c_str() << ", ((IntExpr*)" << expr1.c_str() << "->followDefs())->n );" << std::endl;
indent( os, ind+1 );
os << retModString.c_str() << "new IntExpr(" << ss.str().c_str() << ");" << std::endl;
indent( os, ind );
os << "}else if( " << expr1.c_str() << "->followDefs()->getclass()==RAT_EXPR ){" << std::endl;
indent( os, ind+1 );
os << "mpq_t " << ss.str().c_str() << ";" << std::endl;
indent( os, ind+1 );
os << "mpq_init(" << ss.str().c_str() << ");" << std::endl;
indent( os, ind+1 );
os << "mpq_neg( " << ss.str().c_str() << ", ((RatExpr*)" << expr1.c_str() << "->followDefs())->n );" << std::endl;
indent( os, ind+1 );
os << retModString.c_str() << "new RatExpr(" << ss.str().c_str() << ");" << std::endl;
indent( os, ind );
os << "}" << std::endl;
//clean up memory
write_dec( expr1, os, ind );
}
break;
case IFNEG:
case IFZERO:
{
std::string expr1;
write_expr( ((CExpr*)code)->kids[0], os, ind, expr1 );
indent( os, ind );
os << "if( " << expr1.c_str() << "->followDefs()->getclass()==INT_EXPR ){" << std::endl;
indent( os, ind+1 );
os << "if( mpz_sgn( ((IntExpr *)" << expr1.c_str() << "->followDefs())->n ) ";
if( code->getop()==IFNEG )
os << "<";
else
os << "==";
os << " 0 ){" << std::endl;
write_code( ((CExpr*)code)->kids[1], os, ind+2, retModStr );
indent( os, ind+1 );
os << "}else{" << std::endl;
write_code( ((CExpr*)code)->kids[2], os, ind+2, retModStr );
indent( os, ind+1 );
os << "}" << std::endl;
indent( os, ind );
os << "}else if( " << expr1.c_str() << "->followDefs()->getclass()==RAT_EXPR ){" << std::endl;
indent( os, ind+1 );
os << "if( mpq_sgn( ((RatExpr *)" << expr1.c_str() << "->followDefs())->n ) ";
if( code->getop()==IFNEG )
os << "<";
else
os << "==";
os << " 0 ){" << std::endl;
write_code( ((CExpr*)code)->kids[1], os, ind+2, retModStr );
indent( os, ind+1 );
os << "}else{" << std::endl;
write_code( ((CExpr*)code)->kids[2], os, ind+2, retModStr );
indent( os, ind+1 );
os << "}" << std::endl;
indent( os, ind );
os << "}" << std::endl;
//clean up memory
write_dec( expr1, os, ind );
}
break;
case RUN:/*?*/break;
case PI:/*?*/break;
case LAM:/*?*/break;
case TYPE:/*?*/break;
case KIND:/*?*/break;
case ASCRIBE:/*?*/break;
case MPZ:/*?*/break;
case PROG:/*?*/break;
case PROGVARS:/*?*/break;
case PAT:/*?*/break;
}
break;
}
}
