Blame view

egs/aishell/v1/local/download_and_untar.sh 2.42 KB
8dcb6dfcb   Yannick Estève   first commit
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
  #!/bin/bash
  
  # Copyright   2014  Johns Hopkins University (author: Daniel Povey)
  #             2017  Xingyu Na
  # Apache 2.0
  
  remove_archive=false
  
  if [ "$1" == --remove-archive ]; then
    remove_archive=true
    shift
  fi
  
  if [ $# -ne 3 ]; then
    echo "Usage: $0 [--remove-archive] <data-base> <url-base> <corpus-part>"
    echo "e.g.: $0 /export/a05/xna/data www.openslr.org/resources/33 data_aishell"
    echo "With --remove-archive it will remove the archive after successfully un-tarring it."
    echo "<corpus-part> can be one of: data_aishell, resource_aishell."
  fi
  
  data=$1
  url=$2
  part=$3
  
  if [ ! -d "$data" ]; then
    echo "$0: no such directory $data"
    exit 1;
  fi
  
  part_ok=false
  list="data_aishell resource_aishell"
  for x in $list; do
    if [ "$part" == $x ]; then part_ok=true; fi
  done
  if ! $part_ok; then
    echo "$0: expected <corpus-part> to be one of $list, but got '$part'"
    exit 1;
  fi
  
  if [ -z "$url" ]; then
    echo "$0: empty URL base."
    exit 1;
  fi
  
  if [ -f $data/$part/.complete ]; then
    echo "$0: data part $part was already successfully extracted, nothing to do."
    exit 0;
  fi
  
  # sizes of the archive files in bytes.
  sizes="15582913665 1246920"
  
  if [ -f $data/$part.tgz ]; then
    size=$(/bin/ls -l $data/$part.tgz | awk '{print $5}')
    size_ok=false
    for s in $sizes; do if [ $s == $size ]; then size_ok=true; fi; done
    if ! $size_ok; then
      echo "$0: removing existing file $data/$part.tgz because its size in bytes $size"
      echo "does not equal the size of one of the archives."
      rm $data/$part.gz
    else
      echo "$data/$part.tgz exists and appears to be complete."
    fi
  fi
  
  if [ ! -f $data/$part.tgz ]; then
    if ! which wget >/dev/null; then
      echo "$0: wget is not installed."
      exit 1;
    fi
    full_url=$url/$part.tgz
    echo "$0: downloading data from $full_url.  This may take some time, please be patient."
  
    cd $data
    if ! wget --no-check-certificate $full_url; then
      echo "$0: error executing wget $full_url"
      exit 1;
    fi
  fi
  
  cd $data
  
  if ! tar -xvzf $part.tgz; then
    echo "$0: error un-tarring archive $data/$part.tgz"
    exit 1;
  fi
  
  touch $data/$part/.complete
  
  if [ $part == "data_aishell" ]; then
    cd $data/$part/wav
    for wav in ./*.tar.gz; do
      echo "Extracting wav from $wav"
      tar -zxf $wav && rm $wav
    done
  fi
  
  echo "$0: Successfully downloaded and un-tarred $data/$part.tgz"
  
  if $remove_archive; then
    echo "$0: removing $data/$part.tgz file since --remove-archive option was supplied."
    rm $data/$part.tgz
  fi
  
  exit 0;