#! /bin/bash
# Last edited on 2014-03-03 19:27:40 by stolfilocal

# 05 pixels = 01:00

date="2014-03-03"

function pscale(){
  # Prints to {stdout} the V-coordinate corresponing to a given price.
  
  # Plot scaling parameters:
  pmin="$1"; shift # Low reference price.
  vbot="$1"; shift # V-coordinate of the low reference price.
  pmax="$1"; shift # High reference price.
  vtop="$1"; shift # V-coordinate of the high reference price.
  
  parg="$1"; shift # Given price.
  
  echo "enter pscale ${pmin} ${vbot} ${pmax} ${vtop} ${parg}" 1>&2 

  pdel=$(( ${pmax} - ${pmin} ));
  prnd=$(( ${pdel} / 2 ))
  
  varg=$(( ${vbot} - ( ( ( ${parg} - ${pmin} ) * ( ${vbot} - ${vtop} ) + ${prnd} ) / ${pdel} ) ))
  
  printf "%d\n" "${varg}"
  echo "exit pscale ${varg}" 1>&2 
}

function make_rect_cmds(){
  # Prints to {stdout} the draw commands for a prediction box with label.
  
  # Plot scaling parameters:
  pmin="$1"; shift # Low reference price.
  vbot="$1"; shift # V-coordinate of the low reference price.
  pmax="$1"; shift # High reference price.
  vtop="$1"; shift # V-coordinate of the high reference price.
  
  krect="$1"; shift # "@" for old rectangle, "+" for new.
  hrect="$1"; shift # H-coordinate of rectangle center (already shifted by ${hshift}).
  prect="$1"; shift # Predicted price at ${trect}.
  herr="$1"; shift  # Half-width of rectangle (pixels)
  verr="$1"; shift  # Half-height of rectngle (pixels).

  echo "enter make_rect_cmds ${pmin} ${vbot} ${pmax} ${vtop} ${krect} ${hrect} ${prect} ${herr} ${verr}" 1>&2 
  
  vrect=`pscale ${pmin} ${vbot} ${pmax} ${vtop} ${prect}`
  vrect_top=$(( ${vrect} + ${verr} ))
  vrect_bot=$(( ${vrect} - ${verr} ))
  hrect_lft=$(( ${hrect} - ${herr} ))
  hrect_rht=$(( ${hrect} + ${herr} ))

  if [[ "/${krect}" != "/@" ]]; then crect="#0088ff" ; else crect="#aabbcc" ; fi
  
  printf "%s %s\n" "-color" "${crect}"
  printf "%s %s\n" "-strokewidth" "3"
  printf "%s %s %s  %s %s  %s %s  %s %s\n" "-Q" \
    "${hrect_rht}" "${vrect_top}" \
    "${hrect_rht}" "${vrect_bot}" \
    "${hrect_lft}" "${vrect_bot}" \
    "${hrect_lft}" "${vrect_top}" 
  echo "exit make_rect_cmds" 1>&2 
}     

function annot(){
  # Annotate one image.
  
  ex="$1"; shift       # Exchange code
  which="$1"; shift    # "check" or "make"

  # Input image:
  iimg="${date}-prediction-check-${ex}"
  
  # Output image:
  oimg="${date}-prediction-${which}-${ex}"

  # Vertical scale information
  pmin="$1"; shift # Low reference price.
  vbot="$1"; shift # V-coordinate of the low reference price.
  pmax="$1"; shift # High reference price.
  vtop="$1"; shift # V-coordinate of the high reference price.
  
  hshift=0;  # Value to add or subtract from all h-coordinates.
  if [[ "/$1" == "/-hshift" ]]; then
    shift;
    hshift="$1"; shift
  fi
  
  # Balloon with time posted ("-tpost") or time valid ("-tpred")
  labt_cmds=( )
  while [[ ( "/$1" == "/-tpost" ) || ( "/$1" == "/-tpred" ) ]]; do
    if [[ "/$1" == "/-tpost" ]]; then clabt="#ff44aa" ; else clabt="#33ddff" ; fi
    shift;
  
    # Label with time of last prediction:
    tlabt="$1"; shift # UTC time to show in balloon.
    hlabt="$1"; shift # H-coordinate of balloon center (unshifted).
    vlabt_dot="$1"; shift # V-coordinate of balloon dot.

    hlabt=$(( ${hlabt} + ${hshift} ))

    vlabt_lab=$(( ${vlabt_dot} + 80 ))
    labt_cmds+=( -color "${clabt}" -strokewidth 2 )
    labt_cmds+=( -L ${hlabt} ${vlabt_dot}  ${hlabt} ${vlabt_lab}  "${tlabt}" )
  done
  echo "labt_cmds=( " "${labt_cmds[@]}" " )" 1>&2
  
  # Prediction boxes:
  rect_cmds=( ) # Commands to draw prediction boxes.
  while [[ "/$1" == "/-rect" ]]; do 
    shift;
    krect="$1"; shift # "@" for old, "+" for new.
    hrect="$1"; shift # H-coordinate of rectangle center (unshifted).
    prect="$1"; shift # Price of rectangle center.
    herr="$1"; shift  # Half-width of box (pixels).
    verr="$1"; shift  # Half-height of box (pixels).
    
    hrect=$(( ${hrect} + ${hshift} ))

    rect_cmds+=( `make_rect_cmds ${pmin} ${vbot} ${pmax} ${vtop} "${krect}" ${hrect} ${prect} ${herr} ${verr}` )
  done
  echo "rect_cmds=( " "${rect_cmds[@]}" " )" 1>&2
  
  # Trend lines:
  line_cmds=( )
  while [[ "/$1" == "/-line" ]]; do 
    shift;
    kline="$1"; shift # "@" for bad, "+" for new.
    hline1="$1"; shift # H-coordinate of line start (unshifted).
    pline1="$1"; shift # Price of line start.
    hline2="$1"; shift # H-coordinate of lineend (unshifted).
    pline2="$1"; shift # Price of line end.
    
    if [[ "/${kline}" == "/@" ]]; then cline="#88664466" ; else cline="#ff550099" ; fi
    
    hline1=$(( ${hline1} + ${hshift} ))
    vline1=`pscale ${pmin} ${vbot} ${pmax} ${vtop} ${pline1}`
      
    hline2=$(( ${hline2} + ${hshift} ))
    vline2=`pscale ${pmin} ${vbot} ${pmax} ${vtop} ${pline2}`

    line_cmds+=( -strokewidth 3 -color "${cline}" )
    line_cmds+=( -S ${hline1} ${vline1} ${hline2} ${vline2} ) 
  done
  echo "line_cmds=( " "${line_cmds[@]}" " )" 1>&2

  # Dots for Slumber Points:
  dot_cmds=( ) # Commands to draw dots at the slumber points and slumber times.
  if [[ "/$1" == "/-dots" ]]; then 
    shift;

    vvolb="$1"; shift  # V-coordinate of volume plot baseline.
    vvolb_dot=${vvolb} # V-coordinate of Slumber Times dots.
    
    dots=( "$@" ) # H- an V-cordinates of "slumber points".

    dot_cmds+=( -strokewidth 20 )
    i=0
    while [[ ${i} -lt ${#dots[@]} ]]; do 
      kdot="${dots[${i}]}"; # "@" for bad, "+" for good.
      if [[ "/${kdot}" == "/@" ]]; then cdot="#88664466" ; else cdot="#ff550099" ; fi
      i=$(( ${i} + 1 ))
      
      hdot="${dots[${i}]}"; # H-coordinate of dot (unshifted).
      hdot=$(( ${hdot} + ${hshift} ))
      i=$(( ${i} + 1 ))
      
      pdot="${dots[${i}]}"; # Price at dot.
      vdot=`pscale ${pmin} ${vbot} ${pmax} ${vtop} ${pdot}`
      i=$(( ${i} + 1 ))
  
      dot_cmds+=( -strokewidth 24 -color "${cdot}" )
      dot_cmds+=( -S ${hdot} ${vdot} ) 
      dot_cmds+=( -S ${hdot} ${vvolb_dot} )
    done
    
  fi
  echo "dot_cmds=( " "${dot_cmds[@]}" " )" 1>&2

  set echo
  
  annotate-pic.sh \
      ${iimg}.png 16 Black Yellow \
      "${labt_cmds[@]}" \
      \
      "${rect_cmds[@]}" \
      \
      "${dot_cmds[@]}" \
      \
      "${line_cmds[@]}" \
      \
    > ${oimg}-annotated.png
  gimp ${oimg}-annotated.png
}

# echo `pscale 3700 573  4800 111  4400` 

# Bitstamp slumber dots are computed from Huobi's by the ratios
#   1 USD = 6.40 CNY for 2014-02-07 -- 2014-02-09
#   1 USD = 6.12 CNY for other days.

check=1
make=1

scale_HUBI=( 3100  422  4100   84 )
scale_BSTP=(  400  448   700  101 )

if [[ ${make} -gt 0 ]]; then 

  annot HUBI make \
    ${scale_HUBI[@]} \
    -hshift -1014 \
    \
    -rect "+" 3128  3370 9 7 \
    \
    -line  "@"   -51  4873      8  4780 \
    -line  "@"     8  4780     68  4704 \
    -line  "@"    68  4704    128  4642 \
    -line  "@"   128  4642    188  4592 \
    -line  "@"   188  4592    248  4552 \
    -line  "@"   248  4552    308  4519 \
    -line  "@"   308  4519    368  4492 \
    -line  "@"   368  4492    428  4470 \
    \
    -line  "@"   428  4470    428  4147 \
    \
    -line  "@"   428  4147    488  4101 \
    -line  "@"   488  4101    548  4063 \
    -line  "@"   548  4063    608  4033 \
    -line  "@"   608  4033    668  4008 \
    -line  "@"   668  4008    728  3988 \
    -line  "@"   728  3988    788  3972 \
    -line  "@"   788  3972    848  3958 \
    -line  "@"   848  3958    908  3947 \
    -line  "@"   908  3947    968  3939 \
    -line  "@"   968  3939   1028  3932 \
    -line  "@"  1028  3932   1088  3926 \
    -line  "@"  1088  3926   1148  3921 \
    -line  "@"  1148  3921   1208  3917 \
    -line  "@"  1208  3917   1268  3914 \
    -line  "@"  1268  3914   1328  3912 \
    -line  "@"  1328  3912   1388  3910 \
    -line  "@"  1388  3910   1448  3908 \
    -line  "@"  1448  3908   1508  3907 \
    -line  "@"  1508  3907   1568  3905 \
    -line  "@"  1568  3905   1628  3905 \
    \
    -line  "+"  2348  3567   3188  3355 \
    \
    -dots 555 \
    "+" -2032 5014 \
    "@" -1912 5015 \
    "@" -1792 4966 \
    "+" -1672 4927 \
    "+" -1552 4730 \
    "+" -1432 4839 \
    "@" -1312 5041 \
    "@" -1192 4856 \
    "+" -1072 4829 \
    "@"  -952 4809 \
    "@"  -832 4851 \
    "+"  -712 4873 \
    "@"  -592 4939 \
    "@"  -472 4973 \
    "@"  -352 4898 \
    "@"  -232 4915 \
    "@"  -112 4892 \
    "+"     8 4782 \
    "+"   128 4637 \
    "+"   248 4555 \
    "+"   368 4492 \
    "+"   488 4080 \
    "@"   608 4139 \
    "+"   728 4006 \
    "@"   848 3876 \
    "+"   968 4008 \
    "+"  1088 3961 \
    "@"  1208 3733 \
    "+"  1328 3905 \
    "+"  1448 3817 \
    "@"  1568 3832 \
    "@"  1688 3530 \
    "+"  1808 3425 \
    "@"  1928 3621 \
    "+"  2048 3810 \
    "+"  2168 3615 \
    "+"  2288 3284 \
    "@"  2408 3543 \
    "+"  2528 3533 \
    "+"  2648 3464 \
    "+"  2768 3479 \
    "+"  2888 3427 \
    "@"  3008 3825

fi

if [[ ${check} -gt 0 ]]; then 

  annot HUBI check \
    ${scale_HUBI[@]} \
    -hshift -1014 \
    \
    -tpost "02:35" 2927 350 \
    -tpred "19:00" 3008 350 \
    \
    -rect "@"  368  4510 5 5 \
    -rect "@"  488  4450 5 5 \
    -rect "@"  733  4075 5 5 \
    -rect "@"  848  3970 5 5 \
    -rect "@"  968  3750 5 5 \
    -rect "@" 1088  3630 5 5 \
    -rect "@" 1208  3915 5 5 \
    -rect "@" 1328  3630 5 5 \
    -rect "@" 1448  3890 5 5 \
    -rect "@" 1568  3775 5 5 \
    -rect "@" 1688  3800 5 5 \
    -rect "@" 1808  3772 5 5 \
    -rect "@" 1928  3396 5 5 \
    -rect "@" 2048  3368 5 5 \
    -rect "@" 2168  3809 5 5 \
    -rect "@" 2288  3807 5 5 \
    -rect "@" 2408  3804 5 5 \
    -rect "@" 2528  3802 5 5 \
    -rect "@" 2528  2781 5 5 \
    -rect "@" 2648  3533 5 5 \
    -rect "@" 2768  3522 5 5 \
    -rect "@" 2888  3515 5 5 \
    -rect "+" 3008  3400 5 5 \
    \
    -line  "@"   -51  4873      8  4780 \
    -line  "@"     8  4780     68  4704 \
    -line  "@"    68  4704    128  4642 \
    -line  "@"   128  4642    188  4592 \
    -line  "@"   188  4592    248  4552 \
    -line  "@"   248  4552    308  4519 \
    -line  "@"   308  4519    368  4492 \
    -line  "@"   368  4492    428  4470 \
    \
    -line  "@"   428  4470    428  4147 \
    \
    -line  "@"   428  4147    488  4101 \
    -line  "@"   488  4101    548  4063 \
    -line  "@"   548  4063    608  4033 \
    -line  "@"   608  4033    668  4008 \
    -line  "@"   668  4008    728  3988 \
    -line  "@"   728  3988    788  3972 \
    -line  "@"   788  3972    848  3958 \
    -line  "@"   848  3958    908  3947 \
    -line  "@"   908  3947    968  3939 \
    -line  "@"   968  3939   1028  3932 \
    -line  "@"  1028  3932   1088  3926 \
    -line  "@"  1088  3926   1148  3921 \
    -line  "@"  1148  3921   1208  3917 \
    -line  "@"  1208  3917   1268  3914 \
    -line  "@"  1268  3914   1328  3912 \
    -line  "@"  1328  3912   1388  3910 \
    -line  "@"  1388  3910   1448  3908 \
    -line  "@"  1448  3908   1508  3907 \
    -line  "@"  1508  3907   1568  3905 \
    -line  "@"  1568  3905   1628  3905 \
    \
    -line  "+"  2348  3567   3068  3385 \
    \
    -dots 555 \
    "+" -2032 5014 \
    "@" -1912 5015 \
    "@" -1792 4966 \
    "+" -1672 4927 \
    "+" -1552 4730 \
    "+" -1432 4839 \
    "@" -1312 5041 \
    "@" -1192 4856 \
    "+" -1072 4829 \
    "@"  -952 4809 \
    "@"  -832 4851 \
    "+"  -712 4873 \
    "@"  -592 4939 \
    "@"  -472 4973 \
    "@"  -352 4898 \
    "@"  -232 4915 \
    "@"  -112 4892 \
    "+"     8 4782 \
    "+"   128 4637 \
    "+"   248 4555 \
    "+"   368 4492 \
    "+"   488 4080 \
    "@"   608 4139 \
    "+"   728 4006 \
    "@"   848 3876 \
    "+"   968 4008 \
    "+"  1088 3961 \
    "@"  1208 3733 \
    "+"  1328 3905 \
    "+"  1448 3817 \
    "@"  1568 3832 \
    "@"  1688 3530 \
    "+"  1808 3425 \
    "@"  1928 3621 \
    "+"  2048 3810 \
    "+"  2168 3615 \
    "+"  2288 3284 \
    "@"  2408 3543 \
    "+"  2528 3533 \
    "+"  2648 3464 \
    "+"  2768 3479 \
    "+"  2888 3427 \
    "@"  3008 3825

  annot BSTP check \
    ${scale_BSTP[@]} \
    -hshift -1014 \
    \
    -tpost "02:35" 2927  300 \
    -tpred "19:00" 3008  300 \
    \
    -rect "@"  368  700 5 5 \
    -rect "@"  488  690 5 5 \
    -rect "@"  733  640 5 5 \
    -rect "@"  848  650 5 5 \
    -rect "@"  968  615 5 5 \
    -rect "@" 1088  600 5 5 \
    -rect "@" 1208  640 5 5 \
    -rect "@" 1328  590 5 5 \
    -rect "@" 1448  635 5 5 \
    -rect "@" 1568  615 5 5 \
    -rect "@" 1688  621 5 5 \
    -rect "@" 1808  616 5 5 \
    -rect "@" 1928  555 5 5 \
    -rect "@" 2048  550 5 5 \
    -rect "@" 2168  622 5 5 \
    -rect "@" 2288  622 5 5 \
    -rect "@" 2408  622 5 5 \
    -rect "@" 2528  621 5 5 \
    -rect "@" 2528  454 5 5 \
    -rect "@" 2648  577 5 5 \
    -rect "@" 2768  575 5 5 \
    -rect "@" 2888  572 5 5 \
    -rect "+" 3008  556 5 5 \
    \
    -line  "@"   -51   796      8   781 \
    -line  "@"     8   781     68   768 \
    \
    -line  "@"    68   768     68   735 \
    \
    -line  "@"    68   735    128   725 \
    -line  "@"   128   725    188   717 \
    -line  "@"   188   717    248   711 \
    -line  "@"   248   711    308   706 \
    -line  "@"   308   706    368   701 \
    -line  "@"   368   701    428   698 \
    \
    -line  "@"   428   698    428   677 \
    \
    -line  "@"   428   677    488   670 \
    -line  "@"   488   670    548   663 \
    -line  "@"   548   663    608   658 \
    -line  "@"   608   658    668   654 \
    -line  "@"   668   654    728   651 \
    -line  "@"   728   651    788   649 \
    -line  "@"   788   649    848   646 \
    -line  "@"   848   646    908   644 \
    -line  "@"   908   644    968   643 \
    -line  "@"   968   643   1028   642 \
    -line  "@"  1028   642   1088   641 \
    -line  "@"  1088   641   1148   640 \
    -line  "@"  1148   640   1208   640 \
    -line  "@"  1208   640   1268   639 \
    -line  "@"  1268   639   1328   639 \
    -line  "@"  1328   639   1388   638 \
    -line  "@"  1388   638   1448   638 \
    -line  "@"  1448   638   1508   638 \
    -line  "@"  1508   638   1568   638 \
    -line  "@"  1568   638   1628   638 \
    \
    -line  "+"  2348   583   3068   553 \
    \
    -dots 555 \
    "+" -2032  819 \
    "@" -1912  819 \
    "@" -1792  811 \
    "+" -1672  805 \
    "+" -1552  773 \
    "+" -1432  791 \
    "@" -1312  824 \
    "@" -1192  793 \
    "+" -1072  789 \
    "@"  -952  786 \
    "@"  -832  793 \
    "+"  -712  796 \
    "@"  -592  807 \
    "@"  -472  813 \
    "@"  -352  800 \
    "@"  -232  803 \
    "@"  -112  799 \
    "+"     8  781 \
    "+"   128  724 \
    "+"   248  711 \
    "+"   368  701 \
    "+"   488  666 \
    "@"   608  676 \
    "+"   728  654 \
    "@"   848  633 \
    "+"   968  655 \
    "+"  1088  647 \
    "@"  1208  610 \
    "+"  1328  638 \
    "+"  1448  624 \
    "@"  1568  626 \
    "@"  1688  577 \
    "+"  1808  560 \
    "@"  1928  592 \
    "+"  2048  623 \
    "+"  2168  591 \
    "+"  2288  537 \
    "@"  2408  579 \
    "+"  2528  577 \
    "+"  2648  566 \
    "+"  2768  568 \
    "+"  2888  560 \
    "@"  3008  625

fi