GeneratorInterface/EvtGenInterface/test/ConvertDataCardToPython.pl - master - githubmirror/cmssw

GeneratorInterface/EvtGenInterface/test/ConvertDataCardToPython.pl

70 строк2 KB

inugent

Adding Interface for EvtGen 1.3.0

12 июн 2014, 01:32

12 июн 2014, 01:3284425db

#! /usr/bin/perl

if (@ARGV < 4) {
  print<<HEAD

  This code converts a DataCard to a python fragment for DataCardFileWriter.

  Syntax:

  ConvertDataCardToPython {inputfile} {outputfile} {fragmentname} {datacardname} 

    inputfile    - The name of the input file to be converted to a python fragment for DataCardFileWriter
    outputfile   - The name of the output file  that is to contain DataCardFileWriter python fragment
    fragmentname - The name of the python fragment
    datacardname - The name of the datacard to be written out by DataCardFileWriter

  Example [from within GeneratorInterface/EvtGenInterface/python/]:
    ../test/ConvertDataCardToPython.pl ../data/DECAY_2010.DEC test_cff.py mywriter DECAY_2010.DEC

HEAD
;
}
else {
    $inputfile=$ARGV[0];
    $outputfile=$ARGV[1];
    $fragmentname=$ARGV[2];
    $datacardname=$ARGV[3];

    if(-e $outputfile){
	printf("Output File already exists: $outputfile \nRemove it and try again...");
	exit();
    }

    @DataSets;
    open(InDAT, $inputfile) || die("Could not open file $inputfile!\n");
    while ($item = <InDAT>) {
	chomp($item);
	push(@DataSets,$item);
    }
    close(InDAT);
    
    open(OutDAT, '>>', $outputfile) || die("Could not open file $inputfile!");
    print {OutDAT} "import FWCore.ParameterSet.Config as cms\n";
    print {OutDAT} "$fragmentname = cms.EDAnalyzer(\"DataCardFileWriter\",\n";
    print {OutDAT} "\tFileName = cms.string(\"$datacardname\"),\n";
    print {OutDAT} "\tFileContent = cms.vstring()\n";
    print {OutDAT} "\t)\n";
    $i=0;
    $j=0;
    $nlines = @DataSets;

    foreach $DS (@DataSets){
	if($i==255){
	    $i=0;
	    print {OutDAT} "\t])\n";
	}
	if($i==0){
	    print {OutDAT} "$fragmentname.FileContent.extend([\n";
	}
        $i++;
	$j++;
        $comma=",";
        if($i==255 || $j==$nlines){ $comma="";}
        $DS =~ s/\\/\\\\\\\\/g;
	$DS =~ s/\"/\\"/g;
	print {OutDAT} "\t\"$DS\"$comma\n";
    }
    print {OutDAT} "\t])\n";
    close(OutDAT);
}