#include <iostream>

#include "createTrees.h"
#include "TFile.h"

void processTrees()
{
  TreeBuilder treeB;
  treeB.buildTree("bla");

  TTree *tree1 = treeB.getTree();

  int a0,b0;
  tree1->SetBranchAddress("a",&a0);
  tree1->SetBranchAddress("b",&b0);

  TreeBuilder treeB2;
  treeB2.buildTree("bla2");


  TTree *tree2 = treeB2.getTree();
  int a,b,c;
  
  TFile *f = new TFile("bla2.root","RECREATE");

  tree2->SetBranchAddress("a",&a);
  tree2->SetBranchAddress("b",&b);
  TBranch *b_C = tree2->Branch("c",&c,"c/I");
  
  
  for(int i = 0; i < tree1->GetEntriesFast(); ++i) {
    tree1->GetEntry(i);
    if(i < tree2->GetEntriesFast()) {
      tree2->GetEntry(i);
    } else {
      return;
    }
    c = a + b + a0 + b0;
    //tree2->Fill();
    b_C->Fill();

    
  }
  
  tree2->Write("",TObject::kOverwrite);

}