ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/OSUT3Analysis/AnaTools/scripts/mergeHists
(Generate patch)

Comparing UserCode/OSUT3Analysis/AnaTools/scripts/mergeHists (file contents):
Revision 1.13 by ahart, Thu Jan 3 21:43:31 2013 UTC vs.
Revision 1.23 by lantonel, Wed Apr 10 13:34:17 2013 UTC

# Line 20 | Line 20 | my %weights;
20   my @weights;
21   my $nGoodJobs = 0;
22   my $nBadJobs = 0;
23 + my $nIncompleteJobs = 0;
24   my $counting = 0;
25   my %exitCodes;
26   my %signals;
27 + my %partial;
28   my %crossSections;
29 + my %dirs;
30   my $integratedLuminosity = 10000;
31   $integratedLuminosity = $opt{"luminosity"} if $opt{"luminosity"};
32   my $cutFlow = "cutFlow";
# Line 43 | Line 46 | foreach my $file (@$files)
46          close (FILE);
47          my $fileContents = join ("", @fileContents);
48          $fileContents =~ s/\n/ /g;
49 +        $counting = 1;
50          if ($fileContents =~ m/return value/)
51            {
52              $fileContents =~ s/.*\(return value ([^)]*)\).*/$1/g;
53              $nGoodJobs++;
54              print "WARNING: Nonzero exit code for job $jobNumber. (return value $fileContents)\n" if $fileContents != 0;
55              $exitCodes{$dir}{$jobNumber} = $fileContents;
52            $counting = 1;
56            }
57 <        if ($fileContents =~ m/signal/)
57 >        elsif ($fileContents =~ m/signal/)
58            {
59              $fileContents =~ s/.*\(signal ([^)]*)\).*/$1/g;
60              $nBadJobs++;
61              print "WARNING: Skipping job $jobNumber. (signal $fileContents)\n";
62              $signals{$dir}{$jobNumber} = $fileContents;
63 <            $counting = 1;
63 >          }
64 >        else
65 >          {
66 >            $nIncompleteJobs++;
67 >            $partial{$dir}{$jobNumber} = 1;
68            }
69        }
70      if ($file =~ m/^.*\/crossSectionInPicobarn\.txt$/)
# Line 75 | Line 82 | foreach my $file (@$files)
82      my $dir = $file;
83      $dir =~ s/^(.*)\/[^\/]*$/$1/;
84      my $badJob = 0;
85 +    my $jobNumber;
86      if ($file =~ m/^.*_[^_]*\.root$/)
87        {
88 <        my $jobNumber = $file;
88 >        $jobNumber = $file;
89          $jobNumber =~ s/^.*_([^_]*)\.root$/$1/;
90 <        $badJob = defined $signals{$dir} && defined $signals{$dir}{$jobNumber};
90 >        $badJob = (defined $signals{$dir} && defined $signals{$dir}{$jobNumber}) || (defined $partial{$dir} && defined $partial{$dir}{$jobNumber});
91        }
92      next if $badJob;
93      if ($file =~ m/^.*\.root$/)
# Line 88 | Line 96 | foreach my $file (@$files)
96            {
97              if (substr ($file, 0, length ($arg)) eq $arg)
98                {
99 +                $dirs{$arg} = $dir;
100 +                if (countEvents ($file, $cutFlow) < 0)
101 +                  {
102 +                    $nGoodJobs--;
103 +                    $nBadJobs++;
104 +                    print "WARNING: Skipping job $jobNumber. (bad ROOT file)\n";
105 +                    last;
106 +                  }
107                  push (@{$rootFiles{$arg}}, $file);
108                  push (@weights, $opt{"weight"}) if $opt{"weight"};
109                  push (@weights, $opt{"xsection"} * $integratedLuminosity) if !$opt{"weight"} && $opt{"xsection"};
# Line 117 | Line 133 | foreach my $arg (@ARGV)
133      system ("cutFlowLimits $tmpName");
134      $nTotalEvents{$arg} = $count;
135      $nTotalEvents += $count;
136 <    push (@mergedFiles, "$tmpName");
137 <    push (@mergedWeights, $weights{$arg} / $count);
136 >    $weights{$arg} /= $count if !$opt{"weight"} && $opt{"xsection"};
137 >    $weights{$arg} /= $count if !$opt{"weight"} && !$opt{"xsection"} && defined $crossSections{$dirs{$arg}};
138 >    push (@mergedFiles, $tmpName);
139 >    push (@mergedWeights, $weights{$arg});
140    }
141   my $mergedFiles = join (" ", @mergedFiles);
142   my $mergedWeights = join (",", @mergedWeights);
# Line 127 | Line 145 | foreach my $mergedFile (@mergedFiles)
145    {
146      unlink ("$mergedFile");
147    }
148 < print "Weights:\n";
148 > print "=============================================\n";
149 > my $output = sprintf "Cross-section of samples:\n";
150 > my $printOutput = 0;
151 > foreach my $arg (keys %weights)
152 > {
153 >    my $shortArg = $arg;
154 >    $shortArg =~ s/^.*\/([^\/]*)$/$1/;
155 >    $output = sprintf "%s  $shortArg: %.5g pb\n", $output, ($crossSections{$dirs{$arg}});
156 >    $printOutput = 1 if $weights{$arg} != 1;
157 > }
158 > print $output if $printOutput;
159 >
160 > my $output = sprintf "Effective luminosities of samples:\n";
161 > my $printOutput = 0;
162 > foreach my $arg (keys %weights)
163 >  {
164 >    my $shortArg = $arg;
165 >    $shortArg =~ s/^.*\/([^\/]*)$/$1/;
166 >    $output = sprintf "%s  $shortArg: %.5g/fb\n", $output, ($integratedLuminosity / (1000.0 * $weights{$arg})) if $weights{$arg} != 1;
167 >    $printOutput = 1 if $weights{$arg} != 1;
168 >  }
169 > print $output if $printOutput;
170 >
171 > $output = sprintf "Weights for target luminosity of %g/fb:\n", ($integratedLuminosity / 1000.0);
172 > $printOutput = 0;
173   foreach my $arg (keys %weights)
174    {
175      my $shortArg = $arg;
176      $shortArg =~ s/^.*\/([^\/]*)$/$1/;
177 <    printf "  $shortArg: %.5g (%.5g pb)\n", $weights{$arg}, $weights{$arg} / $integratedLuminosity;
177 >    $output = sprintf "%s  $shortArg: %.5g\n", $output, $weights{$arg} if $weights{$arg} != 1;
178 >    $printOutput = 1 if $weights{$arg} != 1;
179    }
180 + print $output if $printOutput;
181 +
182   my $goodEvents = countEvents ("$opt{'prefix'}.root", $cutFlow);
183   print "$nGoodJobs jobs ran successfully over $nTotalEvents ($goodEvents weighted) events.\n" if $counting;
184   print "$nBadJobs jobs failed to run.\n" if $counting;
185 + print "$nIncompleteJobs jobs have not finished.\n" if $counting;
186 + print "=============================================\n";
187  
188   sub
189   processArgs
# Line 190 | Line 237 | printHelp
237    printf "%-29s%s\n", "  -c, --cutflow HISTOGRAM", "name of histogram to use for the cutflow (default:";
238    printf "%-29s%s\n", "                  ", "cutFlow)";
239    printf "%-29s%s\n", "  -h, --help", "print this help message";
240 <  printf "%-29s%s\n", "  -l, --luminosity", "integrated luminosity to which the histograms are";
194 <  printf "%-29s%s\n", "                  ", "weighted (default: 10000/pb)";
240 >  printf "%-29s%s\n", "  -l, --luminosity", "integrated luminosity to which the histograms are weighted";
241    printf "%-29s%s\n", "  -p, --prefix PREFIX", "output is  named PREFIX.root";
242    printf "%-29s%s\n", "  -w, --weight WEIGHT", "scale the output by WEIGHT, overriding the";
243    printf "%-29s%s\n", "                     ", "automatic weighting using the cross section from";
# Line 228 | Line 274 | countEvents
274    my $cutFlow = shift;
275  
276    my $output = `getEventsFromCutFlow $file $cutFlow`;
277 <  if ($output =~ m/Did not find a histogram named/)
277 >  if ($output =~ m/Did not find a histogram named/ || $output =~ m/appears to be empty/ || $output =~ m/Failed to open/)
278      {
279        print $output;
280        return -1;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines