Index: /trunk/MagicSoft/Mars/mtemp/mifae/programs/controlPlot.cc =================================================================== --- /trunk/MagicSoft/Mars/mtemp/mifae/programs/controlPlot.cc (revision 4395) +++ /trunk/MagicSoft/Mars/mtemp/mifae/programs/controlPlot.cc (revision 4395) @@ -0,0 +1,346 @@ +//////////////////////////////////////////////////////////////////////////////////// +// +// _____ Control Plot program_____ +// +// Take as input root files with hillas parameters and generates several +// control plots. +// Output is a file with : +// - Number of hits plot +// +// Ester Aliu +// Oscar Blanch +// Javier Rico +//////////////////////////////////////////////////////////////////////////////////// + +#include +#include + +#include "TString.h" + +#include "MHillasSrcCalc.h" +#include "MHillasSrc.h" +#include "MSrcRotate.h" +#include "MSrcPosFromFile.h" +#include "MSrcTranslate.h" +#include "MParList.h" +#include "MTaskList.h" +#include "MHillas.h" +#include "MReadTree.h" +#include "MEvtLoop.h" +#include "MLog.h" +#include "MArgs.h" +#include "MWriteRootFile.h" +#include "MTime.h" +#include "MControlPlots.h" +#include "MIslands.h" +#include "MContinue.h" +#include "MF.h" + +using namespace std; + +Bool_t readDatacards(TString& filename); +void controlPlot(); + +//----------------------------------------------------------------------------- +// declaration of variables read from datacards +//----------------------------------------------------------------------------- + +TString inputFile; +TString offFile; +TString outputFile; +TString filter; +TString sizecut; +TString distcut; +TString widthcut; +TString lengthcut; +TString centercut; +ULong_t nmaxevents=999999999; +Float_t xsrcpos=0.; +Float_t ysrcpos=0.; +Double_t mjdpos=0; + +//----------------------------------------------------------------------------- +// constants +//----------------------------------------------------------------------------- + +const TString defaultcard="srcpos.datacard"; +const Float_t conver = 189./0.6; // conversion factor degrees to mm + +//----------------------------------------------------------------------------- + +static void Usage() +{ + gLog <" << endl << endl; + gLog << " : datacards file name (dafault " << defaultcard <<")" << endl; + gLog << " -?/-h: This help" << endl << endl; +} + +//----------------------------------------------------------------------------- +int main(int argc, char **argv) +{ + // evaluate arguments + MArgs arg(argc, argv); + if (arg.HasOnly("-?") || arg.HasOnly("-h") || arg.HasOnly("--help")) + { + Usage(); + return -1; + } + + TString datacard = arg.GetArgumentStr(0); + if(!datacard.Length()) + datacard = defaultcard; + + if(!readDatacards(datacard)) + { + cout << "Error reading datacards. Stoping" << endl; + return -1; + } + + controlPlot(); +} + +//----------------------------------------------------------------------------- +void controlPlot() +{ + // containers + MParList plist; + MTaskList tlist; + MIslands islands; + + // include containers in parameter list + plist.AddToList(&tlist); + plist.AddToList(&islands); + + MF cut(filter); + MF cutS(sizecut); + MF cutD(distcut); + MF cutL(lengthcut); + MF cutW(widthcut); + MF cutC(centercut); + + // tasks + MReadTree read("Parameters", inputFile); + read.DisableAutoScheme(); + + MControlPlots controlplots(outputFile); + controlplots.SetProduceFile(kFALSE); + + MContinue applycutS(&cutS); + applycutS.SetInverted(kTRUE); + + MContinue applycutD(&cutD); + applycutD.SetInverted(kTRUE); + + MContinue applycutL(&cutL); + applycutL.SetInverted(kTRUE); + + MContinue applycutW(&cutW); + applycutW.SetInverted(kTRUE); + + MContinue applycutC(&cutC); + applycutC.SetInverted(kTRUE); + + MContinue applycut(&cut); + applycut.SetInverted(kTRUE); + + + // include tasks in task list + tlist.AddToList(&read); + if(sizecut.Length()) + tlist.AddToList(&applycutS); + if(distcut.Length()) + tlist.AddToList(&applycutD); + if(lengthcut.Length()) + tlist.AddToList(&applycutL); + if(widthcut.Length()) + tlist.AddToList(&applycutW); + if(centercut.Length()) + tlist.AddToList(&applycutC); + if(filter.Length()) + tlist.AddToList(&applycut); + tlist.AddToList(&controlplots); + + // Eventloop + MEvtLoop evtloop; + evtloop.SetParList(&plist); + if (!evtloop.Eventloop(nmaxevents)) + return; + + tlist.PrintStatistics(); + + // do off-data if input file was specified + if(!offFile.Length()) + return; + + MReadTree read2("Parameters", offFile); + read2.DisableAutoScheme(); + tlist.AddToListBefore(&read2, &read, "All"); + tlist.RemoveFromList(&read); + + controlplots.SetMode(MControlPlots::kOff); + controlplots.SetProduceFile(kTRUE); + + if (!evtloop.Eventloop(nmaxevents)) + return; + + tlist.PrintStatistics(); +} +//----------------------------------------------------------------------- + +Bool_t readDatacards(TString& filename) +{ + ifstream ifun(filename.Data()); + if(!ifun) + { + cout << "File " << filename << " not found" << endl; + return kFALSE; + } + + TString word; + + while(ifun >> word) + { + // skip comments + if(word[0]=='/' && word[1]=='/') + { + while(ifun.get()!='\n'); // skip line + continue; + } + + // number of events + if(strcmp(word.Data(),"NEVENTS")==0) + ifun >> nmaxevents; + + // input file name + if(strcmp(word.Data(),"ONFILES")==0) + { + if(inputFile.Length()) + cout << "readDataCards Warning: overriding on-data file name" << endl; + ifun >> inputFile; + } + + // off-data file name + if(strcmp(word.Data(),"OFFFILES")==0) + { + if(offFile.Length()) + cout << "readDataCards Warning: overriding off-data file name" << endl; + ifun >> offFile; + } + + // output file name + if(strcmp(word.Data(),"HITSFILE")==0) + { + if(outputFile.Length()) + cout << "readDataCards Warning: overriding output file name" << endl; + ifun >> outputFile; + } + // exclusion cut + if(strcmp(word.Data(),"FILTER")==0) + { + if(filter.Length()) + cout << "readDataCards Warning: overriding existing cut" << endl; + + char ch; + while((ch=ifun.get())!='\n') + filter.Append(ch); + } + if(strcmp(word.Data(),"SIZECUT")==0) + { + if(sizecut.Length()) + cout << "readDataCards Warning: overriding existing size cut" << endl; + char ch[10]; + sizecut="(MHillas.fSize>"; + ifun >> ch; + sizecut+=ch; + sizecut+=") && (MHillas.fSize<"; + ifun >> ch; + sizecut+=ch; + sizecut+=")"; + } + if(strcmp(word.Data(),"DISTCUT")==0) + { + if(distcut.Length()) + cout << "readDataCards Warning: overriding existing DIST cut" << endl; + char ch[10]; + distcut="(MHillasSrc.fDist>"; + ifun >> ch; + distcut+=ch; + distcut+=") && (MHillasSrc.fDist<"; + ifun >> ch; + distcut+=ch; + distcut+=")"; + } + if(strcmp(word.Data(),"WIDTHCUT")==0) + { + if(widthcut.Length()) + cout << "readDataCards Warning: overriding existing width cut" << endl; + char ch[10]; + widthcut="(MHillas.fWidth>"; + ifun >> ch; + widthcut+=ch; + widthcut+=") && (MHillas.fWidth<"; + ifun >> ch; + widthcut+=ch; + widthcut+=")"; + } + if(strcmp(word.Data(),"LENGTHCUT")==0) + { + if(lengthcut.Length()) + cout << "readDataCards Warning: overriding existing length cut" << endl; + char ch[10]; + lengthcut="(MHillas.fLength>"; + ifun >> ch; + lengthcut+=ch; + lengthcut+=") && (MHillas.fLength<"; + ifun >> ch; + lengthcut+=ch; + lengthcut+=")"; + } + if(strcmp(word.Data(),"CENTERCUT")==0) + { + if(centercut.Length()) + cout << "readDataCards Warning: overriding existing center cut" << endl; + char ch[10]; + centercut="sqrt(MHillas.fMeanX*MHillas.fMeanX+MHillas.fMeanY*MHillas.fMeanY) < 0.00317*"; + ifun >> ch; + centercut+=ch; + } + } + + // check compulsory values + if(!inputFile.Length()) + { + cout << "No on-data file name specified" << endl; + return kFALSE; + } + + if(!offFile.Length()) + { + cout << "No off-data file name specified" << endl; + return kFALSE; + } + + if(!outputFile.Length()) + { + cout << "No output file name specified" << endl; + return kFALSE; + } + + MTime thetime(mjdpos); + + // Dump read values + cout << "************************************************" << endl; + cout << "* Datacards read from file " << filename << endl; + cout << "************************************************" << endl; + cout << "Maximum number of input events: " << nmaxevents << endl; + cout << "Input file name(s): " << inputFile << endl; + cout << "Input file name(s) for off data: " << offFile << endl; + cout << "Output file name for # hits plot : " << outputFile << endl; + if(filter.Length()) + cout << "Applying rejection cut: " << filter << endl; + cout << "***********" << endl << endl; + + return kTRUE; +} Index: /trunk/MagicSoft/Mars/mtemp/mifae/programs/controlplot.datacard =================================================================== --- /trunk/MagicSoft/Mars/mtemp/mifae/programs/controlplot.datacard (revision 4395) +++ /trunk/MagicSoft/Mars/mtemp/mifae/programs/controlplot.datacard (revision 4395) @@ -0,0 +1,38 @@ + +// Maximun number of on and off events to be processed) +NEVENTS 99999999 + +// On-data file name pattern +ONFILES ./srcPosPrueba.root + +// Off-data file name pattern +OFFFILES ./srcPosOffPrueba.root + +// output file name +HITSFILE ./rotateprueba2.ps + +////////// +// CUTS // +////////// + +// Size cut (lower and upper) in # of photons +SIZECUT 2000 9999999 + +// Dist cut (lower and upper) in degrees +DISTCUT 0.2 1.1 + +// Width cut (lower and upper) in degrees +WIDTHCUT 0 0.12 + +// Length cut (lower and upper) in degrees +LENGTHCUT 0 0.26 + +// maxX and maxY upper cut in degrees +CENTERCUT 1.1 + +// Other Additional cut +// Selection cut. +// Condition "==" not supported. +// Enclose all conditions between brakets, like: "(x50)