FelixMcFelix/laughing-ironman

/* global Sketch */
/* global MVM */
//Class definitions for the lookup table.

Sketch.MultiKeyTable = function(){
    this.store = {};
};

Sketch.MultiKeyTable.prototype = {
    add: function(operand, keys, value, predicate, extraSelector){
        var entry = new Sketch.MultiKeyTableEntry(this, operand, keys, value);

        if(!this.store[operand]){
            this.store[operand] = {};
        }

        var cursor = this.store[operand];

        for(var i = 0; i<keys.length; i++){
            if(!cursor[keys[i]]){
                cursor[keys[i]] = {};
            }

            cursor = cursor[keys[i]];
        }

        cursor.content = entry;
        cursor.predicate = predicate;
        cursor.extraSelector = extraSelector;

        return entry;
    },

    lookup: function(operand, keys){
        try{
            var k = this.store[operand];

            for(var i = 0; i< keys.length; i++){
                k = k[keys[i].type];
            }

            var out;

            if (k.content && k.predicate){
                var pred = k.predicate(keys);
                if(pred.answer){
                    out = k.content;
                } else{
                    throw pred.reason;
                }
            } else if (k.content) {
                out = k.content;
            } else{
                throw "No associated entry...";
            }

            if(k.extraSelector){
                out.extra = k.extraSelector(keys);
            }

            return out;
        } catch(e){
            var keysStr = "";
            keys.forEach(function(curr, ind, arr){keysStr+=curr.type; if(ind!==arr.length-1){keysStr+=", ";}});

            throw "Operand and key combination not found for: "+operand+" and "+keysStr+": \n"+e;
        }
    }
};

Sketch.MultiKeyTableEntry = function(table, operand, keys, value){
    this.parent = table;
    this.operand = operand;
    this.keys = keys;
    this.value = value;
};

Sketch.MultiKeyTableEntry.prototype = {
    reflexive: function(){
        this.parent.add(this.operand, this.keys.reverse(), this.value);
    }
};

Sketch.OpCheckValue = function(type, code){
    this.type = type;
    this.code = code;
};

Sketch.SketchGenOperandTable = new Sketch.MultiKeyTable();

/* TEMPLATE
Sketch.SketchGenOperandTable.add("+", ["num", "num"],
                              new Sketch.OpCheckValue("num", MVM.opCodes.FADD)
                            )
                            .reflexive();

Sketch.SketchGenOperandTable.add("+", ["num", "num"], new Sketch.OpCheckValue("num", MVM.opCodes.FADD))
                            .reflexive();
*/

//OUR TYPES ARE num, bool, point, line, polygon, void.

//---//
// + //
//---//
Sketch.SketchGenOperandTable.add("+", ["num", "num"],
                              new Sketch.OpCheckValue("num", MVM.opCodes.FADD)
                            );

var pointPointChecker = function(keys){
    var out;
    if(keys[0].extra.size>2){
        out = {answer: false, reason: "Shapes over 2D are not yet supported."};
    } else{
        out = {
              answer: keys[0].extra.size === keys[1].extra.size,
              reason: "Mismatch between point sizes at addition: "+keys[0].extra.size+" !== "+keys[1].extra.size
          };
    }

    return out;
};

var extraOfSecond = function(keys){
    return keys[1].extra;
};

Sketch.SketchGenOperandTable.add("+", ["point", "point"],
                              new Sketch.OpCheckValue("line", MVM.opCodes.PTADD),
                              pointPointChecker,
                              extraOfSecond
                            );

//TODO: add opcode.
Sketch.SketchGenOperandTable.add("+", ["point", "line"],
                              new Sketch.OpCheckValue("polygon", MVM.opCodes.AUGPT)
                            )
                            .reflexive();

//TODO: add opcode.
Sketch.SketchGenOperandTable.add("+", ["point", "polygon"],
                              new Sketch.OpCheckValue("polygon", MVM.opCodes.AUGPT)
                            )
                            .reflexive();

//---//
// - //
//---//
Sketch.SketchGenOperandTable.add("-", ["num", "num"],
                              new Sketch.OpCheckValue("num", MVM.opCodes.FSUB)
                            );

//---//
// * //
//---//
Sketch.SketchGenOperandTable.add("*", ["num", "num"],
                              new Sketch.OpCheckValue("num", MVM.opCodes.FMUL)
                            );

//---//
// / //
//---//
Sketch.SketchGenOperandTable.add("/", ["num", "num"],
                              new Sketch.OpCheckValue("num", MVM.opCodes.FDIV)
                            );

//---//
// % //
//---//
Sketch.SketchGenOperandTable.add("%", ["num", "num"],
                              new Sketch.OpCheckValue("num", MVM.opCodes.FMOD)
                            );

//----//
// ++ //
//----//
Sketch.SketchGenOperandTable.add("++", ["ident"],
                              new Sketch.OpCheckValue("num", null)
                            );

//----//
// -- //
//----//
Sketch.SketchGenOperandTable.add("--", ["ident"],
                              new Sketch.OpCheckValue("num", null)
                            );

//----//
// && //
//----//
Sketch.SketchGenOperandTable.add("&&", ["bool", "bool"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.BAND)
                            );

//----//
// || //
//----//
Sketch.SketchGenOperandTable.add("||", ["bool", "bool"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.BOR)
                            );


//----//
// ?= //
//----//
Sketch.SketchGenOperandTable.add("?=", ["bool", "bool"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.CMPEQ)
                            );
Sketch.SketchGenOperandTable.add("?=", ["num", "num"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.CMPEQ)
                            );
Sketch.SketchGenOperandTable.add("?=", ["point", "point"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.CMPEQ)
                            );
Sketch.SketchGenOperandTable.add("?=", ["line", "line"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.CMPEQ)
                            );
Sketch.SketchGenOperandTable.add("?=", ["polygon", "polygon"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.CMPEQ)
                            );

//----//
// ?< //
//----//
Sketch.SketchGenOperandTable.add("?<", ["num", "num"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.CMPLT)
                            );

//----//
// ?> //
//----//
Sketch.SketchGenOperandTable.add("?>", ["num", "num"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.CMPGT)
                            );

//---//
// ! //
//---//
Sketch.SketchGenOperandTable.add("!", ["bool"],
                              new Sketch.OpCheckValue("bool", MVM.opCodes.BNEG)
                            );

//------//
// draw //
//------//
Sketch.SketchGenOperandTable.add("draw", ["line"],
                              new Sketch.OpCheckValue(null, MVM.opCodes.LNDRAW)
                            );

Sketch.SketchGenOperandTable.add("draw", ["polygon"],
                              new Sketch.OpCheckValue(null, MVM.opCodes.PGDRAW)
                            );

//---//
// ~ //
//---//
var colourSizeCheck = function(keys){
    var target = keys[keys.length - 1];
    return {
        answer: (target.extra === undefined) || (target.extra.size === 3) || (target.extra.size === 4) , //Assume the user knows what the hell they're doing if they pass a colour to a function.
        reason: "Size of a colour must be 3 or 4 numbers." 
    };
}

Sketch.SketchGenOperandTable.add("~", ["line", "point"],
                              new Sketch.OpCheckValue("line", MVM.opCodes.SETCOLOUR),
                              colourSizeCheck
                            );
Sketch.SketchGenOperandTable.add("~", ["polygon", "point"],
                              new Sketch.OpCheckValue("polygon", MVM.opCodes.SETCOLOUR),
                              colourSizeCheck
                            );

//----//
// -> //
//----//
var pointIs2d = function(keys){
    return {
        answer: (keys[1].extra === undefined) || (keys[1].extra.size === 2),
        reason: "Shapes over 2D are not yet supported."
    };
};

Sketch.SketchGenOperandTable.add("->", ["point", "point"],
                              new Sketch.OpCheckValue("point", MVM.opCodes.TRANSLATEPT),
                              pointPointChecker,
                              extraOfSecond
                            );
Sketch.SketchGenOperandTable.add("->", ["line", "point"],
                              new Sketch.OpCheckValue("line", MVM.opCodes.TRANSLATESTRUCT),
                              pointIs2d
                            );
Sketch.SketchGenOperandTable.add("->", ["polygon", "point"],
                              new Sketch.OpCheckValue("polygon", MVM.opCodes.TRANSLATESTRUCT),
                              pointIs2d
                            );

//-------//
// clear //
//-------//
Sketch.SketchGenOperandTable.add("clear", ["point"],
                              new Sketch.OpCheckValue(null, MVM.opCodes.CLEAR),
                              colourSizeCheck
                            );