/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitModels                                                     *
 * @(#)root/roofit:$Id: RooBlindTools.cxx 36274 2010-10-11 08:05:03Z wouter $
 * Authors:                                                                  *
 *   AR, Aaron Roodman, Stanford University, roodman@slac.stanford.edu       *
 *   WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu       *
 *                                                                           *
 * Copyright (c) 2000-2005, Regents of the University of California          *
 *                          and Stanford University. All rights reserved.    *
 *                                                                           *
 * Redistribution and use in source and binary forms,                        *
 * with or without modification, are permitted according to the terms        *
 * listed in LICENSE (http://roofit.sourceforge.net/license.txt)             *
 *****************************************************************************/

//-----------------------
// This Class's Header --
//-----------------------
#include "RooFit.h"

#include "RooBlindTools.h"
#include "RooBlindTools.h"

#include "RooErrorHandler.h"
#include "Riostream.h"
#include <fstream>
#include <math.h>  
#include <string.h>
#include <ctype.h>

ClassImp(RooBlindTools)


//----------------
// Constructors --
//----------------

//_____________________________________________________________________________
RooBlindTools::RooBlindTools(const char *stSeedIn, blindMode Mode,
			     Double_t centralValue, Double_t sigmaOffset, Bool_t s2bMode) :

  _PrecisionOffsetScale(sigmaOffset),
  _PrecisionCentralValue(centralValue),
  _mode(Mode),
  _s2bMode(s2bMode)
{
  setup(stSeedIn);
}



//_____________________________________________________________________________
RooBlindTools::RooBlindTools(const RooBlindTools& blindTool):
  _PrecisionOffsetScale(blindTool.getPrecisionOffsetScale()),
  _PrecisionCentralValue(blindTool.getPrecisionCentralValue()),
  _mode(blindTool.mode()),
  _s2bMode(blindTool._s2bMode) 
{
  setup(blindTool.stSeed());
}



//_____________________________________________________________________________
void RooBlindTools::setup(const char *stSeedIn)
{

  _stSeed = stSeedIn;

  _DeltaZScale = 1.56;

  _DeltaZOffset = _DeltaZScale*MakeOffset("abcdefghijklmnopqrstuvwxyz");

  _DeltaZSignFlip = MakeSignFlip("ijklmnopqrstuvwxyzabcdefgh");

  _AsymOffset = MakeGaussianOffset("opqrstuvwxyzabcdefghijklmn");

  _AsymSignFlip = MakeSignFlip("zyxwvutsrqponmlkjihgfedcba");

  _DeltaMScale = 0.1;

  _DeltaMOffset = _DeltaMScale*MakeOffset("opqrstuvwxyzabcdefghijklmn");

  _MysteryPhase = 3.14159 * 
                  MakeOffset("wxyzabcdefghijklmnopqrstuv");

  if (_s2bMode) {
    _PrecisionSignFlip = MakeSignFlip("zyxwvutsrqponmlkjihgfedcba");
  } else {
    _PrecisionSignFlip = MakeSignFlip("klmnopqrstuvwxyzabcdefghij");
  }

  _PrecisionOffset = _PrecisionOffsetScale*MakeGaussianOffset("opqrstuvwxyzabcdefghijklmn");

  _PrecisionUniform = _PrecisionOffsetScale*MakeOffset("jihgfedcbazyxwvutsrqponmlk");

  _STagConstant = Randomizer("fghijklmnopqrstuvwxyzabcde");
}


//--------------
// Destructor --
//--------------

//_____________________________________________________________________________
RooBlindTools::~RooBlindTools(){}

//-------------
// Functions --
//-------------


//_____________________________________________________________________________
Double_t RooBlindTools::HideDeltaZ(Double_t DeltaZ, Double_t STag)const{

  Int_t sTag = SignOfTag(STag);
  Double_t DeltaZPrime = _DeltaZSignFlip*DeltaZ*sTag + _DeltaZOffset;

  return DeltaZPrime;
}


//_____________________________________________________________________________
Double_t RooBlindTools::HiDelZPdG(Double_t DeltaZ, Double_t STag, Double_t PdG) const{

  Int_t sTag = SignOfTag(STag);
  Double_t DeltaZPrime = _DeltaZSignFlip*(DeltaZ - PdG)*sTag + _DeltaZOffset;

  return DeltaZPrime;
}


//_____________________________________________________________________________
Double_t RooBlindTools::UnHideDeltaZ(Double_t DeltaZPrime, Double_t STag) const{

  Int_t sTag = SignOfTag(STag);
  Double_t DeltaZ = (DeltaZPrime - _DeltaZOffset)/(sTag*_DeltaZSignFlip);

  return DeltaZ;
}


//_____________________________________________________________________________
Double_t RooBlindTools::UnHiDelZPdG(Double_t DeltaZPrime, Double_t STag, Double_t PdG) const{

  Int_t sTag = SignOfTag(STag);
  Double_t DeltaZ = PdG + (DeltaZPrime - _DeltaZOffset)/(sTag*_DeltaZSignFlip);

  return DeltaZ;
}


//_____________________________________________________________________________
Double_t RooBlindTools::UnHideAsym(Double_t AsymPrime) const{

  if(mode()==dataonly) return AsymPrime;

  Double_t Asym = (AsymPrime - _AsymOffset)/_AsymSignFlip;

  return Asym;
}


//_____________________________________________________________________________
Double_t RooBlindTools::HideAsym(Double_t Asym) const{

  if(mode()==dataonly) return Asym;

  Double_t AsymPrime = Asym*_AsymSignFlip + _AsymOffset;

  return AsymPrime;
}



//_____________________________________________________________________________
Double_t RooBlindTools::UnHideDeltaM(Double_t DeltaMPrime) const{

  if(mode()==dataonly) return DeltaMPrime;

  Double_t DeltaM = DeltaMPrime - _DeltaMOffset;

  return DeltaM;
}


//_____________________________________________________________________________
Double_t RooBlindTools::HideDeltaM(Double_t DeltaM) const{

  if(mode()==dataonly) return DeltaM;

  Double_t DeltaMPrime = DeltaM + _DeltaMOffset;

  return DeltaMPrime;
}


//_____________________________________________________________________________
Double_t RooBlindTools::UnHiAsPdG(Double_t AsymPrime, Double_t PdG) const{

  if(mode()==dataonly) return AsymPrime;

  Double_t Asym = PdG + (AsymPrime - _AsymOffset)/_AsymSignFlip;

  return Asym;
}


//_____________________________________________________________________________
Double_t RooBlindTools::MysteryPhase() const{

  if(mode()==dataonly) return 0.0;

  return _MysteryPhase;
}


//_____________________________________________________________________________
Double_t RooBlindTools::HiAsPdG(Double_t Asym, Double_t PdG) const{

  if(mode()==dataonly) return Asym;

  Double_t AsymPrime = (Asym - PdG)*_AsymSignFlip + _AsymOffset;

  return AsymPrime;
}


//_____________________________________________________________________________
Double_t RooBlindTools::UnHidePrecision(Double_t PrecisionPrime) const{

  if(mode()==dataonly) return PrecisionPrime;

  Double_t Precision(0.);

  if (_PrecisionSignFlip>0) {
    Precision = PrecisionPrime - _PrecisionOffset;
  }
  else {
    Precision = 2.0*_PrecisionCentralValue - PrecisionPrime + _PrecisionOffset;
  }


  return Precision;
}



//_____________________________________________________________________________
Double_t RooBlindTools::HidePrecision(Double_t Precision) const{

  if(mode()==dataonly) return Precision;

  Double_t PrecisionPrime(0.);

  if (_PrecisionSignFlip>0) {
    PrecisionPrime = Precision + _PrecisionOffset;
  }
  else {
    PrecisionPrime = 2.0*_PrecisionCentralValue - Precision + _PrecisionOffset;
  }

  return PrecisionPrime;
}



//_____________________________________________________________________________
Double_t RooBlindTools::UnHideOffset(Double_t PrecisionPrime) const{

  if(mode()==dataonly) return PrecisionPrime;

  return PrecisionPrime - _PrecisionOffset;
}



//_____________________________________________________________________________
Double_t RooBlindTools::HideOffset(Double_t Precision) const{

  if(mode()==dataonly) return Precision;
  
  return Precision + _PrecisionOffset;
}



//_____________________________________________________________________________
Double_t RooBlindTools::UnHideUniform(Double_t PrecisionPrime) const{

  if(mode()==dataonly) return PrecisionPrime;

  return PrecisionPrime - _PrecisionUniform;
}



//_____________________________________________________________________________
Double_t RooBlindTools::HideUniform(Double_t Precision) const{

  if(mode()==dataonly) return Precision;
  
  return Precision + _PrecisionUniform;
}




//_____________________________________________________________________________
Double_t RooBlindTools::RandomizeTag(Double_t STag, Int_t EventNumber) const{

  Int_t Seed = EventNumber % 7997 + 2; 
  Double_t r = PseudoRandom(Seed);
  Double_t STagPrime(0.0);

  if (r < _STagConstant){
    STagPrime = STag;
  } else {
    STagPrime = -1.0 * STag ;
  }

  return STagPrime;

}
   


//_____________________________________________________________________________
Double_t RooBlindTools::Randomizer(const char *StringAlphabet) const{

  char lowerseed[1024] ;
  strlcpy(lowerseed,_stSeed,1024) ;

  Int_t lengthSeed = strlen(lowerseed);

  for (Int_t j=0; j<lengthSeed; j++){
    lowerseed[j] =tolower(_stSeed[j]);
  }                                                                                                                                                                        
  Int_t sumSeed = 0;
  for (Int_t i=0; i<lengthSeed; i++){
    for (Int_t iAlphabet=0; iAlphabet<26; iAlphabet++){
      if ( lowerseed[i] == StringAlphabet[iAlphabet] ){
	if (_s2bMode) {
	  sumSeed =  (iAlphabet<<(5*(i%3)))^sumSeed;
	} else {
	  sumSeed = sumSeed + iAlphabet ;	
	}
      }
    }      
  }

  if (lengthSeed<5 || ((sumSeed<1 || sumSeed>8000)&&!_s2bMode)) {
    cout<< "RooBlindTools::Randomizer: Your String Seed is Bad: '" << _stSeed << "'" << endl ;
    RooErrorHandler::softAbort() ;                                                                                                                     
  }
  
  Int_t ia = 8121;
  Int_t ic = 28411;
  Int_t im = 134456;
  UInt_t jRan = (sumSeed*ia + ic) % im;

  jRan = (jRan*ia + ic) % im;
  jRan = (jRan*ia + ic) % im;
  jRan = (jRan*ia + ic) % im;

  Double_t theRan = (float) jRan / (float) im;     

  return theRan;    //theRan is between 0.0 - 1.0

}


//_____________________________________________________________________________
Double_t RooBlindTools::PseudoRandom(Int_t Seed) const{

  if (Seed<1 || Seed>8000 ) {
    cout<< "RooBlindTools::PseudoRandom: Your integer Seed is Bad" <<endl;
  }
 
  Int_t ia = 8121;
  Int_t ic = 28411;
  Int_t im = 134456;
  UInt_t jRan = (Seed*ia + ic) % im;

  jRan = (jRan*ia + ic) % im;
  jRan = (jRan*ia + ic) % im;
  jRan = (jRan*ia + ic) % im;

  Double_t theRan = (float) jRan / (float) im;     

  return theRan;    //theRan is between 0.0 - 1.0

}



//_____________________________________________________________________________
Double_t RooBlindTools::MakeOffset(const char *StringAlphabet) const{

  Double_t theRan = Randomizer(StringAlphabet);

  Double_t theOffset = (2.0)*theRan - (1.0);

  return theOffset;   //theOffset lies between -1.0 and 1.0
}



//_____________________________________________________________________________
Double_t RooBlindTools::MakeGaussianOffset(const char *StringAlphabet) const{

  Double_t theRan1 = Randomizer(StringAlphabet);
  Double_t theRan2 = Randomizer("cdefghijklmnopqrstuvwxyzab");

  if (theRan1==0.0 || theRan1==1.0){
    theRan1 = 0.5;
  }
  if (theRan2==0.0 || theRan2==1.0){
    theRan2 = 0.5;
  }

  Double_t theOffset = sin(2.0*3.14159*theRan1)*sqrt(-2.0*log(theRan2));

  return theOffset;   //theOffset is Gaussian with mean 0, sigma 1
}



//_____________________________________________________________________________
Double_t RooBlindTools::MakeSignFlip(const char *StringAlphabet) const{

  Double_t theRan = Randomizer(StringAlphabet);

  Double_t theSignFlip = 1.0;
  if (theRan>0.5){
    theSignFlip = 1.0;
  } else {
    theSignFlip = -1.0;
  }

  return theSignFlip;  //theSignFlip is = +1 or -1
}


//_____________________________________________________________________________
Int_t RooBlindTools::SignOfTag(Double_t STag) const{

  Int_t sTag;
  if (STag < 0.0){
    sTag = -1;
  } 
  else if (STag > 0.0) {
    sTag = 1;
  }
  else {
    sTag = 1;
  }

  return sTag;
   
}