sizetree.C

Go to the documentation of this file.

#include "TFile.h"
#include "TTree.h"
#include "TBranch.h"
#include "TString.h"
#include "TKey.h"

#include <iostream>

int sizetree(TString fname)
{
        int totsize = 0;
        TString name[1000];
        int size[1000];
        int n[1000];
        int br[1000];
        int cnt=0;
        
        TFile *f=new TFile(fname,"READ");
        if (!f->IsZombie())
        {               
                TKey *key;
                TIter next(f->GetListOfKeys());
                while ( (key = (TKey*)next()) )
                {
                        TObject *obj = key->ReadObj();
                        
                        if (!obj->InheritsFrom("TTree")) continue;
                        
                        name[cnt] = obj->GetName();
                        TTree* t = (TTree*) obj; 
                        
                        n[cnt]  = t->GetEntries();
                        br[cnt] = t->GetNbranches();
                        int s = 0;
                                                
                        TObjArray* branches = t->GetListOfBranches();
                        for(int i=0; i<=branches->GetLast(); ++i)
                        {
                                TBranch* branch = (TBranch*)branches->UncheckedAt(i);
                                s += branch->GetZipBytes();
                        }
                        size[cnt++]=s;
                        
                        totsize += s;
                }
        }
        f->Close();
        
        cout <<cnt<<endl;
        
        int ar[1000];
        for (int i=0;i<cnt;++i) ar[i]=i;
        
        for (int i=cnt;i>1;--i)
        {
                for (int j=0;j<i-1;++j) 
                {
                        if (size[ar[j]]<size[ar[j+1]])
                        {
                                int tmp=ar[j];ar[j]=ar[j+1];ar[j+1]=tmp;
                        }
                }
        }
        
        for (int i=0;i<cnt;++i)
                printf("%2d. %10s : N = %8d, BR = %5d, N*BR = %7dk, Size = %7dk\n", 
                        ar[i],name[ar[i]].Data(), n[ar[i]], br[ar[i]], (n[ar[i]]*br[ar[i]])/1000, size[ar[i]]/1000);
        
        cout <<"total size: "<<totsize/1000<<"k"<<endl;
  return 0;
}