a |=ic% @sddlZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZmZmZgdZeeeee dZd d Zdd ddddddd e jdd ddZdS)N) ContourGeneratorFillTypeLineTypeMpl2005ContourGeneratorMpl2014ContourGeneratorSerialContourGeneratorThreadedContourGeneratorZInterp max_threads) __version__)calc_chunk_sizes) as_fill_type as_line_type as_z_interp) r contour_generatorr rrrrrrr r )mpl2005mpl2014serialZthreadedcCsTtjj|tjd}tjj|dd}tj|rInput z must be at least a (2, 2) shaped array, but has shape zNumber of dimensions of x (z ) and y (z) do not matchz Length of x (z%) must match number of columns in z ()z Length of y (z") must match number of rows in z (z Shapes of x z and z z do not matchz Shapes of y z+Inputs x and y must be None, 1D or 2D, not Nz=If mask is set it must be a 2D array with the same shape as zz%Unrecognised contour generator name: z4 contour generator does not support corner_mask=Truez. contour generator does not support line_type z. contour generator does not support fill_type z4 contour generator does not support quad_as_tri=Truez- contour generator does not support z_interp )rrz1 contour generator does not support thread_count ) x_chunk_size y_chunk_size)rrr"r#r!r'r(r))rrrrndim TypeErrorshapeZarangeZmeshgridlen ValueError _class_lookupr Zsupports_corner_maskZdefault_line_typerZsupports_line_typeZdefault_fill_typerZsupports_fill_typeZsupports_quad_as_trir LinearrZsupports_z_interpZsupports_threads)xyrr r!r"r#r$r%r&r'r(r)rnynxr.r-clsargskwargsZcont_genrrrr0s=                      r)NNN)numpyrZ _contourpyrrrrrrr r r _versionr chunkr Z enum_utilrrr__all__dictr4rr5rrrrrs ,