QwPromptSummary.cc

Go to the documentation of this file.

#include "QwPromptSummary.h"

#include "QwColor.h"
#include "QwLog.h"

#include "TROOT.h"
#include "TMath.h"

/**
 *  \file   QwPromptSummary.cc
 *  \brief  
 *  \author jhlee@jlab.org
 *  \date   Friday, December 16 10:55:31 EST 2011
 */

//
//
//
//  PromptSummaryElement
//
//
//
//

ClassImp(PromptSummaryElement)
ClassImp(QwPromptSummary)

PromptSummaryElement::PromptSummaryElement()
{
  fElementName         = "";
  
  fYield               = 0.0;
  fYieldError          = 0.0;
  fYieldWidth          = 0.0;
  fYieldUnit           = "";

  fAsymDiff            = 0.0;
  fAsymDiffError       = 0.0;
  fAsymDiffWidth       = 0.0;
  fAsymDiffUnit        = "";

};


PromptSummaryElement::PromptSummaryElement(TString name)
{
  fElementName         = name;
  
  fYield               = 0.0;
  fYieldError          = 0.0;
  fYieldWidth          = 0.0;
  fYieldUnit           = "";

  fAsymDiff            = 0.0;
  fAsymDiffError       = 0.0;
  fAsymDiffWidth       = 0.0;
  fAsymDiffUnit        = "";

};

PromptSummaryElement::~PromptSummaryElement()
{
};


void
PromptSummaryElement::FillData(
             Double_t yield, 
             Double_t yield_err, 
             Double_t yield_width,
             TString  yield_unit,
             Double_t asym_diff, 
             Double_t asym_diff_err, 
             Double_t asym_diff_width, 
             TString  asym_diff_unit
             )
{
  fYield                = yield;
  fYieldError           = yield_err;
  fYieldWidth           = yield_width;
  fYieldUnit            = yield_unit;

  fAsymDiff            = asym_diff;
  fAsymDiffError       = asym_diff_err;
  fAsymDiffWidth       = asym_diff_width;
  fAsymDiffUnit        = asym_diff_unit;

  return;
};


TString
PromptSummaryElement::GetTextSummary()
{
  //  return Form("%14s | %12.2lf +/- %8.2lf | %+12.2lf +/- %8.2lf | %+12.2lf +/- %8.2lf\n", fElementName.Data(), fYield, fYieldError, fAsymDiff, fAsymDiffError, fAsymDiffWidth, fAsymDiffWidthError);
  TString test;
  return test;
};
TString
PromptSummaryElement::GetCSVSummary()
{
  return Form("%s,%e,%e,%e,%e,%e,%e", fElementName.Data(), fYield, fYieldError, fYieldWidth, fAsymDiff, fAsymDiffError, fAsymDiffWidth);
};


void 
PromptSummaryElement::Set(TString type, const Double_t a, const Double_t a_err, const Double_t a_width)
{
  Double_t asymmetry_ppm = 1e-6;

  if(type.Contains("yield")) {
    if (fElementName.Contains("bcm")) {
      this->SetYieldUnit("uA");
    }
    else if (fElementName.Contains("bpm")) {
      this->SetYieldUnit("mm");
    }
    else if (fElementName.Contains("MD")) {
      this->SetYieldUnit("V/uA");
    }
    else if (fElementName.Contains("lumi")) {
      this->SetYieldUnit("V/uA");
    }
    else {
      this->SetYieldUnit("---");
    }
    this->SetYield(a);
    this->SetYieldError(a_err);
    this->SetYieldWidth(a_width);
  } 
  else if(type.Contains("asymmetry")) {
    
    if (fElementName.Contains("bpm")) {
      this->SetDifferenceUnit("nm");
      this->SetDifference(a);
      this->SetDifferenceError(a_err);
      this->SetDifferenceWidth(a_width);
    } 
    else {
      this->SetAsymmetryUnit("ppm");
      this->SetAsymmetry(a/asymmetry_ppm);
      this->SetAsymmetryError(a_err/asymmetry_ppm);
      this->SetAsymmetryWidth(a_width/asymmetry_ppm);
    }
 
  } 
  else if(type.Contains("difference")) {
  } 
  else {
  }
  return;
};

QwPromptSummary::QwPromptSummary()
{
  fRunNumber    = 0;
  fRunletNumber = 0;
  fElementList = new TObjArray();
 
  fElementList -> Clear();
  fElementList -> SetOwner(kTRUE);

  fNElements = 0;

  fLocalDebug = kTRUE;
  
  this->SetupElementList();

};


QwPromptSummary::QwPromptSummary(Int_t run_number, Int_t runlet_number)
{
  fRunNumber    = run_number;
  fRunletNumber = runlet_number;

  fElementList = new TObjArray();
 
  fElementList -> Clear();
  fElementList -> SetOwner(kTRUE);

  fNElements = 0;

  fLocalDebug = kFALSE;
  
  this->SetupElementList();

};


QwPromptSummary::~QwPromptSummary()
{
  if(fElementList) delete fElementList; fElementList = NULL;
};



void 
QwPromptSummary::SetupElementList()
{
  this->AddElement(new PromptSummaryElement("charge"));
  this->AddElement(new PromptSummaryElement("target_x"));
  this->AddElement(new PromptSummaryElement("target_y"));
  this->AddElement(new PromptSummaryElement("angle_x"));
  this->AddElement(new PromptSummaryElement("angle_y"));
  this->AddElement(new PromptSummaryElement("energy"));
  
  this->AddElement(new PromptSummaryElement("bcm1"));
  this->AddElement(new PromptSummaryElement("bcm2"));
  this->AddElement(new PromptSummaryElement("bcm5"));
  this->AddElement(new PromptSummaryElement("bcm6"));
  this->AddElement(new PromptSummaryElement("bcm7"));
  this->AddElement(new PromptSummaryElement("bcm8"));
  this->AddElement(new PromptSummaryElement("bpm3h04_effectivecharge"));
  this->AddElement(new PromptSummaryElement("bpm3h09_effectivecharge"));
  this->AddElement(new PromptSummaryElement("3c12x"));
  this->AddElement(new PromptSummaryElement("3c12y"));
  this->AddElement(new PromptSummaryElement("3h04x"));
  this->AddElement(new PromptSummaryElement("3h04y"));
  this->AddElement(new PromptSummaryElement("3h07cx"));
  this->AddElement(new PromptSummaryElement("3h07cy"));
  this->AddElement(new PromptSummaryElement("3h09x"));
  this->AddElement(new PromptSummaryElement("3h09y"));
  this->AddElement(new PromptSummaryElement("3h09bx"));
  this->AddElement(new PromptSummaryElement("3h09by"));

  this->AddElement(new PromptSummaryElement("bcm1-bcm2"));
  this->AddElement(new PromptSummaryElement("bcm1-bcm5"));
  this->AddElement(new PromptSummaryElement("bcm1-bcm6"));
  this->AddElement(new PromptSummaryElement("bcm2-bcm5"));
  this->AddElement(new PromptSummaryElement("bcm2-bcm6"));
  this->AddElement(new PromptSummaryElement("bcm5-bcm6"));

  this->AddElement(new PromptSummaryElement("bcm1-bcm7"));
  this->AddElement(new PromptSummaryElement("bcm1-bcm8"));

  // this->AddElement(new PromptSummaryElement("bcm2-bcm7"));
  // this->AddElement(new PromptSummaryElement("bcm2-bcm8"));

  // this->AddElement(new PromptSummaryElement("bcm5-bcm7"));
  // this->AddElement(new PromptSummaryElement("bcm5-bcm8"));
  // this->AddElement(new PromptSummaryElement("bcm6-bcm7"));
  // this->AddElement(new PromptSummaryElement("bcm6-bcm8"));

  this->AddElement(new PromptSummaryElement("bcm5-bcm7"));
  this->AddElement(new PromptSummaryElement("bcm7-bcm8"));
  



  this->AddElement(new PromptSummaryElement("MD1"));
  this->AddElement(new PromptSummaryElement("MD2"));
  this->AddElement(new PromptSummaryElement("MD3"));
  this->AddElement(new PromptSummaryElement("MD4"));
  this->AddElement(new PromptSummaryElement("MD5"));
  this->AddElement(new PromptSummaryElement("MD6"));
  this->AddElement(new PromptSummaryElement("MD7"));
  this->AddElement(new PromptSummaryElement("MD8"));

  this->AddElement(new PromptSummaryElement("MD1-MD5"));
  this->AddElement(new PromptSummaryElement("MD2-MD6"));
  this->AddElement(new PromptSummaryElement("MD3-MD7"));
  this->AddElement(new PromptSummaryElement("MD4-MD8"));

  this->AddElement(new PromptSummaryElement("MD_AllBars"));
  this->AddElement(new PromptSummaryElement("MD_EvenBars"));
  this->AddElement(new PromptSummaryElement("MD_OddBars"));

  this->AddElement(new PromptSummaryElement("MD_Even-Odd"));
  this->AddElement(new PromptSummaryElement("MD1/5-MD3/7"));
  this->AddElement(new PromptSummaryElement("MD2/6-MD4/8"));

  this->AddElement(new PromptSummaryElement("dslumi1"));
  this->AddElement(new PromptSummaryElement("dslumi2"));
  this->AddElement(new PromptSummaryElement("dslumi3"));
  this->AddElement(new PromptSummaryElement("dslumi4"));
  this->AddElement(new PromptSummaryElement("dslumi5"));
  this->AddElement(new PromptSummaryElement("dslumi6"));
  this->AddElement(new PromptSummaryElement("dslumi7"));
  this->AddElement(new PromptSummaryElement("dslumi8"));

  this->AddElement(new PromptSummaryElement("dslumi_even"));
  this->AddElement(new PromptSummaryElement("dslumi_odd"));
  this->AddElement(new PromptSummaryElement("dslumi_sum"));
  
  this->AddElement(new PromptSummaryElement("uslumi1"));
  this->AddElement(new PromptSummaryElement("uslumi3"));
  this->AddElement(new PromptSummaryElement("uslumi5"));
  this->AddElement(new PromptSummaryElement("uslumi7"));

  this->AddElement(new PromptSummaryElement("uslumi_sum"));



};


void 
QwPromptSummary::AddElement(PromptSummaryElement *in)
{
  Int_t pos = 0;

  pos = fElementList -> AddAtFree(in);
  if(fLocalDebug) {
    printf("AddElement at pos %d\n", pos);
  }

  fNElements++;
  return;
};


PromptSummaryElement *
QwPromptSummary::GetElementByName(TString name)
{

  PromptSummaryElement* an_element = NULL;
  TObjArrayIter next(fElementList);
  TObject* obj = NULL;
  TString get_name = "";

  while ( (obj = next()) )
    {
      an_element = (PromptSummaryElement*) obj;
      get_name   = an_element->GetName();
      if( get_name.Contains(name) ) {
  if(fLocalDebug) {
    std::cout << "System " << an_element->GetName()
        << " QwPromptSummary::GetElementByName address at" << an_element << std::endl;
  }
  return an_element;
      }
    }
  
  return NULL;
};

TString
QwPromptSummary::PrintTextSummaryHeader()
{
  TString out = "";
  //  TString filename = "";

  //  filename = Form("summary_%d_%d.txt", fRunNumber, fRunletNumber);
  
  out = "======= BEGIN ======= \n";
  out += Form("        RUN = %5d RUNLET %2d \n\n\n\n\n\n", fRunNumber, fRunletNumber);
  return out;
};



TString
QwPromptSummary::PrintTextSummaryTailer()
{
  TString out = "";
  
  out =  " ======== END of SUMMARY======== \n";
  out += " ======== END ======== \n\n";
  out += " Please contact Rakitha Beminiwattha for any queries and suggestions \n";
  out += " rakithab@jlab.org \n";
  return out;
};


TString
QwPromptSummary::PrintCSVHeader()
{
  TString out = "";
  TString filename = "";

  filename = Form("summary_%d_%d.txt", fRunNumber, fRunletNumber);
  
  out = "! This csv file is desinged for making plots easily.\n";
  out += "!See ";
  out += filename;
  out += " in http://qweak.jlab.org/textsummaries/ for theirs units\n";
  out += "!Please contact Jeong Han Lee via jhlee@jlab.org if one has comments or questions.\n";

  return out;
};



void
QwPromptSummary::FillDataInElement(TString  name, 
           Double_t yield, 
           Double_t yield_err, 
           Double_t yield_width, 
           TString  yield_unit, 
           Double_t asym_diff, 
           Double_t asym_diff_err, 
           Double_t asym_diff_width,
           TString  asym_diff_unit
           )
{
 
  PromptSummaryElement* an_element = NULL;
  an_element = this->GetElementByName(name);
  if(an_element) {
    an_element->FillData(yield, yield_err, yield_width, yield_unit, asym_diff, asym_diff_err, asym_diff_width, asym_diff_unit);
  }
  else {
    if(fLocalDebug) {
      std::cout 
  << "QwPromptSummary::FillDataInElement : No Element with the name " 
  << name 
  <<  std::endl;
    }
  }
  return;
};


void
QwPromptSummary::FillYieldToElement(TString name, 
            Double_t yield, 
            Double_t yield_error, 
            Double_t yield_width,
            TString yield_unit
            )
{
  PromptSummaryElement* an_element = NULL;
  an_element = this->GetElementByName(name);
  if(an_element) {
    an_element->SetYield(yield);
    an_element->SetYieldError(yield_error);
    an_element->SetYieldWidth(yield_width);
  }
  else {
    if(fLocalDebug) {
      std::cout 
  << "QwPromptSummary::FillYieldToElement : No Element with the name " 
  << name 
  <<  std::endl;
    }
  }
  return;
};

void 
QwPromptSummary::FillAsymDiffToElement(TString name, 
               Double_t asym_diff, 
               Double_t asym_diff_err, 
               Double_t asym_diff_width,
               TString asym_diff_unit
               )
{
  PromptSummaryElement* an_element = NULL;
  an_element = this->GetElementByName(name);
  if(an_element) {
    an_element->SetAsymmetry(asym_diff);
    an_element->SetAsymmetryError(asym_diff_err);
    an_element->SetAsymmetryWidth(asym_diff_width);
  }
  else {
    if(fLocalDebug) {
      std::cout 
  << "QwPromptSummary::FillYieldToElement : No Element with the name " 
  << name 
  <<  std::endl;
    }
  }

  return;
};

void 
QwPromptSummary::FillDoubleDifference(TString type, TString name1, TString name2)
{
  PromptSummaryElement* an_element = NULL;


  PromptSummaryElement* one_element = NULL;
  PromptSummaryElement* two_element = NULL;

  one_element = this->GetElementByName(name1);
  two_element = this->GetElementByName(name2);


  if(one_element and two_element ) {

    an_element = this->GetElementByName(name1+"-"+name2);

    if(an_element) {
   
      
      Double_t diff       = 0.0;
      Double_t error_diff = 0.0;
      Double_t width_diff = 0.0;
      
      Double_t a = 0.0;
      Double_t b = 0.0;
      Double_t a_err = 0.0;
      Double_t b_err = 0.0;
      Double_t a_wit = 0.0;
      Double_t b_wit = 0.0;
      
      if(type.Contains("yield")) {
  a     = one_element -> GetYield();
  b     = two_element -> GetYield();
  a_err = one_element -> GetYieldError();
  b_err = two_element -> GetYieldError();
  a_wit = one_element -> GetYieldWidth();
  b_wit = two_element -> GetYieldWidth();
  
  diff       = a - b;
  error_diff = TMath::Sqrt(a_err*a_err + b_err*b_err);
  width_diff = a_wit - b_wit;

  an_element -> SetYield(diff);
  an_element -> SetYieldError(error_diff);
  an_element -> SetYieldWidth(width_diff);
  
      }
      else if (type.Contains("asymmetry")) {
  a     = one_element -> GetAsymmetry();
  b     = two_element -> GetAsymmetry();
  a_err = one_element -> GetAsymmetryError();
  b_err = two_element -> GetAsymmetryError();
  a_wit = one_element -> GetAsymmetryWidth();
  b_wit = two_element -> GetAsymmetryWidth();
  
  diff       = a - b;
  error_diff = TMath::Sqrt(a_err*a_err + b_err*b_err);
  width_diff = a_wit - b_wit;

  an_element -> SetAsymmetry(diff);
  an_element -> SetAsymmetryError(error_diff);
  an_element -> SetAsymmetryWidth(width_diff);
      }
    
    }
    else {
      if(fLocalDebug) {
  std::cout 
    << "QwPromptSummary::FillYieldToElement : No Element with the name " 
    << name1 + "-" + name2 
    <<  std::endl;
      }
  }

  }
  else {
    if(fLocalDebug) {
      std::cout 
  << "QwPromptSummary::FillDoubleDifference: No Elements with the name " 
  << name1 << " and " << name2  
  <<  std::endl;
    }
  }
}


void
QwPromptSummary::PrintCSV()
{
  printf("-----------------------\n");
  TObjArrayIter next(fElementList);
  TObject* obj = NULL;
  while ( (obj = next()) )
    {
      PromptSummaryElement* an_element = (PromptSummaryElement*) obj;
      std::cout << an_element -> GetCSVSummary() << std::endl;
    }
  printf("-----------------------\n");
  return;
};

void
QwPromptSummary::PrintTextSummary()
{
  printf("-----------------------\n");
  TObjArrayIter next(fElementList);
  TObject* obj = NULL;
  while ( (obj = next()) )
    {
      PromptSummaryElement* an_element = (PromptSummaryElement*) obj;
      std::cout << an_element -> GetTextSummary() << std::endl;
    }
  printf("-----------------------\n");
  return;
};