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.logger = PluginsRootInstance.logger

    16

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

    17

        self.buildpath = PluginsRootInstance._getBuildPath()

    18

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

    19

        self.md5key = None

    20

        self.srcmd5 = {}

    21

    22

    def getTarget(self):

    23

        target = self.PluginsRootInstance.BeremizRoot.getTargetType()

    24

        if target.getcontent() is None:

    25

            target = self.PluginsRootInstance.GetDefaultTarget()

    26

        return target

    27

    28

    def getBuilderCFLAGS(self):

    29

        """

    30

        Returns list of builder specific CFLAGS

    31

        """

    32

        return [self.getTarget().getcontent()["value"].getCFLAGS()]

    33

    34

    def getBuilderLDFLAGS(self):

    35

        """

    36

        Returns list of builder specific LDFLAGS

    37

        """

    38

        return self.PluginsRootInstance.LDFLAGS + \

    39

               [self.getTarget().getcontent()["value"].getLDFLAGS()]

    40

    41

    def GetBinaryCode(self):

    42

        try:

    43

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

    44

        except Exception, e:

    45

            return None

    46

    47

    def _GetMD5FileName(self):

    48

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

    49

    50

    def GetBinaryCodeMD5(self):

    51

        if self.md5key is not None:

    52

            return self.md5key

    53

        else:

    54

            try:

    55

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

    56

            except Exception, e:

    57

                return None

    58

    59

    def check_and_update_hash_and_deps(self, bn):

    60

        # Get latest computed hash and deps

    61

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

    62

        # read source

    63

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

    64

        # compute new hash

    65

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

    66

        # compare

    67

        match = (oldhash == newhash)

    68

        if not match:

    69

            # file have changed

    70

            # update direct dependencies

    71

            deps = []

    72

            for l in src.splitlines():

    73

                res = includes_re.match(l)

    74

                if res is not None:

    75

                    depfn = res.groups()[0]

    76

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

    77

                        #print bn + " depends on "+depfn

    78

                        deps.append(depfn)

    79

            # store that hashand deps

    80

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

    81

        # recurse through deps

    82

        # TODO detect cicular deps.

    83

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

    84

    85

    def build(self):

    86

        # Retrieve toolchain user parameters

    87

        toolchain_params = self.getTarget().getcontent()["value"]

    88

        self.compiler = toolchain_params.getCompiler()

    89

        self.linker = toolchain_params.getLinker()

    90

    91

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

    92

    93

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

    94

        obns = []

    95

        objs = []

    96

        relink = False

    97

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

    98

            if Location:

    99

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

   100

            else:

   101

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

   102

   103

            for CFile, CFLAGS in CFilesAndCFLAGS:

   104

                bn = os.path.basename(CFile)

   105

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

   106

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

   107

   108

                match = self.check_and_update_hash_and_deps(bn)

   109

   110

                if match:

   111

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

   112

                else:

   113

                    relink = True

   114

   115

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

   116

   117

                    status, result, err_result = ProcessLogger(

   118

                           self.logger,

   119

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

   120

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

   121

                           ).spin()

   122

   123

                    if status :

   124

                        self.srcmd5.pop(bn)

   125

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

   126

                        return False

   127

   128

                obns.append(obn)

   129

                objs.append(objectfilename)

   130

   131

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

   132

        # Link all the object files into one binary file

   133

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

   134

        if relink:

   135

            objstring = []

   136

   137

            # Generate list .o files

   138

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

   139

   140

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

   141

   142

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

   143

   144

            status, result, err_result = ProcessLogger(

   145

                   self.logger,

   146

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

   147

                       (self.linker,

   148

                        listobjstring,

   149

                        self.exe_path,

   150

                        ALLldflags)

   151

                   ).spin()

   152

   153

            if status :

   154

                return False

   155

            else :

   156

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

   157

                data=self.GetBinaryCode()

   158

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

   159

   160

                # Store new PLC filename based on md5 key

   161

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

   162

                f.write(self.md5key)

   163

                f.close()

   164

        else:

   165

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

   166

   167

   168

        return True

   169