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, PuginsRootInstance):

    14

        self.PuginsRootInstance = PuginsRootInstance

    15

        self.logger = PuginsRootInstance.logger

    16

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

    17

        self.buildpath = PuginsRootInstance._getBuildPath()

    18

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

    19

        self.md5key = None

    20

        self.srcmd5 = {}

    21

    22

    def GetBinaryCode(self):

    23

        try:

    24

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

    25

        except Exception, e:

    26

            return None

    27

    28

    def _GetMD5FileName(self):

    29

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

    30

    31

    def GetBinaryCodeMD5(self):

    32

        if self.md5key is not None:

    33

            return self.md5key

    34

        else:

    35

            try:

    36

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

    37

            except Exception, e:

    38

                return None

    39

    40

    def check_and_update_hash_and_deps(self, bn):

    41

        # Get latest computed hash and deps

    42

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

    43

        # read source

    44

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

    45

        # compute new hash

    46

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

    47

        # compare

    48

        match = (oldhash == newhash)

    49

        if not match:

    50

            # file have changed

    51

            # update direct dependencies

    52

            deps = []

    53

            for l in src.splitlines():

    54

                res = includes_re.match(l)

    55

                if res is not None:

    56

                    depfn = res.groups()[0]

    57

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

    58

                        #print bn + " depends on "+depfn

    59

                        deps.append(depfn)

    60

            # store that hashand deps

    61

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

    62

        # recurse through deps

    63

        # TODO detect cicular deps.

    64

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

    65

    66

    def build(self):

    67

        # Retrieve toolchain user parameters

    68

        toolchain_params = self.PuginsRootInstance.BeremizRoot.getTargetType().getcontent()["value"]

    69

        self.compiler = toolchain_params.getCompiler()

    70

        self._CFLAGS = toolchain_params.getCFLAGS()

    71

        self.linker = toolchain_params.getLinker()

    72

        self._LDFLAGS = toolchain_params.getLDFLAGS()

    73

    74

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

    75

        obns = []

    76

        objs = []

    77

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

    78

            if Location:

    79

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

    80

            else:

    81

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

    82

    83

            relink = False

    84

            for CFile, CFLAGS in CFilesAndCFLAGS:

    85

                bn = os.path.basename(CFile)

    86

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

    87

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

    88

    89

                match = self.check_and_update_hash_and_deps(bn)

    90

    91

                if match:

    92

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

    93

                else:

    94

                    relink = True

    95

    96

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

    97

    98

                    status, result, err_result = ProcessLogger(

    99

                           self.logger,

   100

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

   101

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

   102

                           ).spin()

   103

   104

                    if status :

   105

                        self.logger.write_error("C compilation of "+ bn +" failed.\n")

   106

                        return False

   107

   108

                obns.append(obn)

   109

                objs.append(objectfilename)

   110

   111

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

   112

        # Link all the object files into one binary file

   113

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

   114

        if relink:

   115

            objstring = []

   116

   117

            # Generate list .o files

   118

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

   119

   120

            ALLldflags = ' '.join(self.CustomLDFLAGS+self.PuginsRootInstance.LDFLAGS+[self._LDFLAGS])

   121

   122

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

   123

   124

            status, result, err_result = ProcessLogger(

   125

                   self.logger,

   126

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

   127

                       (self.linker,

   128

                        listobjstring,

   129

                        self.exe_path,

   130

                        ALLldflags)

   131

                   ).spin()

   132

   133

            if status :

   134

                return False

   135

            else :

   136

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

   137

                data=self.GetBinaryCode()

   138

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

   139

   140

                # Store new PLC filename based on md5 key

   141

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

   142

                f.write(self.md5key)

   143

                f.close()

   144

        else:

   145

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

   146

   147

   148

        return True

   149