1

#!/usr/bin/env python

     2

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

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

# This program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program is distributed in the hope that it will be useful,

    17

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

import os, re, operator

    26

from util.ProcessLogger import ProcessLogger

    27

import hashlib

    28

    29

includes_re =  re.compile('\s*#include\s*["<]([^">]*)[">].*')

    30

    31

class toolchain_gcc():

    32

    """

    33

    This abstract class contains GCC specific code.

    34

    It cannot be used as this and should be inherited in a target specific

    35

    class such as target_linux or target_win32

    36

    """

    37

    def __init__(self, CTRInstance):

    38

        self.CTRInstance = CTRInstance

    39

        self.buildpath = None

    40

        self.SetBuildPath(self.CTRInstance._getBuildPath())

    41

    42

    def getBuilderCFLAGS(self):

    43

        """

    44

        Returns list of builder specific CFLAGS

    45

        """

    46

        return [self.CTRInstance.GetTarget().getcontent().getCFLAGS()]

    47

    48

    def getBuilderLDFLAGS(self):

    49

        """

    50

        Returns list of builder specific LDFLAGS

    51

        """

    52

        return self.CTRInstance.LDFLAGS + \

    53

               [self.CTRInstance.GetTarget().getcontent().getLDFLAGS()]

    54

    55

    def GetBinaryCode(self):

    56

        try:

    57

            return open(self.exe_path, "rb").read()

    58

        except Exception, e:

    59

            return None

    60

    61

    def _GetMD5FileName(self):

    62

        return os.path.join(self.buildpath, "lastbuildPLC.md5")

    63

    64

    def ResetBinaryCodeMD5(self):

    65

        self.md5key = None

    66

        try:

    67

            os.remove(self._GetMD5FileName())

    68

        except Exception, e:

    69

            pass

    70

    71

    def GetBinaryCodeMD5(self):

    72

        if self.md5key is not None:

    73

            return self.md5key

    74

        else:

    75

            try:

    76

                return open(self._GetMD5FileName(), "r").read()

    77

            except Exception, e:

    78

                return None

    79

    80

    def SetBuildPath(self, buildpath):

    81

        if self.buildpath != buildpath:

    82

            self.buildpath = buildpath

    83

            self.exe = self.CTRInstance.GetProjectName() + self.extension

    84

            self.exe_path = os.path.join(self.buildpath, self.exe)

    85

            self.md5key = None

    86

            self.srcmd5 = {}

    87

    88

    def check_and_update_hash_and_deps(self, bn):

    89

        # Get latest computed hash and deps

    90

        oldhash, deps = self.srcmd5.get(bn,(None,[]))

    91

        # read source

    92

        src = open(os.path.join(self.buildpath, bn)).read()

    93

        # compute new hash

    94

        newhash = hashlib.md5(src).hexdigest()

    95

        # compare

    96

        match = (oldhash == newhash)

    97

        if not match:

    98

            # file have changed

    99

            # update direct dependencies

   100

            deps = []

   101

            for l in src.splitlines():

   102

                res = includes_re.match(l)

   103

                if res is not None:

   104

                    depfn = res.groups()[0]

   105

                    if os.path.exists(os.path.join(self.buildpath, depfn)):

   106

                        #print bn + " depends on "+depfn

   107

                        deps.append(depfn)

   108

            # store that hashand deps

   109

            self.srcmd5[bn] = (newhash, deps)

   110

        # recurse through deps

   111

        # TODO detect cicular deps.

   112

        return reduce(operator.and_, map(self.check_and_update_hash_and_deps, deps), match)

   113

   114

    def build(self):

   115

        # Retrieve toolchain user parameters

   116

        toolchain_params = self.CTRInstance.GetTarget().getcontent()

   117

        self.compiler = toolchain_params.getCompiler()

   118

        self.linker = toolchain_params.getLinker()

   119

   120

        Builder_CFLAGS = ' '.join(self.getBuilderCFLAGS())

   121

   122

        ######### GENERATE OBJECT FILES ########################################

   123

        obns = []

   124

        objs = []

   125

        relink = self.GetBinaryCode() is None

   126

        for Location, CFilesAndCFLAGS, DoCalls in self.CTRInstance.LocationCFilesAndCFLAGS:

   127

            if CFilesAndCFLAGS:

   128

                if Location :

   129

                    self.CTRInstance.logger.write(".".join(map(str,Location))+" :\n")

   130

                else:

   131

                    self.CTRInstance.logger.write(_("PLC :\n"))

   132

   133

            for CFile, CFLAGS in CFilesAndCFLAGS:

   134

                if CFile.endswith(".c"):

   135

                    bn = os.path.basename(CFile)

   136

                    obn = os.path.splitext(bn)[0]+".o"

   137

                    objectfilename = os.path.splitext(CFile)[0]+".o"

   138

   139

                    match = self.check_and_update_hash_and_deps(bn)

   140

   141

                    if match:

   142

                        self.CTRInstance.logger.write("   [pass]  "+bn+" -> "+obn+"\n")

   143

                    else:

   144

                        relink = True

   145

   146

                        self.CTRInstance.logger.write("   [CC]  "+bn+" -> "+obn+"\n")

   147

   148

                        status, result, err_result = ProcessLogger(

   149

                               self.CTRInstance.logger,

   150

                               "\"%s\" -c \"%s\" -o \"%s\" %s %s"%

   151

                                   (self.compiler, CFile, objectfilename, Builder_CFLAGS, CFLAGS)

   152

                               ).spin()

   153

   154

                        if status :

   155

                            self.srcmd5.pop(bn)

   156

                            self.CTRInstance.logger.write_error(_("C compilation of %s failed.\n")%bn)

   157

                            return False

   158

                    obns.append(obn)

   159

                    objs.append(objectfilename)

   160

                elif CFile.endswith(".o"):

   161

                    obns.append(os.path.basename(CFile))

   162

                    objs.append(CFile)

   163

   164

        ######### GENERATE library FILE ########################################

   165

        # Link all the object files into one binary file

   166

        self.CTRInstance.logger.write(_("Linking :\n"))

   167

        if relink:

   168

            objstring = []

   169

   170

            # Generate list .o files

   171

            listobjstring = '"' + '"  "'.join(objs) + '"'

   172

   173

            ALLldflags = ' '.join(self.getBuilderLDFLAGS())

   174

   175

            self.CTRInstance.logger.write("   [CC]  " + ' '.join(obns)+" -> " + self.exe + "\n")

   176

   177

            status, result, err_result = ProcessLogger(

   178

                   self.CTRInstance.logger,

   179

                   "\"%s\" %s -o \"%s\" %s"%

   180

                       (self.linker,

   181

                        listobjstring,

   182

                        self.exe_path,

   183

                        ALLldflags)

   184

                   ).spin()

   185

   186

            if status :

   187

                return False

   188

   189

        else:

   190

            self.CTRInstance.logger.write("   [pass]  " + ' '.join(obns)+" -> " + self.exe + "\n")

   191

   192

        # Calculate md5 key and get data for the new created PLC

   193

        data=self.GetBinaryCode()

   194

        self.md5key = hashlib.md5(data).hexdigest()

   195

   196

        # Store new PLC filename based on md5 key

   197

        f = open(self._GetMD5FileName(), "w")

   198

        f.write(self.md5key)

   199

        f.close()

   200

   201

        return True

   202