CvlQuery.java

package fr.inria.diversify.transformation.query;

import fr.inria.diversify.diversification.InputProgram;
import fr.inria.diversify.transformation.Transformation;
import fr.inria.diversify.transformation.cvl.LinkExistence;
import fr.inria.diversify.transformation.cvl.LinkSubstitution;
import fr.inria.diversify.transformation.cvl.ObjectExistence;
import fr.inria.diversify.transformation.cvl.ObjectSubstitution;
import spoon.reflect.declaration.CtClass;
import spoon.reflect.declaration.CtElement;
import spoon.reflect.declaration.CtField;
import spoon.reflect.reference.CtTypeReference;

import java.util.*;

/**
 * User: Simon
 * Date: 26/02/14
 * Time: 11:10
 */
public class CvlQuery extends TransformationQuery {


    protected InputProgram inputProgram;

    public CvlQuery(InputProgram inputProgram) {
        super(inputProgram);
    }

    @Override
    public Transformation query() {
        try {
            Transformation t = null;
            Random r = new Random();
            int i = r.nextInt(8);
            switch (i) {
                case 0:
                    t = getLinkExistence();
                    break;
                case 1:
                    t = getLinkSubstitution();
                    break;
                case 2:
                case 3:
                    t = getObjectExistence();
                    break;
                case 4:
                case 5:
                case 6:
                case 7:
                    t = getObjectSubstitution();
                    break;
            }
            return t;
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public LinkSubstitution getLinkSubstitution() {
        Random r = new Random();

        if(r.nextInt()%2 == 0)
            return getLSForField();
        else
            return getLSForClass();
    }

    protected LinkSubstitution getLSForField() {
        LinkSubstitution ls = new LinkSubstitution();
        List<CtElement> objects = inputProgram.getAllElement(CtField.class);
        Random r = new Random();

        ls.setTransformationPoint(objects.get(r.nextInt(objects.size())));
        ls.setTransplant(objects.get(r.nextInt(objects.size())));

        return ls;
    }

    protected LinkSubstitution getLSForClass() {
        LinkSubstitution ls = new LinkSubstitution();
        List<CtElement> objects = inputProgram.getAllElement(CtClass.class);
        Random r = new Random();

        CtClass cl = (CtClass) objects.get(r.nextInt(objects.size()));
        List<CtTypeReference> set = new ArrayList<CtTypeReference>();
        while (cl.getSimpleName().equals("") || set.isEmpty()) {
            set.clear();
            cl = (CtClass) objects.get(r.nextInt(objects.size()));
            if(cl.getSuperclass() != null)
                set.add(cl.getSuperclass());
            set.addAll(cl.getSuperInterfaces());
        }
        ls.setTransformationPoint(cl);
        ls.setTransplant(objects.get(r.nextInt(objects.size())));
        ls.setClassOrInterfaceSubstitution(set.get(r.nextInt(set.size())));

        return ls;
    }

    public LinkExistence getLinkExistence() {
        Random r = new Random();

        if(r.nextInt()%2 == 0)
            return getLEForField();
        else
            return getLEForClass();
    }

    protected LinkExistence getLEForField() {
        LinkExistence ls = new LinkExistence();
        List<CtElement> objects = inputProgram.getAllElement(CtField.class);
        Random r = new Random();

        ls.setTransformationPoint(objects.get(r.nextInt(objects.size())));

        return ls;
    }

    protected LinkExistence getLEForClass() {
        LinkExistence ls = new LinkExistence();
        List<CtElement> objects = inputProgram.getAllElement(CtClass.class);
        Random r = new Random();

        CtClass cl = (CtClass) objects.get(r.nextInt(objects.size()));
        List<CtTypeReference> set = new ArrayList<CtTypeReference>();
        while (cl.getSimpleName().equals("") || set.isEmpty()) {
            set.clear();
            cl = (CtClass) objects.get(r.nextInt(objects.size()));
            if(cl.getSuperclass() != null)
                set.add(cl.getSuperclass());
            set.addAll(cl.getSuperInterfaces());
        }
        ls.setTransformationPoint(cl);
        ls.setClassOrInterfaceExistance(set.get(r.nextInt(set.size())));

        return ls;
    }

    public ObjectSubstitution getObjectSubstitution() {
        ObjectSubstitution os = new  ObjectSubstitution();
        Random r = new Random();

        List<CtElement> objects = inputProgram.getAllElement(CtElement.class);
        os.setTransformationPoint(objects.get(r.nextInt(objects.size())));

        List<CtElement> transplants = inputProgram.getAllElement(os.getTransformationPoint().getClass());
        os.setTransplant(transplants.get(r.nextInt(transplants.size())));

        return os;
    }

    public ObjectExistence getObjectExistence() {
        ObjectExistence oe = new  ObjectExistence();
        Random r = new Random();
        List<CtElement> objects = inputProgram.getAllElement(CtElement.class);
        oe.setTransformationPoint(objects.get(r.nextInt(objects.size())));
        return oe;
    }


}