using HeuristicLab.Common;
using HeuristicLab.Core;
using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding;
using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
using HeuristicLab.Problems.DataAnalysis.Symbolic;

namespace HeuristicLab.Problems.GeneticProgramming.GlucosePrediction {
  [StorableClass]
  [Item("Blood glucose level forecast grammar", "See MedGEC Workshop at GECCO 2016")]
  public class Grammar : SymbolicExpressionGrammar {
    [StorableConstructor]
    protected Grammar(bool deserializing) : base(deserializing) { }
    protected Grammar(Grammar original, Cloner cloner) : base(original, cloner) { }

    public Grammar()
      : base("Blood glucose level forecast grammar", "See MedGEC Workshop at GECCO 2016") {
      Initialize();
    }

    public void Initialize() {
      // var func = new SimpleSymbol("Func", "The root for the blood glucose forecasting model.", 3, 3);
      var exprGluc = new SimpleSymbol("ExprGluc", string.Empty, 1, 1);
      // var exprIns = new SimpleSymbol("ExprIns", string.Empty, 1, 1);
      // var exprCh = new SimpleSymbol("ExprCh", string.Empty, 1, 1);

      // var predGlucData = new PredictedGlucoseVariableSymbol();
      var realGlucData = new RealGlucoseVariableSymbol();

      // operators for exprGluc
      var plusGluc = new SimpleSymbol("+", "+", 2, 2);
      var minusGluc = new SimpleSymbol("-", "-", 2, 2);
      var prodGluc = new SimpleSymbol("*", "*", 2, 2);
      var divGluc = new SimpleSymbol("/", "/", 2, 2);
      var expGluc = new SimpleSymbol("Exp", "Exp", 1, 1);
      var sinGluc = new SimpleSymbol("Sin", "Sin", 1, 1);
      var cosGluc = new SimpleSymbol("Cos", "Cos", 1, 1);
      var logGluc = new SimpleSymbol("Log", "Log", 1, 1);
      // var cteGluc = new CteSymbol();

      // operators for exprCh
      // var plusCh = new SimpleSymbol("+Ch", "+", 2, 2);
      // var minusCh = new SimpleSymbol("-Ch", "-", 2, 2);
      // var prodCh = new SimpleSymbol("*Ch", "*", 2, 2);
      // var divCh = new SimpleSymbol("/Ch", "/", 2, 2);
      // var expCh = new SimpleSymbol("ExpCh", "Exp", 1, 1);
      // var sinCh = new SimpleSymbol("SinCh", "Sin", 1, 1);
      // var cosCh = new SimpleSymbol("CosCh", "Cos", 1, 1);
      // var logCh = new SimpleSymbol("LogCh", "Log", 1, 1);
      var curvedCh = new CurvedChVariableSymbol("CurvedCh", "");
      // var cteCh = new CteSymbol();

      // operators for exprIns
      // var plusIns = new SimpleSymbol("+Ins", "+", 2, 2);
      // var minusIns = new SimpleSymbol("-Ins", "-", 2, 2);
      // var prodIns = new SimpleSymbol("*Ins", "*", 2, 2);
      // var divIns = new SimpleSymbol("/Ins", "/", 2, 2);
      // var expIns = new SimpleSymbol("ExpIns", "Exp", 1, 1);
      // var sinIns = new SimpleSymbol("SinIns", "Sin", 1, 1);
      // var cosIns = new SimpleSymbol("CosIns", "Cos", 1, 1);
      // var logIns = new SimpleSymbol("LogIns", "Log", 1, 1);
      var curvedIns = new CurvedInsVariableSymbol("CurvedIns", "");
      // var realInsVar = new RealInsulineVariableSymbol();

      var constSy = new Constant();
      // var cteCh = new CteSymbol();

      // AddSymbol(func);
      AddSymbol(exprGluc);
      // AddSymbol(exprIns);
      // AddSymbol(exprCh);
      // AddSymbol(predGlucData);
      AddSymbol(realGlucData);
      AddSymbol(plusGluc);
      AddSymbol(minusGluc);
      AddSymbol(prodGluc);
      AddSymbol(divGluc);
      AddSymbol(expGluc);
      AddSymbol(sinGluc);
      AddSymbol(cosGluc);
      AddSymbol(logGluc);

      AddSymbol(curvedCh);

      // AddSymbol(plusCh);
      // AddSymbol(minusCh);
      // AddSymbol(prodCh);
      // AddSymbol(divCh);
      // AddSymbol(expCh);
      // AddSymbol(sinCh);
      // AddSymbol(cosCh);
      // AddSymbol(logCh);

      AddSymbol(curvedIns);
      // AddSymbol(plusIns);
      // AddSymbol(minusIns);
      // AddSymbol(prodIns);
      // AddSymbol(divIns);
      // AddSymbol(expIns);
      // AddSymbol(sinIns);
      // AddSymbol(cosIns);
      // AddSymbol(logIns);
      // AddSymbol(realInsVar);
      AddSymbol(constSy);

      // <func> ::= <exprgluc> + <exprch> - <exprins>
      // AddAllowedChildSymbol(func, exprGluc, 0);
      // AddAllowedChildSymbol(func, exprCh, 1);
      // AddAllowedChildSymbol(func, exprIns, 2);

      /* # Glucose
      <exprgluc> ::= (<exprgluc> <op> <exprgluc>)
                    |<preop> (<exprgluc>)
                    |(<cte> <op> <exprgluc>)
                    |predictedData(k-<idxCurr2h>)
                    |realData(k-<idx2hOrMore>)
      */
      var opGlucSet = new Symbol[] { plusGluc, minusGluc, prodGluc, divGluc, expGluc, sinGluc, cosGluc, logGluc, /*predGlucData, */ realGlucData, constSy, curvedCh, curvedIns };
      foreach (var opGluc in opGlucSet) {
        AddAllowedChildSymbol(exprGluc, opGluc);
      }


      foreach (var parentOp in new Symbol[] { plusGluc, minusGluc, prodGluc, divGluc }) {
        foreach (var childOp in opGlucSet) {
          AddAllowedChildSymbol(parentOp, childOp, 0);
          AddAllowedChildSymbol(parentOp, childOp, 1);
        }
      }
      // <exprgluc> ::= <preop> (<exprgluc>)
      foreach (var parentOp in new Symbol[] { expGluc, sinGluc, cosGluc, logGluc }) {
        foreach (var childOp in opGlucSet) {
          AddAllowedChildSymbol(parentOp, childOp, 0);
        }
      }

      /*
      # CH
      <exprch> ::= (<exprch> <op> <exprch>)
                    |<preop> (<exprch>)
                    |(<cte> <op> <exprch>)	
                    |(getPrevData(1,k,1) * <cte> * <curvedCH>)

      # CH in 2 (0.041,24), 3 (0.027,36) and 4 (0.02,48) hours with 3 shapes of beta:
      <curvedCH> ::= beta(0.041*Math.min(24,getPrevDataDistance(1,k,1)),2,5)
              |beta(0.041*Math.min(24,getPrevDataDistance(1,k,1)),3,3)
              |beta(0.041*Math.min(24,getPrevDataDistance(1,k,1)),5,2)
              |beta(0.027*Math.min(36,getPrevDataDistance(1,k,1)),2,5)
              |beta(0.027*Math.min(36,getPrevDataDistance(1,k,1)),3,3)
              |beta(0.027*Math.min(36,getPrevDataDistance(1,k,1)),5,2)
              |beta(0.02*Math.min(48,getPrevDataDistance(1,k,1)),2,5)
              |beta(0.02*Math.min(48,getPrevDataDistance(1,k,1)),3,3)
              |beta(0.02*Math.min(48,getPrevDataDistance(1,k,1)),5,2) 
      */
      // var opChSet = new Symbol[] { plusCh, minusCh, prodCh, divCh, expCh, sinCh, cosCh, logCh, curvedCh, constSy };
      // foreach (var opCh in opChSet) {
      //   AddAllowedChildSymbol(exprCh, opCh);
      // }
      // 
      // foreach (var parentOp in new Symbol[] { plusCh, minusCh, prodCh, divCh }) {
      //   foreach (var childOp in opChSet) {
      //     AddAllowedChildSymbol(parentOp, childOp, 0);
      //     AddAllowedChildSymbol(parentOp, childOp, 1);
      //   }
      // }
      // // <exprch> ::= <preop> (<exprch>)
      // foreach (var parentOp in new Symbol[] { expCh, sinCh, cosCh, logCh }) {
      //   foreach (var childOp in opChSet) {
      //     AddAllowedChildSymbol(parentOp, childOp, 0);
      //   }
      // }

      /*
       # Insulin:
      ## Sum of insulins in past 2h minus the peak
      ## Curve for the peak in past 2h
      <exprins> ::= (<exprins> <op> <exprins>)
        |<preop> (<exprins>)
        |(<cte> <op> <exprins>)		
        |getVariable(2,k-<idx>)
        |(getSumOfValues(24,k,2) - getMaxValue(24,k,2) <op> (getMaxValue(24,k,2) <op> <curvedINS>))

      # INS in 2 (0.041,24), 3 (0.027,36) and 4 (0.02,48) hours with 3 shapes of beta:
      <curvedINS> ::= beta(0.041*Math.min(24,getMaxValueDistance(24,k,2)),2,5)
              |beta(0.041*Math.min(24,getMaxValueDistance(24,k,2)),3,3)
              |beta(0.041*Math.min(24,getMaxValueDistance(24,k,2)),5,2)
              |beta(0.027*Math.min(36,getMaxValueDistance(36,k,2)),2,5)
              |beta(0.027*Math.min(36,getMaxValueDistance(36,k,2)),3,3)
              |beta(0.027*Math.min(36,getMaxValueDistance(36,k,2)),5,2)
              |beta(0.02*Math.min(48,getMaxValueDistance(48,k,2)),2,5)
              |beta(0.02*Math.min(48,getMaxValueDistance(48,k,2)),3,3)
              |beta(0.02*Math.min(48,getMaxValueDistance(48,k,2)),5,2)

      */
      // var opInsSet = new Symbol[] { plusIns, minusIns, prodIns, divIns, expIns, sinIns, cosIns, logIns, /* realInsVar, */ curvedIns, constSy };
      // foreach (var opIns in opInsSet) {
      //   AddAllowedChildSymbol(exprIns, opIns);
      // }
      // 
      // // <exprins> ::= (<exprins> <op> <exprins>)
      // // <exprins> ::= (<cte> <op> <exprins>)		
      // foreach (var parentOp in new Symbol[] { plusIns, minusIns, prodIns, divIns }) {
      //   foreach (var childOp in opInsSet) {
      //     AddAllowedChildSymbol(parentOp, childOp, 0);
      //     AddAllowedChildSymbol(parentOp, childOp, 1);
      //   }
      // }
      // // <exprins> ::= <preop> (<exprins>)
      // foreach (var op in new Symbol[] { expIns, sinIns, cosIns, logIns }) {
      //   foreach (var childOp in opInsSet) {
      //     AddAllowedChildSymbol(op, childOp, 0);
      //   }
      // }

      // root is exprGluc
      AddAllowedChildSymbol(StartSymbol, exprGluc);
    }

    public override IDeepCloneable Clone(Cloner cloner) {
      return new Grammar(this, cloner);
    }
  }
}