// This file is part of the AspectC++ compiler 'ac++'.
// Copyright (C) 1999-2003  The 'ac++' developers (see aspectc.org)
//                                                                
// This program is free software;  you can redistribute it and/or 
// modify it under the terms of the GNU General Public License as 
// published by the Free Software Foundation; either version 2 of 
// the License, or (at your option) any later version.            
//                                                                
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of 
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the  
// GNU General Public License for more details.                   
//                                                                
// You should have received a copy of the GNU General Public      
// License along with this program; if not, write to the Free     
// Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, 
// MA  02111-1307  USA                                            

#include <assert.h>

#include "ThisJoinPoint.h"
#include "JoinPoint.h"
#include "Naming.h"
#include "Binding.h"
#include "JPAdvice.h"
#ifdef ACMODEL
#include "Utils.h"
#else
#include "JoinPointLoc.h"
#endif
#include "WeaverBase.h"

#include "Puma/CTree.h"
#include "Puma/CClassInfo.h"
#include "Puma/CFunctionInfo.h"
#include "Puma/CTypeInfo.h"
#include "Puma/CArgumentInfo.h"

#include <sstream>
using std::stringstream;
using std::endl;
#include <string>
using std::string;

// special setup function for bindings
void ThisJoinPoint::setup (const Binding &binding) {
  _setup = true;
  if (binding._used) {
    _used |= (binding._this != 0) ? THAT : 0;
    _used |= (binding._target != 0) ? TARGET : 0;
    _used |= (binding._result != 0) ? RESULT : 0;
    _used |= (binding._args.length () > 0) ? ARG : 0;
    _used |= (PTR_NEEDED|TYPE_NEEDED);
  }
}

void ThisJoinPoint::setup (CFunctionInfo* func) {
  assert (func->Tree ()->NodeName () == CT_FctDef::NodeId ());

  _setup = true;
  // by starting with the second statement of the body we avoid to iterate
  // through the generated argument list and the generated typedef at the
  // beginning of the function
  CT_CmpdStmt *body = ((CT_FctDef*)func->Tree ())->Body ();
//  for (int i = 1; i < body->Entries (); i++)
// EXPERIMENT
  for (int i = 0; i < body->Entries (); i++)
    setup_search (func, body->Entry (i));
}

void ThisJoinPoint::setup_search (CFunctionInfo *func, CTree *node) {
  CObjectInfo *obj;
  CClassInfo *cls;

  const char *nodename = node->NodeName ();
  
  if (nodename == CT_SimpleName::NodeId () &&
      (obj = ((CT_SimpleName*)node)->Object ())) {
    if (obj->Scope ()->GlobalScope () &&
        strcmp (obj->Name (), "tjp") == 0)
      _used |= (PTR_NEEDED | TYPE_NEEDED);
    else if (obj->Scope ()->GlobalScope () &&
             strcmp (obj->Name (), "thisJoinPoint") == 0)
      _used |= (PTR_NEEDED | PTR_ALIAS_NEEDED | TYPE_NEEDED);
    else if (obj->Scope ()->GlobalScope () &&
             strcmp (obj->Name (), "JoinPoint") == 0)
      _used |= TYPE_NEEDED;
  
    // workaround as long as calls to template functions are not resolved
    if (obj->ClassScope () && 
        strcmp (obj->ClassScope ()->Name (), "JoinPoint") == 0 &&
        obj->FunctionInfo () && obj->FunctionInfo ()->Arguments () == 0 &&
        strcmp (obj->Name (), "arg") == 0) {
      _used |= ARG;
    }
  }

  if (nodename == CT_CallExpr::NodeId () &&
      (obj = ((CT_CallExpr*)node)->Object ()) &&
      (cls = obj->DefObject ()->Scope ()->ClassInfo ()) &&
      strcmp (cls->QualName (), "JoinPoint") == 0) {
    const char *field = obj->Name ();

    if (strcmp (field, "signature") == 0)
      _used |= SIGNATURE;
    else if (strcmp (field, "filename") == 0)
      _used |= FILENAME;
    else if (strcmp (field, "line") == 0)
      _used |= LINE;
    else if (strcmp (field, "args") == 0)
      _used |= ARGS;
    else if (strcmp (field, "arg") == 0)
      _used |= ARG;
    else if (strcmp (field, "argtype") == 0)
      _used |= ARG_TYPE;
    else if (strcmp (field, "type") == 0)
      _used |= TYPE;
    else if (strcmp (field, "id") == 0)
      _used |= ID;
    else if (strcmp (field, "resulttype") == 0)
      _used |= RESULT_TYPE;
    else if (strcmp (field, "that") == 0)
      _used |= THAT;
    else if (strcmp (field, "target") == 0)
      _used |= TARGET;
    else if (strcmp (field, "result") == 0)
      _used |= RESULT;
    else if (strcmp (field, "jptype") == 0)
      _used |= JP_TYPE;
    else if (strcmp (field, "action") == 0)
      _used |= ACTION;
    else if (strcmp (field, "wrapper") == 0)
      _used |= WRAPPER;
    else if (strcmp (field, "proceed") == 0)
    	_used |= PROCEED;
  }

  for (int s = 0; s < node->Sons (); s++)
    setup_search (func, node->Son (s));
}   

#ifdef ACMODEL
string ThisJoinPoint::gen_result_type (ACM_Code *jpl) {
  CTypeInfo *rtype = (has_result_type(*jpl)?
                      TI_Type::of (*get_result_type(*jpl))->type_info ():
                      &CTYPE_UNDEFINED);
#else
string ThisJoinPoint::gen_result_type (JPL_Code *jpl) {
  CTypeInfo *rtype = TI_Type::of (jpl->result_type ())->type_info ();
#endif
  if (rtype->isVoid () || rtype->isUndefined ())
    return "void";
  else if (rtype->isAddress ())
    rtype = rtype->BaseType ();
  stringstream type;
  rtype->TypeText (type, "", true, true);
  return type.str ();
}

#ifdef ACMODEL
void ThisJoinPoint::gen_tjp_struct (ostream &code, ACM_Code *loc,
                                    BackEndProblems &bep, int depth) const {
#else
void ThisJoinPoint::gen_tjp_struct (ostream &code, JPL_Code *loc,
                                    BackEndProblems &bep, int depth) const {
#endif
  
  TI_Code *ti = (TI_Code*)loc->transform_info ();

  // the level 0 struct -> contains all the data and types
  code << "template <typename TResult, typename TThat, typename TTarget, ";
  code << "typename TArgs> struct ";
  Naming::tjp_struct (code, loc, depth);
  if (depth > 0) {
    // the level N > 0 structs -> are derived from level N - 1
    code << " : ";
    Naming::tjp_struct (code, loc, depth - 1);
    code << "<TResult, TThat, TTarget, TArgs>";
  }
  else {
    // the level N == 0 structs are derived from AC::Action, if action() used
    if (useAction())
      code << " : AC::Action";
  }
  code << " {" << endl;
  code << "  typedef ";
  Naming::tjp_struct (code, loc, depth);
  code << " __TJP;" << endl; // internal type definition
  
  stringstream types, data, fct;

  // start: type definitions --------------------------------------------------
#ifdef ACMODEL
  CTypeInfo *rtype  = (has_result_type(*loc) ?
                       TI_Type::of (*get_result_type (*loc))->type_info ():
                       &CTYPE_UNDEFINED);
#else
  CTypeInfo *rtype  = TI_Type::of (loc->result_type ())->type_info ();
#endif
  bool res_is_ref   = rtype->isAddress ();
  bool res_is_undef = rtype->isUndefined ();

  types << "  typedef TResult " << (res_is_ref ? "*" : "") << "Result;" << endl;
  types << "  typedef TThat   That;" << endl;
  types << "  typedef TTarget Target;" << endl;
  
  // argument count and types
  types << "  enum { ARGS = TArgs::ARGS };" << endl;
  types << "  template <int I> struct Arg : AC::Arg<TArgs, I> {};" << endl;
    
  // join point id and type;
#ifdef ACMODEL
  // TODO: handle join-point ID in TJP
//  int jpid = (loc->assigned_id () == -1 ? loc->id () : loc->assigned_id ());
  types << "  static const int JPID = " << 0 /*jpid*/ << ";" << endl;
  types << "  static const AC::JPType JPTYPE = (AC::JPType)" << loc->type_val ()
        << ";" << endl;
#else
  int jpid = (loc->assigned_id () == -1 ? loc->id () : loc->assigned_id ());
  types << "  static const int JPID = " << jpid << ";" << endl;
  types << "  static const AC::JPType JPTYPE = (AC::JPType)" << loc->type ()
        << ";" << endl;
#endif
  
  // result type
  types << "  struct Res {" << endl;
  types << "    typedef " << (res_is_undef ? "void" : "TResult") << " "
        << (res_is_ref ? "&" : "") << "Type;" << endl;
  types << "    typedef " << (res_is_undef ? "void" : "TResult") << " ReferredType;" << endl;
  types << "  };" << endl;
  
  // argument types
#ifdef ACMODEL
  unsigned arg_count = get_arg_count (*loc);
#else
  unsigned arg_count = loc->arg_count ();
#endif
  
  // begin of static data -----------------------------------------------------
  if (signature ()) {
    fct << "  inline static const char *signature () { return \"";
#ifdef ACMODEL
    fct << ::signature (*loc) << "\";}" << endl;
#else
    fct << loc->signature () << "\";}" << endl;
#endif
  }
  if (filename ()) {
    fct << "  inline static const char *filename () { return \"";
#ifdef ACMODEL
    // TODO: implement filename function
    fct << ::filename (*loc) << "\";}" << endl;
#else
    fct << loc->filename () << "\";}" << endl;
#endif
  }
  if (line ()) {
    fct << "  inline static int line () { return ";
#ifdef ACMODEL
    // TODO: implement line function
    fct << ::line(*loc) << ";}" << endl;
#else
    fct << loc->line () << ";}" << endl;
#endif
  }
  if (args ()) {
    fct << "  inline static const int args () { return ";
    fct << arg_count << ";";
    fct << " }" << endl;
  }
  if (type()) {
    fct << "  inline static  AC::Type type() { return ";
    if (ti->tjp_type()) {
      fct << "\"";
      ti->tjp_type()->TypeInfo()->Mangled(fct);
      fct << "\";";
    } else fct << "\"<unknown type>\";";
    fct << "}" << endl;
  }
  if (id()) {
    fct << "  inline static unsigned int id() { return JPID; }" << endl;
  }
  
  if (resulttype()) {
    fct << "  inline static AC::Type resulttype() {return ";
#ifdef ACMODEL
    if (has_result_type(*loc)) {
      fct << "\"";
      TI_Type::of (*get_result_type(*loc))->type_info()->Mangled(fct);
#else
      if (!TI_Type::of (loc->result_type ())->type_info ()->isUndefined ()) {
        fct << "\"";
        TI_Type::of (loc->result_type())->type_info()->Mangled(fct);
#endif
      fct << "\";";
    } else fct << "\"<unknown signature>\";";
    fct << "}" << endl;    
  }
  
  if (jptype()) {
    fct << "  inline static AC::JPType jptype() { return JPTYPE; }" << endl;
  }
  
  if (argtype()) {
    fct << "  inline static AC::Type const argtype(unsigned i) {" << endl;
    if (arg_count > 0) {
      fct << "    static AC::Type const type[" << arg_count << "] = {";
#ifdef ACMODEL
      list<ACM_Type*> tlist;
      get_arg_types (*loc, tlist);
      unsigned i = 0;
      for (list<ACM_Type*>::iterator iter = tlist.begin(); iter != tlist.end();
           ++iter, ++i) {
        if (i > 0) fct << ", ";
        fct << "\"";
        TI_Type::of (**iter)->type_info ()->Mangled(fct);
        fct << "\"";
      }
#else
      for (unsigned i = 0; i < arg_count; i++) {
        if (i > 0) fct << ", ";
        fct << "\"";
        TI_Type::of (loc->arg_type (i))->type_info ()->Mangled(fct);
        fct << "\"";
      }
#endif
      fct << "};" << endl;
      fct << "    return type[i];" << endl;
      fct << "  }" << endl;
    } else {
      fct << "    return \"\";" << endl;
      fct << "  }" << endl;
    }
  }
   
  // begin of dynamic data ----------------------------------------------------
#ifdef ACMODEL
  if (arg () || useAction () || (proceed () && proceed_needs_args (*loc))) {
#else
    if (arg () || useAction () || (proceed () && loc->proceed_needs_args ())) {
#endif
    if (!useAction () && arg_count > 0)
      data << "  void *_args[ARGS];" << endl;
    fct <<  "  inline void *arg (int n) {return "
        << (arg_count > 0 ? "_args[n]" : "0") << ";}" << endl;
    fct <<  "  template <int I> typename Arg<I>::ReferredType *arg () {"
        << endl;
    fct <<  "    return (typename Arg<I>::ReferredType*)arg (I);" << endl;
    fct <<  "  }" << endl;
  }

#ifdef ACMODEL
  if (result() || useAction () || (proceed () && proceed_needs_result (*loc))) {
#else
  if (result() || useAction () || (proceed () && loc->proceed_needs_result ())) {
#endif
    if (!useAction ())
      data << "  Result *_result;" << endl;
    fct <<  "  inline Result *result() {return (Result*)_result;}" << endl;
  }

  if (target() || useAction () || (proceed () && ti->proceed_needs_target ())) {
    if (!useAction ())
      data << "  Target *_target;" << endl;
    fct <<  "  inline Target *target() {return (Target*)_target;}" << endl;
  }

  if (that() || useAction () || (proceed () && ti->proceed_needs_that ())) {
    if (!useAction ())
      data << "  That *_that;" << endl;
    fct <<  "  inline That *that() {return (That*)_that;}" << endl;
  }

  if (proceed () && ti->proceed_needs_fptr ()) {
    data << "  void *_fptr;" << endl;
  }
  
  if (wrapper() || useAction ()) {
    if (!useAction ())
      data << "  void (*_wrapper)(AC::Action &);" << endl;
    fct <<  "  inline void (*wrapper ())(AC::Action &) {return _wrapper;}" << endl;
  }

  // terminate all the strings
  types << endl;
  data << endl;
  fct << endl;
  
  // add all types, attributes, and functions (on level 0)
  if (depth == 0) {
    code << types.str ().data ();
    code << data.str ().data ();
    code << fct.str ().data ();
  }
  
  // the closing bracket is *NOT* generated here -> for external extensions
}

#ifdef ACMODEL
void ThisJoinPoint::gen_tjp_init (ostream &code, ACM_Code *loc,
                                  BackEndProblems &bep, int depth) const {
#else
void ThisJoinPoint::gen_tjp_init (ostream &code, JPL_Code *loc,
                                    BackEndProblems &bep, int depth) const {
#endif
  TI_Code *ti = (TI_Code*)TransformInfo::of (*loc);
  const ThisJoinPoint &tjp = *this; // TODO: not necessary!
#ifdef ACMODEL
  JoinPointType jptype = loc->type_val ();
#else
  JoinPointLoc::join_point_type jptype = loc->type ();
#endif
  
  bool local_class = false, havefuncptr = false;
  // determine the weaving parameters
#ifdef ACMODEL
  if (jptype == JPT_Call) {
#else
  if (jptype == JoinPointLoc::MethodCall) {
#endif
//    local_class = (!_problems._local_class &&
//                   !((JPL_MethodCall*)loc)->in_member_init () &&
//                   ((JPL_MethodCall*)loc)->needs_rights ());
    havefuncptr = (ti->proceed_needs_fptr () &&
                   (!local_class || depth > 0));
  }

  if (tjp.pointer_needed ()) {    

    code << "  __TJP ";
    Naming::tjp_instance(code, loc);
    code << ";" << endl;

#ifdef ACMODEL
    int args = get_arg_count(*loc);
#else
    int args = ((JPL_Code*)loc)->arg_count ();
#endif
    if (tjp.arg() || tjp.useAction() ||
#ifdef ACMODEL
      (tjp.proceed () && proceed_needs_args (*loc))) {
#else
      (tjp.proceed () && loc->proceed_needs_args ())) {
#endif
        if (args) {
      	if (tjp.useAction ()) {
	        code << "  void *";
  	      Naming::tjp_args_array(code, loc);
    	    code << "[] = { ";
      	  for (int i = 0; i < args; i++) {
        	  if (i > 0) code << ", ";
          	code << "(void*)&arg" << i;
        	}
        	code << " };" << endl;
      	}
      }

	if (tjp.useAction ()) {
	    code << "  ";
  	    Naming::tjp_instance(code, loc);
    	  code << "._args = ";
      	if (args)
        	Naming::tjp_args_array(code, loc);
      	else
        	code << "0";
      	code << ";" << endl;
			}
			else {
    	  for (int i = 0; i < args; i++) {
		      code << "  ";
  		    Naming::tjp_instance(code, loc);
    		  code << "._args[" << i << "] = (void*)&arg" << i << ";" << endl;
    	  }
			}
    }
    
    if (tjp.result() || tjp.useAction() ||
#ifdef ACMODEL
      (tjp.proceed () && proceed_needs_result (*loc))) {
#else
      (tjp.proceed () && loc->proceed_needs_result ())) {
#endif
        code << "  ";
      Naming::tjp_instance(code, loc);
      code << "._result = ";
#ifdef ACMODEL
      CTypeInfo *rtype = (has_result_type(*loc) ?
                          TI_Type::of (*get_result_type(*loc))->type_info ():
                          &CTYPE_UNDEFINED);
#else
      CTypeInfo *rtype = TI_Type::of (((JPL_Code*)loc)->result_type ())->type_info ();
#endif
      if (!(rtype->isVoid() || rtype->isUndefined ())) {
        if (tjp.useAction ())
          code << "(void*)";
        code << "&(__TJP::Result&)result";
      } else {
        code << "0";
      }
      code << ";" << endl;
    }
    
    if (tjp.target() || tjp.useAction() ||
        (tjp.proceed () && ti->proceed_needs_target ())) {
      code << "  ";
      Naming::tjp_instance(code, loc);
      code << "._target = ";
      if (ti->tjp_target () && WeaverBase::needs_this (ti->tjp_target())) {
        code << " (";
        if (tjp.useAction ())
          code << "void*)";
        else {
          code << "__TJP::Target*)";
        }
        switch (jptype) {
#ifdef ACMODEL
        case JPT_Construction:
        case JPT_Destruction:
        case JPT_Execution:
          code << "this";
          break;
        case JPT_Call:
#else
        case JoinPointLoc::Construction:
        case JoinPointLoc::Destruction:
        case JoinPointLoc::Method:
          code << "this";
          break;
        case JoinPointLoc::MethodCall:
#endif
          code << "dstthis";
            break;
          default:
            code << " 0";
    	  }
      } else {
        code << " 0";
      }
      code << ";" << endl;
    }

    if (tjp.that() || tjp.useAction() ||
        (tjp.proceed () && ti->proceed_needs_that ())) {
      code << "  ";
      Naming::tjp_instance(code, loc);
      code << "._that = ";
      if (ti->tjp_that () && WeaverBase::needs_this (ti->tjp_that())) {
        code << " (";
        if (tjp.useAction ())
          code << "void*)";
        else {
          code << "__TJP::That*)";
        }
        switch (jptype) {
#ifdef ACMODEL
        case JPT_Construction:
        case JPT_Destruction:
        case JPT_Execution:
          code << "this";
          break;
        case JPT_Call:
#else
        case JoinPointLoc::Construction:
        case JoinPointLoc::Destruction:
        case JoinPointLoc::Method:
          code << "this";
          break;
        case JoinPointLoc::MethodCall:
#endif
          code << "srcthis";
            break;
          default:
            code << " 0";
        }
      } else {
        code << " 0";
      }
      code << ";" << endl;
    }
    
    if (tjp.useAction () ||
        (tjp.proceed () && ti->proceed_needs_fptr ())) {
      code << "  ";
      Naming::tjp_instance(code, loc);
      code << "._fptr = ";
      if (havefuncptr)
        code << "&fptr";
      else  
        code << "0";
      code << ";" << endl;
    }
  }

}

void ThisJoinPoint::dump (ostream &out) const {
  out << "ThisJoinPoint (";
  out << "signature=" << (signature () ? "true" : "false");
  out << ", filename=" << (filename () ? "true" : "false");
  out << ", line=" << (line () ? "true" : "false");
  out << ", args=" << (args () ? "true" : "false");
  out << ", arg=" << (arg() ? "true" : "false");
  out << ", argtype=" << (argtype() ? "true" : "false");
  out << ", type=" << (type() ? "true" : "false");
  out << ", id=" << (id() ? "true" : "false");
  out << ", resulttype=" << (resulttype() ? "true" : "false");
  out << ", that=" << (that() ? "true" : "false");
  out << ", target=" << (target() ? "true" : "false");
  out << ", result=" << (result() ? "true" : "false");
  out << ", jptype=" << (jptype() ? "true" : "false");
  out << ", action=" << (action() ? "true" : "false");
  out << ", proceed=" << (proceed() ? "true" : "false");
  out << ")" << endl;
}