import os
import glob

# path to global data files
base_path = '../orb/'
cfile = './sdlog_generated.h'
mfile_template = './mfile.template'

# there should be one LOGBROADCAST which gives the timing for the logging
logbroadcast_found = 0

# these types can nicely be imported into Matlab
allowed_types = ['uint8_t','int8_t','uint16_t','int16_t','uint32_t','int32_t','uint64_t','int64_t','float','double']

log_entries = []
# loop through global_data_files ending in _t.h and look for LOGME (per variable) and LOGBROADCAST (overall)
for path in glob.glob( os.path.join(base_path, '*_t.h') ):
    # filename is supposed to be global_data_bapedibup_t.h
    if 'global_data' not in path:   
        print 'path: ' + path
        raise 'wrong filename found'
    f = open(path, 'r')
    access_conf_found = False;
    # strip away ../../../../apps/orb/ and _t.h
    data_name = path.lstrip(base_path)[0:-4]
    # strip away ../../../../apps/orb/ and global_data_ and _t.h
    name = path.lstrip(base_path)[12:-4]
    log_entry = {'data_name': data_name,'name':name,'vars': []}
    
    logbroadcast = False;
    
    # loop throug lines
    for line in f:
    
        line_parts = line.split()
        # access_conf is needed to lock the data
        if 'access_conf_t' in line:
            
            # always use the access_conf which has the LOGBROADCAST flag
            if 'LOGBROADCAST' in line:
                access_conf_found = True
                log_entry['access_conf_name'] = line_parts[1].rstrip(';')
                logbroadcast = True
                print 'LOGBROADCAST found in ' + data_name
                logbroadcast_found += 1
            # but use an access_conf anyway 
            elif access_conf_found == False:
                access_conf_found = True
                log_entry['access_conf_name'] = line_parts[1].rstrip(';')
        # variables flagged with LOGME should be logged
        elif 'LOGME' in line:
            var_entry = {'type': line_parts[0]}
        
            # check that it is an allowed type
            if var_entry['type'] not in allowed_types:
                print 'file: '+ path + ', type: ' + var_entry['type']
                raise 'unsupported type'
            
            # save variable name and number for array
            if '[' in line_parts[1]:
                var_entry['name'] = line_parts[1].split('[')[0]
                var_entry['number'] = line_parts[1].split('[')[1].rstrip('];')
            else:
                var_entry['name'] = line_parts[1].rstrip(';')
                var_entry['number'] = 1
                
            # add the variable
            log_entry['vars'].append(var_entry)
    # only use the global data file if any variables have a LOGME
    if logbroadcast == True and len(log_entry['vars']) > 0:
        logbroadcast_entry = log_entry
    elif len(log_entry['vars']) > 0:
        print 'added ' + log_entry['data_name']
        log_entries.append(log_entry)
    f.close()

# check that we have one and only one LOGBROADCAST
if logbroadcast_found > 1:
    raise 'too many LOGBROADCAST found\n'
elif logbroadcast_found == 0:
    raise 'no LOGBROADCAST found\n'

# write function to c file

header = '/* This file is autogenerated in nuttx/configs/px4fmu/include/updatesdlog.py */\n\
\n\
#ifndef SDLOG_GENERATED_H_\n\
#define SDLOG_GENERATED_H_\n\
\n\
\n\
'

cstruct = 'typedef struct\n{\n'

for j in logbroadcast_entry['vars']:
    cstruct += '\t' + j['type'] + ' ' + logbroadcast_entry['name'] + '_' + j['name']
    if j['number'] == 1:
        cstruct += ';\n'
    else:
        cstruct += '[' + j['number'] + '];\n'

for i in log_entries:
    for j in i['vars']:
        cstruct += '\t' + j['type'] + ' ' + i['name'] + '_' + j['name']
        if j['number'] == 1:
            cstruct += ';\n'
        else:
            cstruct += '[' + j['number'] + '];\n'

cstruct += '\tchar check[4];\n} __attribute__((__packed__)) log_block_t;\n\n'


copy_function = 'void copy_block(log_block_t* log_block)\n{\n'
copy_function += '\tif(global_data_wait(&' + logbroadcast_entry['data_name'] + '->' + logbroadcast_entry['access_conf_name'] + ') == 0)\n\t{\n'

for j in logbroadcast_entry['vars']:
    copy_function += '\t\tmemcpy(&log_block->' + logbroadcast_entry['name'] + '_' + j['name'] + ',&' + logbroadcast_entry['data_name'] + '->' + j['name'] + ',sizeof(' + j['type'] + ')*' + str(j['number']) + ');\n'
    #copy_function += '\t\t}\n'

# generate logging MACRO


for i in log_entries:
    copy_function += '\t\tif(global_data_trylock(&' + i['data_name'] + '->' + i['access_conf_name'] + ') == 0)\n\t\t{\n'
    
    for j in i['vars']:
        copy_function += '\t\t\tmemcpy(&log_block->' + i['name'] + '_' + j['name'] + ',&' + i['data_name'] + '->' + j['name'] + ',sizeof(' + j['type'] + ')*' + str(j['number']) + ');\n'
        
    copy_function += '\t\t\tglobal_data_unlock(&' + i['data_name'] + '->' + i['access_conf_name'] + ');\n'
    copy_function += '\t\t}\n'
copy_function += '\t\tglobal_data_unlock(&' + logbroadcast_entry['data_name'] + '->' + logbroadcast_entry['access_conf_name'] + ');\n'
copy_function += '\t}\n'

copy_function += '}\n'

footer = '\n#endif'



# generate mfile

type_bytes = {
'uint8_t' : 1,
'int8_t' : 1,
'uint16_t' : 2,
'int16_t' : 2,
'uint32_t' : 4,
'int32_t' : 4,
'uint64_t' : 8,
'int64_t' : 8,
'float' : 4,
'double' : 8,
}

type_names_matlab = {
'uint8_t' : 'uint8',
'int8_t' : 'int8',
'uint16_t' : 'uint16',
'int16_t' : 'int16',
'uint32_t' : 'uint32',
'int32_t' : 'int32',
'uint64_t' : 'uint64',
'int64_t' : 'int64',
'float' : 'float',
'double' : 'double',
}


# read template mfile
mf = open(mfile_template, 'r')
mfile_template_string = mf.read()

mfile_define = '#define MFILE_STRING "% This file is autogenerated in updatesdlog.py and mfile.template in apps/sdlog\\n\\\n'
mfile_define += '%% Define logged values \\n\\\n\\n\\\nlogTypes = {};\\n\\\n'

for j in logbroadcast_entry['vars']:
    mfile_define += 'logTypes{end+1} = struct(\'data\',\'' + logbroadcast_entry['name'] + '\',\'variable_name\',\'' + j['name'] + '\',\'type_name\',\'' + type_names_matlab.get(j['type']) + '\',\'type_bytes\',' + str(type_bytes.get(j['type'])) + ',\'number_of_array\',' + str(j['number']) + ');\\n\\\n'

for i in log_entries:
    for j in i['vars']:
        mfile_define += 'logTypes{end+1} = struct(\'data\',\'' + i['name'] + '\',\'variable_name\',\'' + j['name'] + '\',\'type_name\',\'' + type_names_matlab.get(j['type']) + '\',\'type_bytes\',' + str(type_bytes.get(j['type'])) + ',\'number_of_array\',' + str(j['number']) + ');\\n\\\n'
        


mfile_define += '\\\n' + mfile_template_string.replace('\n', '\\n\\\n').replace('\"','\\\"') + '"'


# write to c File
cf = open(cfile, 'w')
cf.write(header)
cf.write(cstruct);
cf.write(copy_function)

cf.write(mfile_define)

cf.write(footer)
cf.close()

print 'finished, cleanbuild needed!'