Updates

[u/mrichter/AliRoot.git] / HLT / RCU / AliHLTAltroTimebinAverageComponent.cxx

// $Id$

//**************************************************************************
//* This file is property of and copyright by the ALICE HLT Project        * 
//* ALICE Experiment at CERN, All rights reserved.                         *
//*                                                                        *
//* Primary Authors: Kalliopi Kanaki <Kalliopi.Kanaki@ift.uib.no>          *
//*                  Oystein Djuvsland
//*                  Matthias Richter                                      *
//*                  for The ALICE HLT Project.                            *
//*                                                                        *
//* 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.                  *
//**************************************************************************

/** @file   AliHLTAltroTimebinAverageComponent.cxx
    @author Kalliopi Kanaki, Oystein Djuvsland, Matthias Richter
    @date   26.08.2008
    @brief  
*/

#include "AliHLTAltroTimebinAverageComponent.h"
#include "AliHLTErrorGuard.h"
#include "AliAltroRawStreamV3.h"
#include "AliHLTAltroEncoder.h"
#include "AliRawReaderMemory.h"
#include "AliRawDataHeader.h"
#include <memory>

using namespace std;

/** ROOT macro for the implementation of ROOT specific class methods */
ClassImp(AliHLTAltroTimebinAverageComponent)

AliHLTAltroTimebinAverageComponent::AliHLTAltroTimebinAverageComponent()
    :
    fStartTimeBin(0),
    fEndTimeBin(1024),
    fNTimeBins(1024)
{
  // see header file for class documentation
  // or
  // refer to README to build package
  // or
  // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
}

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

// Public functions to implement AliHLTComponent's interface.
// These functions are required for the registration process

const char* AliHLTAltroTimebinAverageComponent::GetComponentID()
{
  // see header file for class documentation
  return "AltroTimebinAverager";
}

void AliHLTAltroTimebinAverageComponent::GetInputDataTypes( vector<AliHLTComponentDataType>& list)
{
  // see header file for class documentation
  list.clear(); 
  list.push_back( kAliHLTDataTypeDDLRaw | kAliHLTDataOriginTPC );
  list.push_back( kAliHLTDataTypeDDLRaw | kAliHLTDataOriginPHOS );
}

AliHLTComponentDataType AliHLTAltroTimebinAverageComponent::GetOutputDataType()
{
  // see header file for class documentation
  return kAliHLTDataTypeDDLRaw;
}

int AliHLTAltroTimebinAverageComponent::GetOutputDataTypes(AliHLTComponentDataTypeList& tgtList)
{
  // see header file for class documentation
  tgtList.clear();
  tgtList.push_back(kAliHLTDataTypeDDLRaw);
  return tgtList.size();
}

void AliHLTAltroTimebinAverageComponent::GetOutputDataSize( unsigned long& constBase, double& inputMultiplier )
{
  // see header file for class documentation
  constBase=0;
  inputMultiplier=1.0;
}

AliHLTComponent* AliHLTAltroTimebinAverageComponent::Spawn()
{
  // see header file for class documentation
  return new AliHLTAltroTimebinAverageComponent;
}
	
int AliHLTAltroTimebinAverageComponent::DoInit( int argc, const char** argv )
{
  // see header file for class documentation

  Int_t i = 0;
  Char_t* cpErr;

  while ( i < argc ) {      

    // -- number of timebins
    if ( !strcmp( argv[i], "ntimebins" ) ) {
      fNTimeBins = strtoul( argv[i+1], &cpErr ,0);
      if ( *cpErr ) {
	HLTError("Cannot convert ntimebins specifier '%s'.", argv[i+1]);
	return EINVAL;
      }
      i+=2;
      continue;
    }

    // -- first timebin
    if ( !strcmp( argv[i], "start-timebin" ) ) {
      fStartTimeBin = strtoul( argv[i+1], &cpErr ,0);
      if ( *cpErr ) {
	HLTError("Cannot convert start-timebin specifier '%s'.", argv[i+1]);
	return EINVAL;
      }
      i+=2;
      continue;
    }

    // -- last timebin
    if ( !strcmp( argv[i], "end-timebin" ) ) {
      if(strtoul( argv[i+1], &cpErr ,0)<=1024){
	fEndTimeBin = strtoul( argv[i+1], &cpErr ,0);
      }
      if ( *cpErr ) {
	HLTError("Cannot convert end-timebin specifier '%s'.", argv[i+1]);
	return EINVAL;
      }
      i+=2;
      continue;
    }

    HLTError("Unknown option '%s'", argv[i]);
    return -EINVAL;

  }

  return 0;
}

int AliHLTAltroTimebinAverageComponent::DoDeinit()
{
  // see header file for class documentation
  return 0;
}

int AliHLTAltroTimebinAverageComponent::DoEvent( const AliHLTComponentEventData& evtData, 
						const AliHLTComponentBlockData* blocks, 
						AliHLTComponentTriggerData& /*trigData*/, AliHLTUInt8_t* outputPtr, 
						AliHLTUInt32_t& size, 
						vector<AliHLTComponentBlockData>& outputBlocks )
{
  // see header file for class documentation
  int iResult=0;
  AliHLTUInt32_t capacity=size;
  size=0;
  AliHLTUInt32_t offset=0;

  const AliHLTComponentBlockData* iter = NULL;
  unsigned long ndx;

  std::auto_ptr<AliRawReaderMemory> pRawReader(new AliRawReaderMemory);
  if (pRawReader.get()) return -ENOMEM;

  for(ndx = 0; ndx < evtData.fBlockCnt; ndx++) {
    iter = blocks+ndx;
      
    if ( iter->fDataType != kAliHLTDataTypeDDLRaw) {
      continue;
    }

    static AliHLTErrorGuard required("AliHLTAltroTimebinAverageComponent", "DoEvent", "component commission required after major changes, need to extract equipment id from data specification");
    (++required).Throw(1);

    pRawReader->Reset();
    // FIXME: set ddl no
    if (!pRawReader->AddBuffer((UChar_t*)iter->fPtr,iter->fSize, 768)) {
      ALIHLTERRORGUARD(1, "can not set up AltroDecoder for data block %s 0x%08x,"
		       " skipping data block and suppressing further messages",
		       DataType2Text(iter->fDataType).c_str(), iter->fSpecification);
      continue;
    }

    std::auto_ptr<AliAltroRawStreamV3> altroRawStream(new AliAltroRawStreamV3(pRawReader.get()));
    std::auto_ptr<AliHLTAltroEncoder> altroEncoder(new AliHLTAltroEncoder);

    if (!altroRawStream.get() || !altroEncoder.get()) {
      iResult=-ENOMEM;
      break;
    }

    altroRawStream->Reset();
    if (!altroRawStream->NextDDL()) {
      ALIHLTERRORGUARD(1, "internal error, can not read data from AliRawReaderMemory");
      continue;
    }

    UChar_t *RCUTrailer=NULL;
    Int_t RCUTrailerSize=altroRawStream->GetRCUTrailerSize();
    if (RCUTrailerSize<=0 || !altroRawStream->GetRCUTrailerData(RCUTrailer) || RCUTrailer==NULL) {
      ALIHLTERRORGUARD(1, "can not find RCU trailer for data block %s 0x%08x: skipping data block",
		       DataType2Text(iter->fDataType).c_str(), iter->fSpecification);
      continue;
    }

    altroEncoder->SetBuffer(outputPtr+offset,capacity-offset);
    AliRawDataHeader cdh;
    altroEncoder->SetCDH((AliHLTUInt8_t*)iter->fPtr,sizeof(AliRawDataHeader));

    altroEncoder->SetRCUTrailer(RCUTrailer, RCUTrailerSize);

    while (iResult>=0 && altroRawStream->NextChannel()) {
      int hwadd=altroRawStream->GetHWAddress();

      while (iResult>=0 && altroRawStream->NextBunch()) {
	int bunchLength=altroRawStream->GetBunchLength();
	int time=altroRawStream->GetStartTimeBin();
	const  UShort_t* bunchData=altroRawStream->GetSignals();
	for (int bin=bunchLength && iResult>=0; bin>0; ) {
	  bin--;
	  if(bunchData[bin]>0){// disregarding 0 data.
	     
	    if(time+bin>=fStartTimeBin && time+bin<=fEndTimeBin){
	      AliHLTUInt16_t signal=bunchData[bin];
	      if (bin-1>=0) signal+=bunchData[bin-1];
	      altroEncoder->AddSignal((time+bin)/2,signal/2);
	      bin--;
	    } // end if between start and end time bin
	  } // end if bunchData[i]>0
	} // for loop
      } //while loop over bunches
      if (true/*condition deprecated but keep formatting*/) {
	altroEncoder->SetChannel(hwadd);
      }
    } // while loop over channels

    if (true/*condition deprecated but keep formatting*/) {
     int sizeOfData=altroEncoder->SetLength();
     
     if (sizeOfData<0) {
       HLTError("data encoding failed");
       return sizeOfData;
     }
     if(sizeOfData>(int)capacity){
       HLTWarning("Buffer too small to add the altrodata: %d of %d byte(s) already used", sizeOfData, size);
       return -ENOSPC;
     }
   
     AliHLTComponentBlockData bd;
     FillBlockData( bd );
     bd.fOffset = offset;
     bd.fSize = sizeOfData;
     bd.fDataType = iter->fDataType;
     bd.fSpecification = iter->fSpecification;     
     outputBlocks.push_back( bd );
     
     offset+=bd.fSize;
    }

  } // while over data blocks

  if (iResult>=0) size=offset;
  return iResult;
}