#include <math.h>
#include <stdlib.h>
#include <iostream.h>
#include "FourVector.h"
#include "anl_utils.h"

FourVector pjet_from_event(int jet, float event[]){

  float px, py, pz, E;
  if (jet == 0) {
    px = event[0];
    py = event[1];
    pz = event[2];
    E  = event[3]; 
  }
  else if (jet == 1) {
    px = event[5];
    py = event[6];
    pz = event[7];
    E  = event[8]; 
  }
  else if (jet == 2) {
    px = event[10];
    py = event[11];
    pz = event[12];
    E  = event[13]; 
  }
  else if (jet == 3) {
    px = event[15];
    py = event[16];
    pz = event[17];
    E  = event[18]; 
  }

  FourVector* pjetPtr = new FourVector(px, py, pz, E);
  return *pjetPtr;

}

void pair_HZ_masses (float mass[3][4], 
                     int& h1, int& h2, int& z1, int& z2){

  const float MZ = 91.2;
  float min_diff;
  for (int i=0; i<3; i++){
    for (int j=1; j<4; j++){
      float diff = fabs(mass[i][j] - MZ);
      if ( i==0 && j==1 ){
        min_diff = diff; 
        z1 = i;
        z2 = j; 
      }
      else if ( diff < min_diff ) {
        min_diff = diff; 
        z1 = i;
        z2 = j; 
      }
    }
  }

  if ( (z1 == 0 && z2 == 1) || (z1 == 1 && z2 == 0) ) {
    h1 = 2;
    h2 = 3;
  }
  else if ( (z1 == 0 && z2 == 2) || (z1 == 2 && z2 == 0) ) {
    h1 = 1;
    h2 = 3;
  }
  else if ( (z1 == 0 && z2 == 3) || (z1 == 3 && z2 == 0) ) {
    h1 = 1;
    h2 = 2;
  }
  else if ( (z1 == 1 && z2 == 2) || (z1 == 2 && z2 == 1) ) {
    h1 = 0;
    h2 = 3;
  }
  else if ( (z1 == 1 && z2 == 3) || (z1 == 3 && z2 == 1) ) {
    h1 = 0;
    h2 = 2;
  }
  else if ( (z1 == 2 && z2 == 3) || (z1 == 3 && z2 == 2) ) {
    h1 = 0;
    h2 = 1;
  }

}