modification(by Levente) to solve the problem in the QA mentioned in the bug report...

[u/mrichter/AliRoot.git] / HLT / EMCAL / AliHLTEMCALRawHistoMakerComponent.cxx

/**************************************************************************
 * This file is property of and copyright by the ALICE HLT Project        *
 * All rights reserved.                                                   *
 * INFN, Laboratori Nazionali di Frascati                                 *
 * Primary Authors: Federico Ronchetti                                    *
 *                                                                        *
 * 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.                  *
 **************************************************************************/

#include "AliHLTEMCALRawHistoMakerComponent.h"
#include "AliHLTEMCALRawHistoMaker.h"

#include "AliHLTCaloChannelDataHeaderStruct.h"
#include "AliHLTCaloChannelDataStruct.h"

// root stuff
#include "TFile.h"
#include "TProfile2D.h"
#include "TH1F.h"
#include "TH2F.h"
#include <sys/stat.h>
#include <sys/types.h>

#include "AliCDBEntry.h"
#include "AliCDBManager.h"
#include "AliCDBPath.h"
/** 
 * @file   AliHLTEMCALRawHistoMakerComponent.cxx
 * @author Federico Ronchetti
 * @date   
 * @brief  A component to pusk back histograms for EMCAL HLT
 */

//FIXME
AliHLTEMCALRawHistoMakerComponent gAliHLTEMCALRawHistoMakerComponent;

AliHLTEMCALRawHistoMakerComponent::AliHLTEMCALRawHistoMakerComponent() :
				  AliHLTCaloProcessor(),
				  fRawHistoMakerPtr(0),
				  fPushFraction(10),
				  fLocalEventCount(0),
				  fRootFileName("histo_local_dump.root")
{
	//see header file for documentation
}


AliHLTEMCALRawHistoMakerComponent::~AliHLTEMCALRawHistoMakerComponent()
{
	//see header file for documentation
}

int 
AliHLTEMCALRawHistoMakerComponent::Deinit()
{ 
	//see header file for documentation
	if(fRawHistoMakerPtr)
	{
		delete fRawHistoMakerPtr;
		fRawHistoMakerPtr = 0;
	}

	return 0;
}

const char*
AliHLTEMCALRawHistoMakerComponent::GetComponentID()
{
	//see header file for documentation
	return "EMCALRawHistoMaker";
}


void
AliHLTEMCALRawHistoMakerComponent::GetInputDataTypes(vector<AliHLTComponentDataType>& list)
{ 
	//see header file for documentation
	list.clear();
	list.push_back(AliHLTEMCALDefinitions::fgkChannelDataType);
}

AliHLTComponentDataType 
AliHLTEMCALRawHistoMakerComponent::GetOutputDataType()
{
	//see header file for documentation
	return kAliHLTDataTypeHistogram | kAliHLTDataOriginEMCAL;
}

void 
AliHLTEMCALRawHistoMakerComponent::GetOutputDataSize(unsigned long& constBase, double& inputMultiplier)
{
	//see header file for documentation
	constBase = 0;
	// to be reviewed later
	inputMultiplier = 100;
}

int 
AliHLTEMCALRawHistoMakerComponent::DoEvent(const AliHLTComponentEventData& evtData, const AliHLTComponentBlockData* blocks,
		AliHLTComponentTriggerData& /*trigData*/, AliHLTUInt8_t* outputPtr, AliHLTUInt32_t& size,
		std::vector<AliHLTComponentBlockData>& outputBlocks)
{
	//see header file for documentation
	UInt_t mysize           = 0;
	Int_t ret               = 0;


	AliHLTUInt8_t* outBPtr;
	outBPtr = outputPtr;
	const AliHLTComponentBlockData* iter = 0;
	unsigned long ndx;

	UInt_t specification = 0;
	AliHLTCaloChannelDataHeaderStruct* tmpChannelData = 0;

	for( ndx = 0; ndx < evtData.fBlockCnt; ndx++ )
	{
		iter = blocks+ndx;
		//cout << " into do event and into to for loop !!! " << endl;
		//PrintComponentDataTypeInfo(iter->fDataType);

		if(iter->fDataType != AliHLTEMCALDefinitions::fgkChannelDataType)
		{
			HLTDebug("Data block is not of type fgkChannelDataType");

			continue;
		}

		specification |= iter->fSpecification;
		tmpChannelData = reinterpret_cast<AliHLTCaloChannelDataHeaderStruct*>(iter->fPtr);

		//HLTWarning (" channel number %d",tmpChannelData->fNChannels);

		ret = fRawHistoMakerPtr->MakeHisto(tmpChannelData, iter, outputPtr, size);

		//if(ret == -1)
		//{
		//  HLTError("Trying to write over buffer size");
		//  return -ENOBUFS;
		//}
		//digitCount += ret;
	}

	fLocalEventCount++;

	// fRawHistoMakerPtr->Reset();

	TFile rootHistFile(fRootFileName,"recreate");

	fRawHistoMakerPtr->GetHistograms()->Write();

	if (fLocalEventCount%fPushFraction == 0) {
		cout << "pushback done at " << fLocalEventCount << " events " << endl;

		PushBack(fRawHistoMakerPtr->GetHistograms(), kAliHLTDataTypeTObjArray | kAliHLTDataOriginEMCAL , specification);
	}

	size = mysize;

	return 0;
}


int
AliHLTEMCALRawHistoMakerComponent::DoInit(int argc, const char** argv )
{
	//see header file for documentation

	fRawHistoMakerPtr = new AliHLTEMCALRawHistoMaker();

	for(int i = 0; i < argc; i++)
	{
		if(!strcmp("-roothistofilename", argv[i]))
		{
			fRootFileName = argv[i+1];

		}
		if(!strcmp("-pushfraction", argv[i]))
		{
			fPushFraction = atoi(argv[i+1]);
		}
	}

	// cout << rootFileName << endl;

	return 0;
}


AliHLTComponent*
AliHLTEMCALRawHistoMakerComponent::Spawn()
{
	//see header file for documentation
	return new AliHLTEMCALRawHistoMakerComponent();
}