trace_model.cpp

/**************************************************************************
 *
 * Copyright 2010 VMware, Inc.
 * All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 **************************************************************************/


#include "formatter.hpp"
#include "trace_model.hpp"


namespace Trace {


void Null::visit(Visitor &visitor) { visitor.visit(this); }
void Bool::visit(Visitor &visitor) { visitor.visit(this); } 
void SInt::visit(Visitor &visitor) { visitor.visit(this); } 
void UInt::visit(Visitor &visitor) { visitor.visit(this); } 
void Float::visit(Visitor &visitor) { visitor.visit(this); } 
void String::visit(Visitor &visitor) { visitor.visit(this); } 
void Enum::visit(Visitor &visitor) { visitor.visit(this); } 
void Bitmask::visit(Visitor &visitor) { visitor.visit(this); } 
void Struct::visit(Visitor &visitor) { visitor.visit(this); } 
void Array::visit(Visitor &visitor) { visitor.visit(this); } 
void Blob::visit(Visitor &visitor) { visitor.visit(this); } 

class Dumper : public Visitor
{
protected:
   std::ostream &os;
   Formatter::Formatter *formatter;
   Formatter::Attribute *normal;
   Formatter::Attribute *bold;
   Formatter::Attribute *italic;
   Formatter::Attribute *red;
   Formatter::Attribute *pointer;
   Formatter::Attribute *literal;

public:
   Dumper(std::ostream &_os) : os(_os) {
      formatter = Formatter::defaultFormatter();
      normal = formatter->normal();
      bold = formatter->bold();
      italic = formatter->italic();
      red = formatter->color(Formatter::RED);
      pointer = formatter->color(Formatter::GREEN);
      literal = formatter->color(Formatter::BLUE);
   }

   ~Dumper() {
      delete normal;
      delete bold;
      delete italic;
      delete red;
      delete pointer;
      delete literal;
      delete formatter;
   }

   void visit(Null *) {
      os << "NULL";
   }

   void visit(Bool *node) {
      os << literal << (node->value ? "true" : "false") << normal;
   }

   void visit(SInt *node) {
      os << literal << node->value << normal;
   }

   void visit(UInt *node) {
      os << literal << node->value << normal;
   }

   void visit(Float *node) {
      os << literal << node->value << normal;
   }

   void visit(String *node) {
      os << literal << '"' << node->value << '"' << normal;
   }

   void visit(Enum *node) {
      os << literal << node->name << normal;
   }

   void visit(Bitmask *bitmask) {
      unsigned long long value = bitmask->value;
      const Bitmask::Signature *sig = bitmask->sig;
      bool first = true;
      for (Bitmask::Signature::const_iterator it = sig->begin(); value != 0 && it != sig->end(); ++it) {
          assert(it->second);
          if ((value & it->second) == it->second) {
              if (!first) {
                  os << " | ";
              }
              os << literal << it->first << normal;
              value &= ~it->second;
              first = false;
          }
      }
      if (value || first) {
          if (!first) {
              os << " | ";
          }
          os << literal << std::hex << value << std::dec << normal;
      }
   }

   void visit(Struct *s) {
      const char *sep = "";
      os << "{";
      for (unsigned i = 0; i < s->members.size(); ++i) {
         os << sep << italic << s->sig->member_names[i] << normal << " = ";
         _visit(s->members[i]);
         sep = ", ";
      }
      os << "}";
   }

   void visit(Array *array) {
      if (array->values.size() == 1) {
         os << "&";
         _visit(array->values[0]);
      }
      else {
         const char *sep = "";
         os << "{";
         for (std::vector<Value *>::iterator it = array->values.begin(); it != array->values.end(); ++it) {
            os << sep;
            _visit(*it);
            sep = ", ";
         }
         os << "}";
      }
   }
   
   void visit(Blob *blob) {
      os << pointer << "blob(" << blob->size << ")" << normal;
   }

   void visit(Call *call) {
      const char *sep = "";
      os << bold << call->sig->name << normal << "(";
      for (unsigned i = 0; i < call->args.size(); ++i) {
         os << sep << italic << call->sig->arg_names[i] << normal << " = ";
         _visit(call->args[i]);
         sep = ", ";
      }
      os << ")";
      if (call->ret) {
         os << " = ";
         _visit(call->ret);
      }
      os << "\n";
   }
};


std::ostream & operator <<(std::ostream &os, Value *value) {
   Dumper d(os);
   if (value) {
      value->visit(d);
   }
   return os;
}


static inline const Value *unwrap(const Value *node) {
   const Enum *c = dynamic_cast<const Enum *>(node);
   if (c)
      return c->value;
   return node;
}


Value::operator bool(void) const {
   const Bool *b = dynamic_cast<const Bool *>(unwrap(this));
   if (b)
      return b->value;
   assert(0);
   return false;
}

Value::operator signed long long(void) const {
   const SInt *sint = dynamic_cast<const SInt *>(unwrap(this));
   if (sint)
      return sint->value;
   const UInt *uint = dynamic_cast<const UInt *>(unwrap(this));
   if (uint)
      return uint->value;
   assert(0);
   return 0;
}

Value::operator unsigned long long(void) const {
   const UInt *uint = dynamic_cast<const UInt *>(unwrap(this));
   if (uint)
      return uint->value;
   assert(0);
   return 0;
}


Value::operator double(void) const {
   const Float *fl = dynamic_cast<const Float *>(unwrap(this));
   assert(fl);
   return fl->value;
}

static Null null;

const Value & Value::operator[](size_t index) const {
    const Array *array = dynamic_cast<const Array *>(unwrap(this));
    if (array) {
        if (index < array->values.size()) {
            return *array->values[index];
        }
    }
    return null;
}

void * Value::blob(void) const {
   const Blob *blob = dynamic_cast<const Blob *>(unwrap(this));
   if (blob)
       return blob->buf;
   const Null *null = dynamic_cast<const Null *>(unwrap(this));
   if (null)
       return NULL;
   assert(0);
   return NULL;
}

const char * Value::string(void) const {
   const String *string = dynamic_cast<const String *>(unwrap(this));
   if (string)
       return string->value.c_str();
   const Null *null = dynamic_cast<const Null *>(unwrap(this));
   if (null)
       return NULL;
   assert(0);
   return NULL;
}

std::ostream & operator <<(std::ostream &os, Call &call) {
   Dumper d(os);
   d.visit(&call);
   return os;
}


} /* namespace Trace */