//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      pad-wise Digi amplitude correction 
//      based on gain calibration measurement
//      
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Felix Boehmer           TUM      (original author)  
//
//-----------------------------------------------------------

#ifndef TPCAMPCORRECTION_H
#define TPCAMPCORRECTION_H

//Base class
#include "FairTask.h"

#include "TClonesArray.h"
#include "TString.h"

#include <map>

// Collaborating Class Declarations --------------------------------------

class TpcDigi;
class TpcDigiPar;

class TpcDigiAmpCorrectionTask : public FairTask {

public:

// Constructor ----------------------------------------------

  TpcDigiAmpCorrectionTask();			//default constructor
  ~TpcDigiAmpCorrectionTask() {;}
  
  virtual InitStatus Init();	 
  virtual void Exec(Option_t* opt);
  
  virtual void SetParContainers();
  
// Setters --------------------------------------------------
  
  void SetPersistence(Bool_t opt=kTRUE) {fPersistence=opt;} 
  void SetDigiBranchName(TString d) {fDigiBranchName=d;}
  void SetExternalInput(TString file) {fAsciiInput=file; fInputSet=true; readFile();}
  void SetVerbose(Bool_t opt=kTRUE) {fVerbose=opt;}

private:

  void readFile();

  Bool_t fPersistence;
  Bool_t fInputSet;
  Bool_t fVerbose;

  TString fDigiBranchName;
  TString fAsciiInput;
  
  TClonesArray* fDigiArr;
  
  TpcDigiPar* fPar;
  
  std::map<unsigned int, double> fCorrMap;  // < padID, correction factor >


public:
  ClassDef(TpcDigiAmpCorrectionTask,2)
	
};

#endif