saveViews macro fixed. Some macros are now being compiled before using to fix issues...

[u/mrichter/AliRoot.git] / EVE / macros / saveViews.C

/**************************************************************************
 * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
 * full copyright notice.                                                 *
 **************************************************************************/
// Author: Mihai Niculescu 2013


/*
 * This script creates a collage containing all OpenGL views from a running AliEve
 *
 * Given Collage size (width, height), the size for all OpenGL
 * views are computed using the same aspect ratio (width/height) as the main 3D View
 */

#include <TASImage.h>
#include <TGLViewer.h>
#include <TEveViewer.h>
#include <TEveManager.h>
#include <TMath.h>
#include <TSystem.h>
#include <TTimeStamp.h>
#include <TEveElement.h>
#include <TIterator.h>
#include <TList.h>
#include <TROOT.h>

#include <STEER/ESD/AliESDEvent.h>
#include <STEER/ESD/AliESDRun.h>
#include <STEER/STEER/AliGRPObject.h>
#include <STEER/CDB/AliCDBEntry.h>
#include <STEER/CDB/AliCDBManager.h>
#include <RAW/RAWDatarec/AliRawReader.h>
#include <RAW/RAWDatabase/AliRawEventHeaderBase.h>
#include <EVE/EveBase/AliEveEventManager.h>

TString getEventInfo();

/***********Save all OpenGL views into one picture
	compositeImgFileName - save final image to this file
 showLiveBar - whether to show the LIVE bar, useful when not online (using offline)
	width - of the collage image
	height -of the collage image
 */
void saveViews(const char* compositeImgFileName="views.png", Bool_t showLiveBar=kTRUE, Int_t width = 1440, Int_t height= 900)
{
    Info("saveViews.C", "saving views to [%s]", compositeImgFileName);
    
    Int_t heightInfoBar = 65; // hold height of the Information bar
    
    TASImage* compositeImg=0; // this holds the final image
    TASImage* tempImg=0; // temporary used for loading images
    
    TEveViewerList* viewers = gEve->GetViewers();
    Int_t Nviewers = viewers->NumChildren()-2; // remark: 3D view is counted twice
    
    compositeImg = new TASImage(width, height);
    
    // 3D View size
    Int_t width3DView = TMath::FloorNint((float)Nviewers*width/(float)(Nviewers+1)); // the width of the 3D view
    Int_t height3DView= height-heightInfoBar; // the height of the 3D view
    Float_t aspectRatio = (float)width3DView/(float)height3DView; // 3D View aspect ratio
    
    // Children View Size
    Int_t heightChildView = TMath::FloorNint((float)height3DView/(float)Nviewers);
    Int_t widthChildView  = TMath::FloorNint(aspectRatio*heightChildView); // has the same aspect ratio as the 3D view
    
    int index=0; // iteration counter
    int x = width3DView; // x position of the child view
    int y = 0;// y position of the child view
    TString viewFilename; // save view to this file
    
    for(TEveElement::List_i i = (++viewers->BeginChildren()); i != viewers->EndChildren(); i++)
    { // NB: this skips the first children (first 3D View)
        TEveViewer* view = ((TEveViewer*)*i);
        viewFilename = Form("view-%d.png", index);
        
        // Save OpenGL views in files
        if(index==0){
            view->GetGLViewer()->SavePictureUsingFBO(viewFilename, width3DView, height3DView);
        }
        else {
            view->GetGLViewer()->SavePictureUsingFBO(viewFilename, widthChildView, heightChildView);
        }
        
        tempImg = new TASImage(viewFilename);
        
        // copy view image in the composite image
        if(index==0){
            tempImg->CopyArea(compositeImg, 0,0, width3DView, height3DView);
        }
        else {
            tempImg->CopyArea(compositeImg, 0,0, widthChildView, heightChildView, x,y);
            
            // draw a border around child views
            compositeImg->DrawRectangle(x,y, widthChildView, heightChildView, "#C0C0C0");
        }
        
        /*
         final touches inside loop
         */
        delete tempImg;
        if(index>0) // skip 3D View
            y+=heightChildView;
        
        index++;
    }
    
    // Create a glow (bloom) effect
    tempImg = (TASImage*)compositeImg->Clone("tempImg");
    tempImg->Blur(10.0,10.0);
    compositeImg->Merge(tempImg, "lighten");
    delete tempImg; tempImg = 0;
    
    
    // show LIVE bar
    if(showLiveBar){
        TTimeStamp ts;
        TString tNow = ts.AsString("s"); // display date & time
        
        compositeImg->Gradient( 90, "#EAEAEA #D2D2D2 #FFFFFF", 0, 75, 0, 239, 95);
        compositeImg->Gradient( 90, "#D6D6D6 #242424 #000000", 0, 155, 60, 152, 26);
        compositeImg->BeginPaint();
        compositeImg->DrawRectangle(50,0, 264, 94);
        compositeImg->DrawText(162, 6, "LIVE", 70, "#FF2D00", "FreeSansBold.otf");
        compositeImg->DrawText(162, 65, tNow, 16, "#FFFFFF", "arial.ttf");
        compositeImg->EndPaint();
        //include ALICE Logo
        tempImg = new TASImage( Form("%s/picts/2012-Jul-04-4_Color_Logo_small_CB.png", gSystem->Getenv("ALICE_ROOT")) );
        tempImg->Scale(64,86);
        //tempImg->CopyArea(compositeImg, 0,0, 236, 319, 59, 4);
        compositeImg->Merge(tempImg, "alphablend", 82, 4);
        delete tempImg; tempImg = 0;
    }
    
    // show Information bar
    TString stringInfo;
    stringInfo = getEventInfo();
    compositeImg->Gradient( 90, "#1B58BF #1D5CDF #0194FF", 0, 0, height-heightInfoBar, width, heightInfoBar);
    compositeImg->BeginPaint();
    compositeImg->DrawText(10, height-heightInfoBar+15, stringInfo, 28, "#FFFFFF", "FreeSansBold.otf");
    compositeImg->EndPaint();
    
    
    // write composite image to disk
    compositeImg->CopyArea(compositeImg, 0,0, width, height);
    compositeImg->WriteImage(compositeImgFileName);
    
    delete compositeImg;
    
    return;
}

// This function retrieves a string containing some information regarding the current event
TString getEventInfo()
{
    // For general public please show as less or technical information as possible
    
    TString rawInfo, esdInfo;
    
    if (!AliEveEventManager::HasRawReader())
    {
        rawInfo = "";
    }
    else
    {
        AliRawReader* rawReader = AliEveEventManager::AssertRawReader();
        if(!rawReader) return "";
        rawInfo.Form("Run: %d  Event#: %d (%s)",
                     rawReader->GetRunNumber(),
                     AliEveEventManager::CurrentEventId(),
                     AliRawEventHeaderBase::GetTypeName(rawReader->GetType())
                     );
        
        return rawInfo;
    }
    
    if (!AliEveEventManager::HasESD())
    {
        esdInfo = "";
    }
    else
    {
        AliESDEvent* esd =  AliEveEventManager::AssertESD();
        
        esdInfo.Form("Colliding: %s Run: %d  Event: %d (%s)",
                     esd->GetESDRun()->GetBeamType(),
                     esd->GetRunNumber(),
                     AliEveEventManager::CurrentEventId(),
                     "PHYSICS_EVENT"
                     /*AliRawEventHeaderBase::GetTypeName(esd->GetEventType())*/
                     );
    }
    
    return esdInfo;
}