1

import os, re, operator

     2

from wxPopen import ProcessLogger

     3

import hashlib

     4

     5

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

     6

     7

class toolchain_gcc():

     8

    """

     9

    This abstract class contains GCC specific code.

    10

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

    11

    class such as target_linux or target_win32

    12

    """

    13

    def __init__(self, PluginsRootInstance):

    14

        self.PluginsRootInstance = PluginsRootInstance

    15

        self.buildpath = None

    16

        self.SetBuildPath(self.PluginsRootInstance._getBuildPath())

    17

    18

    def getBuilderCFLAGS(self):

    19

        """

    20

        Returns list of builder specific CFLAGS

    21

        """

    22

        return [self.PluginsRootInstance.GetTarget().getcontent()["value"].getCFLAGS()]

    23

    24

    def getBuilderLDFLAGS(self):

    25

        """

    26

        Returns list of builder specific LDFLAGS

    27

        """

    28

        return self.PluginsRootInstance.LDFLAGS + \

    29

               [self.PluginsRootInstance.GetTarget().getcontent()["value"].getLDFLAGS()]

    30

    31

    def GetBinaryCode(self):

    32

        try:

    33

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

    34

        except Exception, e:

    35

            return None

    36

    37

    def _GetMD5FileName(self):

    38

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

    39

    40

    def GetBinaryCodeMD5(self):

    41

        if self.md5key is not None:

    42

            return self.md5key

    43

        else:

    44

            try:

    45

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

    46

            except Exception, e:

    47

                return None

    48

    49

    def SetBuildPath(self, buildpath):

    50

        if self.buildpath != buildpath:

    51

            self.buildpath = buildpath

    52

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

    53

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

    54

            self.md5key = None

    55

            self.srcmd5 = {}

    56

    57

    def check_and_update_hash_and_deps(self, bn):

    58

        # Get latest computed hash and deps

    59

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

    60

        # read source

    61

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

    62

        # compute new hash

    63

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

    64

        # compare

    65

        match = (oldhash == newhash)

    66

        if not match:

    67

            # file have changed

    68

            # update direct dependencies

    69

            deps = []

    70

            for l in src.splitlines():

    71

                res = includes_re.match(l)

    72

                if res is not None:

    73

                    depfn = res.groups()[0]

    74

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

    75

                        #print bn + " depends on "+depfn

    76

                        deps.append(depfn)

    77

            # store that hashand deps

    78

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

    79

        # recurse through deps

    80

        # TODO detect cicular deps.

    81

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

    82

    83

    def build(self):

    84

        # Retrieve toolchain user parameters

    85

        toolchain_params = self.PluginsRootInstance.GetTarget().getcontent()["value"]

    86

        self.compiler = toolchain_params.getCompiler()

    87

        self.linker = toolchain_params.getLinker()

    88

    89

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

    90

    91

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

    92

        obns = []

    93

        objs = []

    94

        relink = False

    95

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

    96

            if Location:

    97

                self.PluginsRootInstance.logger.write(_("Plugin : ") + self.PluginsRootInstance.GetChildByIECLocation(Location).GetCurrentName() + " " + str(Location)+"\n")

    98

            else:

    99

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

   100

   101

            for CFile, CFLAGS in CFilesAndCFLAGS:

   102

                bn = os.path.basename(CFile)

   103

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

   104

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

   105

   106

                match = self.check_and_update_hash_and_deps(bn)

   107

   108

                if match:

   109

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

   110

                else:

   111

                    relink = True

   112

   113

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

   114

   115

                    status, result, err_result = ProcessLogger(

   116

                           self.PluginsRootInstance.logger,

   117

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

   118

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

   119

                           ).spin()

   120

   121

                    if status :

   122

                        self.srcmd5.pop(bn)

   123

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

   124

                        return False

   125

   126

                obns.append(obn)

   127

                objs.append(objectfilename)

   128

   129

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

   130

        # Link all the object files into one binary file

   131

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

   132

        if relink:

   133

            objstring = []

   134

   135

            # Generate list .o files

   136

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

   137

   138

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

   139

   140

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

   141

   142

            status, result, err_result = ProcessLogger(

   143

                   self.PluginsRootInstance.logger,

   144

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

   145

                       (self.linker,

   146

                        listobjstring,

   147

                        self.exe_path,

   148

                        ALLldflags)

   149

                   ).spin()

   150

   151

            if status :

   152

                return False

   153

            else :

   154

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

   155

                data=self.GetBinaryCode()

   156

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

   157

   158

                # Store new PLC filename based on md5 key

   159

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

   160

                f.write(self.md5key)

   161

                f.close()

   162

        else:

   163

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

   164

   165

   166

        return True

   167