/* $Id: sugar.c,v 1.2 2003/11/17 00:48:43 grog Exp $ */
#include <ctype.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/file.h>
#include <sys/param.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <unistd.h>
#include "beer.h"

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

float dme_fraction = 0.94;
float extract_fraction = 0.68;
float malt_fraction = 0.59;

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

float litres;
float Plato;
float SG;

float desired_SG;
float desired_Plato;

int verbose;

int main (int argc, char *argv [])
{
  int i;

  verbose = 0;						    /* until proven otherwise */
  for (i = 1; i < argc; i++)
    {
    if (argv [i] [0] == '-')
      {
      if (memcmp (argv [i], "-d", 2) == 0)		    /* DME weight */
	{
	if (argv [i] [2])				    /* speficied percentage */
	  dme_fraction = atof (&argv [i] [2]) / 100;
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for DME\n\n");
	  usage (argv [0]);
	  return 1;
	  }
	else
	  {
	  double mydme = 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 (memcmp (argv [i], "-e", 2) == 0)		    /* extract weight */
	{
	if (argv [i] [2])				    /* speficied percentage */
	  extract_fraction = atof (&argv [i] [2]) / 100;
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for extract\n\n");
	  usage (argv [0]);
	  return 1;
	  }
	else
	  {
	  double myextract = 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 (memcmp (argv [i], "-m", 2) == 0)		    /* Malt weight */
	{
	if (argv [i] [2])				    /* speficied percentage */
	  malt_fraction = atof (&argv [i] [2]) / 100;
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for malt\n\n");
	  usage (argv [0]);
	  return 1;
	  }
	else
	  {
	  double mydme = 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 (memcmp (argv [i], "-s", 2) == 0)		    /* Sucrose weight */
	{
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for sucrose\n\n");
	  usage (argv [0]);
	  return 1;
	  }
	else
	  {
	  double mysucrose = 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 (memcmp (argv [i], "-S", 2) == 0)		    /* desired SG */
	{
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for desired SG\n\n");
	  usage (argv [0]);
	  return 1;
	  }
	else
	  {
	  desired_SG = atof (argv [i + 1]);
	  i++;
	  }
	}
      else if (memcmp (argv [i], "-P", 2) == 0)		    /* desired Plato */
	{
	if (argc < (i + 2))
	  {
	  fprintf (stderr, "No weight specified for desired Plato\n\n");
	  usage (argv [0]);
	  return 1;
	  }
	else
	  {
	  desired_Plato = atof (argv [i + 1]);
	  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] );
	usage (argv [0]);
	return 1;
	}
      }
    else if (litres == 0.0)
      litres = atof (argv [i]);
    else						    /* too much */
      {
      fprintf (stderr, "Extraneous parameter:%s\n", argv [i]);
      usage (argv [0]);
      return 1;
      }
    }

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

  total_sugar = sucrose + dme * dme_fraction + extract * extract_fraction + malt * malt_fraction;
  Plato = total_sugar / litres * 100;

  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"
	    "SG:\t\t\t%9.4f\n\n",
	    litres,
	    sucrose,
	    dme,
	    extract,
	    malt,
	    total_sugar,
	    Plato,
	    Plato_to_SG (Plato) );
  else
    printf ("Total sugar\tsucrose\tDME\tExtract\tMalt\tVolume (l)\tPlato\tSG\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%2.1f\n",
	    total_sugar,
	    sucrose,
	    dme * dme_fraction,
	    extract * extract_fraction,
	    malt * malt_fraction,
	    litres,
	    Plato,
	    Plato_to_SG (Plato),
	    0.0 );
  if (desired_SG != 0)
    desired_Plato = SG_to_Plato (desired_SG);
  else if (desired_Plato != 0)
    desired_SG = 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 (SG %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_SG);
    }
  return 0;
  }