|
49 | 49 | from titiler.core.models.responses import MultiBaseStatisticsGeoJSON |
50 | 50 | from titiler.core.resources.enums import ImageType, OptionalHeader |
51 | 51 | from titiler.core.resources.responses import GeoJSONResponse |
52 | | -from titiler.core.utils import render_image |
53 | 52 | from titiler.xarray.io import Reader as XarrayReader |
54 | 53 |
|
55 | 54 | jinja2_env = jinja2.Environment( |
56 | | - loader=jinja2.ChoiceLoader([jinja2.PackageLoader(__package__, "templates")]) |
| 55 | + loader=jinja2.ChoiceLoader( |
| 56 | + [ |
| 57 | + jinja2.PackageLoader(__package__, "templates"), |
| 58 | + jinja2.PackageLoader("titiler.core", "templates"), |
| 59 | + ] |
| 60 | + ), |
57 | 61 | ) |
58 | 62 | DEFAULT_TEMPLATES = Jinja2Templates(env=jinja2_env) |
59 | 63 |
|
@@ -497,8 +501,6 @@ def tiles_endpoint( |
497 | 501 | rasterio_params=Depends(self.rasterio_dependency), |
498 | 502 | reader_params=Depends(self.reader_dependency), |
499 | 503 | post_process=Depends(self.process_dependency), |
500 | | - rescale=Depends(self.rescale_dependency), |
501 | | - color_formula=Depends(self.color_formula_dependency), |
502 | 504 | colormap=Depends(self.colormap_dependency), |
503 | 505 | render_params=Depends(self.render_dependency), |
504 | 506 | ) -> Response: |
@@ -538,20 +540,15 @@ def tiles_endpoint( |
538 | 540 | reproject_method=reproject_method, |
539 | 541 | **read_options, |
540 | 542 | ) |
| 543 | + dst_colormap = getattr(src_dst, "colormap", None) |
541 | 544 |
|
542 | 545 | if post_process: |
543 | 546 | image = post_process(image) |
544 | 547 |
|
545 | | - if rescale: |
546 | | - image.rescale(rescale) |
547 | | - |
548 | | - if color_formula: |
549 | | - image.apply_color_formula(color_formula) |
550 | | - |
551 | | - content, media_type = render_image( |
| 548 | + content, media_type = self.render_func( |
552 | 549 | image, |
553 | 550 | output_format=format, |
554 | | - colormap=colormap, |
| 551 | + colormap=colormap or dst_colormap, |
555 | 552 | **render_params.as_dict(), |
556 | 553 | ) |
557 | 554 |
|
@@ -595,8 +592,6 @@ def tilejson_endpoint( # type: ignore |
595 | 592 | rasterio_params=Depends(self.rasterio_dependency), |
596 | 593 | reader_params=Depends(self.reader_dependency), |
597 | 594 | post_process=Depends(available_algorithms.dependency), |
598 | | - rescale=Depends(self.rescale_dependency), |
599 | | - color_formula=Depends(self.color_formula_dependency), |
600 | 595 | colormap=Depends(self.colormap_dependency), |
601 | 596 | render_params=Depends(self.render_dependency), |
602 | 597 | ) -> Dict: |
@@ -679,8 +674,6 @@ def map_endpoint( # type: ignore |
679 | 674 | # Rendering Options |
680 | 675 | ################################################################### |
681 | 676 | post_process=Depends(self.process_dependency), |
682 | | - rescale=Depends(self.rescale_dependency), |
683 | | - color_formula=Depends(self.color_formula_dependency), |
684 | 677 | colormap=Depends(self.colormap_dependency), |
685 | 678 | render_params=Depends(self.render_dependency), |
686 | 679 | ) -> _TemplateResponse: |
@@ -711,14 +704,10 @@ def map_endpoint( # type: ignore |
711 | 704 | request=request, |
712 | 705 | name="map.html", |
713 | 706 | context={ |
| 707 | + "request": request, |
714 | 708 | "tilejson_endpoint": tilejson_url, |
715 | 709 | "tms": tms, |
716 | 710 | "resolutions": [matrix.cellSize for matrix in tms], |
717 | | - "template": { |
718 | | - "api_root": base_url, |
719 | | - "params": request.query_params, |
720 | | - "title": "Map", |
721 | | - }, |
722 | 711 | }, |
723 | 712 | media_type="text/html", |
724 | 713 | ) |
@@ -755,8 +744,6 @@ def bbox_image( |
755 | 744 | reader_params=Depends(self.reader_dependency), |
756 | 745 | post_process=Depends(self.process_dependency), |
757 | 746 | image_params=Depends(self.img_part_dependency), |
758 | | - rescale=Depends(self.rescale_dependency), |
759 | | - color_formula=Depends(self.color_formula_dependency), |
760 | 747 | colormap=Depends(self.colormap_dependency), |
761 | 748 | render_params=Depends(self.render_dependency), |
762 | 749 | ): |
@@ -794,13 +781,7 @@ def bbox_image( |
794 | 781 | if post_process: |
795 | 782 | image = post_process(image) |
796 | 783 |
|
797 | | - if rescale: |
798 | | - image.rescale(rescale) |
799 | | - |
800 | | - if color_formula: |
801 | | - image.apply_color_formula(color_formula) |
802 | | - |
803 | | - content, media_type = render_image( |
| 784 | + content, media_type = self.render_func( |
804 | 785 | image, |
805 | 786 | output_format=format, |
806 | 787 | colormap=colormap or dst_colormap, |
@@ -846,9 +827,7 @@ def feature_image( |
846 | 827 | zarr_params=Depends(self.zarr_dependency), |
847 | 828 | reader_params=Depends(self.reader_dependency), |
848 | 829 | post_process=Depends(self.process_dependency), |
849 | | - rescale=Depends(self.rescale_dependency), |
850 | 830 | image_params=Depends(self.img_part_dependency), |
851 | | - color_formula=Depends(self.color_formula_dependency), |
852 | 831 | colormap=Depends(self.colormap_dependency), |
853 | 832 | render_params=Depends(self.render_dependency), |
854 | 833 | ): |
@@ -882,20 +861,15 @@ def feature_image( |
882 | 861 | dst_crs=dst_crs, |
883 | 862 | **read_options, |
884 | 863 | ) |
| 864 | + dst_colormap = getattr(src_dst, "colormap", None) |
885 | 865 |
|
886 | 866 | if post_process: |
887 | 867 | image = post_process(image) |
888 | 868 |
|
889 | | - if rescale: |
890 | | - image.rescale(rescale) |
891 | | - |
892 | | - if color_formula: |
893 | | - image.apply_color_formula(color_formula) |
894 | | - |
895 | | - content, media_type = render_image( |
| 869 | + content, media_type = self.render_func( |
896 | 870 | image, |
897 | 871 | output_format=format, |
898 | | - colormap=colormap, |
| 872 | + colormap=colormap or dst_colormap, |
899 | 873 | **render_params.as_dict(), |
900 | 874 | ) |
901 | 875 |
|
|
0 commit comments