/* Siconos is a program dedicated to modeling, simulation and control
 * of non smooth dynamical systems.
 *
 * Copyright 2018 INRIA.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
*/
#include <math.h>
#include "projectionOnCone.h"

unsigned projectionOnCone(double* r, double  mu)
{
  double normT = hypot(r[1], r[2]);
  if (mu * normT <= - r[0])
  {
    r[0] = 0.0;
    r[1] = 0.0;
    r[2] = 0.0;
    return PROJCONE_DUAL;
  }
  else if (normT <= mu * r[0])
  {
    return PROJCONE_INSIDE;
  }
  else
  {
    double mu2 = mu * mu;
    r[0] = (mu * normT + r[0]) / (mu2 + 1.0);
    r[1] = mu * r[0] * r[1] / normT;
    r[2] = mu * r[0] * r[2] / normT;
    return PROJCONE_BOUNDARY;
  }
}
unsigned projectionOnDualCone(double* u, double  mu)
{
  double normT = hypot(u[1], u[2]);
  
  if (normT <= - mu * u[0])
  {
    u[0] = 0.0;
    u[1] = 0.0;
    u[2] = 0.0;
    return PROJCONE_DUAL;
  }
  else if (mu * normT <= u[0])
  {
    return PROJCONE_INSIDE;
  }
  else
  {
    double mu2 = mu * mu;
    u[0] = (normT + mu * u[0]) / (mu2 + 1.0);
    u[1] = u[0] * u[1] / normT;
    u[2] = u[0] * u[2] / normT;
    u[0] = mu * u[0];
  
    return PROJCONE_BOUNDARY;
  }
}


void projectionOnSecondOrderCone(double* r, double  mu, int size)
{
  if (size ==3)
  {
    projectionOnCone(r, mu);
  }
  else
  {
    fprintf(stderr, "Numerics, projectionOnSecondOrderCone f not yet implementes for size != 3 \n");
    exit(EXIT_FAILURE);
  }

}
