docs: def assign_stream_id_raster(stream_file, subbasin_file, out_stream_file):
          """Assign stream link ID according to subbasin ID.
          Args:
              stream_file: input stream raster file
              subbasin_file: subbasin raster file
              out_stream_file: output stream raster file
          """
          stream_raster = RasterUtilClass.read_raster(stream_file)
          stream_data = stream_raster.data
          nrows = stream_raster.nRows
          ncols = stream_raster.nCols
          nodata = stream_raster.noDataValue
          subbain_data = RasterUtilClass.read_raster(subbasin_file).data
          nodata_array = ones((nrows, ncols)) * DEFAULT_NODATA
          newstream_data = where((stream_data > 0) & (stream_data != nodata),
                                 subbain_data, nodata_array)
          RasterUtilClass.write_gtiff_file(out_stream_file, nrows, ncols, newstream_data,
                                           stream_raster.geotrans, stream_raster.srs,
                                           DEFAULT_NODATA, GDT_Int16)