WriteInMiniTree3.C

Go to the documentation of this file.

/**************************************************************************
 * Copyright(c) 1998, Agata Experiment, All rights reserved. *
 *                                                                        *
 * Author: The Agata Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/


//
//
#if !defined(__CINT__) || defined(__MAKECINT__)

#include <Riostream.h>
#include "TFile.h"
#include "TMath.h"
#include "TRandom.h"
#include "TTree.h"
#include "TH2F.h"
#include "TObject.h"

#endif

Int_t dim = 256;
Int_t nDiv = 8;
Int_t nTree = 512;

//Int_t dim = 1024;
//Int_t nDiv = 2;
//Int_t nTree = 8;

Int_t GetMiniCubeId(UShort_t* xEnergies, UShort_t* xCube)
{
//   Int_t index[3];
//   TMath::Sort(3, xEnergies, index, false);
   
//   Int_t i = xEnergies[index[0]]/dim;
//   Int_t j = xEnergies[index[1]]/dim;
//   Int_t k = xEnergies[index[2]]/dim;
   
   Int_t i = xEnergies[0]/dim;
   Int_t j = xEnergies[1]/dim;
   Int_t k = xEnergies[2]/dim;
   
   Int_t idx = i*nDiv*nDiv + j*nDiv + k;
   
//   xCube[0] = (xEnergies[index[0]] % dim);
//   xCube[1] = (xEnergies[index[1]] % dim);
//   xCube[2] = (xEnergies[index[2]] % dim);
   
   xCube[0] = (xEnergies[0] % dim);
   xCube[1] = (xEnergies[1] % dim);
   xCube[2] = (xEnergies[2] % dim);

   return idx;
}

void WriteInMiniTree3()
{
  
   UShort_t x0[3];

   UShort_t e[nTree][3];
   UShort_t x[3];
   Float_t Xn[3];
   
   TFile* fileIn = new TFile("CubeTreeR_22_0_67.root ", "READ");
   TTree* tree = (TTree*)fileIn->Get("treecube");
   
   TBranch* b = tree->GetBranch("e");
   if (b == 0x0) {
          printf("Branch not find in tree\n");
          return;
   }
   
   b->SetAddress(&Xn);  
   
   TFile* fileOut = new TFile("test8.root", "RECREATE");
   fileOut->cd();
 
   TTree* oak[nTree];
   for (Int_t i = 0; i < nTree; ++i) {
          oak[i] = new TTree(Form("oak%d", i), "3D tree");
          oak[i]->Branch("e", &e[i], "e[3]/s");
   }
   
   for (Int_t i = 0; i < 1000000; ++i) {
          tree->GetEntry(i);
          printf("%d\r", i);
          
          x[0] = Short_t(Xn[0]);
          x[1] = Short_t(Xn[1]);
          x[2] = Short_t(Xn[2]);
          Int_t idx = GetMiniCubeId(x, x0);
          e[idx][0] = x0[0];
          e[idx][1] = x0[1];
          e[idx][2] = x0[2];
          oak[idx]->Fill();
   }
   
   for (Int_t i = 0; i < nTree; ++i) 
          oak[i]->Write();
   
   printf("\n");
  fileOut->Close();
}