# -*- coding: utf-8 -*-

""" CMake generation module
    @file
"""

import os
import platform
import datetime
from jinja2 import Environment, FileSystemLoader

class CMake (object):
    
    def __init__(self, project, path):
        
        self.path = path
        self.project = project
        self.context = {}
        
    def populateCMake (self):
        """ Generate CMakeList.txt file for building the project
        """

        # For debug run cmake -DCMAKE_BUILD_TYPE=Debug or Release
        cmake = {}
        #fpu = '-mfpu=fpv5-sp-d16 -mfloat-abi=softfp'
        fpu = ''
        
        core = ''
        
        if 'STM32F0' in self.project['chip']:
            core = '-mcpu=cortex-m0'
        elif 'STM32F1' in self.project['chip']:
            core = '-mcpu=cortex-m3'
        elif 'STM32F2' in self.project['chip']:
            core = '-mcpu=cortex-m3'
        elif 'STM32F3' in self.project['chip']:
            core = '-mcpu=cortex-m4'
        elif 'STM32F4' in self.project['chip']:
            core = '-mcpu=cortex-m4'
        elif 'STM32F7' in self.project['chip']:
            core = '-mcpu=cortex-m7'
        elif 'STM32L0' in self.project['chip']:
            core = '-mcpu=cortex-m0plus'
        elif 'STM32L1' in self.project['chip']:
            core = '-mcpu=cortex-m3'
        elif 'STM32L4' in self.project['chip']:
            core = '-mcpu=cortex-m4'
            
        cmake['version'] = '3.1'
        cmake['project'] = self.project['name']
        cmake['incs'] = []
        for inc in self.project['incs']:    
            cmake['incs'].append(inc)    
        cmake['srcs'] = []

        i=0
                
        cmake['files']=[]
        cmake['ass']=[]
        
        for file in self.project['srcs']:
            if file.endswith('.c') or file.endswith('.h'):
                cmake['files'].append({'path': file,'var':'SRC_FILE' + str(i)})  
                i = i+1
            
        for file in self.project['files']:
            print ('Assembly added ' + file)
            cmake['ass'].append({'path': file})  
            cmake['files'].append({'path': file,'var':'SRC_FILE' + str(i)})
            i = i+1
            
        cmake['cxx'] = 'false'
        
        cmake['c_flags'] = '-g -Wextra -Wshadow -Wimplicit-function-declaration -Wredundant-decls -Wmissing-prototypes -Wstrict-prototypes -fno-common -ffunction-sections -fdata-sections -MD -Wall -Wundef -mthumb ' + core + ' ' + fpu

        cmake['cxx_flags'] = '-Wextra -Wshadow -Wredundant-decls  -Weffc++ -fno-common -ffunction-sections -fdata-sections -MD -Wall -Wundef -mthumb ' + core + ' ' + fpu
 
        cmake['asm_flags'] = '-g -mthumb ' + core + ' ' + fpu #+ ' -x assembler-with-cpp'
        cmake['linker_flags'] = '-g -Wl,--gc-sections -Wl,-Map=' + cmake['project'] + '.map -mthumb ' + core + ' ' + fpu
        cmake['linker_script'] = 'STM32FLASH.ld'
        cmake['linker_path'] = ''  
   
        self.linkerScript('STM32FLASH.ld',os.path.join(self.path,'STM32FLASH.ld'))
        
        cmake['oocd_target'] = 'stm32f3x'
        cmake['defines'] = []
        for define in self.project['defs']:
            cmake['defines'].append(define)
            
        cmake['libs'] = []
        
        self.context['cmake'] = cmake
                
        self.generateFile('CMakeLists.txt',os.path.join(self.path,'CMakeLists.txt'))

        print ('Created file CMakeLists.txt')
        
#    def generateFile (self, pathSrc, pathDst='', author='Pegasus', version='v1.0.0', licence='licence.txt', template_dir='../PegasusTemplates'):
    def generateFile (self, pathSrc, pathDst='', author='Pegasus', version='v1.0.0', licence='licence.txt', template_dir='.'):
        
        if (pathDst == ''):
            pathDst = pathSrc
            
        self.context['file'] = os.path.basename(str(pathSrc))
        self.context['author'] = author
        self.context['date'] = datetime.date.today().strftime('%d, %b %Y')
        self.context['version'] = version
        self.context['licence'] = licence
        
        env = Environment(loader=FileSystemLoader(template_dir),trim_blocks=True,lstrip_blocks=True)
        template = env.get_template(str(pathSrc))
        
        generated_code = template.render(self.context)
            
        if platform.system() == 'Window':    

            with open(pathDst, 'wb') as f:
                f.write(generated_code)
        
        elif platform.system() == 'Linux':

            with open(pathDst, 'wb') as f:
                f.write(str.encode(generated_code))        
        else:
            # Different OS than Windows or Linux            
            pass
    
    def linkerScript(self,pathSrc, pathDst='',template_dir='.'):
#    def linkerScript(self,pathSrc, pathDst='',template_dir='.../PegasusTemplates'):
                
        if (pathDst == ''):
            pathDst = pathSrc
            
        self.context['file'] = os.path.basename(str(pathSrc))
        self.context['flash'] = '64'
        self.context['ram'] = '8'        
        
        env = Environment(loader=FileSystemLoader(template_dir),trim_blocks=True,lstrip_blocks=True)
        template = env.get_template(str(pathSrc))
        
        generated_code = template.render(self.context)
            
        if platform.system() == 'Window':    

            with open(pathDst, 'wb') as f:
                f.write(generated_code)
        
        elif platform.system() == 'Linux':

            with open(pathDst, 'wb') as f:
                f.write(str.encode(generated_code))        
        else:
            # Different OS than Windows or Linux            
            pass