#!/opt/chipster/tools/Python-2.7.12/bin/python

"""
Read a file containing a 0 or 1 on each line (`feature_file`), output 
all lines from stdin for which that value was 1

TODO: no need to read the feature_file into memory here, just iterate in
      parallel.

usage: %prog feature_file < ...
"""
from __future__ import print_function

import sys

import psyco_full


def __main__():

    feature_file = sys.argv[1]

    if len( sys.argv ) > 2:
        match = int( sys.argv[2] )
    else:
        match = 1
    
    feature_vector = [ int( line ) for line in open( feature_file ) ]

    for index, line in enumerate( sys.stdin ):
        if feature_vector[ index ] == match: print(line, end='')

if __name__ == "__main__":
    __main__()