/* commands.c: main commands */
/*-
 * Copyright (c) 1997, 1998, 2003
 *	LEMIS (SA) Pty Ltd.
 *
 *  Written by Greg Lehey
 *
 *  This software is distributed under the so-called ``Berkeley
 *  License'':
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * This software is provided ``as is'', and any express or implied
 * warranties, including, but not limited to, the implied warranties of
 * merchantability and fitness for a particular purpose are disclaimed.
 * In no event shall the company or contributors be liable for any
 * direct, indirect, incidental, special, exemplary, or consequential
 * damages (including, but not limited to, procurement of substitute
 * goods or services; loss of use, data, or profits; or business
 * interruption) however caused and on any theory of liability, whether
 * in contract, strict liability, or tort (including negligence or
 * otherwise) arising in any way out of the use of this software, even if
 * advised of the possibility of such damage.
 *
 * $Id: commands.c,v 1.3 2004/12/12 03:08:24 grog Exp grog $
 */

#include "brew.h"

void sugar_usage (char *myname)
{
  fprintf (stderr,
	   "Usage:\n"
	   "%s [-s <sucrose>] [-d <dme>] [-m <malt>]  [-e <extract>] [-F FG] [-P Plato] [-S OG] litres\n\n"
	   "\t<sucrose>, <dme>, <malt> and <extract are weights in kilograms\n"
	   "\tAt least one must be specified\n",
	   myname );
  }

void sugar_command (int argc, char *argv [], char *arg0 [])
{
  int i;
  float dme_fraction = 0.94;
  float extract_fraction = 0.77;
  float malt_fraction = 0.75;
  float attenuation = 0;				    /* set below if no FG */

  float sucrose;
  float dme;
  float extract;
  float malt;
  float total_sugar;

  float litres = 0;
  float Plato;
  float OG = 0;
  float FG = 0;

  float desired_OG = 0;
  float desired_Plato = 0;

  for (i = 1; i < argc; i++)
    {
    if (argv [i] [0] == '-')
      {
      if (strncmp (argv [i], "-a", 2) == 0)		    /* attenuation */
	{
	if (FG)
	  {
	  fprintf (stderr, "Specify FG or attenuation, not both\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	if (argv [i] [2])				    /* speficied percentage */
	  attenuation = percentage (atof (&argv [i] [2]));
	else
	  {
	  i++;
	  attenuation = percentage (atof (argv [i]));
	  }
	}
      else if (strncmp (argv [i], "-d", 2) == 0)	    /* DME weight */
	{
	if (argv [i] [2])				    /* speficied percentage */
	  dme_fraction = percentage (atof (&argv [i] [2]));
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for DME\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	else
	  {
	  double mydme = kgs (atof (argv [i + 1]));

	  dme += mydme;
	  total_sugar += dme_fraction * mydme;
	  if (verbose)
	    printf ("Dried malt extract\t%6.2f\t\t%6.3f\t%6.3f\t\t%6.3f\n",
		    dme_fraction * 100,
		    mydme,
		    dme_fraction * mydme,
		    total_sugar);
	  i++;
	  }
	}
      else if (strncmp (argv [i], "-e", 2) == 0)	    /* extract weight */
	{
	if (argv [i] [2])				    /* speficied percentage */
	  extract_fraction = percentage (atof (&argv [i] [2]));
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for extract\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	else
	  {
	  double myextract = kgs (atof (argv [i + 1]));

	  extract += myextract;
	  total_sugar += extract_fraction * myextract;
	  if (verbose)
	    printf ("Malt extract syrup\t%6.2f\t\t%6.3f\t%6.3f\t\t%6.3f\n",
		    extract_fraction * 100,
		    myextract,
		    extract_fraction * myextract,
		    total_sugar);
	  i++;
	  }
	}
      else if (strncmp (argv [i], "-F", 2) == 0)	    /* desired FG */
	{
	if (attenuation)
	  {
	  fprintf (stderr, "Specify FG or attenuation, not both\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for desired FG\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	else
	  {
	  FG = densityval (atof (argv [i + 1]));
	  i++;
	  }
	}
      else if (strncmp (argv [i], "-m", 2) == 0)	    /* Malt weight */
	{
	if (argv [i] [2])				    /* speficied percentage */
	  malt_fraction = percentage (atof (&argv [i] [2]));
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for malt\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	else
	  {
	  double mymalt = kgs (atof (argv [i + 1]));

	  malt += mymalt;
	  total_sugar += malt_fraction * mymalt;
	  if (verbose)
	    printf ("Grain malt\t\t%6.2f\t\t%6.3f\t%6.3f\t\t%6.3f\n",
		    malt_fraction * 100,
		    mymalt,
		    malt_fraction * mymalt,
		    total_sugar);
	  i++;
	  }
	}
      else if (strncmp (argv [i], "-s", 2) == 0)	    /* Sucrose weight */
	{
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for sucrose\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	else
	  {
	  double mysucrose = kgs (atof (argv [i + 1]));

	  sucrose += mysucrose;
	  total_sugar += mysucrose;
	  if (verbose)
	    printf ("Pure saccharides\t%6.2f\t\t%6.3f\t%6.3f\t\t%6.3f\n",
		    100.0,
		    mysucrose,
		    mysucrose,
		    total_sugar);
	  i++;
	  }
	}
      else if (strncmp (argv [i], "-S", 2) == 0)	    /* desired OG */
	{
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for desired OG\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	else
	  {
	  desired_OG = densityval (atof (argv [i + 1]));
	  i++;
	  }
	}
      else if (strncmp (argv [i], "-P", 2) == 0)	    /* desired Plato */
	{
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for desired Plato\n\n");
	  sugar_usage (argv [0]);
	  return;
	  }
	else
	  {
	  /*
	   * Kludge: Plato is a percentage, i.e a fraction, but all
	   * the functions look at it as a value from 0 to 100.
	   */
	  desired_Plato = percentage (atof (argv [i + 1])) * 100;
	  i++;
	  }
	}
      else if (strcmp (argv [i], "-v") == 0)		    /* verbose */
	{
	verbose++;
	printf ("Sugar\t\t\tconcen-\t\tweight\tequiv\t\ttotal\n"
		"\t\t\ttration %%\t(kg)\tsucrose (kg)\tsugar (kg)\n\n");
	}
      else
	{
	fprintf (stderr,
		 "Invalid option: %s, aborting\n",
		 argv [i] );
	sugar_usage (argv [0]);
	return;
	}
      }
    else if (litres == 0.0)
      litres = atof (argv [i]);
    else						    /* too much */
      {
      fprintf (stderr, "Extraneous parameter:%s\n", argv [i]);
      sugar_usage (argv [0]);
      return;
      }
    }

  if (litres == 0.0)
    {
    fprintf (stderr, "No volume gives no beer\n");
    sugar_usage (argv [0]);
    return;
    }
  if (desired_OG != 0 && desired_Plato !=0)
    {
    fprintf (stderr,
	     "Make up your mind.  Goal OG %g or goal Plato %g?\n",
	     desired_OG,
	     desired_Plato );
    sugar_usage (argv [0]);
    return;
    }

  total_sugar = sucrose + dme * dme_fraction + extract * extract_fraction + malt * malt_fraction;
  Plato = total_sugar / litres * 100;
  OG = Plato_to_SG (Plato);
  if (FG == 0)
    {
    if (attenuation == 0)
      attenuation = 0.74;
    FG = Plato_to_SG (Plato * (1 - attenuation));
    }

  if (verbose)
    printf ("\nTotals:\n\n"
	    "Brew volume:\t\t%6.1f   litres\n"
	    "Pure saccharides:\t%8.3f kg\n"
	    "Dried malt extract:\t%8.3f kg\n"
	    "Malt extract syrup:\t%8.3f kg\n"
	    "Grain malt:\t\t%8.3f kg\n"
	    "Total sugar:\t\t%8.3f kg\n\n"
	    "Plato:\t\t\t%7.2f %%\n"
	    "OG:\t\t\t%9.4f\n"
	    "FG:\t\t\t%9.4f\n"
	    "Alcohol by weight:\t%9.4f %%\n"
	    "Alcohol by volume:\t%9.4f %%\n\n",
	    litres,
	    sucrose,
	    dme,
	    extract,
	    malt,
	    total_sugar,
	    Plato,
	    OG,
	    FG,
	    alcohol_by_weight_fix (OG, FG),
	    abw_to_abv (alcohol_by_weight_fix (OG, FG)) );
  else
    printf ("Total sugar\tsucrose\tDME\tExtract\tMalt\tVolume (l)\tPlato\tOG\tFG\tAlc %%\n"
	    "%-4.2f\t\t%-4.2f\t%-4.2f\t%-4.2f\t%-4.2f\t%-4.2f\t\t%-4.2f\t%-6.4f\t%-6.4f\t%2.1f\n",
	    total_sugar,
	    sucrose,
	    dme * dme_fraction,
	    extract * extract_fraction,
	    malt * malt_fraction,
	    litres,
	    Plato,
	    OG,
	    FG,
	    abw_to_abv (alcohol_by_weight_fix (OG, FG)) );
  if (desired_OG != 0)
    desired_Plato = SG_to_Plato (desired_OG);
  else if (desired_Plato != 0)
    desired_OG = Plato_to_SG (desired_Plato);
  if (desired_Plato != 0)
    {
    float sugar_needed = (desired_Plato - Plato) * litres / 100;
    float water_adjustment = fabs (total_sugar * 100 / desired_Plato - litres);

    if (sugar_needed != 0)
      printf ("\n"
	      "%s %2.3f kg sugar (%2.3f kg DME, %2.3f kg extract or %2.3f kg malt)\n"
	      "or %s %2.3f litres of water to achieve %2.2f Plato (OG %6.4f)\n",
	      sugar_needed > 0? "Add": "Remove",
	      fabs (sugar_needed),
	      fabs (sugar_needed / dme_fraction),
	      fabs (sugar_needed / extract_fraction),
	      fabs (sugar_needed / malt_fraction),
	      sugar_needed > 0? "remove": "add",
	      water_adjustment,
	      desired_Plato,
	      desired_OG);
    }
  }

void hops_usage (char *myname)
{
  fprintf (stderr,
	   "Usage:\n"
	   "%s [-a <alpha acid %%>] [-l <total volume>] [-S <SG>] [-t <boil time>] <hop-weight>\n",
	   myname );
  }


void hops_command (int argc, char *argv [], char *arg0 [])
{
  int i;
  float aaprop = 0.05;					    /* default alpha acid */
  float total_aa = 0;					    /* and total so far */
  float boil_time = 60;					    /* time this component is boiled */
  float total_volume = 0;				    /* total volume of boil */
  float SG = 1.060;					    /* default boil gravity */
  enum
    {
    rager, tinseth
    }
  util_type = tinseth;

  printf ("Hop weight\tAA %%\tBoil time\tUtil\tAA\tTotal AA\n\n");
  if (argc == 1)
  {
    hops_usage (argv [0]);
    return;
  }
  for (i = 1; i < argc; i++)
    {
    if (argv [i] [0] == '-')
      {
      if (strncmp (argv [i], "-a", 2) == 0)		    /* alpha acid */
	{
	if (argv [i] [2])				    /* specified percentage */
	  aaprop = percentage (atof (&argv [i] [2]));
	else  if (argc > (i + 1))
	  {
	  i++;
	  aaprop = percentage (atof (argv [i]));
	  }
	else
	  {
	  fprintf (stderr, "No alpha acid component specified\n\n");
	  hops_usage (argv [0]);
	  return;
	  }
	}
      else if (strncmp (argv [i], "-l", 2) == 0)	    /* litres */
	{
	if (total_volume != 0)				    /* already specified */
	  fprintf (stderr,
		   "Warning: volume redefined, was %4.2f\n",
		   total_volume );
	if (argv [i] [2])				    /* specified percentage */
	  total_volume = atof (&argv [i] [2]);
	else  if (argc > (i + 1))
	  {
	  i++;
	  total_volume = atof (argv [i]);
	  }
	else
	  {
	  fprintf (stderr, "No volume specified\n\n");
	  hops_usage (argv [0]);
	  return;
	  }
	}
      else if (strncmp (argv [i], "-R", 2) == 0)	    /* Rager equations */
	util_type = rager;
      else if (strncmp (argv [i], "-S", 2) == 0)	    /* desired OG */
	{
	if (argv [i] [2])				    /* specified percentage */
	  SG = densityval (atof (&argv [i] [2]));
	else  if (argc > (i + 1))
	  {
	  i++;
	  SG = densityval (atof (argv [i]));
	  }
	else
	  {
	  fprintf (stderr, "No SG specified\n\n");
	  hops_usage (argv [0]);
	  return;
	  }
	}
      else if (strncmp (argv [i], "-T", 2) == 0)	    /* Tinseth equations */
	util_type = tinseth;
      else if (strncmp (argv [i], "-t", 2) == 0)	    /* boil time */
	{
	if (argv [i] [2])				    /* specified percentage */
	  boil_time = atof (&argv [i] [2]);
	else  if (argc > (i + 1))
	  {
	  i++;
	  boil_time = atof (argv [i]);
	  }
	else
	  {
	  fprintf (stderr, "No volume specified\n\n");
	  hops_usage (argv [0]);
	  return;
	  }
	}
      else
	{
	fprintf (stderr, "Unknown option: %s\n\n", argv [i]);
	hops_usage (argv [0]);
	return;
	}
      }
    else						    /* hop weight */
      {
      float util;
      float hop_weight;
      float hop_extract;

      /* Decipher rager and do it */

      if (util_type = tinseth)
	util = tinseth_hop_utilization (SG, boil_time);
      else
	util = 0.2;					    /* XXX */
      hop_weight = atof (argv [i]);
      hop_extract = hop_weight * util * aaprop;
      total_aa += hop_extract;
      printf ("%6.2f\t\t%6.2f\t%6.0f\t\t%4.1f\t%6.3f\t%6.3f\n",
	      hop_weight,
	      aaprop * 100,
	      boil_time,
	      util * 100,
	      hop_extract,
	      total_aa );
      }
    }

  /* Finished: show results */

  if (total_volume)					    /* got a volume */
    printf ("\n"
	    "Boil density:\t\t%7.3f\n"
	    "Total alpha acid:\t%7.3f g\n"
	    "Volume:\t\t\t%7.3f l\n"
	    "IBUs:\t\t\t%3.0f\n",
	    SG,
	    total_aa,
	    total_volume,
	    total_aa / total_volume * 1000 );
  }


void brix_command (int argc, char *argv [], char *arg0 [])
{
  }

void brix2fg_command (int argc, char *argv [], char *arg0 [])
{
  float obrix;
  float fbrix;

  if (argc < 2)
    {
    fprintf (stderr, "Usage: %s original-brix final-brix\n", argv [0]);
    return;
    }

  obrix = atof (argv [1]);
  fbrix = atof (argv [2]);
  printf ("\t\tBrix\tSG\tAttenuation\n"
	  "Original\t%6.2f\t%6.3f\n"
	  "Final\t\t%6.2f\t%6.3f\t%6.0f%%\n",
	  obrix,
	  brix_to_sg (obrix),
	  fbrix,
	  brix_to_fg (obrix, fbrix),
	  100 - (brix_to_fg (obrix, fbrix) - 1) / (brix_to_sg (obrix) - 1) * 100);
  }

void decoct_command (int argc, char *argv [], char *arg0 [])
{
  float totalweight;
  float mashtemp;
  float goaltemp;

  if (argc < 3)
    {
    fprintf (stderr, "Usage: %s totalweight mashtemp goaltemp\n", argv [0]);
    return;
    }

  totalweight = atof (argv [1]);
  mashtemp = atof (argv [2]);
  goaltemp = atof (argv [3]);
  printf ("Mash weight:\t\t%6.3f kg\n"
	  "Mash temperature:\t%6.0f°\n"
	  "Goal temperature:\t%6.0f°\n"
	  "Decoction weight:\t%6.3f kg\n",
	  totalweight,
	  mashtemp,
	  goaltemp,
	  totalweight * (goaltemp - mashtemp) / (100 - mashtemp) );
}

void debug_command (int argc, char *argv [], char *arg0 [])
{
  }