mutation_operator.py

import random
import os
class operator:
    path_list=os.getcwd()+"\plc\plc6_list.txt"
    line_no=0
    check=0
    ori=[]
    def __init__(self, path):
        self.path = path
        self.ori=open(path, 'rU').readlines()
        self.line_no = random.randrange(0,len(self.ori))
        self.line =self.ori[self.line_no].strip('\n')

    def indices(self, element):
        result = []
        offset = -1
        while True:
            try:
                offset = self.line.index(element, offset + 1)
            except ValueError:
                return result
            result.append(offset)

    def choose(self):
        print (self.line)
    def operator_ABS(self):
        numlist=['0','1','2','3','4','5','6','7','8','9']
        result = []
        i=[]
        # print self.line
        for num in numlist:
            result.append(self.indices(num))
        for j in range(0,len(numlist)):
            if result[j]!=[]:
                i.append(j)
        #     print j
        # print i
        if i!=[]:
            numpos=i[random.randrange(0, len(i))]
            numresult= result[numpos]
            cpos=numresult[random.randrange(0, len(numresult))]
            # print "numpos",numpos
            # print numresult
            # print cpos
            newlist = numlist
            del newlist[numpos]
            choosenum = newlist[random.randrange(0, len(newlist))]
            newline=self.line[0:cpos]+choosenum+self.line[cpos+1:]
            self.check=1
            print (newline)
            print (self.line)
            self.line = newline
        else:
            self.check = 0
            print ("error")

    def operator_AOR(self):
        Aolist=['+','-','*','/','%']
        result = []
        i=[]
        # print self.line
        for op in Aolist:
            result.append(self.indices(op))
        for j in range(0,len(Aolist)):
            if result[j]!=[]:
                i.append(j)
        #     print j
        # print i
        if i!=[]:
            oppos=i[random.randrange(0, len(i))]
            opresult= result[oppos]
            cpos=opresult[random.randrange(0, len(opresult))]
            # print "oppos",oppos
            # print opresult
            # print cpos
            newlist = Aolist
            del newlist[oppos]
            chooseop = newlist[random.randrange(0, len(newlist))]
            newline=self.line[0:cpos]+chooseop+self.line[cpos+1:]
            print (newline)
            print (self.line)
            self.line=newline
            self.check = 1
        else:
            print ("error")

    def operator_ASR(self):
        Asolist = ['+=', '-=', '*=', '/=', '%=']
        result = []
        i = []
        # print self.line
        for op in Asolist:
            result.append(self.indices(op))
        for j in range(0, len(Asolist)):
            if result[j] != []:
                i.append(j)
        # print j
        # print i
        if i != []:
            oppos = i[random.randrange(0, len(i))]
            opresult = result[oppos]
            cpos = opresult[random.randrange(0, len(opresult))]
            # print "oppos",oppos
            # print opresult
            # print cpos
            newlist = Asolist
            del newlist[oppos]
            chooseop = newlist[random.randrange(0, len(newlist))]
            newline = self.line[0:cpos] + chooseop + self.line[cpos + 1:]
            print (newline)
            print (self.line)
            self.line = newline
            self.check = 1
        else:
            print ("error")

    def operator_ROR(self):
        Asolist = ['>', '<', '>=', '<=', '==','!=','<>']
        result = []
        i = []
        # print self.line
        for op in Asolist:
            result.append(self.indices(op))
        for j in range(0, len(Asolist)):
            if result[j] != []:
                i.append(j)
        # print j
        # print i
        if i != []:
            oppos = i[random.randrange(0, len(i))]
            opresult = result[oppos]
            cpos = opresult[random.randrange(0, len(opresult))]
            # print "oppos",oppos
            # print opresult
            # print cpos
            newlist = Asolist
            del newlist[oppos]
            chooseop = newlist[random.randrange(0, len(newlist))]
            newline = self.line[0:cpos] + chooseop + self.line[cpos + 1:]
            print (newline)
            print (self.line)
            self.line = newline
            self.check = 1
        else:
            print ("error")

    def operator_COR(self):
        try:
            cpos = self.line.index('if')
            num = random.randrange(0, 2)
            if num == 0:
                newline = self.line[0:cpos] + 'if True:'
            else:
                newline = self.line[0:cpos] + 'if False:'
            print (self.line)
            print (newline)
            self.line = newline
            self.check = 1
        except:
            print ("error")

    def gen(self,mutantnum):
        if self.check == 1:
            with open(self.path_list, 'a') as f2:
                f2.write("Mutation number:"+str(mutantnum)+"\r\n")
                f2.write(self.ori[self.line_no].strip('\n'))
                f2.write("   >>>>>   ")
                f2.write(self.line)
                f2.write('\n\r')
            path_new=self.path[0:len(self.path)-3]+"_"+str(mutantnum)+".py"
            count1 = len(open(self.path, 'rU').readlines())
            self.ori[self.line_no]=self.line+"\r\n"
            with open(path_new, 'a') as f1:
                for i in range(0,len(self.ori)):
                    f1.write(self.ori[i])
            return True
        else:
            print ("No generation")
            return False





# +,-,*,/