from manimlib.imports import * class Surf(ThreeDScene): CONFIG = { "axes_config": { "x_min": 0, "x_max": 8, "y_min": 0, "y_max": 8, "z_min": 0, "z_max": 6, "a":2 ,"b": 6, "c":1 , "d":6, "axes_shift":-3*OUT + 5*LEFT, "x_axis_config": { "tick_frequency": 1, # "include_tip": False, }, "y_axis_config": { "tick_frequency": 1, # "include_tip": False, }, "z_axis_config": { "tick_frequency": 1, # "include_tip": False, }, "num_axis_pieces": 1, }, "default_graph_style": { "stroke_width": 2, "stroke_color": WHITE, }, "default_surface_config": { "fill_opacity": 0.5, "checkerboard_colors": [LIGHT_GREY], "stroke_width": 0.5, "stroke_color": WHITE, "stroke_opacity": 0.5, }, "Func": lambda x,y: 2+y/4+np.sin(x) } def construct(self): self.setup_axes() self.set_camera_orientation(distance=35, phi=80 * DEGREES, theta=-80 * DEGREES, ) fn_text=TextMobject("$S$").set_color(BLUE) self.add_fixed_in_frame_mobjects(fn_text) fn_text.to_edge(TOP,buff=MED_SMALL_BUFF) R=TextMobject("D").set_color(BLACK).scale(3) R.move_to(self.axes.input_plane,IN) self.add(R) #get the surface surface= self.get_surface( self.axes, lambda x , y: self.Func(x,y) ) surface.set_style( fill_opacity=0.8, fill_color=YELLOW, stroke_width=0.8, stroke_color=WHITE, ) self.begin_ambient_camera_rotation(rate=0.05) self.play(Write(surface)) # self.play(LaggedStart(ShowCreation(surface))) self.get_lines() # self.play(FadeIn(self.axes.input_plane)) self.wait(2) self.stop_ambient_camera_rotation() self.wait(1) def get_surface(self,axes, func, **kwargs): config = { "u_min": axes.c, "u_max": axes.d, "v_min": axes.a, "v_max": axes.b, "resolution": ( (axes.y_max - axes.y_min) // axes.y_axis.tick_frequency, (axes.x_max - axes.x_min) // axes.x_axis.tick_frequency, ), } config.update(self.default_surface_config) config.update(kwargs) return ParametricSurface( lambda x,y : axes.c2p( x, y, func(x, y) ), **config ) def get_lines(self): axes = self.axes labels=[axes.x_axis.n2p(axes.a), axes.x_axis.n2p(axes.b), axes.y_axis.n2p(axes.c), axes.y_axis.n2p(axes.d)] surface_corners=[] for x,y,z in self.region_corners: surface_corners.append([x,y,self.Func(x,y)]) lines=VGroup() for start , end in zip(surface_corners, self.region_corners): lines.add(self.draw_lines(start,end,"WHITE")) for start , end in zip(labels, self.region_corners): # lines.add(self.draw_lines(start,end,"BLUE")) # print (start,end) pass self.play(ShowCreation(lines)) def draw_lines(self,start,end,color): start=self.axes.c2p(*start) end=self.axes.c2p(*end) line=DashedLine(start,end,color=color) return line def get_three_d_axes(self, include_labels=True, include_numbers=True, **kwargs): config = dict(self.axes_config) config.update(kwargs) axes = ThreeDAxes(**config) axes.set_stroke(width=2) if include_numbers: self.add_axes_numbers(axes) if include_labels: self.add_axes_labels(axes) # Adjust axis orientation axes.x_axis.rotate( 90 * DEGREES, RIGHT, about_point=axes.c2p(0, 0, 0), ) axes.y_axis.rotate( 90 * DEGREES, UP, about_point=axes.c2p(0, 0, 0), ) # Add xy-plane input_plane = self.get_surface( axes, lambda x, t: 1e-5 ) input_plane.set_style( fill_opacity=0.5, fill_color=TEAL, stroke_width=0, stroke_color=WHITE, ) axes.input_plane = input_plane self.region_corners=[ input_plane.get_corner(pos) for pos in (DL,DR,UR,UL)] return axes def setup_axes(self): axes = self.get_three_d_axes(include_labels=True) axes.add(axes.input_plane) axes.scale(1) # axes.center() axes.shift(axes.axes_shift) self.add(axes) self.axes = axes def add_axes_numbers(self, axes): x_axis = axes.x_axis y_axis = axes.y_axis tex_vals_x = [ ("a", axes.a), ("b", axes.b), ] tex_vals_y=[ ("c", axes.c), ("d", axes.d) ] x_labels = VGroup() y_labels = VGroup() for tex, val in tex_vals_x: label = TexMobject(tex) label.scale(1) label.next_to(x_axis.n2p(val), DOWN) x_labels.add(label) x_axis.add(x_labels) x_axis.numbers = x_labels for tex, val in tex_vals_y: label = TexMobject(tex) label.scale(1.5) label.next_to(y_axis.n2p(val), LEFT) label.rotate(90 * DEGREES) y_labels.add(label) y_axis.add(y_labels) y_axis.numbers = y_labels return axes def add_axes_labels(self, axes): x_label = TexMobject("X") x_label.next_to(axes.x_axis.get_end(), RIGHT) axes.x_axis.label = x_label y_label = TextMobject("Y") y_label.rotate(90 * DEGREES, OUT) y_label.next_to(axes.y_axis.get_end(), UP) axes.y_axis.label = y_label z_label = TextMobject("Z") z_label.rotate(90 * DEGREES, RIGHT) z_label.next_to(axes.z_axis.get_zenith(), RIGHT) axes.z_axis.label = z_label for axis in axes: axis.add(axis.label) return axes ######Code_by_Somnath_Pandit_https://github.com/panditsomnath10016git#########