|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#!/usr/bin/env perl
# R. Mankel, DESY Hamburg 06-Jul-2007
# A. Parenti, DESY Hamburg 27-Mar-2008
# $Revision: 1.3 $ by $Author$
# $Date: 2009/01/07 18:27:49 $
#
# Extract INFI directives from input file
# and write chunk to standard output according to
# specified splitting scheme.
# Lines starting with '#' (preceeded by any number of whitespace)
# will be ignored.
# A first line starting with 'CastorPool=' will always be output as first line.
#
# Usage:
#
# mps_split.pl infile thisChunk numberOfChunks
#
use POSIX;
$inFile = "undefined";
$thisGen = 1;
$totalGen = 0;
# parse the arguments
while (@ARGV) {
$arg = shift(@ARGV);
if ($arg =~ /\A-/) { # check for option
if ($arg =~ "h") {
$helpwanted = 1;
}
elsif ($arg =~ "d") {
$localdir = 1;
}
elsif ($arg =~ "u") {
$updateDb = 1;
}
$optionstring = "$optionstring$arg";
}
else { # parameters not related to options
$i = $i + 1;
if ($i eq 1) {
$inFile = $arg;
}
elsif ($i eq 2) {
$thisGen = $arg;
}
elsif ($i eq 3) {
$totalGen = $arg;
}
}
}
if ($totalGen <= 0) {
print "Insufficient information given\n";
exit 1;
}
if ($thisGen <= 0) {
print "mps_split: zero thisGen --> exit\n";
exit 1;
}
if ($thisGen > $totalGen) {
print "mps_split: thisGen gt totalGen, $thisGen gt $totalGen --> exit\n";
exit 1;
}
# open the input file
open INFILE,"$inFile";
undef $/; # undefining the INPUT-RECORD_SEPARATOR means slurp whole file
$body = <INFILE>; # read whole file
close INFILE;
$/ = "\n"; # back to normal
# split the input into lines
@LINESALL = split "\n",$body;
my @LINES;
foreach my $line (@LINESALL) {
unless ($line =~ /^\s*#/ ) { # exclude lines with comments, i.e.
push (@LINES, $line); # marked with '#' (& ignore whitespace before)
}
}
# how many INFI cards
$ninfi = $#LINES + 1;
my $poolDef = "";
if ($LINES[0] =~ /^CastorPool=/) { # starts with CastorPool
$poolDef = $LINES[0];
$ninfi = $ninfi - 1; # so we have one line less
}
# Now for the arithmetics
$chunkSize = $ninfi / $totalGen;
$startId =floor( ($thisGen - 1) * $chunkSize ) + 1;
$endId = floor( $thisGen * $chunkSize);
if ($thisGen == 1) {
$startId = 1;
}
if ($thisGen == $totalGen) {
$endId = $ninfi;
}
if ($poolDef ne "") { #first line was pool definition, so shift again
$startId = $startId + 1;
$endId = $endId + 1;
}
# print "chunkSize $chunkSize\n ninfi $ninfi startID $startId endId $endId\n";
if ($endId < $startId) {
print "Invalid interval, startId=$startId, endID=$endId\n";
print "ninfi was $#infi\n";
exit 2;
}
print "$poolDef\n" if ($poolDef ne ""); #GF
# prepare the split output
for ($i = $startId; $i <= $endId; ++$i) {
print "@LINES[$i-1]\n";
}
|