binning.cxx File Reference

#include "TCanvas.h"
#include "TFile.h"
#include "TTree.h"
#include "TH1F.h"
#include "TMath.h"
#include "TObjectTable.h"
#include <sstream>
#include <iostream>
#include <fstream>
#include <string>
#include <cstring>
#include <vector>
#include <cmath>

Go to the source code of this file.

Typedefs
typedef std::vector< double >	vdouble

Functions
double	m_left (0)
double	m_right (0)
int	m_bins (0)
bool	m_debug (true)
vdouble	vec_bg (0)
vdouble	v (0)
void	f_opt (double left, double right, int bins)
double	f_cals (double left, double right, double step)
void	f_gen_hists ()
void	print_vec (vdouble v)
void	print_vec_verso (vdouble v)
const double	test_times (2)
double	total_n (0)
double	save_left (0)
double	save_right (0)
double	save_pos (0)
double	save_s (0)
int	main (int argc, char **argv)
void	f_group (double left, double right, int &num, int &index)
int	f_group (double bg)

Variables
TTree *	t = new TTree("t", "tree to save original data")
TTree *	t_save = new TTree("t_save", "tree to save entropy")
int	tot_num_group
double	x_group [2000]
double	hits_group [2000]

Typedef Documentation

vdouble

typedef std::vector<double> vdouble

Definition at line 26 of file binning.cxx.

Function Documentation

f_cals()

double f_cals	(	double	left,
		double	right,
		double	step
	)

Definition at line 157 of file binning.cxx.

                                                    { 
  //  gObjectTable->Print();
  if(m_debug) cout << "f_cal() is happyily running!" << endl;
  for(int i=0; i<t_save->GetEntries(); i++){
    t_save->GetEntry(i);
    if(fabs(save_left-left)<0.0001 && fabs(save_right-right)<0.0001 && fabs(save_pos-pos)<0.0001) return save_s;
  } 
  double left_s(0), right_s(0), total_s(0);
  double left_ratio(0), right_ratio(0);
  TFile *file = new TFile("temp.root", "recreate");
  TTree *tree = t->CopyTree(Form("bg>%f && bg<%f", left, right));
  if(m_debug){ cout << "after open temp.root" << endl; gDirectory->pwd();  gDirectory->ls(); }
  double a_step(0.93*2/10);
  for(int i=0; i<2; i++){// only do test twice
    if(i==0) {m_left = left; m_right = pos;}
    else {m_left = pos; m_right = right;}
    int m_num_group(0);// how many groups will be covered by this bin
    int m_index_group(-1); // the index of left boundary of the groups
    f_group(m_left, m_right, m_num_group, m_index_group);
    if(m_debug) cout << "left " << m_left << " right " << m_right << " num of groups: " << m_num_group << "  boundary index: " << m_index_group << endl;
    for(int j=0; j<10; j++){// index of angle
      TTree *temp_tree = new TTree("tepm_tree", "temp tree");
      temp_tree = tree->CopyTree(Form("bg>%f && bg<%f && costheta>%f && costheta<%f", m_left, m_right, -0.93+j*a_step, -0.93+(j+1)*a_step));
      int nhits(0);
      Float_t m_bg(0);
      double tot_nhits(0);
      temp_tree->SetBranchAddress("nhits", &nhits);
      temp_tree->SetBranchAddress("bg", &m_bg);
      for(int m=0; m<tot_num_group; m++) hits_group[m]=0; 
      for(int k=0; k<temp_tree->GetEntries(); k++){// loop in track
    temp_tree->GetEntry(k);
    tot_nhits += nhits;
    hits_group[f_group(m_bg)] += nhits;
      }
      if(tot_nhits<2000){ file->Close(); return(100000.); } // if too small hits in a bin, bad (large) entropy
      else{
    for(int n=0; n<m_num_group; n++){ // loop in groups
      if(hits_group[m_index_group+n]==0) continue;
      // cout << "group_hits " << hits_group[m_index_group+n] << "  tot_nhits " << tot_nhits << endl;
      double p(hits_group[m_index_group+n]/tot_nhits); // the probability to find a hit of group_k in a bin
      if(p>1) cout << "m_index_group " << m_index_group << "  n " << n << " hits_group[] " <<  hits_group[m_index_group+n] << "  tot_nhits " << tot_nhits << endl;
      if(i==0) left_s += -p*TMath::Log2(p);
      if(i==1) right_s += -p*TMath::Log2(p);
    }// end of loop in groups
    if(i==0) left_ratio += tot_nhits;
    if(i==1) right_ratio += tot_nhits;
      }
      if(m_debug) cout << "m_left:" << m_left << "  m_right:" << m_right << "  j:" << j << "  left_s:" << left_s << "  right_s:" << right_s << endl;
      // temp_tree->Delete();
    }// end of loop angle
    if(fabs(right-pos)<0.00001) break;
  }// end of loop beta-gamma
  //  tree->Delete();
  file->Close();
  left_ratio = left_ratio/(left_ratio+right_ratio);
  right_ratio = 1-left_ratio;
  total_s = left_ratio*left_s + right_ratio*right_s;
  if(m_debug) cout << "left_ratio: " << left_ratio << "  right_ratio: " << right_ratio << "  total_s: " << total_s << endl;
  // save to the tree t_save
  save_left = left;
  save_right = right;
  save_pos = pos;
  save_s = total_s;
  t_save->Fill();
  return(total_s);
}

Referenced by f_opt().

f_gen_hists()

void f_gen_hists

(

)

Definition at line 32 of file binning.cxx.

{};

Referenced by main().

f_group() [1/2]

int f_group

(

double

bg

)

Definition at line 150 of file binning.cxx.

                      {
  for(int i=1; i<tot_num_group; i++){
    if(bg<x_group[i]) return i-1; 
  }
  return tot_num_group;
}

f_group() [2/2]

void f_group	(	double	left,
		double	right,
		int &	num,
		int &	index
	)

Definition at line 141 of file binning.cxx.

                                                             {
  num = 0;
  index = -1;
  for(int i=1; i<tot_num_group; i++){
    if(num==0 && left<x_group[i]) { index = i-1; num = 1; }
    if(right<=x_group[i]){ num = i - index; break;}
  }
}

Referenced by f_cals(), and main().

f_opt()

void f_opt	(	double	left,
		double	right,
		int	bins
	)

Definition at line 88 of file binning.cxx.

                                                   {
  if(m_debug) cout << "f_opt() is happily running!" << endl;
  int total(1); // number of bins
  vec_bg.clear();
  vec_bg.push_back(m_left);
  vec_bg.push_back(m_right);
  TTree *tree = new TTree("tree", "tree to save temp data");
  tree = t->CopyTree(Form("bg>%f && bg<%f", m_left, m_right));
  int m_hits;
  total_n = 0;
  tree->SetBranchAddress("nhits", &m_hits);
  int m_num_total(tree->GetEntries());
  for(int k=0; k<m_num_total; k++){
    tree->GetEntry(k);
    total_n += m_hits;
  }
  tree->Delete();
 
  while(total<bins){
    double temp_s(0), temp_pos(0);
    int temp_i(-1);
    for(unsigned int i=0; i<vec_bg.size(); i++){
      double pos_step((vec_bg[i+1]-vec_bg[i])/test_times);
      double ori_s(-1);
      for(double pos=vec_bg[i]+pos_step; pos<vec_bg[i+1]-0.00001; pos+=pos_step){
    if(ori_s<0) ori_s = f_cals(vec_bg[i], vec_bg[i+1], vec_bg[i+1]);// for the ori one, no mid point
    v.clear();
    v = vec_bg;
    v.insert(v.begin()+i+1, pos);
    double diff_s = ori_s - f_cals(vec_bg[i], vec_bg[i+1], pos); // information gain 
    if(diff_s > temp_s){
      temp_s = diff_s;
      temp_i = i;
      temp_pos = pos;
    }
    if(m_debug){
      cout << "i:" << i << "  pos:" << pos << "  diff_s:" << diff_s << "  temp_s:" << temp_s << endl;
      cout << "temp_v: " << endl;
      print_vec(v);
      cout << "vec_bg: " << endl;
      print_vec(vec_bg);
    }// end of print results
      }// end of loop in inert region
    }// end of loop in whole region
    vec_bg.insert(vec_bg.begin()+temp_i+1, temp_pos);
    cout << "============== vec_bg =============" <<endl;
    print_vec(vec_bg);
    cout << endl << endl;
    total ++;
  }// end of while()
}// end of f_opt()

Referenced by main().

m_bins()

int m_bins

(

0

)

Referenced by main(), and print_vec_verso().

m_debug()

bool m_debug

(

true

)

m_left()

double m_left

(

0

)

Referenced by f_cals(), f_opt(), and main().

m_right()

double m_right

(

0

)

Referenced by f_cals(), f_opt(), and main().

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 43 of file binning.cxx.

                               {
  //  gObjectTable->Print();
  if(argc<2){ cout << "please append config file, ex. binning.exe config.txt" << endl; return 0;}
  TFile *f = new TFile("pre_data.root");
  t = (TTree*)f->Get("n103");
  ifstream config_file(argv[1]);
  if(!config_file){
    cout << "cannot read config file correctly!" << endl;
    return 1;
  }
  TFile *f_group = new TFile("group_points.root");
  if(!f_group){
    cout << "cannot read group points file correctly!" << endl;
    return 1;
  }
  TTree *t_group = (TTree*)f_group->Get("t");
  double m_x_temp(0);
  t_group->SetBranchAddress("x", &m_x_temp);
  tot_num_group = t_group->GetEntries();
  for(int i=0; i<tot_num_group; i++) { t_group->GetEntry(i); x_group[i] = m_x_temp; }
  // t_group->Delete();
  f_group->Close();
  delete f_group;
  // prepare tree t_save
  t_save->Branch("save_left", &save_left, "save_left/D");
  t_save->Branch("save_right", &save_right, "save_right/D");
  t_save->Branch("save_pos", &save_pos, "save_pos/D");
  t_save->Branch("save_s", &save_s, "save_s/D");
  //
  while(m_bins!=-1){ // -1 is the end signal
    config_file >> m_left >> m_right >> m_bins >> m_debug;
    if(m_debug) cout << "config: " << m_left << " " << m_right << " " << m_bins << endl;
    f_opt(m_left, m_right, m_bins);
    cout <<"finial results for this time binning" << endl;
    print_vec_verso(vec_bg);
    f_gen_hists();
  }
  t->Delete();
  t_save->Delete();
  f->Close();
  delete f; 
  return 0;
}

print_vec()

void print_vec

(

vdouble

v

)

Definition at line 224 of file binning.cxx.

                         {
  for(vdouble::iterator i=v.begin(); i<v.end(); i++) cout << *i << "  " ;
  cout << endl;
}

Referenced by f_opt().

print_vec_verso()

void print_vec_verso

(

vdouble

v

)

Definition at line 229 of file binning.cxx.

                               {
  cout << "vec[" << m_bins << "] = {" ;
  for(vdouble::iterator i=v.begin(); i<v.end(); i++) cout << *i << ", " ;
  cout << "};" << endl;
}

Referenced by main().

save_left()

double save_left

(

0

)

Referenced by f_cals(), and main().

save_pos()

double save_pos

(

0

)

Referenced by f_cals(), and main().

save_right()

double save_right

(

0

)

Referenced by f_cals(), and main().

save_s()

double save_s

(

0

)

Referenced by f_cals(), and main().

test_times()

const double test_times

(

2

)

Referenced by f_opt().

total_n()

double total_n

(

0

)

Referenced by f_opt().

v()

vdouble v

(

0

)

vec_bg()

vdouble vec_bg

(

0

)

Referenced by f_opt(), and main().

Variable Documentation

hits_group

double hits_group[2000]

Definition at line 41 of file binning.cxx.

Referenced by f_cals().

t

TTree* t = new TTree("t", "tree to save original data")

Definition at line 23 of file binning.cxx.

Referenced by BesGMenuBar::AddFrameBefore(), BesGeoTrack::AddPoint(), BesGMenuBar::AddPopup(), TrkFitMaker::allReps(), T3DLine::approach_line(), TRunge::approach_line(), T3DLine::approach_point(), XmlRpc::XmlRpcValue::assertTypeOrInvalid(), barrel_con2root(), barrel_conv2root(), G4HepMCInterface::Boost(), TBuilder::build(), TBuilder0::buildRphi(), TBuilder::buildRphi(), TBuilder::buildStereo(), TBuilder0::buildStereo(), TBuilderCosmic::buildStereo(), TBuilder0::buildStereo0(), TBuilder::buildStereoNew(), calib_barrel_left_offset1_bunch0_3::calculate_funcs(), calib_barrel_left_offset1_bunch0_4::calculate_funcs(), calib_barrel_left_offset1_bunch1_3::calculate_funcs(), calib_barrel_left_offset1_bunch1_4::calculate_funcs(), calib_barrel_left_offset1_bunch2_3::calculate_funcs(), calib_barrel_left_offset1_bunch2_4::calculate_funcs(), calib_barrel_left_offset1_bunch3_4::calculate_funcs(), calib_barrel_left_offset2_bunch0_3::calculate_funcs(), calib_barrel_left_offset2_bunch0_4::calculate_funcs(), calib_barrel_left_offset2_bunch1_3::calculate_funcs(), calib_barrel_left_offset2_bunch1_4::calculate_funcs(), calib_barrel_left_offset2_bunch2_3::calculate_funcs(), calib_barrel_left_offset2_bunch2_4::calculate_funcs(), calib_barrel_left_offset2_bunch3_4::calculate_funcs(), calib_barrel_right_offset1_bunch0_3::calculate_funcs(), calib_barrel_right_offset1_bunch0_4::calculate_funcs(), calib_barrel_right_offset1_bunch1_3::calculate_funcs(), calib_barrel_right_offset1_bunch1_4::calculate_funcs(), calib_barrel_right_offset1_bunch2_3::calculate_funcs(), calib_barrel_right_offset1_bunch2_4::calculate_funcs(), calib_barrel_right_offset1_bunch3_4::calculate_funcs(), calib_barrel_right_offset2_bunch0_3::calculate_funcs(), calib_barrel_right_offset2_bunch0_4::calculate_funcs(), calib_barrel_right_offset2_bunch1_3::calculate_funcs(), calib_barrel_right_offset2_bunch1_4::calculate_funcs(), calib_barrel_right_offset2_bunch2_3::calculate_funcs(), calib_barrel_right_offset2_bunch2_4::calculate_funcs(), calib_barrel_right_offset2_bunch3_4::calculate_funcs(), TrkFitMaker::changeDefault(), check_two_sigma_violate(), KalFitTrack::chi2_next(), TTrackManager::closest(), collect(), PackedRawDataCnvSvc::connectOutput(), BesGeoTrack::Construct3DLine(), BesEvent::ConstructMdcTrackFromRec(), Cores(), Clock::count(), Minor2::create(), Minor3::create(), MucGeoMgr::CreateBakelite(), RecExtTrackCnv::DataObjectToTObject(), EvtBto2piCPiso::decay(), EvtBTo3piCP::decay(), EvtBTo4piCP::decay(), EvtBToKpipiCP::decay(), EvtIntervalDecayAmp< T >::decay(), EvtSSDCP::decay(), EvtSSSCP::decay(), EvtSSSCPpng::decay(), EvtSSSCPT::decay(), EvtSTSCP::decay(), EvtSVPCP::decay(), EvtSVSCP::decay(), EvtSVSCPiso::decay(), EvtSVSCPLH::decay(), EvtSVSNONCPEIGEN::decay(), EvtSVVCP::decay(), EvtSVVCPLH::decay(), EvtSVVNONCPEIGEN::decay(), TTrackManager::determineIP(), BesTofDigitizerEcV3::DirectPh(), TTrackManager::divide(), TTrackManager::divideByIp(), Alignment::docaHelixWireNewton(), TFastFinder::doit(), TPerfectFinder::doit(), EvtPHOTOS::doRadCorr(), draw_check_curve(), draw_illustration_hist(), draw_plots(), draw_separation(), drawmom(), TMDC::driftDistance(), endcap_con2root(), endcap_conv2root(), EmcRecCrystal::EndCapCheckout(), etf_conv2root(), TVacuumPol::Evalt(), EvtAmpA1(), EvtAmpA2(), CosmicGenerator::execute(), EsTimeAlg::execute(), TrkReco::execute(), f_cals(), f_opt(), f_real_work(), QCMCFilter::findD0Decay(), TTrackManager::finish(), TCircleFitter::fit(), TLineFitter::fit(), TRobustLineFitter::fit(), T3DLineFitter::fit(), TCosmicFitter::fit(), TRungeFitter::fit(), TLine0::fit2(), TMLine::fit2(), TLine0::fit2p(), TMLine::fit2p(), TMFitter::fitDone(), EvtGen::generateEvent(), EvtKKLambdaCFF::getbaryonff(), TrkFitMaker::getRep(), EvtbTosllAliFF::getScalarFF(), EvtbTosllBallFF::getScalarFF(), EvtHQET2FF::getscalarff(), EvtHQETFF::getscalarff(), EvtISGW2FF::getscalarff(), EvtSLBKPoleFF::getscalarff(), EvtSLPoleFF::getscalarff(), EvtSLBKPoleFF::gettensorff(), EvtSLPoleFF::gettensorff(), EvtISGW2FF::gettensorff(), xmlBase::Dom::getText(), EvtbTosllAliFF::getVectorFF(), EvtbTosllBallFF::getVectorFF(), EvtHQET2FF::getvectorff(), EvtHQETFF::getvectorff(), EvtISGW2FF::getvectorff(), EvtMelikhovFF::getvectorff(), EvtSLBKPoleFF::getvectorff(), EvtSLPoleFF::getvectorff(), EvtVubdGamma::getW1nodelta(), EvtVubdGamma::getW2nodelta(), EvtVubdGamma::getW3nodelta(), EvtVubdGamma::getW4nodelta(), EvtVubdGamma::getW5nodelta(), TTrackManager::goodTrack(), EvtHelSys::Helrotate(), G4HepMCInterface::HepMC2G4(), histgen(), histps(), eformat::helper::DateAndTime::human(), Minor5::I2D2stu(), Minor5::I2D2stui(), Minor5::I2D2stuij(), Minor5::I2D3stu(), Minor5::I2D4stu(), Minor5::I2D5stu(), Minor5::I2D6stu(), Minor5::I2Dstu(), Minor5::I2Dstui(), Minor5::I2stu(), Minor5::I3D2st(), Minor5::I3D2sti(), Minor5::I3D2stij(), Minor5::I3D3st(), Minor5::I3D3sti(), Minor5::I3D3stij(), Minor5::I3D3stijk(), Minor5::I3D4st(), Minor5::I3D5st(), Minor5::I3D6st(), Minor5::I3D7st(), Minor5::I3Dst(), Minor5::I3Dsti(), Minor5::I3st(), EvtCPUtil::incoherentMix(), MagneticFieldSvc::init_params(), EvtParticle::initDecay(), QCMCFilter::initialize(), TBuilder::initialLine(), TBuilder::initialLineOld(), InnerMost(), rdbModel::Rdb::insertLatest(), rdbModel::Rdb::insertRow(), KalFitTrack::insist(), EvtConExc::islgr(), eformat::helper::DateAndTime::iso8601(), ixtime_(), join(), EvtConExc::lgr(), Links(), FTSegment::linkStereoSegments(), EvtConExc::LLr(), main(), MajorLinks(), TTrackManager::makeTds(), TTrackManager::mask(), TTrackManager::maskBadHits(), TTrackManager::maskCurl(), TTrackManager::maskCurlHits(), TTrackManager::maskMultiHits(), TTrackManager::maskNormal(), TTrackManager::maskOut(), KalFitMaterial::mcs_angle(), RkFitMaterial::mcs_angle(), TTrackManager::merge(), std__map_int_int__dict::method_x11(), std__map_int_HepMC__GenVertexp_std__greater_int_s__dict::method_x18(), std__map_int_HepMC__GenParticlep__dict::method_x20(), KalFitTrack::ms(), KalFitTrack::msgasmdc(), ScheduledEvent::name(), BesMucNoise::NoiseSampling(), TrkBase::Predicates::isFitValid::operator()(), TrkBase::Predicates::isFitCurrent::operator()(), TrkBase::Predicates::isHotOnTrack::operator()(), TrkBase::Predicates::hasFitValid::operator()(), TrkBase::Predicates::hasFitCurrent::operator()(), TrkBase::Functors::takeAddress< T >::operator()(), BesError::operator*=(), operator+=(), BesVectorErr::operator-(), BesError::operator/=(), operator<<(), operator>>(), EvtCPUtil::OtherB(), OuterMost(), BesGeoTrack::Paint(), Gen_HEPEVT::position(), G4HepMCInterface::Print(), EvtPolInt::ratint(), BesTofDigitizerEcV3::ReadEffTree(), readTree(), TTrackManager::refit(), TrkFitMaker::repointHypo(), TrkRecoTrk::resetT0(), BesTView::RotateView(), TBuilder::salvage(), TBuilder0::salvage(), TTrackManager::salvageAssociateHits(), TTrackManager::saveMCTables(), TTrackManager::saveTables(), ScheduledEvent::schedule(), BesAngle::sector(), SeparateCores(), TofDataSet::setData(), TagFilterSvc::setDstDataType(), TagParser1::setDstDataType(), TagParser2::setDstDataType(), TagParser3::setDstDataType(), BesEvent::SetEvent(), BesEventHeader::SetEventTof(), TrkFitMaker::setFitNumber(), setregion(), TagFilterSvc::setTagData0(), TagParser1::setTagData0(), TagParser2::setTagData0(), TagParser3::setTagData0(), TagFilterSvc::setTagData1(), TagParser1::setTagData1(), TagParser2::setTagData1(), TagParser3::setTagData1(), TagFilterSvc::setTagData2(), TagParser1::setTagData2(), TagParser2::setTagData2(), TagParser3::setTagData2(), TagFilterSvc::setTagData3(), TagParser1::setTagData3(), TagParser2::setTagData3(), TagParser3::setTagData3(), TagFilterSvc::setTagData4(), TagParser1::setTagData4(), TagParser2::setTagData4(), TagParser3::setTagData4(), TagFilterSvc::setTagData5(), TagParser1::setTagData5(), TagParser2::setTagData5(), TagParser3::setTagData5(), TagFilterSvc::setTagData6(), TagParser1::setTagData6(), TagParser2::setTagData6(), TagParser3::setTagData6(), TagFilterSvc::setTagData7(), TagParser1::setTagData7(), TagParser2::setTagData7(), TagParser3::setTagData7(), TagFilterSvc::setTagData8(), TagParser1::setTagData8(), TagParser2::setTagData8(), TagParser3::setTagData8(), TagFilterSvc::setTagData9(), TagParser1::setTagData9(), TagParser2::setTagData9(), TagParser3::setTagData9(), BesTruthVertex::SetTime(), BesMucHit::SetTime(), BesEmcHit::SetTimeCrystal(), RecBTofCalHit::setTpred(), RecETofCalHit::setTpred(), HoughHitList::setType(), MinorBase::signM2ud(), MinorBase::signM3ud(), sim_conv2root(), sim_endcap_conv2root(), TTrackManager::sortBanksByPt(), TSegment::split(), TSegment0::split(), rdbModel::Rdb::supersedeRow(), FTFinder::t2x(), t_to_h(), testDateTime(), FTWire::time(), MdcHitOnTrack::timeAbsolute(), XmlRpc::XmlRpcValue::timeFromXml(), MdcHitOnTrack::timeResid(), XmlRpc::XmlRpcValue::timeToXml(), RecExtTrackCnv::TObjectToDataObject(), KalFitTrack::tof(), BesTofDigitizerBrV2::TofPmtRspns(), BesTofDigitizerEcV2::TofPmtRspns(), BesTofDigitizerEcV3::TofPmtRspns(), Track2Helix(), TrackDump(), TrackInformation(), TrackKinematics(), TrackLayerUsage(), TrackStatus(), TrackType(), KalFitTrack::trasan_id(), tryQuick(), TMDCWire::tsf(), KalFitTrack::type(), UniqueLinks(), TrkFitMaker::uniqueReps(), rdbModel::Rdb::updateRows(), TrkFundHit::usedOnTrack(), uxtime_(), rdbModel::Assertion::Operator::verify(), vtxFit(), XmlRpc::XmlRpcValue::write(), writeTree(), T3DLine::x(), XtMdcCalib::xtFun(), and xtFun().

t_save

TTree* t_save = new TTree("t_save", "tree to save entropy")

Definition at line 37 of file binning.cxx.

Referenced by f_cals(), and main().

tot_num_group

int tot_num_group

Definition at line 39 of file binning.cxx.

Referenced by f_cals(), f_group(), and main().

x_group

double x_group[2000]

Definition at line 40 of file binning.cxx.

Referenced by f_group(), and main().