#!/usr/bin/python
#

try:
    import gdb
except ImportError as e:
    raise ImportError("This script must be run in GDB: ", str(e))
import traceback
import datetime
import sys
import os
sys.path.append(os.getcwd())
import common_helpers


class LogSizeCommand (gdb.Command):
    """calc size of the memory used by the object and write it to file"""

    def __init__ (self):
        super (LogSizeCommand, self).__init__ ("logsize", gdb.COMMAND_USER)

    def invoke(self, arg, from_tty):
        try:
            args = gdb.string_to_argv(arg)
            obj = gdb.parse_and_eval(args[0])
            logfile = open(args[1], 'a')
            size = common_helpers.get_instance_size(obj)
            logfile.write("%s %s: %d\n" % (str(datetime.datetime.now()), args[0], size))
            logfile.close()
        except Exception as e:
            print(traceback.format_exc())
            raise e

LogSizeCommand()