71216c3bba5f48b74db93dce9673d8a44be3420f
[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / Fit / AliLatexTable.cxx
1 // ------------------------------------------------------------------
2 //
3 //                           AliLatexTable
4 //
5 // This class is used to produce tables using latex sintax. It can
6 // than print the table as Latex source code (to be pasted on a paper)
7 // or ASCII.
8 //
9 // The basic idea is that you add columns one after the other with the
10 // SetNextCol method and than you insert this row. The SetNextCol
11 // method comes in different flavours to alow the insertion of
12 // different type of values (numbers with or without errors,
13 // strings...).
14 //
15 // TODO:
16 // 1. Make the class drawable
17 // 2. Implement vertical lines in ascii print
18 // 3. Print output in HTML format
19 //
20 // Author: Michele Floris, CERN
21 // ------------------------------------------------------------------
22
23 #include "AliLatexTable.h"
24 #include "TString.h"
25 #include "TRegexp.h"
26 #include "TObjString.h"
27 #include "TObjArray.h"
28 #include <stdarg.h>
29 #include "snprintf.h"
30 #include "Varargs.h"
31 #include "TPRegexp.h"
32 #include "TMath.h"
33 #include <iostream>
34 #include <fstream>
35 #include "AliLog.h"
36
37 using namespace std;
38
39 ClassImp(AliLatexTable)
40
41 AliLatexTable::AliLatexTable() : fNcol(0), fFormat(""), fRows(0), fCols(0), fNcolReady(0){
42   // default constructor
43   fNcol = 1;
44   fFormat = "c";
45   fNcolReady = 0;
46   fRows = new TObjArray;
47   fCols = new TObjArray;
48
49   fRows->SetOwner();
50   fCols->SetOwner();
51
52 }
53
54 AliLatexTable::AliLatexTable(Int_t ncol, TString format) : fNcol(0), fFormat(""), fRows(0), fCols(0), fNcolReady(0){
55   // constructor, specify number of cols 
56   fNcol = ncol;
57   fFormat = format;
58   fNcolReady = 0;
59   fRows = new TObjArray;
60   fCols = new TObjArray;
61
62   fRows->SetOwner();
63   fCols->SetOwner();
64 }
65
66
67 AliLatexTable::~AliLatexTable() {
68
69   // dtor
70   if (fRows) delete fRows;
71   if (fCols) delete fCols;
72
73 }
74
75 void AliLatexTable::SetNextCol(Int_t val){
76   // Set next column in current row - integer
77   char col[200];
78   sprintf(col, " %d ", val);
79   SetNextCol(col);
80
81
82
83 void AliLatexTable::SetNextCol(Int_t val, Int_t err){
84
85   // Set next column in current row - int +- int
86   char col[200];
87   sprintf(col, " $%d \\pm %d$ ", val, err);
88   SetNextCol(col);
89
90
91
92 void AliLatexTable::SetNextCol(Double_t val, Int_t scientificNotation, Bool_t rounding){
93
94   // Set next column in current row - double, specify resolution
95
96   char col[200];
97   if(rounding) {
98     if(scientificNotation >= 0) {
99       char format[100];
100
101       //    cout << format << endl;    
102       Double_t mantissa, exp;
103       GetMantissaAndExpBase10(val, mantissa, exp);
104     
105       if (exp == 0) {
106         sprintf(format, " $%%%d.%df $", scientificNotation, scientificNotation);
107         sprintf(col, format, mantissa);      
108       }
109       else  {
110         sprintf(format, " $%%%d.%df \\cdot 10^{%%0.0f} $", scientificNotation, scientificNotation);
111         sprintf(col, format, mantissa, exp);
112       }
113
114
115     
116
117       SetNextCol(col);
118
119     } else {
120       char format[100];
121       sprintf(format, " $%%%d.%df $", -scientificNotation,-scientificNotation);    
122       sprintf(col, format , TMath::Nint(val*TMath::Power(10,-scientificNotation))/TMath::Power(10,-scientificNotation));
123       SetNextCol(col);
124     } 
125   }else {
126     sprintf(col, " %f ", val);
127     SetNextCol(col);
128   }
129
130
131 void AliLatexTable::SetNextCol(Double_t val, Double_t err, Int_t scientificNotation, Bool_t rounding){
132
133   // Set next column in current row - double +- double, specify resolution
134
135   // scientific notation is used to determine number of
136   // digits in error 
137
138   //if it is > 0 exp notation is used 
139
140
141   char col[200];
142   if(rounding) {
143     if(scientificNotation >=0 ) {
144       
145       Double_t mantissa, exp;
146       GetMantissaAndExpBase10(val, mantissa, exp);
147       
148       Double_t mantissa_err, exp_err;
149       GetMantissaAndExpBase10(err, mantissa_err, exp_err);
150
151       Int_t nSigDigits =TMath::Nint(exp - exp_err);
152       if(scientificNotation != 0) nSigDigits =  nSigDigits + scientificNotation - 1;
153
154
155       char format[100];
156       if (exp == 0) {
157         sprintf(format, " $%%%d.%df \\pm %%%d.%df $", nSigDigits, nSigDigits, nSigDigits, nSigDigits);
158         sprintf(col, format, mantissa, mantissa_err/TMath::Power(10,exp - exp_err));
159       }
160       else  {
161         sprintf(format, " $%%%d.%df \\pm %%%d.%df \\cdot 10^{%%0.0f}$", nSigDigits, nSigDigits, nSigDigits, nSigDigits);
162         sprintf(col, format, mantissa, mantissa_err/TMath::Power(10,exp - exp_err), exp);
163       }
164
165
166       //cout << format << endl;
167     
168       SetNextCol(col);
169
170     } else  {
171       char format[100];
172       sprintf(format, " $%%%d.%df \\pm %%%d.%df $", -scientificNotation,-scientificNotation,-scientificNotation,-scientificNotation);    
173       sprintf(col, format , TMath::Nint(val*TMath::Power(10,-scientificNotation))/TMath::Power(10,-scientificNotation), TMath::Nint(err*TMath::Power(10,-scientificNotation))/TMath::Power(10,-scientificNotation));
174       SetNextCol(col);
175     }
176   }
177   else {
178     sprintf(col, " $%f \\pm %f$ ", val, err);
179     SetNextCol(col);
180   }
181
182
183 void AliLatexTable::SetNextCol(Double_t val, Double_t err, Double_t errSyst, Int_t scientificNotation, Bool_t rounding){
184
185   // Set next column in current row - double +- double +- double, specify resolution
186
187   // if scientific notation is != -1 is used to determine number of
188   // digits in error 
189
190   //if it is > 0 exp notation is used 
191
192   //if it is < -1 number is truncated to the number of digits after
193   //point dictated by scientificNotation,
194
195   char col[200];
196   if (rounding) {
197     if(scientificNotation >=0 ) {
198
199       Double_t mantissa, exp;
200       GetMantissaAndExpBase10(val, mantissa, exp);
201
202       Double_t mantissa_err, exp_err;
203       GetMantissaAndExpBase10(err, mantissa_err, exp_err);
204
205       Double_t mantissa_errsyst, exp_errsyst;
206       GetMantissaAndExpBase10(errSyst, mantissa_errsyst, exp_errsyst);
207
208       Int_t nSigDigits =TMath::Nint(exp - exp_err);
209       if(scientificNotation != 0) nSigDigits =  nSigDigits + scientificNotation - 1;
210
211
212       char format[100];
213       if (exp == 0) {
214         sprintf(format, " $%%%d.%df \\pm %%%d.%df \\pm %%%d.%df $", 
215                 nSigDigits, nSigDigits, nSigDigits, nSigDigits, nSigDigits, nSigDigits);
216         sprintf(col, format, mantissa, 
217                 mantissa_err/TMath::Power(10,exp - exp_err), 
218                 mantissa_errsyst/TMath::Power(10,exp - exp_errsyst));
219       }
220       else  {
221         sprintf(format, " $%%%d.%df \\pm %%%d.%df  \\pm %%%d.%df \\cdot 10^{%%0.0f}$", 
222                 nSigDigits, nSigDigits, nSigDigits, nSigDigits, nSigDigits, nSigDigits);
223         sprintf(col, format, mantissa, 
224                 mantissa_err/TMath::Power(10,exp - exp_err), 
225                 mantissa_errsyst/TMath::Power(10,exp - exp_errsyst),
226                 exp);
227       }
228
229
230       //cout << format << endl;
231     
232       SetNextCol(col);
233
234     } else  {
235       char format[100];
236       sprintf(format, " $%%%d.%df \\pm %%%d.%df \\pm %%%d.%df $", -scientificNotation,-scientificNotation,-scientificNotation,-scientificNotation, -scientificNotation,-scientificNotation);    
237       sprintf(col, format ,TMath::Nint(val*TMath::Power(10,-scientificNotation))/TMath::Power(10,-scientificNotation), TMath::Nint(err*TMath::Power(10,-scientificNotation))/TMath::Power(10,-scientificNotation), TMath::Nint(errSyst*TMath::Power(10,-scientificNotation))/TMath::Power(10,-scientificNotation));
238       SetNextCol(col);
239     } 
240   }
241   else {
242     sprintf(col, " $%f \\pm %f  \\pm %f$ ", val, err, errSyst);
243     SetNextCol(col);
244   }
245
246
247 // void AliLatexTable::SetNextColPrintf(const char *va_(fmt), ...){
248
249
250 //   const Int_t buf_size = 1024;  // hope it's long enough
251 //   char colBuffer[buf_size];
252
253 //   va_list ap;
254 //   va_start(ap, va_(fmt));
255 //   Int_t n = vsnprintf(colBuffer, buf_size, fmt, ap);
256
257 //   if (n == -1 || n >= buf_size) {
258 //     Error("SetNextCol", "vsnprintf error!");
259 //   }
260
261 //   va_end(ap);
262      
263 // //   for(int iobj=0; iobj<num; iobj++){   //Loop until all objects are added
264 // //     TH1 * obj = (TH1 *) va_arg(arguments, void *);
265 // //     returnValue = returnValue && AddObject(obj,(char*)fOptMany.Data(), EColor(iobj+1));
266 // //   }
267 //   SetNextCol(colBuffer);
268 // }
269
270
271 void AliLatexTable::SetNextCol(TString val){
272
273   // Set next column in current row - tstring
274
275   fCols->Add(new TObjString(val));
276   fNcolReady++;
277
278
279
280 void AliLatexTable::InsertCustomRow(TString row){
281   // insert a full row from string. Make sure you have all the columns
282   fRows->Add(new TObjString(row));
283 }
284
285
286
287 void AliLatexTable::InsertRow(){
288
289   // Insert the row, based on all the individual columns
290
291   if ( fNcolReady != fNcol) {
292     Warning("InsertRow", "Wrong number of cols: %d (!= %d)", fNcolReady, fNcol); 
293   }
294
295   TString row = "";
296   for(Int_t icol = 0; icol < fNcol; icol++){
297     row = row + ((TObjString*) fCols->At(icol))->String();
298     if(icol != (fNcol-1)) row = row + " & ";
299   }
300   row+="\\\\";
301
302   fRows->Add(new TObjString(row));
303
304   fNcolReady = 0;
305   fCols->Clear();
306
307 }
308
309 void AliLatexTable::InsertHline(){
310
311   // insert an horizontal line
312   fRows->Add(new TObjString("\\hline"));
313   
314
315 }
316
317 Int_t * AliLatexTable::GetColWidths() {
318
319   // Compute the width of columns, for nice ascii printing
320
321   Int_t * col_widths= new Int_t[fNcol];
322   Int_t nrow = fRows->GetEntriesFast();
323
324   for(Int_t icol = 0; icol < fNcol; icol++){
325     col_widths[icol] = 0;
326   }
327   
328
329   for(Int_t irow = 0; irow < nrow; irow++){
330     TString row(((TObjString*) fRows->At(irow))->String());
331     if(row.Contains("\\hline"))continue;
332     
333     StripLatex(row, "ASCII");
334
335     TObjArray * cols = row.Tokenize("&");
336     if(cols->GetEntries() != fNcol) {
337       Error("GetColWidths", "Wrong number of cols in row %s: %d - %d", row.Data(), cols->GetEntries(), fNcol);      
338     }
339     for(Int_t icol = 0; icol < fNcol; icol++){
340       Int_t w = ((TObjString *) cols->At(icol))->String().Length();
341       if (w>col_widths[icol]) col_widths[icol] = w;
342     }
343     
344   }
345
346   return col_widths;
347 }
348
349 void AliLatexTable::PrintTable(Option_t * opt){
350
351   // Print the table on screen. You can specify the format with opt.
352   // Currently supported:
353   // "" -> LaTeX
354   // "ASCII" -> plain ASCII
355   // "HTML"  -> HTML, to be improved
356   // "CSV"   -> skips hline, usefult for importing in excell 
357
358   if(TString(opt) == "ASCII" || TString(opt)=="HTML" ||  TString(opt)=="CSV") {
359     
360     Int_t nrow = fRows->GetEntriesFast();
361
362     Int_t * col_widths = GetColWidths();
363
364     Int_t total_lenght = 0;
365     for(Int_t icol = 0; icol < fNcol; icol++) total_lenght = total_lenght + col_widths[icol] + 2 ;
366     
367
368     for(Int_t irow = 0; irow < nrow; irow++){
369       TString row = ((TObjString*) fRows->At(irow))->String();
370       if (row.Contains("\\hline")){     
371         if (TString(opt)!="CSV") {
372           for(Int_t il = 0; il < total_lenght; il++) printf("-");
373           printf("\n");   
374         }
375         continue;
376       }
377       StripLatex(row, opt);
378       TObjArray * cols = row.Tokenize("&");
379       for(Int_t icol = 0; icol < fNcol; icol++){
380         const char * colstr = ((TObjString *) cols->At(icol))->String().Data();
381         char format [200];
382         if (TString(opt)!="CSV") {
383           sprintf(format, "%%%ds", col_widths[icol] + 2);       
384         } else {
385           sprintf(format, "%%s");       
386         }
387         printf(format, colstr);
388         if (TString(opt)=="CSV") printf(", ");
389
390       }
391       printf ("\n");
392     }
393     
394
395     return;
396   }
397   
398
399   cout << "\\begin{tabular}{"<<fFormat<<"}"<<endl;
400
401   Int_t nrow = fRows->GetEntriesFast();
402
403   for(Int_t irow = 0; irow < nrow; irow++){
404     cout << ((TObjString*) fRows->At(irow))->String() << endl;
405   }
406   
407   cout << "\\end{tabular}" << endl;
408
409
410 }
411
412 // void AliLatexTable::ParseExponent(TString &expo){
413 // //    TString parseExponent = col;
414 //   TRegexp re = "e[+-][0-9][0-9]";
415 //   //  cout << col << endl;
416   
417 //   if(expo.Contains(re)){
418 //     Int_t index = expo.Index(re);
419 //     TString replacement = "\\cdot 10^{";
420 //     replacement = replacement + expo(index+1, 3) +"}";
421 //     //    cout << replacement <<" --- "<< endl;
422 //     replacement.ReplaceAll("+","");
423 // //     cout << "B: " << expo << endl;      
424 // //     cout << "RE " << expo(re) << endl;
425     
426 //     expo.ReplaceAll(expo(re), replacement);
427 //     //    cout << "A: " << expo << endl;      
428 //     // recursion to parse all exponents
429 //     if (expo.Contains(re)) ParseExponent(expo);
430 //   }
431 //   else Warning("", "Error parsing exponent");
432 // }
433
434 void AliLatexTable::GetMantissaAndExpBase10(Double_t num, Double_t &man, Double_t &exp) {
435
436   // Helper used to get mantissa and exponent
437
438   exp = TMath::Floor(TMath::Log10(TMath::Abs(num)));  
439   man = num / TMath::Power(10, exp);
440
441 //   cout << "" << endl;
442 //   cout << num << " = " << man << " x10^{"<<exp<<"} "   << endl;
443   
444
445 }
446
447 void AliLatexTable::StripLatex(TString &text, TString format) {
448
449   // Strip latex away for ascii and html printing. Replaces latex
450   // command with corresponding text/tags
451
452   text.ReplaceAll("\\cdot", "x");
453   text.ReplaceAll("\\pm", "+-");
454   text.ReplaceAll("$", "");
455   if (format == "ASCII") {
456     text.ReplaceAll("\\right>", ">");
457     text.ReplaceAll("\\left<", "<");
458     text.ReplaceAll("\\rangle", ">");
459     text.ReplaceAll("\\langle", "<");
460   } else if (format == "HTML") {
461     text.ReplaceAll("\\right>", "&rang;");
462     text.ReplaceAll("\\left<",  "&lang;");
463     text.ReplaceAll("\\rangle", "&rang;");
464     text.ReplaceAll("\\langle", "&lang;");
465   } 
466   if(text.Contains("multicolumn")) {
467     //    cout << "col " << text.Data() << endl;
468     // in case of multicol span, replace first column with content and
469     // add n empty cols
470     TObjArray * array = TPRegexp("multicolumn\\{(.*)\\}\\{.*\\}\\{(.*)\\}").MatchS(text); // Added double \\ because gcc 4 triggers hard error on unknown escape sequence. Hope it still works...
471     const TString content = ((TObjString *)array->At(2))->GetString();
472     Int_t nspan   = ((TObjString *)array->At(1))->GetString().Atoi();
473     text = content;
474 //     cout << "ns " << nspan <<  ((TObjString *)array->At(1))->GetString() << endl;
475 //     cout << "t " << text << endl;
476     
477     for(Int_t ispan = 1; ispan < nspan; ispan++){
478       text+=" & ";
479     }
480     //    cout << "t " << text << endl;
481     
482   
483   }
484
485   text.ReplaceAll("\\mathrm", "");
486
487   text.ReplaceAll("\\", "");
488   text.Strip(TString::EStripType(1), ' ');
489
490 }
491
492 void AliLatexTable::LoadTeXFromFileAndPrintASCII(const char * filename) {
493
494   // opens a file containing only a latex table and prints it on screen as ASCII
495
496   ifstream file (filename);
497   if (!file.is_open()) {
498     AliError(Form("Cannot open file %s", filename));
499   }
500   TString line;
501   while (line.ReadLine(file)) {
502     if (line.Contains("begin") && line.Contains("tabular")) {
503       // We need to get and parse the format
504       //      TPRegexp re("\\begin\\{tabular\\}\\{([^\\}]*)\\}");
505       TPRegexp re(".*begin{tabular}{(.*)}");
506       TObjArray * arr = re.MatchS(line);
507       if (arr->GetLast() > 0){ 
508         //      cout << "Size: " << arr->GetSize() << " " << arr->GetLast() << endl;
509         
510         TString subStr = ((TObjString *)arr->At(1))->GetString();
511         subStr.ReplaceAll("|","");
512         subStr.ReplaceAll(" ","");
513         subStr.ReplaceAll("\t","");
514         //      subStr.ReplaceAll(" ","");      
515         //      cout << subStr.Data() << " " << subStr.Length()<< endl;
516         fNcol = subStr.Length();
517         delete arr;
518       }
519     }
520
521     // Skip stuff we don't parse
522     if (line.Contains("begin")) continue;
523     if (line.Contains("end")) continue;
524     if (line.Contains("tabular")) continue;
525     // add line
526     InsertCustomRow(line.Data());
527     
528   }
529   PrintTable("ASCII");
530 }