-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathadd_to_start_and_end_of_values_in_columns.pl
132 lines (109 loc) · 3.43 KB
/
add_to_start_and_end_of_values_in_columns.pl
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
124
125
126
127
128
129
130
131
132
#!/usr/bin/env perl
# Pads non-empty values in specified columns with parameter start and end text.
# Usage:
# perl add_to_start_and_end_of_values_in_columns.pl [table]
# "[text to add to start of each column value]" "[text to add to end of each column value]"
# "[title of column to search]" "[title of another column to search]"
# "[title of another column to search]" [etc.]
# Prints to console. To print to file, use
# perl add_to_start_and_end_of_values_in_columns.pl [table]
# "[text to add to start of each column value]" "[text to add to end of each column value]"
# "[title of column to search]" "[title of another column to search]"
# "[title of another column to search]" [etc.] > [output table path]
use strict;
use warnings;
my $table = $ARGV[0];
my $start_text = $ARGV[1];
my $end_text = $ARGV[2];
my @titles_of_columns_to_add_to = @ARGV[3..$#ARGV];
my $NEWLINE = "\n";
my $DELIMITER = "\t";
# verifies that input file exists and is not empty
if(!$table or !-e $table or -z $table)
{
print STDERR "Error: table not provided, does not exist, or empty:\n\t"
.$table."\nExiting.\n";
die;
}
# verifies that we have been provided column titles
if(!scalar @titles_of_columns_to_add_to)
{
print STDERR "Error: no column titles provided. Exiting.\n";
die;
}
# converts array of column titles to a hash
my %title_is_of_column_to_add_to = (); # key: column title -> value: 1 if column has dates
my %column_title_to_column = (); # key: included column title -> value: column
foreach my $column_title(@titles_of_columns_to_add_to)
{
$title_is_of_column_to_add_to{$column_title} = 1;
$column_title_to_column{$column_title} = -1;
}
# reads in and processes input table
my $first_line = 1;
my %column_is_column_to_add_to = (); # key: column (0-indexed) -> value: 1 if column has dates
open TABLE, "<$table" || die "Could not open $table to read; terminating =(\n";
while(<TABLE>) # for each row in the file
{
chomp;
my $line = $_;
if($line =~ /\S/) # if row not empty
{
my @items_in_line = split($DELIMITER, $line, -1);
if($first_line) # column titles
{
# identifies columns to add to
my $column = 0;
foreach my $column_title(@items_in_line)
{
if(defined $column_title and $title_is_of_column_to_add_to{$column_title})
{
$column_is_column_to_add_to{$column} = 1;
$column_title_to_column{$column_title} = $column;
}
$column++;
}
# verifies that we have found all columns to add to
foreach my $column_title(keys %column_title_to_column)
{
if($column_title_to_column{$column_title} == -1)
{
print STDERR "Error: expected column title ".$column_title
." not found in table ".$table."\nExiting.\n";
die;
}
}
# print header line as is
print $line.$NEWLINE;
$first_line = 0; # next line is not column titles
}
else # column values (not column titles)
{
# prints all values, padding non-empty values in columns to add to
my $column = 0;
foreach my $value(@items_in_line)
{
# prints delimiter
if($column > 0)
{
print $DELIMITER;
}
# pads values if this is a column to search and value is non-empty
if($column_is_column_to_add_to{$column}
and defined $value and length $value and $value =~ /\S/)
{
$value = $start_text.$value.$end_text;
}
# prints value
if(defined $value and length $value)
{
print $value;
}
$column++;
}
print $NEWLINE;
}
}
}
close TABLE;
# August 24, 2021