@@ -523,6 +523,36 @@ <h1>Source code for ete4.smartview.faces</h1><div class="highlight"><pre>
523523 < span class ="k "> return</ span > < span class ="n "> graphics</ span > < span class ="p "> ,</ span > < span class ="n "> size</ span > </ div >
524524</ div >
525525
526+
527+
528+ < div class ="viewcode-block " id ="LegendFace ">
529+ < a class ="viewcode-back " href ="../../../reference/reference_smartview.html#ete4.smartview.faces.LegendFace "> [docs]</ a >
530+ < span class ="k "> class</ span > < span class ="w "> </ span > < span class ="nc "> LegendFace</ span > < span class ="p "> (</ span > < span class ="n "> Face</ span > < span class ="p "> ):</ span >
531+ < span class ="w "> </ span > < span class ="sd "> """A legend with information about the data we are visualizing."""</ span >
532+
533+ < div class ="viewcode-block " id ="LegendFace.__init__ ">
534+ < a class ="viewcode-back " href ="../../../reference/reference_smartview.html#ete4.smartview.faces.LegendFace.__init__ "> [docs]</ a >
535+ < span class ="k "> def</ span > < span class ="w "> </ span > < span class ="fm "> __init__</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> title</ span > < span class ="p "> ,</ span > < span class ="n "> variable</ span > < span class ="p "> ,</ span >
536+ < span class ="n "> colormap</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> value_range</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> color_range</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ):</ span >
537+ < span class ="c1 "> # Do some very basic consistency checks first.</ span >
538+ < span class ="k "> if</ span > < span class ="n "> variable</ span > < span class ="o "> ==</ span > < span class ="s1 "> 'discrete'</ span > < span class ="p "> :</ span >
539+ < span class ="k "> assert</ span > < span class ="n "> colormap</ span > < span class ="ow "> and</ span > < span class ="n "> value_range</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="ow "> and</ span > < span class ="n "> color_range</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > \
540+ < span class ="s1 "> 'discrete variable needs a colormap (and no more)'</ span >
541+ < span class ="k "> elif</ span > < span class ="n "> variable</ span > < span class ="o "> ==</ span > < span class ="s1 "> 'continuous'</ span > < span class ="p "> :</ span >
542+ < span class ="k "> assert</ span > < span class ="n "> value_range</ span > < span class ="ow "> and</ span > < span class ="n "> color_range</ span > < span class ="ow "> and</ span > < span class ="n "> colormap</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > \
543+ < span class ="s1 "> 'continuous variable needs value and color ranges (and no more)'</ span >
544+ < span class ="k "> else</ span > < span class ="p "> :</ span >
545+ < span class ="k "> raise</ span > < span class ="ne "> ValueError</ span > < span class ="p "> (</ span > < span class ="sa "> f</ span > < span class ="s1 "> 'invalid variable value: </ span > < span class ="si "> {</ span > < span class ="n "> variable</ span > < span class ="si "> }</ span > < span class ="s1 "> '</ span > < span class ="p "> )</ span >
546+
547+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> title</ span > < span class ="o "> =</ span > < span class ="n "> title</ span >
548+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> variable</ span > < span class ="o "> =</ span > < span class ="n "> variable</ span > < span class ="c1 "> # can be "discrete" or "continuous"</ span >
549+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> colormap</ span > < span class ="o "> =</ span > < span class ="n "> colormap</ span > < span class ="c1 "> # dict {name: color}</ span >
550+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> value_range</ span > < span class ="o "> =</ span > < span class ="n "> value_range</ span > < span class ="c1 "> # (min, max)</ span >
551+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> color_range</ span > < span class ="o "> =</ span > < span class ="n "> color_range</ span > < span class ="c1 "> # (min, max)</ span > </ div >
552+ </ div >
553+
554+
555+ < span class ="c1 "> # NOTE: We don't need a special draw() function, we use the info directly.</ span >
526556</ pre > </ div >
527557
528558 </ div >
0 commit comments